Scheduler for momentum?

I know that one can use something like torch.optim.lr_scheduler.LambdaLR to schedule lr.
Is there any way to schedule momentum hyperparameter, like in SGD for example?

With a bit of work, the pytorch schedulers can be extended to handle momentum as well. I’ve had a go implementing a version of LambdaLR that includes momentum, along with a few other scheduler types.

Is this the sort of thing you are looking for?

Alternatively, the pytorch/ignite project seems to cover this use case: https://github.com/pytorch/ignite