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

Method __init__

guided_diffusion/train_util.py:34–123  ·  view source on GitHub ↗
(
        self,
        *,
        model,
        classifier,
        diffusion,
        data,
        dataloader,
        batch_size,
        microbatch,
        lr,
        ema_rate,
        log_interval,
        save_interval,
        resume_checkpoint,
        use_fp16=False,
        fp16_scale_growth=1e-3,
        schedule_sampler=None,
        weight_decay=0.0,
        lr_anneal_steps=0,
    )

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 5

_load_optimizer_stateMethod · 0.95
_load_ema_parametersMethod · 0.95
UniformSamplerClass · 0.85

Tested by

no test coverage detected