lucidrains / lightweight-gan

Implementation of 'lightweight' GAN, proposed in ICLR 2021, in Pytorch. High resolution image generations that can be trained within a day or two
MIT License
1.62k stars 220 forks source link

Any way to do multi-gpu over LAN? #110

Closed msglm closed 1 year ago

msglm commented 2 years ago

https://pytorch.org/docs/stable/distributed.html#tcp-initialization I see that pytorch can do this, but I don't know if or how this is implemented on lightweight-gan.