typo in valid_loss in utils.train

nkern · Mar 11, 2024 · 64ab10f · 64ab10f
1 parent 3c1652a
commit 64ab10f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py21cmnet/utils.py b/py21cmnet/utils.py
@@ -127,7 +127,7 @@ def train(model, train_dloader, loss_fn, optim, optim_kwargs={}, track_mini=True
                         train_loss.append(loss.cpu())
                         train_acc.append(acc.cpu())
                     else:
-                        valid_loss.append(acc.cpu())
+                        valid_loss.append(loss.cpu())
                         valid_acc.append(acc.cpu())