Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.nn.utils.spectral_norm.SpectralNormStateDictHook Class Reference
Inheritance diagram for torch.nn.utils.spectral_norm.SpectralNormStateDictHook:

Public Member Functions

def __init__ (self, fn)
def __call__ (self, module, state_dict, prefix, local_metadata)

Public Attributes


Detailed Description

Definition at line 177 of file

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