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