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

reduce # wandb calls

Browse Source
This commit is contained in:
Nicklas Hansen
2024-02-02 15:56:54 -08:00
parent 02b18a48b1
commit 718966c28d
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tdmpc2/common/logger.py
View File

@@ -227,8 +227,10 @@ class Logger:
xkey = "step" xkey = "step"
elif category == "pretrain": elif category == "pretrain":
xkey = "iteration" xkey = "iteration"
_d = dict()
for k, v in d.items(): for k, v in d.items():
self._wandb.log({category + "/" + k: v}, step=d[xkey]) _d[category + "/" + k] = v
self._wandb.log(_d, step=d[xkey])
if category == "eval" and self._save_csv: if category == "eval" and self._save_csv:
keys = ["step", "episode_reward"] keys = ["step", "episode_reward"]
self._eval.append(np.array([d[keys[0]], d[keys[1]]])) self._eval.append(np.array([d[keys[0]], d[keys[1]]]))