File size: 802 Bytes
4455348 |
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 27 28 29 30 31 32 |
{
"model_name_or_path": "gsarti/it5-base",
"tokenizer_name": "gsarti/it5-base",
"dataset_name": "/content/drive/MyDrive/_colab/squad_code/my_squad_it",
"do_train": true,
"do_eval":false,
"push_to_hub" : false,
"per_device_train_batch_size": 3,
"per_device_eval_batch_size": 3,
"learning_rate": 3e-5,
"num_train_epochs": 3,
"max_seq_length": 384,
"doc_stride": 128,
"max_answer_length":30,
"n_best_size":5,
"save_steps": 500000,
"save_total_limit": 3,
"save_strategy":"steps",
"context_column": "context",
"question_column": "question",
"answer_column": "answers",
"output_dir": "./tmp/quanIta3-t5/",
"version_2_with_negative": true,
"eval_steps": 2000,
"evaluation_strategy": "steps",
"predict_with_generate":true,
"max_eval_samples": 1000
}
|