search

NVIDIA-AI-IOT / torch2trt

An easy to use PyTorch to TensorRT converter
MIT License
4.56k stars 672 forks source link

AttributeError: 'Tensor' object has no attribute '_trt' #710

Open wxthu opened 2 years ago

wxthu commented 2 years ago

when I run yolov5x.pt model after converting to tensorRT using torch2trt, I found the following error and very confused image

200102200001 commented 2 years ago

There are some problems about the detect.py.

wxthu commented 2 years ago

There are some problems about the detect.py.

you mean that yolov5 series models could be supported by tensorrt?

wxthu commented 2 years ago

It looks as if yolov5 series models cannot be supported by tensorrt, right ?

LeoRainly commented 2 years ago

I had the same problem

awarebayes commented 2 years ago

Hello