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

Make sure pca_policy knowns, we are using pca

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-08-22 01:10:41 +02:00
parent d6c05668cb
commit 0af83eb7dc
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
sbBrix/common/off_policy_algorithm.py
View File

@ -85,6 +85,7 @@ class BetterOffPolicyAlgorithm(OffPolicyAlgorithm):
): ):
assert not (use_sde and use_pca) assert not (use_sde and use_pca)
self.use_pca = use_pca self.use_pca = use_pca
policy_kwargs["use_pca"] = self.use_pca
super().__init__( super().__init__(
policy=policy, policy=policy,
env=env, env=env,