Defaults for image training.
()
| 41 | |
| 42 | |
| 43 | def model_and_diffusion_defaults(): |
| 44 | """ |
| 45 | Defaults for image training. |
| 46 | """ |
| 47 | res = dict( |
| 48 | image_size=64, |
| 49 | num_channels=128, |
| 50 | num_res_blocks=2, |
| 51 | num_heads=4, |
| 52 | num_heads_upsample=-1, |
| 53 | num_head_channels=-1, |
| 54 | attention_resolutions="16,8", |
| 55 | channel_mult="", |
| 56 | dropout=0.0, |
| 57 | class_cond=False, |
| 58 | use_checkpoint=False, |
| 59 | use_scale_shift_norm=True, |
| 60 | resblock_updown=False, |
| 61 | use_fp16=False, |
| 62 | use_new_attention_order=False, |
| 63 | ) |
| 64 | res.update(diffusion_defaults()) |
| 65 | return res |
| 66 | |
| 67 | |
| 68 | def classifier_and_diffusion_defaults(): |
no test coverage detected