fixes lr_scheduler warning

Commit

5 years ago

fixes lr_scheduler warning For more details, see https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate