def METHOD_NAME(epoch):
    lr_adj = 1.
    if epoch < warmup_epochs:
        lr_adj = 1. / workers * (epoch * (workers - 1) / warmup_epochs + 1)
    else:
        decay_schedule.sort(reverse=True)
        for e in decay_schedule:
            if epoch >= e:
                lr_adj *= alpha
    return lr_adj