scheduler.step()用法

for i in range(10):

    lr1=lr

    scheduler.step()

    lr2=scheduler.get_last_lr()

    print(lr1,lr2)

lr是最开始的学习率。

输出:

0.0001 [0.0001]
0.0001 [0.0001]
0.0001 [1e-05]
0.0001 [1e-05]
0.0001 [1e-05]
0.0001 [1.0000000000000002e-06]

我设置的是3次一变学习率，step相当于计数