hub / github.com/ImprintLab/MedSegDiff / create_classifier

Function create_classifier

guided_diffusion/script_util.py:257–293 · 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,
)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

create_classifier_and_diffusionFunction · 0.85

Calls 1

EncoderUNetModelClass · 0.85

Tested by

no test coverage detected