BUG? Batch normalization moving average is zero error parsing layer type BatchNorm index 2

AI & Data Science Deep Learning (Training & Inference) TensorRT
1

I can run the model using caffe.But error occur in TRT.
How can I solve the problem?

2

Hi,

Please refer below link:
https://devtalk.nvidia.com/default/topic/966759/tensorrt/tensorrt-batchnorm-error/

Can you provide the following information so we can better help?
Provide details on the platforms you are using:
o Linux distro and version
o GPU type
o Nvidia driver version
o CUDA version
o CUDNN version
o Python version [if using python]
o Tensorflow and PyTorch version
o TensorRT version

Also, if possible please share the script & model file to reproduce the issue.

Meanwhile, could you please try to use “trtexec” command in verbose mode to test the model.
https://github.com/NVIDIA/TensorRT/tree/master/samples/opensource/trtexec

Thanks