tiny-tensorrt icon indicating copy to clipboard operation
tiny-tensorrt copied to clipboard

Always got nan result.

Open EEWenbinWu opened this issue 2 years ago • 2 comments

the model is resnet50 from TensorRT8 folder. onnx can produce the correct result, but tinytrt can't. image the onnx result image the tinytrt result image

EEWenbinWu avatar Jun 08 '23 08:06 EEWenbinWu

Could you please share the reproduce script then I can debug it. Thanks!

zerollzeng avatar Jun 11 '23 14:06 zerollzeng

I have to say I didn't put much effort on the python api and I didn't add accuracy test for it. So it might have bugs.

Source code in https://github.com/zerollzeng/tiny-tensorrt/blob/master/src/PyTrt.cpp

zerollzeng avatar Jun 11 '23 14:06 zerollzeng