add one more print for the result

MannLabs · May 16, 2024 · 80f956f · 80f956f
1 parent 4dcaabc
commit 80f956f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/sparcscore/ml/plmodels.py b/src/sparcscore/ml/plmodels.py
@@ -204,10 +204,12 @@ def configure_optimizers(self):
 
     def training_step(self, batch):
         data, target = batch
-        
+
         output = self.network(data) # Forward pass, only one output
         loss = F.mse_loss(output, target) # L2 loss
 
+        print("Result is: ", output, target)
+
         # accuracy metrics for regression???
 
         self.log('loss/train', loss, on_step=False, on_epoch=True, prog_bar=True)