fix save check point

2025-09-08 23:21:03 +08:00
parent 64b41cc4f6
commit 49af1cf154
2 changed files with 6 additions and 3 deletions
@@ -197,6 +197,9 @@ class QwenVlAct_Trainer:
            self.train_loop(epoch)
            self.accelerator.wait_for_everyone()
            
+            if (epoch + 1) % self.config.get("epoch_save_interval", 10) == 0:
+                self.save_checkpoint(epoch)
+            
            # Validation after each epoch
            self.val_loop()
            self.accelerator.wait_for_everyone()