欢迎光临散文网 会员登陆 & 注册

深度学习:optimizer.step()和scheduler.step()的区别——转

2020-12-28 21:42 作者:囹之圄  | 我要投稿

optimizer.step()通常用在每个mini-batch之中,而scheduler.step()通常用在epoch里面,但是不绝对,可以根据具体的需求来做。只有用了optimizer.step(),模型才会更新,而scheduler.step()是对lr进行调整。通常我们有

在scheduler的step_size表示scheduler.step()每调用step_size次,对应的学习率就会按照策略调整一次。所以如果scheduler.step()是放在mini-batch里面,那么step_size指的是经过这么多次迭代,学习率改变一次。


深度学习:optimizer.step()和scheduler.step()的区别——转的评论 (共 条)

分享到微博请遵守国家法律