File size: 613 Bytes
93a40be
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
{
  "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": "TunedModel"
}