UserWarning: To get the last learning rate computed by the scheduler, please use `get_last_lr()`

将 lr = scheduler.get_lr()

改为 lr = scheduler.get_last_lr()