Eli Browne
Initial commit: mixed up models, THIS is plain ColBERT; other model is LegalBERT; names are updated to reflect this
3c6447c
{ | |
"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": 2, | |
"kmeans_niters": 20, | |
"resume": false, | |
"similarity": "cosine", | |
"bsize": 32, | |
"accumsteps": 1, | |
"lr": 5e-6, | |
"maxsteps": 500000, | |
"save_every": 73, | |
"warmup": 73, | |
"warmup_bert": null, | |
"relu": false, | |
"nway": 2, | |
"use_ib_negatives": true, | |
"reranker": false, | |
"distillation_alpha": 1.0, | |
"ignore_scores": false, | |
"model_name": "baseColBERT", | |
"query_maxlen": 32, | |
"attend_to_mask_tokens": false, | |
"interaction": "colbert", | |
"dim": 128, | |
"doc_maxlen": 256, | |
"mask_punctuation": true, | |
"checkpoint": "colbert-ir\/colbertv2.0", | |
"triples": ".\/data\/triples.train.colbert.jsonl", | |
"collection": ".\/data\/corpus.train.colbert.tsv", | |
"queries": ".\/data\/queries.train.colbert.tsv", | |
"index_name": null, | |
"overwrite": false, | |
"root": ".ragatouille\/", | |
"experiment": "colbert", | |
"index_root": null, | |
"name": "2024-07\/17\/13.39.56", | |
"rank": 0, | |
"nranks": 1, | |
"amp": true, | |
"gpus": 1, | |
"avoid_fork_if_possible": false, | |
"meta": { | |
"hostname": "jagupard28.stanford.edu", | |
"git_branch": "main", | |
"git_hash": "4e1a837629bb109bf7e4b6dc2650dced9c0b9bff", | |
"git_commit_datetime": "2024-07-17 12:04:00-07:00", | |
"current_datetime": "Jul 17, 2024 ; 1:44PM PDT (-0700)", | |
"cmd": "fine_tune.py", | |
"version": "colbert-v0.4" | |
} | |
} | |