pytorch查看应用指数衰减后的学习率

在pytorch中使用指数衰减后,想着将每轮训练后的学习率打印出来,看指数衰减是否起作用了,于是在每轮训练后通过下面代码查看学习率:

print("lr:",net.lr)

发现不管训练多少轮,打印出来的学习率都为初始设定的值,但是检查发现指数衰减的使用是没有问题的,最后发现需要通过下面代码来查看应用指数衰减后的学习率:

print("lr:",net.scheduler.get_lr())

猜你喜欢

转载自blog.csdn.net/weixin_38314865/article/details/106032250
今日推荐