From 64325ec82765ef6d2b40990f2d0c296bf46394f2 Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Thu, 10 Mar 2022 15:53:14 +0100
Subject: [PATCH] removed selu from initializer
---
mlair/model_modules/convolutional_networks.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/mlair/model_modules/convolutional_networks.py b/mlair/model_modules/convolutional_networks.py
index 3e87bb7c..a388f368 100644
--- a/mlair/model_modules/convolutional_networks.py
+++ b/mlair/model_modules/convolutional_networks.py
@@ -18,8 +18,7 @@ class CNNfromConfig(AbstractModelClass):
"prelu": partial(keras.layers.PReLU, alpha_initializer=keras.initializers.constant(value=0.25)),
"leakyrelu": partial(keras.layers.LeakyReLU)}
_initializer = {"tanh": "glorot_uniform", "sigmoid": "glorot_uniform", "linear": "glorot_uniform",
- "relu": keras.initializers.he_normal(), "selu": keras.initializers.lecun_normal(),
- "prelu": keras.initializers.he_normal()}
+ "relu": keras.initializers.he_normal(), "prelu": keras.initializers.he_normal()}
_optimizer = {"adam": keras.optimizers.Adam, "sgd": keras.optimizers.SGD}
_regularizer = {"l1": keras.regularizers.l1, "l2": keras.regularizers.l2, "l1_l2": keras.regularizers.l1_l2}
_requirements = ["lr", "beta_1", "beta_2", "epsilon", "decay", "amsgrad", "momentum", "nesterov", "l1", "l2"]
--
GitLab