From 29c65d634d9107b6251ad4d83fc72e4b391f08bc Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Mon, 8 Mar 2021 15:02:20 +0100
Subject: [PATCH] custom objects were missing
---
mlair/model_modules/fully_connected_networks.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mlair/model_modules/fully_connected_networks.py b/mlair/model_modules/fully_connected_networks.py
index 45b8eb63..9e3657c3 100644
--- a/mlair/model_modules/fully_connected_networks.py
+++ b/mlair/model_modules/fully_connected_networks.py
@@ -95,7 +95,7 @@ class FCN(AbstractModelClass):
# apply to model
self.set_model()
self.set_compile_options()
- # self.set_custom_objects(loss=self.compile_options['loss'])
+ self.set_custom_objects(loss=custom_loss([keras.losses.mean_squared_error, var_loss]), var_loss=var_loss)
def _set_activation(self, activation):
try:
--
GitLab