diff --git a/src/run_modules/model_setup.py b/src/run_modules/model_setup.py index 0563fc88948c7281f389ec300b294116abd7a491..e8259b2847ea4ede1b365f49778f019c004fa7f1 100644 --- a/src/run_modules/model_setup.py +++ b/src/run_modules/model_setup.py @@ -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)