PairDistill-colbertv2 / artifact.metadata
chaoweihuang's picture
Upload folder using huggingface_hub
e3e42f8 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,
"nbits": 1,
"kmeans_niters": 20,
"resume": false,
"similarity": "cosine",
"bsize": 8,
"accumsteps": 2,
"lr": 1e-5,
"maxsteps": 30000,
"save_every": 10000,
"warmup": 3000,
"warmup_bert": null,
"relu": false,
"nway": 64,
"use_ib_negatives": true,
"reranker": false,
"distillation_alpha": 1.0,
"ignore_scores": false,
"model_name": null,
"rank_loss": null,
"ranknet_weight_type": null,
"use_pair_scores": true,
"pair_loss_alpha": 3.0,
"only_pair_loss": false,
"query_maxlen": 32,
"attend_to_mask_tokens": false,
"interaction": "colbert",
"dim": 128,
"doc_maxlen": 180,
"mask_punctuation": true,
"checkpoint": "experiments\/official-neg63-point-mini-pair-r2-4-colbertv2-pair3.0-cont\/none\/2024-01-31_14.16.16\/checkpoints\/colbert-15000",
"triples": "..\/data\/msmarco\/official_neg63_point_mini_pair_r2-4_onlypair.jsonl",
"collection": "..\/data\/msmarco\/collection.tsv",
"queries": "..\/data\/msmarco\/queries.train.tsv",
"index_name": null,
"overwrite": false,
"root": "experiments",
"experiment": "official-neg63-point-mini-pair-r2-4-colbertv2-pair3.0-cont2-2",
"index_root": null,
"name": "2024-02-02_00.40.15",
"rank": 0,
"nranks": 4,
"amp": true,
"gpus": 4,
"meta": {
"hostname": "e9z4hictr1706184263273-zz9wq",
"current_datetime": "Feb 02, 2024 ; 1:56PM CST (+0800)",
"cmd": "train.py",
"version": "colbert-v0.4"
}
}