diff --git a/mlair/model_modules/convolutional_networks.py b/mlair/model_modules/convolutional_networks.py index 3e87bb7c595e4a0f709ca73449043ad74e9b6e9d..a388f368366ff00a975f54e927b9a2b693cd7a4a 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"]