CoMAE icon indicating copy to clipboard operation
CoMAE copied to clipboard

Published 20 hours ago • verified publisher icon MCG-NJU

Multi gpus pre-training

Open ShaohuaDong2021 opened this issue 1 year ago • 1 comments

Hi, can you tell me how to use multi gpus to train the model? When I runing "python main_pretrain_cpc.py", it only uses the first gpu. Thanks a lot.

ShaohuaDong2021 avatar May 14 '23 22:05 ShaohuaDong2021