diff --git a/mlair/model_modules/model_class.py b/mlair/model_modules/model_class.py index 004fd217411eb0d0e69ce51eb2260a7b120b5b93..d9527b84ef745874f2e849ecb3eaccd55c3c9ed4 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'],