convert model to tensorrt

dmburd / S-DCNet

Unofficial Pytorch implementation of S-DCNet and SS-DCNet

17 stars 9 forks source link

convert model to tensorrt #10

Open darissa opened 3 years ago

darissa commented 3 years ago

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.

dmburd commented 3 years ago

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

darissa commented 3 years ago

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