Custom scheduler class?

1

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 step(self):
        ...

Is something like this possible? How do other people handle custom schedulers?

2

hey @jw3126

you might want to check out lr_scheduler_step hook.

Also, we have moved the discussions to GitHub Discussions. You might want to check that out instead to get quick response. The forums will be marked read-only after some time.

Thank you