stylegan2-pytorch
stylegan2-pytorch copied to clipboard
Nice work! Any GPU memory consume suggestions?
Original version train a 1024 input size model on 8 GPUs, will OOM.
I have 4 GPUs, all with 10G memory, what's the max input size I can train?
Wait for me to finish https://github.com/lucidrains/lightweight-gan ;)
@lucidrains look promising! waiting for your results!
@jinfagang its done, give it a try :)