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

clean up model setup

parent 33e294b0
Branches
Tags v2.0.0
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