diff --git a/emote/callbacks/logging.py b/emote/callbacks/logging.py index a540c74f..37a43f99 100644 --- a/emote/callbacks/logging.py +++ b/emote/callbacks/logging.py @@ -53,7 +53,7 @@ def end_cycle(self, bp_step, bp_samples): for k, v in cb.windowed_scalar_cumulative.items(): k_split = k.split("/") - k_split[0] = k_split[0] + "_" + suffix + k_split[0] = f"cumulative/{k_split[0]}_{suffix}" k = "/".join(k_split) self._writer.add_scalar(f"{k}/cumulative", v, bp_step) diff --git a/emote/callbacks/wb_logger.py b/emote/callbacks/wb_logger.py index 4748173f..f362378c 100644 --- a/emote/callbacks/wb_logger.py +++ b/emote/callbacks/wb_logger.py @@ -63,7 +63,7 @@ def end_cycle(self, bp_step, bp_samples): for k, v in cb.windowed_scalar_cumulative.items(): k_split = k.split("/") - k_split[0] = k_split[0] + "_" + suffix + k_split[0] = f"cumulative/{k_split[0]}_{suffix}" k = "/".join(k_split) log_dict[f"{k}/cumulative"] = v