From 13cac077595d9eb600cbc6634d31335ab738c08c Mon Sep 17 00:00:00 2001 From: Nicklas Hansen Date: Wed, 3 Jan 2024 10:12:28 -0800 Subject: [PATCH] fix minor logging issue in offline trainer --- tdmpc2/trainer/offline_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tdmpc2/trainer/offline_trainer.py b/tdmpc2/trainer/offline_trainer.py index 9ed4fd4..1bace8e 100755 --- a/tdmpc2/trainer/offline_trainer.py +++ b/tdmpc2/trainer/offline_trainer.py @@ -76,7 +76,7 @@ class OfflineTrainer(Trainer): train_metrics = self.agent.update(self.buffer) # Evaluate agent periodically - if i % self.cfg.eval_freq == 0 or i == 10_000: + if i % self.cfg.eval_freq == 0 or i % 10_000 == 0: metrics = { 'iteration': i, 'total_time': time() - self._start_time,