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.