2nd subter network arch

Deep-SAD-PyTorch/src/base/base_net.py

@@ -1,6 +1,5 @@
 import logging
 
-import numpy as np
 import torch.nn as nn
 import torchscan
 
@@ -32,8 +31,5 @@ class BaseNet(nn.Module):
                 "Input dimension is not set. Please set input_dim before calling summary."
             )
             return
-        self.logger.info(
-            torchscan.summary(self, self.input_dim, receptive_field=receptive_field)
-        )
-        module_info = torchscan.crawl_module(self, self.input_dim)
-        pass
+        self.logger.info("torchscan:\n")
+        torchscan.summary(self, self.input_dim, receptive_field=receptive_field)