search

okankop / Efficient-3DCNNs

PyTorch Implementation of "Resource Efficient 3D Convolutional Neural Networks", codes and pretrained models.
MIT License
773 stars 149 forks source link

Can I convert the 3DCNN to tensorrt? #16

Open Nioolek opened 4 years ago

Nioolek commented 4 years ago

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

okankop commented 4 years ago

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

Nioolek commented 4 years ago

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