What GPU was it trained on?

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

What GPU was it trained on? #3

Closed hexicle closed 5 years ago

hexicle commented 5 years ago

I am training it on an NVIDIA Tesla K80 but it runs out of memory even when I put the --light True parameter.

taki0112 commented 5 years ago

One NVIDIA V100 32GB GPU. It takes 4 days.

hexicle commented 5 years ago

Thank you, I ran out of memory because the process was running and I needed to reboot the server to free up the memory. It is not a recurring problem.

FantasyJXF commented 5 years ago

One NVIDIA V100 32GB GPU. It takes 4 days.

Hello, did you train with light?