awslabs / keras-apache-mxnet

[DEPRECATED] Amazon Deep Learning's Keras with Apache MXNet support
https://github.com/awslabs/keras-apache-mxnet/wiki
Other
289 stars 65 forks source link

fix batchnorm freeze mean and var #252

Closed roywei closed 4 years ago

roywei commented 4 years ago

As per user feedback, Keras-MXNet was not able to freeze the running mean and var weights during fine tune.

This PR fix the bug and added unit test Reference: https://discuss.mxnet.io/t/how-freeze-batchnorm-layer-in-symbolblock/3949

sandeep-krishnamurthy commented 4 years ago

Thanks.