From 695cdc63a44b52b642759398f3835bfe73e90cc2 Mon Sep 17 00:00:00 2001 From: "Raffin, Antonin" Date: Thu, 5 Dec 2019 14:52:59 +0100 Subject: [PATCH] Update torchy_baselines/common/base_class.py --- torchy_baselines/common/base_class.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchy_baselines/common/base_class.py b/torchy_baselines/common/base_class.py index c70c2a2..8f14cb3 100644 --- a/torchy_baselines/common/base_class.py +++ b/torchy_baselines/common/base_class.py @@ -558,7 +558,7 @@ class BaseRLModel(object): """ Save all the attributes of the object and the model parameters in a zip-file for continuous learning - :param path: (str) path to the file where the data should be saved + :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 :param include: ([str]) name of parameters that might be excluded but should be included anyway """