bclavie commited on
Commit
6868316
1 Parent(s): fa8c5fd

Upload artifact.metadata

Browse files
Files changed (1) hide show
  1. artifact.metadata +75 -0
artifact.metadata ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "bclavie\/mini-base",
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
+ }