Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
[ghstack-poisoned]
  • Loading branch information
vmoens committed Dec 3, 2024
1 parent c5c75aa commit b4dd118
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tutorials/sphinx-tutorials/multiagent_ppo.py
Original file line number Diff line number Diff line change
Expand Up @@ -450,8 +450,8 @@
out_keys=[env.action_key],
distribution_class=TanhNormal,
distribution_kwargs={
"low": env.action_spec_unbatched[env.action_key].space.low,
"high": env.action_spec_unbatched[env.action_key].space.high,
"low": env.full_action_spec_unbatched[env.action_key].space.low,
"high": env.full_action_spec_unbatched[env.action_key].space.high,
},
return_log_prob=True,
log_prob_key=("agents", "sample_log_prob"),
Expand Down

0 comments on commit b4dd118

Please sign in to comment.