The same pytorch unet model, on jetpack 4.6 is normal, on jetpack 5.0.2, the inference result is incorrect

[kangwenkui]

The same pytorch unet model, on jetpack 4.6 is normal, on jetpack 5.0.2, the inference result is incorrect.
Model transformation and inference based on thishttps://github.com/wang-xinyu/tensorrtx

[AastaLLL]

Hi,

Please share more information with us.

• The reproduced steps on both JetPack 4.6 and JetPack 5.0.2.
• The output (on JetPack 5.0.2) and expected output (on JetPack 4.6.2) result.

It looks like you are using a third-party implementation, have you also checked this issue with the source author?
Thanks.