torch2trt icon indicating copy to clipboard operation
torch2trt copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA-AI-IOT

How to convert the pytorch model (.pt) into the TRT engine file (.engine)

Open qin53027 opened this issue 3 years ago • 1 comments

Hi,

I have installed this torch2trt and still cannot get the instruction how to covert .pt file into the .engine file so that I can use that in the DeepStream apps.

Now I have 4light_test_ep4.pt (resnet model). Please let me know how I can convert this file into. 4light_test_ep4.engine format?

Thanks so much Jimmy

qin53027 avatar Jan 13 '22 16:01 qin53027

I have similar question. Can anyone suggest for us?

DavidNguyen95 avatar Feb 16 '22 14:02 DavidNguyen95