From fdb544e775c989d6724f8829eab55a136054cf35 Mon Sep 17 00:00:00 2001 From: "Dormann, Noah" Date: Thu, 5 Dec 2019 08:10:14 +0100 Subject: [PATCH] comment refactoring Co-Authored-By: Raffin, Antonin --- 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 842aa5b..2554e89 100644 --- a/torchy_baselines/common/base_class.py +++ b/torchy_baselines/common/base_class.py @@ -503,7 +503,7 @@ class BaseRLModel(object): def save(self, path, exclude=None, include=None): """ - saves all the params from init and pytorch params in a zip-file for continuous learning + 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 exclude: (list) name of parameters that should be excluded, use standard exclude params if None