File size: 1,633 Bytes
c8116d8 |
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 43 44 45 46 47 48 |
{
"nprobe": 2,
"ncandidates": 8192,
"index_path": null,
"nbits": 1,
"kmeans_niters": 20,
"similarity": "cosine",
"bsize": 8,
"accumsteps": 1,
"lr": 1e-5,
"maxsteps": 400000,
"save_every": null,
"resume": false,
"warmup": 20000,
"warmup_bert": null,
"relu": false,
"nway": 64,
"use_ib_negatives": true,
"query_maxlen": 32,
"attend_to_mask_tokens": false,
"dim": 128,
"doc_maxlen": 180,
"mask_punctuation": true,
"checkpoint": "\/future\/u\/okhattab\/root\/unit\/experiments\/2021.10\/none\/kld.v3.nway32.l2.ib\/checkpoints\/colbert-150000\/",
"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,
"meta": {
"hostname": "future-hgx-1.stanford.edu",
"git_branch": "api_residual_compression3_KLD",
"git_hash": "62198f65e45bc7887eda4fdd57f9d3a103c72a7b",
"git_commit_datetime": "2021-10-29 07:05:32-07:00",
"current_datetime": "Oct 29, 2021 ; 9:05AM PDT (-0700)",
"cmd": "\/future\/u\/okhattab\/repos\/QolBERT2\/tests\/train_kldR2_nway64.py",
"version": "colbert-v0.4"
}
}
|