Defaults for classifier models.
()
| 25 | |
| 26 | |
| 27 | def classifier_defaults(): |
| 28 | """ |
| 29 | Defaults for classifier models. |
| 30 | """ |
| 31 | return dict( |
| 32 | image_size=64, |
| 33 | classifier_use_fp16=False, |
| 34 | classifier_width=128, |
| 35 | classifier_depth=2, |
| 36 | classifier_attention_resolutions="32,16,8", # 16 |
| 37 | classifier_use_scale_shift_norm=True, # False |
| 38 | classifier_resblock_updown=True, # False |
| 39 | classifier_pool="attention", |
| 40 | ) |
| 41 | |
| 42 | |
| 43 | def model_and_diffusion_defaults(): |
no outgoing calls
no test coverage detected