Giters

WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

update mmcv to solve out of memory

marearth opened this issue · comments

commented

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.