YOLOF icon indicating copy to clipboard operation
YOLOF copied to clipboard

How can i run this project in a single gpu

Open KongBaiHC opened this issue 3 years ago • 1 comments

I delete IMS_PER_DEVICE, and change IMS_PER_BATCH from 64 to 16 and use pods_train --num_gpu 1 to train. Then Default process group is not initialized is reported. It seems to YOLOF use SyncBN, but I didn't find it in config.py. Thanks.

KongBaiHC avatar Aug 19 '21 02:08 KongBaiHC

YOLOF does not use SyncBN by default. This error is caused by the calculation of num_foreground. You can replace it with the new version to support training with one GPU.

chensnathan avatar Aug 19 '21 09:08 chensnathan