Cian-H/symbolic_nn_tests
1
0
Fork 0
mirror of https://github.com/Cian-H/symbolic_nn_tests.git synced 2025-12-22 14:11:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added wandb.finish to properly start new runs

Browse Source
This commit is contained in:
Cian Hughes
2024-05-15 18:40:20 +01:00
parent 8a87e864fb
commit e3690b0425
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
symbolic_nn_tests/model.py
View File

@@ -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)