wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-08-06 13:20:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

initializations -> initializers

Browse Source
This commit is contained in:
farizrahman4u
2017-03-19 08:49:50 +05:30
parent 28a0f55611
commit 87b43828e0
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/layers/normalization.py
+4 -4
View File
@@ -44,11 +44,11 @@ class BatchRenormalization(Layer):
`[(input_shape,), (input_shape,)]`
Note that the order of this list is [gamma, beta, mean, std]
beta_init: name of initialization function for shift parameter
(see [initializations](../initializations.md)), or alternatively,
(see [initializer](../initializer.md)), or alternatively,
Theano/TensorFlow function to use for weights initialization.
This parameter is only relevant if you don't pass a `weights` argument.
gamma_init: name of initialization function for scale parameter (see
[initializations](../initializations.md)), or alternatively,
[initializer](../initializer.md)), or alternatively,
Theano/TensorFlow function to use for weights initialization.
This parameter is only relevant if you don't pass a `weights` argument.
gamma_regularizer: instance of [WeightRegularizer](../regularizers.md)
@@ -73,8 +73,8 @@ class BatchRenormalization(Layer):
gamma_init='one', gamma_regularizer=None, beta_regularizer=None,
**kwargs):
self.supports_masking = True
self.beta_init = initializations.get(beta_init)
self.gamma_init = initializations.get(gamma_init)
self.beta_init = initializer.get(beta_init)
self.gamma_init = initializer.get(gamma_init)
self.epsilon = epsilon
self.mode = mode
self.axis = axis