def remove(module):
     KernelNorm.remove(module, SpectralNorm.NAME)
Example #2
0
 def remove(module):
     KernelNorm.remove(module, WeightNorm.NAME)