tiny-tensorrt
tiny-tensorrt copied to clipboard
Always got nan result.
the model is resnet50 from TensorRT8 folder. onnx can produce the correct result, but tinytrt can't.
the onnx result
the tinytrt result
Could you please share the reproduce script then I can debug it. Thanks!
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