another bug

montrealrobotics · Sep 25, 2023 · 4813f4e · 4813f4e
1 parent 3281b23
commit 4813f4e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cleanrl/dqn.py b/cleanrl/dqn.py
@@ -418,6 +418,8 @@ def get_random_action():
                 if "episode" not in info:
                     continue
 
+
+                ep_cost = 0
                 last_step = global_step
                 print(f"global_step={global_step}, episodic_return={info['episode']['r']}")
                 writer.add_scalar("charts/episodic_return", info["episode"]["r"], global_step)