Defaults for image training.
()
| 9 | |
| 10 | |
| 11 | def model_and_diffusion_defaults(): |
| 12 | """ |
| 13 | Defaults for image training. |
| 14 | """ |
| 15 | return dict( |
| 16 | image_size=64, |
| 17 | num_channels=128, |
| 18 | num_res_blocks=2, |
| 19 | num_heads=4, |
| 20 | num_heads_upsample=-1, |
| 21 | attention_resolutions="16,8", |
| 22 | dropout=0.0, |
| 23 | learn_sigma=False, |
| 24 | sigma_small=False, |
| 25 | class_cond=False, |
| 26 | diffusion_steps=1000, |
| 27 | noise_schedule="linear", |
| 28 | timestep_respacing="", |
| 29 | use_kl=False, |
| 30 | predict_xstart=False, |
| 31 | rescale_timesteps=True, |
| 32 | rescale_learned_sigmas=True, |
| 33 | use_checkpoint=False, |
| 34 | use_scale_shift_norm=True, |
| 35 | ) |
| 36 | |
| 37 | |
| 38 | def create_model_and_diffusion( |
no outgoing calls
no test coverage detected