MCPcopy Index your code
hub / github.com/openai/guided-diffusion / set_annealed_lr

Function set_annealed_lr

scripts/classifier_train.py:170–173  ·  view source on GitHub ↗
(opt, base_lr, frac_done)

Source from the content-addressed store, hash-verified

168
169
170def set_annealed_lr(opt, base_lr, frac_done):
171 lr = base_lr * (1 - frac_done)
172 for param_group in opt.param_groups:
173 param_group["lr"] = lr
174
175
176def save_model(mp_trainer, opt, step):

Callers 1

mainFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected