Remove torchsummary

siamese_nn.py

@@ -9,7 +9,6 @@ two vectors of similarity which then in turn are being passed to Contrastive Los
 indicates whether or not the samples were similar or dissimilar. 
 """
 import torch.nn as nn
-from torchsummary import summary
 
 class Siamese_nn(nn.Module):
     """
@@ -75,9 +74,4 @@ class Siamese_nn(nn.Module):
         """
         output1 = self.forwardOne(input1)
         output2 = self.forwardOne(input2)
-        return output1, output2
-
-if __name__ == '__main__':
-    model = Siamese_nn().to('cuda')
-    summary(model, [(1, 153, 136), (1, 153, 136)], 1)
-# %%
+        return output1, output2