diff --git a/sbBrix/sac/sac.py b/sbBrix/sac/sac.py index b3ee34e..52213c2 100644 --- 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,