gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 03:30:19 +01:00
Code Issues Projects Releases Wiki Activity

fixes to rainbow dqn + a cartpole based golden test (#253)

Browse Source
This commit is contained in:
Gal Leibovich
2019-03-21 12:57:56 +02:00
committed by GitHub
parent 83741fa92a
commit abec59f367
6 changed files with 127 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
rl_coach/agents/rainbow_dqn_agent.py
View File

@@ -60,9 +60,12 @@ class RainbowDQNAgentParameters(CategoricalDQNAgentParameters):
def __init__(self):
super().__init__()
self.algorithm = RainbowDQNAlgorithmParameters()
# ParameterNoiseParameters is changing the network wrapper parameters. This line needs to be done first.
self.network_wrappers = {"main": RainbowDQNNetworkParameters()}
self.exploration = ParameterNoiseParameters(self)
self.memory = PrioritizedExperienceReplayParameters()
self.network_wrappers = {"main": RainbowDQNNetworkParameters()}
@property
def path(self):