I guess the initial learning rate of the optimizer is too high and you might lower it.
The warning is raised, since it’s a new design decision to call the scheduler after the first optimization, such that the first step will use the user-defined learning rate from the optimizer.