Allow CosineSchedule to be a schedule that gradually increases.
Allow CosineSchedule to be a schedule that gradually increases.

PiperOrigin-RevId: 366938093

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo