highdeffrepo / tokenizer_config.json
highdeff's picture
Upload 4 files
93a40be
raw
history blame contribute delete
613 Bytes
{
"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"
}