openai / guided-diffusion

MIT License
6.06k stars 807 forks source link

What were the training flags used for the released ImageNet models? #11

Open crowsonkb opened 3 years ago

crowsonkb commented 3 years ago

They don't seem to be documented either in this repo or in improved-diffusion. I'd like to match the training flags as closely as possible when fine-tuning (I'm trying to fine-tune the 512x512 ImageNet model).