diff --git a/train.py b/train.py
index c714bc27..da102e4d 100644
--- a/train.py
+++ b/train.py
@@ -170,7 +170,7 @@ def train(
             best_loss = mloss['total']
 
         # Save training results
-        save = False
+        save = True
         if save:
             # Save latest checkpoint
             checkpoint = {'epoch': epoch,