octree-nn / ocnn-pytorch

Octree-based Sparse Convolutional Neural Networks
MIT License
150 stars 16 forks source link

deploy #32

Closed cama36 closed 4 months ago

cama36 commented 4 months ago

Hi, thank you very much for your work! As we all know, the deployment of sparse convolution consistently is a very difficult problem.But in o-cnn all the ops are pytorch supported, so I would like to ask you if you have tried to export o-cnn to onnx? and if so, I would very much appreciate if you could provide an example, thanks!

wang-ps commented 4 months ago

Sorry I have not tried to export o-cnn to onnx. Theoretically, it would be easy, since ocnn is implemented with pure Pytorch.