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.

  
      
      GitHub
  
  
    

timesler/lr-momentum-scheduler 60

Pytorch implementation of arbitrary learning rate and momentum schedules, including the One Cycle Policy - timesler/lr-momentum-scheduler


  
  
    
    
  
  


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 31