remove env name from hparams tag (#309)

cpnota · Feb 27, 2024 · dc295ab · dc295ab
1 parent 1119aad
commit dc295ab
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/all/logging/experiment.py b/all/logging/experiment.py
@@ -62,11 +62,8 @@ def add_schedule(self, name, value, step="frame"):
     def add_hparams(self, hparam_dict, metric_dict, step="frame"):
         allowed_types = (int, float, str, bool, torch.Tensor)
         hparams = {k: v for k, v in hparam_dict.items() if isinstance(v, allowed_types)}
-        metrics = {
-            f"{self.env_name}/{metric}": value for metric, value in metric_dict.items()
-        }
         super().add_hparams(
-            hparams, metrics, run_name=".", global_step=self._get_step("frame")
+            hparams, metric_dict, run_name=".", global_step=self._get_step("frame")
         )
 
     def _add_scalar(self, name, value, step="frame"):