evaluate at end of training

YerevaNN · Dec 12, 2023 · 649fc83 · 649fc83
1 parent 0d4224c
commit 649fc83
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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__":