INNER CODE UNIT · Python

curr_lr

OpenMeshLab/MeshXL · engine.py:17

        curr_lr = args.final_lr + 0.5 * (args.base_lr - args.final_lr) * (
            1 + math.cos(math.pi * curr_iter / max_iters)
        )
    return curr_lr



def adjust_learning_rate(args, optimizer, curr_iter, max_iters):
    curr_lr = compute_learning_rate(args, curr_iter, max_iters)
    for param_group in optimizer.param_groups:
        param_group["lr"] = curr_lr
    return curr_lr



def do_train(
    args,
    model,

View source record →