NVIDIA / MinkowskiEngine

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

Home Page:https://nvidia.github.io/MinkowskiEngine

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Export Minkowski Nets to Torchscript

a-krawciw opened this issue · comments

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.

Hi, have you solve it ?