save_states

class braincore.save_states(target, **kwargs)

Save all states in the target as a dictionary for later disk serialization.

Parameters:

target (Module) – Module. The node to save its states.

Return type:

Dict

Returns:

Dict. The state dict for serialization.