eps696 / stargan2 Goto Github PK

Hello,
Where to download pretrained celebA HQ to run with test.py ?

I managed to get an RTX3090 which has 16G. With Tensorflow version of Clovaai, I managed to get a batch size of 6 on dataset 512x512 (eats up 18G GPU) and still training as it run extremely slowly (and seems heavy) compares to StyleGan V2 on Pytorch.

What do you think of your codes on batch size 6 would it be worth trying? My dataset is purely indoor scene and furniture, close to million of images.

Anyhow, would you make your code work on multi GPUs, please ?

Thanks,
Steve

Have you managed to run your code with multi-GPU ?

Thanks,
Steve

Do you broadcast this repo on Facebook or Reddit? I want to know how to access the wonderful repo for the first time.

Thank you for your interesting comments, I find them very meaningfull.

Can you please clarify what exactly are you trying to achieve by lowmem parameter as from my perspective it just removes a style vector (matrix ~ [Bs x StyleDim]), which is pretty lightweight.

It would be interesting to hear some numbers (like gpu mem savings), how it helps.

Thanks!👍🏻