convert model to tensorrt

Open darissa opened this issue 5 years ago • 2 comments

Hi,

Last time I have successfully run this code. Just curious if u have tried on convert it to tensorrt using torch2trt. Coz i have tried but encounter an error seems like due to weight of this model. But i am not sure about that. Have u tried on this? Thank you.

Nov 09 '20 03:11 darissa

Hi, No, I have not tried to convert pytorch -> tensorrt. Another possible option is to try pytorch -> ONNX -> tensorrt conversion.

Nov 10 '20 08:11 dmburd

ok. Thank you for response. I will find way.

Nov 10 '20 08:11 darissa