saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-28 20:11:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix saving

Browse source
This commit is contained in:
Antonin Raffin 2020-09-24 17:12:31 +02:00
parent 6a0ee7f321
commit ccaf9c3882
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
stable_baselines3/tqc/tqc.py
View file

@ -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]]:
"""