fix eval mode

2025-10-01 12:08:36 +08:00 · 2025-10-01 12:08:36 +08:00 · 52e144f20d
commit 52e144f20d
parent bd1480df33
1 changed files with 1 additions and 0 deletions
--- a/YOPO/policy/yopo_trainer.py
+++ b/YOPO/policy/yopo_trainer.py
@ -70,6 +70,7 @@ class YopoTrainer:
            for self.epoch_i in range(epoch):
                self.policy.train()
                self.train_one_epoch(self.epoch_i, total_progress)
+                self.policy.eval()
                self.eval_one_epoch(self.epoch_i)
                if save_interval is not None and (self.epoch_i + 1) % save_interval == 0:
                    self.progress_log.console.log("Saving model...")