Pytorch to onnx to trt, BN layer output wrong result

part of the model is as following.:

        nn.Conv2d(3, 64, 3, stride=1, padding=1),
        nn.MaxPool2d(3, 2, padding=1),
    when running the BN layer, output by pytorch and onnx is different.the

ckpt.t7 (43.9 MB)


Just want to confirm first.
Do you find the difference between PyTorch vs. ONNX or PyTorch vs. TensorRT?

If the difference occurs when converting the PyTorch model into ONNX.
It’s recommended to check this issue with the PyTorch team for better help.


This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.