can it support for mutil-gpu training ?

eladrich / pixel2style2pixel

Official Implementation for "Encoding in Style: a StyleGAN Encoder for Image-to-Image Translation" (CVPR 2021) presenting the pixel2style2pixel (pSp) framework

https://eladrich.github.io/pixel2style2pixel/

MIT License

3.19k stars 570 forks source link

can it support for mutil-gpu training ? #219

Closed vicentowang closed 2 years ago

vicentowang commented 2 years ago

my gpu memory is 8G and cannot run it even for batch size of 1 with 512x512 resolution. now i got 2 gpu of 8G，how can i get for training? thanks a lot.

yuval-alaluf commented 2 years ago

We do not support multi-gpu training but adding this feature should be possible. Please take a look at: https://github.com/eladrich/pixel2style2pixel/issues/154

vicentowang commented 2 years ago

still GPU memory is not enough for two GPU of 8G.