remove print statements

Author: namsaraeva

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)
 

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