| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.019436345966958212, | |
| "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.0, | |
| "grad_norm": 4.318624496459961, | |
| "learning_rate": 0.00019805941782534764, | |
| "loss": 9.3465, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 6.591656684875488, | |
| "learning_rate": 0.00019605881764529358, | |
| "loss": 12.8608, | |
| "step": 2 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 9.56568431854248, | |
| "learning_rate": 0.00019405821746523957, | |
| "loss": 11.289, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 7.295264720916748, | |
| "learning_rate": 0.00019205761728518557, | |
| "loss": 6.2966, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 6.040774822235107, | |
| "learning_rate": 0.00019005701710513156, | |
| "loss": 4.4317, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 13.841460227966309, | |
| "learning_rate": 0.00018805641692507753, | |
| "loss": 5.0921, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 12.43684196472168, | |
| "learning_rate": 0.00018605581674502352, | |
| "loss": 3.8523, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 7.349277973175049, | |
| "learning_rate": 0.00018405521656496952, | |
| "loss": 3.0653, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 2.4729762077331543, | |
| "learning_rate": 0.00018205461638491548, | |
| "loss": 2.5825, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 3.187196969985962, | |
| "learning_rate": 0.00018005401620486148, | |
| "loss": 2.1595, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 2.3855996131896973, | |
| "learning_rate": 0.00017805341602480744, | |
| "loss": 2.1515, | |
| "step": 11 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 2.119424343109131, | |
| "learning_rate": 0.00017605281584475344, | |
| "loss": 2.3592, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 1.1523925065994263, | |
| "learning_rate": 0.00017405221566469943, | |
| "loss": 2.3854, | |
| "step": 13 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 2.134817361831665, | |
| "learning_rate": 0.00017205161548464542, | |
| "loss": 1.4468, | |
| "step": 14 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 1.452102780342102, | |
| "learning_rate": 0.00017005101530459136, | |
| "loss": 2.0123, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 4.131059646606445, | |
| "learning_rate": 0.00016805041512453736, | |
| "loss": 2.1094, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 2.092371940612793, | |
| "learning_rate": 0.00016604981494448335, | |
| "loss": 1.5696, | |
| "step": 17 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 1.253164529800415, | |
| "learning_rate": 0.00016404921476442935, | |
| "loss": 1.4767, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 1.338577151298523, | |
| "learning_rate": 0.0001620486145843753, | |
| "loss": 2.338, | |
| "step": 19 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 1.121963381767273, | |
| "learning_rate": 0.0001600480144043213, | |
| "loss": 2.0752, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 1.5833094120025635, | |
| "learning_rate": 0.0001580474142242673, | |
| "loss": 1.4675, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 1.428231954574585, | |
| "learning_rate": 0.00015604681404421327, | |
| "loss": 1.3554, | |
| "step": 22 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 1.5906881093978882, | |
| "learning_rate": 0.00015404621386415926, | |
| "loss": 1.703, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 1.2706618309020996, | |
| "learning_rate": 0.00015204561368410523, | |
| "loss": 1.7487, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 0.8932807445526123, | |
| "learning_rate": 0.00015004501350405122, | |
| "loss": 1.8345, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.8607711791992188, | |
| "learning_rate": 0.00014804441332399721, | |
| "loss": 1.1681, | |
| "step": 26 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.011381983757019, | |
| "learning_rate": 0.0001460438131439432, | |
| "loss": 1.7756, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.0185519456863403, | |
| "learning_rate": 0.00014404321296388918, | |
| "loss": 1.3468, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.6411266326904297, | |
| "learning_rate": 0.00014204261278383514, | |
| "loss": 2.116, | |
| "step": 29 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.9967129826545715, | |
| "learning_rate": 0.00014004201260378114, | |
| "loss": 1.6774, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.3136436939239502, | |
| "learning_rate": 0.00013804141242372713, | |
| "loss": 2.0461, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.4330521821975708, | |
| "learning_rate": 0.00013604081224367312, | |
| "loss": 2.1365, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.1006214618682861, | |
| "learning_rate": 0.0001340402120636191, | |
| "loss": 0.4195, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.9954245686531067, | |
| "learning_rate": 0.00013203961188356508, | |
| "loss": 1.4915, | |
| "step": 34 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.8714771270751953, | |
| "learning_rate": 0.00013003901170351108, | |
| "loss": 1.4862, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.8504552841186523, | |
| "learning_rate": 0.00012803841152345704, | |
| "loss": 1.2124, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.5393089056015015, | |
| "learning_rate": 0.000126037811343403, | |
| "loss": 2.0633, | |
| "step": 37 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.018890142440796, | |
| "learning_rate": 0.000124037211163349, | |
| "loss": 0.6224, | |
| "step": 38 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.2864466905593872, | |
| "learning_rate": 0.000122036610983295, | |
| "loss": 1.795, | |
| "step": 39 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.0033133029937744, | |
| "learning_rate": 0.00012003601080324098, | |
| "loss": 0.7465, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 3.63820481300354, | |
| "learning_rate": 0.00011803541062318697, | |
| "loss": 1.3544, | |
| "step": 41 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 3.668487787246704, | |
| "learning_rate": 0.00011603481044313295, | |
| "loss": 1.9775, | |
| "step": 42 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 11.158123016357422, | |
| "learning_rate": 0.00011403421026307892, | |
| "loss": 1.2352, | |
| "step": 43 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 3.3200602531433105, | |
| "learning_rate": 0.00011203361008302491, | |
| "loss": 1.5836, | |
| "step": 44 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.0530873537063599, | |
| "learning_rate": 0.0001100330099029709, | |
| "loss": 1.5381, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.1884853839874268, | |
| "learning_rate": 0.00010803240972291689, | |
| "loss": 1.6451, | |
| "step": 46 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.3888977766036987, | |
| "learning_rate": 0.00010603180954286287, | |
| "loss": 1.5273, | |
| "step": 47 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.9774441123008728, | |
| "learning_rate": 0.00010403120936280886, | |
| "loss": 1.9541, | |
| "step": 48 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.068190097808838, | |
| "learning_rate": 0.00010203060918275482, | |
| "loss": 1.6002, | |
| "step": 49 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.28312087059021, | |
| "learning_rate": 0.00010003000900270081, | |
| "loss": 2.0254, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.3461538553237915, | |
| "learning_rate": 9.802940882264679e-05, | |
| "loss": 1.3561, | |
| "step": 51 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.3271141052246094, | |
| "learning_rate": 9.602880864259278e-05, | |
| "loss": 1.361, | |
| "step": 52 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.3316106796264648, | |
| "learning_rate": 9.402820846253876e-05, | |
| "loss": 1.7272, | |
| "step": 53 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.97672039270401, | |
| "learning_rate": 9.202760828248476e-05, | |
| "loss": 0.708, | |
| "step": 54 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.9566367864608765, | |
| "learning_rate": 9.002700810243074e-05, | |
| "loss": 1.5139, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.0057029724121094, | |
| "learning_rate": 8.802640792237672e-05, | |
| "loss": 1.2497, | |
| "step": 56 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 9.65938663482666, | |
| "learning_rate": 8.602580774232271e-05, | |
| "loss": 1.4, | |
| "step": 57 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 2.8687500953674316, | |
| "learning_rate": 8.402520756226868e-05, | |
| "loss": 2.0894, | |
| "step": 58 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 7.262545108795166, | |
| "learning_rate": 8.202460738221467e-05, | |
| "loss": 0.9389, | |
| "step": 59 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.3187130689620972, | |
| "learning_rate": 8.002400720216065e-05, | |
| "loss": 1.3847, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 5.589620590209961, | |
| "learning_rate": 7.802340702210663e-05, | |
| "loss": 0.4888, | |
| "step": 61 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.8035283088684082, | |
| "learning_rate": 7.602280684205261e-05, | |
| "loss": 1.3295, | |
| "step": 62 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.8805381059646606, | |
| "learning_rate": 7.402220666199861e-05, | |
| "loss": 1.6203, | |
| "step": 63 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.222285509109497, | |
| "learning_rate": 7.202160648194459e-05, | |
| "loss": 1.3659, | |
| "step": 64 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.1526193618774414, | |
| "learning_rate": 7.002100630189057e-05, | |
| "loss": 1.9816, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.8536458015441895, | |
| "learning_rate": 6.802040612183656e-05, | |
| "loss": 1.0255, | |
| "step": 66 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.1652932167053223, | |
| "learning_rate": 6.601980594178254e-05, | |
| "loss": 0.9737, | |
| "step": 67 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.8228743076324463, | |
| "learning_rate": 6.401920576172852e-05, | |
| "loss": 0.9365, | |
| "step": 68 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.7666674852371216, | |
| "learning_rate": 6.20186055816745e-05, | |
| "loss": 1.1326, | |
| "step": 69 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.9048521518707275, | |
| "learning_rate": 6.001800540162049e-05, | |
| "loss": 1.8278, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.013771891593933, | |
| "learning_rate": 5.801740522156648e-05, | |
| "loss": 1.2261, | |
| "step": 71 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.0149589776992798, | |
| "learning_rate": 5.601680504151246e-05, | |
| "loss": 1.7701, | |
| "step": 72 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.3616000413894653, | |
| "learning_rate": 5.4016204861458444e-05, | |
| "loss": 1.3506, | |
| "step": 73 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.3497377634048462, | |
| "learning_rate": 5.201560468140443e-05, | |
| "loss": 1.1989, | |
| "step": 74 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.9258934855461121, | |
| "learning_rate": 5.0015004501350405e-05, | |
| "loss": 1.5837, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.7765821814537048, | |
| "learning_rate": 4.801440432129639e-05, | |
| "loss": 1.3821, | |
| "step": 76 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 1.5330684185028076, | |
| "learning_rate": 4.601380414124238e-05, | |
| "loss": 1.2167, | |
| "step": 77 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.8043363690376282, | |
| "learning_rate": 4.401320396118836e-05, | |
| "loss": 1.0048, | |
| "step": 78 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.0334686040878296, | |
| "learning_rate": 4.201260378113434e-05, | |
| "loss": 1.0847, | |
| "step": 79 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.02166748046875, | |
| "learning_rate": 4.0012003601080326e-05, | |
| "loss": 1.1226, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.7887734770774841, | |
| "learning_rate": 3.801140342102631e-05, | |
| "loss": 0.7872, | |
| "step": 81 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.9185380339622498, | |
| "learning_rate": 3.6010803240972294e-05, | |
| "loss": 0.9887, | |
| "step": 82 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.8893826007843018, | |
| "learning_rate": 3.401020306091828e-05, | |
| "loss": 1.3784, | |
| "step": 83 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.1666996479034424, | |
| "learning_rate": 3.200960288086426e-05, | |
| "loss": 1.5805, | |
| "step": 84 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.330802321434021, | |
| "learning_rate": 3.0009002700810245e-05, | |
| "loss": 1.2388, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.092350959777832, | |
| "learning_rate": 2.800840252075623e-05, | |
| "loss": 1.7721, | |
| "step": 86 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.0547586679458618, | |
| "learning_rate": 2.6007802340702216e-05, | |
| "loss": 1.8803, | |
| "step": 87 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.0033807754516602, | |
| "learning_rate": 2.4007202160648196e-05, | |
| "loss": 1.6825, | |
| "step": 88 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.6005492210388184, | |
| "learning_rate": 2.200660198059418e-05, | |
| "loss": 1.9344, | |
| "step": 89 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.8459662795066833, | |
| "learning_rate": 2.0006001800540163e-05, | |
| "loss": 1.0833, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.7953637838363647, | |
| "learning_rate": 1.8005401620486147e-05, | |
| "loss": 1.2859, | |
| "step": 91 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.0880411863327026, | |
| "learning_rate": 1.600480144043213e-05, | |
| "loss": 1.3714, | |
| "step": 92 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.3695777654647827, | |
| "learning_rate": 1.4004201260378114e-05, | |
| "loss": 1.3642, | |
| "step": 93 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.9539559483528137, | |
| "learning_rate": 1.2003601080324098e-05, | |
| "loss": 1.2392, | |
| "step": 94 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.958608865737915, | |
| "learning_rate": 1.0003000900270082e-05, | |
| "loss": 1.0437, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.9105627536773682, | |
| "learning_rate": 8.002400720216065e-06, | |
| "loss": 1.0925, | |
| "step": 96 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.4394004344940186, | |
| "learning_rate": 6.001800540162049e-06, | |
| "loss": 1.2812, | |
| "step": 97 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.986015260219574, | |
| "learning_rate": 4.001200360108033e-06, | |
| "loss": 1.1349, | |
| "step": 98 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.9465481042861938, | |
| "learning_rate": 2.0006001800540163e-06, | |
| "loss": 1.1348, | |
| "step": 99 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.0053468942642212, | |
| "learning_rate": 0.0, | |
| "loss": 1.1546, | |
| "step": 100 | |
| } | |
| ], | |
| "logging_steps": 1, | |
| "max_steps": 100, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 500, | |
| "total_flos": 5736360454225920.0, | |
| "train_batch_size": 1, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |