断点恢复训练,LRScheduler如何更新


#1

在gluoncv中训练yolov3,载入断点保存的模型后,learning rate还是从epoch0开始。不知道gluon.Trainer.step()的更新机制是什么样的。

目前是直接set的:

lr_scheduler = LRScheduler(...)
trainer.set_learning_rate(lr_schedule(epoch*num_batches+ idx_batch))

有没有更方便的方法