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