NVIDIA / MinkowskiEngine

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

The model used contains deepcopy, and a TypeError: cannot pickle 'MinkowskiConvolutionFunction' object error was reported when using this project #519

Open Joker9194 opened 1 year ago

Joker9194 commented 1 year ago

Describe the bug I installed it according to the installation steps. I used ema in my project, which used deepcopy code, and this error occurred when the project was running. “TypeError: cannot pickle 'MinkowskiConvolutionFunction' object”


Expected behavior The project is running normally


Desktop (please complete the following information):

==========MinkowskiEngine========== 0.5.4 MinkowskiEngine compiled with CUDA Support: True NVCC version MinkowskiEngine is compiled: 11030 CUDART version MinkowskiEngine is compiled: 11030

Charlescai123 commented 1 year ago

The same issue happens to me when I try to use MinkowskiEngine for multi-GPU training work

Hlings commented 1 year ago

You can try build the model in the ModelEmaV2 class directly, that works for me. Eg: forward the config into the Ema class and build the model in the "init" rather than using deepcopy :)