train_arguments.json

{"num_epochs": 2, "train_batch_size": 64, "evaluation_steps": 1000, "scheduler": "WarmupLinear", "warmup_steps": 288, "optimizer_params": {"lr": 2e-05}, "use_amp": true, "weight_decay": 0.05}