From ac43287650acc95a2f16e7ce18d56378031b9551 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Mon, 5 Jul 2021 07:45:21 +0200
Subject: [PATCH] comment out stand alone optimizer in IntelliO3TsArchitecture
---
mlair/model_modules/model_class.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mlair/model_modules/model_class.py b/mlair/model_modules/model_class.py
index 004fd217..d9527b84 100644
--- a/mlair/model_modules/model_class.py
+++ b/mlair/model_modules/model_class.py
@@ -461,7 +461,7 @@ class IntelliO3TsArchitecture(AbstractModelClass):
self.model = keras.utils.multi_gpu_model(model, self.avail_gpus)
def set_compile_options(self):
- self.optimizer = keras.optimizers.SGD(lr=self.initial_lr, momentum=0.9)
+ # self.optimizer = keras.optimizers.SGD(lr=self.initial_lr, momentum=0.9)
self.compile_options = {"optimizer": keras.optimizers.adam(lr=self.initial_lr, amsgrad=True),
"loss": [l_p_loss(4), keras.losses.mean_squared_error],
"metrics": ['mse'],
--
GitLab