Caffe2 - Python API
A deep learning, cross platform ML framework
Classes | Functions
torch.nn.utils.spectral_norm Namespace Reference

Classes

class  SpectralNorm
 
class  SpectralNormLoadStateDictPreHook
 
class  SpectralNormStateDictHook
 

Functions

def spectral_norm (module, name='weight', n_power_iterations=1, eps=1e-12, dim=None)
 
def remove_spectral_norm (module, name='weight')
 

Detailed Description

Spectral Normalization from https://arxiv.org/abs/1802.05957