search

xuebinqin / U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."
Apache License 2.0
8.31k stars 1.43k forks source link

TensorRT conversion produces different outputs #339

Open aviadmx opened 1 year ago

aviadmx commented 1 year ago

Hi, Tried to convert the pretrained model to TensorRT with either torch/TensorRT and torch2trt from Nvidia. Despite using the same precision (FP32) the model outputs differ significantly on large images.

Was it something that done successfully by anyone yet?