Commit a2554881 authored by Mayank Mittal's avatar Mayank Mittal

renames tb namespace for episode info in skrl

parent a46c7d6d
......@@ -138,7 +138,7 @@ suppress_warnings = [
# of these warnings when parsing PEP-compliant type hints via static
# analysis. Since those hints are actual hints that *CANNOT* by definition
# by canonicalized, our only recourse is to squelch warnings altogether.
'ref.python',
"ref.python",
]
# -- Options for HTML output -------------------------------------------------
......
......@@ -156,7 +156,7 @@ class SkrlSequentialLogTrainer(Trainer):
if "episode" in infos:
for k, v in infos["episode"].items():
if isinstance(v, torch.Tensor) and v.numel() == 1:
self.agents.track_data(f"Episode / {k}", v.item())
self.agents.track_data(f"EpisodeInfo / {k}", v.item())
# post-interaction
self.agents.post_interaction(timestep=timestep, timesteps=self.timesteps)
# reset the environments
......@@ -220,7 +220,7 @@ class SkrlSequentialLogTrainer(Trainer):
if "episode" in infos:
for k, v in infos["episode"].items():
if isinstance(v, torch.Tensor) and v.numel() == 1:
agent.track_data(f"Episode / {k}", v.item())
agent.track_data(f"EpisodeInfo / {k}", v.item())
# perform post-interaction
super(type(agent), agent).post_interaction(timestep=timestep, timesteps=self.timesteps)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment