stylegan2-pytorch icon indicating copy to clipboard operation
stylegan2-pytorch copied to clipboard

Nice work! Any GPU memory consume suggestions?

Open lucasjinreal opened this issue 4 years ago • 3 comments

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?

lucasjinreal avatar Nov 15 '20 03:11 lucasjinreal

Wait for me to finish https://github.com/lucidrains/lightweight-gan ;)

lucidrains avatar Nov 15 '20 03:11 lucidrains

@lucidrains look promising! waiting for your results!

lucasjinreal avatar Nov 15 '20 05:11 lucasjinreal

@jinfagang its done, give it a try :)

lucidrains avatar Nov 17 '20 00:11 lucidrains