mirror of
https://github.com/gryf/coach.git
synced 2025-12-19 04:00:18 +01:00
fixes to rainbow dqn + a cartpole based golden test (#253)
This commit is contained in:
@@ -49,6 +49,7 @@ class Middleware(object):
|
||||
# layers order is conv -> batchnorm -> activation -> dropout
|
||||
if isinstance(self.scheme, MiddlewareScheme):
|
||||
self.layers_params = copy.copy(self.schemes[self.scheme])
|
||||
self.layers_params = [convert_layer(l) for l in self.layers_params]
|
||||
else:
|
||||
# if scheme is specified directly, convert to TF layer if it's not a callable object
|
||||
# NOTE: if layer object is callable, it must return a TF tensor when invoked
|
||||
|
||||
Reference in New Issue
Block a user