search

lucidrains / pixel-level-contrastive-learning

Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch
MIT License
252 stars 27 forks source link

about distributed #6

Closed datuizhuang closed 3 years ago

datuizhuang commented 3 years ago

hi, thanks for your work, does the code support DistributedDataParallel/DataParallel or not?

lucidrains commented 3 years ago

@datuizhuang it should work the same as https://github.com/lucidrains/byol-pytorch/tree/master/examples/lightning so yes, just use pytorch lightning!