Fixing sde's bugs

2022-08-14 16:10:22 +02:00
parent 0e4eedae5e
commit 0ee65e789b
3 changed files with 32 additions and 16 deletions
@@ -140,10 +140,9 @@ class ActorCriticPolicy(BasePolicy):
        # Keyword arguments for gSDE distribution
        if use_sde:
            add_dist_kwargs = {
-                "full_std": full_std,
-                "squash_output": squash_output,
-                "use_expln": use_expln,
-                "learn_features": False,
+                'use_sde': True,
+                # "use_expln": use_expln,
+                # "learn_features": False,
            }
            for k in add_dist_kwargs:
                dist_kwargs[k] = add_dist_kwargs[k]