File size: 1,797 Bytes
7df2438 104f39b 74cc722 7df2438 6d0bf8f 7df2438 | 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 | {
"setting": "KD + ground truth CE",
"alpha": 0.5,
"order": "math_code",
"seed": 1,
"exp": 2,
"student": "Qwen/Qwen2.5-1.5B-Instruct",
"gpu_student": "NVIDIA RTX A6000",
"gpu_teachers": "NVIDIA RTX 6000 Ada Generation",
"load_h": 0.003992888331413269,
"stages": {
"stage1_math": {
"domain": "math",
"steps": 896,
"train_h": 3.425248695876863,
"val_start": {
"step": 0,
"math_kl": 0.15874307828215575,
"math_ce": 0.6515277156879916,
"code_kl": 0.14786912634071003,
"code_ce": 0.30340147782401555,
"math_hard_kl": 0.24940142917923272,
"math_hard_ce": 0.6765219455834559
},
"val_end": {
"step": 896,
"math_kl": 0.08436640837325561,
"math_ce": 0.4846152951909727,
"code_kl": 0.1533881911288058,
"code_ce": 0.3149686746307666,
"math_hard_kl": 0.09327549027114479,
"math_hard_ce": 0.6137640068253667
},
"peak_gb_student": 36.57978296279907,
"peak_gb_teachers": 30.618666172027588
},
"stage2_code": {
"domain": "code",
"steps": 922,
"train_h": 3.420455579492781,
"val_start": {
"step": 0,
"math_kl": 0.08435009805638696,
"math_ce": 0.48463048857849117,
"code_kl": 0.15332529827721086,
"code_ce": 0.31500937218458064,
"math_hard_kl": 0.09329687484267526,
"math_hard_ce": 0.6136897417741203
},
"val_end": {
"step": 922,
"math_kl": 0.10094955866903507,
"math_ce": 0.47546204311290524,
"code_kl": 0.08989427999834172,
"code_ce": 0.23964554178732264,
"math_hard_kl": 0.12081789152162996,
"math_hard_ce": 0.6168563930018784
},
"peak_gb_student": 33.86883306503296,
"peak_gb_teachers": 30.005428314208984,
"bench_h": 0.6256002970536549,
"gpu_bench": "NVIDIA RTX A6000"
}
},
"train_h": 6.845704275369644,
"done": true
} |