move sn to engine

util/misc.py

@@ -4,16 +4,6 @@ import pkgutil
 from pathlib import Path
 from typing import Optional
 
-import torch.nn as nn
-
-
-def add_spectral_norm(module):
-    if isinstance(module, (nn.Linear, nn.Conv2d, nn.ConvTranspose2d)) and not hasattr(module, 'weight_u'):
-        return nn.utils.spectral_norm(module)
-    else:
-        return module
-
-
 def import_submodules(package, recursive=True):
     """ Import all submodules of a module, recursively, including subpackages