diff --git a/torchy_baselines/ppo/ppo.py b/torchy_baselines/ppo/ppo.py index 3b4734e..c4e0838 100644 --- a/torchy_baselines/ppo/ppo.py +++ b/torchy_baselines/ppo/ppo.py @@ -303,9 +303,7 @@ class PPO(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. diff --git a/torchy_baselines/td3/td3.py b/torchy_baselines/td3/td3.py index 035d344..e21c65a 100644 --- a/torchy_baselines/td3/td3.py +++ b/torchy_baselines/td3/td3.py @@ -248,12 +248,9 @@ class TD3(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 """