Update base_runner.py

runners/base_runner.py

@@ -19,20 +19,6 @@
 from torch.nn.parallel import DistributedDataParallel as DDP
 
 
-def weights_init(mod):
-    """
-    Custom weights initialization called on netG, netD and netE
-    :param m:
-    :return:
-    """
-    classname = mod.__class__.__name__
-    if classname.find('Conv') != -1:
-        mod.weight.data.normal_(0.0, 0.02)
-    elif classname.find('BatchNorm') != -1:
-        mod.weight.data.normal_(1.0, 0.02)
-        mod.bias.data.fill_(0)
-
-
 class BaseRunner(ABC):
     def __init__(self, cfg: DictConfig, working_dir: str) -> None:
         super().__init__()
@@ -142,4 +128,4 @@ def _get_attr(self, name: str) -> Any:
 
         module_path, attr_name = name.split(" - ")
         module = import_module(module_path)
-        return getattr(module, attr_name)
+        return getattr(module, attr_name)