diff --git a/torchy_baselines/ppo/ppo.py b/torchy_baselines/ppo/ppo.py index bc66171..b0e9de2 100644 --- a/torchy_baselines/ppo/ppo.py +++ b/torchy_baselines/ppo/ppo.py @@ -317,7 +317,7 @@ class PPO(BaseRLModel): """ Returns a dict of all the optimizers and their parameters - :return: (Dict) of optimizer names and their state_dict + :return: (dict) of optimizer names and their state_dict """ return {"opt": self.policy.optimizer.state_dict()}