From e96d4ae1a6fe36d5e74cf7b5c1f4a86d180f92ec Mon Sep 17 00:00:00 2001 From: Nicklas Hansen Date: Fri, 2 Feb 2024 15:56:54 -0800 Subject: [PATCH] reduce # wandb calls --- tdmpc2/common/logger.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tdmpc2/common/logger.py b/tdmpc2/common/logger.py index ea26996..4dce7ca 100755 --- a/tdmpc2/common/logger.py +++ b/tdmpc2/common/logger.py @@ -227,8 +227,10 @@ class Logger: xkey = "step" elif category == "pretrain": xkey = "iteration" + _d = dict() 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: keys = ["step", "episode_reward"] self._eval.append(np.array([d[keys[0]], d[keys[1]]]))