BN layer result wrong

part of the model is as following.:

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

ckpt.t7 (43.9 MB)

Pytorch to onnx to trt, BN layer output wrong result - Jetson & Embedded Systems / Jetson Xavier NX - NVIDIA Developer Forums

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