search

VCIP-RGBD / DFormer

[ICLR 2024] DFormer: Rethinking RGBD Representation Learning for Semantic Segmentation
https://yinbow.github.io/Projects/DFormer/index.html
MIT License
142 stars 24 forks source link

Single GPU training #2

Closed lili0620 closed 1 year ago

lili0620 commented 1 year ago

Thanks for your excellent work! I melt the issue when training the model with a single 3090 GPU. I cannot sort out the source of the issue.

RuntimeError: Default process group has not been initialized, please make sure to call init_process_group.

yinbow commented 1 year ago

I was using 'syncBN' in the model, but I hadn't switched it to standard BN when working with a single GPU. I've fixed that now. Thanks for the reminder.