diff --git a/torchy_baselines/common/base_class.py b/torchy_baselines/common/base_class.py index 40c3061..c70c2a2 100644 --- a/torchy_baselines/common/base_class.py +++ b/torchy_baselines/common/base_class.py @@ -547,7 +547,9 @@ class BaseRLModel(object): @staticmethod def excluded_save_params(): """ - returns the names of the parameters that should be excluded from save + Returns the names of the parameters that should be excluded by default + when saving the model. + :return: ([str]) List of parameters that should be excluded from save """ return ["env", "eval_env", "replay_buffer", "rollout_buffer"]