File size: 2,402 Bytes
111f4cd | 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 | {
"run": "pilot_Bpc_s2",
"step": 6000,
"model_config": {
"vocab_size": 256,
"d_enc": 256,
"d_main": 512,
"n_enc": 2,
"n_dec": 2,
"n_main": 6,
"n_heads_enc": 4,
"n_heads_main": 8,
"seq_len": 1024,
"dropout": 0.0,
"target_ratio": 6.0,
"max_chunk_mult": 3.0,
"dechunk": "ema",
"router_noise": 0.0,
"rope_base": 10000.0
},
"tied_weights": {
"head.weight": "emb.weight"
},
"train_config": {
"condition": "B",
"seed": 2,
"out_dir": null,
"run_name": "pilot_Bpc_s2",
"corpus": "corpus",
"langs": null,
"seq_len": 1024,
"batch_size": 32,
"grad_accum": 1,
"steps": 6000,
"lr": 0.0003,
"min_lr_frac": 0.1,
"warmup": 300,
"weight_decay": 0.1,
"grad_clip": 1.0,
"betas": [
0.9,
0.95
],
"router_lr_mult": 1.0,
"d_enc": 256,
"d_main": 512,
"n_enc": 2,
"n_dec": 2,
"n_main": 6,
"n_heads_enc": 4,
"n_heads_main": 8,
"dechunk": "ema",
"target_ratio": 6.0,
"alpha": 0.03,
"alpha_parity_cap": 0.06,
"global_anchor": 0.0,
"premium_clip": 4.0,
"premium_kind": "flores_per_char",
"mixture": null,
"doc_filters": null,
"tau1": 0.4,
"tau2": 0.7,
"freeze_scope": "router",
"ckpt_every": 200,
"eval_every": 200,
"log_every": 50,
"probe_sents": 128,
"eval_batches": 8,
"keep_ckpt": false,
"amp": "bf16",
"compile": false,
"cpc_decay": 0.9,
"device": "cuda",
"log_spaced_probe": true,
"probe_bpb_batches": 4,
"dump_reps": false,
"dump_reps_thresholds": [
0.3,
0.4,
0.5,
0.6,
0.7
],
"router_noise": 0.0,
"genreg_target_entropy": null,
"genreg_eta": 0.05
},
"probe_agg": {
"gini_cpc": 0.05947751695357484,
"gini_cps": 0.10514840600952735,
"bpic": 9.37537559545621,
"worst_lang": "ru",
"cpc_parity_vs_en": {
"en": 1.0,
"de": 1.0199480456016647,
"vi": 1.0056935311382222,
"tr": 1.0146574383001157,
"fi": 0.9878722240593089,
"ru": 1.2588239101357477,
"el": 0.9734078650680693,
"hi": 1.1679111674836358,
"ta": 1.1526405179899044,
"th": 1.121282957018114,
"zh": 1.186838330558305,
"my": 0.8244921548326407
}
},
"source": "AdaptiveChunking/hnet-chunking-pilots/pilot_Bpc_s2/model.safetensors"
} |