lr_cosine_annealing function

Set the learning rate of each parameter group using a cosine annealing schedule