_target_: transformers.get_cosine_schedule_with_warmup
num_warmup_steps: 2500
num_training_steps: 1000000