training_args.json

{"model_path": "bert-base-uncased", "output_dir": "./finetuned_kaggle/", "data_dir": "./datasets/kaggle/", "max_seq_length": 128, "epoch": 5, "local_rank": -1, "task_name": "kaggle", "overwrite_cache": false, "learning_rate": 5e-05, "adam_epsilon": 1e-08, "train_batch_size": 16, "eval_batch_size": 64, "weight_decay": 0.1, "num_labels": 4, "train": true, "eval": true, "n_gpu": 1, "device": "cuda"}