diff --git a/mlair/model_modules/model_class.py b/mlair/model_modules/model_class.py index ec7f217902d7e29b1092ecd1ef51b3fc30602f62..bf97864feefc732c2d39bb0a7dd69644d5fc496c 100644 --- a/mlair/model_modules/model_class.py +++ b/mlair/model_modules/model_class.py @@ -458,7 +458,7 @@ class IntelliO3_ts_architecture(AbstractModelClass): def set_compile_options(self): self.compile_options = {"optimizer": keras.optimizers.adam(lr=self.initial_lr, amsgrad=True), - "loss": [l_p_loss(4), keras.losses.mean_squared_error], + "loss": [keras.losses.mean_squared_error, keras.losses.mean_squared_error], "metrics": ['mse'], "loss_weights": [.01, .99] } \ No newline at end of file