Skip to content
Snippets Groups Projects
Commit e0f2649b authored by Felix Kleinert's avatar Felix Kleinert
Browse files

clean up model setup

parent 33e294b0
No related branches found
No related tags found
3 merge requests!125Release v0.10.0,!124Update Master to new version v0.10.0,!92introduce 'compile_options' in model_class which combines all allowed args of...
Pipeline #36808 passed
...@@ -60,10 +60,11 @@ class ModelSetup(RunEnvironment): ...@@ -60,10 +60,11 @@ class ModelSetup(RunEnvironment):
self.data_store.set("channels", channels, self.scope) self.data_store.set("channels", channels, self.scope)
def compile_model(self): 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 compile_options = self.model.compile_options
# self.model.compile(optimizer=optimizer, loss=loss, **compile_options)
self.model.compile(**compile_options) self.model.compile(**compile_options)
self.data_store.set("model", self.model, self.scope) self.data_store.set("model", self.model, self.scope)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment