mirror of
https://github.com/gryf/coach.git
synced 2025-12-17 19:20:19 +01:00
fixes to rainbow dqn + a cartpole based golden test (#253)
This commit is contained in:
@@ -59,6 +59,7 @@ class InputEmbedder(object):
|
||||
# layers order is conv -> batchnorm -> activation -> dropout
|
||||
if isinstance(self.scheme, EmbedderScheme):
|
||||
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