From f44f914b3335d4b4f3c7cbff4e37c155d18c6e7f Mon Sep 17 00:00:00 2001 From: namsaraeva Date: Thu, 16 May 2024 14:41:38 +0200 Subject: [PATCH] add batch size --- src/sparcscore/ml/plmodels.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/sparcscore/ml/plmodels.py b/src/sparcscore/ml/plmodels.py index 6ff8fbf..fa124e0 100644 --- a/src/sparcscore/ml/plmodels.py +++ b/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