From 649fc835d8eaa1f446dc02d9914de8c1baddfaee Mon Sep 17 00:00:00 2001
From: Philipp Guevorguian <pguevorg@uwo.ca>
Date: Tue, 12 Dec 2023 15:14:31 +0400
Subject: [PATCH] evaluate at end of training

---
 src/train.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/train.py b/src/train.py
index b44c105..23aef48 100644
--- a/src/train.py
+++ b/src/train.py
@@ -274,6 +274,8 @@ def train(
             except KeyboardInterrupt:
                 with accelerator.main_process_first():
                     logger.error("KeyboardInterrupt")
+            if not (max_steps % eval_steps == 0):
+                trainer.evaluate()
 
 
 if __name__ == "__main__":