Colbert-Finetuned / artifact.metadata
dutta18's picture
Add ColBERT checkpoint
46ca325 verified
{
"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,
"index_bsize": 64,
"nbits": 1,
"kmeans_niters": 20,
"resume": false,
"pool_factor": 1,
"clustering_mode": "hierarchical",
"protected_tokens": 0,
"similarity": "cosine",
"bsize": 32,
"accumsteps": 1,
"lr": 0.0001,
"maxsteps": 400000,
"save_every": null,
"warmup": 20000,
"warmup_bert": null,
"relu": false,
"nway": 2,
"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": "colbert-ir\/colbertv1.9",
"triples": "train_files_colbert\/testData160000Final\/mapping_train_FAISS_1pos_1neg.jsonl",
"collection": "train_files_colbert\/testData160000Final\/collection_train_FAISS_1pos_1neg.tsv",
"queries": "train_files_colbert\/testData160000Final\/queries_train_FAISS_1pos_1neg.tsv",
"index_name": null,
"overwrite": false,
"root": "\/home\/aritra\/experiments",
"experiment": "default",
"index_root": null,
"name": "2025-09\/11\/03.22.38",
"rank": 0,
"nranks": 1,
"amp": true,
"gpus": 1,
"avoid_fork_if_possible": false,
"meta": {
"hostname": "gpu",
"current_datetime": "Sep 11, 2025 ; 3:31AM UTC (+0000)",
"cmd": "\/home\/aritra\/miniconda3\/envs\/colbert-test1\/lib\/python3.9\/site-packages\/ipykernel_launcher.py -f \/home\/aritra\/.local\/share\/jupyter\/runtime\/kernel-9f9cbc82-4453-4ede-b728-9c73c8627f46.json",
"version": "colbert-v0.4"
}
}