The bug of the batchnorm in the ndarray


#1

I work on the scala with mxnet 1.1.0

I would like to replace the symbol.batchnorm with the NDArray.batchnorm

First, “set_global_stats” could not be controlled. I knew the default value is false and I set is as true. But I found it didn’t work.

Test: the data

val after_bn=NDArray.BatchNorm(test,gamma,beta,moving_mean,moving_variance,0.01,0,false,false,true,1,true)

result

mean:5.0

val after_bn=NDArray.BatchNorm(test,gamma,beta,moving_mean,moving_variance,0.01,0,false,true,true,1,true)

result

mean:5.0


Second, if the "set_global_stats" work, how to get the data_mean and data_variance.