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
	fn

Detailed Description

Definition at line 177 of file spectral_norm.py.

---

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

• torch/nn/utils/spectral_norm.py