Fixed SDE: sampling had dimension mismatches

2022-08-14 20:09:10 +02:00
parent e1c59cffd0
commit 28d0c609bc
3 changed files with 16 additions and 10 deletions
@@ -203,8 +203,7 @@ class ActorCriticPolicy(BasePolicy):

        if isinstance(
                self.action_dist, UniversalGaussianDistribution):
-            self.action_dist.sample_weights(
-                get_action_dim(self.action_space), batch_size=n_envs)
+            self.action_dist.sample_weights(batch_size=n_envs)

    def _build_mlp_extractor(self) -> None:
        """