From 03a0d437ef54cb5e9278d7bd0f62a22fc6042496 Mon Sep 17 00:00:00 2001 From: Noah Dormann Date: Thu, 21 Nov 2019 16:49:49 +0100 Subject: [PATCH] refactor --- torchy_baselines/sac/sac.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/torchy_baselines/sac/sac.py b/torchy_baselines/sac/sac.py index 2497523..ad36f64 100644 --- a/torchy_baselines/sac/sac.py +++ b/torchy_baselines/sac/sac.py @@ -288,12 +288,9 @@ class SAC(BaseRLModel): def load_parameters(self, load_dict, opt_params): """ Load model parameters and optimizer parameters from a dictionary - Dictionary should be of shape torch model.state_dict() - This does not load agent's hyper-parameters. - :param load_dict: (dict) dict of parameters from model.state_dict() :param opt_params: (dict of dicts) dict of optimizer state_dicts should be handled in child_class """