Ejemplos de KernelNorm.add en Python

Ejemplo n.º 1

Archivo: spectral_normalization.py Proyecto: tallamjr/google-research

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

Ejemplo n.º 2

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