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

Update torchy_baselines/common/base_class.py

Browse source
This commit is contained in:
Raffin, Antonin 2019-12-05 14:53:14 +01:00 committed by GitHub Enterprise
parent 695cdc63a4
commit bac9d4efed
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

@ -556,7 +556,7 @@ class BaseRLModel(object):
def save(self, path, exclude=None, include=None):
"""
Save all the attributes of the object and the model parameters in a zip-file for continuous learning
Save all the attributes of the object and the model parameters in a zip-file.
:param path: (str) path to the file where the rl agent should be saved
:param exclude: ([str]) name of parameters that should be excluded in addition to the default one