{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.4358161648177496, "global_step": 1100, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0, "learning_rate": 3.846153846153847e-06, "loss": 2.7825, "theoretical_loss": 3.3518567762894107, "tokens_seen": 2969305088 }, { "epoch": 0.0, "learning_rate": 3.846153846153846e-05, "loss": 2.4663, "theoretical_loss": 3.3517550749686795, "tokens_seen": 2970484736 }, { "epoch": 0.01, "learning_rate": 7.692307692307693e-05, "loss": 1.9063, "theoretical_loss": 3.351642134115535, "tokens_seen": 2971795456 }, { "epoch": 0.01, "learning_rate": 9.983987189751803e-05, "loss": 1.4439, "theoretical_loss": 3.351529257004948, "tokens_seen": 2973106176 }, { "epoch": 0.02, "learning_rate": 9.943955164131305e-05, "loss": 1.2731, "theoretical_loss": 3.3514164435728655, "tokens_seen": 2974416896 }, { "epoch": 0.02, "learning_rate": 9.90392313851081e-05, "loss": 1.2079, "theoretical_loss": 3.3513036937553267, "tokens_seen": 2975727616 }, { "epoch": 0.02, "learning_rate": 9.863891112890312e-05, "loss": 1.2085, "theoretical_loss": 3.3511910074884628, "tokens_seen": 2977038336 }, { "epoch": 0.03, "learning_rate": 9.823859087269817e-05, "loss": 1.2031, "theoretical_loss": 3.3510783847084977, "tokens_seen": 2978349056 }, { "epoch": 0.03, "learning_rate": 9.78382706164932e-05, "loss": 1.2176, "theoretical_loss": 3.350965825351748, "tokens_seen": 2979659776 }, { "epoch": 0.04, "learning_rate": 9.743795036028824e-05, "loss": 1.1891, "theoretical_loss": 3.3508533293546208, "tokens_seen": 2980970496 }, { "epoch": 0.04, "learning_rate": 9.703763010408326e-05, "loss": 1.2086, "theoretical_loss": 3.3507408966536154, "tokens_seen": 2982281216 }, { "epoch": 0.04, "learning_rate": 9.663730984787832e-05, "loss": 1.18, "theoretical_loss": 3.3506285271853233, "tokens_seen": 2983591936 }, { "epoch": 0.05, "learning_rate": 9.623698959167334e-05, "loss": 1.1709, "theoretical_loss": 3.350516220886427, "tokens_seen": 2984902656 }, { "epoch": 0.05, "objective/train/docs_used": 1000073, "objective/train/instantaneous_batch_size": 16, "objective/train/instantaneous_microbatch_size": 16384, "objective/train/original_loss": 1.024436354637146, "objective/train/theoretical_loss": 3.3504600914057403, "objective/train/tokens_used": 36842976, "theoretical_loss": 3.3504600914057403, "tokens_seen": 2985558016 }, { "epoch": 0.05, "learning_rate": 9.583666933546838e-05, "loss": 1.1659, "theoretical_loss": 3.3504039776936994, "tokens_seen": 2986213376 }, { "epoch": 0.06, "learning_rate": 9.543634907926342e-05, "loss": 1.1826, "theoretical_loss": 3.350291797544005, "tokens_seen": 2987524096 }, { "epoch": 0.06, "learning_rate": 9.503602882305846e-05, "loss": 1.1854, "theoretical_loss": 3.3501796803742994, "tokens_seen": 2988834816 }, { "epoch": 0.06, "learning_rate": 9.463570856685348e-05, "loss": 1.1901, "theoretical_loss": 3.3500676261216285, "tokens_seen": 2990145536 }, { "epoch": 0.07, "learning_rate": 9.423538831064852e-05, "loss": 1.1854, "theoretical_loss": 3.3499556347231287, "tokens_seen": 2991456256 }, { "epoch": 0.07, "learning_rate": 9.383506805444356e-05, "loss": 1.2042, "theoretical_loss": 3.349843706116027, "tokens_seen": 2992766976 }, { "epoch": 0.08, "learning_rate": 9.34347477982386e-05, "loss": 1.1804, "theoretical_loss": 3.3497318402376397, "tokens_seen": 2994077696 }, { "epoch": 0.08, "learning_rate": 9.303442754203363e-05, "loss": 1.1869, "theoretical_loss": 3.349620037025374, "tokens_seen": 2995388416 }, { "epoch": 0.08, "learning_rate": 9.263410728582867e-05, "loss": 1.1783, "theoretical_loss": 3.349508296416727, "tokens_seen": 2996699136 }, { "epoch": 0.09, "learning_rate": 9.22337870296237e-05, "loss": 1.1659, "theoretical_loss": 3.3493966183492847, "tokens_seen": 2998009856 }, { "epoch": 0.09, "learning_rate": 9.183346677341874e-05, "loss": 1.1763, "theoretical_loss": 3.349285002760723, "tokens_seen": 2999320576 }, { "epoch": 0.1, "learning_rate": 9.143314651721377e-05, "loss": 1.151, "theoretical_loss": 3.3491734495888066, "tokens_seen": 3000631296 }, { "epoch": 0.1, "objective/train/docs_used": 1005438, "objective/train/instantaneous_batch_size": 16, "objective/train/instantaneous_microbatch_size": 16384, "objective/train/original_loss": 1.0497773885726929, "objective/train/theoretical_loss": 3.3490619587713906, "objective/train/tokens_used": 53226976, "theoretical_loss": 3.3490619587713906, "tokens_seen": 3001942016 }, { "epoch": 0.1, "learning_rate": 9.103282626100881e-05, "loss": 1.1792, "theoretical_loss": 3.3490619587713906, "tokens_seen": 3001942016 }, { "epoch": 0.1, "learning_rate": 9.063250600480385e-05, "loss": 1.1536, "theoretical_loss": 3.348950530246418, "tokens_seen": 3003252736 }, { "epoch": 0.11, "learning_rate": 9.023218574859889e-05, "loss": 1.1929, "theoretical_loss": 3.348839163951921, "tokens_seen": 3004563456 }, { "epoch": 0.11, "learning_rate": 8.983186549239391e-05, "loss": 1.1682, "theoretical_loss": 3.3487278598260195, "tokens_seen": 3005874176 }, { "epoch": 0.11, "learning_rate": 8.943154523618895e-05, "loss": 1.148, "theoretical_loss": 3.348616617806923, "tokens_seen": 3007184896 }, { "epoch": 0.12, "learning_rate": 8.903122497998399e-05, "loss": 1.1768, "theoretical_loss": 3.34850543783293, "tokens_seen": 3008495616 }, { "epoch": 0.12, "learning_rate": 8.863090472377903e-05, "loss": 1.1853, "theoretical_loss": 3.3483943198424253, "tokens_seen": 3009806336 }, { "epoch": 0.13, "learning_rate": 8.823058446757406e-05, "loss": 1.1791, "theoretical_loss": 3.348283263773882, "tokens_seen": 3011117056 }, { "epoch": 0.13, "learning_rate": 8.78302642113691e-05, "loss": 1.18, "theoretical_loss": 3.3481722695658633, "tokens_seen": 3012427776 }, { "epoch": 0.13, "learning_rate": 8.742994395516414e-05, "loss": 1.1761, "theoretical_loss": 3.3480613371570165, "tokens_seen": 3013738496 }, { "epoch": 0.14, "learning_rate": 8.702962369895917e-05, "loss": 1.1834, "theoretical_loss": 3.347950466486079, "tokens_seen": 3015049216 }, { "epoch": 0.14, "learning_rate": 8.66293034427542e-05, "loss": 1.1986, "theoretical_loss": 3.3478396574918747, "tokens_seen": 3016359936 }, { "epoch": 0.15, "learning_rate": 8.622898318654924e-05, "loss": 1.1895, "theoretical_loss": 3.347728910113315, "tokens_seen": 3017670656 }, { "epoch": 0.15, "objective/train/docs_used": 1011213, "objective/train/instantaneous_batch_size": 16, "objective/train/instantaneous_microbatch_size": 16384, "objective/train/original_loss": 1.2239086627960205, "objective/train/theoretical_loss": 3.3476735595108353, "objective/train/tokens_used": 69610976, "theoretical_loss": 3.3476735595108353, "tokens_seen": 3018326016 }, { "epoch": 0.15, "learning_rate": 8.582866293034428e-05, "loss": 1.1745, "theoretical_loss": 3.3476182242893975, "tokens_seen": 3018981376 }, { "epoch": 0.15, "learning_rate": 8.542834267413932e-05, "loss": 1.1915, "theoretical_loss": 3.3475075999592074, "tokens_seen": 3020292096 }, { "epoch": 0.16, "learning_rate": 8.502802241793434e-05, "loss": 1.1925, "theoretical_loss": 3.347397037061916, "tokens_seen": 3021602816 }, { "epoch": 0.16, "learning_rate": 8.462770216172938e-05, "loss": 1.1872, "theoretical_loss": 3.3472865355367825, "tokens_seen": 3022913536 }, { "epoch": 0.17, "learning_rate": 8.422738190552442e-05, "loss": 1.1753, "theoretical_loss": 3.3471760953231504, "tokens_seen": 3024224256 }, { "epoch": 0.17, "learning_rate": 8.382706164931946e-05, "loss": 1.2088, "theoretical_loss": 3.3470657163604507, "tokens_seen": 3025534976 }, { "epoch": 0.17, "learning_rate": 8.34267413931145e-05, "loss": 1.1959, "theoretical_loss": 3.3469553985882, "tokens_seen": 3026845696 }, { "epoch": 0.18, "learning_rate": 8.302642113690953e-05, "loss": 1.2173, "theoretical_loss": 3.3468451419460017, "tokens_seen": 3028156416 }, { "epoch": 0.18, "learning_rate": 8.262610088070457e-05, "loss": 1.1876, "theoretical_loss": 3.3467349463735436, "tokens_seen": 3029467136 }, { "epoch": 0.19, "learning_rate": 8.22257806244996e-05, "loss": 1.1839, "theoretical_loss": 3.3466248118105995, "tokens_seen": 3030777856 }, { "epoch": 0.19, "learning_rate": 8.182546036829464e-05, "loss": 1.198, "theoretical_loss": 3.3465147381970293, "tokens_seen": 3032088576 }, { "epoch": 0.19, "learning_rate": 8.142514011208967e-05, "loss": 1.1891, "theoretical_loss": 3.346404725472777, "tokens_seen": 3033399296 }, { "epoch": 0.2, "objective/train/docs_used": 1016482, "objective/train/instantaneous_batch_size": 16, "objective/train/instantaneous_microbatch_size": 16384, "objective/train/original_loss": 1.2322360277175903, "objective/train/theoretical_loss": 3.3462947735778727, "objective/train/tokens_used": 85994976, "theoretical_loss": 3.3462947735778727, "tokens_seen": 3034710016 }, { "epoch": 0.2, "learning_rate": 8.102481985588471e-05, "loss": 1.1985, "theoretical_loss": 3.3462947735778727, "tokens_seen": 3034710016 }, { "epoch": 0.2, "learning_rate": 8.062449959967975e-05, "loss": 1.1688, "theoretical_loss": 3.346184882452431, "tokens_seen": 3036020736 }, { "epoch": 0.21, "learning_rate": 8.022417934347479e-05, "loss": 1.1874, "theoretical_loss": 3.3460750520366513, "tokens_seen": 3037331456 }, { "epoch": 0.21, "learning_rate": 7.982385908726981e-05, "loss": 1.1835, "theoretical_loss": 3.3459652822708175, "tokens_seen": 3038642176 }, { "epoch": 0.21, "learning_rate": 7.942353883106487e-05, "loss": 1.1842, "theoretical_loss": 3.345855573095298, "tokens_seen": 3039952896 }, { "epoch": 0.22, "learning_rate": 7.902321857485989e-05, "loss": 1.1882, "theoretical_loss": 3.3457459244505463, "tokens_seen": 3041263616 }, { "epoch": 0.22, "learning_rate": 7.862289831865493e-05, "loss": 1.2044, "theoretical_loss": 3.3456363362770984, "tokens_seen": 3042574336 }, { "epoch": 0.23, "learning_rate": 7.822257806244996e-05, "loss": 1.168, "theoretical_loss": 3.345526808515576, "tokens_seen": 3043885056 }, { "epoch": 0.23, "learning_rate": 7.782225780624501e-05, "loss": 1.1908, "theoretical_loss": 3.3454173411066828, "tokens_seen": 3045195776 }, { "epoch": 0.23, "learning_rate": 7.742193755004003e-05, "loss": 1.1942, "theoretical_loss": 3.345307933991209, "tokens_seen": 3046506496 }, { "epoch": 0.24, "learning_rate": 7.702161729383507e-05, "loss": 1.1965, "theoretical_loss": 3.3451985871100254, "tokens_seen": 3047817216 }, { "epoch": 0.24, "learning_rate": 7.662129703763011e-05, "loss": 1.1727, "theoretical_loss": 3.345089300404087, "tokens_seen": 3049127936 }, { "epoch": 0.25, "learning_rate": 7.622097678142515e-05, "loss": 1.1993, "theoretical_loss": 3.344980073814434, "tokens_seen": 3050438656 }, { "epoch": 0.25, "objective/train/docs_used": 1022217, "objective/train/instantaneous_batch_size": 16, "objective/train/instantaneous_microbatch_size": 16384, "objective/train/original_loss": 0.9348351955413818, "objective/train/theoretical_loss": 3.3449254830448116, "objective/train/tokens_used": 102378976, "theoretical_loss": 3.3449254830448116, "tokens_seen": 3051094016 }, { "epoch": 0.25, "learning_rate": 7.582065652522018e-05, "loss": 1.1904, "theoretical_loss": 3.3448709072821874, "tokens_seen": 3051749376 }, { "epoch": 0.25, "learning_rate": 7.542033626901522e-05, "loss": 1.1939, "theoretical_loss": 3.344761800748551, "tokens_seen": 3053060096 }, { "epoch": 0.26, "learning_rate": 7.502001601281026e-05, "loss": 1.1901, "theoretical_loss": 3.3446527541548132, "tokens_seen": 3054370816 }, { "epoch": 0.26, "learning_rate": 7.46196957566053e-05, "loss": 1.2002, "theoretical_loss": 3.3445437674423437, "tokens_seen": 3055681536 }, { "epoch": 0.27, "learning_rate": 7.421937550040032e-05, "loss": 1.2222, "theoretical_loss": 3.3444348405525948, "tokens_seen": 3056992256 }, { "epoch": 0.27, "learning_rate": 7.381905524419536e-05, "loss": 1.2025, "theoretical_loss": 3.3443259734271016, "tokens_seen": 3058302976 }, { "epoch": 0.27, "learning_rate": 7.34187349879904e-05, "loss": 1.1916, "theoretical_loss": 3.344217166007481, "tokens_seen": 3059613696 }, { "epoch": 0.28, "learning_rate": 7.301841473178544e-05, "loss": 1.1965, "theoretical_loss": 3.344108418235432, "tokens_seen": 3060924416 }, { "epoch": 0.28, "learning_rate": 7.261809447558046e-05, "loss": 1.1738, "theoretical_loss": 3.343999730052735, "tokens_seen": 3062235136 }, { "epoch": 0.29, "learning_rate": 7.22177742193755e-05, "loss": 1.1847, "theoretical_loss": 3.343891101401253, "tokens_seen": 3063545856 }, { "epoch": 0.29, "learning_rate": 7.181745396317054e-05, "loss": 1.18, "theoretical_loss": 3.3437825322229298, "tokens_seen": 3064856576 }, { "epoch": 0.29, "learning_rate": 7.141713370696558e-05, "loss": 1.1817, "theoretical_loss": 3.3436740224597914, "tokens_seen": 3066167296 }, { "epoch": 0.3, "objective/train/docs_used": 1028068, "objective/train/instantaneous_batch_size": 16, "objective/train/instantaneous_microbatch_size": 16384, "objective/train/original_loss": 1.3122754096984863, "objective/train/theoretical_loss": 3.3435655720539446, "objective/train/tokens_used": 118762976, "theoretical_loss": 3.3435655720539446, "tokens_seen": 3067478016 }, { "epoch": 0.3, "learning_rate": 7.101681345076061e-05, "loss": 1.2002, "theoretical_loss": 3.3435655720539446, "tokens_seen": 3067478016 }, { "epoch": 0.3, "learning_rate": 7.061649319455565e-05, "loss": 1.1951, "theoretical_loss": 3.343457180947577, "tokens_seen": 3068788736 }, { "epoch": 0.31, "learning_rate": 7.021617293835069e-05, "loss": 1.1923, "theoretical_loss": 3.343348849082957, "tokens_seen": 3070099456 }, { "epoch": 0.31, "learning_rate": 6.981585268214572e-05, "loss": 1.1902, "theoretical_loss": 3.3432405764024353, "tokens_seen": 3071410176 }, { "epoch": 0.31, "learning_rate": 6.941553242594075e-05, "loss": 1.1889, "theoretical_loss": 3.3431323628484417, "tokens_seen": 3072720896 }, { "epoch": 0.32, "learning_rate": 6.901521216973579e-05, "loss": 1.1913, "theoretical_loss": 3.343024208363487, "tokens_seen": 3074031616 }, { "epoch": 0.32, "learning_rate": 6.861489191353083e-05, "loss": 1.18, "theoretical_loss": 3.3429161128901628, "tokens_seen": 3075342336 }, { "epoch": 0.32, "learning_rate": 6.821457165732587e-05, "loss": 1.1859, "theoretical_loss": 3.3428080763711403, "tokens_seen": 3076653056 }, { "epoch": 0.33, "learning_rate": 6.78142514011209e-05, "loss": 1.1819, "theoretical_loss": 3.3427000987491717, "tokens_seen": 3077963776 }, { "epoch": 0.33, "learning_rate": 6.741393114491593e-05, "loss": 1.1947, "theoretical_loss": 3.342592179967088, "tokens_seen": 3079274496 }, { "epoch": 0.34, "learning_rate": 6.701361088871097e-05, "loss": 1.196, "theoretical_loss": 3.3424843199678005, "tokens_seen": 3080585216 }, { "epoch": 0.34, "learning_rate": 6.661329063250601e-05, "loss": 1.1844, "theoretical_loss": 3.3423765186943006, "tokens_seen": 3081895936 }, { "epoch": 0.34, "learning_rate": 6.621297037630104e-05, "loss": 1.1972, "theoretical_loss": 3.3422687760896586, "tokens_seen": 3083206656 }, { "epoch": 0.35, "objective/train/docs_used": 1033433, "objective/train/instantaneous_batch_size": 16, "objective/train/instantaneous_microbatch_size": 16384, "objective/train/original_loss": 1.273867130279541, "objective/train/theoretical_loss": 3.3422149267703904, "objective/train/tokens_used": 135146976, "theoretical_loss": 3.3422149267703904, "tokens_seen": 3083862016 }, { "epoch": 0.35, "learning_rate": 6.581265012009608e-05, "loss": 1.1819, "theoretical_loss": 3.342161092097024, "tokens_seen": 3084517376 }, { "epoch": 0.35, "learning_rate": 6.541232986389112e-05, "loss": 1.1736, "theoretical_loss": 3.342053466659626, "tokens_seen": 3085828096 }, { "epoch": 0.36, "learning_rate": 6.501200960768615e-05, "loss": 1.1701, "theoretical_loss": 3.3419458997207734, "tokens_seen": 3087138816 }, { "epoch": 0.36, "learning_rate": 6.461168935148118e-05, "loss": 1.1776, "theoretical_loss": 3.3418383912238525, "tokens_seen": 3088449536 }, { "epoch": 0.36, "learning_rate": 6.421136909527622e-05, "loss": 1.1794, "theoretical_loss": 3.341730941112329, "tokens_seen": 3089760256 }, { "epoch": 0.37, "learning_rate": 6.381104883907126e-05, "loss": 1.1969, "theoretical_loss": 3.341623549329748, "tokens_seen": 3091070976 }, { "epoch": 0.37, "learning_rate": 6.34107285828663e-05, "loss": 1.1738, "theoretical_loss": 3.3415162158197327, "tokens_seen": 3092381696 }, { "epoch": 0.38, "learning_rate": 6.301040832666132e-05, "loss": 1.1815, "theoretical_loss": 3.3414089405259837, "tokens_seen": 3093692416 }, { "epoch": 0.38, "learning_rate": 6.261008807045636e-05, "loss": 1.176, "theoretical_loss": 3.341301723392281, "tokens_seen": 3095003136 }, { "epoch": 0.38, "learning_rate": 6.22097678142514e-05, "loss": 1.1834, "theoretical_loss": 3.3411945643624814, "tokens_seen": 3096313856 }, { "epoch": 0.39, "learning_rate": 6.180944755804644e-05, "loss": 1.1951, "theoretical_loss": 3.3410874633805214, "tokens_seen": 3097624576 }, { "epoch": 0.39, "learning_rate": 6.140912730184148e-05, "loss": 1.1936, "theoretical_loss": 3.3409804203904145, "tokens_seen": 3098935296 }, { "epoch": 0.4, "objective/train/docs_used": 1039074, "objective/train/instantaneous_batch_size": 16, "objective/train/instantaneous_microbatch_size": 16384, "objective/train/original_loss": 1.3948986530303955, "objective/train/theoretical_loss": 3.3408734353362513, "objective/train/tokens_used": 151530976, "theoretical_loss": 3.3408734353362513, "tokens_seen": 3100246016 }, { "epoch": 0.4, "learning_rate": 6.1008807045636506e-05, "loss": 1.1987, "theoretical_loss": 3.3408734353362513, "tokens_seen": 3100246016 }, { "epoch": 0.4, "learning_rate": 6.060848678943155e-05, "loss": 1.1874, "theoretical_loss": 3.3407665081622, "tokens_seen": 3101556736 }, { "epoch": 0.4, "learning_rate": 6.0208166533226584e-05, "loss": 1.184, "theoretical_loss": 3.340659638812507, "tokens_seen": 3102867456 }, { "epoch": 0.41, "learning_rate": 5.9807846277021617e-05, "loss": 1.1919, "theoretical_loss": 3.340552827231495, "tokens_seen": 3104178176 }, { "epoch": 0.41, "learning_rate": 5.940752602081665e-05, "loss": 1.1835, "theoretical_loss": 3.3404460733635646, "tokens_seen": 3105488896 }, { "epoch": 0.42, "learning_rate": 5.9007205764611695e-05, "loss": 1.18, "theoretical_loss": 3.340339377153193, "tokens_seen": 3106799616 }, { "epoch": 0.42, "learning_rate": 5.860688550840673e-05, "loss": 1.1818, "theoretical_loss": 3.340232738544933, "tokens_seen": 3108110336 }, { "epoch": 0.42, "learning_rate": 5.820656525220176e-05, "loss": 1.1953, "theoretical_loss": 3.340126157483417, "tokens_seen": 3109421056 }, { "epoch": 0.43, "learning_rate": 5.7806244995996806e-05, "loss": 1.1923, "theoretical_loss": 3.340019633913351, "tokens_seen": 3110731776 }, { "epoch": 0.43, "learning_rate": 5.740592473979184e-05, "loss": 1.2147, "theoretical_loss": 3.3399131677795184, "tokens_seen": 3112042496 }, { "epoch": 0.44, "learning_rate": 5.700560448358687e-05, "loss": 1.2106, "theoretical_loss": 3.3398067590267804, "tokens_seen": 3113353216 } ], "max_steps": 2524, "num_train_epochs": 9223372036854775807, "total_flos": 7.35798362112e+16, "trial_name": null, "trial_params": null }