|
{ |
|
"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": 4, |
|
"resume": false, |
|
"similarity": "cosine", |
|
"bsize": 64, |
|
"accumsteps": 1, |
|
"lr": 5e-6, |
|
"maxsteps": 1000, |
|
"save_every": null, |
|
"warmup": 0, |
|
"warmup_bert": null, |
|
"relu": false, |
|
"nway": 2, |
|
"use_ib_negatives": true, |
|
"reranker": false, |
|
"distillation_alpha": 1.0, |
|
"ignore_scores": true, |
|
"model_name": "antoinelouis\/colbertv1-camembert-base-mmarcoFR", |
|
"query_maxlen": 64, |
|
"attend_to_mask_tokens": true, |
|
"interaction": "colbert", |
|
"dim": 128, |
|
"doc_maxlen": 512, |
|
"mask_punctuation": true, |
|
"checkpoint": "antoinelouis\/colbertv1-camembert-base-mmarcoFR", |
|
"triples": "data\/lleqa\/train-64.0K-triplets-hard_negs_me5.jsonl", |
|
"collection": "data\/lleqa\/collection.tsv", |
|
"queries": "data\/lleqa\/queries.train.tsv", |
|
"index_name": null, |
|
"overwrite": false, |
|
"root": "output\/training\/lleqa", |
|
"experiment": "colbert", |
|
"index_root": null, |
|
"name": "2024-05-24_23.13-antoinelouis-colbertv1-camembert-base-mmarcoFR-lleqa", |
|
"rank": 0, |
|
"nranks": 1, |
|
"amp": true, |
|
"gpus": 1, |
|
"avoid_fork_if_possible": false, |
|
"meta": { |
|
"hostname": "jupyterlab-gpu-2-13-djqhv", |
|
"git_branch": "main", |
|
"git_hash": "73bea556edb4e68a9fa96e46084a2260827fb742", |
|
"git_commit_datetime": "2024-05-24 22:52:51+02:00", |
|
"current_datetime": "May 24, 2024 ; 11:24PM CEST (+0200)", |
|
"cmd": "src\/retrievers\/multi_dense_biencoder.py --dataset lleqa --data_dir data --do_train --model_name antoinelouis\/colbertv1-camembert-base-mmarcoFR --dim 128 --similarity cosine --doc_maxlen 512 --query_maxlen 64 --mask_punctuation --attend_to_mask_tokens --maxsteps 1000 --warmup 0 --lr 5e-6 --bsize 64 --accumsteps 1 --nway 2 --use_ib_negatives --ignore_scores --distillation_alpha 1.0 --seed 42 --do_test --nbits 2 --kmeans_niters 4 --output_dir output\/training", |
|
"version": "colbert-v0.4" |
|
} |
|
} |
|
|