ColBERT-NQ / artifact.metadata
danf's picture
Model artifacts
6eddc80
{
"ncells": null,
"centroid_score_threshold": null,
"ndocs": null,
"index_path": null,
"nbits": 1,
"kmeans_niters": 4,
"resume": false,
"similarity": "cosine",
"bsize": 32,
"accumsteps": 1,
"lr": 3e-6,
"maxsteps": 500000,
"save_every": null,
"warmup": null,
"warmup_bert": null,
"relu": false,
"nway": 2,
"use_ib_negatives": true,
"reranker": false,
"distillation_alpha": 1.0,
"ignore_scores": false,
"epochs": 1,
"query_maxlen": 32,
"attend_to_mask_tokens": false,
"interaction": "colbert",
"dim": 128,
"doc_maxlen": 220,
"mask_punctuation": true,
"checkpoint": "bert-base-uncased",
"triples": "\/workdisk\/nosnap\/datasets\/open_domain_data\/colbert\/nq-triplets_train.json",
"collection": "\/workdisk\/nosnap\/datasets\/open_domain_data\/colbert\/collection.tsv",
"queries": "\/workdisk\/nosnap\/datasets\/open_domain_data\/colbert\/nq-queries_train.tsv",
"index_name": null,
"overwrite": false,
"root": "\/workdisk\/code\/colbertv2\/experiments",
"experiment": "wiki-nq",
"index_root": null,
"name": "2022-11\/08\/22.14.30",
"rank": 0,
"nranks": 4,
"amp": true,
"gpus": 4,
"wandb_project": "colbert_v2",
"meta": {
"hostname": "pizsak-research-tc2tt-zppl2",
"git_branch": "peter-enable-training",
"git_hash": "80838542bbe7e08d5e37985f54b5ec81636a4913",
"git_commit_datetime": "2022-11-08 22:21:50+02:00",
"current_datetime": "Nov 09, 2022 ; 8:00PM IST (+0200)",
"cmd": "train_model.py",
"version": "colbert-v0.4"
}
}