vsooda / mxnet_merge_bn

remove batchnorm when deploy mxnet model

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

this is a demo for removing batchnorm layer for mxnet trained model

currently, this is a demo for remove bn layer for mxnet-ssd. you can easily to change for another model.

the accuracy had been tested. there is no problem in this script.

todo: add demo for another more common example.

About

remove batchnorm when deploy mxnet model


Languages

Language:Python 100.0%