| { |
| "query_token_id": "[unused0]", |
| "doc_token_id": "[unused1]", |
| "query_token": "[Q]", |
| "doc_token": "[D]", |
| "ncells": null, |
| "centroid_score_threshold": null, |
| "ndocs": null, |
| "load_index_with_mmap": false, |
| "index_path": null, |
| "nbits": 1, |
| "kmeans_niters": 20, |
| "resume": false, |
| "similarity": "cosine", |
| "bsize": 8, |
| "accumsteps": 1, |
| "lr": 1e-05, |
| "maxsteps": 400000, |
| "save_every": null, |
| "warmup": 20000, |
| "warmup_bert": null, |
| "relu": false, |
| "nway": 1, |
| "use_ib_negatives": true, |
| "reranker": false, |
| "distillation_alpha": 1.0, |
| "ignore_scores": false, |
| "model_name": null, |
| "query_maxlen": 32, |
| "attend_to_mask_tokens": false, |
| "interaction": "colbert", |
| "dim": 128, |
| "doc_maxlen": 180, |
| "mask_punctuation": true, |
| "checkpoint": "ckpts/colbertv2.0", |
| "triples": "/future/u/okhattab/root/unit/experiments/2021.10/downstream.distillation.round2.2_score/round2.nway6.cosine.ib/examples.64.json", |
| "collection": "/future/u/okhattab/data/MSMARCO/collection.tsv", |
| "queries": "/future/u/okhattab/data/MSMARCO/queries.train.tsv", |
| "index_name": null, |
| "overwrite": false, |
| "root": "/future/u/okhattab/root/unit/experiments", |
| "experiment": "2021.10", |
| "index_root": null, |
| "name": "kldR2.nway64.ib", |
| "rank": 0, |
| "nranks": 4, |
| "amp": true, |
| "gpus": 8 |
| } |
