replace the print statement

MannLabs · May 16, 2024 · a7210f5 · a7210f5
1 parent 80f956f
commit a7210f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/sparcscore/ml/plmodels.py b/src/sparcscore/ml/plmodels.py
@@ -206,10 +206,11 @@ 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)
 
+        loss = F.mse_loss(output, target) # L2 loss
+
         # accuracy metrics for regression???
 
         self.log('loss/train', loss, on_step=False, on_epoch=True, prog_bar=True)