saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-09-17 22:30:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cleaned comments on model specific get and load functions

Browse source
This commit is contained in:
Noah Dormann 2019-11-21 16:50:59 +01:00
parent 03a0d437ef
commit 924ba9aea6
2 changed files with 0 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
torchy_baselines/ppo/ppo.py
View file

@ -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.

3
torchy_baselines/td3/td3.py
View file

@ -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
"""