fix

vwxyzjn · Oct 15, 2023 · 47ac0a1 · 47ac0a1
1 parent c9ea9a9
commit 47ac0a1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cleanrl/ppo_continuous_action.py b/cleanrl/ppo_continuous_action.py
@@ -339,6 +339,7 @@ def get_action_and_value(self, x, action=None):
             run_name=f"{run_name}-eval",
             Model=Agent,
             device=device,
+            gamma=args.gamma,
         )
         for idx, episodic_return in enumerate(episodic_returns):
             writer.add_scalar("eval/episodic_return", episodic_return, idx)