File size: 867 Bytes
2f044c1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
train_dataset_path: "/root/relik-sapienzanlp/data/reader/retriever-relik-entity-linking-aida-wikipedia-base-question-encoder/train_windowed_candidates.jsonl"
val_dataset_path: "/root/relik-sapienzanlp/data/reader/retriever-relik-entity-linking-aida-wikipedia-base-question-encoder/testa_windowed_candidates.jsonl"
train_dataset:
_target_: "relik.reader.data.relik_reader_data.RelikDataset"
transformer_model: "${model.model.transformer_model}"
materialize_samples: False
shuffle_candidates: 0.5
random_drop_gold_candidates: 0.05
noise_param: 0.0
for_inference: False
tokens_per_batch: 4096
special_symbols: null
val_dataset:
_target_: "relik.reader.data.relik_reader_data.RelikDataset"
transformer_model: "${model.model.transformer_model}"
materialize_samples: False
shuffle_candidates: False
for_inference: True
special_symbols: null
|