Skip to content

Food Platter 512 πŸͺπŸ°πŸ£πŸΉπŸ”

Latest
Compare
Choose a tag to compare
@MasterScrat MasterScrat released this 27 Oct 15:55
· 2 commits to main since this release
5f15806

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