Defaults for image and classifier training.
()
| 9 | |
| 10 | |
| 11 | def diffusion_defaults(): |
| 12 | """ |
| 13 | Defaults for image and classifier training. |
| 14 | """ |
| 15 | return dict( |
| 16 | learn_sigma=False, |
| 17 | diffusion_steps=1000, |
| 18 | noise_schedule="linear", |
| 19 | timestep_respacing="", |
| 20 | use_kl=False, |
| 21 | predict_xstart=False, |
| 22 | rescale_timesteps=False, |
| 23 | rescale_learned_sigmas=False, |
| 24 | ) |
| 25 | |
| 26 | |
| 27 | def classifier_defaults(): |
no outgoing calls
no test coverage detected