remove print statements

MannLabs · May 16, 2024 · 5b21fd0 · 5b21fd0
1 parent f44f914
commit 5b21fd0
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/src/sparcscore/ml/models.py b/src/sparcscore/ml/models.py
@@ -181,7 +181,6 @@ def vgg(cfg, in_channels,  **kwargs):
         return model
 
     def forward(self, x):
-        print("x shape: ", x.shape)
         x = self.norm(x)
         x = self.features(x)
 

diff --git a/src/sparcscore/ml/plmodels.py b/src/sparcscore/ml/plmodels.py
@@ -204,9 +204,7 @@ def configure_optimizers(self):
 
     def training_step(self, batch):
         data, target = batch
-        print("Training data shape: ", data.shape)
-        print("Batch size: ", data.size(0))
-
+
         output = self.network(data) # Forward pass, only one output
         loss = F.mse_loss(output, target) # L2 loss