saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-06-05 00:00:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

comment refactoring

Browse source 
Co-Authored-By: Raffin, Antonin <Antonin.Raffin@dlr.de>
This commit is contained in:
Dormann, Noah 2019-12-05 08:08:20 +01:00 committed by GitHub Enterprise
parent 362bba73ba
commit 03bf513e5e
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
torchy_baselines/ppo/ppo.py
View file

@ -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()}