How to implement batchnorm with pytorch model

I am importing a pytorch model for inference.But does not like import from caffe or tensorflow.It looks like there is no batchnorm layer here.

I follow the pytorch example of tensorrt 5.0

Can someone provide some advice?


You can use the IScaleLayer to implement the function of batchnorm.