GPUs

https://www.chrisplaysgames.com/gadgets/2019/02/26/training-at-home-and-in-the-cloud/ Es un blog, donde el autor probó stylegan2 en su casa y en google cloud. Compara distintas GPUs. Les digo un par de cosas que me parecieron interesantes.

La “mejor” que se probó fue V100:

Y cuando compara V100 con GTX 1080:

Para 60 iteraciones, el tiempo que tarda GTX 1080 es 44 horas, mientras que V100 necesitó 18 horas.

La diferencia entre estas pruebas y las que realizaremos nosotras están en que él entrenó con muchas fotos y consideró “aceptable” el resultado cuando se obtienen fotos que sin hacer zoom parecen coherentes. Lo cual es lo contrario a lo que nos vamos a enfocar nosotras, porque me imagino que entrenaremos con pocas fotos y el resultado que queremos es fotos de alta resolución con caras que no parezcan creadas por stylegan2. Supongo que vamos a hacer más iteraciones pero con menos fotos.

En el readme del repositorio del stylegan2 me pareció útil la siguiente tabla. Probando con NVIDIA DGX-1 con hasta 8 GPUs Tesla V100. Config-f y config-e son redes que ya fueron pre entrenadas con datasets de stylegan2: