|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 3.0, |
|
"eval_steps": 500, |
|
"global_step": 105, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 5e-05, |
|
"loss": 1.41, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 0.0001, |
|
"loss": 1.3381, |
|
"step": 2 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 0.00015000000000000001, |
|
"loss": 0.3565, |
|
"step": 3 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 0.0002, |
|
"loss": 0.3117, |
|
"step": 4 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 0.00019995162822919883, |
|
"loss": 0.3793, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 0.00019980655971335945, |
|
"loss": 0.3194, |
|
"step": 6 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 0.0001995649347969019, |
|
"loss": 0.3543, |
|
"step": 7 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 0.00019922698723632767, |
|
"loss": 0.3057, |
|
"step": 8 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 0.0001987930439740757, |
|
"loss": 0.2841, |
|
"step": 9 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 0.00019826352482222638, |
|
"loss": 0.3207, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 0.00019763894205636072, |
|
"loss": 0.2432, |
|
"step": 11 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 0.00019691989991996663, |
|
"loss": 0.3355, |
|
"step": 12 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 0.00019610709403987246, |
|
"loss": 0.2753, |
|
"step": 13 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 0.00019520131075327298, |
|
"loss": 0.2159, |
|
"step": 14 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 0.0001942034263469989, |
|
"loss": 0.1977, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 0.00019311440620976597, |
|
"loss": 0.2275, |
|
"step": 16 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 0.00019193530389822363, |
|
"loss": 0.2853, |
|
"step": 17 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 0.00019066726011770726, |
|
"loss": 0.1527, |
|
"step": 18 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 0.00018931150161867916, |
|
"loss": 0.2492, |
|
"step": 19 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 0.00018786934000992688, |
|
"loss": 0.2089, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 0.00018634217048966637, |
|
"loss": 0.1417, |
|
"step": 21 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 0.00018473147049577774, |
|
"loss": 0.2058, |
|
"step": 22 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 0.00018303879827647975, |
|
"loss": 0.1367, |
|
"step": 23 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 0.00018126579138282503, |
|
"loss": 0.208, |
|
"step": 24 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 0.00017941416508447536, |
|
"loss": 0.1991, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 0.000177485710710289, |
|
"loss": 0.2163, |
|
"step": 26 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 0.00017548229391532572, |
|
"loss": 0.1497, |
|
"step": 27 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 0.00017340585287594604, |
|
"loss": 0.1983, |
|
"step": 28 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 0.00017125839641475072, |
|
"loss": 0.1899, |
|
"step": 29 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 0.0001690420020571747, |
|
"loss": 0.166, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 0.00016675881402161536, |
|
"loss": 0.1466, |
|
"step": 31 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 0.0001644110411450398, |
|
"loss": 0.1478, |
|
"step": 32 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 0.00016200095474607753, |
|
"loss": 0.2104, |
|
"step": 33 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 0.0001595308864276666, |
|
"loss": 0.0953, |
|
"step": 34 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 0.00015700322582137827, |
|
"loss": 0.101, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 1.03, |
|
"learning_rate": 0.00015442041827560274, |
|
"loss": 0.1058, |
|
"step": 36 |
|
}, |
|
{ |
|
"epoch": 1.06, |
|
"learning_rate": 0.00015178496248983254, |
|
"loss": 0.0695, |
|
"step": 37 |
|
}, |
|
{ |
|
"epoch": 1.09, |
|
"learning_rate": 0.00014909940809733222, |
|
"loss": 0.0608, |
|
"step": 38 |
|
}, |
|
{ |
|
"epoch": 1.11, |
|
"learning_rate": 0.00014636635319853275, |
|
"loss": 0.0414, |
|
"step": 39 |
|
}, |
|
{ |
|
"epoch": 1.14, |
|
"learning_rate": 0.00014358844184753712, |
|
"loss": 0.0492, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 1.17, |
|
"learning_rate": 0.00014076836149416887, |
|
"loss": 0.0994, |
|
"step": 41 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"learning_rate": 0.00013790884038403795, |
|
"loss": 0.1392, |
|
"step": 42 |
|
}, |
|
{ |
|
"epoch": 1.23, |
|
"learning_rate": 0.00013501264491913906, |
|
"loss": 0.0333, |
|
"step": 43 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"learning_rate": 0.00013208257698153677, |
|
"loss": 0.0522, |
|
"step": 44 |
|
}, |
|
{ |
|
"epoch": 1.29, |
|
"learning_rate": 0.00012912147122272523, |
|
"loss": 0.1114, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 1.31, |
|
"learning_rate": 0.00012613219232128608, |
|
"loss": 0.0667, |
|
"step": 46 |
|
}, |
|
{ |
|
"epoch": 1.34, |
|
"learning_rate": 0.000123117632211497, |
|
"loss": 0.0855, |
|
"step": 47 |
|
}, |
|
{ |
|
"epoch": 1.37, |
|
"learning_rate": 0.00012008070728557186, |
|
"loss": 0.056, |
|
"step": 48 |
|
}, |
|
{ |
|
"epoch": 1.4, |
|
"learning_rate": 0.00011702435557223987, |
|
"loss": 0.0742, |
|
"step": 49 |
|
}, |
|
{ |
|
"epoch": 1.43, |
|
"learning_rate": 0.00011395153389439233, |
|
"loss": 0.077, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 1.46, |
|
"learning_rate": 0.00011086521500854745, |
|
"loss": 0.0927, |
|
"step": 51 |
|
}, |
|
{ |
|
"epoch": 1.49, |
|
"learning_rate": 0.00010776838472890065, |
|
"loss": 0.0612, |
|
"step": 52 |
|
}, |
|
{ |
|
"epoch": 1.51, |
|
"learning_rate": 0.00010466403903874176, |
|
"loss": 0.0613, |
|
"step": 53 |
|
}, |
|
{ |
|
"epoch": 1.54, |
|
"learning_rate": 0.0001015551811920351, |
|
"loss": 0.0311, |
|
"step": 54 |
|
}, |
|
{ |
|
"epoch": 1.57, |
|
"learning_rate": 9.844481880796491e-05, |
|
"loss": 0.0606, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 1.6, |
|
"learning_rate": 9.533596096125825e-05, |
|
"loss": 0.042, |
|
"step": 56 |
|
}, |
|
{ |
|
"epoch": 1.63, |
|
"learning_rate": 9.223161527109937e-05, |
|
"loss": 0.047, |
|
"step": 57 |
|
}, |
|
{ |
|
"epoch": 1.66, |
|
"learning_rate": 8.913478499145254e-05, |
|
"loss": 0.0401, |
|
"step": 58 |
|
}, |
|
{ |
|
"epoch": 1.69, |
|
"learning_rate": 8.604846610560771e-05, |
|
"loss": 0.0409, |
|
"step": 59 |
|
}, |
|
{ |
|
"epoch": 1.71, |
|
"learning_rate": 8.297564442776014e-05, |
|
"loss": 0.0585, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 1.74, |
|
"learning_rate": 7.991929271442817e-05, |
|
"loss": 0.0765, |
|
"step": 61 |
|
}, |
|
{ |
|
"epoch": 1.77, |
|
"learning_rate": 7.688236778850306e-05, |
|
"loss": 0.0525, |
|
"step": 62 |
|
}, |
|
{ |
|
"epoch": 1.8, |
|
"learning_rate": 7.386780767871397e-05, |
|
"loss": 0.0291, |
|
"step": 63 |
|
}, |
|
{ |
|
"epoch": 1.83, |
|
"learning_rate": 7.087852877727481e-05, |
|
"loss": 0.0533, |
|
"step": 64 |
|
}, |
|
{ |
|
"epoch": 1.86, |
|
"learning_rate": 6.791742301846326e-05, |
|
"loss": 0.0793, |
|
"step": 65 |
|
}, |
|
{ |
|
"epoch": 1.89, |
|
"learning_rate": 6.498735508086093e-05, |
|
"loss": 0.049, |
|
"step": 66 |
|
}, |
|
{ |
|
"epoch": 1.91, |
|
"learning_rate": 6.209115961596208e-05, |
|
"loss": 0.0329, |
|
"step": 67 |
|
}, |
|
{ |
|
"epoch": 1.94, |
|
"learning_rate": 5.923163850583113e-05, |
|
"loss": 0.0172, |
|
"step": 68 |
|
}, |
|
{ |
|
"epoch": 1.97, |
|
"learning_rate": 5.6411558152462894e-05, |
|
"loss": 0.0504, |
|
"step": 69 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"learning_rate": 5.363364680146725e-05, |
|
"loss": 0.0378, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 2.03, |
|
"learning_rate": 5.090059190266779e-05, |
|
"loss": 0.0514, |
|
"step": 71 |
|
}, |
|
{ |
|
"epoch": 2.06, |
|
"learning_rate": 4.821503751016746e-05, |
|
"loss": 0.0286, |
|
"step": 72 |
|
}, |
|
{ |
|
"epoch": 2.09, |
|
"learning_rate": 4.5579581724397255e-05, |
|
"loss": 0.0494, |
|
"step": 73 |
|
}, |
|
{ |
|
"epoch": 2.11, |
|
"learning_rate": 4.2996774178621736e-05, |
|
"loss": 0.0181, |
|
"step": 74 |
|
}, |
|
{ |
|
"epoch": 2.14, |
|
"learning_rate": 4.046911357233343e-05, |
|
"loss": 0.0308, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 2.17, |
|
"learning_rate": 3.79990452539225e-05, |
|
"loss": 0.0375, |
|
"step": 76 |
|
}, |
|
{ |
|
"epoch": 2.2, |
|
"learning_rate": 3.558895885496023e-05, |
|
"loss": 0.0085, |
|
"step": 77 |
|
}, |
|
{ |
|
"epoch": 2.23, |
|
"learning_rate": 3.324118597838464e-05, |
|
"loss": 0.028, |
|
"step": 78 |
|
}, |
|
{ |
|
"epoch": 2.26, |
|
"learning_rate": 3.0957997942825336e-05, |
|
"loss": 0.0153, |
|
"step": 79 |
|
}, |
|
{ |
|
"epoch": 2.29, |
|
"learning_rate": 2.874160358524931e-05, |
|
"loss": 0.0378, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 2.31, |
|
"learning_rate": 2.659414712405398e-05, |
|
"loss": 0.0233, |
|
"step": 81 |
|
}, |
|
{ |
|
"epoch": 2.34, |
|
"learning_rate": 2.451770608467432e-05, |
|
"loss": 0.0047, |
|
"step": 82 |
|
}, |
|
{ |
|
"epoch": 2.37, |
|
"learning_rate": 2.251428928971102e-05, |
|
"loss": 0.0045, |
|
"step": 83 |
|
}, |
|
{ |
|
"epoch": 2.4, |
|
"learning_rate": 2.058583491552465e-05, |
|
"loss": 0.0161, |
|
"step": 84 |
|
}, |
|
{ |
|
"epoch": 2.43, |
|
"learning_rate": 1.8734208617174988e-05, |
|
"loss": 0.0424, |
|
"step": 85 |
|
}, |
|
{ |
|
"epoch": 2.46, |
|
"learning_rate": 1.696120172352025e-05, |
|
"loss": 0.0167, |
|
"step": 86 |
|
}, |
|
{ |
|
"epoch": 2.49, |
|
"learning_rate": 1.526852950422226e-05, |
|
"loss": 0.0125, |
|
"step": 87 |
|
}, |
|
{ |
|
"epoch": 2.51, |
|
"learning_rate": 1.3657829510333654e-05, |
|
"loss": 0.0236, |
|
"step": 88 |
|
}, |
|
{ |
|
"epoch": 2.54, |
|
"learning_rate": 1.2130659990073146e-05, |
|
"loss": 0.0089, |
|
"step": 89 |
|
}, |
|
{ |
|
"epoch": 2.57, |
|
"learning_rate": 1.0688498381320855e-05, |
|
"loss": 0.0202, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 2.6, |
|
"learning_rate": 9.332739882292752e-06, |
|
"loss": 0.0111, |
|
"step": 91 |
|
}, |
|
{ |
|
"epoch": 2.63, |
|
"learning_rate": 8.064696101776358e-06, |
|
"loss": 0.0025, |
|
"step": 92 |
|
}, |
|
{ |
|
"epoch": 2.66, |
|
"learning_rate": 6.8855937902340576e-06, |
|
"loss": 0.0209, |
|
"step": 93 |
|
}, |
|
{ |
|
"epoch": 2.69, |
|
"learning_rate": 5.7965736530010916e-06, |
|
"loss": 0.0318, |
|
"step": 94 |
|
}, |
|
{ |
|
"epoch": 2.71, |
|
"learning_rate": 4.798689246727006e-06, |
|
"loss": 0.0334, |
|
"step": 95 |
|
}, |
|
{ |
|
"epoch": 2.74, |
|
"learning_rate": 3.892905960127546e-06, |
|
"loss": 0.0117, |
|
"step": 96 |
|
}, |
|
{ |
|
"epoch": 2.77, |
|
"learning_rate": 3.0801000800333877e-06, |
|
"loss": 0.0134, |
|
"step": 97 |
|
}, |
|
{ |
|
"epoch": 2.8, |
|
"learning_rate": 2.3610579436393e-06, |
|
"loss": 0.0279, |
|
"step": 98 |
|
}, |
|
{ |
|
"epoch": 2.83, |
|
"learning_rate": 1.7364751777736332e-06, |
|
"loss": 0.0071, |
|
"step": 99 |
|
}, |
|
{ |
|
"epoch": 2.86, |
|
"learning_rate": 1.2069560259243328e-06, |
|
"loss": 0.0162, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 2.89, |
|
"learning_rate": 7.730127636723539e-07, |
|
"loss": 0.0029, |
|
"step": 101 |
|
}, |
|
{ |
|
"epoch": 2.91, |
|
"learning_rate": 4.3506520309813947e-07, |
|
"loss": 0.005, |
|
"step": 102 |
|
}, |
|
{ |
|
"epoch": 2.94, |
|
"learning_rate": 1.9344028664056713e-07, |
|
"loss": 0.0189, |
|
"step": 103 |
|
}, |
|
{ |
|
"epoch": 2.97, |
|
"learning_rate": 4.837177080119215e-08, |
|
"loss": 0.0099, |
|
"step": 104 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"learning_rate": 0.0, |
|
"loss": 0.0075, |
|
"step": 105 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"step": 105, |
|
"total_flos": 14279656734720.0, |
|
"train_loss": 0.12493001309977401, |
|
"train_runtime": 603.9084, |
|
"train_samples_per_second": 21.743, |
|
"train_steps_per_second": 0.174 |
|
} |
|
], |
|
"logging_steps": 1.0, |
|
"max_steps": 105, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 3, |
|
"save_steps": 50000, |
|
"total_flos": 14279656734720.0, |
|
"train_batch_size": 16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|