Using Cosine LR scheduler via TrainingArguments in Trainer

Thank you so much @edwarddgao. It’s working perfectly.