add batch size

src/sparcscore/ml/plmodels.py

@@ -205,6 +205,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