diff --git a/mlair/model_modules/recurrent_networks.py b/mlair/model_modules/recurrent_networks.py index 55a3d58576c1062e6f852d1c8fb33612679b51a9..7adc91119ef93adc3db22fe8a390c7e84c62630e 100644 --- a/mlair/model_modules/recurrent_networks.py +++ b/mlair/model_modules/recurrent_networks.py @@ -136,7 +136,7 @@ class RNN(AbstractModelClass): raise AttributeError(f"Given activation {activation} is not supported in this model class.") def set_compile_options(self): - self.compile_options = {"loss": [custom_loss([keras.losses.mean_squared_error, var_loss])], + self.compile_options = {"loss": [keras.losses.mean_squared_error], "metrics": ["mse", "mae", var_loss]} def _set_optimizer(self, optimizer, **kwargs):