saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-06-03 23:49:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Cleared base_class.load description

Browse source
This commit is contained in:
Noah Dormann 2019-11-28 10:36:21 +01:00
parent 4f8f936451
commit b75ffe166a
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
torchy_baselines/common/base_class.py
View file

@ -275,7 +275,7 @@ class BaseRLModel(object):
"""
Load the model from file
:param load_path: (str or file-like) the saved parameter location
:param load_path: (str) the saved parameter location
:param env: (Gym Envrionment) the new environment to run the loaded model on
(can be None if you only need prediction from a trained model)
:param kwargs: extra arguments to change the model when loading
@ -507,7 +507,7 @@ class BaseRLModel(object):
"""
saves all the params from init and pytorch params in a file for continuous learning
:param path: path to the file where the data should be saved
:param path: (str) path to the file where the data should be saved
:return:
"""
data = self.__dict__