每隔一定的epoch调整学习率 def adjust_learning_rate(optimizer, epoch): """Sets the learning rate to the initial LR decayed by 10 every 30 epochs""" lr = args.lr * (0.1 ** (epoch // 30)) for param_group in optimizer.param_groups: param_g
深度炼丹如同炖排骨一般,需要先大火全局加热,紧接着中火炖出营养,最后转小火收汁.本文给出炼丹中的 “火候控制器”-- 学习率的几种调节方法,框架基于 pytorch 1. 自定义根据 epoch 改变学习率. 这种方法在开源代码中常见,此处引用 pytorch 官方实例中的代码 adjust_lr def adjust_learning_rate(optimizer, epoch): """Sets the learning rate to the initial LR de
参考:https://pytorch.org/docs/master/optim.html#how-to-adjust-learning-rate torch.optim.lr_scheduler提供了几种方法来根据迭代的数量来调整学习率 自己手动定义一个学习率衰减函数: def adjust_learning_rate(optimizer, epoch, lr): """Sets the learning rate to the initial LR decayed by