LR Schedulers shouldn't get `epoch` argument in `step` function
🐛 Bug
PyTorch LR schedulers now shouldn't get any arguments in step function, see here and here.
Looks like the calls in PytorchLightning are not in line with the new interface, see here.
This results in unexpected LR changes. Removing the epoch argument from step call solves the issue for me.
Environment
PyTorch 1.4
PyTorchLightning 0.5.3.2