|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.0161290322580645, |
|
"eval_steps": 25, |
|
"global_step": 126, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.01, |
|
"learning_rate": 0.0, |
|
"loss": 4.0553, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.01, |
|
"eval_loss": 4.052073955535889, |
|
"eval_runtime": 57.3905, |
|
"eval_samples_per_second": 4.356, |
|
"eval_steps_per_second": 4.356, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 2e-05, |
|
"loss": 4.3696, |
|
"step": 2 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 4e-05, |
|
"loss": 4.2401, |
|
"step": 3 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 6e-05, |
|
"loss": 4.3077, |
|
"step": 4 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 8e-05, |
|
"loss": 4.0741, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 0.0001, |
|
"loss": 4.1673, |
|
"step": 6 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 0.00012, |
|
"loss": 3.9378, |
|
"step": 7 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 0.00014, |
|
"loss": 4.0167, |
|
"step": 8 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 0.00016, |
|
"loss": 3.5818, |
|
"step": 9 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"learning_rate": 0.00018, |
|
"loss": 3.9022, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 0.0002, |
|
"loss": 3.7906, |
|
"step": 11 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 0.0001999979107245606, |
|
"loss": 3.6776, |
|
"step": 12 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 0.00019999164298554375, |
|
"loss": 3.4291, |
|
"step": 13 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 0.00019998119704485014, |
|
"loss": 3.6101, |
|
"step": 14 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 0.00019996657333896877, |
|
"loss": 3.5537, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 0.00019994777247895855, |
|
"loss": 3.386, |
|
"step": 16 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 0.00019992479525042303, |
|
"loss": 3.6359, |
|
"step": 17 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 0.00019989764261347736, |
|
"loss": 3.6132, |
|
"step": 18 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 0.00019986631570270832, |
|
"loss": 3.4794, |
|
"step": 19 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 0.00019983081582712685, |
|
"loss": 3.2395, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 0.00019979114447011323, |
|
"loss": 3.2974, |
|
"step": 21 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 0.00019974730328935534, |
|
"loss": 3.2698, |
|
"step": 22 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 0.0001996992941167792, |
|
"loss": 3.3091, |
|
"step": 23 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 0.00019964711895847258, |
|
"loss": 3.2017, |
|
"step": 24 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 0.00019959077999460095, |
|
"loss": 3.2469, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"eval_loss": 3.0720062255859375, |
|
"eval_runtime": 61.8274, |
|
"eval_samples_per_second": 4.044, |
|
"eval_steps_per_second": 4.044, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 0.00019953027957931658, |
|
"loss": 3.1102, |
|
"step": 26 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 0.00019946562024066014, |
|
"loss": 3.1641, |
|
"step": 27 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 0.00019939680468045499, |
|
"loss": 2.9779, |
|
"step": 28 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 0.00019932383577419432, |
|
"loss": 3.104, |
|
"step": 29 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 0.00019924671657092096, |
|
"loss": 2.7981, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 0.00019916545029310012, |
|
"loss": 2.7673, |
|
"step": 31 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 0.00019908004033648453, |
|
"loss": 2.8478, |
|
"step": 32 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 0.00019899049026997272, |
|
"loss": 2.7539, |
|
"step": 33 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 0.00019889680383545973, |
|
"loss": 2.726, |
|
"step": 34 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 0.00019879898494768093, |
|
"loss": 2.81, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 0.00019869703769404828, |
|
"loss": 2.6177, |
|
"step": 36 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 0.00019859096633447965, |
|
"loss": 2.6046, |
|
"step": 37 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 0.00019848077530122083, |
|
"loss": 2.5004, |
|
"step": 38 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 0.0001983664691986601, |
|
"loss": 2.4993, |
|
"step": 39 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 0.00019824805280313623, |
|
"loss": 2.2775, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 0.00019812553106273847, |
|
"loss": 2.3306, |
|
"step": 41 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 0.00019799890909710013, |
|
"loss": 2.236, |
|
"step": 42 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 0.00019786819219718443, |
|
"loss": 2.2798, |
|
"step": 43 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 0.0001977333858250636, |
|
"loss": 2.2178, |
|
"step": 44 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 0.00019759449561369038, |
|
"loss": 1.9915, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 0.00019745152736666302, |
|
"loss": 2.3066, |
|
"step": 46 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 0.00019730448705798239, |
|
"loss": 2.1409, |
|
"step": 47 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 0.00019715338083180269, |
|
"loss": 1.9976, |
|
"step": 48 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 0.00019699821500217434, |
|
"loss": 2.1843, |
|
"step": 49 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 0.0001968389960527806, |
|
"loss": 2.1539, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"eval_loss": 1.9486372470855713, |
|
"eval_runtime": 61.7944, |
|
"eval_samples_per_second": 4.046, |
|
"eval_steps_per_second": 4.046, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 0.0001966757306366662, |
|
"loss": 2.2096, |
|
"step": 51 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 0.00019650842557595967, |
|
"loss": 2.0241, |
|
"step": 52 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 0.00019633708786158806, |
|
"loss": 1.8814, |
|
"step": 53 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 0.00019616172465298492, |
|
"loss": 1.9953, |
|
"step": 54 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 0.00019598234327779118, |
|
"loss": 1.9807, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 0.0001957989512315489, |
|
"loss": 1.9743, |
|
"step": 56 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 0.00019561155617738797, |
|
"loss": 1.867, |
|
"step": 57 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 0.00019542016594570615, |
|
"loss": 1.8289, |
|
"step": 58 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 0.00019522478853384155, |
|
"loss": 1.9712, |
|
"step": 59 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 0.00019502543210573881, |
|
"loss": 1.6946, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 0.00019482210499160765, |
|
"loss": 1.8674, |
|
"step": 61 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 0.00019461481568757506, |
|
"loss": 1.6849, |
|
"step": 62 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 0.00019440357285533, |
|
"loss": 1.7003, |
|
"step": 63 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 0.00019418838532176173, |
|
"loss": 1.7021, |
|
"step": 64 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 0.00019396926207859084, |
|
"loss": 1.8302, |
|
"step": 65 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 0.0001937462122819935, |
|
"loss": 1.8549, |
|
"step": 66 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 0.000193519245252219, |
|
"loss": 1.8946, |
|
"step": 67 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 0.0001932883704732001, |
|
"loss": 1.799, |
|
"step": 68 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"learning_rate": 0.00019305359759215685, |
|
"loss": 1.7541, |
|
"step": 69 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"learning_rate": 0.00019281493641919368, |
|
"loss": 1.7145, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 0.00019257239692688907, |
|
"loss": 1.8254, |
|
"step": 71 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 0.00019232598924987903, |
|
"loss": 1.6161, |
|
"step": 72 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 0.00019207572368443385, |
|
"loss": 1.5698, |
|
"step": 73 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 0.00019182161068802741, |
|
"loss": 1.8155, |
|
"step": 74 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 0.0001915636608789006, |
|
"loss": 1.6997, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"eval_loss": 1.6153544187545776, |
|
"eval_runtime": 61.8069, |
|
"eval_samples_per_second": 4.045, |
|
"eval_steps_per_second": 4.045, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 0.00019130188503561741, |
|
"loss": 1.6507, |
|
"step": 76 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 0.0001910362940966147, |
|
"loss": 1.6035, |
|
"step": 77 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 0.00019076689915974488, |
|
"loss": 1.4493, |
|
"step": 78 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 0.00019049371148181253, |
|
"loss": 1.5494, |
|
"step": 79 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 0.0001902167424781038, |
|
"loss": 1.5386, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 0.00018993600372190932, |
|
"loss": 1.467, |
|
"step": 81 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 0.00018965150694404094, |
|
"loss": 1.6658, |
|
"step": 82 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 0.00018936326403234125, |
|
"loss": 1.6175, |
|
"step": 83 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 0.00018907128703118695, |
|
"loss": 1.6551, |
|
"step": 84 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 0.00018877558814098561, |
|
"loss": 1.409, |
|
"step": 85 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 0.00018847617971766577, |
|
"loss": 1.4879, |
|
"step": 86 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 0.0001881730742721608, |
|
"loss": 1.3067, |
|
"step": 87 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 0.00018786628446988593, |
|
"loss": 1.6869, |
|
"step": 88 |
|
}, |
|
{ |
|
"epoch": 0.72, |
|
"learning_rate": 0.0001875558231302091, |
|
"loss": 1.5962, |
|
"step": 89 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 0.00018724170322591537, |
|
"loss": 1.3352, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 0.00018692393788266479, |
|
"loss": 1.5016, |
|
"step": 91 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 0.00018660254037844388, |
|
"loss": 1.6951, |
|
"step": 92 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 0.00018627752414301086, |
|
"loss": 1.5388, |
|
"step": 93 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 0.00018594890275733453, |
|
"loss": 1.3656, |
|
"step": 94 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 0.00018561668995302667, |
|
"loss": 1.4599, |
|
"step": 95 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 0.0001852808996117683, |
|
"loss": 1.3428, |
|
"step": 96 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 0.00018494154576472976, |
|
"loss": 1.2147, |
|
"step": 97 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 0.0001845986425919841, |
|
"loss": 1.4761, |
|
"step": 98 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 0.00018425220442191495, |
|
"loss": 1.4571, |
|
"step": 99 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 0.00018390224573061747, |
|
"loss": 1.5076, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"eval_loss": 1.49380362033844, |
|
"eval_runtime": 61.7975, |
|
"eval_samples_per_second": 4.045, |
|
"eval_steps_per_second": 4.045, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 0.00018354878114129367, |
|
"loss": 1.3776, |
|
"step": 101 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 0.00018319182542364117, |
|
"loss": 1.437, |
|
"step": 102 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 0.00018283139349323634, |
|
"loss": 1.3558, |
|
"step": 103 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"learning_rate": 0.0001824675004109107, |
|
"loss": 1.4502, |
|
"step": 104 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 0.00018210016138212187, |
|
"loss": 1.3451, |
|
"step": 105 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 0.00018172939175631808, |
|
"loss": 1.403, |
|
"step": 106 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 0.00018135520702629675, |
|
"loss": 1.1523, |
|
"step": 107 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 0.00018097762282755727, |
|
"loss": 1.2302, |
|
"step": 108 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 0.00018059665493764743, |
|
"loss": 1.4151, |
|
"step": 109 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 0.0001802123192755044, |
|
"loss": 1.4853, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 0.0001798246319007893, |
|
"loss": 1.2571, |
|
"step": 111 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 0.0001794336090132164, |
|
"loss": 1.3594, |
|
"step": 112 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 0.00017903926695187595, |
|
"loss": 1.3997, |
|
"step": 113 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 0.00017864162219455163, |
|
"loss": 1.5718, |
|
"step": 114 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 0.00017824069135703198, |
|
"loss": 1.314, |
|
"step": 115 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 0.00017783649119241602, |
|
"loss": 1.3595, |
|
"step": 116 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 0.00017742903859041325, |
|
"loss": 1.4467, |
|
"step": 117 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"learning_rate": 0.00017701835057663798, |
|
"loss": 1.2837, |
|
"step": 118 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 0.0001766044443118978, |
|
"loss": 1.2669, |
|
"step": 119 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 0.00017618733709147662, |
|
"loss": 1.3115, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 0.0001757670463444118, |
|
"loss": 1.1836, |
|
"step": 121 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 0.00017534358963276607, |
|
"loss": 1.6092, |
|
"step": 122 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 0.00017491698465089362, |
|
"loss": 1.2288, |
|
"step": 123 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 0.00017448724922470058, |
|
"loss": 1.2305, |
|
"step": 124 |
|
}, |
|
{ |
|
"epoch": 1.01, |
|
"learning_rate": 0.00017405440131090048, |
|
"loss": 1.3794, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 1.01, |
|
"eval_loss": 1.4039632081985474, |
|
"eval_runtime": 61.7879, |
|
"eval_samples_per_second": 4.046, |
|
"eval_steps_per_second": 4.046, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 1.02, |
|
"learning_rate": 0.00017361845899626355, |
|
"loss": 1.2283, |
|
"step": 126 |
|
} |
|
], |
|
"logging_steps": 1, |
|
"max_steps": 496, |
|
"num_train_epochs": 4, |
|
"save_steps": 500, |
|
"total_flos": 4.12261207202857e+16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|