CRNN-Pytorch2TensorRT-via-ONNX icon indicating copy to clipboard operation
CRNN-Pytorch2TensorRT-via-ONNX copied to clipboard

the tensorrt is slow than pytorch

Open archwolf118 opened this issue 3 years ago • 0 comments

Thank you for this wonderful project. But I met a problem in python api of tensorrt to inference the model which is trained on pytorch. I found the speed is slower in tensorrt than pytorch. Like this project, I have 7226 classes of char, so the speed is 2.9ms/pic (32*100) in tensorrt (if dynamic input, the speed is 9ms/pic) . But when I test the same picture in pytorch(pytorch is dynamic input), it show only 3ms/pic. Have you test the speed in pytorch?

archwolf118 avatar Apr 02 '21 01:04 archwolf118