How much gpu memory is required to run the full UGATIT code?

taki0112 / UGATIT

Official Tensorflow implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation (ICLR 2020)

MIT License

6.17k stars 1.04k forks source link

How much gpu memory is required to run the full UGATIT code? #125

Open xx-xuan opened 1 year ago

xx-xuan commented 1 year ago

I failed to use the gpu device with 8G memory to run the full UGATIT code. How can I modify the code to use two gpu devices?