training_args.json

{
  "output_dir": "./results",
  "num_train_epochs": 5,
  "per_device_train_batch_size": 32,
  "per_device_eval_batch_size": 64,
  "learning_rate": 2e-05,
  "weight_decay": 0.01,
  "evaluation_strategy": "epoch",
  "save_strategy": "epoch",
  "load_best_model_at_end": true,
  "metric_for_best_model": "f1",
  "fp16": true
}