From b75ffe166a2c78efa7a9e9c183055c03cce0f033 Mon Sep 17 00:00:00 2001 From: Noah Dormann Date: Thu, 28 Nov 2019 10:36:21 +0100 Subject: [PATCH] Cleared base_class.load description --- torchy_baselines/common/base_class.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchy_baselines/common/base_class.py b/torchy_baselines/common/base_class.py index 57c27b0..3384775 100644 --- a/torchy_baselines/common/base_class.py +++ b/torchy_baselines/common/base_class.py @@ -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__