Fixed parameter scheduler bug with CosineAnnealingWarmRestarts
#2938
+107
−7
CosineAnnealingWarmRestarts
#2938