hongshaorou/tdmpc2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix minor logging issue in offline trainer

Browse Source
This commit is contained in:
Nicklas Hansen
2024-01-03 10:12:28 -08:00
parent 1d224cec3a
commit 13cac07759
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tdmpc2/trainer/offline_trainer.py
View File

@@ -76,7 +76,7 @@ class OfflineTrainer(Trainer):
train_metrics = self.agent.update(self.buffer) train_metrics = self.agent.update(self.buffer)
# Evaluate agent periodically # 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 = { metrics = {
'iteration': i, 'iteration': i,
'total_time': time() - self._start_time, 'total_time': time() - self._start_time,