Python KernelNorm Examples

Example #1

File: spectral_normalization.py Project: tallamjr/google-research

 def remove(module):
     KernelNorm.remove(module, SpectralNorm.NAME)

Example #2

 def remove(module):
     KernelNorm.remove(module, WeightNorm.NAME)

Example #3

File: spectral_normalization.py Project: tallamjr/google-research

 def add(module, norm=1):
     KernelNorm.add(module, SpectralNorm(norm))

Example #4

 def add(module, scale=True, axis=-1, epsilon=1e-5):
     KernelNorm.add(module, WeightNorm(scale, axis, epsilon))