Open PatrickkZ opened 1 year ago
Thanks for your feedback, and I think this modification is reasonable, would you mind posting a PR to fix this?
Thanks for your feedback, and I think this modification is reasonable, would you mind posting a PR to fix this?
sure
Checklist
description
Hi, I'm trying to train the SoloV2 provide by
mmdet
on CPU with distributed mode. I know it's not recommended to train on CPU, but it's possible to do it? I change some code intools/train.py
to train the model on cpu. Then i got this errorAfter debugging, is there an inconsistency in MMDDP between CPU and GPU? because i use CPU, so the
device_ids
in MMDDP is None, then this line will be executed. if it's GPU, the input will bescatter
firstso i change the line 69 to
then this can works on CPU while training SoloV2. I want to know if it's right to do this? or how can I train with MMDDP on CPU cluster?
env