Bugs

metastable_baselines/distributions/distributions.py

@@ -111,6 +111,7 @@ def make_proba_distribution(
     if dist_kwargs is None:
         dist_kwargs = {}
 
+    if not use_pca:
         dist_kwargs['use_sde'] = use_sde
 
     if isinstance(action_space, gym.spaces.Box):

metastable_baselines/ppo/policies.py

@@ -156,6 +156,7 @@ class ActorCriticPolicy(BasePolicy):
                 "sde_net_arch is deprecated and will be removed in SB3 v2.4.0.", DeprecationWarning)
 
         self.use_sde = use_sde
+        print('[i] Use PCA? '+['No', 'Yes'][use_pca])
         self.use_pca = use_pca
         self.dist_kwargs = dist_kwargs