torch.distributed ---> torchrun

facebookresearch / vicreg

VICReg official code base

MIT License

525 stars 88 forks source link

Closed BalajiAI closed 1 year ago

BalajiAI commented 1 year ago

torch.distributed is deprecated and it throws off an error. So kindly change it to torchrun.