matthias-wright/flaxmodels

How to train styleGAN on Colab TPU for custom dataset?

waleedrazakhan92 opened this issue · 2 comments

waleedrazakhan92 commented

How do i run the training on my custom dataset while utilizing colab TPU.

matthias-wright commented

Hi @waleedrazakhan92, I don't have access to TPUs right now. Check out this repository. The author's adopted the flaxmodels stylegan2 implementation for TPU training. Hope that helps!

  • ❤️1
waleedrazakhan92 commented

@matthias-wright thankyou for clearing this.