State Class Reference

Inheritance diagram for State:

Public Member Functions
def	__init__ (self, model=None, **kwargs)

Public Attributes
	model
	torch model
	collection_keys
	list of keys to save.

Detailed Description

torch state

Definition at line 18 of file torch.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			model = None,
		**	kwargs
	)

 Constructor of the state object 

Definition at line 23 of file torch.py.

    def __init__(self, model=None, **kwargs):
        """ Constructor of the state object """
        
        self.model = model
 
        
        self.collection_keys = []
 
        # other possible things to save that are needed by the model
        for key, value in kwargs.items():
            self.collection_keys.append(key)
            setattr(self, key, value)
 
 

Member Data Documentation

collection_keys

collection_keys

list of keys to save.

Any attribute in this collection is written to the weightfile and recovered during loading.

Definition at line 29 of file torch.py.

model

model

torch model

Definition at line 26 of file torch.py.

---

The documentation for this class was generated from the following file:

• mva/scripts/basf2_mva_python_interface/torch.py