search

facebookresearch / moco

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722
MIT License
4.82k stars 794 forks source link

One question about single GPU #133

Open Sangh0 opened 2 years ago

Sangh0 commented 2 years ago

Hi, thank you for your implementations.

I have a question about single GPU. Can I use a single gpu when use shuffle BN?

Thank you.

ly9802 commented 1 year ago

I also have a single GPU, how to implement the code by using single GPU?

Sangh0 commented 1 year ago

I tried using torch.randperm() when use shuffle BN with single GPU in training roof. I refered to this repository https://github.com/leftthomas/MoCo