《深度学习笔记》——Pytorch调整学习率

Posted 2021-08-28 songyuc

1 定义调整学习率函数

def adjust_learning_rate(optimizer, epoch, lr):
    """Sets the learning rate to the initial LR decayed by 10 every 2 epochs"""
    # optimizer表示优化器对象
    lr *= (0.1 ** (epoch // 2))
    for param_group in optimizer.param_groups:
        param_group['lr'] = lr