From e0f2649b424f39c10d6f106b09bb6d99e15f23e6 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Tue, 12 May 2020 14:48:54 +0200
Subject: [PATCH] clean up model setup
---
src/run_modules/model_setup.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/src/run_modules/model_setup.py b/src/run_modules/model_setup.py
index 0563fc88..e8259b28 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)
--
GitLab