Skip to content

Releases: nyx-ai/stylegan2-flax-tpu

Food Platter 512 🍪🍰🍣🍹🍔

27 Oct 15:55
5f15806
Compare
Choose a tag to compare

This release contains a new 5-class 512x512 pretrained models with classes: burger, cheesecake, cocktail, cookie, sushi.

See samples here: https://nyx-ai.github.io/stylegan2-flax-tpu/

Model changelog:

  • Single 5-class model (burger/cheesecake/cocktail/cookie/sushi) instead of 1-class models
  • Resolution of 512x512 instead of 256x256
  • Trained for much longer: 8 days at 256x256 then 28 days at 512x512 instead of 10 hours
  • Trained on more data: 558k 512x512 images instead of 100k 256x256 images

Code changelog: see stylegan2-flax-tpu#-changelog

📊 Wandb training plots

Food Platter 256 🍪🍰🍣🍹

20 Jul 09:01
Compare
Choose a tag to compare

This release contains 4 256x256 pretrained models: cookie, cheesecake, sushi and cocktail.

See samples here: https://nyx-ai.github.io/stylegan2-flax-tpu/

Generate more images using the Colab notebook: https://colab.research.google.com/github/nyx-ai/stylegan2-flax-tpu/blob/master/notebook/image_generation.ipynb