From f68995dfed31461ad2f3d9524a77a8842dec84bc Mon Sep 17 00:00:00 2001 From: Ole Schwen Date: Tue, 26 Nov 2024 10:16:09 +0100 Subject: [PATCH] removed torchinfo usage --- .../app/custom/models/mini_model.py | 2 -- 1 file changed, 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 index 569b3bd..d87bfe8 100644 --- 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)