I want to keep lr at 5e-4 before N epoch, and then use cosineAnnealingLR in the rest of training. What should I do?