From e3690b042501d2e2e1d446248a0ec41903bb33b9 Mon Sep 17 00:00:00 2001
From: Cian Hughes <chughes000@gmail.com>
Date: Wed, 15 May 2024 18:40:20 +0100
Subject: [PATCH] Added wandb.finish to properly start new runs

---
 symbolic_nn_tests/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/symbolic_nn_tests/model.py b/symbolic_nn_tests/model.py
index 4849bc8..285c33d 100644
--- a/symbolic_nn_tests/model.py
+++ b/symbolic_nn_tests/model.py
@@ -32,7 +32,7 @@ def main(loss_func=nn.functional.cross_entropy, logger=None, **kwargs):
     train, val, test = get_singleton_dataset()
     lmodel = TrainingWrapper(model, loss_func=loss_func)
     lmodel.configure_optimizers(**kwargs)
-    trainer = L.Trainer(max_epochs=1, logger=logger)
+    trainer = L.Trainer(max_epochs=20, logger=logger)
     trainer.fit(model=lmodel, train_dataloaders=train, val_dataloaders=val)
     trainer.test(dataloaders=test)