MCPcopy Create free account
hub / github.com/ImprintLab/MedSegDiff / __init__

Method __init__

guided_diffusion/fp16_util.py:149–171  ·  view source on GitHub ↗
(
        self,
        *,
        model,
        use_fp16=False,
        fp16_scale_growth=1e-3,
        initial_lg_loss_scale=INITIAL_LOG_LOSS_SCALE,
    )

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 3

get_param_groups_and_shapesFunction · 0.85
make_master_paramsFunction · 0.85
convert_to_fp16Method · 0.45

Tested by

no test coverage detected