| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.0379066075955365, | |
| "eval_steps": 500, | |
| "global_step": 100, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.000379066075955365, | |
| "grad_norm": 2.788883924484253, | |
| "learning_rate": 2e-05, | |
| "loss": 1.4892, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.00075813215191073, | |
| "grad_norm": 2.803041934967041, | |
| "learning_rate": 4e-05, | |
| "loss": 1.4967, | |
| "step": 2 | |
| }, | |
| { | |
| "epoch": 0.001137198227866095, | |
| "grad_norm": 2.5324907302856445, | |
| "learning_rate": 6e-05, | |
| "loss": 1.3553, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 0.00151626430382146, | |
| "grad_norm": 1.693648099899292, | |
| "learning_rate": 8e-05, | |
| "loss": 0.983, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 0.0018953303797768248, | |
| "grad_norm": 1.2373262643814087, | |
| "learning_rate": 0.0001, | |
| "loss": 0.6743, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.00227439645573219, | |
| "grad_norm": 0.9896576404571533, | |
| "learning_rate": 0.00012, | |
| "loss": 0.4505, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 0.0026534625316875548, | |
| "grad_norm": 1.0764743089675903, | |
| "learning_rate": 0.00014, | |
| "loss": 0.3412, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 0.00303252860764292, | |
| "grad_norm": 0.4975992739200592, | |
| "learning_rate": 0.00016, | |
| "loss": 0.2131, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 0.0034115946835982847, | |
| "grad_norm": 0.3490230143070221, | |
| "learning_rate": 0.00018, | |
| "loss": 0.1971, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 0.0037906607595536495, | |
| "grad_norm": 0.2667368948459625, | |
| "learning_rate": 0.0002, | |
| "loss": 0.1833, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.004169726835509015, | |
| "grad_norm": 0.25213170051574707, | |
| "learning_rate": 0.0001999390827019096, | |
| "loss": 0.1652, | |
| "step": 11 | |
| }, | |
| { | |
| "epoch": 0.00454879291146438, | |
| "grad_norm": 0.18969492614269257, | |
| "learning_rate": 0.00019975640502598244, | |
| "loss": 0.1606, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 0.004927858987419744, | |
| "grad_norm": 0.15731994807720184, | |
| "learning_rate": 0.00019945218953682734, | |
| "loss": 0.152, | |
| "step": 13 | |
| }, | |
| { | |
| "epoch": 0.0053069250633751095, | |
| "grad_norm": 0.14271649718284607, | |
| "learning_rate": 0.00019902680687415705, | |
| "loss": 0.1435, | |
| "step": 14 | |
| }, | |
| { | |
| "epoch": 0.005685991139330475, | |
| "grad_norm": 0.16122481226921082, | |
| "learning_rate": 0.00019848077530122083, | |
| "loss": 0.1609, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.00606505721528584, | |
| "grad_norm": 0.15835386514663696, | |
| "learning_rate": 0.00019781476007338058, | |
| "loss": 0.1444, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 0.006444123291241204, | |
| "grad_norm": 0.11050277203321457, | |
| "learning_rate": 0.00019702957262759965, | |
| "loss": 0.128, | |
| "step": 17 | |
| }, | |
| { | |
| "epoch": 0.0068231893671965695, | |
| "grad_norm": 0.12347878515720367, | |
| "learning_rate": 0.0001961261695938319, | |
| "loss": 0.1326, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 0.007202255443151935, | |
| "grad_norm": 0.09975384920835495, | |
| "learning_rate": 0.00019510565162951537, | |
| "loss": 0.1187, | |
| "step": 19 | |
| }, | |
| { | |
| "epoch": 0.007581321519107299, | |
| "grad_norm": 0.1068282350897789, | |
| "learning_rate": 0.00019396926207859084, | |
| "loss": 0.1202, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.007960387595062665, | |
| "grad_norm": 0.1021302118897438, | |
| "learning_rate": 0.00019271838545667876, | |
| "loss": 0.1297, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 0.00833945367101803, | |
| "grad_norm": 0.09192782640457153, | |
| "learning_rate": 0.0001913545457642601, | |
| "loss": 0.1227, | |
| "step": 22 | |
| }, | |
| { | |
| "epoch": 0.008718519746973394, | |
| "grad_norm": 0.07424570620059967, | |
| "learning_rate": 0.0001898794046299167, | |
| "loss": 0.1104, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 0.00909758582292876, | |
| "grad_norm": 0.09622485190629959, | |
| "learning_rate": 0.00018829475928589271, | |
| "loss": 0.1279, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 0.009476651898884124, | |
| "grad_norm": 0.08250657469034195, | |
| "learning_rate": 0.00018660254037844388, | |
| "loss": 0.1152, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.009855717974839489, | |
| "grad_norm": 0.09128892421722412, | |
| "learning_rate": 0.0001848048096156426, | |
| "loss": 0.1072, | |
| "step": 26 | |
| }, | |
| { | |
| "epoch": 0.010234784050794855, | |
| "grad_norm": 0.11655224114656448, | |
| "learning_rate": 0.00018290375725550417, | |
| "loss": 0.1156, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 0.010613850126750219, | |
| "grad_norm": 0.09212372452020645, | |
| "learning_rate": 0.00018090169943749476, | |
| "loss": 0.107, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 0.010992916202705583, | |
| "grad_norm": 0.1035194918513298, | |
| "learning_rate": 0.00017880107536067218, | |
| "loss": 0.1169, | |
| "step": 29 | |
| }, | |
| { | |
| "epoch": 0.01137198227866095, | |
| "grad_norm": 0.09886885434389114, | |
| "learning_rate": 0.0001766044443118978, | |
| "loss": 0.114, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.011751048354616314, | |
| "grad_norm": 0.07283884286880493, | |
| "learning_rate": 0.00017431448254773944, | |
| "loss": 0.0962, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 0.01213011443057168, | |
| "grad_norm": 0.09377507120370865, | |
| "learning_rate": 0.0001719339800338651, | |
| "loss": 0.0925, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 0.012509180506527044, | |
| "grad_norm": 0.09097617864608765, | |
| "learning_rate": 0.00016946583704589973, | |
| "loss": 0.1045, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 0.012888246582482409, | |
| "grad_norm": 0.10535702854394913, | |
| "learning_rate": 0.00016691306063588583, | |
| "loss": 0.1005, | |
| "step": 34 | |
| }, | |
| { | |
| "epoch": 0.013267312658437775, | |
| "grad_norm": 0.09806367754936218, | |
| "learning_rate": 0.00016427876096865394, | |
| "loss": 0.102, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.013646378734393139, | |
| "grad_norm": 0.09789954870939255, | |
| "learning_rate": 0.0001615661475325658, | |
| "loss": 0.0955, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 0.014025444810348503, | |
| "grad_norm": 0.1131892129778862, | |
| "learning_rate": 0.00015877852522924732, | |
| "loss": 0.0853, | |
| "step": 37 | |
| }, | |
| { | |
| "epoch": 0.01440451088630387, | |
| "grad_norm": 0.08628207445144653, | |
| "learning_rate": 0.0001559192903470747, | |
| "loss": 0.0901, | |
| "step": 38 | |
| }, | |
| { | |
| "epoch": 0.014783576962259234, | |
| "grad_norm": 0.11414898931980133, | |
| "learning_rate": 0.0001529919264233205, | |
| "loss": 0.0993, | |
| "step": 39 | |
| }, | |
| { | |
| "epoch": 0.015162643038214598, | |
| "grad_norm": 0.07424012571573257, | |
| "learning_rate": 0.00015000000000000001, | |
| "loss": 0.1037, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.015541709114169964, | |
| "grad_norm": 0.08380600810050964, | |
| "learning_rate": 0.00014694715627858908, | |
| "loss": 0.0839, | |
| "step": 41 | |
| }, | |
| { | |
| "epoch": 0.01592077519012533, | |
| "grad_norm": 0.06833647936582565, | |
| "learning_rate": 0.00014383711467890774, | |
| "loss": 0.0916, | |
| "step": 42 | |
| }, | |
| { | |
| "epoch": 0.016299841266080695, | |
| "grad_norm": 0.06410624086856842, | |
| "learning_rate": 0.00014067366430758004, | |
| "loss": 0.0845, | |
| "step": 43 | |
| }, | |
| { | |
| "epoch": 0.01667890734203606, | |
| "grad_norm": 0.06120501831173897, | |
| "learning_rate": 0.00013746065934159123, | |
| "loss": 0.0812, | |
| "step": 44 | |
| }, | |
| { | |
| "epoch": 0.017057973417991423, | |
| "grad_norm": 0.06658154726028442, | |
| "learning_rate": 0.00013420201433256689, | |
| "loss": 0.0868, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.017437039493946788, | |
| "grad_norm": 0.07054657489061356, | |
| "learning_rate": 0.00013090169943749476, | |
| "loss": 0.0885, | |
| "step": 46 | |
| }, | |
| { | |
| "epoch": 0.017816105569902152, | |
| "grad_norm": 0.05874604731798172, | |
| "learning_rate": 0.0001275637355816999, | |
| "loss": 0.0845, | |
| "step": 47 | |
| }, | |
| { | |
| "epoch": 0.01819517164585752, | |
| "grad_norm": 0.07008881121873856, | |
| "learning_rate": 0.00012419218955996676, | |
| "loss": 0.0775, | |
| "step": 48 | |
| }, | |
| { | |
| "epoch": 0.018574237721812884, | |
| "grad_norm": 0.07047414779663086, | |
| "learning_rate": 0.00012079116908177593, | |
| "loss": 0.0839, | |
| "step": 49 | |
| }, | |
| { | |
| "epoch": 0.01895330379776825, | |
| "grad_norm": 0.07158780097961426, | |
| "learning_rate": 0.00011736481776669306, | |
| "loss": 0.0993, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.019332369873723613, | |
| "grad_norm": 0.06991136819124222, | |
| "learning_rate": 0.00011391731009600654, | |
| "loss": 0.0873, | |
| "step": 51 | |
| }, | |
| { | |
| "epoch": 0.019711435949678977, | |
| "grad_norm": 0.06695719808340073, | |
| "learning_rate": 0.00011045284632676536, | |
| "loss": 0.0876, | |
| "step": 52 | |
| }, | |
| { | |
| "epoch": 0.020090502025634345, | |
| "grad_norm": 0.07722793519496918, | |
| "learning_rate": 0.00010697564737441252, | |
| "loss": 0.0977, | |
| "step": 53 | |
| }, | |
| { | |
| "epoch": 0.02046956810158971, | |
| "grad_norm": 0.06521150469779968, | |
| "learning_rate": 0.00010348994967025012, | |
| "loss": 0.0856, | |
| "step": 54 | |
| }, | |
| { | |
| "epoch": 0.020848634177545074, | |
| "grad_norm": 0.07508374005556107, | |
| "learning_rate": 0.0001, | |
| "loss": 0.0876, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.021227700253500438, | |
| "grad_norm": 0.05650763958692551, | |
| "learning_rate": 9.651005032974994e-05, | |
| "loss": 0.0837, | |
| "step": 56 | |
| }, | |
| { | |
| "epoch": 0.021606766329455802, | |
| "grad_norm": 0.07390966266393661, | |
| "learning_rate": 9.302435262558747e-05, | |
| "loss": 0.0879, | |
| "step": 57 | |
| }, | |
| { | |
| "epoch": 0.021985832405411167, | |
| "grad_norm": 0.058437567204236984, | |
| "learning_rate": 8.954715367323468e-05, | |
| "loss": 0.0773, | |
| "step": 58 | |
| }, | |
| { | |
| "epoch": 0.022364898481366535, | |
| "grad_norm": 0.06248459964990616, | |
| "learning_rate": 8.608268990399349e-05, | |
| "loss": 0.0788, | |
| "step": 59 | |
| }, | |
| { | |
| "epoch": 0.0227439645573219, | |
| "grad_norm": 0.0653211697936058, | |
| "learning_rate": 8.263518223330697e-05, | |
| "loss": 0.0855, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.023123030633277263, | |
| "grad_norm": 0.062243081629276276, | |
| "learning_rate": 7.920883091822408e-05, | |
| "loss": 0.0886, | |
| "step": 61 | |
| }, | |
| { | |
| "epoch": 0.023502096709232628, | |
| "grad_norm": 0.055676043033599854, | |
| "learning_rate": 7.580781044003324e-05, | |
| "loss": 0.0897, | |
| "step": 62 | |
| }, | |
| { | |
| "epoch": 0.023881162785187992, | |
| "grad_norm": 0.05823361128568649, | |
| "learning_rate": 7.243626441830009e-05, | |
| "loss": 0.0753, | |
| "step": 63 | |
| }, | |
| { | |
| "epoch": 0.02426022886114336, | |
| "grad_norm": 0.047967329621315, | |
| "learning_rate": 6.909830056250527e-05, | |
| "loss": 0.0804, | |
| "step": 64 | |
| }, | |
| { | |
| "epoch": 0.024639294937098724, | |
| "grad_norm": 0.04651477187871933, | |
| "learning_rate": 6.579798566743314e-05, | |
| "loss": 0.0737, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.02501836101305409, | |
| "grad_norm": 0.06440019607543945, | |
| "learning_rate": 6.25393406584088e-05, | |
| "loss": 0.0931, | |
| "step": 66 | |
| }, | |
| { | |
| "epoch": 0.025397427089009453, | |
| "grad_norm": 0.0673041045665741, | |
| "learning_rate": 5.9326335692419995e-05, | |
| "loss": 0.0926, | |
| "step": 67 | |
| }, | |
| { | |
| "epoch": 0.025776493164964817, | |
| "grad_norm": 0.06460334360599518, | |
| "learning_rate": 5.616288532109225e-05, | |
| "loss": 0.1055, | |
| "step": 68 | |
| }, | |
| { | |
| "epoch": 0.02615555924092018, | |
| "grad_norm": 0.06001191958785057, | |
| "learning_rate": 5.305284372141095e-05, | |
| "loss": 0.0815, | |
| "step": 69 | |
| }, | |
| { | |
| "epoch": 0.02653462531687555, | |
| "grad_norm": 0.05296599864959717, | |
| "learning_rate": 5.000000000000002e-05, | |
| "loss": 0.0837, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.026913691392830914, | |
| "grad_norm": 0.0675460696220398, | |
| "learning_rate": 4.700807357667952e-05, | |
| "loss": 0.0848, | |
| "step": 71 | |
| }, | |
| { | |
| "epoch": 0.027292757468786278, | |
| "grad_norm": 0.05409557744860649, | |
| "learning_rate": 4.4080709652925336e-05, | |
| "loss": 0.0779, | |
| "step": 72 | |
| }, | |
| { | |
| "epoch": 0.027671823544741642, | |
| "grad_norm": 0.05015251785516739, | |
| "learning_rate": 4.12214747707527e-05, | |
| "loss": 0.0788, | |
| "step": 73 | |
| }, | |
| { | |
| "epoch": 0.028050889620697007, | |
| "grad_norm": 0.05301973596215248, | |
| "learning_rate": 3.843385246743417e-05, | |
| "loss": 0.0764, | |
| "step": 74 | |
| }, | |
| { | |
| "epoch": 0.02842995569665237, | |
| "grad_norm": 0.05640785023570061, | |
| "learning_rate": 3.5721239031346066e-05, | |
| "loss": 0.0867, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.02880902177260774, | |
| "grad_norm": 0.055719390511512756, | |
| "learning_rate": 3.308693936411421e-05, | |
| "loss": 0.0788, | |
| "step": 76 | |
| }, | |
| { | |
| "epoch": 0.029188087848563103, | |
| "grad_norm": 0.06163398548960686, | |
| "learning_rate": 3.053416295410026e-05, | |
| "loss": 0.0807, | |
| "step": 77 | |
| }, | |
| { | |
| "epoch": 0.029567153924518468, | |
| "grad_norm": 0.06289924681186676, | |
| "learning_rate": 2.8066019966134904e-05, | |
| "loss": 0.0808, | |
| "step": 78 | |
| }, | |
| { | |
| "epoch": 0.029946220000473832, | |
| "grad_norm": 0.056900832802057266, | |
| "learning_rate": 2.5685517452260567e-05, | |
| "loss": 0.0762, | |
| "step": 79 | |
| }, | |
| { | |
| "epoch": 0.030325286076429196, | |
| "grad_norm": 0.06545019149780273, | |
| "learning_rate": 2.339555568810221e-05, | |
| "loss": 0.0929, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.030704352152384564, | |
| "grad_norm": 0.060016706585884094, | |
| "learning_rate": 2.119892463932781e-05, | |
| "loss": 0.0752, | |
| "step": 81 | |
| }, | |
| { | |
| "epoch": 0.03108341822833993, | |
| "grad_norm": 0.055878885090351105, | |
| "learning_rate": 1.9098300562505266e-05, | |
| "loss": 0.0753, | |
| "step": 82 | |
| }, | |
| { | |
| "epoch": 0.03146248430429529, | |
| "grad_norm": 0.06015196815133095, | |
| "learning_rate": 1.7096242744495837e-05, | |
| "loss": 0.086, | |
| "step": 83 | |
| }, | |
| { | |
| "epoch": 0.03184155038025066, | |
| "grad_norm": 0.05178290605545044, | |
| "learning_rate": 1.5195190384357404e-05, | |
| "loss": 0.0793, | |
| "step": 84 | |
| }, | |
| { | |
| "epoch": 0.032220616456206025, | |
| "grad_norm": 0.05161258578300476, | |
| "learning_rate": 1.339745962155613e-05, | |
| "loss": 0.0809, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 0.03259968253216139, | |
| "grad_norm": 0.05504770204424858, | |
| "learning_rate": 1.1705240714107302e-05, | |
| "loss": 0.0772, | |
| "step": 86 | |
| }, | |
| { | |
| "epoch": 0.032978748608116754, | |
| "grad_norm": 0.05566380172967911, | |
| "learning_rate": 1.0120595370083318e-05, | |
| "loss": 0.0726, | |
| "step": 87 | |
| }, | |
| { | |
| "epoch": 0.03335781468407212, | |
| "grad_norm": 0.05389956384897232, | |
| "learning_rate": 8.645454235739903e-06, | |
| "loss": 0.0779, | |
| "step": 88 | |
| }, | |
| { | |
| "epoch": 0.03373688076002748, | |
| "grad_norm": 0.05445749685168266, | |
| "learning_rate": 7.281614543321269e-06, | |
| "loss": 0.076, | |
| "step": 89 | |
| }, | |
| { | |
| "epoch": 0.03411594683598285, | |
| "grad_norm": 0.04557236656546593, | |
| "learning_rate": 6.030737921409169e-06, | |
| "loss": 0.075, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.03449501291193821, | |
| "grad_norm": 0.050466809421777725, | |
| "learning_rate": 4.8943483704846475e-06, | |
| "loss": 0.0747, | |
| "step": 91 | |
| }, | |
| { | |
| "epoch": 0.034874078987893575, | |
| "grad_norm": 0.04975885897874832, | |
| "learning_rate": 3.873830406168111e-06, | |
| "loss": 0.0746, | |
| "step": 92 | |
| }, | |
| { | |
| "epoch": 0.03525314506384894, | |
| "grad_norm": 0.055442556738853455, | |
| "learning_rate": 2.970427372400353e-06, | |
| "loss": 0.0807, | |
| "step": 93 | |
| }, | |
| { | |
| "epoch": 0.035632211139804304, | |
| "grad_norm": 0.05043969675898552, | |
| "learning_rate": 2.1852399266194314e-06, | |
| "loss": 0.0741, | |
| "step": 94 | |
| }, | |
| { | |
| "epoch": 0.036011277215759675, | |
| "grad_norm": 0.05411124229431152, | |
| "learning_rate": 1.5192246987791981e-06, | |
| "loss": 0.0741, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 0.03639034329171504, | |
| "grad_norm": 0.06104011833667755, | |
| "learning_rate": 9.731931258429638e-07, | |
| "loss": 0.0822, | |
| "step": 96 | |
| }, | |
| { | |
| "epoch": 0.036769409367670404, | |
| "grad_norm": 0.05103430524468422, | |
| "learning_rate": 5.478104631726711e-07, | |
| "loss": 0.0716, | |
| "step": 97 | |
| }, | |
| { | |
| "epoch": 0.03714847544362577, | |
| "grad_norm": 0.057701822370290756, | |
| "learning_rate": 2.4359497401758024e-07, | |
| "loss": 0.0767, | |
| "step": 98 | |
| }, | |
| { | |
| "epoch": 0.03752754151958113, | |
| "grad_norm": 0.05738105624914169, | |
| "learning_rate": 6.09172980904238e-08, | |
| "loss": 0.0837, | |
| "step": 99 | |
| }, | |
| { | |
| "epoch": 0.0379066075955365, | |
| "grad_norm": 0.05406734347343445, | |
| "learning_rate": 0.0, | |
| "loss": 0.0856, | |
| "step": 100 | |
| } | |
| ], | |
| "logging_steps": 1, | |
| "max_steps": 100, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 50, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 3.888051858648392e+17, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |