tabensemb.model.AbstractNN._save_to_state_dict

method

AbstractNN._save_to_state_dict(destination, prefix, keep_vars)

Saves module state to destination dictionary, containing a state of the module, but not its descendants. This is called on every submodule in state_dict().

In rare cases, subclasses can achieve class-specific behavior by overriding this method with custom logic.

Parameters:

• destination (dict) – a dict where state will be stored

• prefix (str) – the prefix for parameters and buffers used in this module