MnistModel.set_extra_state

MnistModel.set_extra_state(state: Any)

This function is called from load_state_dict() to handle any extra state found within the state_dict. Implement this function and a corresponding get_extra_state() for your module if you need to store extra state within its state_dict.

Args:

state (dict): Extra state from the state_dict

previous

MnistModel.requires_grad_

next

MnistModel.share_memory