From 52e144f20d97d704cce2ab23f2358d71a4d90fbc Mon Sep 17 00:00:00 2001 From: TJU-Lu Date: Wed, 1 Oct 2025 12:08:36 +0800 Subject: [PATCH] fix eval mode --- YOPO/policy/yopo_trainer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/YOPO/policy/yopo_trainer.py b/YOPO/policy/yopo_trainer.py index 96d837e..626cbcc 100644 --- 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...")