Spconv with torchscript

jit
1

Hi,

I’m having trouble using torchscript with my model which uses spconv. I need spconv, since I’m working with large sparse matrices. However,

traced_script_module= torch.jit.trace(model, example)

throws the following error:

RuntimeError: Found an unsupported argument type c10::List<at::Tensor> in the JIT tracer.

I’m also getting a lot of transformation errors like:

TracerWarning: Converting a tensor to a NumPy array might cause the trace to be incorrect.

Any Idea what could cause this?

Thanks.