update model class for IntelliO3Ar

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'],