Skip to content
Snippets Groups Projects

introduce 'compile_options' in model_class which combines all allowed args of...

Merged Ghost User requested to merge felix_issue110_custom_compile_options into develop
1 file
+ 4
3
Compare changes
  • Side-by-side
  • Inline
@@ -60,10 +60,11 @@ class ModelSetup(RunEnvironment):
self.data_store.set("channels", channels, self.scope)
def compile_model(self):
# optimizer = self.data_store.get("optimizer", self.scope)
# loss = self.model.loss
"""
Compiles the keras model. Compile options are mandetory and have to be set by implementing set_compile() method
in child class of AbstractModelClass.
"""
compile_options = self.model.compile_options
# self.model.compile(optimizer=optimizer, loss=loss, **compile_options)
self.model.compile(**compile_options)
self.data_store.set("model", self.model, self.scope)
Loading