SAC support for PCA

2023-08-22 00:10:54 +02:00 · 2023-08-22 00:10:54 +02:00 · 8384d32287
commit 8384d32287
parent 82c6674615
1 changed files with 2 additions and 0 deletions
--- a/sbBrix/sac/sac.py
+++ b/sbBrix/sac/sac.py
@ -106,6 +106,7 @@ class SAC(BetterOffPolicyAlgorithm):
        use_sde: bool = False,
        sde_sample_freq: int = -1,
        use_sde_at_warmup: bool = False,
+        use_pca: bool = False,
        stats_window_size: int = 100,
        tensorboard_log: Optional[str] = None,
        policy_kwargs: Optional[Dict[str, Any]] = None,
@ -137,6 +138,7 @@ class SAC(BetterOffPolicyAlgorithm):
            use_sde=use_sde,
            sde_sample_freq=sde_sample_freq,
            use_sde_at_warmup=use_sde_at_warmup,
+            use_pca=use_pca,
            optimize_memory_usage=optimize_memory_usage,
            supported_action_spaces=(spaces.Box),
            support_multi_env=True,