Support for Scheduler taking a value to the step function
Looking to the code in training_loop.py it seems like the only scheduler that can take values to the step function is ReduceLROnPlateau, however, there is CosineAnnealingWarmRestarts scheduler and custom schedulers that can take epoch/step id or any other value to the step function.

  
    
      pytorch-lightning/pytorch_lightning/trainer/training_loop.py
    
    
         Line 1148
      in
      9ab0715
    
  
  
    

        
          
           if lr_scheduler['reduce_on_plateau']: