Skip to content

Commit

Permalink
added random_normal -> keras.initializers.RandomNormal to initialisers
Browse files Browse the repository at this point in the history
  • Loading branch information
comane committed Dec 13, 2024
1 parent 050d723 commit 9aaad26
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion n3fit/src/n3fit/backends/keras_backend/MetaLayer.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,14 +8,15 @@
For instance: np_to_tensor is just a call to K.constant
"""

from keras.initializers import Constant, RandomUniform, glorot_normal, glorot_uniform
from keras.initializers import Constant, RandomUniform, glorot_normal, glorot_uniform, RandomNormal
from keras.layers import Layer

# Define in this dictionary new initializers as well as the arguments they accept (with default values if needed be)
initializers = {
"random_uniform": (RandomUniform, {"minval": -0.5, "maxval": 0.5}),
"glorot_uniform": (glorot_uniform, {}),
"glorot_normal": (glorot_normal, {}),
"random_normal": (RandomNormal, {"mean": 0.0, "stddev": 0.05}),
}


Expand Down

0 comments on commit 9aaad26

Please sign in to comment.