MnistModel.register_module

MnistModel.register_module(name: str, module: Optional[Module]) → None

Alias for add_module().

previous

MnistModel.register_load_state_dict_post_hook

next

MnistModel.register_parameter