File size: 2,164 Bytes
cd3d68a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 | {
"phase": "D",
"init_repo": "electroglyph/berty-phaseC",
"sources": [
{
"name": "msmarco-bm25",
"repo": "sentence-transformers/msmarco-bm25",
"pats": [
"triplet-50/*.parquet"
],
"weight": 0.26,
"w_end": 0.14,
"fmt": "m50"
},
{
"name": "msmarco-dense",
"repo": "sentence-transformers/msmarco-co-condenser-margin-mse-cls-v1",
"pats": [
"triplet-50/*.parquet"
],
"weight": 0.26,
"w_end": 0.38,
"fmt": "m50"
},
{
"name": "hotpotqa",
"repo": "sentence-transformers/hotpotqa",
"pats": [
"triplet-20/*.parquet"
],
"weight": 0.12,
"w_end": 0.12,
"fmt": "h20"
},
{
"name": "nq",
"repo": "Tevatron/wikipedia-nq",
"pats": [
"nq-train.jsonl.gz"
],
"weight": 0.1,
"w_end": 0.1,
"fmt": "dpr"
},
{
"name": "quora",
"repo": "sentence-transformers/quora-duplicates",
"pats": [
"triplet-all/*.parquet",
"pair-class/*.parquet"
],
"weight": 0.09,
"w_end": 0.09,
"fmt": "quora"
},
{
"name": "nli",
"repo": "sentence-transformers/all-nli",
"pats": [
"triplet/train-*.parquet"
],
"weight": 0.07,
"w_end": 0.07,
"fmt": "nli"
},
{
"name": "pairs",
"repo": "sentence-transformers/eli5",
"pats": [
"pair/*.parquet"
],
"weight": 0.04,
"w_end": 0.04,
"fmt": "qa"
},
{
"name": "squad",
"repo": "sentence-transformers/squad",
"pats": [
"pair/*.parquet"
],
"weight": 0.02,
"w_end": 0.02,
"fmt": "qa"
},
{
"name": "stackex",
"repo": "sentence-transformers/stackexchange-duplicates",
"pats": [
"post-post-pair/*.parquet"
],
"weight": 0.04,
"w_end": 0.04,
"fmt": "t12",
"cols": [
"post1",
"post2"
]
}
],
"base_lr": 5e-05,
"seq_q": 64,
"seq_d": 512,
"h_frozen": [
7,
4
],
"n_ance": [
2,
6
],
"curr_anneal_steps": 15000,
"h_hard_eval": 15,
"n_pos": 2,
"nce_lambda": 1.0,
"bwd_lambda": 0.5,
"kl_lambda": 0.4,
"kl_temp": 1.0,
"distill_lambda": 0.05,
"reservoir": 262144,
"scifact_probe": true,
"batch_cfg": {
"micro_q": 64,
"accum": 2,
"q_target": "128 queries/step",
"vram": 95.0
}
} |