How do I incorporate a scheduler with "step" and "batch step"?
Hi,
I am trying to incorporate the scheduler provided here.
I went through the documentation of PyTorch lightning and it allows only scheduler interval "epoch" or "step". But in the scheduler attached, it needs to call "step" every epoch and  "batch step" after each iteration. Could someone help me figure out a way to do this? Any help will be appreciated.
Thank you,
Best,
Shreyas Kamath
It is currently not supported that a scheduler can step on both batch and epoch. The scheduler you are linking to are also different from a standard scheduler since it both has an step method and a batch_step method, where lightning only supports the step method.
To get this working in lightning, you would need to split the scheduler into two new schedulers that both operate on the same optimizer: one that steps on batch and one the steps on epoch.