Skip to content
Snippets Groups Projects
Commit 64325ec8 authored by leufen1's avatar leufen1
Browse files

removed selu from initializer

parent 1fff2055
No related branches found
No related tags found
5 merge requests!430update recent developments,!413update release branch,!412Resolve "release v2.0.0",!407Lukas issue368 feat prepare cnn class for filter benchmarking,!403Resolve "prepare CNN class for filter benchmarking"
Checking pipeline status
......@@ -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"]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment