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

added get_parameter_list function

Browse source
This commit is contained in:
Noah Dormann 2019-12-05 08:50:11 +01:00
parent ff7c4d24f4
commit 7c8d375bcb
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
torchy_baselines/common/base_class.py
View file

@ -174,6 +174,13 @@ class BaseRLModel(object):
"""
pass
def get_parameter_list(self):
"""
Returns policy and optimizer parameters as a tuple
:return: (dict,dict) policy_parameters, opt_parameters
"""
return self.get_policy_parameters(),self.get_opt_parameters()
def get_policy_parameters(self):
"""
Get current model policy parameters as dictionary of variable name -> tensors.