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
@@ -185,8 +185,7 @@ class Actor(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 get_action_dist_params(self, obs: th.Tensor) -> Tuple[th.Tensor, th.Tensor, Dict[str, th.Tensor]]:
        """