WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
Other
1.7k stars 306 forks source link

update mmcv to solve out of memory #202

Closed marearth closed 2 years ago

marearth commented 2 years ago

Older version mmcv 0.2.16 tends to consume more GPU memory in the pytorch distributed learning. It's easy to lead to out-of -memory. Older implementation of MMDistributedDataParallel mmcv module is not efficient, please use latest mmcv.