How to set weights of the batch normalization layer?

[proni...@gmail.com]

Hi all,

How do I set weights of the batch normalization layer?

I am bit confused by the documentation 

• weights: Initialization weights. List of 2 Numpy arrays, with shapes: [(input_shape,), (input_shape,)] Note that the order of this list is [gamma, beta, mean, std]

Do we need all four [gamma, beta, mean, std]? Is there a way to set weights using only [gamma, beta].


Thanks for the help.

[François Chollet]

The docs need to be clarified. You do need all 4 arrays in a call to `set_weights`.

Otherwise you can call `K.set_value` on specific attributes of the layer (e.g. `layer.gamma`).

[proni...@gmail.com]

Thanks Francois! I will try that.