debug: debug nan in loss

Floto-Lab · Feb 1, 2025 · e40d0a9 · e40d0a9
1 parent 53695c1
commit e40d0a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bactgraph/modeling/model.py b/bactgraph/modeling/model.py
@@ -137,7 +137,7 @@ def training_step(self, batch, batch_idx):
         x_batch, edge_index_batch, y = batch
         preds = self.forward(x_batch, edge_index_batch.type(torch.long))
         loss = F.mse_loss(preds, y.view(-1))
-        print("Loss:", loss)
+        print("preds", preds, "y", y)
         self.log("train_loss", loss, on_step=True, on_epoch=True, prog_bar=True)
         return loss