hub / github.com/ImprintLab/MedSegDiff / create_classifier_and_diffusion

Function create_classifier_and_diffusion

guided_diffusion/script_util.py:216–254 · view source on GitHub ↗

(
    image_size,
    classifier_use_fp16,
    classifier_width,
    classifier_depth,
    classifier_attention_resolutions,
    classifier_use_scale_shift_norm,
    classifier_resblock_updown,
    classifier_pool,
    learn_sigma,
    diffusion_steps,
    noise_schedule,
    timestep_respacing,
    use_kl,
    predict_xstart,
    rescale_timesteps,
    rescale_learned_sigmas,
)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 2

create_classifierFunction · 0.85

create_gaussian_diffusionFunction · 0.85

Tested by

no test coverage detected