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

This commit is contained in:
Gal Leibovich authored and GitHub committed 2019-03-21 12:57:56 +02:00
1 parent 83741fa92a
commit abec59f367
6 files changed
+127 -23

No files matched your search

+4 -1
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):