saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-09-16 22:20:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

added standart exclude parameters

Browse source 
Co-Authored-By: Raffin, Antonin <Antonin.Raffin@dlr.de>
This commit is contained in:
Dormann, Noah 2019-12-05 08:09:26 +01:00 committed by GitHub Enterprise
parent 03bf513e5e
commit 85d3743224
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
torchy_baselines/common/base_class.py
View file

@ -499,7 +499,7 @@ class BaseRLModel(object):
returns the names of the parameters that should be excluded from save
:return: (list) List of parameters that should be excluded from save
"""
return ["replay_buffer"]
return ["env", "eval_env", "replay_buffer", "rollout_buffer"]
def save(self, path, exclude=None, include=None):
"""