removed torchinfo usage

KatherLab · Nov 26, 2024 · f68995d · f68995d
1 parent 9cecb48
commit f68995d
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/application/jobs/minimal_training_pytorch_cnn/app/custom/models/mini_model.py b/application/jobs/minimal_training_pytorch_cnn/app/custom/models/mini_model.py
@@ -1,7 +1,6 @@
 from models.base_model import BasicClassifier
 import torch
 import torch.nn as nn
-import torchinfo
 import math
 
 class MiniCNNForTesting(BasicClassifier):
@@ -35,7 +34,6 @@ def __init__(self,
             nn.Linear(linear_waste_of_memory, 1)
         )
         print(self.model)
-        torchinfo.summary(self.model, input_size=(1, 1, 18, 18))
 
     def forward(self, x_in: torch.Tensor, **kwargs) -> torch.Tensor:
         return self.model(x_in)