File size: 1,360 Bytes
33b288a | 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 33 34 35 36 37 38 39 40 41 42 | {
"monitor_metric": "eval_MRR@NM",
"trainer_kwargs": {
"class_name": "IR"
},
"trainee_kwargs": {
"class_name": "DPRBiEncoder",
"loss": {
"class_name": "NLLLoss"
}
},
"data_module_kwargs": {
"class_name": "mono_modal_DataModule",
"data_processor": {
"class_name": "evqa_data_processor",
"dataset_path": "../../data/evqa/",
"kb_path": "../../data/evqa/passages",
"entity_kb_path": "../../data/evqa/kb"
},
"use_image": false,
"relevant_indices_key": "BM25_provenance_indices",
"irrelevant_indices_key": "BM25_irrelevant_indices",
"input_key": "question",
"passage_key": "passage",
"dataloader_kwargs": {
"num_workers": 6,
"prefetch_factor": 2
}
},
"question_model_kwargs": {
"class_name": "DPRQuestionEncoder",
"checkpoint_name": "question_model",
"pretrained_model_name_or_path": "/home/data/meerqat/my_transformers_cache/dpr_question_encoder_triviaqa_without_viquae",
"inference_path": "saved_models/DPR_evqa_1/question_model"
},
"context_model_kwargs": {
"class_name": "DPRContextEncoder",
"checkpoint_name": "context_model",
"pretrained_model_name_or_path": "/home/data/meerqat/my_transformers_cache/dpr_context_encoder_triviaqa_without_viquae",
"inference_path": "saved_models/DPR_evqa_1/context_model"
}
} |