File size: 1,791 Bytes
f6d99cf e9b4e05 2b72233 f6d99cf 6e1ffcf f6d99cf | 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 only",
"alpha": 1.0,
"order": "math_code",
"seed": 1,
"exp": 2,
"student": "Qwen/Qwen2.5-1.5B-Instruct",
"gpu_student": "NVIDIA RTX 6000 Ada Generation",
"gpu_teachers": "NVIDIA RTX A6000",
"load_h": 0.004078492522239685,
"stages": {
"stage1_math": {
"domain": "math",
"steps": 896,
"train_h": 2.7449205807844796,
"val_start": {
"step": 0,
"math_kl": 0.15902198004314483,
"math_ce": 0.6553564874444496,
"code_kl": 0.14797331712898912,
"code_ce": 0.3033394576202258,
"math_hard_kl": 0.24902600898986357,
"math_hard_ce": 0.6769691187959327
},
"val_end": {
"step": 896,
"math_kl": 0.0756993714180177,
"math_ce": 0.858021463785887,
"code_kl": 0.1720647986767575,
"code_ce": 0.3236746077973602,
"math_hard_kl": 0.08558017192474197,
"math_hard_ce": 0.8342651866385676
},
"peak_gb_student": 36.57978296279907,
"peak_gb_teachers": 30.618666172027588
},
"stage2_code": {
"domain": "code",
"steps": 922,
"train_h": 2.6334087841378317,
"val_start": {
"step": 0,
"math_kl": 0.0757168118165207,
"math_ce": 0.858557205345532,
"code_kl": 0.17211650957781463,
"code_ce": 0.32364794010142417,
"math_hard_kl": 0.08564324635022826,
"math_hard_ce": 0.8344598311839683
},
"val_end": {
"step": 922,
"math_kl": 0.0941693826888755,
"math_ce": 0.7592589879130468,
"code_kl": 0.07876284398144404,
"code_ce": 0.3094142575261208,
"math_hard_kl": 0.11899033841110994,
"math_hard_ce": 0.7571135998629719
},
"peak_gb_student": 33.86883306503296,
"peak_gb_teachers": 30.005428314208984,
"bench_h": 0.5363972081740698,
"gpu_bench": "NVIDIA RTX 6000 Ada Generation"
}
},
"train_h": 5.378329364922312,
"done": true
} |