autonomousvision / stylegan-xl

[SIGGRAPH'22] StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasets
MIT License
961 stars 113 forks source link

Steps required to train a StyleGan-XL for a custom dataset #114

Open MiguelCosta94 opened 11 months ago

MiguelCosta94 commented 11 months ago

What would be the steps required to train the network for a new dataset like the Visual Wake Words?

xhlho commented 8 months ago

Hello, have you completed the training on your own dataset? If it is implemented, please let me know what needs to be noted. Thank you in advance!

MiguelCosta94 commented 8 months ago

Sorry, I did not finish the training. I ended up using another GAN architecture.

xhlho commented 8 months ago

Thank you for your answer. I have another small request. Could you please let me know which open source code it is? Thank you!

MiguelCosta94 commented 8 months ago

I've used this framework to implement my GAN https://github.com/POSTECH-CVLab/PyTorch-StudioGAN/tree/master