From ccaf9c388224fa50bd26344392d53d51f7daa157 Mon Sep 17 00:00:00 2001 From: Antonin Raffin Date: Thu, 24 Sep 2020 17:12:31 +0200 Subject: [PATCH] Fix saving --- stable_baselines3/tqc/tqc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stable_baselines3/tqc/tqc.py b/stable_baselines3/tqc/tqc.py index 750c97b..1f36246 100644 --- a/stable_baselines3/tqc/tqc.py +++ b/stable_baselines3/tqc/tqc.py @@ -476,7 +476,7 @@ class TQC(OffPolicyAlgorithm): :return: (List[str]) List of parameters that should be excluded from save """ # Exclude aliases - return super(TQC, self).excluded_save_params() + ["actor", "critic", "critic_target"] + return super(TQC, self)._excluded_save_params() + ["actor", "critic", "critic_target"] def _get_torch_save_params(self) -> Tuple[List[str], List[str]]: """