From 6a9d7012c7b9afc3dfe40c76187fa3c600b38435 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 21 May 2023 17:03:22 +0200 Subject: [PATCH] Bugs --- metastable_baselines/distributions/distributions.py | 3 ++- metastable_baselines/ppo/policies.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/metastable_baselines/distributions/distributions.py b/metastable_baselines/distributions/distributions.py index 30cfab4..b65c25f 100644 --- a/metastable_baselines/distributions/distributions.py +++ b/metastable_baselines/distributions/distributions.py @@ -111,7 +111,8 @@ def make_proba_distribution( if dist_kwargs is None: dist_kwargs = {} - dist_kwargs['use_sde'] = use_sde + if not use_pca: + dist_kwargs['use_sde'] = use_sde if isinstance(action_space, gym.spaces.Box): assert len( diff --git a/metastable_baselines/ppo/policies.py b/metastable_baselines/ppo/policies.py index 108108b..644c56a 100644 --- a/metastable_baselines/ppo/policies.py +++ b/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