search

raoyongming / DenseCLIP

[CVPR 2022] DenseCLIP: Language-Guided Dense Prediction with Context-Aware Prompting
505 stars 38 forks source link

Single GPU error #24

Closed Virgilzz closed 2 years ago

Virgilzz commented 2 years ago

Hi，I've modified the settings for single or multiple GPUs

norm_cfg = dict(type='BN', requires_grad=True)

But there were still such errors. ![Uploading image.png…]() Does that mean I need to modify the training section in the mmseg source?

Virgilzz commented 2 years ago

gitgubbb

raoyongming commented 2 years ago

Hi @Virgilzz, thanks for your interest in our work.

It seems the error comes from the SyncBN layer. We convert all batch norm layers to SyncBN here, which only supports DistributedDataParallel (DDP) environment. You may need to comment out this line if you want to train the model on a single GPU or without DDP.

Virgilzz commented 2 years ago

It has been solved. Thank you!