dodox/metastable-baselines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Bugs

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-05-21 17:03:22 +02:00
parent 43418a5e53
commit 6a9d7012c7
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
metastable_baselines/distributions/distributions.py
View File

@ -111,7 +111,8 @@ def make_proba_distribution(
if dist_kwargs is None: if dist_kwargs is None:
dist_kwargs = {} 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): if isinstance(action_space, gym.spaces.Box):
assert len( assert len(

1
metastable_baselines/ppo/policies.py
View File

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