facebookresearch / swav

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882
Other
2.01k stars 280 forks source link

multiple GPUs #87

Open AziMtmd opened 3 years ago

AziMtmd commented 3 years ago

Hi,

Thank you very much for releasing your code. I wanted to ask since you are using 'torch.distributed', is it necessary to have multiple GPUs to use your code? I only have one GPU. Thank you,