amend

Signed-off-by: Matteo Bettini <[email protected]>
facebookresearch · Oct 6, 2023 · 3f43db6 · 3f43db6
1 parent e764b02
commit 3f43db6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/benchmarl/experiment/logger.py b/benchmarl/experiment/logger.py
@@ -190,10 +190,10 @@ def log_evaluation(
         ).mean(0)
         to_log.update(
             {
-                "eval/episode_reward_min": mean_group_return.min().item(),
-                "eval/episode_reward_mean": mean_group_return.mean().item(),
-                "eval/episode_reward_max": mean_group_return.max().item(),
-                "eval/episode_len_mean": sum(td.batch_size[0] for td in rollouts)
+                "eval/reward/episode_reward_min": mean_group_return.min().item(),
+                "eval/reward/episode_reward_mean": mean_group_return.mean().item(),
+                "eval/reward/episode_reward_max": mean_group_return.max().item(),
+                "eval/reward/episode_len_mean": sum(td.batch_size[0] for td in rollouts)
                 / len(rollouts),
             }
         )