From 01127de4b3373718c0d2f6dd12603283535b2afb Mon Sep 17 00:00:00 2001
From: Cian-H <chughes000@gmail.com>
Date: Wed, 15 May 2024 12:46:48 +0100
Subject: [PATCH] Added testing step at end of training

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

diff --git a/symbolic_nn_tests/model.py b/symbolic_nn_tests/model.py
index 51abc69..285c33d 100644
--- a/symbolic_nn_tests/model.py
+++ b/symbolic_nn_tests/model.py
@@ -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__":