|
{ |
|
"best_metric": 0.03033074364066124, |
|
"best_model_checkpoint": "runs/deepseek_lora_20240423-133229/checkpoint-1000", |
|
"epoch": 0.31330774653403304, |
|
"eval_steps": 500, |
|
"global_step": 1000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.0, |
|
"grad_norm": 1.5461480617523193, |
|
"learning_rate": 4.0000000000000003e-07, |
|
"loss": 1.5239, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.01, |
|
"grad_norm": 2.797250747680664, |
|
"learning_rate": 8.000000000000001e-07, |
|
"loss": 1.4966, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.01, |
|
"grad_norm": 4.274460315704346, |
|
"learning_rate": 1.2000000000000002e-06, |
|
"loss": 1.4788, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.01, |
|
"grad_norm": 1.9981650114059448, |
|
"learning_rate": 1.6000000000000001e-06, |
|
"loss": 1.4001, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"grad_norm": 1.2322547435760498, |
|
"learning_rate": 2.0000000000000003e-06, |
|
"loss": 1.4683, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"grad_norm": 1.250878930091858, |
|
"learning_rate": 2.4000000000000003e-06, |
|
"loss": 1.1961, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"grad_norm": 4.917982578277588, |
|
"learning_rate": 2.8000000000000003e-06, |
|
"loss": 1.4261, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"grad_norm": 3.885340929031372, |
|
"learning_rate": 3.2000000000000003e-06, |
|
"loss": 1.2742, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"grad_norm": 1.170296311378479, |
|
"learning_rate": 3.6000000000000003e-06, |
|
"loss": 1.253, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"grad_norm": 6.992724895477295, |
|
"learning_rate": 4.000000000000001e-06, |
|
"loss": 1.2521, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"grad_norm": 0.9540568590164185, |
|
"learning_rate": 4.4e-06, |
|
"loss": 1.0828, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"grad_norm": 0.9779148697853088, |
|
"learning_rate": 4.800000000000001e-06, |
|
"loss": 1.1255, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"grad_norm": 1.2271130084991455, |
|
"learning_rate": 5.2e-06, |
|
"loss": 0.9496, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"grad_norm": 2.1939122676849365, |
|
"learning_rate": 5.600000000000001e-06, |
|
"loss": 0.953, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"grad_norm": 1.5918376445770264, |
|
"learning_rate": 6e-06, |
|
"loss": 0.9348, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"grad_norm": 0.772602379322052, |
|
"learning_rate": 6.4000000000000006e-06, |
|
"loss": 0.7762, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"grad_norm": 0.8226258158683777, |
|
"learning_rate": 6.800000000000001e-06, |
|
"loss": 0.6798, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"grad_norm": 1.0945135354995728, |
|
"learning_rate": 7.2000000000000005e-06, |
|
"loss": 0.5936, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"grad_norm": 0.6972842812538147, |
|
"learning_rate": 7.600000000000001e-06, |
|
"loss": 0.4161, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"grad_norm": 3.9560859203338623, |
|
"learning_rate": 8.000000000000001e-06, |
|
"loss": 0.3959, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"grad_norm": 0.4297197759151459, |
|
"learning_rate": 8.400000000000001e-06, |
|
"loss": 0.3166, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"grad_norm": 0.4494841992855072, |
|
"learning_rate": 8.8e-06, |
|
"loss": 0.2056, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"grad_norm": 0.6981032490730286, |
|
"learning_rate": 9.200000000000002e-06, |
|
"loss": 0.3024, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"grad_norm": 0.5783653259277344, |
|
"learning_rate": 9.600000000000001e-06, |
|
"loss": 0.4392, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"grad_norm": 0.33407163619995117, |
|
"learning_rate": 1e-05, |
|
"loss": 0.3708, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"grad_norm": 0.3096323311328888, |
|
"learning_rate": 1.04e-05, |
|
"loss": 0.1637, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"grad_norm": 1.6187375783920288, |
|
"learning_rate": 1.0800000000000002e-05, |
|
"loss": 0.1078, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"grad_norm": 0.27954748272895813, |
|
"learning_rate": 1.1200000000000001e-05, |
|
"loss": 0.2668, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"grad_norm": 0.4082590937614441, |
|
"learning_rate": 1.16e-05, |
|
"loss": 0.1843, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"grad_norm": 0.2993428707122803, |
|
"learning_rate": 1.2e-05, |
|
"loss": 0.1969, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"grad_norm": 3.0335004329681396, |
|
"learning_rate": 1.2400000000000002e-05, |
|
"loss": 0.122, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"grad_norm": 0.0972365066409111, |
|
"learning_rate": 1.2800000000000001e-05, |
|
"loss": 0.1354, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"grad_norm": 0.26203760504722595, |
|
"learning_rate": 1.3200000000000002e-05, |
|
"loss": 0.17, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"grad_norm": 5.179258346557617, |
|
"learning_rate": 1.3600000000000002e-05, |
|
"loss": 0.1409, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"grad_norm": 0.032190825790166855, |
|
"learning_rate": 1.4e-05, |
|
"loss": 0.076, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"grad_norm": 0.02556140348315239, |
|
"learning_rate": 1.4400000000000001e-05, |
|
"loss": 0.0468, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"grad_norm": 5.667189121246338, |
|
"learning_rate": 1.48e-05, |
|
"loss": 0.1345, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"grad_norm": 0.07244217395782471, |
|
"learning_rate": 1.5200000000000002e-05, |
|
"loss": 0.0411, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"grad_norm": 0.11255475878715515, |
|
"learning_rate": 1.5600000000000003e-05, |
|
"loss": 0.1907, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"grad_norm": 1.9602845907211304, |
|
"learning_rate": 1.6000000000000003e-05, |
|
"loss": 0.0717, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"grad_norm": 0.013551376760005951, |
|
"learning_rate": 1.64e-05, |
|
"loss": 0.0808, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"grad_norm": 0.14421305060386658, |
|
"learning_rate": 1.6800000000000002e-05, |
|
"loss": 0.0604, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"grad_norm": 0.01705547235906124, |
|
"learning_rate": 1.72e-05, |
|
"loss": 0.1742, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"grad_norm": 0.021360402926802635, |
|
"learning_rate": 1.76e-05, |
|
"loss": 0.0886, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"grad_norm": 0.04369840770959854, |
|
"learning_rate": 1.8e-05, |
|
"loss": 0.1063, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"grad_norm": 2.2816083431243896, |
|
"learning_rate": 1.8400000000000003e-05, |
|
"loss": 0.1061, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"grad_norm": 0.00989863183349371, |
|
"learning_rate": 1.88e-05, |
|
"loss": 0.2317, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"grad_norm": 0.014644264243543148, |
|
"learning_rate": 1.9200000000000003e-05, |
|
"loss": 0.1054, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"grad_norm": 0.13385985791683197, |
|
"learning_rate": 1.9600000000000002e-05, |
|
"loss": 0.1196, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"grad_norm": 2.099637985229492, |
|
"learning_rate": 2e-05, |
|
"loss": 0.1584, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"eval_loss": 0.05546905845403671, |
|
"eval_runtime": 62.1757, |
|
"eval_samples_per_second": 16.083, |
|
"eval_steps_per_second": 16.083, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"grad_norm": 0.013259253464639187, |
|
"learning_rate": 1.9955555555555557e-05, |
|
"loss": 0.0373, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"grad_norm": 0.1753215193748474, |
|
"learning_rate": 1.9911111111111112e-05, |
|
"loss": 0.0986, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"grad_norm": 0.006998735014349222, |
|
"learning_rate": 1.9866666666666667e-05, |
|
"loss": 0.1372, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"grad_norm": 0.3990759253501892, |
|
"learning_rate": 1.9822222222222226e-05, |
|
"loss": 0.127, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"grad_norm": 1.8248735666275024, |
|
"learning_rate": 1.977777777777778e-05, |
|
"loss": 0.1239, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"grad_norm": 0.005832708440721035, |
|
"learning_rate": 1.9733333333333336e-05, |
|
"loss": 0.1181, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"grad_norm": 0.10705375671386719, |
|
"learning_rate": 1.968888888888889e-05, |
|
"loss": 0.0972, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"grad_norm": 3.7429046630859375, |
|
"learning_rate": 1.9644444444444447e-05, |
|
"loss": 0.1543, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"grad_norm": 0.010058793239295483, |
|
"learning_rate": 1.9600000000000002e-05, |
|
"loss": 0.1146, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"grad_norm": 0.012124909088015556, |
|
"learning_rate": 1.9555555555555557e-05, |
|
"loss": 0.1582, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"grad_norm": 0.012700803577899933, |
|
"learning_rate": 1.9511111111111113e-05, |
|
"loss": 0.0711, |
|
"step": 610 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"grad_norm": 0.2153177261352539, |
|
"learning_rate": 1.9466666666666668e-05, |
|
"loss": 0.0452, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"grad_norm": 0.01289213914424181, |
|
"learning_rate": 1.9422222222222223e-05, |
|
"loss": 0.067, |
|
"step": 630 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"grad_norm": 0.007021366152912378, |
|
"learning_rate": 1.9377777777777778e-05, |
|
"loss": 0.0776, |
|
"step": 640 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"grad_norm": 0.005274974275380373, |
|
"learning_rate": 1.9333333333333333e-05, |
|
"loss": 0.2159, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"grad_norm": 1.6235843896865845, |
|
"learning_rate": 1.928888888888889e-05, |
|
"loss": 0.0664, |
|
"step": 660 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"grad_norm": 0.009710369631648064, |
|
"learning_rate": 1.9244444444444444e-05, |
|
"loss": 0.1125, |
|
"step": 670 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"grad_norm": 0.0918051227927208, |
|
"learning_rate": 1.9200000000000003e-05, |
|
"loss": 0.0755, |
|
"step": 680 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"grad_norm": 0.012988925911486149, |
|
"learning_rate": 1.9155555555555558e-05, |
|
"loss": 0.095, |
|
"step": 690 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"grad_norm": 0.09063850343227386, |
|
"learning_rate": 1.9111111111111113e-05, |
|
"loss": 0.0419, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"grad_norm": 3.1540229320526123, |
|
"learning_rate": 1.9066666666666668e-05, |
|
"loss": 0.0338, |
|
"step": 710 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"grad_norm": 0.009780442342162132, |
|
"learning_rate": 1.9022222222222223e-05, |
|
"loss": 0.0935, |
|
"step": 720 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"grad_norm": 0.011248274706304073, |
|
"learning_rate": 1.897777777777778e-05, |
|
"loss": 0.017, |
|
"step": 730 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"grad_norm": 0.03275947645306587, |
|
"learning_rate": 1.8933333333333334e-05, |
|
"loss": 0.1128, |
|
"step": 740 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"grad_norm": 2.224815607070923, |
|
"learning_rate": 1.888888888888889e-05, |
|
"loss": 0.1213, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"grad_norm": 1.9777376651763916, |
|
"learning_rate": 1.8844444444444444e-05, |
|
"loss": 0.1128, |
|
"step": 760 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"grad_norm": 7.6119561195373535, |
|
"learning_rate": 1.88e-05, |
|
"loss": 0.0557, |
|
"step": 770 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"grad_norm": 3.7092978954315186, |
|
"learning_rate": 1.8755555555555558e-05, |
|
"loss": 0.1196, |
|
"step": 780 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"grad_norm": 0.08959495276212692, |
|
"learning_rate": 1.8711111111111113e-05, |
|
"loss": 0.0231, |
|
"step": 790 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"grad_norm": 2.716322183609009, |
|
"learning_rate": 1.866666666666667e-05, |
|
"loss": 0.1253, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"grad_norm": 0.006216592621058226, |
|
"learning_rate": 1.8622222222222224e-05, |
|
"loss": 0.0377, |
|
"step": 810 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"grad_norm": 0.005503402091562748, |
|
"learning_rate": 1.857777777777778e-05, |
|
"loss": 0.1072, |
|
"step": 820 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"grad_norm": 0.4769572913646698, |
|
"learning_rate": 1.8533333333333334e-05, |
|
"loss": 0.0888, |
|
"step": 830 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"grad_norm": 0.034854818135499954, |
|
"learning_rate": 1.848888888888889e-05, |
|
"loss": 0.0424, |
|
"step": 840 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"grad_norm": 0.00536635797470808, |
|
"learning_rate": 1.8444444444444448e-05, |
|
"loss": 0.0161, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"grad_norm": 0.002790963975712657, |
|
"learning_rate": 1.8400000000000003e-05, |
|
"loss": 0.025, |
|
"step": 860 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"grad_norm": 0.003568426240235567, |
|
"learning_rate": 1.835555555555556e-05, |
|
"loss": 0.0597, |
|
"step": 870 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"grad_norm": 0.005008562933653593, |
|
"learning_rate": 1.8311111111111114e-05, |
|
"loss": 0.1817, |
|
"step": 880 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"grad_norm": 0.006756258197128773, |
|
"learning_rate": 1.826666666666667e-05, |
|
"loss": 0.0084, |
|
"step": 890 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"grad_norm": 2.3732385635375977, |
|
"learning_rate": 1.8222222222222224e-05, |
|
"loss": 0.072, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"grad_norm": 0.003554440103471279, |
|
"learning_rate": 1.817777777777778e-05, |
|
"loss": 0.0507, |
|
"step": 910 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"grad_norm": 0.06785377860069275, |
|
"learning_rate": 1.8133333333333335e-05, |
|
"loss": 0.0428, |
|
"step": 920 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"grad_norm": 0.009546133689582348, |
|
"learning_rate": 1.808888888888889e-05, |
|
"loss": 0.0179, |
|
"step": 930 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"grad_norm": 4.595461368560791, |
|
"learning_rate": 1.8044444444444445e-05, |
|
"loss": 0.0506, |
|
"step": 940 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"grad_norm": 2.310696601867676, |
|
"learning_rate": 1.8e-05, |
|
"loss": 0.0719, |
|
"step": 950 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"grad_norm": 1.8953649997711182, |
|
"learning_rate": 1.7955555555555556e-05, |
|
"loss": 0.1718, |
|
"step": 960 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"grad_norm": 1.706213116645813, |
|
"learning_rate": 1.791111111111111e-05, |
|
"loss": 0.0807, |
|
"step": 970 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"grad_norm": 0.06022366136312485, |
|
"learning_rate": 1.7866666666666666e-05, |
|
"loss": 0.0792, |
|
"step": 980 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"grad_norm": 0.033781811594963074, |
|
"learning_rate": 1.782222222222222e-05, |
|
"loss": 0.0864, |
|
"step": 990 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"grad_norm": 0.014879009686410427, |
|
"learning_rate": 1.7777777777777777e-05, |
|
"loss": 0.0527, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"eval_loss": 0.03033074364066124, |
|
"eval_runtime": 62.1716, |
|
"eval_samples_per_second": 16.085, |
|
"eval_steps_per_second": 16.085, |
|
"step": 1000 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 5000, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 2, |
|
"save_steps": 1000, |
|
"total_flos": 1.6102125993984e+16, |
|
"train_batch_size": 1, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|