From 7e6681d770a77484a2b70a1c935e7ea44e7e83c9 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Mon, 5 Jul 2021 10:20:39 +0200
Subject: [PATCH] update model class for IntelliO3Ar
---
mlair/model_modules/model_class.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/mlair/model_modules/model_class.py b/mlair/model_modules/model_class.py
index 004fd217..7e0895fb 100644
--- a/mlair/model_modules/model_class.py
+++ b/mlair/model_modules/model_class.py
@@ -455,13 +455,13 @@ class IntelliO3TsArchitecture(AbstractModelClass):
)
model = keras.Model(inputs=X_input, outputs=[out_minor1, out_main])
- if self.avail_gpus == 0:
+ if self.avail_gpus <= 1:
self.model = model
else:
self.model = keras.utils.multi_gpu_model(model, self.avail_gpus)
+ print(f"Set multi_gpu model with {self.avail_gpus} GPUs")
def set_compile_options(self):
- 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'],
@@ -663,4 +663,4 @@ class MyCNNModelSect(AbstractModelClass):
# drop=.94,
# epochs_drop=10)
self.loss = keras.losses.mean_squared_error
- self.compile_options = {"metrics": ["mse", "mae"]}
\ No newline at end of file
+ self.compile_options = {"metrics": ["mse", "mae"]}
--
GitLab