Cosine annealed warm restart learning schedule