fix minor logging issue in offline trainer
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user