answerdotai/answerai-colbert-small-v1

1{2    "query_token_id": "[unused0]",3    "doc_token_id": "[unused1]",4    "query_token": "[Q]",5    "doc_token": "[D]",6    "ncells": null,7    "centroid_score_threshold": null,8    "ndocs": null,9    "load_index_with_mmap": false,10    "index_path": null,11    "index_bsize": 64,12    "nbits": 1,13    "kmeans_niters": 4,14    "resume": false,15    "pool_factor": 1,16    "clustering_mode": "hierarchical",17    "protected_tokens": 0,18    "similarity": "cosine",19    "bsize": 32,20    "accumsteps": 1,21    "lr": 1e-5,22    "maxsteps": 15626,23    "save_every": null,24    "warmup": 781,25    "warmup_bert": null,26    "relu": false,27    "nway": 32,28    "use_ib_negatives": false,29    "reranker": false,30    "distillation_alpha": 1.0,31    "ignore_scores": false,32    "model_name": "answerdotai/AnswerAI-ColBERTv2.5-small",33    "schedule_free": false,34    "schedule_free_wd": 0.0,35    "kldiv_loss": true,36    "marginmse_loss": false,37    "kldiv_weight": 1.0,38    "marginmse_weight": 0.05,39    "ib_loss_weight": 1.0,40    "normalise_training_scores": true,41    "normalization_method": "minmax",42    "quant_aware": false,43    "highest_quant_level": 8,44    "lowest_quant_level": 2,45    "query_maxlen": 32,46    "attend_to_mask_tokens": false,47    "interaction": "colbert",48    "cap_padding": 0,49    "dynamic_query_maxlen": false,50    "dynamic_querylen_multiples": 32,51    "dim": 96,52    "doc_maxlen": 300,53    "mask_punctuation": true,54    "checkpoint": "answerdotai/AnswerAI-ColBERTv2.5-small",55    "triples": "\/home\/bclavie\/colbertv2.5_en\/data\/msmarco\/triplets.jsonl",56    "collection": "\/home\/bclavie\/colbertv2.5_en\/data\/msmarco\/collection.tsv",57    "queries": "\/home\/bclavie\/colbertv2.5_en\/data\/msmarco\/queries.tsv",58    "index_name": null,59    "overwrite": false,60    "root": "\/home\/bclavie\/colbertv2.5_en\/experiments",61    "experiment": "minicolbertv2.5",62    "index_root": null,63    "name": "2024-08\/07\/08.16.20",64    "rank": 0,65    "nranks": 4,66    "amp": true,67    "gpus": 4,68    "avoid_fork_if_possible": false,69    "meta": {70        "hostname": "a100-80-4x",71        "current_datetime": "Aug 07, 2024 ; 12:13PM UTC (+0000)",72        "cmd": "train_v2.5_mini.py",73        "version": "colbert-v0.4"74    }75}76