TrainingArguments cosine_with_restarts: how to define the restart count?
Fine-tuning SegFormer. The arguments look like:training_args = TrainingArguments(
    output_dir=outputs_dir,
    overwrite_output_dir=True,
    learning_rate=lr,
    num_train_epochs=epochs,
    lr_scheduler_type="cosine_with_restarts",
    warmup_ratio=1.0 / epochs,
    # ...
)How do I specify the number of restarts for the scheduler?
2023-04-03T05:56:07Z
You can seethis git issue.