Update

[ghstack-poisoned]
pytorch · Nov 18, 2024 · 79a005c · 79a005c
2 parents 0fa015f + 425d671
commit 79a005c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sota-implementations/a2c/utils_mujoco.py b/sota-implementations/a2c/utils_mujoco.py
@@ -60,7 +60,7 @@ def make_ppo_models_state(proof_environment, device, *, compile: bool = False):
         "low": proof_environment.action_spec.space.low.to(device),
         "high": proof_environment.action_spec.space.high.to(device),
         "tanh_loc": False,
-        "safe_tanh": not compile,
+        "safe_tanh": True,
     }
 
     # Define policy architecture