Conversion to Tensorrt - Githubissues

roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

MIT License

2.82k stars 567 forks source link

Conversion to Tensorrt #227

Open ma-siddiqui opened 4 years ago

ma-siddiqui commented 4 years ago

Hi, I want to convert this model to tensorrt. If someone has done please share guidance. It would be great help. In addition to this I am using torch2trt and facing issue while building cuda engine. Its throws error. Unused input. Kindly let me know the input and output names.

Thanks