{
  "training_args": {
    "output_dir": "./output",
    "overwrite_output_dir": true,
    "num_train_epochs": 3,
    "per_device_train_batch_size": 16,
    "per_device_eval_batch_size": 64,
    "save_steps": 10000,
    "save_total_limit": 1
  },
  "model_args": {
    "num_labels": 2,
    "hidden_dropout_prob": 0.1,
    "hidden_size": 768,
    "num_hidden_layers": 12,
    "num_attention_heads": 12,
    "max_position_embeddings": 512
  },
  "data_args": {
    "train_file": "./data/train.csv",
    "test_file": "./data/test.csv",
    "validation_file": "./data/val.csv"
  },
  "tokenizer_name": "config.json"
}