search

NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
https://nvidia.github.io/MinkowskiEngine
Other
2.47k stars 367 forks source link

Compatibility with multiprocess #525

Open jrockholt3 opened 1 year ago

jrockholt3 commented 1 year ago

I am trying to run a distributed learning system with a copy of my network on multiple cpu's using pythons multiprocessing pool.map. I always get error: cannot pickle "MinkowskiConvolutionFunction" object.

Is there a way to solve this?

marina-gonzalezalvarez commented 10 months ago

Have you managed to resolve this issue, @jrockholt3 ?

zhixuanli commented 5 months ago

Still have this problem. Could anyone give some suggestions? Thanks!