Export Minkowski Nets to Torchscript

NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

https://nvidia.github.io/MinkowskiEngine

Other

2.43k stars 360 forks source link

Export Minkowski Nets to Torchscript #550

Open a-krawciw opened 1 year ago

a-krawciw commented 1 year ago

Is your feature request related to a problem? Please describe. I would like to integrate a MinkUnet model into an existing C++ pipeline. Currently this fails at the MinkowskiEngine statements

Describe the solution you'd like It would be great if either torch.jit.script or torch.script.trace could be supported. I'm not completely sure how much effort this would be, but it might be quite a bit.

cama36 commented 6 months ago

Hi, have you solve it ?