search

jbwang1997 / OBBDetection

OBBDetection is an oriented object detection library, which is based on MMdetection.
Apache License 2.0
522 stars 112 forks source link

多显卡训练错误 #59

Closed gys1287009045 closed 2 years ago

gys1287009045 commented 2 years ago

您好，在多显卡训练时我报了如下错误： RuntimeError: replicas_[0].size() == rebuilt_paramindices.size() INTERNAL ASSERT FAILED at "/opt/conda/conda-bld/pytorch_1627336327116/work/torch/lib/c10d/reducer.cpp":1376, please report a bug to PyTorch. rebuilt parameter indices size is not same as original model parameters size.Original model param size is: 156 versus rebuilt params size of: 189852 image

jbwang1997 commented 2 years ago

我在pytorch的issue查到了这个回答，这个可能是pytorch的版本导致的

gys1287009045 commented 2 years ago

好的谢谢，我尝试一下