Custom scheduler
I would like to provide my own learning rate scheduler. What I would love is do is doing this in the lightning style, e.g. implementing some hooks:
class MyScheduler(pl.LightningScheduler):
    ...
    def on_step(self...):
        ...
    def on_epoch(self...):
        ...
Is something like this possible? How do other people handle custom schedulers?
PS: I asked this question before on the [deprecated forum.]
(https://forums.pytorchlightning.ai/t/custom-scheduler-class/1238)
can you please elaborate on the usage more? like what do you want to do inside on_step and on_epoch methods?