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 testing step at end of training

Browse Source
This commit is contained in:
Cian Hughes
2024-05-15 12:46:48 +01:00
parent 62649a39da
commit 01127de4b3
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
symbolic_nn_tests/model.py
View File

@@ -34,6 +34,7 @@ def main(loss_func=nn.functional.cross_entropy, logger=None, **kwargs):
lmodel.configure_optimizers(**kwargs)
trainer = L.Trainer(max_epochs=20, logger=logger)
trainer.fit(model=lmodel, train_dataloaders=train, val_dataloaders=val)
trainer.test(dataloaders=test)
if __name__ == "__main__":