| { |
| "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" |
| } |
| } |
