Efficient-3DCNNs icon indicating copy to clipboard operation
Efficient-3DCNNs copied to clipboard

Can I convert the 3DCNN to tensorrt?

Open Nioolek opened this issue 4 years ago • 2 comments

Thank you for your contribution. Can I convert the 3DCNN to tensorrt?

Nioolek avatar Apr 07 '20 09:04 Nioolek

I did not try tensorrt on these models. However, if the layers in the models are supported by tensorrt, you can convert them.

okankop avatar Apr 07 '20 09:04 okankop

I did not try tensorrt on these models. However, if the layers in the models are supported by tensorrt, you can convert them.

Thank you. I tried to convert 3DCNN to tensorrt and onnx. When I tried onnx, the problem is the result of pytorch and onnx is different. I think it's because of the differennt between pytorch op and onnx op. I do not not how to solve the problem. When I tried to convert to tensorrt, the problem is that there is not a official implement to convert PYTORCH 3DCONV/3D BN to TENSORRT. If anyone succeeds, I would like to know how to implement it

Nioolek avatar Apr 07 '20 10:04 Nioolek