roytseng-tw / Detectron.pytorch

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Conversion to Tensorrt

ma-siddiqui opened this issue · comments

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