train_arguments.json

{"num_epochs": 1, "train_batch_size": 64, "evaluation_steps": 1000, "scheduler": "WarmupLinear", "warmup_steps": 144, "optimizer_params": {"lr": 4e-05}, "use_amp": true, "weight_decay": 0.01}