| { | |
| "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": 2, | |
| "kmeans_niters": 4, | |
| "resume": false, | |
| "similarity": "cosine", | |
| "bsize": 128, | |
| "accumsteps": 1, | |
| "lr": 5e-6, | |
| "maxsteps": 200000, | |
| "save_every": null, | |
| "warmup": 20000, | |
| "warmup_bert": null, | |
| "relu": false, | |
| "nway": 2, | |
| "use_ib_negatives": true, | |
| "reranker": false, | |
| "distillation_alpha": 1.0, | |
| "ignore_scores": true, | |
| "model_name": "facebook\/xmod-base", | |
| "query_maxlen": 32, | |
| "attend_to_mask_tokens": true, | |
| "interaction": "colbert", | |
| "dim": 128, | |
| "doc_maxlen": 256, | |
| "mask_punctuation": true, | |
| "checkpoint": "facebook\/xmod-base", | |
| "triples": "data\/mmarco\/tuples.train.scores-ids.2nway.25.6M.jsonl", | |
| "collection": "data\/mmarco\/english_collection.tsv", | |
| "queries": "data\/mmarco\/english_queries.train.tsv", | |
| "index_name": null, | |
| "overwrite": false, | |
| "root": "output\/training\/mmarco", | |
| "experiment": "colbert", | |
| "index_root": null, | |
| "name": "2024-01-24_18.33-facebook-xmod-base-mmarco-en", | |
| "rank": 0, | |
| "nranks": 1, | |
| "amp": true, | |
| "gpus": 1, | |
| "avoid_fork_if_possible": false, | |
| "meta": { | |
| "hostname": "jupyterlab-gpu-2-4-tljq4", | |
| "git_branch": "main", | |
| "git_hash": "2760099557eff9c5ca8d58bd23198a71161eaacf", | |
| "git_commit_datetime": "2024-01-24 17:11:38+01:00", | |
| "current_datetime": "Jan 25, 2024 ; 2:08AM CET (+0100)", | |
| "cmd": "src\/training\/colbertv2.py --dataset mmarco --language en --do_train --model_name facebook\/xmod-base --dim 128 --similarity cosine --doc_maxlen 256 --query_maxlen 32 --mask_punctuation --attend_to_mask_tokens --maxsteps 200000 --warmup 20000 --lr 5e-6 --bsize 128 --accumsteps 1 --nway 2 --use_ib_negatives --ignore_scores --distillation_alpha 1.0 --nbits 2 --kmeans_niters 4 --data_dir data --output_dir output\/training", | |
| "version": "colbert-v0.4" | |
| } | |
| } | |