(optimizer, i_iter, total_iters)
return lr
Sets the learning rate to the initial LR divided by 5 at 60th, 120th and 160th epochs
Sets the learning rate to the initial LR divided by 5 at 60th, 120th and 160th epochs
Sets the learning rate to the initial LR divided by 5 at 60th 120th and 160th epochs
def adjust_learning_rate(optimizer, i_iter, total_iters):
    """Sets the learning rate to the initial LR divided by 5 at 60th, 120th and 160th epochs"""
    lr = lr_poly(args.learning_rate, i_iter, total_iters, args.power)
    optimizer.param_groups[0]['lr'] = lr
    return lr
def adjust_learning_rate ( optimizer , i_iter , total_iters ) : lr = lr_poly ( args . learning_rate , i_iter , total_iters , args . power ) optimizer . param_groups [ 0 ] [ 'lr' ] = lr return lr