diff --git "a/checkpoint-25177/trainer_state.json" "b/checkpoint-25177/trainer_state.json" new file mode 100644--- /dev/null +++ "b/checkpoint-25177/trainer_state.json" @@ -0,0 +1,199526 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 0.5, + "global_step": 25177, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0, + "learning_rate": 0.0, + "loss": 755.8631, + "theoretical_loss": 20.81281629304291, + "tokens_seen": 65536 + }, + { + "epoch": 0.0, + "learning_rate": 0.0, + "loss": 767.2727, + "theoretical_loss": 17.566199595824404, + "tokens_seen": 131072 + }, + { + "epoch": 0.0, + "learning_rate": 0.0, + "loss": 763.0899, + "theoretical_loss": 15.939475584332328, + "tokens_seen": 196608 + }, + { + "epoch": 0.0, + "learning_rate": 9.92063492063492e-07, + "loss": 753.3647, + "theoretical_loss": 14.89231524748433, + "tokens_seen": 262144 + }, + { + "epoch": 0.0, + "learning_rate": 9.92063492063492e-07, + "loss": 755.1718, + "theoretical_loss": 14.136215429258735, + "tokens_seen": 327680 + }, + { + "epoch": 0.0, + "learning_rate": 1.984126984126984e-06, + "loss": 766.758, + "theoretical_loss": 13.552559964045983, + "tokens_seen": 393216 + }, + { + "epoch": 0.0, + "learning_rate": 2.9761904761904763e-06, + "loss": 755.9789, + "theoretical_loss": 13.08180749289695, + "tokens_seen": 458752 + }, + { + "epoch": 0.0, + "learning_rate": 3.968253968253968e-06, + "loss": 737.8781, + "theoretical_loss": 12.690128116979084, + "tokens_seen": 524288 + }, + { + "epoch": 0.0, + "learning_rate": 4.96031746031746e-06, + "loss": 724.6478, + "theoretical_loss": 12.356590955369384, + "tokens_seen": 589824 + }, + { + "epoch": 0.0, + "learning_rate": 5.9523809523809525e-06, + "loss": 708.7361, + "theoretical_loss": 12.067411098530838, + "tokens_seen": 655360 + }, + { + "epoch": 0.0, + "learning_rate": 6.944444444444444e-06, + "loss": 693.5477, + "theoretical_loss": 11.813064722597437, + "tokens_seen": 720896 + }, + { + "epoch": 0.0, + "learning_rate": 7.936507936507936e-06, + "loss": 673.1252, + "theoretical_loss": 11.58671770020249, + "tokens_seen": 786432 + }, + { + "epoch": 0.0, + "learning_rate": 8.928571428571428e-06, + "loss": 647.0286, + "theoretical_loss": 11.383312631682548, + "tokens_seen": 851968 + }, + { + "epoch": 0.0, + "learning_rate": 9.92063492063492e-06, + "loss": 648.8019, + "theoretical_loss": 11.199010193607633, + "tokens_seen": 917504 + }, + { + "epoch": 0.0, + "learning_rate": 1.0912698412698412e-05, + "loss": 616.7948, + "theoretical_loss": 11.030832409473673, + "tokens_seen": 983040 + }, + { + "epoch": 0.0, + "learning_rate": 1.1904761904761905e-05, + "loss": 608.8694, + "theoretical_loss": 10.87642657795271, + "tokens_seen": 1048576 + }, + { + "epoch": 0.0, + "learning_rate": 1.2896825396825396e-05, + "loss": 613.2443, + "theoretical_loss": 10.733904231558485, + "tokens_seen": 1114112 + }, + { + "epoch": 0.0, + "learning_rate": 1.3888888888888888e-05, + "loss": 617.6806, + "theoretical_loss": 10.60172836772604, + "tokens_seen": 1179648 + }, + { + "epoch": 0.0, + "learning_rate": 1.4880952380952381e-05, + "loss": 613.5583, + "theoretical_loss": 10.478632663852402, + "tokens_seen": 1245184 + }, + { + "epoch": 0.0, + "learning_rate": 1.5873015873015872e-05, + "loss": 580.1763, + "theoretical_loss": 10.36356243525909, + "tokens_seen": 1310720 + }, + { + "epoch": 0.0, + "learning_rate": 1.6865079365079364e-05, + "loss": 554.4431, + "theoretical_loss": 10.255630712392506, + "tokens_seen": 1376256 + }, + { + "epoch": 0.0, + "learning_rate": 1.7857142857142855e-05, + "loss": 558.8197, + "theoretical_loss": 10.154085044765779, + "tokens_seen": 1441792 + }, + { + "epoch": 0.0, + "learning_rate": 1.884920634920635e-05, + "loss": 550.8199, + "theoretical_loss": 10.058282053228359, + "tokens_seen": 1507328 + }, + { + "epoch": 0.0, + "learning_rate": 1.984126984126984e-05, + "loss": 544.7081, + "theoretical_loss": 9.967667670336038, + "tokens_seen": 1572864 + }, + { + "epoch": 0.0, + "objective/train/advantage_avg": 0.4763805568218231, + "objective/train/docs_used": 13112, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 8.302227973937988, + "objective/train/original_loss": 8.302227020263672, + "objective/train/theoretical_loss": 9.881761617888868, + "objective/train/tokens_used": 22098400, + "objective/train/value_avg": -0.481201171875, + "objective/train/value_loss": 0.22699925303459167, + "objective/train/value_max": -0.479736328125, + "objective/train/value_min": -0.48291015625, + "objective/train/value_reward_corr": -0.01800956963855145, + "objective/train/value_std": 0.0005345344543457031, + "objective/train/weight_avg": 117.51708984375, + "objective/train/weighted_lm_loss": 966.7343139648438, + "objective/train/weights_max": 125.09851837158203, + "objective/train/weights_min": 29.544034957885742, + "theoretical_loss": 9.881761617888868, + "tokens_seen": 1638400 + }, + { + "epoch": 0.0, + "learning_rate": 2.0833333333333333e-05, + "loss": 516.3743, + "theoretical_loss": 9.881761617888868, + "tokens_seen": 1638400 + }, + { + "epoch": 0.0, + "learning_rate": 2.1825396825396824e-05, + "loss": 479.2585, + "theoretical_loss": 9.80014508303632, + "tokens_seen": 1703936 + }, + { + "epoch": 0.0, + "learning_rate": 2.2817460317460315e-05, + "loss": 480.6397, + "theoretical_loss": 9.722450838403205, + "tokens_seen": 1769472 + }, + { + "epoch": 0.0, + "learning_rate": 2.380952380952381e-05, + "loss": 471.6678, + "theoretical_loss": 9.648355250577305, + "tokens_seen": 1835008 + }, + { + "epoch": 0.0, + "learning_rate": 2.48015873015873e-05, + "loss": 470.6309, + "theoretical_loss": 9.5775717626414, + "tokens_seen": 1900544 + }, + { + "epoch": 0.0, + "learning_rate": 2.5793650793650793e-05, + "loss": 391.9027, + "theoretical_loss": 9.509845538260612, + "tokens_seen": 1966080 + }, + { + "epoch": 0.0, + "learning_rate": 2.6785714285714284e-05, + "loss": 415.0912, + "theoretical_loss": 9.444949029127695, + "tokens_seen": 2031616 + }, + { + "epoch": 0.0, + "learning_rate": 2.7777777777777776e-05, + "loss": 407.0198, + "theoretical_loss": 9.382678282406216, + "tokens_seen": 2097152 + }, + { + "epoch": 0.0, + "learning_rate": 2.876984126984127e-05, + "loss": 382.5104, + "theoretical_loss": 9.32284984572974, + "tokens_seen": 2162688 + }, + { + "epoch": 0.0, + "learning_rate": 2.9761904761904762e-05, + "loss": 378.358, + "theoretical_loss": 9.265298158156035, + "tokens_seen": 2228224 + }, + { + "epoch": 0.0, + "learning_rate": 3.075396825396825e-05, + "loss": 331.4446, + "theoretical_loss": 9.209873338940517, + "tokens_seen": 2293760 + }, + { + "epoch": 0.0, + "learning_rate": 3.1746031746031745e-05, + "loss": 313.841, + "theoretical_loss": 9.156439304004051, + "tokens_seen": 2359296 + }, + { + "epoch": 0.0, + "learning_rate": 3.273809523809524e-05, + "loss": 293.332, + "theoretical_loss": 9.10487215390911, + "tokens_seen": 2424832 + }, + { + "epoch": 0.0, + "learning_rate": 3.373015873015873e-05, + "loss": 272.0188, + "theoretical_loss": 9.055058788029493, + "tokens_seen": 2490368 + }, + { + "epoch": 0.0, + "learning_rate": 3.472222222222222e-05, + "loss": 260.2195, + "theoretical_loss": 9.006895708139588, + "tokens_seen": 2555904 + }, + { + "epoch": 0.0, + "learning_rate": 3.571428571428571e-05, + "loss": 237.5644, + "theoretical_loss": 8.960287981405116, + "tokens_seen": 2621440 + }, + { + "epoch": 0.0, + "learning_rate": 3.670634920634921e-05, + "loss": 234.5823, + "theoretical_loss": 8.91514833813637, + "tokens_seen": 2686976 + }, + { + "epoch": 0.0, + "learning_rate": 3.76984126984127e-05, + "loss": 224.3307, + "theoretical_loss": 8.871396383973984, + "tokens_seen": 2752512 + }, + { + "epoch": 0.0, + "learning_rate": 3.8690476190476195e-05, + "loss": 205.798, + "theoretical_loss": 8.828957909649258, + "tokens_seen": 2818048 + }, + { + "epoch": 0.0, + "learning_rate": 3.968253968253968e-05, + "loss": 195.3919, + "theoretical_loss": 8.787764284274173, + "tokens_seen": 2883584 + }, + { + "epoch": 0.0, + "learning_rate": 4.067460317460318e-05, + "loss": 169.7055, + "theoretical_loss": 8.747751920407216, + "tokens_seen": 2949120 + }, + { + "epoch": 0.0, + "learning_rate": 4.1666666666666665e-05, + "loss": 158.7231, + "theoretical_loss": 8.708861801016592, + "tokens_seen": 3014656 + }, + { + "epoch": 0.0, + "learning_rate": 4.265873015873016e-05, + "loss": 136.9043, + "theoretical_loss": 8.671039060004606, + "tokens_seen": 3080192 + }, + { + "epoch": 0.0, + "learning_rate": 4.365079365079365e-05, + "loss": 124.3726, + "theoretical_loss": 8.634232609231233, + "tokens_seen": 3145728 + }, + { + "epoch": 0.0, + "learning_rate": 4.464285714285714e-05, + "loss": 119.2385, + "theoretical_loss": 8.598394806032085, + "tokens_seen": 3211264 + }, + { + "epoch": 0.0, + "objective/train/advantage_avg": 0.3406403660774231, + "objective/train/docs_used": 13755, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 7.455689907073975, + "objective/train/original_loss": 7.455689907073975, + "objective/train/theoretical_loss": 8.563481156106828, + "objective/train/tokens_used": 23736800, + "objective/train/value_avg": -0.34619140625, + "objective/train/value_loss": 0.11673914641141891, + "objective/train/value_max": -0.3212890625, + "objective/train/value_min": -0.4091796875, + "objective/train/value_reward_corr": -0.10507654884505241, + "objective/train/value_std": 0.0241241455078125, + "objective/train/weight_avg": 31.315393447875977, + "objective/train/weighted_lm_loss": 206.46487426757812, + "objective/train/weights_max": 57.836334228515625, + "objective/train/weights_min": 6.207902908325195, + "theoretical_loss": 8.563481156106828, + "tokens_seen": 3276800 + }, + { + "epoch": 0.0, + "learning_rate": 4.563492063492063e-05, + "loss": 112.5491, + "theoretical_loss": 8.563481156106828, + "tokens_seen": 3276800 + }, + { + "epoch": 0.0, + "learning_rate": 4.6626984126984126e-05, + "loss": 95.8402, + "theoretical_loss": 8.529450047390876, + "tokens_seen": 3342336 + }, + { + "epoch": 0.0, + "learning_rate": 4.761904761904762e-05, + "loss": 83.9444, + "theoretical_loss": 8.496262511141762, + "tokens_seen": 3407872 + }, + { + "epoch": 0.0, + "learning_rate": 4.8611111111111115e-05, + "loss": 81.7756, + "theoretical_loss": 8.463882006992947, + "tokens_seen": 3473408 + }, + { + "epoch": 0.0, + "learning_rate": 4.96031746031746e-05, + "loss": 73.9694, + "theoretical_loss": 8.43227422916854, + "tokens_seen": 3538944 + }, + { + "epoch": 0.0, + "learning_rate": 5.05952380952381e-05, + "loss": 65.3385, + "theoretical_loss": 8.401406931426477, + "tokens_seen": 3604480 + }, + { + "epoch": 0.0, + "learning_rate": 5.1587301587301586e-05, + "loss": 58.0164, + "theoretical_loss": 8.37124976861597, + "tokens_seen": 3670016 + }, + { + "epoch": 0.0, + "learning_rate": 5.257936507936508e-05, + "loss": 51.143, + "theoretical_loss": 8.341774153006837, + "tokens_seen": 3735552 + }, + { + "epoch": 0.0, + "learning_rate": 5.357142857142857e-05, + "loss": 50.4984, + "theoretical_loss": 8.312953123781089, + "tokens_seen": 3801088 + }, + { + "epoch": 0.0, + "learning_rate": 5.4563492063492063e-05, + "loss": 45.1839, + "theoretical_loss": 8.284761228276942, + "tokens_seen": 3866624 + }, + { + "epoch": 0.0, + "learning_rate": 5.555555555555555e-05, + "loss": 38.8596, + "theoretical_loss": 8.257174413747624, + "tokens_seen": 3932160 + }, + { + "epoch": 0.0, + "learning_rate": 5.6547619047619046e-05, + "loss": 34.8879, + "theoretical_loss": 8.230169928545873, + "tokens_seen": 3997696 + }, + { + "epoch": 0.0, + "learning_rate": 5.753968253968254e-05, + "loss": 33.038, + "theoretical_loss": 8.20372623177373, + "tokens_seen": 4063232 + }, + { + "epoch": 0.0, + "learning_rate": 5.8531746031746036e-05, + "loss": 26.7032, + "theoretical_loss": 8.177822910548805, + "tokens_seen": 4128768 + }, + { + "epoch": 0.0, + "learning_rate": 5.9523809523809524e-05, + "loss": 24.4537, + "theoretical_loss": 8.152440604135377, + "tokens_seen": 4194304 + }, + { + "epoch": 0.0, + "learning_rate": 6.051587301587302e-05, + "loss": 21.8622, + "theoretical_loss": 8.127560934273259, + "tokens_seen": 4259840 + }, + { + "epoch": 0.0, + "learning_rate": 6.15079365079365e-05, + "loss": 20.5318, + "theoretical_loss": 8.103166441111469, + "tokens_seen": 4325376 + }, + { + "epoch": 0.0, + "learning_rate": 6.25e-05, + "loss": 20.076, + "theoretical_loss": 8.0792405242184, + "tokens_seen": 4390912 + }, + { + "epoch": 0.0, + "learning_rate": 6.349206349206349e-05, + "loss": 16.7188, + "theoretical_loss": 8.055767388197175, + "tokens_seen": 4456448 + }, + { + "epoch": 0.0, + "learning_rate": 6.448412698412699e-05, + "loss": 15.1535, + "theoretical_loss": 8.032731992484768, + "tokens_seen": 4521984 + }, + { + "epoch": 0.0, + "learning_rate": 6.547619047619048e-05, + "loss": 14.1059, + "theoretical_loss": 8.010120004957596, + "tokens_seen": 4587520 + }, + { + "epoch": 0.0, + "learning_rate": 6.646825396825397e-05, + "loss": 12.3631, + "theoretical_loss": 7.987917759005139, + "tokens_seen": 4653056 + }, + { + "epoch": 0.0, + "learning_rate": 6.746031746031745e-05, + "loss": 11.9625, + "theoretical_loss": 7.966112213767561, + "tokens_seen": 4718592 + }, + { + "epoch": 0.0, + "learning_rate": 6.845238095238096e-05, + "loss": 10.9243, + "theoretical_loss": 7.944690917263747, + "tokens_seen": 4784128 + }, + { + "epoch": 0.0, + "learning_rate": 6.944444444444444e-05, + "loss": 10.6636, + "theoretical_loss": 7.923641972163278, + "tokens_seen": 4849664 + }, + { + "epoch": 0.0, + "objective/train/advantage_avg": 0.11417747288942337, + "objective/train/docs_used": 15099, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 6.844882965087891, + "objective/train/original_loss": 6.844882965087891, + "objective/train/theoretical_loss": 7.902954003979827, + "objective/train/tokens_used": 25375200, + "objective/train/value_avg": -0.11865234375, + "objective/train/value_loss": 0.015311661176383495, + "objective/train/value_max": -0.08404541015625, + "objective/train/value_min": -0.403076171875, + "objective/train/value_reward_corr": -2.201807067110824e-05, + "objective/train/value_std": 0.0472412109375, + "objective/train/weight_avg": 3.695091485977173, + "objective/train/weighted_lm_loss": 22.475915908813477, + "objective/train/weights_max": 52.099212646484375, + "objective/train/weights_min": 0.856390118598938, + "theoretical_loss": 7.902954003979827, + "tokens_seen": 4915200 + }, + { + "epoch": 0.0, + "learning_rate": 7.043650793650793e-05, + "loss": 9.6012, + "theoretical_loss": 7.902954003979827, + "tokens_seen": 4915200 + }, + { + "epoch": 0.0, + "learning_rate": 7.142857142857142e-05, + "loss": 8.3989, + "theoretical_loss": 7.882616131484962, + "tokens_seen": 4980736 + }, + { + "epoch": 0.0, + "learning_rate": 7.242063492063492e-05, + "loss": 8.0256, + "theoretical_loss": 7.862617939160387, + "tokens_seen": 5046272 + }, + { + "epoch": 0.0, + "learning_rate": 7.341269841269842e-05, + "loss": 7.6668, + "theoretical_loss": 7.8429494515236975, + "tokens_seen": 5111808 + }, + { + "epoch": 0.0, + "learning_rate": 7.440476190476191e-05, + "loss": 7.1967, + "theoretical_loss": 7.823601109178073, + "tokens_seen": 5177344 + }, + { + "epoch": 0.0, + "learning_rate": 7.53968253968254e-05, + "loss": 6.5698, + "theoretical_loss": 7.804563746449924, + "tokens_seen": 5242880 + }, + { + "epoch": 0.0, + "learning_rate": 7.63888888888889e-05, + "loss": 6.6817, + "theoretical_loss": 7.785828570490832, + "tokens_seen": 5308416 + }, + { + "epoch": 0.0, + "learning_rate": 7.738095238095239e-05, + "loss": 6.3491, + "theoretical_loss": 7.767387141731124, + "tokens_seen": 5373952 + }, + { + "epoch": 0.0, + "learning_rate": 7.837301587301588e-05, + "loss": 5.9305, + "theoretical_loss": 7.7492313555823795, + "tokens_seen": 5439488 + }, + { + "epoch": 0.0, + "learning_rate": 7.936507936507937e-05, + "loss": 5.8608, + "theoretical_loss": 7.731353425295078, + "tokens_seen": 5505024 + }, + { + "epoch": 0.0, + "learning_rate": 8.035714285714287e-05, + "loss": 5.6862, + "theoretical_loss": 7.713745865885679, + "tokens_seen": 5570560 + }, + { + "epoch": 0.0, + "learning_rate": 8.134920634920635e-05, + "loss": 5.6001, + "theoretical_loss": 7.696401479054694, + "tokens_seen": 5636096 + }, + { + "epoch": 0.0, + "learning_rate": 8.234126984126984e-05, + "loss": 5.3625, + "theoretical_loss": 7.67931333902394, + "tokens_seen": 5701632 + }, + { + "epoch": 0.0, + "learning_rate": 8.333333333333333e-05, + "loss": 5.1213, + "theoretical_loss": 7.662474779227088, + "tokens_seen": 5767168 + }, + { + "epoch": 0.0, + "learning_rate": 8.432539682539683e-05, + "loss": 5.0387, + "theoretical_loss": 7.645879379793039, + "tokens_seen": 5832704 + }, + { + "epoch": 0.0, + "learning_rate": 8.531746031746032e-05, + "loss": 4.7109, + "theoretical_loss": 7.629520955766621, + "tokens_seen": 5898240 + }, + { + "epoch": 0.0, + "learning_rate": 8.630952380952381e-05, + "loss": 4.3837, + "theoretical_loss": 7.613393546015455, + "tokens_seen": 5963776 + }, + { + "epoch": 0.0, + "learning_rate": 8.73015873015873e-05, + "loss": 4.6911, + "theoretical_loss": 7.59749140277604, + "tokens_seen": 6029312 + }, + { + "epoch": 0.0, + "learning_rate": 8.82936507936508e-05, + "loss": 4.8691, + "theoretical_loss": 7.581808981795648, + "tokens_seen": 6094848 + }, + { + "epoch": 0.0, + "learning_rate": 8.928571428571429e-05, + "loss": 4.593, + "theoretical_loss": 7.56634093303013, + "tokens_seen": 6160384 + }, + { + "epoch": 0.0, + "learning_rate": 9.027777777777777e-05, + "loss": 4.4468, + "theoretical_loss": 7.5510820918607084, + "tokens_seen": 6225920 + }, + { + "epoch": 0.0, + "learning_rate": 9.126984126984126e-05, + "loss": 4.5837, + "theoretical_loss": 7.536027470795679, + "tokens_seen": 6291456 + }, + { + "epoch": 0.0, + "learning_rate": 9.226190476190476e-05, + "loss": 4.6296, + "theoretical_loss": 7.521172251625522, + "tokens_seen": 6356992 + }, + { + "epoch": 0.0, + "learning_rate": 9.325396825396825e-05, + "loss": 4.1704, + "theoretical_loss": 7.506511778002256, + "tokens_seen": 6422528 + }, + { + "epoch": 0.0, + "learning_rate": 9.424603174603175e-05, + "loss": 4.1529, + "theoretical_loss": 7.492041548416008, + "tokens_seen": 6488064 + }, + { + "epoch": 0.0, + "objective/train/advantage_avg": 0.0269852913916111, + "objective/train/docs_used": 15777, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 7.0349931716918945, + "objective/train/original_loss": 7.0349931716918945, + "objective/train/theoretical_loss": 7.477757209543791, + "objective/train/tokens_used": 27013600, + "objective/train/value_avg": -0.032440185546875, + "objective/train/value_loss": 0.001031366060487926, + "objective/train/value_max": -0.01934814453125, + "objective/train/value_min": -0.2216796875, + "objective/train/value_reward_corr": -0.16112656527733699, + "objective/train/value_std": 0.0131378173828125, + "objective/train/weight_avg": 1.3339617252349854, + "objective/train/weighted_lm_loss": 9.211831092834473, + "objective/train/weights_max": 9.177886009216309, + "objective/train/weights_min": 0.4551447033882141, + "theoretical_loss": 7.477757209543791, + "tokens_seen": 6553600 + }, + { + "epoch": 0.0, + "learning_rate": 9.523809523809524e-05, + "loss": 4.2583, + "theoretical_loss": 7.477757209543791, + "tokens_seen": 6553600 + }, + { + "epoch": 0.0, + "learning_rate": 9.623015873015874e-05, + "loss": 4.2566, + "theoretical_loss": 7.463654549947222, + "tokens_seen": 6619136 + }, + { + "epoch": 0.0, + "learning_rate": 9.722222222222223e-05, + "loss": 4.2653, + "theoretical_loss": 7.449729494097676, + "tokens_seen": 6684672 + }, + { + "epoch": 0.0, + "learning_rate": 9.821428571428572e-05, + "loss": 4.3759, + "theoretical_loss": 7.435978096708778, + "tokens_seen": 6750208 + }, + { + "epoch": 0.0, + "learning_rate": 9.92063492063492e-05, + "loss": 4.3069, + "theoretical_loss": 7.422396537357665, + "tokens_seen": 6815744 + }, + { + "epoch": 0.0, + "learning_rate": 0.00010019841269841271, + "loss": 4.2534, + "theoretical_loss": 7.408981115377635, + "tokens_seen": 6881280 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001011904761904762, + "loss": 4.3419, + "theoretical_loss": 7.395728245006104, + "tokens_seen": 6946816 + }, + { + "epoch": 0.0, + "learning_rate": 0.00010218253968253968, + "loss": 4.2462, + "theoretical_loss": 7.382634450772793, + "tokens_seen": 7012352 + }, + { + "epoch": 0.0, + "learning_rate": 0.00010317460317460317, + "loss": 4.3021, + "theoretical_loss": 7.369696363114134, + "tokens_seen": 7077888 + }, + { + "epoch": 0.0, + "learning_rate": 0.00010416666666666667, + "loss": 4.178, + "theoretical_loss": 7.356910714200849, + "tokens_seen": 7143424 + }, + { + "epoch": 0.0, + "learning_rate": 0.00010515873015873016, + "loss": 4.0157, + "theoretical_loss": 7.344274333966448, + "tokens_seen": 7208960 + }, + { + "epoch": 0.0, + "learning_rate": 0.00010615079365079365, + "loss": 4.2783, + "theoretical_loss": 7.331784146325278, + "tokens_seen": 7274496 + }, + { + "epoch": 0.0, + "learning_rate": 0.00010714285714285714, + "loss": 3.9572, + "theoretical_loss": 7.319437165569436, + "tokens_seen": 7340032 + }, + { + "epoch": 0.0, + "learning_rate": 0.00010813492063492064, + "loss": 4.101, + "theoretical_loss": 7.307230492934584, + "tokens_seen": 7405568 + }, + { + "epoch": 0.0, + "learning_rate": 0.00010912698412698413, + "loss": 4.312, + "theoretical_loss": 7.295161313325323, + "tokens_seen": 7471104 + }, + { + "epoch": 0.0, + "learning_rate": 0.00011011904761904761, + "loss": 4.2068, + "theoretical_loss": 7.283226892191411, + "tokens_seen": 7536640 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001111111111111111, + "loss": 3.956, + "theoretical_loss": 7.271424572546591, + "tokens_seen": 7602176 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001121031746031746, + "loss": 4.1139, + "theoretical_loss": 7.259751772122383, + "tokens_seen": 7667712 + }, + { + "epoch": 0.0, + "learning_rate": 0.00011309523809523809, + "loss": 4.1602, + "theoretical_loss": 7.2482059806496295, + "tokens_seen": 7733248 + }, + { + "epoch": 0.0, + "learning_rate": 0.00011408730158730158, + "loss": 3.8485, + "theoretical_loss": 7.236784757261022, + "tokens_seen": 7798784 + }, + { + "epoch": 0.0, + "learning_rate": 0.00011507936507936508, + "loss": 4.0966, + "theoretical_loss": 7.225485728008257, + "tokens_seen": 7864320 + }, + { + "epoch": 0.0, + "learning_rate": 0.00011607142857142858, + "loss": 4.1708, + "theoretical_loss": 7.214306583487879, + "tokens_seen": 7929856 + }, + { + "epoch": 0.0, + "learning_rate": 0.00011706349206349207, + "loss": 3.8745, + "theoretical_loss": 7.203245076570161, + "tokens_seen": 7995392 + }, + { + "epoch": 0.0, + "learning_rate": 0.00011805555555555556, + "loss": 4.1223, + "theoretical_loss": 7.192299020225775, + "tokens_seen": 8060928 + }, + { + "epoch": 0.0, + "learning_rate": 0.00011904761904761905, + "loss": 4.1086, + "theoretical_loss": 7.181466285445275, + "tokens_seen": 8126464 + }, + { + "epoch": 0.0, + "objective/train/advantage_avg": 0.017898762598633766, + "objective/train/docs_used": 17166, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 6.344177722930908, + "objective/train/original_loss": 6.344177722930908, + "objective/train/theoretical_loss": 7.170744799246725, + "objective/train/tokens_used": 28652000, + "objective/train/value_avg": -0.0254974365234375, + "objective/train/value_loss": 0.0005556456744670868, + "objective/train/value_max": -0.0099334716796875, + "objective/train/value_min": -0.1573486328125, + "objective/train/value_reward_corr": -0.03335615364555736, + "objective/train/value_std": 0.010711669921875, + "objective/train/weight_avg": 1.210811972618103, + "objective/train/weighted_lm_loss": 7.6226630210876465, + "objective/train/weights_max": 4.823434829711914, + "objective/train/weights_min": 0.5281758308410645, + "theoretical_loss": 7.170744799246725, + "tokens_seen": 8192000 + }, + { + "epoch": 0.0, + "learning_rate": 0.00012003968253968255, + "loss": 3.9962, + "theoretical_loss": 7.170744799246725, + "tokens_seen": 8192000 + }, + { + "epoch": 0.0, + "learning_rate": 0.00012103174603174604, + "loss": 4.0706, + "theoretical_loss": 7.160132542767032, + "tokens_seen": 8257536 + }, + { + "epoch": 0.0, + "learning_rate": 0.00012202380952380953, + "loss": 3.943, + "theoretical_loss": 7.149627549432898, + "tokens_seen": 8323072 + }, + { + "epoch": 0.0, + "learning_rate": 0.000123015873015873, + "loss": 3.9462, + "theoretical_loss": 7.139227903207399, + "tokens_seen": 8388608 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001240079365079365, + "loss": 3.9087, + "theoretical_loss": 7.128931736908553, + "tokens_seen": 8454144 + }, + { + "epoch": 0.0, + "learning_rate": 0.000125, + "loss": 3.7923, + "theoretical_loss": 7.118737230596375, + "tokens_seen": 8519680 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001259920634920635, + "loss": 4.052, + "theoretical_loss": 7.108642610025104, + "tokens_seen": 8585216 + }, + { + "epoch": 0.0, + "learning_rate": 0.00012698412698412698, + "loss": 4.0549, + "theoretical_loss": 7.098646145157529, + "tokens_seen": 8650752 + }, + { + "epoch": 0.0, + "learning_rate": 0.00012797619047619048, + "loss": 4.0965, + "theoretical_loss": 7.088746148738453, + "tokens_seen": 8716288 + }, + { + "epoch": 0.0, + "learning_rate": 0.00012896825396825398, + "loss": 4.0041, + "theoretical_loss": 7.0789409749245085, + "tokens_seen": 8781824 + }, + { + "epoch": 0.0, + "learning_rate": 0.00012996031746031748, + "loss": 3.9969, + "theoretical_loss": 7.069229017967725, + "tokens_seen": 8847360 + }, + { + "epoch": 0.0, + "learning_rate": 0.00013095238095238096, + "loss": 3.9119, + "theoretical_loss": 7.059608710950327, + "tokens_seen": 8912896 + }, + { + "epoch": 0.0, + "learning_rate": 0.00013194444444444446, + "loss": 3.782, + "theoretical_loss": 7.050078524568429, + "tokens_seen": 8978432 + }, + { + "epoch": 0.0, + "learning_rate": 0.00013293650793650793, + "loss": 3.7713, + "theoretical_loss": 7.040636965962385, + "tokens_seen": 9043968 + }, + { + "epoch": 0.0, + "learning_rate": 0.00013392857142857144, + "loss": 3.8475, + "theoretical_loss": 7.031282577591693, + "tokens_seen": 9109504 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001349206349206349, + "loss": 3.8014, + "theoretical_loss": 7.0220139361524385, + "tokens_seen": 9175040 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001359126984126984, + "loss": 4.0792, + "theoretical_loss": 7.01282965153537, + "tokens_seen": 9240576 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001369047619047619, + "loss": 4.0961, + "theoretical_loss": 7.00372836582283, + "tokens_seen": 9306112 + }, + { + "epoch": 0.0, + "learning_rate": 0.00013789682539682541, + "loss": 4.0131, + "theoretical_loss": 6.994708752322817, + "tokens_seen": 9371648 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001388888888888889, + "loss": 3.9374, + "theoretical_loss": 6.985769514638539, + "tokens_seen": 9437184 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001398809523809524, + "loss": 4.003, + "theoretical_loss": 6.976909385771949, + "tokens_seen": 9502720 + }, + { + "epoch": 0.0, + "learning_rate": 0.00014087301587301586, + "loss": 4.019, + "theoretical_loss": 6.968127127259775, + "tokens_seen": 9568256 + }, + { + "epoch": 0.0, + "learning_rate": 0.00014186507936507937, + "loss": 3.9256, + "theoretical_loss": 6.959421528340654, + "tokens_seen": 9633792 + }, + { + "epoch": 0.0, + "learning_rate": 0.00014285714285714284, + "loss": 3.9328, + "theoretical_loss": 6.950791405152069, + "tokens_seen": 9699328 + }, + { + "epoch": 0.0, + "learning_rate": 0.00014384920634920634, + "loss": 4.0271, + "theoretical_loss": 6.942235599955788, + "tokens_seen": 9764864 + }, + { + "epoch": 0.0, + "objective/train/advantage_avg": 0.024732397869229317, + "objective/train/docs_used": 17861, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.353109836578369, + "objective/train/original_loss": 5.353109359741211, + "objective/train/theoretical_loss": 6.9337529803906595, + "objective/train/tokens_used": 30290400, + "objective/train/value_avg": -0.0302734375, + "objective/train/value_loss": 0.0009083261829800904, + "objective/train/value_max": -0.0082855224609375, + "objective/train/value_min": -0.146728515625, + "objective/train/value_reward_corr": -0.031235661265766747, + "objective/train/value_std": 0.0156707763671875, + "objective/train/weight_avg": 1.3012911081314087, + "objective/train/weighted_lm_loss": 6.98408842086792, + "objective/train/weights_max": 3.974698305130005, + "objective/train/weights_min": 0.6351094245910645, + "theoretical_loss": 6.9337529803906595, + "tokens_seen": 9830400 + }, + { + "epoch": 0.0, + "learning_rate": 0.00014484126984126984, + "loss": 3.6403, + "theoretical_loss": 6.9337529803906595, + "tokens_seen": 9830400 + }, + { + "epoch": 0.0, + "learning_rate": 0.00014583333333333335, + "loss": 3.8147, + "theoretical_loss": 6.925342438751576, + "tokens_seen": 9895936 + }, + { + "epoch": 0.0, + "learning_rate": 0.00014682539682539685, + "loss": 3.9681, + "theoretical_loss": 6.917002891293558, + "tokens_seen": 9961472 + }, + { + "epoch": 0.0, + "learning_rate": 0.00014781746031746032, + "loss": 3.9242, + "theoretical_loss": 6.908733277559907, + "tokens_seen": 10027008 + }, + { + "epoch": 0.0, + "learning_rate": 0.00014880952380952382, + "loss": 3.7781, + "theoretical_loss": 6.900532559733447, + "tokens_seen": 10092544 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001498015873015873, + "loss": 3.6177, + "theoretical_loss": 6.89239972200992, + "tokens_seen": 10158080 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001507936507936508, + "loss": 3.6489, + "theoretical_loss": 6.884333769992631, + "tokens_seen": 10223616 + }, + { + "epoch": 0.0, + "learning_rate": 0.00015178571428571427, + "loss": 3.8545, + "theoretical_loss": 6.876333730107509, + "tokens_seen": 10289152 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001527777777777778, + "loss": 3.9385, + "theoretical_loss": 6.868398649037756, + "tokens_seen": 10354688 + }, + { + "epoch": 0.0, + "learning_rate": 0.00015376984126984128, + "loss": 3.7811, + "theoretical_loss": 6.860527593177311, + "tokens_seen": 10420224 + }, + { + "epoch": 0.0, + "learning_rate": 0.00015476190476190478, + "loss": 3.6559, + "theoretical_loss": 6.85271964810239, + "tokens_seen": 10485760 + }, + { + "epoch": 0.0, + "learning_rate": 0.00015575396825396825, + "loss": 3.8556, + "theoretical_loss": 6.844973918060402, + "tokens_seen": 10551296 + }, + { + "epoch": 0.0, + "learning_rate": 0.00015674603174603175, + "loss": 3.7076, + "theoretical_loss": 6.837289525475555, + "tokens_seen": 10616832 + }, + { + "epoch": 0.0, + "learning_rate": 0.00015773809523809523, + "loss": 3.5885, + "theoretical_loss": 6.829665610470508, + "tokens_seen": 10682368 + }, + { + "epoch": 0.0, + "learning_rate": 0.00015873015873015873, + "loss": 3.8269, + "theoretical_loss": 6.822101330403479, + "tokens_seen": 10747904 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001597222222222222, + "loss": 3.8527, + "theoretical_loss": 6.814595859420155, + "tokens_seen": 10813440 + }, + { + "epoch": 0.0, + "learning_rate": 0.00016071428571428573, + "loss": 3.6036, + "theoretical_loss": 6.807148388019941, + "tokens_seen": 10878976 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001617063492063492, + "loss": 3.6202, + "theoretical_loss": 6.799758122635906, + "tokens_seen": 10944512 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001626984126984127, + "loss": 3.7236, + "theoretical_loss": 6.792424285227984, + "tokens_seen": 11010048 + }, + { + "epoch": 0.0, + "learning_rate": 0.00016369047619047618, + "loss": 3.5578, + "theoretical_loss": 6.785146112888908, + "tokens_seen": 11075584 + }, + { + "epoch": 0.0, + "learning_rate": 0.00016468253968253969, + "loss": 3.3895, + "theoretical_loss": 6.777922857462398, + "tokens_seen": 11141120 + }, + { + "epoch": 0.0, + "learning_rate": 0.00016567460317460316, + "loss": 3.5414, + "theoretical_loss": 6.7707537851731825, + "tokens_seen": 11206656 + }, + { + "epoch": 0.0, + "learning_rate": 0.00016666666666666666, + "loss": 3.5707, + "theoretical_loss": 6.763638176268385, + "tokens_seen": 11272192 + }, + { + "epoch": 0.0, + "learning_rate": 0.00016765873015873016, + "loss": 3.6671, + "theoretical_loss": 6.756575324669884, + "tokens_seen": 11337728 + }, + { + "epoch": 0.0, + "learning_rate": 0.00016865079365079366, + "loss": 3.8127, + "theoretical_loss": 6.749564537637246, + "tokens_seen": 11403264 + }, + { + "epoch": 0.0, + "objective/train/advantage_avg": 0.01785166561603546, + "objective/train/docs_used": 18493, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 6.146810054779053, + "objective/train/original_loss": 6.146810531616211, + "objective/train/theoretical_loss": 6.742605135440851, + "objective/train/tokens_used": 31928800, + "objective/train/value_avg": -0.024688720703125, + "objective/train/value_loss": 0.0005992697551846504, + "objective/train/value_max": -0.00653839111328125, + "objective/train/value_min": -0.135009765625, + "objective/train/value_reward_corr": 0.03187241226798395, + "objective/train/value_std": 0.01470184326171875, + "objective/train/weight_avg": 1.2136759757995605, + "objective/train/weighted_lm_loss": 7.5370259284973145, + "objective/train/weights_max": 3.8109962940216064, + "objective/train/weights_min": 0.5201064348220825, + "theoretical_loss": 6.742605135440851, + "tokens_seen": 11468800 + }, + { + "epoch": 0.0, + "learning_rate": 0.00016964285714285717, + "loss": 3.698, + "theoretical_loss": 6.742605135440851, + "tokens_seen": 11468800 + }, + { + "epoch": 0.0, + "learning_rate": 0.00017063492063492064, + "loss": 3.898, + "theoretical_loss": 6.735696451044834, + "tokens_seen": 11534336 + }, + { + "epoch": 0.0, + "learning_rate": 0.00017162698412698414, + "loss": 3.548, + "theoretical_loss": 6.72883782979952, + "tokens_seen": 11599872 + }, + { + "epoch": 0.0, + "learning_rate": 0.00017261904761904762, + "loss": 3.7642, + "theoretical_loss": 6.7220286291429865, + "tokens_seen": 11665408 + }, + { + "epoch": 0.0, + "learning_rate": 0.00017361111111111112, + "loss": 3.3953, + "theoretical_loss": 6.715268218311449, + "tokens_seen": 11730944 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001746031746031746, + "loss": 3.808, + "theoretical_loss": 6.708555978058172, + "tokens_seen": 11796480 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001755952380952381, + "loss": 3.242, + "theoretical_loss": 6.701891300380584, + "tokens_seen": 11862016 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001765873015873016, + "loss": 3.3468, + "theoretical_loss": 6.6952735882553185, + "tokens_seen": 11927552 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001775793650793651, + "loss": 3.6783, + "theoretical_loss": 6.6887022553809405, + "tokens_seen": 11993088 + }, + { + "epoch": 0.0, + "learning_rate": 0.00017857142857142857, + "loss": 3.7245, + "theoretical_loss": 6.682176725928034, + "tokens_seen": 12058624 + }, + { + "epoch": 0.0, + "learning_rate": 0.00017956349206349207, + "loss": 3.4552, + "theoretical_loss": 6.675696434296475, + "tokens_seen": 12124160 + }, + { + "epoch": 0.0, + "learning_rate": 0.00018055555555555555, + "loss": 3.7103, + "theoretical_loss": 6.669260824879574, + "tokens_seen": 12189696 + }, + { + "epoch": 0.0, + "learning_rate": 0.00018154761904761905, + "loss": 3.6315, + "theoretical_loss": 6.662869351834917, + "tokens_seen": 12255232 + }, + { + "epoch": 0.0, + "learning_rate": 0.00018253968253968252, + "loss": 3.6235, + "theoretical_loss": 6.656521478861638, + "tokens_seen": 12320768 + }, + { + "epoch": 0.0, + "learning_rate": 0.00018353174603174602, + "loss": 3.574, + "theoretical_loss": 6.650216678983931, + "tokens_seen": 12386304 + }, + { + "epoch": 0.0, + "learning_rate": 0.00018452380952380953, + "loss": 3.4108, + "theoretical_loss": 6.643954434340591, + "tokens_seen": 12451840 + }, + { + "epoch": 0.0, + "learning_rate": 0.00018551587301587303, + "loss": 3.5704, + "theoretical_loss": 6.6377342359803855, + "tokens_seen": 12517376 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001865079365079365, + "loss": 3.4075, + "theoretical_loss": 6.631555583663063, + "tokens_seen": 12582912 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001875, + "loss": 3.5662, + "theoretical_loss": 6.625417985665809, + "tokens_seen": 12648448 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001884920634920635, + "loss": 3.2934, + "theoretical_loss": 6.619320958594983, + "tokens_seen": 12713984 + }, + { + "epoch": 0.0, + "learning_rate": 0.00018948412698412698, + "loss": 3.4387, + "theoretical_loss": 6.6132640272029715, + "tokens_seen": 12779520 + }, + { + "epoch": 0.0, + "learning_rate": 0.00019047619047619048, + "loss": 3.4155, + "theoretical_loss": 6.607246724209972, + "tokens_seen": 12845056 + }, + { + "epoch": 0.0, + "learning_rate": 0.00019146825396825398, + "loss": 3.4582, + "theoretical_loss": 6.601268590130569, + "tokens_seen": 12910592 + }, + { + "epoch": 0.0, + "learning_rate": 0.00019246031746031748, + "loss": 3.3766, + "theoretical_loss": 6.59532917310494, + "tokens_seen": 12976128 + }, + { + "epoch": 0.0, + "learning_rate": 0.00019345238095238096, + "loss": 3.4424, + "theoretical_loss": 6.589428028734545, + "tokens_seen": 13041664 + }, + { + "epoch": 0.0, + "objective/train/advantage_avg": 0.017791178077459335, + "objective/train/docs_used": 19731, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.6331467628479, + "objective/train/original_loss": 5.633145809173584, + "objective/train/theoretical_loss": 6.583564719922174, + "objective/train/tokens_used": 33567200, + "objective/train/value_avg": -0.02490234375, + "objective/train/value_loss": 0.0006902085733599961, + "objective/train/value_max": -0.00498199462890625, + "objective/train/value_min": -0.1302490234375, + "objective/train/value_reward_corr": -0.07877530466446529, + "objective/train/value_std": 0.0162506103515625, + "objective/train/weight_avg": 1.2188433408737183, + "objective/train/weighted_lm_loss": 7.0526814460754395, + "objective/train/weights_max": 3.618537187576294, + "objective/train/weights_min": 0.37747922539711, + "theoretical_loss": 6.583564719922174, + "tokens_seen": 13107200 + }, + { + "epoch": 0.0, + "learning_rate": 0.00019444444444444446, + "loss": 3.5604, + "theoretical_loss": 6.583564719922174, + "tokens_seen": 13107200 + }, + { + "epoch": 0.0, + "learning_rate": 0.00019543650793650793, + "loss": 3.5484, + "theoretical_loss": 6.57773881671619, + "tokens_seen": 13172736 + }, + { + "epoch": 0.0, + "learning_rate": 0.00019642857142857144, + "loss": 3.4668, + "theoretical_loss": 6.571949896158857, + "tokens_seen": 13238272 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001974206349206349, + "loss": 3.5171, + "theoretical_loss": 6.566197542138623, + "tokens_seen": 13303808 + }, + { + "epoch": 0.0, + "learning_rate": 0.0001984126984126984, + "loss": 3.4753, + "theoretical_loss": 6.560481345246222, + "tokens_seen": 13369344 + }, + { + "epoch": 0.0, + "learning_rate": 0.00019940476190476191, + "loss": 3.3732, + "theoretical_loss": 6.554800902634504, + "tokens_seen": 13434880 + }, + { + "epoch": 0.0, + "learning_rate": 0.00020039682539682542, + "loss": 3.4985, + "theoretical_loss": 6.54915581788185, + "tokens_seen": 13500416 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002013888888888889, + "loss": 3.6148, + "theoretical_loss": 6.54354570085909, + "tokens_seen": 13565952 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002023809523809524, + "loss": 3.4329, + "theoretical_loss": 6.537970167599786, + "tokens_seen": 13631488 + }, + { + "epoch": 0.0, + "learning_rate": 0.00020337301587301587, + "loss": 3.3553, + "theoretical_loss": 6.532428840173829, + "tokens_seen": 13697024 + }, + { + "epoch": 0.0, + "learning_rate": 0.00020436507936507937, + "loss": 3.3952, + "theoretical_loss": 6.52692134656418, + "tokens_seen": 13762560 + }, + { + "epoch": 0.0, + "learning_rate": 0.00020535714285714284, + "loss": 3.4005, + "theoretical_loss": 6.521447320546738, + "tokens_seen": 13828096 + }, + { + "epoch": 0.0, + "learning_rate": 0.00020634920634920634, + "loss": 3.4942, + "theoretical_loss": 6.516006401573176, + "tokens_seen": 13893632 + }, + { + "epoch": 0.0, + "learning_rate": 0.00020734126984126985, + "loss": 3.5705, + "theoretical_loss": 6.51059823465671, + "tokens_seen": 13959168 + }, + { + "epoch": 0.0, + "learning_rate": 0.00020833333333333335, + "loss": 3.5429, + "theoretical_loss": 6.505222470260669, + "tokens_seen": 14024704 + }, + { + "epoch": 0.0, + "learning_rate": 0.00020932539682539685, + "loss": 3.2291, + "theoretical_loss": 6.499878764189829, + "tokens_seen": 14090240 + }, + { + "epoch": 0.0, + "learning_rate": 0.00021031746031746032, + "loss": 3.3168, + "theoretical_loss": 6.494566777484378, + "tokens_seen": 14155776 + }, + { + "epoch": 0.0, + "learning_rate": 0.00021130952380952382, + "loss": 3.6775, + "theoretical_loss": 6.489286176316504, + "tokens_seen": 14221312 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002123015873015873, + "loss": 3.44, + "theoretical_loss": 6.48403663188946, + "tokens_seen": 14286848 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002132936507936508, + "loss": 3.141, + "theoretical_loss": 6.478817820339071, + "tokens_seen": 14352384 + }, + { + "epoch": 0.0, + "learning_rate": 0.00021428571428571427, + "loss": 3.3508, + "theoretical_loss": 6.473629422637629, + "tokens_seen": 14417920 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002152777777777778, + "loss": 3.3139, + "theoretical_loss": 6.468471124500068, + "tokens_seen": 14483456 + }, + { + "epoch": 0.0, + "learning_rate": 0.00021626984126984128, + "loss": 3.2412, + "theoretical_loss": 6.463342616292376, + "tokens_seen": 14548992 + }, + { + "epoch": 0.0, + "learning_rate": 0.00021726190476190478, + "loss": 3.3389, + "theoretical_loss": 6.458243592942189, + "tokens_seen": 14614528 + }, + { + "epoch": 0.0, + "learning_rate": 0.00021825396825396825, + "loss": 3.2499, + "theoretical_loss": 6.453173753851489, + "tokens_seen": 14680064 + }, + { + "epoch": 0.0, + "objective/train/advantage_avg": 0.011923457495868206, + "objective/train/docs_used": 20428, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.278204917907715, + "objective/train/original_loss": 5.278204917907715, + "objective/train/theoretical_loss": 6.448132802811353, + "objective/train/tokens_used": 35205600, + "objective/train/value_avg": -0.017364501953125, + "objective/train/value_loss": 0.0003513648989610374, + "objective/train/value_max": -0.0036792755126953125, + "objective/train/value_min": -0.10577392578125, + "objective/train/value_reward_corr": 0.07631083084594213, + "objective/train/value_std": 0.0122833251953125, + "objective/train/weight_avg": 1.1390091180801392, + "objective/train/weighted_lm_loss": 6.085953712463379, + "objective/train/weights_max": 2.654543876647949, + "objective/train/weights_min": 0.4326308071613312, + "theoretical_loss": 6.448132802811353, + "tokens_seen": 14745600 + }, + { + "epoch": 0.0, + "learning_rate": 0.00021924603174603176, + "loss": 3.1505, + "theoretical_loss": 6.448132802811353, + "tokens_seen": 14745600 + }, + { + "epoch": 0.0, + "learning_rate": 0.00022023809523809523, + "loss": 3.3554, + "theoretical_loss": 6.443120447918698, + "tokens_seen": 14811136 + }, + { + "epoch": 0.0, + "learning_rate": 0.00022123015873015873, + "loss": 3.4953, + "theoretical_loss": 6.438136401494974, + "tokens_seen": 14876672 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002222222222222222, + "loss": 3.1914, + "theoretical_loss": 6.433180380006723, + "tokens_seen": 14942208 + }, + { + "epoch": 0.0, + "learning_rate": 0.00022321428571428573, + "loss": 3.4241, + "theoretical_loss": 6.428252103987999, + "tokens_seen": 15007744 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002242063492063492, + "loss": 3.3862, + "theoretical_loss": 6.423351297964546, + "tokens_seen": 15073280 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002251984126984127, + "loss": 3.2943, + "theoretical_loss": 6.41847769037973, + "tokens_seen": 15138816 + }, + { + "epoch": 0.0, + "learning_rate": 0.00022619047619047618, + "loss": 3.2048, + "theoretical_loss": 6.413631013522151, + "tokens_seen": 15204352 + }, + { + "epoch": 0.0, + "learning_rate": 0.00022718253968253969, + "loss": 3.1993, + "theoretical_loss": 6.4088110034549, + "tokens_seen": 15269888 + }, + { + "epoch": 0.0, + "learning_rate": 0.00022817460317460316, + "loss": 3.3796, + "theoretical_loss": 6.404017399946415, + "tokens_seen": 15335424 + }, + { + "epoch": 0.0, + "learning_rate": 0.00022916666666666666, + "loss": 3.4797, + "theoretical_loss": 6.399249946402892, + "tokens_seen": 15400960 + }, + { + "epoch": 0.0, + "learning_rate": 0.00023015873015873016, + "loss": 3.2126, + "theoretical_loss": 6.394508389802212, + "tokens_seen": 15466496 + }, + { + "epoch": 0.0, + "learning_rate": 0.00023115079365079367, + "loss": 3.1957, + "theoretical_loss": 6.3897924806293345, + "tokens_seen": 15532032 + }, + { + "epoch": 0.0, + "learning_rate": 0.00023214285714285717, + "loss": 3.2256, + "theoretical_loss": 6.385101972813146, + "tokens_seen": 15597568 + }, + { + "epoch": 0.0, + "learning_rate": 0.00023313492063492064, + "loss": 3.1022, + "theoretical_loss": 6.380436623664683, + "tokens_seen": 15663104 + }, + { + "epoch": 0.0, + "learning_rate": 0.00023412698412698414, + "loss": 3.2668, + "theoretical_loss": 6.3757961938167265, + "tokens_seen": 15728640 + }, + { + "epoch": 0.0, + "learning_rate": 0.00023511904761904762, + "loss": 3.3856, + "theoretical_loss": 6.3711804471647255, + "tokens_seen": 15794176 + }, + { + "epoch": 0.0, + "learning_rate": 0.00023611111111111112, + "loss": 3.1293, + "theoretical_loss": 6.366589150809007, + "tokens_seen": 15859712 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002371031746031746, + "loss": 3.2271, + "theoretical_loss": 6.36202207499824, + "tokens_seen": 15925248 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002380952380952381, + "loss": 3.1905, + "theoretical_loss": 6.357478993074131, + "tokens_seen": 15990784 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002390873015873016, + "loss": 3.0126, + "theoretical_loss": 6.352959681417312, + "tokens_seen": 16056320 + }, + { + "epoch": 0.0, + "learning_rate": 0.0002400793650793651, + "loss": 3.0417, + "theoretical_loss": 6.348463919394389, + "tokens_seen": 16121856 + }, + { + "epoch": 0.0, + "learning_rate": 0.00024107142857142857, + "loss": 3.3571, + "theoretical_loss": 6.343991489306125, + "tokens_seen": 16187392 + }, + { + "epoch": 0.0, + "learning_rate": 0.00024206349206349207, + "loss": 3.2193, + "theoretical_loss": 6.33954217633673, + "tokens_seen": 16252928 + }, + { + "epoch": 0.0, + "learning_rate": 0.00024305555555555555, + "loss": 3.301, + "theoretical_loss": 6.33511576850422, + "tokens_seen": 16318464 + }, + { + "epoch": 0.0, + "objective/train/advantage_avg": 0.011138406582176685, + "objective/train/docs_used": 21641, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.063250541687012, + "objective/train/original_loss": 5.063250541687012, + "objective/train/theoretical_loss": 6.330712056611843, + "objective/train/tokens_used": 36844000, + "objective/train/value_avg": -0.01715087890625, + "objective/train/value_loss": 0.000311657931888476, + "objective/train/value_max": -0.0038089752197265625, + "objective/train/value_min": -0.08770751953125, + "objective/train/value_reward_corr": 0.1304287920908867, + "objective/train/value_std": 0.01087188720703125, + "objective/train/weight_avg": 1.1282998323440552, + "objective/train/weighted_lm_loss": 5.783185005187988, + "objective/train/weights_max": 2.2725961208343506, + "objective/train/weights_min": 0.3413560688495636, + "theoretical_loss": 6.330712056611843, + "tokens_seen": 16384000 + }, + { + "epoch": 0.0, + "learning_rate": 0.00024404761904761905, + "loss": 3.2327, + "theoretical_loss": 6.330712056611843, + "tokens_seen": 16384000 + }, + { + "epoch": 0.0, + "learning_rate": 0.00024503968253968255, + "loss": 3.3487, + "theoretical_loss": 6.32633083420051, + "tokens_seen": 16449536 + }, + { + "epoch": 0.01, + "learning_rate": 0.000246031746031746, + "loss": 3.2043, + "theoretical_loss": 6.32197189750223, + "tokens_seen": 16515072 + }, + { + "epoch": 0.01, + "learning_rate": 0.00024702380952380955, + "loss": 2.9662, + "theoretical_loss": 6.317635045394523, + "tokens_seen": 16580608 + }, + { + "epoch": 0.01, + "learning_rate": 0.000248015873015873, + "loss": 3.1484, + "theoretical_loss": 6.3133200793557815, + "tokens_seen": 16646144 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002490079365079365, + "loss": 2.8924, + "theoretical_loss": 6.3090268034215455, + "tokens_seen": 16711680 + }, + { + "epoch": 0.01, + "learning_rate": 0.00025, + "loss": 3.1533, + "theoretical_loss": 6.304755024141699, + "tokens_seen": 16777216 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002509920634920635, + "loss": 3.3463, + "theoretical_loss": 6.300504550538534, + "tokens_seen": 16842752 + }, + { + "epoch": 0.01, + "learning_rate": 0.000251984126984127, + "loss": 3.1454, + "theoretical_loss": 6.296275194065678, + "tokens_seen": 16908288 + }, + { + "epoch": 0.01, + "learning_rate": 0.00025297619047619046, + "loss": 3.2773, + "theoretical_loss": 6.292066768567858, + "tokens_seen": 16973824 + }, + { + "epoch": 0.01, + "learning_rate": 0.00025396825396825396, + "loss": 2.9542, + "theoretical_loss": 6.28787909024149, + "tokens_seen": 17039360 + }, + { + "epoch": 0.01, + "learning_rate": 0.00025496031746031746, + "loss": 3.1307, + "theoretical_loss": 6.283711977596056, + "tokens_seen": 17104896 + }, + { + "epoch": 0.01, + "learning_rate": 0.00025595238095238096, + "loss": 3.2373, + "theoretical_loss": 6.279565251416267, + "tokens_seen": 17170432 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002569444444444444, + "loss": 3.0126, + "theoretical_loss": 6.275438734724988, + "tokens_seen": 17235968 + }, + { + "epoch": 0.01, + "learning_rate": 0.00025793650793650796, + "loss": 3.2568, + "theoretical_loss": 6.271332252746902, + "tokens_seen": 17301504 + }, + { + "epoch": 0.01, + "learning_rate": 0.00025892857142857146, + "loss": 3.2219, + "theoretical_loss": 6.267245632872896, + "tokens_seen": 17367040 + }, + { + "epoch": 0.01, + "learning_rate": 0.00025992063492063497, + "loss": 3.0633, + "theoretical_loss": 6.263178704625169, + "tokens_seen": 17432576 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002609126984126984, + "loss": 3.1378, + "theoretical_loss": 6.2591312996230055, + "tokens_seen": 17498112 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002619047619047619, + "loss": 3.0554, + "theoretical_loss": 6.255103251549257, + "tokens_seen": 17563648 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002628968253968254, + "loss": 3.2136, + "theoretical_loss": 6.251094396117449, + "tokens_seen": 17629184 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002638888888888889, + "loss": 3.0597, + "theoretical_loss": 6.247104571039561, + "tokens_seen": 17694720 + }, + { + "epoch": 0.01, + "learning_rate": 0.00026488095238095237, + "loss": 3.0393, + "theoretical_loss": 6.243133615994411, + "tokens_seen": 17760256 + }, + { + "epoch": 0.01, + "learning_rate": 0.00026587301587301587, + "loss": 3.1251, + "theoretical_loss": 6.239181372596676, + "tokens_seen": 17825792 + }, + { + "epoch": 0.01, + "learning_rate": 0.00026686507936507937, + "loss": 3.0325, + "theoretical_loss": 6.235247684366492, + "tokens_seen": 17891328 + }, + { + "epoch": 0.01, + "learning_rate": 0.00026785714285714287, + "loss": 3.0744, + "theoretical_loss": 6.231332396699658, + "tokens_seen": 17956864 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.009136986918747425, + "objective/train/docs_used": 22253, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.189753532409668, + "objective/train/original_loss": 5.189753532409668, + "objective/train/theoretical_loss": 6.227435356838402, + "objective/train/tokens_used": 38482400, + "objective/train/value_avg": -0.01374053955078125, + "objective/train/value_loss": 0.0002494758809916675, + "objective/train/value_max": -0.0027904510498046875, + "objective/train/value_min": -0.11181640625, + "objective/train/value_reward_corr": -0.025826456766141204, + "objective/train/value_std": 0.0098419189453125, + "objective/train/weight_avg": 1.1050573587417603, + "objective/train/weighted_lm_loss": 5.802955150604248, + "objective/train/weights_max": 3.059232473373413, + "objective/train/weights_min": 0.30825307965278625, + "theoretical_loss": 6.227435356838402, + "tokens_seen": 18022400 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002688492063492063, + "loss": 3.2456, + "theoretical_loss": 6.227435356838402, + "tokens_seen": 18022400 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002698412698412698, + "loss": 3.0582, + "theoretical_loss": 6.223556413842715, + "tokens_seen": 18087936 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002708333333333333, + "loss": 3.2341, + "theoretical_loss": 6.219695418562217, + "tokens_seen": 18153472 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002718253968253968, + "loss": 3.0604, + "theoretical_loss": 6.215852223608581, + "tokens_seen": 18219008 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002728174603174603, + "loss": 3.151, + "theoretical_loss": 6.212026683328462, + "tokens_seen": 18284544 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002738095238095238, + "loss": 2.9625, + "theoretical_loss": 6.208218653776937, + "tokens_seen": 18350080 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002748015873015873, + "loss": 3.062, + "theoretical_loss": 6.2044279926914605, + "tokens_seen": 18415616 + }, + { + "epoch": 0.01, + "learning_rate": 0.00027579365079365083, + "loss": 3.0468, + "theoretical_loss": 6.20065455946629, + "tokens_seen": 18481152 + }, + { + "epoch": 0.01, + "learning_rate": 0.00027678571428571433, + "loss": 3.1169, + "theoretical_loss": 6.1968982151273995, + "tokens_seen": 18546688 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002777777777777778, + "loss": 3.0494, + "theoretical_loss": 6.193158822307858, + "tokens_seen": 18612224 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002787698412698413, + "loss": 2.802, + "theoretical_loss": 6.1894362452236615, + "tokens_seen": 18677760 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002797619047619048, + "loss": 2.9126, + "theoretical_loss": 6.185730349650021, + "tokens_seen": 18743296 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002807539682539683, + "loss": 3.0061, + "theoretical_loss": 6.182041002898073, + "tokens_seen": 18808832 + }, + { + "epoch": 0.01, + "learning_rate": 0.00028174603174603173, + "loss": 2.9626, + "theoretical_loss": 6.17836807379203, + "tokens_seen": 18874368 + }, + { + "epoch": 0.01, + "learning_rate": 0.00028273809523809523, + "loss": 2.9603, + "theoretical_loss": 6.174711432646743, + "tokens_seen": 18939904 + }, + { + "epoch": 0.01, + "learning_rate": 0.00028373015873015873, + "loss": 2.8214, + "theoretical_loss": 6.171070951245673, + "tokens_seen": 19005440 + }, + { + "epoch": 0.01, + "learning_rate": 0.00028472222222222223, + "loss": 3.0407, + "theoretical_loss": 6.167446502819269, + "tokens_seen": 19070976 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002857142857142857, + "loss": 2.9263, + "theoretical_loss": 6.163837962023724, + "tokens_seen": 19136512 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002867063492063492, + "loss": 3.1286, + "theoretical_loss": 6.160245204920132, + "tokens_seen": 19202048 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002876984126984127, + "loss": 3.0395, + "theoretical_loss": 6.156668108954003, + "tokens_seen": 19267584 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002886904761904762, + "loss": 3.0142, + "theoretical_loss": 6.1531065529351565, + "tokens_seen": 19333120 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002896825396825397, + "loss": 3.101, + "theoretical_loss": 6.149560417017971, + "tokens_seen": 19398656 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002906746031746032, + "loss": 2.9247, + "theoretical_loss": 6.146029582681981, + "tokens_seen": 19464192 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002916666666666667, + "loss": 3.1222, + "theoretical_loss": 6.1425139327128235, + "tokens_seen": 19529728 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002926587301587302, + "loss": 3.0376, + "theoretical_loss": 6.139013351183521, + "tokens_seen": 19595264 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.011530731804668903, + "objective/train/docs_used": 23612, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.75862455368042, + "objective/train/original_loss": 5.75862455368042, + "objective/train/theoretical_loss": 6.135527723436086, + "objective/train/tokens_used": 40120800, + "objective/train/value_avg": -0.0174102783203125, + "objective/train/value_loss": 0.0003715897910296917, + "objective/train/value_max": -0.0030994415283203125, + "objective/train/value_min": -0.20166015625, + "objective/train/value_reward_corr": 0.013559516065810732, + "objective/train/value_std": 0.01337432861328125, + "objective/train/weight_avg": 1.1371591091156006, + "objective/train/weighted_lm_loss": 6.633453369140625, + "objective/train/weights_max": 7.512750148773193, + "objective/train/weights_min": 0.5943675637245178, + "theoretical_loss": 6.135527723436086, + "tokens_seen": 19660800 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002936507936507937, + "loss": 3.0539, + "theoretical_loss": 6.135527723436086, + "tokens_seen": 19660800 + }, + { + "epoch": 0.01, + "learning_rate": 0.00029464285714285714, + "loss": 2.9936, + "theoretical_loss": 6.132056936063465, + "tokens_seen": 19726336 + }, + { + "epoch": 0.01, + "learning_rate": 0.00029563492063492064, + "loss": 2.9642, + "theoretical_loss": 6.128600876891781, + "tokens_seen": 19791872 + }, + { + "epoch": 0.01, + "learning_rate": 0.00029662698412698414, + "loss": 2.9739, + "theoretical_loss": 6.125159434962898, + "tokens_seen": 19857408 + }, + { + "epoch": 0.01, + "learning_rate": 0.00029761904761904765, + "loss": 3.0348, + "theoretical_loss": 6.121732500517281, + "tokens_seen": 19922944 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002986111111111111, + "loss": 2.9833, + "theoretical_loss": 6.118319964977157, + "tokens_seen": 19988480 + }, + { + "epoch": 0.01, + "learning_rate": 0.0002996031746031746, + "loss": 3.0073, + "theoretical_loss": 6.114921720929972, + "tokens_seen": 20054016 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003005952380952381, + "loss": 2.9815, + "theoretical_loss": 6.111537662112119, + "tokens_seen": 20119552 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003015873015873016, + "loss": 2.8465, + "theoretical_loss": 6.1081676833929555, + "tokens_seen": 20185088 + }, + { + "epoch": 0.01, + "learning_rate": 0.00030257936507936505, + "loss": 2.8008, + "theoretical_loss": 6.104811680759095, + "tokens_seen": 20250624 + }, + { + "epoch": 0.01, + "learning_rate": 0.00030357142857142855, + "loss": 2.9418, + "theoretical_loss": 6.101469551298964, + "tokens_seen": 20316160 + }, + { + "epoch": 0.01, + "learning_rate": 0.00030456349206349205, + "loss": 2.9553, + "theoretical_loss": 6.098141193187615, + "tokens_seen": 20381696 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003055555555555556, + "loss": 2.9052, + "theoretical_loss": 6.094826505671813, + "tokens_seen": 20447232 + }, + { + "epoch": 0.01, + "learning_rate": 0.00030654761904761905, + "loss": 2.9609, + "theoretical_loss": 6.091525389055353, + "tokens_seen": 20512768 + }, + { + "epoch": 0.01, + "learning_rate": 0.00030753968253968255, + "loss": 2.9579, + "theoretical_loss": 6.088237744684644, + "tokens_seen": 20578304 + }, + { + "epoch": 0.01, + "learning_rate": 0.00030853174603174605, + "loss": 2.9849, + "theoretical_loss": 6.084963474934522, + "tokens_seen": 20643840 + }, + { + "epoch": 0.01, + "learning_rate": 0.00030952380952380956, + "loss": 2.9352, + "theoretical_loss": 6.081702483194306, + "tokens_seen": 20709376 + }, + { + "epoch": 0.01, + "learning_rate": 0.000310515873015873, + "loss": 2.8845, + "theoretical_loss": 6.078454673854084, + "tokens_seen": 20774912 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003115079365079365, + "loss": 2.9011, + "theoretical_loss": 6.075219952291231, + "tokens_seen": 20840448 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003125, + "loss": 2.9672, + "theoretical_loss": 6.071998224857146, + "tokens_seen": 20905984 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003134920634920635, + "loss": 3.1649, + "theoretical_loss": 6.068789398864208, + "tokens_seen": 20971520 + }, + { + "epoch": 0.01, + "learning_rate": 0.000314484126984127, + "loss": 2.7794, + "theoretical_loss": 6.065593382572951, + "tokens_seen": 21037056 + }, + { + "epoch": 0.01, + "learning_rate": 0.00031547619047619046, + "loss": 2.8333, + "theoretical_loss": 6.062410085179447, + "tokens_seen": 21102592 + }, + { + "epoch": 0.01, + "learning_rate": 0.00031646825396825396, + "loss": 2.888, + "theoretical_loss": 6.059239416802894, + "tokens_seen": 21168128 + }, + { + "epoch": 0.01, + "learning_rate": 0.00031746031746031746, + "loss": 2.8476, + "theoretical_loss": 6.0560812884734085, + "tokens_seen": 21233664 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.00934158731251955, + "objective/train/docs_used": 24330, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.932507514953613, + "objective/train/original_loss": 5.93250846862793, + "objective/train/theoretical_loss": 6.052935612120018, + "objective/train/tokens_used": 41759200, + "objective/train/value_avg": -0.0136260986328125, + "objective/train/value_loss": 0.0002475394867360592, + "objective/train/value_max": -0.0027675628662109375, + "objective/train/value_min": -0.156982421875, + "objective/train/value_reward_corr": 0.06626976650281922, + "objective/train/value_std": 0.01030731201171875, + "objective/train/weight_avg": 1.1071875095367432, + "objective/train/weighted_lm_loss": 6.621048927307129, + "objective/train/weights_max": 4.805803298950195, + "objective/train/weights_min": 0.300566166639328, + "theoretical_loss": 6.052935612120018, + "tokens_seen": 21299200 + }, + { + "epoch": 0.01, + "learning_rate": 0.00031845238095238096, + "loss": 2.9942, + "theoretical_loss": 6.052935612120018, + "tokens_seen": 21299200 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003194444444444444, + "loss": 2.9034, + "theoretical_loss": 6.049802300558843, + "tokens_seen": 21364736 + }, + { + "epoch": 0.01, + "learning_rate": 0.00032043650793650796, + "loss": 2.8588, + "theoretical_loss": 6.046681267481475, + "tokens_seen": 21430272 + }, + { + "epoch": 0.01, + "learning_rate": 0.00032142857142857147, + "loss": 2.8696, + "theoretical_loss": 6.043572427443539, + "tokens_seen": 21495808 + }, + { + "epoch": 0.01, + "learning_rate": 0.00032242063492063497, + "loss": 2.8936, + "theoretical_loss": 6.040475695853447, + "tokens_seen": 21561344 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003234126984126984, + "loss": 2.8837, + "theoretical_loss": 6.037390988961312, + "tokens_seen": 21626880 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003244047619047619, + "loss": 3.0365, + "theoretical_loss": 6.034318223848068, + "tokens_seen": 21692416 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003253968253968254, + "loss": 2.9921, + "theoretical_loss": 6.031257318414738, + "tokens_seen": 21757952 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003263888888888889, + "loss": 2.8684, + "theoretical_loss": 6.028208191371885, + "tokens_seen": 21823488 + }, + { + "epoch": 0.01, + "learning_rate": 0.00032738095238095237, + "loss": 2.8434, + "theoretical_loss": 6.025170762229224, + "tokens_seen": 21889024 + }, + { + "epoch": 0.01, + "learning_rate": 0.00032837301587301587, + "loss": 2.8185, + "theoretical_loss": 6.022144951285407, + "tokens_seen": 21954560 + }, + { + "epoch": 0.01, + "learning_rate": 0.00032936507936507937, + "loss": 2.8408, + "theoretical_loss": 6.019130679617959, + "tokens_seen": 22020096 + }, + { + "epoch": 0.01, + "learning_rate": 0.00033035714285714287, + "loss": 2.8999, + "theoretical_loss": 6.016127869073374, + "tokens_seen": 22085632 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003313492063492063, + "loss": 2.9195, + "theoretical_loss": 6.01313644225737, + "tokens_seen": 22151168 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003323412698412698, + "loss": 3.0036, + "theoretical_loss": 6.010156322525292, + "tokens_seen": 22216704 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003333333333333333, + "loss": 2.7858, + "theoretical_loss": 6.007187433972668, + "tokens_seen": 22282240 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003343253968253968, + "loss": 2.7208, + "theoretical_loss": 6.004229701425903, + "tokens_seen": 22347776 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003353174603174603, + "loss": 2.9428, + "theoretical_loss": 6.001283050433128, + "tokens_seen": 22413312 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003363095238095238, + "loss": 2.8381, + "theoretical_loss": 5.998347407255185, + "tokens_seen": 22478848 + }, + { + "epoch": 0.01, + "learning_rate": 0.00033730158730158733, + "loss": 2.889, + "theoretical_loss": 5.995422698856746, + "tokens_seen": 22544384 + }, + { + "epoch": 0.01, + "learning_rate": 0.00033829365079365083, + "loss": 2.8622, + "theoretical_loss": 5.992508852897577, + "tokens_seen": 22609920 + }, + { + "epoch": 0.01, + "learning_rate": 0.00033928571428571433, + "loss": 2.7967, + "theoretical_loss": 5.989605797723927, + "tokens_seen": 22675456 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003402777777777778, + "loss": 2.7702, + "theoretical_loss": 5.986713462360052, + "tokens_seen": 22740992 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003412698412698413, + "loss": 2.7932, + "theoretical_loss": 5.983831776499871, + "tokens_seen": 22806528 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003422619047619048, + "loss": 2.8741, + "theoretical_loss": 5.980960670498742, + "tokens_seen": 22872064 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.00702398456633091, + "objective/train/docs_used": 24983, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.591332912445068, + "objective/train/original_loss": 5.591332912445068, + "objective/train/theoretical_loss": 5.978100075365368, + "objective/train/tokens_used": 43397600, + "objective/train/value_avg": -0.0139312744140625, + "objective/train/value_loss": 0.00028285259031690657, + "objective/train/value_max": -0.0023326873779296875, + "objective/train/value_min": -0.1795654296875, + "objective/train/value_reward_corr": 0.001762871428231544, + "objective/train/value_std": 0.01090240478515625, + "objective/train/weight_avg": 1.0857257843017578, + "objective/train/weighted_lm_loss": 6.122445583343506, + "objective/train/weights_max": 5.906911373138428, + "objective/train/weights_min": 0.2668619453907013, + "theoretical_loss": 5.978100075365368, + "tokens_seen": 22937600 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003432539682539683, + "loss": 2.9173, + "theoretical_loss": 5.978100075365368, + "tokens_seen": 22937600 + }, + { + "epoch": 0.01, + "learning_rate": 0.00034424603174603173, + "loss": 2.8758, + "theoretical_loss": 5.975249922753816, + "tokens_seen": 23003136 + }, + { + "epoch": 0.01, + "learning_rate": 0.00034523809523809523, + "loss": 2.9084, + "theoretical_loss": 5.972410144955672, + "tokens_seen": 23068672 + }, + { + "epoch": 0.01, + "learning_rate": 0.00034623015873015873, + "loss": 2.8624, + "theoretical_loss": 5.969580674892294, + "tokens_seen": 23134208 + }, + { + "epoch": 0.01, + "learning_rate": 0.00034722222222222224, + "loss": 2.6836, + "theoretical_loss": 5.966761446107192, + "tokens_seen": 23199744 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003482142857142857, + "loss": 2.7024, + "theoretical_loss": 5.963952392758523, + "tokens_seen": 23265280 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003492063492063492, + "loss": 2.6056, + "theoretical_loss": 5.961153449611697, + "tokens_seen": 23330816 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003501984126984127, + "loss": 2.7083, + "theoretical_loss": 5.958364552032075, + "tokens_seen": 23396352 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003511904761904762, + "loss": 2.8809, + "theoretical_loss": 5.955585635977804, + "tokens_seen": 23461888 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003521825396825397, + "loss": 2.821, + "theoretical_loss": 5.952816637992738, + "tokens_seen": 23527424 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003531746031746032, + "loss": 2.8019, + "theoretical_loss": 5.950057495199465, + "tokens_seen": 23592960 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003541666666666667, + "loss": 2.7, + "theoretical_loss": 5.947308145292439, + "tokens_seen": 23658496 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003551587301587302, + "loss": 2.9498, + "theoretical_loss": 5.944568526531218, + "tokens_seen": 23724032 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003561507936507937, + "loss": 2.8854, + "theoretical_loss": 5.941838577733787, + "tokens_seen": 23789568 + }, + { + "epoch": 0.01, + "learning_rate": 0.00035714285714285714, + "loss": 2.7905, + "theoretical_loss": 5.939118238269988, + "tokens_seen": 23855104 + }, + { + "epoch": 0.01, + "learning_rate": 0.00035813492063492064, + "loss": 2.9265, + "theoretical_loss": 5.936407448055045, + "tokens_seen": 23920640 + }, + { + "epoch": 0.01, + "learning_rate": 0.00035912698412698415, + "loss": 2.7672, + "theoretical_loss": 5.933706147543174, + "tokens_seen": 23986176 + }, + { + "epoch": 0.01, + "learning_rate": 0.00036011904761904765, + "loss": 2.7659, + "theoretical_loss": 5.9310142777212995, + "tokens_seen": 24051712 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003611111111111111, + "loss": 2.8054, + "theoretical_loss": 5.928331780102846, + "tokens_seen": 24117248 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003621031746031746, + "loss": 2.857, + "theoretical_loss": 5.925658596721627, + "tokens_seen": 24182784 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003630952380952381, + "loss": 2.8701, + "theoretical_loss": 5.922994670125819, + "tokens_seen": 24248320 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003640873015873016, + "loss": 2.7272, + "theoretical_loss": 5.920339943372026, + "tokens_seen": 24313856 + }, + { + "epoch": 0.01, + "learning_rate": 0.00036507936507936505, + "loss": 2.7936, + "theoretical_loss": 5.917694360019418, + "tokens_seen": 24379392 + }, + { + "epoch": 0.01, + "learning_rate": 0.00036607142857142855, + "loss": 2.7818, + "theoretical_loss": 5.915057864123962, + "tokens_seen": 24444928 + }, + { + "epoch": 0.01, + "learning_rate": 0.00036706349206349205, + "loss": 2.7538, + "theoretical_loss": 5.912430400232731, + "tokens_seen": 24510464 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.006513331551104784, + "objective/train/docs_used": 26439, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.757891654968262, + "objective/train/original_loss": 4.757890701293945, + "objective/train/theoretical_loss": 5.909811913378293, + "objective/train/tokens_used": 45036000, + "objective/train/value_avg": -0.012725830078125, + "objective/train/value_loss": 0.0002583596797194332, + "objective/train/value_max": -0.0022602081298828125, + "objective/train/value_min": -0.2286376953125, + "objective/train/value_reward_corr": -0.056101801618329034, + "objective/train/value_std": 0.0101318359375, + "objective/train/weight_avg": 1.0796440839767456, + "objective/train/weighted_lm_loss": 5.261064052581787, + "objective/train/weights_max": 9.839224815368652, + "objective/train/weights_min": 0.2501309812068939, + "theoretical_loss": 5.909811913378293, + "tokens_seen": 24576000 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003680555555555556, + "loss": 2.6913, + "theoretical_loss": 5.909811913378293, + "tokens_seen": 24576000 + }, + { + "epoch": 0.01, + "learning_rate": 0.00036904761904761905, + "loss": 2.784, + "theoretical_loss": 5.907202349073183, + "tokens_seen": 24641536 + }, + { + "epoch": 0.01, + "learning_rate": 0.00037003968253968255, + "loss": 2.6375, + "theoretical_loss": 5.904601653304437, + "tokens_seen": 24707072 + }, + { + "epoch": 0.01, + "learning_rate": 0.00037103174603174606, + "loss": 2.7039, + "theoretical_loss": 5.902009772528231, + "tokens_seen": 24772608 + }, + { + "epoch": 0.01, + "learning_rate": 0.00037202380952380956, + "loss": 2.8754, + "theoretical_loss": 5.899426653664568, + "tokens_seen": 24838144 + }, + { + "epoch": 0.01, + "learning_rate": 0.000373015873015873, + "loss": 2.7599, + "theoretical_loss": 5.896852244092045, + "tokens_seen": 24903680 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003740079365079365, + "loss": 2.881, + "theoretical_loss": 5.894286491642708, + "tokens_seen": 24969216 + }, + { + "epoch": 0.01, + "learning_rate": 0.000375, + "loss": 2.7771, + "theoretical_loss": 5.891729344596959, + "tokens_seen": 25034752 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003759920634920635, + "loss": 2.7416, + "theoretical_loss": 5.8891807516785395, + "tokens_seen": 25100288 + }, + { + "epoch": 0.01, + "learning_rate": 0.000376984126984127, + "loss": 2.9635, + "theoretical_loss": 5.886640662049593, + "tokens_seen": 25165824 + }, + { + "epoch": 0.01, + "learning_rate": 0.00037797619047619046, + "loss": 2.7399, + "theoretical_loss": 5.884109025305776, + "tokens_seen": 25231360 + }, + { + "epoch": 0.01, + "learning_rate": 0.00037896825396825396, + "loss": 2.6505, + "theoretical_loss": 5.881585791471453, + "tokens_seen": 25296896 + }, + { + "epoch": 0.01, + "learning_rate": 0.00037996031746031746, + "loss": 2.8805, + "theoretical_loss": 5.879070910994954, + "tokens_seen": 25362432 + }, + { + "epoch": 0.01, + "learning_rate": 0.00038095238095238096, + "loss": 2.6802, + "theoretical_loss": 5.876564334743884, + "tokens_seen": 25427968 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003819444444444444, + "loss": 2.8124, + "theoretical_loss": 5.874066014000512, + "tokens_seen": 25493504 + }, + { + "epoch": 0.01, + "learning_rate": 0.00038293650793650797, + "loss": 2.7989, + "theoretical_loss": 5.871575900457216, + "tokens_seen": 25559040 + }, + { + "epoch": 0.01, + "learning_rate": 0.00038392857142857147, + "loss": 2.747, + "theoretical_loss": 5.86909394621199, + "tokens_seen": 25624576 + }, + { + "epoch": 0.01, + "learning_rate": 0.00038492063492063497, + "loss": 2.6896, + "theoretical_loss": 5.8666201037640064, + "tokens_seen": 25690112 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003859126984126984, + "loss": 2.8097, + "theoretical_loss": 5.864154326009246, + "tokens_seen": 25755648 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003869047619047619, + "loss": 2.6902, + "theoretical_loss": 5.861696566236189, + "tokens_seen": 25821184 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003878968253968254, + "loss": 2.8652, + "theoretical_loss": 5.859246778121547, + "tokens_seen": 25886720 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003888888888888889, + "loss": 2.8373, + "theoretical_loss": 5.856804915726074, + "tokens_seen": 25952256 + }, + { + "epoch": 0.01, + "learning_rate": 0.00038988095238095237, + "loss": 2.774, + "theoretical_loss": 5.854370933490414, + "tokens_seen": 26017792 + }, + { + "epoch": 0.01, + "learning_rate": 0.00039087301587301587, + "loss": 2.732, + "theoretical_loss": 5.851944786231018, + "tokens_seen": 26083328 + }, + { + "epoch": 0.01, + "learning_rate": 0.00039186507936507937, + "loss": 2.783, + "theoretical_loss": 5.849526429136105, + "tokens_seen": 26148864 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.005724027752876282, + "objective/train/docs_used": 27060, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.126623630523682, + "objective/train/original_loss": 5.126623630523682, + "objective/train/theoretical_loss": 5.847115817761683, + "objective/train/tokens_used": 46674400, + "objective/train/value_avg": -0.01268768310546875, + "objective/train/value_loss": 0.0002081326674669981, + "objective/train/value_max": -0.001972198486328125, + "objective/train/value_min": -0.12432861328125, + "objective/train/value_reward_corr": 0.00018191887065582835, + "objective/train/value_std": 0.0086212158203125, + "objective/train/weight_avg": 1.0679926872253418, + "objective/train/weighted_lm_loss": 5.504868984222412, + "objective/train/weights_max": 3.267685651779175, + "objective/train/weights_min": 0.2630084455013275, + "theoretical_loss": 5.847115817761683, + "tokens_seen": 26214400 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003928571428571429, + "loss": 2.6611, + "theoretical_loss": 5.847115817761683, + "tokens_seen": 26214400 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003938492063492063, + "loss": 2.6435, + "theoretical_loss": 5.844712908027621, + "tokens_seen": 26279936 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003948412698412698, + "loss": 2.7253, + "theoretical_loss": 5.842317656213765, + "tokens_seen": 26345472 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003958333333333333, + "loss": 2.8448, + "theoretical_loss": 5.83993001895612, + "tokens_seen": 26411008 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003968253968253968, + "loss": 2.7171, + "theoretical_loss": 5.83754995324307, + "tokens_seen": 26476544 + }, + { + "epoch": 0.01, + "learning_rate": 0.0003978174603174603, + "loss": 2.6854, + "theoretical_loss": 5.835177416411649, + "tokens_seen": 26542080 + }, + { + "epoch": 0.01, + "learning_rate": 0.00039880952380952383, + "loss": 2.6924, + "theoretical_loss": 5.8328123661438624, + "tokens_seen": 26607616 + }, + { + "epoch": 0.01, + "learning_rate": 0.00039980158730158733, + "loss": 2.855, + "theoretical_loss": 5.830454760463059, + "tokens_seen": 26673152 + }, + { + "epoch": 0.01, + "learning_rate": 0.00040079365079365083, + "loss": 2.6461, + "theoretical_loss": 5.828104557730348, + "tokens_seen": 26738688 + }, + { + "epoch": 0.01, + "learning_rate": 0.00040178571428571433, + "loss": 2.6315, + "theoretical_loss": 5.825761716641054, + "tokens_seen": 26804224 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004027777777777778, + "loss": 2.5866, + "theoretical_loss": 5.823426196221233, + "tokens_seen": 26869760 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004037698412698413, + "loss": 2.7586, + "theoretical_loss": 5.821097955824221, + "tokens_seen": 26935296 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004047619047619048, + "loss": 2.6385, + "theoretical_loss": 5.818776955127233, + "tokens_seen": 27000832 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004057539682539683, + "loss": 2.7534, + "theoretical_loss": 5.816463154128003, + "tokens_seen": 27066368 + }, + { + "epoch": 0.01, + "learning_rate": 0.00040674603174603173, + "loss": 2.6019, + "theoretical_loss": 5.814156513141467, + "tokens_seen": 27131904 + }, + { + "epoch": 0.01, + "learning_rate": 0.00040773809523809523, + "loss": 2.7691, + "theoretical_loss": 5.811856992796489, + "tokens_seen": 27197440 + }, + { + "epoch": 0.01, + "learning_rate": 0.00040873015873015874, + "loss": 2.8545, + "theoretical_loss": 5.809564554032628, + "tokens_seen": 27262976 + }, + { + "epoch": 0.01, + "learning_rate": 0.00040972222222222224, + "loss": 2.6061, + "theoretical_loss": 5.80727915809695, + "tokens_seen": 27328512 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004107142857142857, + "loss": 2.7078, + "theoretical_loss": 5.805000766540871, + "tokens_seen": 27394048 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004117063492063492, + "loss": 2.6524, + "theoretical_loss": 5.802729341217047, + "tokens_seen": 27459584 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004126984126984127, + "loss": 2.7397, + "theoretical_loss": 5.800464844276306, + "tokens_seen": 27525120 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004136904761904762, + "loss": 2.5529, + "theoretical_loss": 5.798207238164607, + "tokens_seen": 27590656 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004146825396825397, + "loss": 2.5932, + "theoretical_loss": 5.795956485620051, + "tokens_seen": 27656192 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004156746031746032, + "loss": 2.6782, + "theoretical_loss": 5.793712549669918, + "tokens_seen": 27721728 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004166666666666667, + "loss": 2.6544, + "theoretical_loss": 5.791475393627746, + "tokens_seen": 27787264 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.005918544251471758, + "objective/train/docs_used": 28453, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.874495506286621, + "objective/train/original_loss": 4.874495983123779, + "objective/train/theoretical_loss": 5.789244981090448, + "objective/train/tokens_used": 48312800, + "objective/train/value_avg": -0.00992584228515625, + "objective/train/value_loss": 0.0001469232956878841, + "objective/train/value_max": -0.001934051513671875, + "objective/train/value_min": -0.25537109375, + "objective/train/value_reward_corr": -0.020185511404128637, + "objective/train/value_std": 0.00778961181640625, + "objective/train/weight_avg": 1.0674448013305664, + "objective/train/weighted_lm_loss": 5.245066165924072, + "objective/train/weights_max": 12.854718208312988, + "objective/train/weights_min": 0.30690377950668335, + "theoretical_loss": 5.789244981090448, + "tokens_seen": 27852800 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004176587301587302, + "loss": 2.6495, + "theoretical_loss": 5.789244981090448, + "tokens_seen": 27852800 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004186507936507937, + "loss": 2.9286, + "theoretical_loss": 5.7870212759354605, + "tokens_seen": 27918336 + }, + { + "epoch": 0.01, + "learning_rate": 0.00041964285714285714, + "loss": 2.8268, + "theoretical_loss": 5.7848042423179304, + "tokens_seen": 27983872 + }, + { + "epoch": 0.01, + "learning_rate": 0.00042063492063492065, + "loss": 2.5442, + "theoretical_loss": 5.782593844667936, + "tokens_seen": 28049408 + }, + { + "epoch": 0.01, + "learning_rate": 0.00042162698412698415, + "loss": 2.5971, + "theoretical_loss": 5.780390047687737, + "tokens_seen": 28114944 + }, + { + "epoch": 0.01, + "learning_rate": 0.00042261904761904765, + "loss": 2.5259, + "theoretical_loss": 5.778192816349071, + "tokens_seen": 28180480 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004236111111111111, + "loss": 2.6342, + "theoretical_loss": 5.776002115890471, + "tokens_seen": 28246016 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004246031746031746, + "loss": 2.7628, + "theoretical_loss": 5.773817911814618, + "tokens_seen": 28311552 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004255952380952381, + "loss": 2.4708, + "theoretical_loss": 5.771640169885731, + "tokens_seen": 28377088 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004265873015873016, + "loss": 2.6787, + "theoretical_loss": 5.769468856126985, + "tokens_seen": 28442624 + }, + { + "epoch": 0.01, + "learning_rate": 0.00042757936507936505, + "loss": 2.6837, + "theoretical_loss": 5.76730393681796, + "tokens_seen": 28508160 + }, + { + "epoch": 0.01, + "learning_rate": 0.00042857142857142855, + "loss": 2.595, + "theoretical_loss": 5.765145378492123, + "tokens_seen": 28573696 + }, + { + "epoch": 0.01, + "learning_rate": 0.00042956349206349205, + "loss": 2.7707, + "theoretical_loss": 5.762993147934338, + "tokens_seen": 28639232 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004305555555555556, + "loss": 2.5189, + "theoretical_loss": 5.76084721217841, + "tokens_seen": 28704768 + }, + { + "epoch": 0.01, + "learning_rate": 0.00043154761904761905, + "loss": 2.8279, + "theoretical_loss": 5.7587075385046536, + "tokens_seen": 28770304 + }, + { + "epoch": 0.01, + "learning_rate": 0.00043253968253968256, + "loss": 2.7622, + "theoretical_loss": 5.756574094437491, + "tokens_seen": 28835840 + }, + { + "epoch": 0.01, + "learning_rate": 0.00043353174603174606, + "loss": 2.7448, + "theoretical_loss": 5.75444684774309, + "tokens_seen": 28901376 + }, + { + "epoch": 0.01, + "learning_rate": 0.00043452380952380956, + "loss": 2.4451, + "theoretical_loss": 5.752325766427008, + "tokens_seen": 28966912 + }, + { + "epoch": 0.01, + "learning_rate": 0.000435515873015873, + "loss": 2.8616, + "theoretical_loss": 5.750210818731889, + "tokens_seen": 29032448 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004365079365079365, + "loss": 2.8594, + "theoretical_loss": 5.748101973135171, + "tokens_seen": 29097984 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004375, + "loss": 2.6214, + "theoretical_loss": 5.745999198346823, + "tokens_seen": 29163520 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004384920634920635, + "loss": 2.6073, + "theoretical_loss": 5.743902463307119, + "tokens_seen": 29229056 + }, + { + "epoch": 0.01, + "learning_rate": 0.000439484126984127, + "loss": 2.7024, + "theoretical_loss": 5.7418117371844275, + "tokens_seen": 29294592 + }, + { + "epoch": 0.01, + "learning_rate": 0.00044047619047619046, + "loss": 2.7047, + "theoretical_loss": 5.739726989373027, + "tokens_seen": 29360128 + }, + { + "epoch": 0.01, + "learning_rate": 0.00044146825396825396, + "loss": 2.6527, + "theoretical_loss": 5.737648189490956, + "tokens_seen": 29425664 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.0021934083197265863, + "objective/train/docs_used": 29115, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.427516937255859, + "objective/train/original_loss": 5.427517414093018, + "objective/train/theoretical_loss": 5.735575307377884, + "objective/train/tokens_used": 49951200, + "objective/train/value_avg": -0.009613037109375, + "objective/train/value_loss": 0.00015654387243557721, + "objective/train/value_max": -0.001483917236328125, + "objective/train/value_min": -0.147705078125, + "objective/train/value_reward_corr": -0.06079807870890153, + "objective/train/value_std": 0.006984710693359375, + "objective/train/weight_avg": 1.029848575592041, + "objective/train/weighted_lm_loss": 5.588486671447754, + "objective/train/weights_max": 4.229508399963379, + "objective/train/weights_min": 0.45167794823646545, + "theoretical_loss": 5.735575307377884, + "tokens_seen": 29491200 + }, + { + "epoch": 0.01, + "learning_rate": 0.00044246031746031746, + "loss": 2.5274, + "theoretical_loss": 5.735575307377884, + "tokens_seen": 29491200 + }, + { + "epoch": 0.01, + "learning_rate": 0.00044345238095238096, + "loss": 2.743, + "theoretical_loss": 5.733508313093, + "tokens_seen": 29556736 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004444444444444444, + "loss": 2.7439, + "theoretical_loss": 5.731447176912937, + "tokens_seen": 29622272 + }, + { + "epoch": 0.01, + "learning_rate": 0.00044543650793650797, + "loss": 2.5585, + "theoretical_loss": 5.729391869329716, + "tokens_seen": 29687808 + }, + { + "epoch": 0.01, + "learning_rate": 0.00044642857142857147, + "loss": 2.7316, + "theoretical_loss": 5.727342361048713, + "tokens_seen": 29753344 + }, + { + "epoch": 0.01, + "learning_rate": 0.00044742063492063497, + "loss": 2.6802, + "theoretical_loss": 5.725298622986648, + "tokens_seen": 29818880 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004484126984126984, + "loss": 2.5962, + "theoretical_loss": 5.723260626269605, + "tokens_seen": 29884416 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004494047619047619, + "loss": 2.57, + "theoretical_loss": 5.721228342231065, + "tokens_seen": 29949952 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004503968253968254, + "loss": 2.746, + "theoretical_loss": 5.719201742409968, + "tokens_seen": 30015488 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004513888888888889, + "loss": 2.6077, + "theoretical_loss": 5.717180798548798, + "tokens_seen": 30081024 + }, + { + "epoch": 0.01, + "learning_rate": 0.00045238095238095237, + "loss": 2.8024, + "theoretical_loss": 5.715165482591681, + "tokens_seen": 30146560 + }, + { + "epoch": 0.01, + "learning_rate": 0.00045337301587301587, + "loss": 2.5172, + "theoretical_loss": 5.713155766682521, + "tokens_seen": 30212096 + }, + { + "epoch": 0.01, + "learning_rate": 0.00045436507936507937, + "loss": 2.3865, + "theoretical_loss": 5.71115162316314, + "tokens_seen": 30277632 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004553571428571429, + "loss": 2.6617, + "theoretical_loss": 5.709153024571448, + "tokens_seen": 30343168 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004563492063492063, + "loss": 2.6915, + "theoretical_loss": 5.707159943639638, + "tokens_seen": 30408704 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004573412698412698, + "loss": 2.5696, + "theoretical_loss": 5.705172353292392, + "tokens_seen": 30474240 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004583333333333333, + "loss": 2.5007, + "theoretical_loss": 5.703190226645112, + "tokens_seen": 30539776 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004593253968253968, + "loss": 2.7007, + "theoretical_loss": 5.70121353700217, + "tokens_seen": 30605312 + }, + { + "epoch": 0.01, + "learning_rate": 0.00046031746031746033, + "loss": 2.6077, + "theoretical_loss": 5.699242257855182, + "tokens_seen": 30670848 + }, + { + "epoch": 0.01, + "learning_rate": 0.00046130952380952383, + "loss": 2.6505, + "theoretical_loss": 5.697276362881294, + "tokens_seen": 30736384 + }, + { + "epoch": 0.01, + "learning_rate": 0.00046230158730158733, + "loss": 2.7648, + "theoretical_loss": 5.6953158259414955, + "tokens_seen": 30801920 + }, + { + "epoch": 0.01, + "learning_rate": 0.00046329365079365083, + "loss": 2.6878, + "theoretical_loss": 5.693360621078943, + "tokens_seen": 30867456 + }, + { + "epoch": 0.01, + "learning_rate": 0.00046428571428571433, + "loss": 2.6205, + "theoretical_loss": 5.691410722517309, + "tokens_seen": 30932992 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004652777777777778, + "loss": 2.6069, + "theoretical_loss": 5.689466104659148, + "tokens_seen": 30998528 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004662698412698413, + "loss": 2.5697, + "theoretical_loss": 5.6875267420842786, + "tokens_seen": 31064064 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.0045348345302045345, + "objective/train/docs_used": 30152, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.067525863647461, + "objective/train/original_loss": 5.067525386810303, + "objective/train/theoretical_loss": 5.685592609548186, + "objective/train/tokens_used": 51589600, + "objective/train/value_avg": -0.0102386474609375, + "objective/train/value_loss": 0.00015573075506836176, + "objective/train/value_max": -0.0015850067138671875, + "objective/train/value_min": -0.188232421875, + "objective/train/value_reward_corr": -0.026114925255221882, + "objective/train/value_std": 0.008575439453125, + "objective/train/weight_avg": 1.0536197423934937, + "objective/train/weighted_lm_loss": 5.37288761138916, + "objective/train/weights_max": 6.56875467300415, + "objective/train/weights_min": 0.08743437379598618, + "theoretical_loss": 5.685592609548186, + "tokens_seen": 31129600 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004672619047619048, + "loss": 2.5565, + "theoretical_loss": 5.685592609548186, + "tokens_seen": 31129600 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004682539682539683, + "loss": 2.5478, + "theoretical_loss": 5.683663681980443, + "tokens_seen": 31195136 + }, + { + "epoch": 0.01, + "learning_rate": 0.00046924603174603173, + "loss": 2.5461, + "theoretical_loss": 5.68173993448314, + "tokens_seen": 31260672 + }, + { + "epoch": 0.01, + "learning_rate": 0.00047023809523809523, + "loss": 2.5502, + "theoretical_loss": 5.679821342329351, + "tokens_seen": 31326208 + }, + { + "epoch": 0.01, + "learning_rate": 0.00047123015873015874, + "loss": 2.4969, + "theoretical_loss": 5.677907880961591, + "tokens_seen": 31391744 + }, + { + "epoch": 0.01, + "learning_rate": 0.00047222222222222224, + "loss": 2.536, + "theoretical_loss": 5.6759995259903135, + "tokens_seen": 31457280 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004732142857142857, + "loss": 2.5488, + "theoretical_loss": 5.674096253192413, + "tokens_seen": 31522816 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004742063492063492, + "loss": 2.4891, + "theoretical_loss": 5.672198038509743, + "tokens_seen": 31588352 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004751984126984127, + "loss": 2.6767, + "theoretical_loss": 5.670304858047658, + "tokens_seen": 31653888 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004761904761904762, + "loss": 2.6424, + "theoretical_loss": 5.66841668807356, + "tokens_seen": 31719424 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004771825396825397, + "loss": 2.5659, + "theoretical_loss": 5.666533505015475, + "tokens_seen": 31784960 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004781746031746032, + "loss": 2.5028, + "theoretical_loss": 5.664655285460632, + "tokens_seen": 31850496 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004791666666666667, + "loss": 2.6857, + "theoretical_loss": 5.662782006154064, + "tokens_seen": 31916032 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004801587301587302, + "loss": 2.5796, + "theoretical_loss": 5.660913643997225, + "tokens_seen": 31981568 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004811507936507937, + "loss": 2.469, + "theoretical_loss": 5.659050176046617, + "tokens_seen": 32047104 + }, + { + "epoch": 0.01, + "learning_rate": 0.00048214285714285715, + "loss": 2.5667, + "theoretical_loss": 5.657191579512437, + "tokens_seen": 32112640 + }, + { + "epoch": 0.01, + "learning_rate": 0.00048313492063492065, + "loss": 2.6954, + "theoretical_loss": 5.655337831757233, + "tokens_seen": 32178176 + }, + { + "epoch": 0.01, + "learning_rate": 0.00048412698412698415, + "loss": 2.5685, + "theoretical_loss": 5.653488910294585, + "tokens_seen": 32243712 + }, + { + "epoch": 0.01, + "learning_rate": 0.00048511904761904765, + "loss": 2.5485, + "theoretical_loss": 5.651644792787781, + "tokens_seen": 32309248 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004861111111111111, + "loss": 2.5313, + "theoretical_loss": 5.649805457048533, + "tokens_seen": 32374784 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004871031746031746, + "loss": 2.7141, + "theoretical_loss": 5.647970881035681, + "tokens_seen": 32440320 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004880952380952381, + "loss": 2.5846, + "theoretical_loss": 5.646141042853927, + "tokens_seen": 32505856 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004890873015873016, + "loss": 2.4726, + "theoretical_loss": 5.644315920752579, + "tokens_seen": 32571392 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004900793650793651, + "loss": 2.4042, + "theoretical_loss": 5.642495493124309, + "tokens_seen": 32636928 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004910714285714286, + "loss": 2.584, + "theoretical_loss": 5.640679738503916, + "tokens_seen": 32702464 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.004206159617751837, + "objective/train/docs_used": 30758, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.193688869476318, + "objective/train/original_loss": 5.193688869476318, + "objective/train/theoretical_loss": 5.638868635567113, + "objective/train/tokens_used": 53228000, + "objective/train/value_avg": -0.01041412353515625, + "objective/train/value_loss": 0.0001617267553228885, + "objective/train/value_max": -0.0016546249389648438, + "objective/train/value_min": -0.115966796875, + "objective/train/value_reward_corr": 0.07104848301984104, + "objective/train/value_std": 0.007595062255859375, + "objective/train/weight_avg": 1.0500991344451904, + "objective/train/weighted_lm_loss": 5.477589130401611, + "objective/train/weights_max": 3.1232264041900635, + "objective/train/weights_min": 0.3066580295562744, + "theoretical_loss": 5.638868635567113, + "tokens_seen": 32768000 + }, + { + "epoch": 0.01, + "learning_rate": 0.000492063492063492, + "loss": 2.6855, + "theoretical_loss": 5.638868635567113, + "tokens_seen": 32768000 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004930555555555556, + "loss": 2.61, + "theoretical_loss": 5.6370621631293245, + "tokens_seen": 32833536 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004940476190476191, + "loss": 2.4987, + "theoretical_loss": 5.635260300144488, + "tokens_seen": 32899072 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004950396825396826, + "loss": 2.511, + "theoretical_loss": 5.633463025703879, + "tokens_seen": 32964608 + }, + { + "epoch": 0.01, + "learning_rate": 0.000496031746031746, + "loss": 2.6181, + "theoretical_loss": 5.631670319034946, + "tokens_seen": 33030144 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004970238095238095, + "loss": 2.5711, + "theoretical_loss": 5.6298821595001485, + "tokens_seen": 33095680 + }, + { + "epoch": 0.01, + "learning_rate": 0.000498015873015873, + "loss": 2.5656, + "theoretical_loss": 5.628098526595821, + "tokens_seen": 33161216 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004990079365079365, + "loss": 2.5207, + "theoretical_loss": 5.626319399951038, + "tokens_seen": 33226752 + }, + { + "epoch": 0.01, + "learning_rate": 0.0005, + "loss": 2.5021, + "theoretical_loss": 5.624544759326499, + "tokens_seen": 33292288 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004999899699097292, + "loss": 2.6166, + "theoretical_loss": 5.622774584613412, + "tokens_seen": 33357824 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004999799398194584, + "loss": 2.4842, + "theoretical_loss": 5.621008855832411, + "tokens_seen": 33423360 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004999699097291876, + "loss": 2.5376, + "theoretical_loss": 5.619247553132459, + "tokens_seen": 33488896 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004999598796389167, + "loss": 2.5029, + "theoretical_loss": 5.61749065678978, + "tokens_seen": 33554432 + }, + { + "epoch": 0.01, + "learning_rate": 0.000499949849548646, + "loss": 2.6695, + "theoretical_loss": 5.615738147206796, + "tokens_seen": 33619968 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004999398194583751, + "loss": 2.5515, + "theoretical_loss": 5.613990004911081, + "tokens_seen": 33685504 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004999297893681044, + "loss": 2.5566, + "theoretical_loss": 5.612246210554302, + "tokens_seen": 33751040 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004999197592778335, + "loss": 2.4623, + "theoretical_loss": 5.610506744911213, + "tokens_seen": 33816576 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004999097291875627, + "loss": 2.6104, + "theoretical_loss": 5.608771588878618, + "tokens_seen": 33882112 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004998996990972919, + "loss": 2.6496, + "theoretical_loss": 5.60704072347437, + "tokens_seen": 33947648 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004998896690070211, + "loss": 2.6862, + "theoretical_loss": 5.605314129836366, + "tokens_seen": 34013184 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004998796389167503, + "loss": 2.5419, + "theoretical_loss": 5.603591789221566, + "tokens_seen": 34078720 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004998696088264795, + "loss": 2.6103, + "theoretical_loss": 5.601873683005008, + "tokens_seen": 34144256 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004998595787362087, + "loss": 2.5221, + "theoretical_loss": 5.600159792678844, + "tokens_seen": 34209792 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004998495486459378, + "loss": 2.5686, + "theoretical_loss": 5.598450099851373, + "tokens_seen": 34275328 + }, + { + "epoch": 0.01, + "learning_rate": 0.000499839518555667, + "loss": 2.6023, + "theoretical_loss": 5.596744586246102, + "tokens_seen": 34340864 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.003202206688001752, + "objective/train/docs_used": 31355, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.528264999389648, + "objective/train/original_loss": 4.528265476226807, + "objective/train/theoretical_loss": 5.595043233700797, + "objective/train/tokens_used": 54866400, + "objective/train/value_avg": -0.0086822509765625, + "objective/train/value_loss": 0.00011574759992072359, + "objective/train/value_max": -0.0014324188232421875, + "objective/train/value_min": -0.1380615234375, + "objective/train/value_reward_corr": 0.015298859315155532, + "objective/train/value_std": 0.00677490234375, + "objective/train/weight_avg": 1.0381278991699219, + "objective/train/weighted_lm_loss": 4.707478046417236, + "objective/train/weights_max": 3.9192001819610596, + "objective/train/weights_min": 0.325858473777771, + "theoretical_loss": 5.595043233700797, + "tokens_seen": 34406400 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004998294884653962, + "loss": 2.5203, + "theoretical_loss": 5.595043233700797, + "tokens_seen": 34406400 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004998194583751254, + "loss": 2.5837, + "theoretical_loss": 5.59334602416656, + "tokens_seen": 34471936 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004998094282848546, + "loss": 2.6541, + "theoretical_loss": 5.591652939706902, + "tokens_seen": 34537472 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004997993981945837, + "loss": 2.6202, + "theoretical_loss": 5.589963962496837, + "tokens_seen": 34603008 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004997893681043129, + "loss": 2.4839, + "theoretical_loss": 5.588279074821972, + "tokens_seen": 34668544 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004997793380140421, + "loss": 2.5916, + "theoretical_loss": 5.586598259077619, + "tokens_seen": 34734080 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004997693079237714, + "loss": 2.4203, + "theoretical_loss": 5.58492149776791, + "tokens_seen": 34799616 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004997592778335005, + "loss": 2.5574, + "theoretical_loss": 5.583248773504918, + "tokens_seen": 34865152 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004997492477432298, + "loss": 2.6054, + "theoretical_loss": 5.581580069007791, + "tokens_seen": 34930688 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004997392176529588, + "loss": 2.4771, + "theoretical_loss": 5.5799153671018935, + "tokens_seen": 34996224 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004997291875626881, + "loss": 2.555, + "theoretical_loss": 5.578254650717955, + "tokens_seen": 35061760 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004997191574724173, + "loss": 2.6089, + "theoretical_loss": 5.576597902891232, + "tokens_seen": 35127296 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004997091273821465, + "loss": 2.4991, + "theoretical_loss": 5.574945106760667, + "tokens_seen": 35192832 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004996990972918757, + "loss": 2.4969, + "theoretical_loss": 5.573296245568071, + "tokens_seen": 35258368 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004996890672016048, + "loss": 2.5365, + "theoretical_loss": 5.571651302657301, + "tokens_seen": 35323904 + }, + { + "epoch": 0.01, + "learning_rate": 0.000499679037111334, + "loss": 2.5066, + "theoretical_loss": 5.570010261473453, + "tokens_seen": 35389440 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004996690070210632, + "loss": 2.4213, + "theoretical_loss": 5.568373105562058, + "tokens_seen": 35454976 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004996589769307924, + "loss": 2.4546, + "theoretical_loss": 5.566739818568294, + "tokens_seen": 35520512 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004996489468405216, + "loss": 2.3983, + "theoretical_loss": 5.565110384236193, + "tokens_seen": 35586048 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004996389167502507, + "loss": 2.5415, + "theoretical_loss": 5.563484786407864, + "tokens_seen": 35651584 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004996288866599799, + "loss": 2.4172, + "theoretical_loss": 5.56186300902273, + "tokens_seen": 35717120 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004996188565697091, + "loss": 2.4423, + "theoretical_loss": 5.560245036116753, + "tokens_seen": 35782656 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004996088264794383, + "loss": 2.587, + "theoretical_loss": 5.55863085182169, + "tokens_seen": 35848192 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004995987963891675, + "loss": 2.6093, + "theoretical_loss": 5.557020440364331, + "tokens_seen": 35913728 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004995887662988968, + "loss": 2.3846, + "theoretical_loss": 5.555413786065771, + "tokens_seen": 35979264 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.0038662729784846306, + "objective/train/docs_used": 32385, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.818417072296143, + "objective/train/original_loss": 4.818417072296143, + "objective/train/theoretical_loss": 5.553810873340668, + "objective/train/tokens_used": 56504800, + "objective/train/value_avg": -0.00962066650390625, + "objective/train/value_loss": 0.00017477427900303155, + "objective/train/value_max": -0.0016040802001953125, + "objective/train/value_min": -0.35546875, + "objective/train/value_reward_corr": 0.051374766104316356, + "objective/train/value_std": 0.00939178466796875, + "objective/train/weight_avg": 1.0509169101715088, + "objective/train/weighted_lm_loss": 5.054445743560791, + "objective/train/weights_max": 30.86699676513672, + "objective/train/weights_min": 0.37638944387435913, + "theoretical_loss": 5.553810873340668, + "tokens_seen": 36044800 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004995787362086258, + "loss": 2.5454, + "theoretical_loss": 5.553810873340668, + "tokens_seen": 36044800 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004995687061183551, + "loss": 2.5003, + "theoretical_loss": 5.5522116866965145, + "tokens_seen": 36110336 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004995586760280842, + "loss": 2.5923, + "theoretical_loss": 5.550616210732926, + "tokens_seen": 36175872 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004995486459378135, + "loss": 2.5456, + "theoretical_loss": 5.549024430140915, + "tokens_seen": 36241408 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004995386158475427, + "loss": 2.3233, + "theoretical_loss": 5.547436329702196, + "tokens_seen": 36306944 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004995285857572718, + "loss": 2.65, + "theoretical_loss": 5.545851894288477, + "tokens_seen": 36372480 + }, + { + "epoch": 0.01, + "learning_rate": 0.000499518555667001, + "loss": 2.7075, + "theoretical_loss": 5.544271108860773, + "tokens_seen": 36438016 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004995085255767302, + "loss": 2.3954, + "theoretical_loss": 5.542693958468716, + "tokens_seen": 36503552 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004994984954864594, + "loss": 2.4595, + "theoretical_loss": 5.5411204282498705, + "tokens_seen": 36569088 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004994884653961886, + "loss": 2.4779, + "theoretical_loss": 5.539550503429072, + "tokens_seen": 36634624 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004994784353059178, + "loss": 2.5359, + "theoretical_loss": 5.537984169317745, + "tokens_seen": 36700160 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004994684052156469, + "loss": 2.3554, + "theoretical_loss": 5.536421411313254, + "tokens_seen": 36765696 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004994583751253761, + "loss": 2.4614, + "theoretical_loss": 5.534862214898242, + "tokens_seen": 36831232 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004994483450351053, + "loss": 2.4647, + "theoretical_loss": 5.533306565639981, + "tokens_seen": 36896768 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004994383149448345, + "loss": 2.3798, + "theoretical_loss": 5.531754449189732, + "tokens_seen": 36962304 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004994282848545637, + "loss": 2.4839, + "theoretical_loss": 5.530205851282113, + "tokens_seen": 37027840 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004994182547642928, + "loss": 2.4372, + "theoretical_loss": 5.528660757734458, + "tokens_seen": 37093376 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004994082246740221, + "loss": 2.4969, + "theoretical_loss": 5.527119154446201, + "tokens_seen": 37158912 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004993981945837512, + "loss": 2.4326, + "theoretical_loss": 5.525581027398248, + "tokens_seen": 37224448 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004993881644934805, + "loss": 2.4605, + "theoretical_loss": 5.524046362652378, + "tokens_seen": 37289984 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004993781344032096, + "loss": 2.514, + "theoretical_loss": 5.522515146350619, + "tokens_seen": 37355520 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004993681043129389, + "loss": 2.4401, + "theoretical_loss": 5.520987364714657, + "tokens_seen": 37421056 + }, + { + "epoch": 0.01, + "learning_rate": 0.000499358074222668, + "loss": 2.3969, + "theoretical_loss": 5.519463004045237, + "tokens_seen": 37486592 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004993480441323972, + "loss": 2.5104, + "theoretical_loss": 5.517942050721569, + "tokens_seen": 37552128 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004993380140421264, + "loss": 2.353, + "theoretical_loss": 5.5164244912007465, + "tokens_seen": 37617664 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.0029271587263792753, + "objective/train/docs_used": 32883, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 5.02232551574707, + "objective/train/original_loss": 5.02232551574707, + "objective/train/theoretical_loss": 5.514910312017166, + "objective/train/tokens_used": 58143200, + "objective/train/value_avg": -0.00873565673828125, + "objective/train/value_loss": 0.00011130507482448593, + "objective/train/value_max": -0.0011510848999023438, + "objective/train/value_min": -0.1195068359375, + "objective/train/value_reward_corr": 0.029837154417291595, + "objective/train/value_std": 0.007205963134765625, + "objective/train/weight_avg": 1.0351099967956543, + "objective/train/weighted_lm_loss": 5.210063934326172, + "objective/train/weights_max": 2.788703680038452, + "objective/train/weights_min": 0.39119651913642883, + "theoretical_loss": 5.514910312017166, + "tokens_seen": 37683200 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004993279839518556, + "loss": 2.514, + "theoretical_loss": 5.514910312017166, + "tokens_seen": 37683200 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004993179538615848, + "loss": 2.4852, + "theoretical_loss": 5.513399499781949, + "tokens_seen": 37748736 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004993079237713139, + "loss": 2.5185, + "theoretical_loss": 5.511892041182376, + "tokens_seen": 37814272 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004992978936810431, + "loss": 2.5971, + "theoretical_loss": 5.5103879229813195, + "tokens_seen": 37879808 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004992878635907723, + "loss": 2.5341, + "theoretical_loss": 5.508887132016688, + "tokens_seen": 37945344 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004992778335005015, + "loss": 2.5937, + "theoretical_loss": 5.507389655200869, + "tokens_seen": 38010880 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004992678034102307, + "loss": 2.5162, + "theoretical_loss": 5.505895479520182, + "tokens_seen": 38076416 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004992577733199598, + "loss": 2.4711, + "theoretical_loss": 5.504404592034333, + "tokens_seen": 38141952 + }, + { + "epoch": 0.01, + "learning_rate": 0.000499247743229689, + "loss": 2.4462, + "theoretical_loss": 5.5029169798758755, + "tokens_seen": 38207488 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004992377131394183, + "loss": 2.4484, + "theoretical_loss": 5.5014326302496785, + "tokens_seen": 38273024 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004992276830491475, + "loss": 2.3546, + "theoretical_loss": 5.499951530432394, + "tokens_seen": 38338560 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004992176529588767, + "loss": 2.4719, + "theoretical_loss": 5.498473667771936, + "tokens_seen": 38404096 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004992076228686059, + "loss": 2.4214, + "theoretical_loss": 5.496999029686955, + "tokens_seen": 38469632 + }, + { + "epoch": 0.01, + "learning_rate": 0.000499197592778335, + "loss": 2.5629, + "theoretical_loss": 5.4955276036663285, + "tokens_seen": 38535168 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004991875626880642, + "loss": 2.5706, + "theoretical_loss": 5.494059377268647, + "tokens_seen": 38600704 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004991775325977934, + "loss": 2.3813, + "theoretical_loss": 5.492594338121711, + "tokens_seen": 38666240 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004991675025075226, + "loss": 2.5128, + "theoretical_loss": 5.4911324739220255, + "tokens_seen": 38731776 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004991574724172518, + "loss": 2.4006, + "theoretical_loss": 5.4896737724343065, + "tokens_seen": 38797312 + }, + { + "epoch": 0.01, + "learning_rate": 0.000499147442326981, + "loss": 2.4849, + "theoretical_loss": 5.488218221490987, + "tokens_seen": 38862848 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004991374122367101, + "loss": 2.5404, + "theoretical_loss": 5.486765808991725, + "tokens_seen": 38928384 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004991273821464393, + "loss": 2.5796, + "theoretical_loss": 5.48531652290293, + "tokens_seen": 38993920 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004991173520561685, + "loss": 2.4334, + "theoretical_loss": 5.483870351257271, + "tokens_seen": 39059456 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004991073219658977, + "loss": 2.5248, + "theoretical_loss": 5.482427282153209, + "tokens_seen": 39124992 + }, + { + "epoch": 0.01, + "learning_rate": 0.000499097291875627, + "loss": 2.4328, + "theoretical_loss": 5.480987303754524, + "tokens_seen": 39190528 + }, + { + "epoch": 0.01, + "learning_rate": 0.000499087261785356, + "loss": 2.4091, + "theoretical_loss": 5.479550404289846, + "tokens_seen": 39256064 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.0012067611096426845, + "objective/train/docs_used": 34166, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.663817882537842, + "objective/train/original_loss": 4.663817882537842, + "objective/train/theoretical_loss": 5.478116572052198, + "objective/train/tokens_used": 59781600, + "objective/train/value_avg": -0.008941650390625, + "objective/train/value_loss": 0.0001277915434911847, + "objective/train/value_max": -0.001201629638671875, + "objective/train/value_min": -0.1295166015625, + "objective/train/value_reward_corr": 0.10170990635605719, + "objective/train/value_std": 0.00713348388671875, + "objective/train/weight_avg": 1.0184701681137085, + "objective/train/weighted_lm_loss": 4.779131889343262, + "objective/train/weights_max": 3.1694703102111816, + "objective/train/weights_min": 0.2001967877149582, + "theoretical_loss": 5.478116572052198, + "tokens_seen": 39321600 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004990772316950853, + "loss": 2.5146, + "theoretical_loss": 5.478116572052198, + "tokens_seen": 39321600 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004990672016048144, + "loss": 2.5263, + "theoretical_loss": 5.476685795398528, + "tokens_seen": 39387136 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004990571715145437, + "loss": 2.345, + "theoretical_loss": 5.475258062749262, + "tokens_seen": 39452672 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004990471414242729, + "loss": 2.3256, + "theoretical_loss": 5.4738333625878495, + "tokens_seen": 39518208 + }, + { + "epoch": 0.01, + "learning_rate": 0.000499037111334002, + "loss": 2.3995, + "theoretical_loss": 5.472411683460317, + "tokens_seen": 39583744 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004990270812437312, + "loss": 2.3539, + "theoretical_loss": 5.470993013974829, + "tokens_seen": 39649280 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004990170511534604, + "loss": 2.6222, + "theoretical_loss": 5.469577342801242, + "tokens_seen": 39714816 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004990070210631896, + "loss": 2.5134, + "theoretical_loss": 5.468164658670672, + "tokens_seen": 39780352 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004989969909729188, + "loss": 2.4578, + "theoretical_loss": 5.466754950375066, + "tokens_seen": 39845888 + }, + { + "epoch": 0.01, + "learning_rate": 0.000498986960882648, + "loss": 2.563, + "theoretical_loss": 5.465348206766773, + "tokens_seen": 39911424 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004989769307923771, + "loss": 2.4059, + "theoretical_loss": 5.463944416758115, + "tokens_seen": 39976960 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004989669007021063, + "loss": 2.4066, + "theoretical_loss": 5.4625435693209745, + "tokens_seen": 40042496 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004989568706118355, + "loss": 2.4844, + "theoretical_loss": 5.461145653486371, + "tokens_seen": 40108032 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004989468405215647, + "loss": 2.565, + "theoretical_loss": 5.459750658344051, + "tokens_seen": 40173568 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004989368104312939, + "loss": 2.505, + "theoretical_loss": 5.45835857304208, + "tokens_seen": 40239104 + }, + { + "epoch": 0.01, + "learning_rate": 0.000498926780341023, + "loss": 2.4042, + "theoretical_loss": 5.456969386786434, + "tokens_seen": 40304640 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004989167502507523, + "loss": 2.3886, + "theoretical_loss": 5.455583088840594, + "tokens_seen": 40370176 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004989067201604814, + "loss": 2.4299, + "theoretical_loss": 5.454199668525155, + "tokens_seen": 40435712 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004988966900702107, + "loss": 2.4683, + "theoretical_loss": 5.452819115217421, + "tokens_seen": 40501248 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004988866599799398, + "loss": 2.4971, + "theoretical_loss": 5.4514414183510205, + "tokens_seen": 40566784 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004988766298896691, + "loss": 2.1796, + "theoretical_loss": 5.4500665674155115, + "tokens_seen": 40632320 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004988665997993982, + "loss": 2.5129, + "theoretical_loss": 5.448694551956001, + "tokens_seen": 40697856 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004988565697091274, + "loss": 2.3502, + "theoretical_loss": 5.447325361572755, + "tokens_seen": 40763392 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004988465396188566, + "loss": 2.3441, + "theoretical_loss": 5.445958985920832, + "tokens_seen": 40828928 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004988365095285858, + "loss": 2.4401, + "theoretical_loss": 5.444595414709691, + "tokens_seen": 40894464 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.0014040537644177675, + "objective/train/docs_used": 34866, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.626951217651367, + "objective/train/original_loss": 4.626951217651367, + "objective/train/theoretical_loss": 5.443234637702833, + "objective/train/tokens_used": 61420000, + "objective/train/value_avg": -0.00870513916015625, + "objective/train/value_loss": 0.0001555603666929528, + "objective/train/value_max": -0.0014047622680664062, + "objective/train/value_min": -0.12188720703125, + "objective/train/value_reward_corr": 0.07283570578976965, + "objective/train/value_std": 0.006694793701171875, + "objective/train/weight_avg": 1.0214276313781738, + "objective/train/weighted_lm_loss": 4.684600830078125, + "objective/train/weights_max": 3.1468701362609863, + "objective/train/weights_min": 0.34594273567199707, + "theoretical_loss": 5.443234637702833, + "tokens_seen": 40960000 + }, + { + "epoch": 0.01, + "learning_rate": 0.000498826479438315, + "loss": 2.3616, + "theoretical_loss": 5.443234637702833, + "tokens_seen": 40960000 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004988164493480441, + "loss": 2.4622, + "theoretical_loss": 5.441876644717422, + "tokens_seen": 41025536 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004988064192577733, + "loss": 2.4564, + "theoretical_loss": 5.440521425623924, + "tokens_seen": 41091072 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004987963891675025, + "loss": 2.4592, + "theoretical_loss": 5.439168970345737, + "tokens_seen": 41156608 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004987863590772317, + "loss": 2.5733, + "theoretical_loss": 5.4378192688588385, + "tokens_seen": 41222144 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004987763289869609, + "loss": 2.3775, + "theoretical_loss": 5.436472311191425, + "tokens_seen": 41287680 + }, + { + "epoch": 0.01, + "learning_rate": 0.00049876629889669, + "loss": 2.4665, + "theoretical_loss": 5.435128087423554, + "tokens_seen": 41353216 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004987562688064192, + "loss": 2.4815, + "theoretical_loss": 5.433786587686798, + "tokens_seen": 41418752 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004987462387161484, + "loss": 2.4147, + "theoretical_loss": 5.432447802163894, + "tokens_seen": 41484288 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004987362086258777, + "loss": 2.4442, + "theoretical_loss": 5.431111721088394, + "tokens_seen": 41549824 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004987261785356068, + "loss": 2.4783, + "theoretical_loss": 5.429778334744331, + "tokens_seen": 41615360 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004987161484453361, + "loss": 2.4246, + "theoretical_loss": 5.428447633465867, + "tokens_seen": 41680896 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004987061183550651, + "loss": 2.4961, + "theoretical_loss": 5.427119607636971, + "tokens_seen": 41746432 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004986960882647944, + "loss": 2.428, + "theoretical_loss": 5.42579424769107, + "tokens_seen": 41811968 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004986860581745236, + "loss": 2.4896, + "theoretical_loss": 5.424471544110728, + "tokens_seen": 41877504 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004986760280842528, + "loss": 2.3841, + "theoretical_loss": 5.423151487427312, + "tokens_seen": 41943040 + }, + { + "epoch": 0.01, + "learning_rate": 0.000498665997993982, + "loss": 2.3076, + "theoretical_loss": 5.421834068220666, + "tokens_seen": 42008576 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004986559679037111, + "loss": 2.2999, + "theoretical_loss": 5.420519277118791, + "tokens_seen": 42074112 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004986459378134403, + "loss": 2.4086, + "theoretical_loss": 5.419207104797523, + "tokens_seen": 42139648 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004986359077231695, + "loss": 2.4895, + "theoretical_loss": 5.41789754198021, + "tokens_seen": 42205184 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004986258776328987, + "loss": 2.3738, + "theoretical_loss": 5.416590579437406, + "tokens_seen": 42270720 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004986158475426279, + "loss": 2.4476, + "theoretical_loss": 5.415286207986547, + "tokens_seen": 42336256 + }, + { + "epoch": 0.01, + "learning_rate": 0.000498605817452357, + "loss": 2.3335, + "theoretical_loss": 5.413984418491651, + "tokens_seen": 42401792 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004985957873620862, + "loss": 2.3659, + "theoretical_loss": 5.412685201863004, + "tokens_seen": 42467328 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004985857572718154, + "loss": 2.3632, + "theoretical_loss": 5.411388549056857, + "tokens_seen": 42532864 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.00220158277079463, + "objective/train/docs_used": 35868, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.510434627532959, + "objective/train/original_loss": 4.510434150695801, + "objective/train/theoretical_loss": 5.410094451075121, + "objective/train/tokens_used": 63058400, + "objective/train/value_avg": -0.007495880126953125, + "objective/train/value_loss": 0.0001072565937647596, + "objective/train/value_max": -0.0008692741394042969, + "objective/train/value_min": -0.1114501953125, + "objective/train/value_reward_corr": 0.06575674663814889, + "objective/train/value_std": 0.005878448486328125, + "objective/train/weight_avg": 1.0272542238235474, + "objective/train/weighted_lm_loss": 4.645576477050781, + "objective/train/weights_max": 2.796630859375, + "objective/train/weights_min": 0.3165469169616699, + "theoretical_loss": 5.410094451075121, + "tokens_seen": 42598400 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004985757271815446, + "loss": 2.3484, + "theoretical_loss": 5.410094451075121, + "tokens_seen": 42598400 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004985656970912738, + "loss": 2.4515, + "theoretical_loss": 5.408802898965068, + "tokens_seen": 42663936 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004985556670010031, + "loss": 2.4695, + "theoretical_loss": 5.4075138838190355, + "tokens_seen": 42729472 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004985456369107321, + "loss": 2.4215, + "theoretical_loss": 5.406227396774128, + "tokens_seen": 42795008 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004985356068204614, + "loss": 2.3548, + "theoretical_loss": 5.4049434290119205, + "tokens_seen": 42860544 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004985255767301905, + "loss": 2.4084, + "theoretical_loss": 5.4036619717581775, + "tokens_seen": 42926080 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004985155466399198, + "loss": 2.4093, + "theoretical_loss": 5.402383016282556, + "tokens_seen": 42991616 + }, + { + "epoch": 0.01, + "learning_rate": 0.000498505516549649, + "loss": 2.1951, + "theoretical_loss": 5.401106553898321, + "tokens_seen": 43057152 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004984954864593782, + "loss": 2.5143, + "theoretical_loss": 5.3998325759620664, + "tokens_seen": 43122688 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004984854563691073, + "loss": 2.3791, + "theoretical_loss": 5.3985610738734255, + "tokens_seen": 43188224 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004984754262788365, + "loss": 2.1752, + "theoretical_loss": 5.397292039074801, + "tokens_seen": 43253760 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004984653961885657, + "loss": 2.3829, + "theoretical_loss": 5.396025463051079, + "tokens_seen": 43319296 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004984553660982949, + "loss": 2.3989, + "theoretical_loss": 5.394761337329362, + "tokens_seen": 43384832 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004984453360080241, + "loss": 2.5157, + "theoretical_loss": 5.3934996534786865, + "tokens_seen": 43450368 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004984353059177532, + "loss": 2.4185, + "theoretical_loss": 5.392240403109765, + "tokens_seen": 43515904 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004984252758274825, + "loss": 2.4498, + "theoretical_loss": 5.390983577874708, + "tokens_seen": 43581440 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004984152457372116, + "loss": 2.4127, + "theoretical_loss": 5.3897291694667615, + "tokens_seen": 43646976 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004984052156469409, + "loss": 2.3951, + "theoretical_loss": 5.388477169620044, + "tokens_seen": 43712512 + }, + { + "epoch": 0.01, + "learning_rate": 0.00049839518555667, + "loss": 2.4236, + "theoretical_loss": 5.38722757010928, + "tokens_seen": 43778048 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004983851554663993, + "loss": 2.3756, + "theoretical_loss": 5.385980362749544, + "tokens_seen": 43843584 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004983751253761284, + "loss": 2.3156, + "theoretical_loss": 5.384735539396003, + "tokens_seen": 43909120 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004983650952858576, + "loss": 2.2923, + "theoretical_loss": 5.383493091943657, + "tokens_seen": 43974656 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004983550651955868, + "loss": 2.2889, + "theoretical_loss": 5.382253012327088, + "tokens_seen": 44040192 + }, + { + "epoch": 0.01, + "learning_rate": 0.000498345035105316, + "loss": 2.4222, + "theoretical_loss": 5.3810152925202095, + "tokens_seen": 44105728 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004983350050150452, + "loss": 2.2468, + "theoretical_loss": 5.379779924536011, + "tokens_seen": 44171264 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.00193874747492373, + "objective/train/docs_used": 36544, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.517274379730225, + "objective/train/original_loss": 4.517274856567383, + "objective/train/theoretical_loss": 5.3785469004263184, + "objective/train/tokens_used": 64696800, + "objective/train/value_avg": -0.00818634033203125, + "objective/train/value_loss": 0.00012367957970127463, + "objective/train/value_max": -0.0010280609130859375, + "objective/train/value_min": -0.09600830078125, + "objective/train/value_reward_corr": 0.002103036062236297, + "objective/train/value_std": 0.006175994873046875, + "objective/train/weight_avg": 1.025285243988037, + "objective/train/weighted_lm_loss": 4.571944713592529, + "objective/train/weights_max": 2.6119132041931152, + "objective/train/weights_min": 0.1411619335412979, + "theoretical_loss": 5.3785469004263184, + "tokens_seen": 44236800 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004983249749247743, + "loss": 2.3819, + "theoretical_loss": 5.3785469004263184, + "tokens_seen": 44236800 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004983149448345035, + "loss": 2.2184, + "theoretical_loss": 5.377316212281538, + "tokens_seen": 44302336 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004983049147442327, + "loss": 2.3705, + "theoretical_loss": 5.3760878522304205, + "tokens_seen": 44367872 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004982948846539619, + "loss": 2.3599, + "theoretical_loss": 5.374861812439816, + "tokens_seen": 44433408 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004982848545636911, + "loss": 2.2354, + "theoretical_loss": 5.373638085114435, + "tokens_seen": 44498944 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004982748244734202, + "loss": 2.3574, + "theoretical_loss": 5.372416662496606, + "tokens_seen": 44564480 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004982647943831494, + "loss": 2.4422, + "theoretical_loss": 5.371197536866044, + "tokens_seen": 44630016 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004982547642928786, + "loss": 2.3968, + "theoretical_loss": 5.36998070053961, + "tokens_seen": 44695552 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004982447342026079, + "loss": 2.5014, + "theoretical_loss": 5.368766145871087, + "tokens_seen": 44761088 + }, + { + "epoch": 0.01, + "learning_rate": 0.000498234704112337, + "loss": 2.3415, + "theoretical_loss": 5.367553865250939, + "tokens_seen": 44826624 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004982246740220663, + "loss": 2.4239, + "theoretical_loss": 5.3663438511060875, + "tokens_seen": 44892160 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004982146439317953, + "loss": 2.4924, + "theoretical_loss": 5.3651360958996825, + "tokens_seen": 44957696 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004982046138415246, + "loss": 2.459, + "theoretical_loss": 5.3639305921308775, + "tokens_seen": 45023232 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004981945837512538, + "loss": 2.3695, + "theoretical_loss": 5.362727332334602, + "tokens_seen": 45088768 + }, + { + "epoch": 0.01, + "learning_rate": 0.000498184553660983, + "loss": 2.3397, + "theoretical_loss": 5.361526309081345, + "tokens_seen": 45154304 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004981745235707122, + "loss": 2.2312, + "theoretical_loss": 5.3603275149769285, + "tokens_seen": 45219840 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004981644934804413, + "loss": 2.4281, + "theoretical_loss": 5.3591309426622935, + "tokens_seen": 45285376 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004981544633901705, + "loss": 2.278, + "theoretical_loss": 5.357936584813277, + "tokens_seen": 45350912 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004981444332998997, + "loss": 2.3345, + "theoretical_loss": 5.3567444341404045, + "tokens_seen": 45416448 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004981344032096289, + "loss": 2.4156, + "theoretical_loss": 5.355554483388666, + "tokens_seen": 45481984 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004981243731193581, + "loss": 2.4196, + "theoretical_loss": 5.354366725337309, + "tokens_seen": 45547520 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004981143430290873, + "loss": 2.3122, + "theoretical_loss": 5.353181152799634, + "tokens_seen": 45613056 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004981043129388164, + "loss": 2.3156, + "theoretical_loss": 5.351997758622769, + "tokens_seen": 45678592 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004980942828485456, + "loss": 2.4585, + "theoretical_loss": 5.35081653568748, + "tokens_seen": 45744128 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004980842527582748, + "loss": 2.2808, + "theoretical_loss": 5.3496374769079535, + "tokens_seen": 45809664 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": -0.0002522535214666277, + "objective/train/docs_used": 37915, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.8409218788146973, + "objective/train/original_loss": 3.8409221172332764, + "objective/train/theoretical_loss": 5.348460575231594, + "objective/train/tokens_used": 66335200, + "objective/train/value_avg": -0.006229400634765625, + "objective/train/value_loss": 0.00010480458877282217, + "objective/train/value_max": -0.0007495880126953125, + "objective/train/value_min": -0.1602783203125, + "objective/train/value_reward_corr": 0.08717547091912564, + "objective/train/value_std": 0.00487518310546875, + "objective/train/weight_avg": 1.00237238407135, + "objective/train/weighted_lm_loss": 3.8486862182617188, + "objective/train/weights_max": 4.966837406158447, + "objective/train/weights_min": 0.24555516242980957, + "theoretical_loss": 5.348460575231594, + "tokens_seen": 45875200 + }, + { + "epoch": 0.01, + "learning_rate": 0.000498074222668004, + "loss": 2.3003, + "theoretical_loss": 5.348460575231594, + "tokens_seen": 45875200 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004980641925777333, + "loss": 2.2703, + "theoretical_loss": 5.347285823638824, + "tokens_seen": 45940736 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004980541624874623, + "loss": 2.3953, + "theoretical_loss": 5.3461132151428785, + "tokens_seen": 46006272 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004980441323971916, + "loss": 2.4035, + "theoretical_loss": 5.344942742789613, + "tokens_seen": 46071808 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004980341023069207, + "loss": 2.3835, + "theoretical_loss": 5.343774399657292, + "tokens_seen": 46137344 + }, + { + "epoch": 0.01, + "learning_rate": 0.00049802407221665, + "loss": 2.3915, + "theoretical_loss": 5.342608178856405, + "tokens_seen": 46202880 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004980140421263792, + "loss": 2.3899, + "theoretical_loss": 5.341444073529464, + "tokens_seen": 46268416 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004980040120361084, + "loss": 2.4363, + "theoretical_loss": 5.340282076850814, + "tokens_seen": 46333952 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004979939819458375, + "loss": 2.3573, + "theoretical_loss": 5.339122182026433, + "tokens_seen": 46399488 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004979839518555667, + "loss": 2.3063, + "theoretical_loss": 5.337964382293748, + "tokens_seen": 46465024 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004979739217652959, + "loss": 2.2889, + "theoretical_loss": 5.336808670921445, + "tokens_seen": 46530560 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004979638916750251, + "loss": 2.289, + "theoretical_loss": 5.335655041209275, + "tokens_seen": 46596096 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004979538615847543, + "loss": 2.3458, + "theoretical_loss": 5.3345034864878755, + "tokens_seen": 46661632 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004979438314944834, + "loss": 2.3284, + "theoretical_loss": 5.333354000118574, + "tokens_seen": 46727168 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004979338014042126, + "loss": 2.4645, + "theoretical_loss": 5.332206575493219, + "tokens_seen": 46792704 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004979237713139418, + "loss": 2.2713, + "theoretical_loss": 5.33106120603398, + "tokens_seen": 46858240 + }, + { + "epoch": 0.01, + "learning_rate": 0.000497913741223671, + "loss": 2.3906, + "theoretical_loss": 5.329917885193181, + "tokens_seen": 46923776 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004979037111334002, + "loss": 2.2787, + "theoretical_loss": 5.328776606453111, + "tokens_seen": 46989312 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004978936810431293, + "loss": 2.3474, + "theoretical_loss": 5.327637363325849, + "tokens_seen": 47054848 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004978836509528586, + "loss": 2.278, + "theoretical_loss": 5.326500149353086, + "tokens_seen": 47120384 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004978736208625877, + "loss": 2.2791, + "theoretical_loss": 5.325364958105951, + "tokens_seen": 47185920 + }, + { + "epoch": 0.01, + "learning_rate": 0.000497863590772317, + "loss": 2.3718, + "theoretical_loss": 5.324231783184828, + "tokens_seen": 47251456 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004978535606820461, + "loss": 2.3515, + "theoretical_loss": 5.323100618219199, + "tokens_seen": 47316992 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004978435305917754, + "loss": 2.374, + "theoretical_loss": 5.32197145686745, + "tokens_seen": 47382528 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004978335005015045, + "loss": 2.4021, + "theoretical_loss": 5.3208442928167194, + "tokens_seen": 47448064 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.0027397058438509703, + "objective/train/docs_used": 38651, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.269894123077393, + "objective/train/original_loss": 4.269894123077393, + "objective/train/theoretical_loss": 5.319719119782715, + "objective/train/tokens_used": 67973600, + "objective/train/value_avg": -0.00809478759765625, + "objective/train/value_loss": 0.00010214724170509726, + "objective/train/value_max": -0.0009255409240722656, + "objective/train/value_min": -0.095703125, + "objective/train/value_reward_corr": 0.11476559680043805, + "objective/train/value_std": 0.005313873291015625, + "objective/train/weight_avg": 1.032437801361084, + "objective/train/weighted_lm_loss": 4.404292106628418, + "objective/train/weights_max": 2.433824062347412, + "objective/train/weights_min": 0.30578190088272095, + "theoretical_loss": 5.319719119782715, + "tokens_seen": 47513600 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004978234704112337, + "loss": 2.3064, + "theoretical_loss": 5.319719119782715, + "tokens_seen": 47513600 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004978134403209629, + "loss": 2.2631, + "theoretical_loss": 5.318595931509554, + "tokens_seen": 47579136 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004978034102306921, + "loss": 2.3259, + "theoretical_loss": 5.317474721769591, + "tokens_seen": 47644672 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004977933801404213, + "loss": 2.2942, + "theoretical_loss": 5.3163554843632514, + "tokens_seen": 47710208 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004977833500501504, + "loss": 2.3123, + "theoretical_loss": 5.31523821311887, + "tokens_seen": 47775744 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004977733199598796, + "loss": 2.3822, + "theoretical_loss": 5.314122901892526, + "tokens_seen": 47841280 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004977632898696088, + "loss": 2.3694, + "theoretical_loss": 5.31300954456788, + "tokens_seen": 47906816 + }, + { + "epoch": 0.01, + "learning_rate": 0.000497753259779338, + "loss": 2.3108, + "theoretical_loss": 5.31189813505601, + "tokens_seen": 47972352 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004977432296890672, + "loss": 2.3412, + "theoretical_loss": 5.310788667295256, + "tokens_seen": 48037888 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004977331995987965, + "loss": 2.3349, + "theoretical_loss": 5.3096811352510604, + "tokens_seen": 48103424 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004977231695085255, + "loss": 2.4182, + "theoretical_loss": 5.308575532915805, + "tokens_seen": 48168960 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004977131394182548, + "loss": 2.3131, + "theoretical_loss": 5.307471854308661, + "tokens_seen": 48234496 + }, + { + "epoch": 0.01, + "learning_rate": 0.000497703109327984, + "loss": 2.2801, + "theoretical_loss": 5.306370093475426, + "tokens_seen": 48300032 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004976930792377132, + "loss": 2.4255, + "theoretical_loss": 5.305270244488379, + "tokens_seen": 48365568 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004976830491474424, + "loss": 2.3931, + "theoretical_loss": 5.304172301446117, + "tokens_seen": 48431104 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004976730190571715, + "loss": 2.2598, + "theoretical_loss": 5.303076258473412, + "tokens_seen": 48496640 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004976629889669007, + "loss": 2.2705, + "theoretical_loss": 5.30198210972105, + "tokens_seen": 48562176 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004976529588766299, + "loss": 2.324, + "theoretical_loss": 5.300889849365689, + "tokens_seen": 48627712 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004976429287863591, + "loss": 2.263, + "theoretical_loss": 5.299799471609704, + "tokens_seen": 48693248 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004976328986960883, + "loss": 2.2814, + "theoretical_loss": 5.298710970681048, + "tokens_seen": 48758784 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004976228686058175, + "loss": 2.2259, + "theoretical_loss": 5.2976243408330905, + "tokens_seen": 48824320 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004976128385155466, + "loss": 2.1668, + "theoretical_loss": 5.296539576344487, + "tokens_seen": 48889856 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004976028084252758, + "loss": 2.3059, + "theoretical_loss": 5.295456671519022, + "tokens_seen": 48955392 + }, + { + "epoch": 0.01, + "learning_rate": 0.000497592778335005, + "loss": 2.3199, + "theoretical_loss": 5.294375620685475, + "tokens_seen": 49020928 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004975827482447342, + "loss": 2.2698, + "theoretical_loss": 5.293296418197466, + "tokens_seen": 49086464 + }, + { + "epoch": 0.01, + "objective/train/advantage_avg": 0.0017358529148623347, + "objective/train/docs_used": 39958, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.235271453857422, + "objective/train/original_loss": 4.23527193069458, + "objective/train/theoretical_loss": 5.292219058433327, + "objective/train/tokens_used": 69612000, + "objective/train/value_avg": -0.0081329345703125, + "objective/train/value_loss": 9.017179399961606e-05, + "objective/train/value_max": -0.0009183883666992188, + "objective/train/value_min": -0.1312255859375, + "objective/train/value_reward_corr": 0.06389590762855538, + "objective/train/value_std": 0.006954193115234375, + "objective/train/weight_avg": 1.0221257209777832, + "objective/train/weighted_lm_loss": 4.359708786010742, + "objective/train/weights_max": 3.221693277359009, + "objective/train/weights_min": 0.5517849922180176, + "theoretical_loss": 5.292219058433327, + "tokens_seen": 49152000 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004975727181544635, + "loss": 2.1813, + "theoretical_loss": 5.292219058433327, + "tokens_seen": 49152000 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004975626880641925, + "loss": 2.2712, + "theoretical_loss": 5.29114353579595, + "tokens_seen": 49217536 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004975526579739218, + "loss": 2.1758, + "theoretical_loss": 5.290069844712654, + "tokens_seen": 49283072 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004975426278836509, + "loss": 2.2662, + "theoretical_loss": 5.2889979796350435, + "tokens_seen": 49348608 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004975325977933802, + "loss": 2.0871, + "theoretical_loss": 5.28792793503887, + "tokens_seen": 49414144 + }, + { + "epoch": 0.01, + "learning_rate": 0.0004975225677031094, + "loss": 2.2468, + "theoretical_loss": 5.286859705423897, + "tokens_seen": 49479680 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004975125376128386, + "loss": 2.3282, + "theoretical_loss": 5.285793285313759, + "tokens_seen": 49545216 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004975025075225677, + "loss": 2.1041, + "theoretical_loss": 5.284728669255836, + "tokens_seen": 49610752 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004974924774322969, + "loss": 2.4015, + "theoretical_loss": 5.283665851821111, + "tokens_seen": 49676288 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004974824473420261, + "loss": 2.2362, + "theoretical_loss": 5.282604827604038, + "tokens_seen": 49741824 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004974724172517553, + "loss": 2.1816, + "theoretical_loss": 5.2815455912224145, + "tokens_seen": 49807360 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004974623871614845, + "loss": 2.3054, + "theoretical_loss": 5.280488137317242, + "tokens_seen": 49872896 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004974523570712136, + "loss": 2.3092, + "theoretical_loss": 5.2794324605526075, + "tokens_seen": 49938432 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004974423269809428, + "loss": 2.3566, + "theoretical_loss": 5.278378555615541, + "tokens_seen": 50003968 + }, + { + "epoch": 0.02, + "learning_rate": 0.000497432296890672, + "loss": 2.2204, + "theoretical_loss": 5.277326417215896, + "tokens_seen": 50069504 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004974222668004012, + "loss": 2.2423, + "theoretical_loss": 5.276276040086216, + "tokens_seen": 50135040 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004974122367101304, + "loss": 2.2391, + "theoretical_loss": 5.275227418981615, + "tokens_seen": 50200576 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004974022066198595, + "loss": 2.2748, + "theoretical_loss": 5.27418054867964, + "tokens_seen": 50266112 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004973921765295888, + "loss": 2.3384, + "theoretical_loss": 5.273135423980159, + "tokens_seen": 50331648 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004973821464393179, + "loss": 2.341, + "theoretical_loss": 5.272092039705226, + "tokens_seen": 50397184 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004973721163490472, + "loss": 2.241, + "theoretical_loss": 5.271050390698967, + "tokens_seen": 50462720 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004973620862587763, + "loss": 2.305, + "theoretical_loss": 5.270010471827446, + "tokens_seen": 50528256 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004973520561685056, + "loss": 2.2668, + "theoretical_loss": 5.268972277978554, + "tokens_seen": 50593792 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004973420260782347, + "loss": 2.0843, + "theoretical_loss": 5.267935804061883, + "tokens_seen": 50659328 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004973319959879639, + "loss": 2.3984, + "theoretical_loss": 5.266901045008607, + "tokens_seen": 50724864 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": 0.0009306073188781738, + "objective/train/docs_used": 40584, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.072620868682861, + "objective/train/original_loss": 4.072620868682861, + "objective/train/theoretical_loss": 5.265867995771362, + "objective/train/tokens_used": 71250400, + "objective/train/value_avg": -0.00673675537109375, + "objective/train/value_loss": 0.00012664221867453307, + "objective/train/value_max": -0.000720977783203125, + "objective/train/value_min": -0.1815185546875, + "objective/train/value_reward_corr": 0.07069674699527387, + "objective/train/value_std": 0.0055999755859375, + "objective/train/weight_avg": 1.0154167413711548, + "objective/train/weighted_lm_loss": 4.133606910705566, + "objective/train/weights_max": 6.142215728759766, + "objective/train/weights_min": 0.45227837562561035, + "theoretical_loss": 5.265867995771362, + "tokens_seen": 50790400 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004973219658976931, + "loss": 2.2748, + "theoretical_loss": 5.265867995771362, + "tokens_seen": 50790400 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004973119358074223, + "loss": 2.1413, + "theoretical_loss": 5.26483665132413, + "tokens_seen": 50855936 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004973019057171515, + "loss": 2.3044, + "theoretical_loss": 5.263807006662114, + "tokens_seen": 50921472 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004972918756268806, + "loss": 2.2692, + "theoretical_loss": 5.2627790568016355, + "tokens_seen": 50987008 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004972818455366098, + "loss": 2.2808, + "theoretical_loss": 5.2617527967800015, + "tokens_seen": 51052544 + }, + { + "epoch": 0.02, + "learning_rate": 0.000497271815446339, + "loss": 2.2647, + "theoretical_loss": 5.260728221655401, + "tokens_seen": 51118080 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004972617853560682, + "loss": 2.2796, + "theoretical_loss": 5.259705326506787, + "tokens_seen": 51183616 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004972517552657974, + "loss": 2.2909, + "theoretical_loss": 5.258684106433764, + "tokens_seen": 51249152 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004972417251755266, + "loss": 2.3133, + "theoretical_loss": 5.257664556556469, + "tokens_seen": 51314688 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004972316950852557, + "loss": 2.3273, + "theoretical_loss": 5.256646672015468, + "tokens_seen": 51380224 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004972216649949849, + "loss": 2.1321, + "theoretical_loss": 5.255630447971641, + "tokens_seen": 51445760 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004972116349047142, + "loss": 2.3838, + "theoretical_loss": 5.25461587960607, + "tokens_seen": 51511296 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004972016048144433, + "loss": 2.3111, + "theoretical_loss": 5.253602962119929, + "tokens_seen": 51576832 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004971915747241726, + "loss": 2.3913, + "theoretical_loss": 5.252591690734379, + "tokens_seen": 51642368 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004971815446339017, + "loss": 2.1086, + "theoretical_loss": 5.251582060690451, + "tokens_seen": 51707904 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004971715145436309, + "loss": 2.2566, + "theoretical_loss": 5.250574067248951, + "tokens_seen": 51773440 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004971614844533601, + "loss": 2.3992, + "theoretical_loss": 5.249567705690339, + "tokens_seen": 51838976 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004971514543630893, + "loss": 2.0297, + "theoretical_loss": 5.248562971314636, + "tokens_seen": 51904512 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004971414242728185, + "loss": 2.3023, + "theoretical_loss": 5.247559859441305, + "tokens_seen": 51970048 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004971313941825477, + "loss": 2.3871, + "theoretical_loss": 5.2465583654091565, + "tokens_seen": 52035584 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004971213640922768, + "loss": 2.248, + "theoretical_loss": 5.245558484576243, + "tokens_seen": 52101120 + }, + { + "epoch": 0.02, + "learning_rate": 0.000497111334002006, + "loss": 2.3411, + "theoretical_loss": 5.244560212319749, + "tokens_seen": 52166656 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004971013039117352, + "loss": 2.1473, + "theoretical_loss": 5.243563544035896, + "tokens_seen": 52232192 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004970912738214644, + "loss": 2.3245, + "theoretical_loss": 5.242568475139834, + "tokens_seen": 52297728 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004970812437311936, + "loss": 2.3228, + "theoretical_loss": 5.241575001065544, + "tokens_seen": 52363264 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": 0.002423921599984169, + "objective/train/docs_used": 41941, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.341817855834961, + "objective/train/original_loss": 4.341817855834961, + "objective/train/theoretical_loss": 5.240583117265738, + "objective/train/tokens_used": 72888800, + "objective/train/value_avg": -0.007778167724609375, + "objective/train/value_loss": 9.976723958970979e-05, + "objective/train/value_max": -0.0006189346313476562, + "objective/train/value_min": -0.15966796875, + "objective/train/value_reward_corr": 0.030618838302674332, + "objective/train/value_std": 0.006420135498046875, + "objective/train/weight_avg": 1.0293971300125122, + "objective/train/weighted_lm_loss": 4.438074588775635, + "objective/train/weights_max": 4.062808513641357, + "objective/train/weights_min": 0.5450382232666016, + "theoretical_loss": 5.240583117265738, + "tokens_seen": 52428800 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004970712136409227, + "loss": 2.2909, + "theoretical_loss": 5.240583117265738, + "tokens_seen": 52428800 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004970611835506519, + "loss": 2.2643, + "theoretical_loss": 5.239592819211753, + "tokens_seen": 52494336 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004970511534603811, + "loss": 2.314, + "theoretical_loss": 5.238604102393458, + "tokens_seen": 52559872 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004970411233701103, + "loss": 2.2524, + "theoretical_loss": 5.237616962319154, + "tokens_seen": 52625408 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004970310932798396, + "loss": 2.4115, + "theoretical_loss": 5.236631394515473, + "tokens_seen": 52690944 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004970210631895686, + "loss": 2.3361, + "theoretical_loss": 5.235647394527281, + "tokens_seen": 52756480 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004970110330992979, + "loss": 2.2186, + "theoretical_loss": 5.234664957917584, + "tokens_seen": 52822016 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004970010030090271, + "loss": 2.0735, + "theoretical_loss": 5.2336840802674285, + "tokens_seen": 52887552 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004969909729187563, + "loss": 2.1809, + "theoretical_loss": 5.232704757175809, + "tokens_seen": 52953088 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004969809428284855, + "loss": 2.2622, + "theoretical_loss": 5.23172698425957, + "tokens_seen": 53018624 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004969709127382147, + "loss": 2.2487, + "theoretical_loss": 5.230750757153314, + "tokens_seen": 53084160 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004969608826479438, + "loss": 2.3059, + "theoretical_loss": 5.2297760715093045, + "tokens_seen": 53149696 + }, + { + "epoch": 0.02, + "learning_rate": 0.000496950852557673, + "loss": 2.2362, + "theoretical_loss": 5.228802922997378, + "tokens_seen": 53215232 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004969408224674022, + "loss": 2.215, + "theoretical_loss": 5.2278313073048475, + "tokens_seen": 53280768 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004969307923771314, + "loss": 2.2039, + "theoretical_loss": 5.22686122013641, + "tokens_seen": 53346304 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004969207622868606, + "loss": 2.2288, + "theoretical_loss": 5.22589265721406, + "tokens_seen": 53411840 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004969107321965897, + "loss": 2.2475, + "theoretical_loss": 5.224925614276991, + "tokens_seen": 53477376 + }, + { + "epoch": 0.02, + "learning_rate": 0.000496900702106319, + "loss": 2.2066, + "theoretical_loss": 5.223960087081515, + "tokens_seen": 53542912 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004968906720160481, + "loss": 2.1998, + "theoretical_loss": 5.222996071400964, + "tokens_seen": 53608448 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004968806419257774, + "loss": 2.2382, + "theoretical_loss": 5.2220335630256045, + "tokens_seen": 53673984 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004968706118355065, + "loss": 2.245, + "theoretical_loss": 5.221072557762554, + "tokens_seen": 53739520 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004968605817452358, + "loss": 2.2, + "theoretical_loss": 5.220113051435683, + "tokens_seen": 53805056 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004968505516549649, + "loss": 2.2938, + "theoretical_loss": 5.2191550398855355, + "tokens_seen": 53870592 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004968405215646941, + "loss": 2.1647, + "theoretical_loss": 5.21819851896924, + "tokens_seen": 53936128 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004968304914744233, + "loss": 2.3471, + "theoretical_loss": 5.2172434845604245, + "tokens_seen": 54001664 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": 0.002176029374822974, + "objective/train/docs_used": 42708, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.5303802490234375, + "objective/train/original_loss": 4.5303802490234375, + "objective/train/theoretical_loss": 5.2162899325491265, + "objective/train/tokens_used": 74527200, + "objective/train/value_avg": -0.0077362060546875, + "objective/train/value_loss": 0.00010403218766441569, + "objective/train/value_max": -0.0007014274597167969, + "objective/train/value_min": -0.1124267578125, + "objective/train/value_reward_corr": 0.10983025069171316, + "objective/train/value_std": 0.00557708740234375, + "objective/train/weight_avg": 1.0267761945724487, + "objective/train/weighted_lm_loss": 4.657341957092285, + "objective/train/weights_max": 3.0779616832733154, + "objective/train/weights_min": 0.3532910645008087, + "theoretical_loss": 5.2162899325491265, + "tokens_seen": 54067200 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004968204613841525, + "loss": 2.2053, + "theoretical_loss": 5.2162899325491265, + "tokens_seen": 54067200 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004968104312938817, + "loss": 2.2757, + "theoretical_loss": 5.215337858841714, + "tokens_seen": 54132736 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004968004012036108, + "loss": 2.1055, + "theoretical_loss": 5.214387259360796, + "tokens_seen": 54198272 + }, + { + "epoch": 0.02, + "learning_rate": 0.00049679037111334, + "loss": 2.1475, + "theoretical_loss": 5.213438130045141, + "tokens_seen": 54263808 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004967803410230692, + "loss": 2.1352, + "theoretical_loss": 5.212490466849594, + "tokens_seen": 54329344 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004967703109327984, + "loss": 2.0766, + "theoretical_loss": 5.211544265744993, + "tokens_seen": 54394880 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004967602808425276, + "loss": 2.2423, + "theoretical_loss": 5.210599522718084, + "tokens_seen": 54460416 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004967502507522568, + "loss": 2.1714, + "theoretical_loss": 5.209656233771442, + "tokens_seen": 54525952 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004967402206619859, + "loss": 2.1936, + "theoretical_loss": 5.2087143949233905, + "tokens_seen": 54591488 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004967301905717151, + "loss": 2.164, + "theoretical_loss": 5.207774002207918, + "tokens_seen": 54657024 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004967201604814444, + "loss": 2.1947, + "theoretical_loss": 5.206835051674599, + "tokens_seen": 54722560 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004967101303911735, + "loss": 2.0924, + "theoretical_loss": 5.205897539388513, + "tokens_seen": 54788096 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004967001003009028, + "loss": 2.1245, + "theoretical_loss": 5.204961461430168, + "tokens_seen": 54853632 + }, + { + "epoch": 0.02, + "learning_rate": 0.000496690070210632, + "loss": 2.3155, + "theoretical_loss": 5.2040268138954175, + "tokens_seen": 54919168 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004966800401203611, + "loss": 2.0884, + "theoretical_loss": 5.203093592895387, + "tokens_seen": 54984704 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004966700100300903, + "loss": 2.1577, + "theoretical_loss": 5.2021617945563925, + "tokens_seen": 55050240 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004966599799398195, + "loss": 2.1993, + "theoretical_loss": 5.201231415019864, + "tokens_seen": 55115776 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004966499498495487, + "loss": 2.1692, + "theoretical_loss": 5.200302450442269, + "tokens_seen": 55181312 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004966399197592779, + "loss": 2.2502, + "theoretical_loss": 5.199374896995037, + "tokens_seen": 55246848 + }, + { + "epoch": 0.02, + "learning_rate": 0.000496629889669007, + "loss": 2.108, + "theoretical_loss": 5.198448750864481, + "tokens_seen": 55312384 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004966198595787362, + "loss": 2.2342, + "theoretical_loss": 5.197524008251724, + "tokens_seen": 55377920 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004966098294884654, + "loss": 2.2533, + "theoretical_loss": 5.196600665372626, + "tokens_seen": 55443456 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004965997993981946, + "loss": 2.2116, + "theoretical_loss": 5.195678718457707, + "tokens_seen": 55508992 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004965897693079238, + "loss": 2.144, + "theoretical_loss": 5.194758163752068, + "tokens_seen": 55574528 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004965797392176529, + "loss": 2.1672, + "theoretical_loss": 5.19383899751533, + "tokens_seen": 55640064 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": 0.0006768309394828975, + "objective/train/docs_used": 44116, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.823684215545654, + "objective/train/original_loss": 4.823685169219971, + "objective/train/theoretical_loss": 5.192921216021549, + "objective/train/tokens_used": 76165600, + "objective/train/value_avg": -0.007602691650390625, + "objective/train/value_loss": 0.00014043629926163703, + "objective/train/value_max": -0.0006513595581054688, + "objective/train/value_min": -0.1165771484375, + "objective/train/value_reward_corr": 0.058363122371924814, + "objective/train/value_std": 0.006298065185546875, + "objective/train/weight_avg": 1.0133771896362305, + "objective/train/weighted_lm_loss": 4.899722576141357, + "objective/train/weights_max": 3.208397150039673, + "objective/train/weights_min": 0.29270219802856445, + "theoretical_loss": 5.192921216021549, + "tokens_seen": 55705600 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004965697091273821, + "loss": 2.2717, + "theoretical_loss": 5.192921216021549, + "tokens_seen": 55705600 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004965596790371113, + "loss": 2.0406, + "theoretical_loss": 5.19200481555915, + "tokens_seen": 55771136 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004965496489468405, + "loss": 1.9606, + "theoretical_loss": 5.19108979243085, + "tokens_seen": 55836672 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004965396188565698, + "loss": 2.1817, + "theoretical_loss": 5.190176142953593, + "tokens_seen": 55902208 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004965295887662988, + "loss": 2.1996, + "theoretical_loss": 5.189263863458471, + "tokens_seen": 55967744 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004965195586760281, + "loss": 2.0844, + "theoretical_loss": 5.188352950290662, + "tokens_seen": 56033280 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004965095285857573, + "loss": 2.0202, + "theoretical_loss": 5.1874433998093465, + "tokens_seen": 56098816 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004964994984954865, + "loss": 1.9207, + "theoretical_loss": 5.186535208387652, + "tokens_seen": 56164352 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004964894684052157, + "loss": 2.287, + "theoretical_loss": 5.185628372412573, + "tokens_seen": 56229888 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004964794383149449, + "loss": 2.242, + "theoretical_loss": 5.1847228882849095, + "tokens_seen": 56295424 + }, + { + "epoch": 0.02, + "learning_rate": 0.000496469408224674, + "loss": 2.3428, + "theoretical_loss": 5.1838187524191905, + "tokens_seen": 56360960 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004964593781344032, + "loss": 2.3375, + "theoretical_loss": 5.18291596124361, + "tokens_seen": 56426496 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004964493480441324, + "loss": 2.0213, + "theoretical_loss": 5.182014511199963, + "tokens_seen": 56492032 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004964393179538616, + "loss": 2.1669, + "theoretical_loss": 5.181114398743569, + "tokens_seen": 56557568 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004964292878635908, + "loss": 2.0801, + "theoretical_loss": 5.180215620343211, + "tokens_seen": 56623104 + }, + { + "epoch": 0.02, + "learning_rate": 0.00049641925777332, + "loss": 2.2375, + "theoretical_loss": 5.17931817248107, + "tokens_seen": 56688640 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004964092276830491, + "loss": 2.2198, + "theoretical_loss": 5.178422051652653, + "tokens_seen": 56754176 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004963991975927783, + "loss": 2.1657, + "theoretical_loss": 5.1775272543667334, + "tokens_seen": 56819712 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004963891675025075, + "loss": 2.1958, + "theoretical_loss": 5.17663377714528, + "tokens_seen": 56885248 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004963791374122367, + "loss": 2.2096, + "theoretical_loss": 5.175741616523398, + "tokens_seen": 56950784 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004963691073219659, + "loss": 2.0955, + "theoretical_loss": 5.174850769049258, + "tokens_seen": 57016320 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004963590772316951, + "loss": 2.1146, + "theoretical_loss": 5.173961231284037, + "tokens_seen": 57081856 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004963490471414242, + "loss": 2.1363, + "theoretical_loss": 5.1730729998018505, + "tokens_seen": 57147392 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004963390170511535, + "loss": 2.0983, + "theoretical_loss": 5.172186071189692, + "tokens_seen": 57212928 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004963289869608827, + "loss": 2.203, + "theoretical_loss": 5.171300442047371, + "tokens_seen": 57278464 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": -0.00013127195416018367, + "objective/train/docs_used": 44698, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.8118560314178467, + "objective/train/original_loss": 3.8118555545806885, + "objective/train/theoretical_loss": 5.1704161089874425, + "objective/train/tokens_used": 77804000, + "objective/train/value_avg": -0.006008148193359375, + "objective/train/value_loss": 9.51085748965852e-05, + "objective/train/value_max": -0.0006771087646484375, + "objective/train/value_min": -0.1572265625, + "objective/train/value_reward_corr": 0.01816564425008857, + "objective/train/value_std": 0.0049285888671875, + "objective/train/weight_avg": 1.0034019947052002, + "objective/train/weighted_lm_loss": 3.8111071586608887, + "objective/train/weights_max": 4.8175506591796875, + "objective/train/weights_min": 0.5587560534477234, + "theoretical_loss": 5.1704161089874425, + "tokens_seen": 57344000 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004963189568706119, + "loss": 2.114, + "theoretical_loss": 5.1704161089874425, + "tokens_seen": 57344000 + }, + { + "epoch": 0.02, + "learning_rate": 0.000496308926780341, + "loss": 2.0951, + "theoretical_loss": 5.169533068635155, + "tokens_seen": 57409536 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004962988966900702, + "loss": 2.2772, + "theoretical_loss": 5.168651317628383, + "tokens_seen": 57475072 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004962888665997994, + "loss": 2.2551, + "theoretical_loss": 5.167770852617565, + "tokens_seen": 57540608 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004962788365095286, + "loss": 2.1538, + "theoretical_loss": 5.1668916702656436, + "tokens_seen": 57606144 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004962688064192578, + "loss": 2.1677, + "theoretical_loss": 5.166013767248007, + "tokens_seen": 57671680 + }, + { + "epoch": 0.02, + "learning_rate": 0.000496258776328987, + "loss": 2.2163, + "theoretical_loss": 5.165137140252424, + "tokens_seen": 57737216 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004962487462387161, + "loss": 2.1594, + "theoretical_loss": 5.1642617859789866, + "tokens_seen": 57802752 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004962387161484453, + "loss": 2.408, + "theoretical_loss": 5.1633877011400475, + "tokens_seen": 57868288 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004962286860581746, + "loss": 2.1474, + "theoretical_loss": 5.162514882460167, + "tokens_seen": 57933824 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004962186559679037, + "loss": 2.2789, + "theoretical_loss": 5.161643326676048, + "tokens_seen": 57999360 + }, + { + "epoch": 0.02, + "learning_rate": 0.000496208625877633, + "loss": 2.1252, + "theoretical_loss": 5.160773030536475, + "tokens_seen": 58064896 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004961985957873621, + "loss": 2.1514, + "theoretical_loss": 5.15990399080227, + "tokens_seen": 58130432 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004961885656970913, + "loss": 2.1003, + "theoretical_loss": 5.159036204246215, + "tokens_seen": 58195968 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004961785356068205, + "loss": 2.1791, + "theoretical_loss": 5.158169667653005, + "tokens_seen": 58261504 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004961685055165497, + "loss": 2.1312, + "theoretical_loss": 5.1573043778191945, + "tokens_seen": 58327040 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004961584754262789, + "loss": 2.0207, + "theoretical_loss": 5.1564403315531315, + "tokens_seen": 58392576 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004961484453360081, + "loss": 2.2571, + "theoretical_loss": 5.155577525674904, + "tokens_seen": 58458112 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004961384152457372, + "loss": 2.2307, + "theoretical_loss": 5.15471595701629, + "tokens_seen": 58523648 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004961283851554664, + "loss": 2.3032, + "theoretical_loss": 5.153855622420689, + "tokens_seen": 58589184 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004961183550651956, + "loss": 2.1856, + "theoretical_loss": 5.152996518743078, + "tokens_seen": 58654720 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004961083249749248, + "loss": 2.1942, + "theoretical_loss": 5.152138642849951, + "tokens_seen": 58720256 + }, + { + "epoch": 0.02, + "learning_rate": 0.000496098294884654, + "loss": 2.144, + "theoretical_loss": 5.151281991619264, + "tokens_seen": 58785792 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004960882647943831, + "loss": 2.2079, + "theoretical_loss": 5.150426561940381, + "tokens_seen": 58851328 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004960782347041123, + "loss": 2.1367, + "theoretical_loss": 5.1495723507140205, + "tokens_seen": 58916864 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": 0.001023741438984871, + "objective/train/docs_used": 46033, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4973127841949463, + "objective/train/original_loss": 3.4973127841949463, + "objective/train/theoretical_loss": 5.148719354852201, + "objective/train/tokens_used": 79442400, + "objective/train/value_avg": -0.00733184814453125, + "objective/train/value_loss": 0.00013510156713891774, + "objective/train/value_max": -0.0005593299865722656, + "objective/train/value_min": -0.186279296875, + "objective/train/value_reward_corr": 0.05250465352359981, + "objective/train/value_std": 0.006847381591796875, + "objective/train/weight_avg": 1.0172332525253296, + "objective/train/weighted_lm_loss": 3.5470492839813232, + "objective/train/weights_max": 6.023562431335449, + "objective/train/weights_min": 0.555488646030426, + "theoretical_loss": 5.148719354852201, + "tokens_seen": 58982400 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004960682046138415, + "loss": 1.9608, + "theoretical_loss": 5.148719354852201, + "tokens_seen": 58982400 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004960581745235707, + "loss": 2.1725, + "theoretical_loss": 5.147867571278184, + "tokens_seen": 59047936 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004960481444333, + "loss": 2.2291, + "theoretical_loss": 5.147016996926427, + "tokens_seen": 59113472 + }, + { + "epoch": 0.02, + "learning_rate": 0.000496038114343029, + "loss": 2.1139, + "theoretical_loss": 5.146167628742525, + "tokens_seen": 59179008 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004960280842527583, + "loss": 2.1364, + "theoretical_loss": 5.145319463683162, + "tokens_seen": 59244544 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004960180541624875, + "loss": 2.2416, + "theoretical_loss": 5.144472498716053, + "tokens_seen": 59310080 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004960080240722167, + "loss": 2.094, + "theoretical_loss": 5.143626730819898, + "tokens_seen": 59375616 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004959979939819459, + "loss": 2.0751, + "theoretical_loss": 5.142782156984327, + "tokens_seen": 59441152 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004959879638916751, + "loss": 2.0604, + "theoretical_loss": 5.141938774209849, + "tokens_seen": 59506688 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004959779338014042, + "loss": 2.2093, + "theoretical_loss": 5.1410965795077965, + "tokens_seen": 59572224 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004959679037111334, + "loss": 2.1623, + "theoretical_loss": 5.140255569900284, + "tokens_seen": 59637760 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004959578736208626, + "loss": 2.2378, + "theoretical_loss": 5.13941574242015, + "tokens_seen": 59703296 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004959478435305918, + "loss": 2.2218, + "theoretical_loss": 5.138577094110906, + "tokens_seen": 59768832 + }, + { + "epoch": 0.02, + "learning_rate": 0.000495937813440321, + "loss": 2.0759, + "theoretical_loss": 5.137739622026694, + "tokens_seen": 59834368 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004959277833500501, + "loss": 2.0308, + "theoretical_loss": 5.136903323232229, + "tokens_seen": 59899904 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004959177532597793, + "loss": 2.1694, + "theoretical_loss": 5.13606819480275, + "tokens_seen": 59965440 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004959077231695085, + "loss": 2.2467, + "theoretical_loss": 5.135234233823978, + "tokens_seen": 60030976 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004958976930792377, + "loss": 2.1069, + "theoretical_loss": 5.134401437392057, + "tokens_seen": 60096512 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004958876629889669, + "loss": 2.2235, + "theoretical_loss": 5.133569802613515, + "tokens_seen": 60162048 + }, + { + "epoch": 0.02, + "learning_rate": 0.000495877632898696, + "loss": 2.2206, + "theoretical_loss": 5.132739326605209, + "tokens_seen": 60227584 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004958676028084253, + "loss": 2.1052, + "theoretical_loss": 5.131910006494278, + "tokens_seen": 60293120 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004958575727181544, + "loss": 2.2095, + "theoretical_loss": 5.131081839418099, + "tokens_seen": 60358656 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004958475426278837, + "loss": 2.278, + "theoretical_loss": 5.130254822524234, + "tokens_seen": 60424192 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004958375125376129, + "loss": 1.9707, + "theoretical_loss": 5.129428952970388, + "tokens_seen": 60489728 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004958274824473421, + "loss": 2.2479, + "theoretical_loss": 5.128604227924356, + "tokens_seen": 60555264 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": 0.0015134832356125116, + "objective/train/docs_used": 46630, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.824455738067627, + "objective/train/original_loss": 3.824455976486206, + "objective/train/theoretical_loss": 5.127780644563984, + "objective/train/tokens_used": 81080800, + "objective/train/value_avg": -0.006683349609375, + "objective/train/value_loss": 0.00011408379941713065, + "objective/train/value_max": -0.0004189014434814453, + "objective/train/value_min": -0.19873046875, + "objective/train/value_reward_corr": 0.06807154500106755, + "objective/train/value_std": 0.007152557373046875, + "objective/train/weight_avg": 1.0210613012313843, + "objective/train/weighted_lm_loss": 3.917862892150879, + "objective/train/weights_max": 6.899967193603516, + "objective/train/weights_min": 0.19592435657978058, + "theoretical_loss": 5.127780644563984, + "tokens_seen": 60620800 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004958174523570712, + "loss": 2.0318, + "theoretical_loss": 5.127780644563984, + "tokens_seen": 60620800 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004958074222668004, + "loss": 2.0541, + "theoretical_loss": 5.126958200077116, + "tokens_seen": 60686336 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004957973921765296, + "loss": 1.9625, + "theoretical_loss": 5.126136891661549, + "tokens_seen": 60751872 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004957873620862588, + "loss": 2.1469, + "theoretical_loss": 5.12531671652499, + "tokens_seen": 60817408 + }, + { + "epoch": 0.02, + "learning_rate": 0.000495777331995988, + "loss": 2.1946, + "theoretical_loss": 5.124497671885009, + "tokens_seen": 60882944 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004957673019057172, + "loss": 2.0229, + "theoretical_loss": 5.1236797549689905, + "tokens_seen": 60948480 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004957572718154463, + "loss": 2.2513, + "theoretical_loss": 5.122862963014094, + "tokens_seen": 61014016 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004957472417251755, + "loss": 2.06, + "theoretical_loss": 5.122047293267204, + "tokens_seen": 61079552 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004957372116349047, + "loss": 2.1536, + "theoretical_loss": 5.121232742984887, + "tokens_seen": 61145088 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004957271815446339, + "loss": 2.2646, + "theoretical_loss": 5.120419309433351, + "tokens_seen": 61210624 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004957171514543631, + "loss": 2.0732, + "theoretical_loss": 5.119606989888393, + "tokens_seen": 61276160 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004957071213640923, + "loss": 2.1637, + "theoretical_loss": 5.118795781635365, + "tokens_seen": 61341696 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004956970912738214, + "loss": 2.1981, + "theoretical_loss": 5.11798568196912, + "tokens_seen": 61407232 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004956870611835507, + "loss": 2.2019, + "theoretical_loss": 5.117176688193979, + "tokens_seen": 61472768 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004956770310932798, + "loss": 2.1491, + "theoretical_loss": 5.1163687976236805, + "tokens_seen": 61538304 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004956670010030091, + "loss": 2.2101, + "theoretical_loss": 5.115562007581341, + "tokens_seen": 61603840 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004956569709127383, + "loss": 2.0527, + "theoretical_loss": 5.114756315399407, + "tokens_seen": 61669376 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004956469408224674, + "loss": 2.1426, + "theoretical_loss": 5.113951718419624, + "tokens_seen": 61734912 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004956369107321966, + "loss": 2.2579, + "theoretical_loss": 5.113148213992981, + "tokens_seen": 61800448 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004956268806419258, + "loss": 2.0754, + "theoretical_loss": 5.112345799479678, + "tokens_seen": 61865984 + }, + { + "epoch": 0.02, + "learning_rate": 0.000495616850551655, + "loss": 2.0909, + "theoretical_loss": 5.111544472249082, + "tokens_seen": 61931520 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004956068204613842, + "loss": 2.0458, + "theoretical_loss": 5.110744229679679, + "tokens_seen": 61997056 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004955967903711133, + "loss": 2.175, + "theoretical_loss": 5.1099450691590445, + "tokens_seen": 62062592 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004955867602808425, + "loss": 2.2109, + "theoretical_loss": 5.109146988083792, + "tokens_seen": 62128128 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004955767301905717, + "loss": 2.2185, + "theoretical_loss": 5.108349983859538, + "tokens_seen": 62193664 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": 0.0002240528992842883, + "objective/train/docs_used": 47896, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.348313808441162, + "objective/train/original_loss": 4.348313808441162, + "objective/train/theoretical_loss": 5.107554053900861, + "objective/train/tokens_used": 82719200, + "objective/train/value_avg": -0.006885528564453125, + "objective/train/value_loss": 9.446393232792616e-05, + "objective/train/value_max": -0.000396728515625, + "objective/train/value_min": -0.1650390625, + "objective/train/value_reward_corr": 0.13026612535751456, + "objective/train/value_std": 0.0065460205078125, + "objective/train/weight_avg": 1.007053256034851, + "objective/train/weighted_lm_loss": 4.3681793212890625, + "objective/train/weights_max": 5.209013938903809, + "objective/train/weights_min": 0.24593952298164368, + "theoretical_loss": 5.107554053900861, + "tokens_seen": 62259200 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004955667001003009, + "loss": 2.1508, + "theoretical_loss": 5.107554053900861, + "tokens_seen": 62259200 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004955566700100301, + "loss": 2.1423, + "theoretical_loss": 5.106759195631259, + "tokens_seen": 62324736 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004955466399197592, + "loss": 2.1876, + "theoretical_loss": 5.105965406483108, + "tokens_seen": 62390272 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004955366098294884, + "loss": 2.1224, + "theoretical_loss": 5.105172683897628, + "tokens_seen": 62455808 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004955265797392177, + "loss": 2.1023, + "theoretical_loss": 5.104381025324839, + "tokens_seen": 62521344 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004955165496489468, + "loss": 2.004, + "theoretical_loss": 5.103590428223521, + "tokens_seen": 62586880 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004955065195586761, + "loss": 2.0772, + "theoretical_loss": 5.102800890061178, + "tokens_seen": 62652416 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004954964894684052, + "loss": 2.0656, + "theoretical_loss": 5.102012408313994, + "tokens_seen": 62717952 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004954864593781344, + "loss": 2.0651, + "theoretical_loss": 5.101224980466801, + "tokens_seen": 62783488 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004954764292878636, + "loss": 2.0918, + "theoretical_loss": 5.100438604013036, + "tokens_seen": 62849024 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004954663991975928, + "loss": 2.0106, + "theoretical_loss": 5.0996532764547, + "tokens_seen": 62914560 + }, + { + "epoch": 0.02, + "learning_rate": 0.000495456369107322, + "loss": 1.9911, + "theoretical_loss": 5.098868995302327, + "tokens_seen": 62980096 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004954463390170512, + "loss": 2.0062, + "theoretical_loss": 5.098085758074939, + "tokens_seen": 63045632 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004954363089267803, + "loss": 2.0743, + "theoretical_loss": 5.097303562300015, + "tokens_seen": 63111168 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004954262788365095, + "loss": 2.2011, + "theoretical_loss": 5.096522405513447, + "tokens_seen": 63176704 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004954162487462387, + "loss": 2.1528, + "theoretical_loss": 5.095742285259507, + "tokens_seen": 63242240 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004954062186559679, + "loss": 2.1499, + "theoretical_loss": 5.094963199090807, + "tokens_seen": 63307776 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004953961885656971, + "loss": 2.1218, + "theoretical_loss": 5.094185144568264, + "tokens_seen": 63373312 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004953861584754263, + "loss": 2.126, + "theoretical_loss": 5.093408119261066, + "tokens_seen": 63438848 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004953761283851555, + "loss": 2.0813, + "theoretical_loss": 5.092632120746626, + "tokens_seen": 63504384 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004953660982948846, + "loss": 2.0304, + "theoretical_loss": 5.091857146610557, + "tokens_seen": 63569920 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004953560682046139, + "loss": 2.1269, + "theoretical_loss": 5.091083194446627, + "tokens_seen": 63635456 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004953460381143431, + "loss": 2.151, + "theoretical_loss": 5.0903102618567315, + "tokens_seen": 63700992 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004953360080240723, + "loss": 2.1461, + "theoretical_loss": 5.089538346450848, + "tokens_seen": 63766528 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004953259779338014, + "loss": 2.0792, + "theoretical_loss": 5.088767445847009, + "tokens_seen": 63832064 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": -0.00016378660802729428, + "objective/train/docs_used": 48551, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.074954509735107, + "objective/train/original_loss": 4.074954509735107, + "objective/train/theoretical_loss": 5.087997557671262, + "objective/train/tokens_used": 84357600, + "objective/train/value_avg": -0.005924224853515625, + "objective/train/value_loss": 0.00011391686712158844, + "objective/train/value_max": -0.00036835670471191406, + "objective/train/value_min": -0.1622314453125, + "objective/train/value_reward_corr": 0.058423284858464396, + "objective/train/value_std": 0.00681304931640625, + "objective/train/weight_avg": 1.0042766332626343, + "objective/train/weighted_lm_loss": 4.078671455383301, + "objective/train/weights_max": 5.0647993087768555, + "objective/train/weights_min": 0.29577764868736267, + "theoretical_loss": 5.087997557671262, + "tokens_seen": 63897600 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004953159478435306, + "loss": 2.1329, + "theoretical_loss": 5.087997557671262, + "tokens_seen": 63897600 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004953059177532598, + "loss": 2.1127, + "theoretical_loss": 5.087228679557634, + "tokens_seen": 63963136 + }, + { + "epoch": 0.02, + "learning_rate": 0.000495295887662989, + "loss": 2.0736, + "theoretical_loss": 5.0864608091481, + "tokens_seen": 64028672 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004952858575727182, + "loss": 2.0945, + "theoretical_loss": 5.085693944092547, + "tokens_seen": 64094208 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004952758274824474, + "loss": 2.1166, + "theoretical_loss": 5.084928082048736, + "tokens_seen": 64159744 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004952657973921765, + "loss": 2.1293, + "theoretical_loss": 5.084163220682272, + "tokens_seen": 64225280 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004952557673019057, + "loss": 2.0287, + "theoretical_loss": 5.083399357666567, + "tokens_seen": 64290816 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004952457372116349, + "loss": 2.1924, + "theoretical_loss": 5.082636490682806, + "tokens_seen": 64356352 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004952357071213641, + "loss": 2.1279, + "theoretical_loss": 5.081874617419919, + "tokens_seen": 64421888 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004952256770310933, + "loss": 2.0253, + "theoretical_loss": 5.081113735574535, + "tokens_seen": 64487424 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004952156469408225, + "loss": 2.0607, + "theoretical_loss": 5.080353842850963, + "tokens_seen": 64552960 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004952056168505516, + "loss": 2.0098, + "theoretical_loss": 5.079594936961146, + "tokens_seen": 64618496 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004951955867602809, + "loss": 1.9171, + "theoretical_loss": 5.078837015624638, + "tokens_seen": 64684032 + }, + { + "epoch": 0.02, + "learning_rate": 0.00049518555667001, + "loss": 2.0481, + "theoretical_loss": 5.078080076568562, + "tokens_seen": 64749568 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004951755265797393, + "loss": 2.0011, + "theoretical_loss": 5.077324117527585, + "tokens_seen": 64815104 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004951654964894685, + "loss": 2.0952, + "theoretical_loss": 5.076569136243883, + "tokens_seen": 64880640 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004951554663991976, + "loss": 2.037, + "theoretical_loss": 5.075815130467101, + "tokens_seen": 64946176 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004951454363089268, + "loss": 2.0089, + "theoretical_loss": 5.075062097954335, + "tokens_seen": 65011712 + }, + { + "epoch": 0.02, + "learning_rate": 0.000495135406218656, + "loss": 2.0717, + "theoretical_loss": 5.0743100364700915, + "tokens_seen": 65077248 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004951253761283852, + "loss": 2.092, + "theoretical_loss": 5.07355894378625, + "tokens_seen": 65142784 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004951153460381144, + "loss": 2.0662, + "theoretical_loss": 5.072808817682045, + "tokens_seen": 65208320 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004951053159478435, + "loss": 1.9005, + "theoretical_loss": 5.072059655944021, + "tokens_seen": 65273856 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004950952858575727, + "loss": 2.1029, + "theoretical_loss": 5.071311456366011, + "tokens_seen": 65339392 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004950852557673019, + "loss": 2.005, + "theoretical_loss": 5.070564216749103, + "tokens_seen": 65404928 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004950752256770311, + "loss": 2.1625, + "theoretical_loss": 5.069817934901602, + "tokens_seen": 65470464 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": 0.0014170599170029163, + "objective/train/docs_used": 49128, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.127939224243164, + "objective/train/original_loss": 4.127939224243164, + "objective/train/theoretical_loss": 5.069072608639006, + "objective/train/tokens_used": 85996000, + "objective/train/value_avg": -0.007076263427734375, + "objective/train/value_loss": 0.00011187413474544883, + "objective/train/value_max": -0.000583648681640625, + "objective/train/value_min": -0.18798828125, + "objective/train/value_reward_corr": 0.014551581689580284, + "objective/train/value_std": 0.006313323974609375, + "objective/train/weight_avg": 1.0199280977249146, + "objective/train/weighted_lm_loss": 4.187434673309326, + "objective/train/weights_max": 6.552737236022949, + "objective/train/weights_min": 0.374116986989975, + "theoretical_loss": 5.069072608639006, + "tokens_seen": 65536000 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004950651955867603, + "loss": 2.0314, + "theoretical_loss": 5.069072608639006, + "tokens_seen": 65536000 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004950551654964894, + "loss": 2.0492, + "theoretical_loss": 5.0683282357839765, + "tokens_seen": 65601536 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004950451354062186, + "loss": 2.1585, + "theoretical_loss": 5.067584814166302, + "tokens_seen": 65667072 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004950351053159479, + "loss": 1.8994, + "theoretical_loss": 5.066842341622871, + "tokens_seen": 65732608 + }, + { + "epoch": 0.02, + "learning_rate": 0.000495025075225677, + "loss": 1.9735, + "theoretical_loss": 5.066100815997643, + "tokens_seen": 65798144 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004950150451354063, + "loss": 2.07, + "theoretical_loss": 5.065360235141615, + "tokens_seen": 65863680 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004950050150451354, + "loss": 2.045, + "theoretical_loss": 5.0646205969127935, + "tokens_seen": 65929216 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004949949849548646, + "loss": 2.0546, + "theoretical_loss": 5.063881899176165, + "tokens_seen": 65994752 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004949849548645938, + "loss": 1.9649, + "theoretical_loss": 5.063144139803664, + "tokens_seen": 66060288 + }, + { + "epoch": 0.02, + "learning_rate": 0.000494974924774323, + "loss": 1.9825, + "theoretical_loss": 5.062407316674149, + "tokens_seen": 66125824 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004949648946840522, + "loss": 1.959, + "theoretical_loss": 5.061671427673364, + "tokens_seen": 66191360 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004949548645937814, + "loss": 1.9924, + "theoretical_loss": 5.060936470693919, + "tokens_seen": 66256896 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004949448345035105, + "loss": 1.9168, + "theoretical_loss": 5.060202443635257, + "tokens_seen": 66322432 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004949348044132397, + "loss": 2.033, + "theoretical_loss": 5.05946934440362, + "tokens_seen": 66387968 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004949247743229689, + "loss": 2.1589, + "theoretical_loss": 5.058737170912032, + "tokens_seen": 66453504 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004949147442326981, + "loss": 1.9658, + "theoretical_loss": 5.058005921080258, + "tokens_seen": 66519040 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004949047141424273, + "loss": 1.9364, + "theoretical_loss": 5.057275592834783, + "tokens_seen": 66584576 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004948946840521565, + "loss": 1.9307, + "theoretical_loss": 5.0565461841087815, + "tokens_seen": 66650112 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004948846539618856, + "loss": 1.8114, + "theoretical_loss": 5.055817692842091, + "tokens_seen": 66715648 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004948746238716148, + "loss": 2.1175, + "theoretical_loss": 5.05509011698118, + "tokens_seen": 66781184 + }, + { + "epoch": 0.02, + "learning_rate": 0.000494864593781344, + "loss": 1.9963, + "theoretical_loss": 5.054363454479127, + "tokens_seen": 66846720 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004948545636910733, + "loss": 1.9646, + "theoretical_loss": 5.053637703295585, + "tokens_seen": 66912256 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004948445336008024, + "loss": 2.0687, + "theoretical_loss": 5.0529128613967575, + "tokens_seen": 66977792 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004948345035105316, + "loss": 2.0104, + "theoretical_loss": 5.052188926755374, + "tokens_seen": 67043328 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004948244734202607, + "loss": 2.1381, + "theoretical_loss": 5.051465897350656, + "tokens_seen": 67108864 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": 0.0018457121914252639, + "objective/train/docs_used": 50658, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.6496641635894775, + "objective/train/original_loss": 3.6496641635894775, + "objective/train/theoretical_loss": 5.0507437711683, + "objective/train/tokens_used": 87634400, + "objective/train/value_avg": -0.00634002685546875, + "objective/train/value_loss": 0.00019463311764411628, + "objective/train/value_max": -0.00029587745666503906, + "objective/train/value_min": -0.259765625, + "objective/train/value_reward_corr": 0.03024750496725171, + "objective/train/value_std": 0.00978851318359375, + "objective/train/weight_avg": 1.029559850692749, + "objective/train/weighted_lm_loss": 3.7486555576324463, + "objective/train/weights_max": 9.18894100189209, + "objective/train/weights_min": 0.19299429655075073, + "theoretical_loss": 5.0507437711683, + "tokens_seen": 67174400 + }, + { + "epoch": 0.02, + "learning_rate": 0.00049481444332999, + "loss": 1.9461, + "theoretical_loss": 5.0507437711683, + "tokens_seen": 67174400 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004948044132397192, + "loss": 1.9773, + "theoretical_loss": 5.050022546200437, + "tokens_seen": 67239936 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004947943831494484, + "loss": 2.025, + "theoretical_loss": 5.049302220445618, + "tokens_seen": 67305472 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004947843530591776, + "loss": 2.0921, + "theoretical_loss": 5.048582791908779, + "tokens_seen": 67371008 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004947743229689067, + "loss": 2.005, + "theoretical_loss": 5.04786425860122, + "tokens_seen": 67436544 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004947642928786359, + "loss": 2.0902, + "theoretical_loss": 5.047146618540575, + "tokens_seen": 67502080 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004947542627883651, + "loss": 2.105, + "theoretical_loss": 5.046429869750787, + "tokens_seen": 67567616 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004947442326980943, + "loss": 2.1111, + "theoretical_loss": 5.045714010262081, + "tokens_seen": 67633152 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004947342026078235, + "loss": 2.0977, + "theoretical_loss": 5.044999038110941, + "tokens_seen": 67698688 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004947241725175527, + "loss": 2.0489, + "theoretical_loss": 5.044284951340078, + "tokens_seen": 67764224 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004947141424272818, + "loss": 1.996, + "theoretical_loss": 5.043571747998412, + "tokens_seen": 67829760 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004947041123370111, + "loss": 2.2042, + "theoretical_loss": 5.042859426141039, + "tokens_seen": 67895296 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004946940822467402, + "loss": 1.9212, + "theoretical_loss": 5.042147983829212, + "tokens_seen": 67960832 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004946840521564695, + "loss": 1.9989, + "theoretical_loss": 5.041437419130309, + "tokens_seen": 68026368 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004946740220661987, + "loss": 2.0064, + "theoretical_loss": 5.040727730117813, + "tokens_seen": 68091904 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004946639919759278, + "loss": 2.1156, + "theoretical_loss": 5.040018914871285, + "tokens_seen": 68157440 + }, + { + "epoch": 0.02, + "learning_rate": 0.000494653961885657, + "loss": 1.9717, + "theoretical_loss": 5.039310971476339, + "tokens_seen": 68222976 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004946439317953862, + "loss": 2.1671, + "theoretical_loss": 5.038603898024617, + "tokens_seen": 68288512 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004946339017051154, + "loss": 2.0643, + "theoretical_loss": 5.037897692613765, + "tokens_seen": 68354048 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004946238716148446, + "loss": 2.2134, + "theoretical_loss": 5.037192353347406, + "tokens_seen": 68419584 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004946138415245737, + "loss": 1.9487, + "theoretical_loss": 5.0364878783351195, + "tokens_seen": 68485120 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004946038114343029, + "loss": 1.9702, + "theoretical_loss": 5.035784265692413, + "tokens_seen": 68550656 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004945937813440321, + "loss": 2.0813, + "theoretical_loss": 5.035081513540704, + "tokens_seen": 68616192 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004945837512537613, + "loss": 2.1141, + "theoretical_loss": 5.034379620007284, + "tokens_seen": 68681728 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004945737211634905, + "loss": 2.0655, + "theoretical_loss": 5.03367858322531, + "tokens_seen": 68747264 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": -0.0004463552322704345, + "objective/train/docs_used": 51428, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4726946353912354, + "objective/train/original_loss": 3.472694158554077, + "objective/train/theoretical_loss": 5.032978401333766, + "objective/train/tokens_used": 89272800, + "objective/train/value_avg": -0.005031585693359375, + "objective/train/value_loss": 0.0001221809070557356, + "objective/train/value_max": -0.0002065896987915039, + "objective/train/value_min": -0.079345703125, + "objective/train/value_reward_corr": 0.07334069935886453, + "objective/train/value_std": 0.0043487548828125, + "objective/train/weight_avg": 1.0010077953338623, + "objective/train/weighted_lm_loss": 3.4683103561401367, + "objective/train/weights_max": 2.211026906967163, + "objective/train/weights_min": 0.3324393332004547, + "theoretical_loss": 5.032978401333766, + "tokens_seen": 68812800 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004945636910732196, + "loss": 1.9579, + "theoretical_loss": 5.032978401333766, + "tokens_seen": 68812800 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004945536609829488, + "loss": 2.075, + "theoretical_loss": 5.0322790724774515, + "tokens_seen": 68878336 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004945436308926781, + "loss": 2.042, + "theoretical_loss": 5.031580594806948, + "tokens_seen": 68943872 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004945336008024072, + "loss": 2.0147, + "theoretical_loss": 5.030882966478602, + "tokens_seen": 69009408 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004945235707121365, + "loss": 2.0999, + "theoretical_loss": 5.0301861856544985, + "tokens_seen": 69074944 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004945135406218656, + "loss": 2.1198, + "theoretical_loss": 5.029490250502439, + "tokens_seen": 69140480 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004945035105315948, + "loss": 2.0735, + "theoretical_loss": 5.028795159195919, + "tokens_seen": 69206016 + }, + { + "epoch": 0.02, + "learning_rate": 0.000494493480441324, + "loss": 2.0973, + "theoretical_loss": 5.0281009099141, + "tokens_seen": 69271552 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004944834503510532, + "loss": 2.1701, + "theoretical_loss": 5.027407500841795, + "tokens_seen": 69337088 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004944734202607824, + "loss": 2.0351, + "theoretical_loss": 5.026714930169438, + "tokens_seen": 69402624 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004944633901705116, + "loss": 2.0906, + "theoretical_loss": 5.026023196093064, + "tokens_seen": 69468160 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004944533600802407, + "loss": 2.0241, + "theoretical_loss": 5.02533229681429, + "tokens_seen": 69533696 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004944433299899699, + "loss": 2.0335, + "theoretical_loss": 5.0246422305402865, + "tokens_seen": 69599232 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004944332998996991, + "loss": 2.0702, + "theoretical_loss": 5.023952995483757, + "tokens_seen": 69664768 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004944232698094283, + "loss": 2.397, + "theoretical_loss": 5.0232645898629205, + "tokens_seen": 69730304 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004944132397191575, + "loss": 2.0356, + "theoretical_loss": 5.022577011901481, + "tokens_seen": 69795840 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004944032096288867, + "loss": 2.0369, + "theoretical_loss": 5.021890259828613, + "tokens_seen": 69861376 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004943931795386158, + "loss": 1.9744, + "theoretical_loss": 5.021204331878934, + "tokens_seen": 69926912 + }, + { + "epoch": 0.02, + "learning_rate": 0.000494383149448345, + "loss": 2.1171, + "theoretical_loss": 5.020519226292487, + "tokens_seen": 69992448 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004943731193580742, + "loss": 1.9941, + "theoretical_loss": 5.019834941314716, + "tokens_seen": 70057984 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004943630892678035, + "loss": 1.872, + "theoretical_loss": 5.0191514751964466, + "tokens_seen": 70123520 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004943530591775326, + "loss": 1.9045, + "theoretical_loss": 5.01846882619386, + "tokens_seen": 70189056 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004943430290872618, + "loss": 1.9434, + "theoretical_loss": 5.01778699256848, + "tokens_seen": 70254592 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004943329989969909, + "loss": 2.0755, + "theoretical_loss": 5.017105972587139, + "tokens_seen": 70320128 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004943229689067202, + "loss": 2.0388, + "theoretical_loss": 5.016425764521972, + "tokens_seen": 70385664 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": -0.00012673618039116263, + "objective/train/docs_used": 52769, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.61791729927063, + "objective/train/original_loss": 3.6179165840148926, + "objective/train/theoretical_loss": 5.015746366650382, + "objective/train/tokens_used": 90911200, + "objective/train/value_avg": -0.00518035888671875, + "objective/train/value_loss": 9.252568270312622e-05, + "objective/train/value_max": -0.00030303001403808594, + "objective/train/value_min": -0.38427734375, + "objective/train/value_reward_corr": -0.012224030727961287, + "objective/train/value_std": 0.00540924072265625, + "objective/train/weight_avg": 1.0050456523895264, + "objective/train/weighted_lm_loss": 3.657482385635376, + "objective/train/weights_max": 43.31175231933594, + "objective/train/weights_min": 0.33351585268974304, + "theoretical_loss": 5.015746366650382, + "tokens_seen": 70451200 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004943129388164494, + "loss": 1.9303, + "theoretical_loss": 5.015746366650382, + "tokens_seen": 70451200 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004943029087261786, + "loss": 2.0833, + "theoretical_loss": 5.0150677772550285, + "tokens_seen": 70516736 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004942928786359078, + "loss": 1.9278, + "theoretical_loss": 5.014389994623801, + "tokens_seen": 70582272 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004942828485456369, + "loss": 1.9969, + "theoretical_loss": 5.013713017049801, + "tokens_seen": 70647808 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004942728184553661, + "loss": 2.0073, + "theoretical_loss": 5.013036842831319, + "tokens_seen": 70713344 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004942627883650953, + "loss": 1.9924, + "theoretical_loss": 5.012361470271816, + "tokens_seen": 70778880 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004942527582748245, + "loss": 1.987, + "theoretical_loss": 5.0116868976799065, + "tokens_seen": 70844416 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004942427281845537, + "loss": 2.0383, + "theoretical_loss": 5.0110131233693265, + "tokens_seen": 70909952 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004942326980942828, + "loss": 1.9329, + "theoretical_loss": 5.010340145658926, + "tokens_seen": 70975488 + }, + { + "epoch": 0.02, + "learning_rate": 0.000494222668004012, + "loss": 1.9235, + "theoretical_loss": 5.009667962872642, + "tokens_seen": 71041024 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004942126379137412, + "loss": 1.8889, + "theoretical_loss": 5.008996573339481, + "tokens_seen": 71106560 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004942026078234704, + "loss": 2.0542, + "theoretical_loss": 5.008325975393495, + "tokens_seen": 71172096 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004941925777331996, + "loss": 2.0038, + "theoretical_loss": 5.00765616737377, + "tokens_seen": 71237632 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004941825476429289, + "loss": 1.977, + "theoretical_loss": 5.006987147624395, + "tokens_seen": 71303168 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004941725175526579, + "loss": 1.9952, + "theoretical_loss": 5.006318914494451, + "tokens_seen": 71368704 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004941624874623872, + "loss": 2.1473, + "theoretical_loss": 5.00565146633799, + "tokens_seen": 71434240 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004941524573721163, + "loss": 2.0091, + "theoretical_loss": 5.004984801514011, + "tokens_seen": 71499776 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004941424272818456, + "loss": 1.8381, + "theoretical_loss": 5.004318918386446, + "tokens_seen": 71565312 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004941323971915748, + "loss": 1.9927, + "theoretical_loss": 5.003653815324136, + "tokens_seen": 71630848 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004941223671013039, + "loss": 1.9672, + "theoretical_loss": 5.002989490700816, + "tokens_seen": 71696384 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004941123370110331, + "loss": 2.0891, + "theoretical_loss": 5.002325942895094, + "tokens_seen": 71761920 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004941023069207623, + "loss": 1.9329, + "theoretical_loss": 5.001663170290431, + "tokens_seen": 71827456 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004940922768304915, + "loss": 1.8968, + "theoretical_loss": 5.0010011712751234, + "tokens_seen": 71892992 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004940822467402207, + "loss": 2.0814, + "theoretical_loss": 5.000339944242283, + "tokens_seen": 71958528 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004940722166499498, + "loss": 2.0528, + "theoretical_loss": 4.9996794875898205, + "tokens_seen": 72024064 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": 0.0008325845701619983, + "objective/train/docs_used": 53229, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.7424936294555664, + "objective/train/original_loss": 3.7424936294555664, + "objective/train/theoretical_loss": 4.999019799720424, + "objective/train/tokens_used": 92549600, + "objective/train/value_avg": -0.00559234619140625, + "objective/train/value_loss": 9.033548849401996e-05, + "objective/train/value_max": -0.0002269744873046875, + "objective/train/value_min": -0.1900634765625, + "objective/train/value_reward_corr": 0.08932904706913883, + "objective/train/value_std": 0.0054931640625, + "objective/train/weight_avg": 1.0128018856048584, + "objective/train/weighted_lm_loss": 3.794332504272461, + "objective/train/weights_max": 5.476353168487549, + "objective/train/weights_min": 0.2922336161136627, + "theoretical_loss": 4.999019799720424, + "tokens_seen": 72089600 + }, + { + "epoch": 0.02, + "learning_rate": 0.000494062186559679, + "loss": 2.0413, + "theoretical_loss": 4.999019799720424, + "tokens_seen": 72089600 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004940521564694082, + "loss": 1.8753, + "theoretical_loss": 4.998360879041542, + "tokens_seen": 72155136 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004940421263791374, + "loss": 2.045, + "theoretical_loss": 4.997702723965366, + "tokens_seen": 72220672 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004940320962888666, + "loss": 2.0263, + "theoretical_loss": 4.997045332908808, + "tokens_seen": 72286208 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004940220661985958, + "loss": 2.1133, + "theoretical_loss": 4.996388704293487, + "tokens_seen": 72351744 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004940120361083249, + "loss": 2.014, + "theoretical_loss": 4.995732836545709, + "tokens_seen": 72417280 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004940020060180542, + "loss": 1.817, + "theoretical_loss": 4.995077728096446, + "tokens_seen": 72482816 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004939919759277834, + "loss": 1.8329, + "theoretical_loss": 4.994423377381324, + "tokens_seen": 72548352 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004939819458375126, + "loss": 2.0521, + "theoretical_loss": 4.993769782840598, + "tokens_seen": 72613888 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004939719157472418, + "loss": 1.9146, + "theoretical_loss": 4.993116942919141, + "tokens_seen": 72679424 + }, + { + "epoch": 0.02, + "learning_rate": 0.000493961885656971, + "loss": 2.1105, + "theoretical_loss": 4.9924648560664195, + "tokens_seen": 72744960 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004939518555667001, + "loss": 1.9477, + "theoretical_loss": 4.99181352073648, + "tokens_seen": 72810496 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004939418254764293, + "loss": 2.0811, + "theoretical_loss": 4.991162935387935, + "tokens_seen": 72876032 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004939317953861585, + "loss": 1.9812, + "theoretical_loss": 4.990513098483933, + "tokens_seen": 72941568 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004939217652958877, + "loss": 2.0345, + "theoretical_loss": 4.989864008492155, + "tokens_seen": 73007104 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004939117352056169, + "loss": 1.9683, + "theoretical_loss": 4.98921566388479, + "tokens_seen": 73072640 + }, + { + "epoch": 0.02, + "learning_rate": 0.000493901705115346, + "loss": 1.9359, + "theoretical_loss": 4.9885680631385165, + "tokens_seen": 73138176 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004938916750250752, + "loss": 1.8298, + "theoretical_loss": 4.987921204734489, + "tokens_seen": 73203712 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004938816449348044, + "loss": 1.9863, + "theoretical_loss": 4.987275087158322, + "tokens_seen": 73269248 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004938716148445337, + "loss": 1.9336, + "theoretical_loss": 4.986629708900064, + "tokens_seen": 73334784 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004938615847542628, + "loss": 2.1127, + "theoretical_loss": 4.985985068454193, + "tokens_seen": 73400320 + }, + { + "epoch": 0.02, + "learning_rate": 0.000493851554663992, + "loss": 2.1247, + "theoretical_loss": 4.985341164319592, + "tokens_seen": 73465856 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004938415245737211, + "loss": 2.1771, + "theoretical_loss": 4.984697994999531, + "tokens_seen": 73531392 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004938314944834504, + "loss": 2.021, + "theoretical_loss": 4.984055559001657, + "tokens_seen": 73596928 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004938214643931796, + "loss": 2.1193, + "theoretical_loss": 4.983413854837971, + "tokens_seen": 73662464 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": -0.0004779342270921916, + "objective/train/docs_used": 53865, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.033031463623047, + "objective/train/original_loss": 4.033030986785889, + "objective/train/theoretical_loss": 4.982772881024813, + "objective/train/tokens_used": 94188000, + "objective/train/value_avg": -0.005100250244140625, + "objective/train/value_loss": 0.00010871110862353817, + "objective/train/value_max": -0.0002359151840209961, + "objective/train/value_min": -0.09039306640625, + "objective/train/value_reward_corr": 0.015207086004974944, + "objective/train/value_std": 0.00405120849609375, + "objective/train/weight_avg": 0.99991375207901, + "objective/train/weighted_lm_loss": 4.036130905151367, + "objective/train/weights_max": 2.379913091659546, + "objective/train/weights_min": 0.1907527595758438, + "theoretical_loss": 4.982772881024813, + "tokens_seen": 73728000 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004938114343029088, + "loss": 2.0414, + "theoretical_loss": 4.982772881024813, + "tokens_seen": 73728000 + }, + { + "epoch": 0.02, + "learning_rate": 0.000493801404212638, + "loss": 2.1048, + "theoretical_loss": 4.982132636082847, + "tokens_seen": 73793536 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004937913741223671, + "loss": 2.0769, + "theoretical_loss": 4.981493118537046, + "tokens_seen": 73859072 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004937813440320963, + "loss": 2.0101, + "theoretical_loss": 4.9808543269166705, + "tokens_seen": 73924608 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004937713139418255, + "loss": 2.0069, + "theoretical_loss": 4.980216259755256, + "tokens_seen": 73990144 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004937612838515547, + "loss": 2.0896, + "theoretical_loss": 4.979578915590595, + "tokens_seen": 74055680 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004937512537612839, + "loss": 1.9633, + "theoretical_loss": 4.978942292964726, + "tokens_seen": 74121216 + }, + { + "epoch": 0.02, + "learning_rate": 0.000493741223671013, + "loss": 2.0351, + "theoretical_loss": 4.978306390423909, + "tokens_seen": 74186752 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004937311935807422, + "loss": 1.8715, + "theoretical_loss": 4.977671206518615, + "tokens_seen": 74252288 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004937211634904714, + "loss": 2.0001, + "theoretical_loss": 4.977036739803509, + "tokens_seen": 74317824 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004937111334002006, + "loss": 1.9753, + "theoretical_loss": 4.976402988837435, + "tokens_seen": 74383360 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004937011033099298, + "loss": 2.0028, + "theoretical_loss": 4.9757699521834, + "tokens_seen": 74448896 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004936910732196591, + "loss": 1.9435, + "theoretical_loss": 4.975137628408555, + "tokens_seen": 74514432 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004936810431293881, + "loss": 2.0576, + "theoretical_loss": 4.9745060160841845, + "tokens_seen": 74579968 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004936710130391174, + "loss": 1.9484, + "theoretical_loss": 4.973875113785686, + "tokens_seen": 74645504 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004936609829488465, + "loss": 1.8943, + "theoretical_loss": 4.9732449200925615, + "tokens_seen": 74711040 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004936509528585758, + "loss": 1.9926, + "theoretical_loss": 4.972615433588394, + "tokens_seen": 74776576 + }, + { + "epoch": 0.02, + "learning_rate": 0.000493640922768305, + "loss": 1.8715, + "theoretical_loss": 4.971986652860837, + "tokens_seen": 74842112 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004936308926780341, + "loss": 1.9317, + "theoretical_loss": 4.971358576501599, + "tokens_seen": 74907648 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004936208625877633, + "loss": 1.9339, + "theoretical_loss": 4.970731203106427, + "tokens_seen": 74973184 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004936108324974925, + "loss": 1.9432, + "theoretical_loss": 4.970104531275091, + "tokens_seen": 75038720 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004936008024072217, + "loss": 1.9637, + "theoretical_loss": 4.96947855961137, + "tokens_seen": 75104256 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004935907723169509, + "loss": 1.9162, + "theoretical_loss": 4.968853286723041, + "tokens_seen": 75169792 + }, + { + "epoch": 0.02, + "learning_rate": 0.00049358074222668, + "loss": 1.7711, + "theoretical_loss": 4.968228711221853, + "tokens_seen": 75235328 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004935707121364092, + "loss": 1.8056, + "theoretical_loss": 4.967604831723525, + "tokens_seen": 75300864 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": -0.0004275987739674747, + "objective/train/docs_used": 54881, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4652066230773926, + "objective/train/original_loss": 3.4652061462402344, + "objective/train/theoretical_loss": 4.966981646847723, + "objective/train/tokens_used": 95826400, + "objective/train/value_avg": -0.005161285400390625, + "objective/train/value_loss": 0.00011201648885617033, + "objective/train/value_max": -0.0002148151397705078, + "objective/train/value_min": -0.30859375, + "objective/train/value_reward_corr": 0.0015010440245383084, + "objective/train/value_std": 0.007068634033203125, + "objective/train/weight_avg": 1.0030624866485596, + "objective/train/weighted_lm_loss": 3.4808480739593506, + "objective/train/weights_max": 20.230621337890625, + "objective/train/weights_min": 0.0833415612578392, + "theoretical_loss": 4.966981646847723, + "tokens_seen": 75366400 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004935606820461384, + "loss": 1.8594, + "theoretical_loss": 4.966981646847723, + "tokens_seen": 75366400 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004935506519558676, + "loss": 2.0302, + "theoretical_loss": 4.966359155218047, + "tokens_seen": 75431936 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004935406218655968, + "loss": 1.9233, + "theoretical_loss": 4.96573735546202, + "tokens_seen": 75497472 + }, + { + "epoch": 0.02, + "learning_rate": 0.000493530591775326, + "loss": 1.8869, + "theoretical_loss": 4.965116246211068, + "tokens_seen": 75563008 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004935205616850551, + "loss": 1.9938, + "theoretical_loss": 4.964495826100508, + "tokens_seen": 75628544 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004935105315947844, + "loss": 1.9218, + "theoretical_loss": 4.963876093769536, + "tokens_seen": 75694080 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004935005015045135, + "loss": 1.9438, + "theoretical_loss": 4.963257047861209, + "tokens_seen": 75759616 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004934904714142428, + "loss": 1.9908, + "theoretical_loss": 4.962638687022432, + "tokens_seen": 75825152 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004934804413239719, + "loss": 1.8567, + "theoretical_loss": 4.962021009903943, + "tokens_seen": 75890688 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004934704112337011, + "loss": 1.7497, + "theoretical_loss": 4.9614040151603, + "tokens_seen": 75956224 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004934603811434303, + "loss": 1.9717, + "theoretical_loss": 4.960787701449868, + "tokens_seen": 76021760 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004934503510531595, + "loss": 2.1152, + "theoretical_loss": 4.960172067434801, + "tokens_seen": 76087296 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004934403209628887, + "loss": 2.0156, + "theoretical_loss": 4.959557111781034, + "tokens_seen": 76152832 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004934302908726179, + "loss": 2.0249, + "theoretical_loss": 4.958942833158262, + "tokens_seen": 76218368 + }, + { + "epoch": 0.02, + "learning_rate": 0.000493420260782347, + "loss": 2.0033, + "theoretical_loss": 4.958329230239932, + "tokens_seen": 76283904 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004934102306920762, + "loss": 1.8541, + "theoretical_loss": 4.9577163017032255, + "tokens_seen": 76349440 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004934002006018054, + "loss": 1.9554, + "theoretical_loss": 4.957104046229048, + "tokens_seen": 76414976 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004933901705115346, + "loss": 1.7877, + "theoretical_loss": 4.9564924625020135, + "tokens_seen": 76480512 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004933801404212638, + "loss": 2.0871, + "theoretical_loss": 4.955881549210428, + "tokens_seen": 76546048 + }, + { + "epoch": 0.02, + "learning_rate": 0.000493370110330993, + "loss": 1.7994, + "theoretical_loss": 4.955271305046283, + "tokens_seen": 76611584 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004933600802407221, + "loss": 1.9282, + "theoretical_loss": 4.954661728705236, + "tokens_seen": 76677120 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004933500501504513, + "loss": 1.8981, + "theoretical_loss": 4.9540528188866, + "tokens_seen": 76742656 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004933400200601805, + "loss": 1.95, + "theoretical_loss": 4.953444574293328, + "tokens_seen": 76808192 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004933299899699098, + "loss": 1.9679, + "theoretical_loss": 4.952836993632001, + "tokens_seen": 76873728 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004933199598796389, + "loss": 1.8382, + "theoretical_loss": 4.952230075612816, + "tokens_seen": 76939264 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": 0.0005203356849960983, + "objective/train/docs_used": 55424, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.044203281402588, + "objective/train/original_loss": 4.044203281402588, + "objective/train/theoretical_loss": 4.951623818949572, + "objective/train/tokens_used": 97464800, + "objective/train/value_avg": -0.00547027587890625, + "objective/train/value_loss": 7.87184399086982e-05, + "objective/train/value_max": -0.0002846717834472656, + "objective/train/value_min": -0.154541015625, + "objective/train/value_reward_corr": 0.1133809513618442, + "objective/train/value_std": 0.00632476806640625, + "objective/train/weight_avg": 1.0092936754226685, + "objective/train/weighted_lm_loss": 4.105243682861328, + "objective/train/weights_max": 4.308821201324463, + "objective/train/weights_min": 0.4399987757205963, + "theoretical_loss": 4.951623818949572, + "tokens_seen": 77004800 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004933099297893682, + "loss": 1.9967, + "theoretical_loss": 4.951623818949572, + "tokens_seen": 77004800 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004932998996990972, + "loss": 1.9529, + "theoretical_loss": 4.951018222359654, + "tokens_seen": 77070336 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004932898696088265, + "loss": 1.9653, + "theoretical_loss": 4.950413284564025, + "tokens_seen": 77135872 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004932798395185557, + "loss": 2.008, + "theoretical_loss": 4.949809004287209, + "tokens_seen": 77201408 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004932698094282849, + "loss": 1.7854, + "theoretical_loss": 4.949205380257279, + "tokens_seen": 77266944 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004932597793380141, + "loss": 1.8197, + "theoretical_loss": 4.948602411205847, + "tokens_seen": 77332480 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004932497492477432, + "loss": 2.0282, + "theoretical_loss": 4.948000095868046, + "tokens_seen": 77398016 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004932397191574724, + "loss": 2.003, + "theoretical_loss": 4.9473984329825225, + "tokens_seen": 77463552 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004932296890672016, + "loss": 1.8385, + "theoretical_loss": 4.946797421291418, + "tokens_seen": 77529088 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004932196589769308, + "loss": 2.0129, + "theoretical_loss": 4.946197059540362, + "tokens_seen": 77594624 + }, + { + "epoch": 0.02, + "learning_rate": 0.00049320962888666, + "loss": 1.9027, + "theoretical_loss": 4.945597346478458, + "tokens_seen": 77660160 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004931995987963893, + "loss": 1.8539, + "theoretical_loss": 4.944998280858268, + "tokens_seen": 77725696 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004931895687061183, + "loss": 1.9323, + "theoretical_loss": 4.944399861435803, + "tokens_seen": 77791232 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004931795386158476, + "loss": 1.9154, + "theoretical_loss": 4.943802086970509, + "tokens_seen": 77856768 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004931695085255767, + "loss": 2.047, + "theoretical_loss": 4.943204956225257, + "tokens_seen": 77922304 + }, + { + "epoch": 0.02, + "learning_rate": 0.000493159478435306, + "loss": 1.8269, + "theoretical_loss": 4.942608467966326, + "tokens_seen": 77987840 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004931494483450352, + "loss": 1.9994, + "theoretical_loss": 4.942012620963396, + "tokens_seen": 78053376 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004931394182547643, + "loss": 1.9054, + "theoretical_loss": 4.941417413989534, + "tokens_seen": 78118912 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004931293881644935, + "loss": 1.7874, + "theoretical_loss": 4.940822845821179, + "tokens_seen": 78184448 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004931193580742227, + "loss": 1.9305, + "theoretical_loss": 4.940228915238132, + "tokens_seen": 78249984 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004931093279839519, + "loss": 1.9337, + "theoretical_loss": 4.939635621023548, + "tokens_seen": 78315520 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004930992978936811, + "loss": 1.8127, + "theoretical_loss": 4.939042961963916, + "tokens_seen": 78381056 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004930892678034102, + "loss": 1.7974, + "theoretical_loss": 4.938450936849053, + "tokens_seen": 78446592 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004930792377131394, + "loss": 1.9946, + "theoretical_loss": 4.937859544472092, + "tokens_seen": 78512128 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004930692076228686, + "loss": 1.829, + "theoretical_loss": 4.937268783629463, + "tokens_seen": 78577664 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": -0.0009810628835111856, + "objective/train/docs_used": 56825, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.6470417976379395, + "objective/train/original_loss": 3.6470417976379395, + "objective/train/theoretical_loss": 4.936678653120895, + "objective/train/tokens_used": 99103200, + "objective/train/value_avg": -0.004558563232421875, + "objective/train/value_loss": 0.00011408594582462683, + "objective/train/value_max": -0.0001499652862548828, + "objective/train/value_min": -0.06585693359375, + "objective/train/value_reward_corr": 0.16880493424660706, + "objective/train/value_std": 0.004367828369140625, + "objective/train/weight_avg": 0.9953742623329163, + "objective/train/weighted_lm_loss": 3.6373140811920166, + "objective/train/weights_max": 1.8339755535125732, + "objective/train/weights_min": 0.37673419713974, + "theoretical_loss": 4.936678653120895, + "tokens_seen": 78643200 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004930591775325978, + "loss": 1.9142, + "theoretical_loss": 4.936678653120895, + "tokens_seen": 78643200 + }, + { + "epoch": 0.02, + "learning_rate": 0.000493049147442327, + "loss": 2.1408, + "theoretical_loss": 4.936089151749389, + "tokens_seen": 78708736 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004930391173520562, + "loss": 2.1293, + "theoretical_loss": 4.935500278321216, + "tokens_seen": 78774272 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004930290872617853, + "loss": 1.8762, + "theoretical_loss": 4.934912031645903, + "tokens_seen": 78839808 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004930190571715146, + "loss": 1.8095, + "theoretical_loss": 4.934324410536221, + "tokens_seen": 78905344 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004930090270812437, + "loss": 1.874, + "theoretical_loss": 4.933737413808172, + "tokens_seen": 78970880 + }, + { + "epoch": 0.02, + "learning_rate": 0.000492998996990973, + "loss": 1.9744, + "theoretical_loss": 4.933151040280983, + "tokens_seen": 79036416 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004929889669007021, + "loss": 1.9525, + "theoretical_loss": 4.932565288777084, + "tokens_seen": 79101952 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004929789368104313, + "loss": 1.9597, + "theoretical_loss": 4.93198015812211, + "tokens_seen": 79167488 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004929689067201605, + "loss": 1.9399, + "theoretical_loss": 4.931395647144881, + "tokens_seen": 79233024 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004929588766298897, + "loss": 1.8466, + "theoretical_loss": 4.930811754677391, + "tokens_seen": 79298560 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004929488465396189, + "loss": 2.1423, + "theoretical_loss": 4.930228479554799, + "tokens_seen": 79364096 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004929388164493481, + "loss": 1.841, + "theoretical_loss": 4.929645820615419, + "tokens_seen": 79429632 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004929287863590773, + "loss": 1.8528, + "theoretical_loss": 4.929063776700706, + "tokens_seen": 79495168 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004929187562688064, + "loss": 1.8544, + "theoretical_loss": 4.928482346655246, + "tokens_seen": 79560704 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004929087261785356, + "loss": 1.8169, + "theoretical_loss": 4.927901529326744, + "tokens_seen": 79626240 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004928986960882648, + "loss": 1.9464, + "theoretical_loss": 4.927321323566017, + "tokens_seen": 79691776 + }, + { + "epoch": 0.02, + "learning_rate": 0.000492888665997994, + "loss": 1.9494, + "theoretical_loss": 4.926741728226978, + "tokens_seen": 79757312 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004928786359077232, + "loss": 1.9882, + "theoretical_loss": 4.926162742166628, + "tokens_seen": 79822848 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004928686058174523, + "loss": 1.8223, + "theoretical_loss": 4.925584364245044, + "tokens_seen": 79888384 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004928585757271815, + "loss": 1.883, + "theoretical_loss": 4.925006593325367, + "tokens_seen": 79953920 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004928485456369107, + "loss": 1.9581, + "theoretical_loss": 4.924429428273798, + "tokens_seen": 80019456 + }, + { + "epoch": 0.02, + "learning_rate": 0.00049283851554664, + "loss": 1.9045, + "theoretical_loss": 4.923852867959576, + "tokens_seen": 80084992 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004928284854563691, + "loss": 2.0143, + "theoretical_loss": 4.923276911254978, + "tokens_seen": 80150528 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004928184553660984, + "loss": 1.9379, + "theoretical_loss": 4.9227015570353005, + "tokens_seen": 80216064 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": -0.0006200098432600498, + "objective/train/docs_used": 57296, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.1317315101623535, + "objective/train/original_loss": 4.131731986999512, + "objective/train/theoretical_loss": 4.9221268041788555, + "objective/train/tokens_used": 100741600, + "objective/train/value_avg": -0.00490570068359375, + "objective/train/value_loss": 8.11806385172531e-05, + "objective/train/value_max": -0.000293731689453125, + "objective/train/value_min": -0.0592041015625, + "objective/train/value_reward_corr": 0.06486236362378375, + "objective/train/value_std": 0.004177093505859375, + "objective/train/weight_avg": 0.9975805282592773, + "objective/train/weighted_lm_loss": 4.121804714202881, + "objective/train/weights_max": 1.8076740503311157, + "objective/train/weights_min": 0.2452181726694107, + "theoretical_loss": 4.9221268041788555, + "tokens_seen": 80281600 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004928084252758274, + "loss": 2.0193, + "theoretical_loss": 4.9221268041788555, + "tokens_seen": 80281600 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004927983951855567, + "loss": 1.7487, + "theoretical_loss": 4.921552651566954, + "tokens_seen": 80347136 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004927883650952859, + "loss": 1.9384, + "theoretical_loss": 4.920979098083901, + "tokens_seen": 80412672 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004927783350050151, + "loss": 1.9413, + "theoretical_loss": 4.920406142616981, + "tokens_seen": 80478208 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004927683049147443, + "loss": 1.7998, + "theoretical_loss": 4.919833784056449, + "tokens_seen": 80543744 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004927582748244734, + "loss": 1.7986, + "theoretical_loss": 4.919262021295519, + "tokens_seen": 80609280 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004927482447342026, + "loss": 1.8543, + "theoretical_loss": 4.918690853230357, + "tokens_seen": 80674816 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004927382146439318, + "loss": 1.8409, + "theoretical_loss": 4.918120278760069, + "tokens_seen": 80740352 + }, + { + "epoch": 0.02, + "learning_rate": 0.000492728184553661, + "loss": 1.848, + "theoretical_loss": 4.917550296786688, + "tokens_seen": 80805888 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004927181544633902, + "loss": 1.9698, + "theoretical_loss": 4.9169809062151675, + "tokens_seen": 80871424 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004927081243731193, + "loss": 1.8513, + "theoretical_loss": 4.916412105953371, + "tokens_seen": 80936960 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004926980942828485, + "loss": 2.0034, + "theoretical_loss": 4.915843894912058, + "tokens_seen": 81002496 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004926880641925777, + "loss": 1.8616, + "theoretical_loss": 4.915276272004883, + "tokens_seen": 81068032 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004926780341023069, + "loss": 1.8484, + "theoretical_loss": 4.914709236148374, + "tokens_seen": 81133568 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004926680040120361, + "loss": 1.7362, + "theoretical_loss": 4.9141427862619285, + "tokens_seen": 81199104 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004926579739217654, + "loss": 1.818, + "theoretical_loss": 4.913576921267806, + "tokens_seen": 81264640 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004926479438314944, + "loss": 1.841, + "theoretical_loss": 4.913011640091115, + "tokens_seen": 81330176 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004926379137412237, + "loss": 1.8679, + "theoretical_loss": 4.9124469416598, + "tokens_seen": 81395712 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004926278836509528, + "loss": 1.8958, + "theoretical_loss": 4.91188282490464, + "tokens_seen": 81461248 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004926178535606821, + "loss": 1.7645, + "theoretical_loss": 4.9113192887592305, + "tokens_seen": 81526784 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004926078234704113, + "loss": 1.8085, + "theoretical_loss": 4.91075633215998, + "tokens_seen": 81592320 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004925977933801404, + "loss": 1.8655, + "theoretical_loss": 4.910193954046093, + "tokens_seen": 81657856 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004925877632898696, + "loss": 1.8253, + "theoretical_loss": 4.9096321533595715, + "tokens_seen": 81723392 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004925777331995988, + "loss": 1.9131, + "theoretical_loss": 4.909070929045194, + "tokens_seen": 81788928 + }, + { + "epoch": 0.02, + "learning_rate": 0.000492567703109328, + "loss": 1.836, + "theoretical_loss": 4.908510280050512, + "tokens_seen": 81854464 + }, + { + "epoch": 0.02, + "objective/train/advantage_avg": -0.0008281050832010806, + "objective/train/docs_used": 58605, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.166916370391846, + "objective/train/original_loss": 4.166916370391846, + "objective/train/theoretical_loss": 4.907950205325841, + "objective/train/tokens_used": 102380000, + "objective/train/value_avg": -0.005367279052734375, + "objective/train/value_loss": 0.00010128566646017134, + "objective/train/value_max": -0.0001150369644165039, + "objective/train/value_min": -0.140869140625, + "objective/train/value_reward_corr": 0.1170327068579143, + "objective/train/value_std": 0.00623321533203125, + "objective/train/weight_avg": 0.9967740178108215, + "objective/train/weighted_lm_loss": 4.168039321899414, + "objective/train/weights_max": 4.090599060058594, + "objective/train/weights_min": 0.28894829750061035, + "theoretical_loss": 4.907950205325841, + "tokens_seen": 81920000 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004925576730190572, + "loss": 2.0228, + "theoretical_loss": 4.907950205325841, + "tokens_seen": 81920000 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004925476429287864, + "loss": 1.8024, + "theoretical_loss": 4.907390703824248, + "tokens_seen": 81985536 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004925376128385155, + "loss": 2.01, + "theoretical_loss": 4.906831774501544, + "tokens_seen": 82051072 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004925275827482447, + "loss": 1.9374, + "theoretical_loss": 4.9062734163162744, + "tokens_seen": 82116608 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004925175526579739, + "loss": 1.8054, + "theoretical_loss": 4.905715628229709, + "tokens_seen": 82182144 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004925075225677031, + "loss": 1.8938, + "theoretical_loss": 4.905158409205833, + "tokens_seen": 82247680 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004924974924774323, + "loss": 2.0212, + "theoretical_loss": 4.904601758211339, + "tokens_seen": 82313216 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004924874623871615, + "loss": 1.8964, + "theoretical_loss": 4.904045674215614, + "tokens_seen": 82378752 + }, + { + "epoch": 0.02, + "learning_rate": 0.0004924774322968907, + "loss": 1.8887, + "theoretical_loss": 4.9034901561907365, + "tokens_seen": 82444288 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004924674022066199, + "loss": 1.8136, + "theoretical_loss": 4.902935203111462, + "tokens_seen": 82509824 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004924573721163491, + "loss": 1.6795, + "theoretical_loss": 4.902380813955215, + "tokens_seen": 82575360 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004924473420260783, + "loss": 1.8711, + "theoretical_loss": 4.901826987702082, + "tokens_seen": 82640896 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004924373119358075, + "loss": 1.9964, + "theoretical_loss": 4.9012737233348, + "tokens_seen": 82706432 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004924272818455366, + "loss": 2.1167, + "theoretical_loss": 4.900721019838748, + "tokens_seen": 82771968 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004924172517552658, + "loss": 1.9, + "theoretical_loss": 4.90016887620194, + "tokens_seen": 82837504 + }, + { + "epoch": 0.03, + "learning_rate": 0.000492407221664995, + "loss": 1.8639, + "theoretical_loss": 4.899617291415016, + "tokens_seen": 82903040 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004923971915747242, + "loss": 1.7801, + "theoretical_loss": 4.899066264471228, + "tokens_seen": 82968576 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004923871614844534, + "loss": 2.0203, + "theoretical_loss": 4.898515794366439, + "tokens_seen": 83034112 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004923771313941825, + "loss": 1.8518, + "theoretical_loss": 4.897965880099106, + "tokens_seen": 83099648 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004923671013039117, + "loss": 2.0382, + "theoretical_loss": 4.897416520670279, + "tokens_seen": 83165184 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004923570712136409, + "loss": 1.8375, + "theoretical_loss": 4.896867715083587, + "tokens_seen": 83230720 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004923470411233702, + "loss": 1.6803, + "theoretical_loss": 4.896319462345232, + "tokens_seen": 83296256 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004923370110330993, + "loss": 1.8352, + "theoretical_loss": 4.895771761463979, + "tokens_seen": 83361792 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004923269809428286, + "loss": 1.841, + "theoretical_loss": 4.895224611451146, + "tokens_seen": 83427328 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004923169508525576, + "loss": 1.8714, + "theoretical_loss": 4.8946780113206, + "tokens_seen": 83492864 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.002330695278942585, + "objective/train/docs_used": 59310, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.9291229248046875, + "objective/train/original_loss": 3.9291226863861084, + "objective/train/theoretical_loss": 4.894131960088744, + "objective/train/tokens_used": 104018400, + "objective/train/value_avg": -0.004459381103515625, + "objective/train/value_loss": 0.00015995626745279878, + "objective/train/value_max": -0.00016605854034423828, + "objective/train/value_min": -0.1812744140625, + "objective/train/value_reward_corr": 0.18213510784478068, + "objective/train/value_std": 0.00539398193359375, + "objective/train/weight_avg": 0.9838688373565674, + "objective/train/weighted_lm_loss": 3.8833889961242676, + "objective/train/weights_max": 4.648962497711182, + "objective/train/weights_min": 0.29077136516571045, + "theoretical_loss": 4.894131960088744, + "tokens_seen": 83558400 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004923069207622869, + "loss": 1.9377, + "theoretical_loss": 4.894131960088744, + "tokens_seen": 83558400 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004922968906720161, + "loss": 1.9144, + "theoretical_loss": 4.89358645677451, + "tokens_seen": 83623936 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004922868605817453, + "loss": 1.8848, + "theoretical_loss": 4.8930415003993515, + "tokens_seen": 83689472 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004922768304914745, + "loss": 1.881, + "theoretical_loss": 4.892497089987232, + "tokens_seen": 83755008 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004922668004012036, + "loss": 1.9126, + "theoretical_loss": 4.891953224564623, + "tokens_seen": 83820544 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004922567703109328, + "loss": 2.0439, + "theoretical_loss": 4.891409903160486, + "tokens_seen": 83886080 + }, + { + "epoch": 0.03, + "learning_rate": 0.000492246740220662, + "loss": 1.6764, + "theoretical_loss": 4.890867124806276, + "tokens_seen": 83951616 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004922367101303912, + "loss": 1.9329, + "theoretical_loss": 4.890324888535918, + "tokens_seen": 84017152 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004922266800401204, + "loss": 1.8651, + "theoretical_loss": 4.889783193385815, + "tokens_seen": 84082688 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004922166499498495, + "loss": 1.8866, + "theoretical_loss": 4.88924203839483, + "tokens_seen": 84148224 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004922066198595787, + "loss": 1.9714, + "theoretical_loss": 4.88870142260428, + "tokens_seen": 84213760 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004921965897693079, + "loss": 1.8985, + "theoretical_loss": 4.888161345057926, + "tokens_seen": 84279296 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004921865596790371, + "loss": 1.801, + "theoretical_loss": 4.88762180480197, + "tokens_seen": 84344832 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004921765295887663, + "loss": 1.8487, + "theoretical_loss": 4.8870828008850395, + "tokens_seen": 84410368 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004921664994984956, + "loss": 1.7548, + "theoretical_loss": 4.886544332358189, + "tokens_seen": 84475904 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004921564694082246, + "loss": 1.7552, + "theoretical_loss": 4.886006398274881, + "tokens_seen": 84541440 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004921464393179539, + "loss": 1.7885, + "theoretical_loss": 4.885468997690987, + "tokens_seen": 84606976 + }, + { + "epoch": 0.03, + "learning_rate": 0.000492136409227683, + "loss": 1.8487, + "theoretical_loss": 4.884932129664774, + "tokens_seen": 84672512 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004921263791374123, + "loss": 1.8209, + "theoretical_loss": 4.8843957932569, + "tokens_seen": 84738048 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004921163490471415, + "loss": 1.7422, + "theoretical_loss": 4.883859987530404, + "tokens_seen": 84803584 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004921063189568706, + "loss": 1.7788, + "theoretical_loss": 4.8833247115507, + "tokens_seen": 84869120 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004920962888665998, + "loss": 1.9511, + "theoretical_loss": 4.882789964385566, + "tokens_seen": 84934656 + }, + { + "epoch": 0.03, + "learning_rate": 0.000492086258776329, + "loss": 1.8807, + "theoretical_loss": 4.88225574510514, + "tokens_seen": 85000192 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004920762286860582, + "loss": 1.7785, + "theoretical_loss": 4.8817220527819085, + "tokens_seen": 85065728 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004920661985957874, + "loss": 1.9105, + "theoretical_loss": 4.881188886490703, + "tokens_seen": 85131264 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0010243294527754188, + "objective/train/docs_used": 60492, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.5053906440734863, + "objective/train/original_loss": 3.5053908824920654, + "objective/train/theoretical_loss": 4.880656245308686, + "objective/train/tokens_used": 105656800, + "objective/train/value_avg": -0.004566192626953125, + "objective/train/value_loss": 8.53120582178235e-05, + "objective/train/value_max": -0.00016605854034423828, + "objective/train/value_min": -0.0982666015625, + "objective/train/value_reward_corr": 0.10819632261235229, + "objective/train/value_std": 0.004253387451171875, + "objective/train/weight_avg": 0.9937407374382019, + "objective/train/weighted_lm_loss": 3.4720346927642822, + "objective/train/weights_max": 2.651299238204956, + "objective/train/weights_min": 0.24455957114696503, + "theoretical_loss": 4.880656245308686, + "tokens_seen": 85196800 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004920561685055166, + "loss": 1.867, + "theoretical_loss": 4.880656245308686, + "tokens_seen": 85196800 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004920461384152457, + "loss": 1.8684, + "theoretical_loss": 4.880124128315353, + "tokens_seen": 85262336 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004920361083249749, + "loss": 1.8716, + "theoretical_loss": 4.879592534592515, + "tokens_seen": 85327872 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004920260782347041, + "loss": 1.8236, + "theoretical_loss": 4.879061463224295, + "tokens_seen": 85393408 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004920160481444333, + "loss": 1.7845, + "theoretical_loss": 4.87853091329712, + "tokens_seen": 85458944 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004920060180541625, + "loss": 1.9577, + "theoretical_loss": 4.878000883899718, + "tokens_seen": 85524480 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004919959879638916, + "loss": 1.9209, + "theoretical_loss": 4.877471374123102, + "tokens_seen": 85590016 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004919859578736209, + "loss": 1.8677, + "theoretical_loss": 4.876942383060569, + "tokens_seen": 85655552 + }, + { + "epoch": 0.03, + "learning_rate": 0.00049197592778335, + "loss": 1.9069, + "theoretical_loss": 4.876413909807688, + "tokens_seen": 85721088 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004919658976930793, + "loss": 1.7822, + "theoretical_loss": 4.875885953462296, + "tokens_seen": 85786624 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004919558676028084, + "loss": 1.8006, + "theoretical_loss": 4.8753585131244925, + "tokens_seen": 85852160 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004919458375125377, + "loss": 1.7812, + "theoretical_loss": 4.874831587896624, + "tokens_seen": 85917696 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004919358074222668, + "loss": 1.8394, + "theoretical_loss": 4.874305176883285, + "tokens_seen": 85983232 + }, + { + "epoch": 0.03, + "learning_rate": 0.000491925777331996, + "loss": 1.6942, + "theoretical_loss": 4.873779279191307, + "tokens_seen": 86048768 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004919157472417252, + "loss": 1.9804, + "theoretical_loss": 4.87325389392975, + "tokens_seen": 86114304 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004919057171514544, + "loss": 1.8535, + "theoretical_loss": 4.872729020209899, + "tokens_seen": 86179840 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004918956870611836, + "loss": 1.9341, + "theoretical_loss": 4.872204657145253, + "tokens_seen": 86245376 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004918856569709127, + "loss": 1.9169, + "theoretical_loss": 4.871680803851521, + "tokens_seen": 86310912 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004918756268806419, + "loss": 1.8296, + "theoretical_loss": 4.8711574594466125, + "tokens_seen": 86376448 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004918655967903711, + "loss": 1.9237, + "theoretical_loss": 4.87063462305063, + "tokens_seen": 86441984 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004918555667001003, + "loss": 1.947, + "theoretical_loss": 4.870112293785865, + "tokens_seen": 86507520 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004918455366098295, + "loss": 1.8497, + "theoretical_loss": 4.869590470776787, + "tokens_seen": 86573056 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004918355065195586, + "loss": 1.8039, + "theoretical_loss": 4.869069153150041, + "tokens_seen": 86638592 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004918254764292878, + "loss": 1.773, + "theoretical_loss": 4.868548340034435, + "tokens_seen": 86704128 + }, + { + "epoch": 0.03, + "learning_rate": 0.000491815446339017, + "loss": 1.6949, + "theoretical_loss": 4.868028030560937, + "tokens_seen": 86769664 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0029073741752654314, + "objective/train/docs_used": 61005, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4351682662963867, + "objective/train/original_loss": 3.4351677894592285, + "objective/train/theoretical_loss": 4.867508223862666, + "objective/train/tokens_used": 107295200, + "objective/train/value_avg": -0.00420379638671875, + "objective/train/value_loss": 0.00010291047510690987, + "objective/train/value_max": -0.00014317035675048828, + "objective/train/value_min": -0.06573486328125, + "objective/train/value_reward_corr": 0.00190853537460672, + "objective/train/value_std": 0.003936767578125, + "objective/train/weight_avg": 0.9757164716720581, + "objective/train/weighted_lm_loss": 3.391615629196167, + "objective/train/weights_max": 1.9296692609786987, + "objective/train/weights_min": 0.3700840175151825, + "theoretical_loss": 4.867508223862666, + "tokens_seen": 86835200 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004918054162487463, + "loss": 1.8299, + "theoretical_loss": 4.867508223862666, + "tokens_seen": 86835200 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004917953861584754, + "loss": 1.8588, + "theoretical_loss": 4.866988919074889, + "tokens_seen": 86900736 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004917853560682047, + "loss": 1.8659, + "theoretical_loss": 4.866470115335007, + "tokens_seen": 86966272 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004917753259779337, + "loss": 1.6963, + "theoretical_loss": 4.865951811782555, + "tokens_seen": 87031808 + }, + { + "epoch": 0.03, + "learning_rate": 0.000491765295887663, + "loss": 1.7591, + "theoretical_loss": 4.865434007559189, + "tokens_seen": 87097344 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004917552657973922, + "loss": 1.724, + "theoretical_loss": 4.864916701808687, + "tokens_seen": 87162880 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004917452357071214, + "loss": 1.8306, + "theoretical_loss": 4.864399893676933, + "tokens_seen": 87228416 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004917352056168506, + "loss": 1.8436, + "theoretical_loss": 4.863883582311917, + "tokens_seen": 87293952 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004917251755265797, + "loss": 1.6312, + "theoretical_loss": 4.863367766863728, + "tokens_seen": 87359488 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004917151454363089, + "loss": 1.7838, + "theoretical_loss": 4.86285244648454, + "tokens_seen": 87425024 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004917051153460381, + "loss": 1.9164, + "theoretical_loss": 4.862337620328617, + "tokens_seen": 87490560 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004916950852557673, + "loss": 1.7117, + "theoretical_loss": 4.861823287552293, + "tokens_seen": 87556096 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004916850551654965, + "loss": 1.9437, + "theoretical_loss": 4.861309447313978, + "tokens_seen": 87621632 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004916750250752258, + "loss": 1.8487, + "theoretical_loss": 4.860796098774145, + "tokens_seen": 87687168 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004916649949849548, + "loss": 1.7712, + "theoretical_loss": 4.8602832410953205, + "tokens_seen": 87752704 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004916549648946841, + "loss": 1.8333, + "theoretical_loss": 4.859770873442082, + "tokens_seen": 87818240 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004916449348044132, + "loss": 1.9301, + "theoretical_loss": 4.859258994981058, + "tokens_seen": 87883776 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004916349047141425, + "loss": 1.8616, + "theoretical_loss": 4.858747604880904, + "tokens_seen": 87949312 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004916248746238717, + "loss": 1.8814, + "theoretical_loss": 4.858236702312313, + "tokens_seen": 88014848 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004916148445336008, + "loss": 1.8086, + "theoretical_loss": 4.857726286448001, + "tokens_seen": 88080384 + }, + { + "epoch": 0.03, + "learning_rate": 0.00049160481444333, + "loss": 1.8075, + "theoretical_loss": 4.857216356462702, + "tokens_seen": 88145920 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004915947843530592, + "loss": 1.9236, + "theoretical_loss": 4.856706911533163, + "tokens_seen": 88211456 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004915847542627884, + "loss": 1.783, + "theoretical_loss": 4.85619795083813, + "tokens_seen": 88276992 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004915747241725176, + "loss": 2.0414, + "theoretical_loss": 4.855689473558357, + "tokens_seen": 88342528 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004915646940822468, + "loss": 2.0851, + "theoretical_loss": 4.855181478876584, + "tokens_seen": 88408064 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": 0.0011010210728272796, + "objective/train/docs_used": 61597, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.8659422397613525, + "objective/train/original_loss": 3.8659420013427734, + "objective/train/theoretical_loss": 4.854673965977539, + "objective/train/tokens_used": 108933600, + "objective/train/value_avg": -0.00582122802734375, + "objective/train/value_loss": 0.00015187459939625114, + "objective/train/value_max": -0.0002868175506591797, + "objective/train/value_min": -0.406005859375, + "objective/train/value_reward_corr": 0.054094799613430104, + "objective/train/value_std": 0.01055145263671875, + "objective/train/weight_avg": 1.0247594118118286, + "objective/train/weighted_lm_loss": 3.938272476196289, + "objective/train/weights_max": 49.36723709106445, + "objective/train/weights_min": 0.5578980445861816, + "theoretical_loss": 4.854673965977539, + "tokens_seen": 88473600 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004915546639919759, + "loss": 1.9574, + "theoretical_loss": 4.854673965977539, + "tokens_seen": 88473600 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004915446339017051, + "loss": 1.7892, + "theoretical_loss": 4.85416693404793, + "tokens_seen": 88539136 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004915346038114343, + "loss": 1.8649, + "theoretical_loss": 4.8536603822764395, + "tokens_seen": 88604672 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004915245737211635, + "loss": 1.7523, + "theoretical_loss": 4.853154309853716, + "tokens_seen": 88670208 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004915145436308927, + "loss": 1.8139, + "theoretical_loss": 4.85264871597237, + "tokens_seen": 88735744 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004915045135406218, + "loss": 1.7844, + "theoretical_loss": 4.852143599826965, + "tokens_seen": 88801280 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004914944834503511, + "loss": 1.8576, + "theoretical_loss": 4.851638960614015, + "tokens_seen": 88866816 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004914844533600802, + "loss": 1.9099, + "theoretical_loss": 4.851134797531978, + "tokens_seen": 88932352 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004914744232698095, + "loss": 1.8946, + "theoretical_loss": 4.850631109781246, + "tokens_seen": 88997888 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004914643931795386, + "loss": 2.3187, + "theoretical_loss": 4.850127896564143, + "tokens_seen": 89063424 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004914543630892679, + "loss": 1.9438, + "theoretical_loss": 4.849625157084915, + "tokens_seen": 89128960 + }, + { + "epoch": 0.03, + "learning_rate": 0.000491444332998997, + "loss": 1.8697, + "theoretical_loss": 4.8491228905497294, + "tokens_seen": 89194496 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004914343029087262, + "loss": 1.9536, + "theoretical_loss": 4.848621096166663, + "tokens_seen": 89260032 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004914242728184554, + "loss": 1.8037, + "theoretical_loss": 4.848119773145701, + "tokens_seen": 89325568 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004914142427281846, + "loss": 1.8374, + "theoretical_loss": 4.847618920698728, + "tokens_seen": 89391104 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004914042126379138, + "loss": 1.8894, + "theoretical_loss": 4.847118538039522, + "tokens_seen": 89456640 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004913941825476429, + "loss": 1.8907, + "theoretical_loss": 4.846618624383751, + "tokens_seen": 89522176 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004913841524573721, + "loss": 1.8832, + "theoretical_loss": 4.846119178948966, + "tokens_seen": 89587712 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004913741223671013, + "loss": 1.913, + "theoretical_loss": 4.84562020095459, + "tokens_seen": 89653248 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004913640922768305, + "loss": 1.9766, + "theoretical_loss": 4.845121689621922, + "tokens_seen": 89718784 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004913540621865597, + "loss": 1.8256, + "theoretical_loss": 4.844623644174124, + "tokens_seen": 89784320 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004913440320962888, + "loss": 1.8067, + "theoretical_loss": 4.844126063836216, + "tokens_seen": 89849856 + }, + { + "epoch": 0.03, + "learning_rate": 0.000491334002006018, + "loss": 1.9421, + "theoretical_loss": 4.84362894783507, + "tokens_seen": 89915392 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004913239719157472, + "loss": 1.72, + "theoretical_loss": 4.8431322953994105, + "tokens_seen": 89980928 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004913139418254765, + "loss": 1.7719, + "theoretical_loss": 4.8426361057598, + "tokens_seen": 90046464 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0011066634906455874, + "objective/train/docs_used": 63098, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.686882734298706, + "objective/train/original_loss": 3.686882734298706, + "objective/train/theoretical_loss": 4.8421403781486365, + "objective/train/tokens_used": 110572000, + "objective/train/value_avg": -0.0045928955078125, + "objective/train/value_loss": 8.297982276417315e-05, + "objective/train/value_max": -0.0002758502960205078, + "objective/train/value_min": -0.2264404296875, + "objective/train/value_reward_corr": 0.05826503084635767, + "objective/train/value_std": 0.00495147705078125, + "objective/train/weight_avg": 0.9932082891464233, + "objective/train/weighted_lm_loss": 3.669044017791748, + "objective/train/weights_max": 9.625389099121094, + "objective/train/weights_min": 0.44298815727233887, + "theoretical_loss": 4.8421403781486365, + "tokens_seen": 90112000 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004913039117352056, + "loss": 1.8052, + "theoretical_loss": 4.8421403781486365, + "tokens_seen": 90112000 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004912938816449349, + "loss": 1.6751, + "theoretical_loss": 4.8416451118001484, + "tokens_seen": 90177536 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004912838515546639, + "loss": 1.6941, + "theoretical_loss": 4.841150305950391, + "tokens_seen": 90243072 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004912738214643932, + "loss": 1.662, + "theoretical_loss": 4.840655959837236, + "tokens_seen": 90308608 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004912637913741224, + "loss": 1.7549, + "theoretical_loss": 4.8401620727003705, + "tokens_seen": 90374144 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004912537612838516, + "loss": 1.7625, + "theoretical_loss": 4.839668643781287, + "tokens_seen": 90439680 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004912437311935808, + "loss": 1.8209, + "theoretical_loss": 4.839175672323279, + "tokens_seen": 90505216 + }, + { + "epoch": 0.03, + "learning_rate": 0.00049123370110331, + "loss": 1.7895, + "theoretical_loss": 4.838683157571442, + "tokens_seen": 90570752 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004912236710130391, + "loss": 1.8075, + "theoretical_loss": 4.838191098772656, + "tokens_seen": 90636288 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004912136409227683, + "loss": 1.7965, + "theoretical_loss": 4.837699495175589, + "tokens_seen": 90701824 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004912036108324975, + "loss": 1.8405, + "theoretical_loss": 4.837208346030691, + "tokens_seen": 90767360 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004911935807422267, + "loss": 1.7205, + "theoretical_loss": 4.836717650590181, + "tokens_seen": 90832896 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004911835506519559, + "loss": 2.0265, + "theoretical_loss": 4.836227408108052, + "tokens_seen": 90898432 + }, + { + "epoch": 0.03, + "learning_rate": 0.000491173520561685, + "loss": 1.823, + "theoretical_loss": 4.835737617840058, + "tokens_seen": 90963968 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004911634904714142, + "loss": 1.8125, + "theoretical_loss": 4.835248279043711, + "tokens_seen": 91029504 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004911534603811434, + "loss": 1.9062, + "theoretical_loss": 4.834759390978274, + "tokens_seen": 91095040 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004911434302908726, + "loss": 1.9366, + "theoretical_loss": 4.834270952904761, + "tokens_seen": 91160576 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004911334002006019, + "loss": 1.9823, + "theoretical_loss": 4.833782964085925, + "tokens_seen": 91226112 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004911233701103309, + "loss": 1.8861, + "theoretical_loss": 4.8332954237862555, + "tokens_seen": 91291648 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004911133400200602, + "loss": 1.8113, + "theoretical_loss": 4.832808331271973, + "tokens_seen": 91357184 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004911033099297893, + "loss": 1.9627, + "theoretical_loss": 4.832321685811024, + "tokens_seen": 91422720 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004910932798395186, + "loss": 1.9915, + "theoretical_loss": 4.831835486673079, + "tokens_seen": 91488256 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004910832497492478, + "loss": 1.756, + "theoretical_loss": 4.831349733129516, + "tokens_seen": 91553792 + }, + { + "epoch": 0.03, + "learning_rate": 0.000491073219658977, + "loss": 1.8357, + "theoretical_loss": 4.83086442445343, + "tokens_seen": 91619328 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004910631895687061, + "loss": 1.9421, + "theoretical_loss": 4.830379559919616, + "tokens_seen": 91684864 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.002883288310840726, + "objective/train/docs_used": 63840, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.67038893699646, + "objective/train/original_loss": 3.67038893699646, + "objective/train/theoretical_loss": 4.829895138804573, + "objective/train/tokens_used": 112210400, + "objective/train/value_avg": -0.004024505615234375, + "objective/train/value_loss": 0.00011046992585761473, + "objective/train/value_max": -0.00017404556274414062, + "objective/train/value_min": -0.1300048828125, + "objective/train/value_reward_corr": 0.07343504722860769, + "objective/train/value_std": 0.00400543212890625, + "objective/train/weight_avg": 0.9762313961982727, + "objective/train/weighted_lm_loss": 3.5723280906677246, + "objective/train/weights_max": 3.575392007827759, + "objective/train/weights_min": 0.2910293638706207, + "theoretical_loss": 4.829895138804573, + "tokens_seen": 91750400 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004910531594784353, + "loss": 1.7923, + "theoretical_loss": 4.829895138804573, + "tokens_seen": 91750400 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004910431293881645, + "loss": 1.7353, + "theoretical_loss": 4.829411160386489, + "tokens_seen": 91815936 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004910330992978937, + "loss": 1.7847, + "theoretical_loss": 4.828927623945246, + "tokens_seen": 91881472 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004910230692076229, + "loss": 1.7397, + "theoretical_loss": 4.828444528762407, + "tokens_seen": 91947008 + }, + { + "epoch": 0.03, + "learning_rate": 0.000491013039117352, + "loss": 1.9279, + "theoretical_loss": 4.827961874121215, + "tokens_seen": 92012544 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004910030090270812, + "loss": 1.8329, + "theoretical_loss": 4.827479659306585, + "tokens_seen": 92078080 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004909929789368104, + "loss": 1.9189, + "theoretical_loss": 4.826997883605104, + "tokens_seen": 92143616 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004909829488465397, + "loss": 1.7439, + "theoretical_loss": 4.826516546305019, + "tokens_seen": 92209152 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004909729187562688, + "loss": 1.8183, + "theoretical_loss": 4.826035646696238, + "tokens_seen": 92274688 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004909628886659981, + "loss": 1.9833, + "theoretical_loss": 4.825555184070321, + "tokens_seen": 92340224 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004909528585757272, + "loss": 1.9386, + "theoretical_loss": 4.825075157720478, + "tokens_seen": 92405760 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004909428284854564, + "loss": 1.7678, + "theoretical_loss": 4.824595566941559, + "tokens_seen": 92471296 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004909327983951856, + "loss": 1.7819, + "theoretical_loss": 4.824116411030056, + "tokens_seen": 92536832 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004909227683049148, + "loss": 1.8404, + "theoretical_loss": 4.823637689284094, + "tokens_seen": 92602368 + }, + { + "epoch": 0.03, + "learning_rate": 0.000490912738214644, + "loss": 1.8291, + "theoretical_loss": 4.823159401003425, + "tokens_seen": 92667904 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004909027081243731, + "loss": 1.8057, + "theoretical_loss": 4.8226815454894245, + "tokens_seen": 92733440 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004908926780341023, + "loss": 1.8049, + "theoretical_loss": 4.822204122045089, + "tokens_seen": 92798976 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004908826479438315, + "loss": 1.8346, + "theoretical_loss": 4.821727129975027, + "tokens_seen": 92864512 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004908726178535607, + "loss": 1.7228, + "theoretical_loss": 4.821250568585455, + "tokens_seen": 92930048 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004908625877632899, + "loss": 1.7965, + "theoretical_loss": 4.8207744371841965, + "tokens_seen": 92995584 + }, + { + "epoch": 0.03, + "learning_rate": 0.000490852557673019, + "loss": 1.7783, + "theoretical_loss": 4.820298735080673, + "tokens_seen": 93061120 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004908425275827482, + "loss": 1.775, + "theoretical_loss": 4.819823461585898, + "tokens_seen": 93126656 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004908324974924774, + "loss": 1.7189, + "theoretical_loss": 4.819348616012476, + "tokens_seen": 93192192 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004908224674022067, + "loss": 1.9796, + "theoretical_loss": 4.818874197674601, + "tokens_seen": 93257728 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004908124373119358, + "loss": 1.7489, + "theoretical_loss": 4.8184002058880395, + "tokens_seen": 93323264 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0006242537638172507, + "objective/train/docs_used": 65224, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.583801031112671, + "objective/train/original_loss": 3.583800792694092, + "objective/train/theoretical_loss": 4.817926639970137, + "objective/train/tokens_used": 113848800, + "objective/train/value_avg": -0.003902435302734375, + "objective/train/value_loss": 7.408454257529229e-05, + "objective/train/value_max": -7.545948028564453e-05, + "objective/train/value_min": -0.12054443359375, + "objective/train/value_reward_corr": 0.035096100518476, + "objective/train/value_std": 0.004268646240234375, + "objective/train/weight_avg": 0.9972555637359619, + "objective/train/weighted_lm_loss": 3.5843794345855713, + "objective/train/weights_max": 2.7923202514648438, + "objective/train/weights_min": 0.24406330287456512, + "theoretical_loss": 4.817926639970137, + "tokens_seen": 93388800 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004908024072216651, + "loss": 1.8261, + "theoretical_loss": 4.817926639970137, + "tokens_seen": 93388800 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004907923771313941, + "loss": 1.7712, + "theoretical_loss": 4.817453499239813, + "tokens_seen": 93454336 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004907823470411234, + "loss": 1.7154, + "theoretical_loss": 4.816980783017545, + "tokens_seen": 93519872 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004907723169508526, + "loss": 1.8064, + "theoretical_loss": 4.8165084906253774, + "tokens_seen": 93585408 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004907622868605818, + "loss": 1.7602, + "theoretical_loss": 4.81603662138691, + "tokens_seen": 93650944 + }, + { + "epoch": 0.03, + "learning_rate": 0.000490752256770311, + "loss": 1.748, + "theoretical_loss": 4.815565174627296, + "tokens_seen": 93716480 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004907422266800401, + "loss": 1.8759, + "theoretical_loss": 4.815094149673232, + "tokens_seen": 93782016 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004907321965897693, + "loss": 1.8517, + "theoretical_loss": 4.8146235458529585, + "tokens_seen": 93847552 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004907221664994985, + "loss": 1.785, + "theoretical_loss": 4.814153362496257, + "tokens_seen": 93913088 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004907121364092277, + "loss": 1.6559, + "theoretical_loss": 4.813683598934439, + "tokens_seen": 93978624 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004907021063189569, + "loss": 1.7639, + "theoretical_loss": 4.813214254500345, + "tokens_seen": 94044160 + }, + { + "epoch": 0.03, + "learning_rate": 0.000490692076228686, + "loss": 1.8572, + "theoretical_loss": 4.812745328528342, + "tokens_seen": 94109696 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004906820461384152, + "loss": 1.7958, + "theoretical_loss": 4.812276820354314, + "tokens_seen": 94175232 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004906720160481444, + "loss": 1.7462, + "theoretical_loss": 4.811808729315663, + "tokens_seen": 94240768 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004906619859578736, + "loss": 1.5742, + "theoretical_loss": 4.811341054751299, + "tokens_seen": 94306304 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004906519558676028, + "loss": 1.783, + "theoretical_loss": 4.810873796001641, + "tokens_seen": 94371840 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004906419257773321, + "loss": 1.9637, + "theoretical_loss": 4.810406952408607, + "tokens_seen": 94437376 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004906318956870611, + "loss": 1.764, + "theoretical_loss": 4.809940523315612, + "tokens_seen": 94502912 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004906218655967904, + "loss": 1.8315, + "theoretical_loss": 4.809474508067569, + "tokens_seen": 94568448 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004906118355065195, + "loss": 1.6623, + "theoretical_loss": 4.809008906010872, + "tokens_seen": 94633984 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004906018054162488, + "loss": 1.866, + "theoretical_loss": 4.808543716493404, + "tokens_seen": 94699520 + }, + { + "epoch": 0.03, + "learning_rate": 0.000490591775325978, + "loss": 1.9562, + "theoretical_loss": 4.808078938864527, + "tokens_seen": 94765056 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004905817452357072, + "loss": 1.7813, + "theoretical_loss": 4.8076145724750745, + "tokens_seen": 94830592 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004905717151454363, + "loss": 1.8154, + "theoretical_loss": 4.807150616677357, + "tokens_seen": 94896128 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004905616850551655, + "loss": 1.8249, + "theoretical_loss": 4.806687070825147, + "tokens_seen": 94961664 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.001263303798623383, + "objective/train/docs_used": 65494, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.833336114883423, + "objective/train/original_loss": 3.833336353302002, + "objective/train/theoretical_loss": 4.80622393427368, + "objective/train/tokens_used": 115487200, + "objective/train/value_avg": -0.00431060791015625, + "objective/train/value_loss": 7.592602196382359e-05, + "objective/train/value_max": -0.0001767873764038086, + "objective/train/value_min": -0.06744384765625, + "objective/train/value_reward_corr": 0.16588962337716126, + "objective/train/value_std": 0.00395965576171875, + "objective/train/weight_avg": 0.9908163547515869, + "objective/train/weighted_lm_loss": 3.8045289516448975, + "objective/train/weights_max": 1.8137328624725342, + "objective/train/weights_min": 0.08257336169481277, + "theoretical_loss": 4.80622393427368, + "tokens_seen": 95027200 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004905516549648947, + "loss": 1.7644, + "theoretical_loss": 4.80622393427368, + "tokens_seen": 95027200 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004905416248746239, + "loss": 1.8021, + "theoretical_loss": 4.80576120637965, + "tokens_seen": 95092736 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004905315947843531, + "loss": 1.8251, + "theoretical_loss": 4.8052988865012045, + "tokens_seen": 95158272 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004905215646940822, + "loss": 1.7186, + "theoretical_loss": 4.80483697399794, + "tokens_seen": 95223808 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004905115346038114, + "loss": 1.7912, + "theoretical_loss": 4.804375468230898, + "tokens_seen": 95289344 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004905015045135406, + "loss": 1.7483, + "theoretical_loss": 4.803914368562561, + "tokens_seen": 95354880 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004904914744232698, + "loss": 1.888, + "theoretical_loss": 4.803453674356847, + "tokens_seen": 95420416 + }, + { + "epoch": 0.03, + "learning_rate": 0.000490481444332999, + "loss": 1.8223, + "theoretical_loss": 4.802993384979108, + "tokens_seen": 95485952 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004904714142427281, + "loss": 1.8646, + "theoretical_loss": 4.802533499796123, + "tokens_seen": 95551488 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004904613841524574, + "loss": 1.7791, + "theoretical_loss": 4.8020740181760955, + "tokens_seen": 95617024 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004904513540621865, + "loss": 1.7023, + "theoretical_loss": 4.8016149394886485, + "tokens_seen": 95682560 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004904413239719158, + "loss": 1.8372, + "theoretical_loss": 4.80115626310482, + "tokens_seen": 95748096 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004904312938816449, + "loss": 1.7993, + "theoretical_loss": 4.800697988397061, + "tokens_seen": 95813632 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004904212637913742, + "loss": 1.8462, + "theoretical_loss": 4.800240114739227, + "tokens_seen": 95879168 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004904112337011033, + "loss": 1.835, + "theoretical_loss": 4.799782641506579, + "tokens_seen": 95944704 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004904012036108325, + "loss": 1.7241, + "theoretical_loss": 4.799325568075776, + "tokens_seen": 96010240 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004903911735205617, + "loss": 1.8207, + "theoretical_loss": 4.798868893824874, + "tokens_seen": 96075776 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004903811434302909, + "loss": 1.8811, + "theoretical_loss": 4.798412618133318, + "tokens_seen": 96141312 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004903711133400201, + "loss": 1.9207, + "theoretical_loss": 4.797956740381938, + "tokens_seen": 96206848 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004903610832497492, + "loss": 1.7925, + "theoretical_loss": 4.797501259952952, + "tokens_seen": 96272384 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004903510531594784, + "loss": 1.9193, + "theoretical_loss": 4.797046176229951, + "tokens_seen": 96337920 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004903410230692076, + "loss": 1.7513, + "theoretical_loss": 4.796591488597907, + "tokens_seen": 96403456 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004903309929789368, + "loss": 1.7104, + "theoretical_loss": 4.796137196443157, + "tokens_seen": 96468992 + }, + { + "epoch": 0.03, + "learning_rate": 0.000490320962888666, + "loss": 1.8867, + "theoretical_loss": 4.795683299153408, + "tokens_seen": 96534528 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004903109327983952, + "loss": 1.8636, + "theoretical_loss": 4.795229796117728, + "tokens_seen": 96600064 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0013454204890877008, + "objective/train/docs_used": 66055, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.7626914978027344, + "objective/train/original_loss": 3.762690782546997, + "objective/train/theoretical_loss": 4.7947766867265464, + "objective/train/tokens_used": 117125600, + "objective/train/value_avg": -0.0054779052734375, + "objective/train/value_loss": 0.00013774906983599067, + "objective/train/value_max": -0.0002551078796386719, + "objective/train/value_min": -0.19091796875, + "objective/train/value_reward_corr": 0.05072681714511358, + "objective/train/value_std": 0.00787353515625, + "objective/train/weight_avg": 0.9943771958351135, + "objective/train/weighted_lm_loss": 3.7166779041290283, + "objective/train/weights_max": 6.002528667449951, + "objective/train/weights_min": 0.41104039549827576, + "theoretical_loss": 4.7947766867265464, + "tokens_seen": 96665600 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004903009027081243, + "loss": 1.8331, + "theoretical_loss": 4.7947766867265464, + "tokens_seen": 96665600 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004902908726178535, + "loss": 1.8497, + "theoretical_loss": 4.794323970371646, + "tokens_seen": 96731136 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004902808425275828, + "loss": 1.7979, + "theoretical_loss": 4.793871646446159, + "tokens_seen": 96796672 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004902708124373119, + "loss": 1.7081, + "theoretical_loss": 4.793419714344568, + "tokens_seen": 96862208 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004902607823470412, + "loss": 1.7897, + "theoretical_loss": 4.792968173462697, + "tokens_seen": 96927744 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004902507522567703, + "loss": 1.8505, + "theoretical_loss": 4.792517023197708, + "tokens_seen": 96993280 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004902407221664995, + "loss": 1.854, + "theoretical_loss": 4.792066262948102, + "tokens_seen": 97058816 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004902306920762287, + "loss": 1.8711, + "theoretical_loss": 4.7916158921137075, + "tokens_seen": 97124352 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004902206619859579, + "loss": 1.7531, + "theoretical_loss": 4.791165910095685, + "tokens_seen": 97189888 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004902106318956871, + "loss": 1.8396, + "theoretical_loss": 4.790716316296514, + "tokens_seen": 97255424 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004902006018054163, + "loss": 1.8751, + "theoretical_loss": 4.790267110119999, + "tokens_seen": 97320960 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004901905717151454, + "loss": 1.6885, + "theoretical_loss": 4.789818290971258, + "tokens_seen": 97386496 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004901805416248746, + "loss": 1.7548, + "theoretical_loss": 4.7893698582567215, + "tokens_seen": 97452032 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004901705115346038, + "loss": 1.8676, + "theoretical_loss": 4.788921811384128, + "tokens_seen": 97517568 + }, + { + "epoch": 0.03, + "learning_rate": 0.000490160481444333, + "loss": 1.745, + "theoretical_loss": 4.788474149762525, + "tokens_seen": 97583104 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004901504513540623, + "loss": 1.7834, + "theoretical_loss": 4.7880268728022575, + "tokens_seen": 97648640 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004901404212637913, + "loss": 1.7906, + "theoretical_loss": 4.787579979914966, + "tokens_seen": 97714176 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004901303911735206, + "loss": 1.8223, + "theoretical_loss": 4.7871334705135915, + "tokens_seen": 97779712 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004901203610832497, + "loss": 1.8708, + "theoretical_loss": 4.78668734401236, + "tokens_seen": 97845248 + }, + { + "epoch": 0.03, + "learning_rate": 0.000490110330992979, + "loss": 1.8784, + "theoretical_loss": 4.786241599826784, + "tokens_seen": 97910784 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004901003009027082, + "loss": 1.6851, + "theoretical_loss": 4.78579623737366, + "tokens_seen": 97976320 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004900902708124374, + "loss": 1.8772, + "theoretical_loss": 4.785351256071063, + "tokens_seen": 98041856 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004900802407221665, + "loss": 1.657, + "theoretical_loss": 4.784906655338345, + "tokens_seen": 98107392 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004900702106318957, + "loss": 1.7935, + "theoretical_loss": 4.7844624345961275, + "tokens_seen": 98172928 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004900601805416249, + "loss": 1.8379, + "theoretical_loss": 4.7840185932663, + "tokens_seen": 98238464 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0010161909740418196, + "objective/train/docs_used": 66835, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.056639194488525, + "objective/train/original_loss": 4.056640148162842, + "objective/train/theoretical_loss": 4.783575130772016, + "objective/train/tokens_used": 118764000, + "objective/train/value_avg": -0.004566192626953125, + "objective/train/value_loss": 8.297957538161427e-05, + "objective/train/value_max": -0.0001780986785888672, + "objective/train/value_min": -0.07794189453125, + "objective/train/value_reward_corr": 0.03336605092111482, + "objective/train/value_std": 0.0038890838623046875, + "objective/train/weight_avg": 0.9937987327575684, + "objective/train/weighted_lm_loss": 4.014974594116211, + "objective/train/weights_max": 2.1802051067352295, + "objective/train/weights_min": 0.5664100050926208, + "theoretical_loss": 4.783575130772016, + "tokens_seen": 98304000 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004900501504513541, + "loss": 1.8419, + "theoretical_loss": 4.783575130772016, + "tokens_seen": 98304000 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004900401203610833, + "loss": 1.7234, + "theoretical_loss": 4.783132046537694, + "tokens_seen": 98369536 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004900300902708124, + "loss": 1.7872, + "theoretical_loss": 4.782689339989005, + "tokens_seen": 98435072 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004900200601805416, + "loss": 1.732, + "theoretical_loss": 4.782247010552876, + "tokens_seen": 98500608 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004900100300902708, + "loss": 1.7491, + "theoretical_loss": 4.781805057657483, + "tokens_seen": 98566144 + }, + { + "epoch": 0.03, + "learning_rate": 0.00049, + "loss": 1.8823, + "theoretical_loss": 4.7813634807322485, + "tokens_seen": 98631680 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004899899699097292, + "loss": 1.8404, + "theoretical_loss": 4.780922279207839, + "tokens_seen": 98697216 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004899799398194583, + "loss": 1.931, + "theoretical_loss": 4.780481452516161, + "tokens_seen": 98762752 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004899699097291876, + "loss": 1.8505, + "theoretical_loss": 4.780041000090356, + "tokens_seen": 98828288 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004899598796389167, + "loss": 1.8102, + "theoretical_loss": 4.779600921364795, + "tokens_seen": 98893824 + }, + { + "epoch": 0.03, + "learning_rate": 0.000489949849548646, + "loss": 1.7073, + "theoretical_loss": 4.779161215775083, + "tokens_seen": 98959360 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004899398194583751, + "loss": 1.8219, + "theoretical_loss": 4.778721882758049, + "tokens_seen": 99024896 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004899297893681044, + "loss": 1.7305, + "theoretical_loss": 4.7782829217517415, + "tokens_seen": 99090432 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004899197592778335, + "loss": 1.8306, + "theoretical_loss": 4.77784433219543, + "tokens_seen": 99155968 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004899097291875627, + "loss": 1.7996, + "theoretical_loss": 4.777406113529597, + "tokens_seen": 99221504 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004898996990972919, + "loss": 1.7791, + "theoretical_loss": 4.77696826519594, + "tokens_seen": 99287040 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004898896690070211, + "loss": 1.8398, + "theoretical_loss": 4.776530786637362, + "tokens_seen": 99352576 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004898796389167503, + "loss": 1.9486, + "theoretical_loss": 4.776093677297971, + "tokens_seen": 99418112 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004898696088264794, + "loss": 1.7682, + "theoretical_loss": 4.775656936623077, + "tokens_seen": 99483648 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004898595787362086, + "loss": 1.7799, + "theoretical_loss": 4.77522056405919, + "tokens_seen": 99549184 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004898495486459378, + "loss": 1.7602, + "theoretical_loss": 4.774784559054009, + "tokens_seen": 99614720 + }, + { + "epoch": 0.03, + "learning_rate": 0.000489839518555667, + "loss": 1.7937, + "theoretical_loss": 4.774348921056429, + "tokens_seen": 99680256 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004898294884653962, + "loss": 1.8092, + "theoretical_loss": 4.773913649516535, + "tokens_seen": 99745792 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004898194583751254, + "loss": 1.8167, + "theoretical_loss": 4.77347874388559, + "tokens_seen": 99811328 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004898094282848545, + "loss": 1.7978, + "theoretical_loss": 4.773044203616044, + "tokens_seen": 99876864 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.002438762690871954, + "objective/train/docs_used": 67525, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.446929693222046, + "objective/train/original_loss": 3.4469292163848877, + "objective/train/theoretical_loss": 4.7726100281615205, + "objective/train/tokens_used": 120402400, + "objective/train/value_avg": -0.004611968994140625, + "objective/train/value_loss": 0.00010555711924098432, + "objective/train/value_max": -0.00010973215103149414, + "objective/train/value_min": -0.1326904296875, + "objective/train/value_reward_corr": 0.024972286005578024, + "objective/train/value_std": 0.004222869873046875, + "objective/train/weight_avg": 0.980546772480011, + "objective/train/weighted_lm_loss": 3.3898255825042725, + "objective/train/weights_max": 3.5222771167755127, + "objective/train/weights_min": 0.1913357824087143, + "theoretical_loss": 4.7726100281615205, + "tokens_seen": 99942400 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004897993981945837, + "loss": 1.776, + "theoretical_loss": 4.7726100281615205, + "tokens_seen": 99942400 + }, + { + "epoch": 0.03, + "learning_rate": 0.000489789368104313, + "loss": 1.7132, + "theoretical_loss": 4.772176216976822, + "tokens_seen": 100007936 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004897793380140421, + "loss": 1.7784, + "theoretical_loss": 4.771742769517919, + "tokens_seen": 100073472 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004897693079237714, + "loss": 1.8034, + "theoretical_loss": 4.771309685241952, + "tokens_seen": 100139008 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004897592778335005, + "loss": 1.8507, + "theoretical_loss": 4.7708769636072255, + "tokens_seen": 100204544 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004897492477432297, + "loss": 1.796, + "theoretical_loss": 4.770444604073205, + "tokens_seen": 100270080 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004897392176529589, + "loss": 1.8264, + "theoretical_loss": 4.770012606100515, + "tokens_seen": 100335616 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004897291875626881, + "loss": 1.7649, + "theoretical_loss": 4.769580969150937, + "tokens_seen": 100401152 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004897191574724173, + "loss": 1.8028, + "theoretical_loss": 4.769149692687401, + "tokens_seen": 100466688 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004897091273821465, + "loss": 1.8577, + "theoretical_loss": 4.768718776173987, + "tokens_seen": 100532224 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004896990972918756, + "loss": 1.8489, + "theoretical_loss": 4.76828821907592, + "tokens_seen": 100597760 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004896890672016048, + "loss": 1.8333, + "theoretical_loss": 4.76785802085957, + "tokens_seen": 100663296 + }, + { + "epoch": 0.03, + "learning_rate": 0.000489679037111334, + "loss": 1.8004, + "theoretical_loss": 4.767428180992441, + "tokens_seen": 100728832 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004896690070210632, + "loss": 1.7834, + "theoretical_loss": 4.766998698943177, + "tokens_seen": 100794368 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004896589769307924, + "loss": 1.7246, + "theoretical_loss": 4.766569574181554, + "tokens_seen": 100859904 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004896489468405215, + "loss": 1.699, + "theoretical_loss": 4.766140806178475, + "tokens_seen": 100925440 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004896389167502507, + "loss": 1.7685, + "theoretical_loss": 4.765712394405973, + "tokens_seen": 100990976 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004896288866599799, + "loss": 1.7383, + "theoretical_loss": 4.765284338337203, + "tokens_seen": 101056512 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004896188565697091, + "loss": 1.9126, + "theoretical_loss": 4.7648566374464405, + "tokens_seen": 101122048 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004896088264794384, + "loss": 1.7617, + "theoretical_loss": 4.764429291209074, + "tokens_seen": 101187584 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004895987963891674, + "loss": 1.8618, + "theoretical_loss": 4.764002299101612, + "tokens_seen": 101253120 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004895887662988967, + "loss": 1.6813, + "theoretical_loss": 4.763575660601672, + "tokens_seen": 101318656 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004895787362086259, + "loss": 1.7347, + "theoretical_loss": 4.763149375187977, + "tokens_seen": 101384192 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004895687061183551, + "loss": 1.7367, + "theoretical_loss": 4.7627234423403575, + "tokens_seen": 101449728 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004895586760280843, + "loss": 1.9147, + "theoretical_loss": 4.762297861539745, + "tokens_seen": 101515264 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0012628933181986213, + "objective/train/docs_used": 68617, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.655167818069458, + "objective/train/original_loss": 3.655168056488037, + "objective/train/theoretical_loss": 4.761872632268167, + "objective/train/tokens_used": 122040800, + "objective/train/value_avg": -0.0046844482421875, + "objective/train/value_loss": 9.621829667594284e-05, + "objective/train/value_max": -0.0001926422119140625, + "objective/train/value_min": -0.1513671875, + "objective/train/value_reward_corr": 0.06009327734416299, + "objective/train/value_std": 0.00476837158203125, + "objective/train/weight_avg": 0.9918888211250305, + "objective/train/weighted_lm_loss": 3.646021842956543, + "objective/train/weights_max": 4.1627326011657715, + "objective/train/weights_min": 0.08314625918865204, + "theoretical_loss": 4.761872632268167, + "tokens_seen": 101580800 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004895486459378135, + "loss": 1.8119, + "theoretical_loss": 4.761872632268167, + "tokens_seen": 101580800 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004895386158475426, + "loss": 1.7308, + "theoretical_loss": 4.761447754008753, + "tokens_seen": 101646336 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004895285857572718, + "loss": 1.8016, + "theoretical_loss": 4.761023226245718, + "tokens_seen": 101711872 + }, + { + "epoch": 0.03, + "learning_rate": 0.000489518555667001, + "loss": 1.9187, + "theoretical_loss": 4.76059904846437, + "tokens_seen": 101777408 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004895085255767302, + "loss": 1.9496, + "theoretical_loss": 4.760175220151105, + "tokens_seen": 101842944 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004894984954864594, + "loss": 1.7342, + "theoretical_loss": 4.7597517407934, + "tokens_seen": 101908480 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004894884653961885, + "loss": 1.6501, + "theoretical_loss": 4.759328609879812, + "tokens_seen": 101974016 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004894784353059178, + "loss": 1.7782, + "theoretical_loss": 4.75890582689998, + "tokens_seen": 102039552 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004894684052156469, + "loss": 1.7958, + "theoretical_loss": 4.758483391344614, + "tokens_seen": 102105088 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004894583751253762, + "loss": 1.864, + "theoretical_loss": 4.758061302705496, + "tokens_seen": 102170624 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004894483450351053, + "loss": 1.7963, + "theoretical_loss": 4.757639560475478, + "tokens_seen": 102236160 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004894383149448346, + "loss": 1.9014, + "theoretical_loss": 4.757218164148476, + "tokens_seen": 102301696 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004894282848545637, + "loss": 1.7382, + "theoretical_loss": 4.756797113219473, + "tokens_seen": 102367232 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004894182547642929, + "loss": 1.7795, + "theoretical_loss": 4.756376407184508, + "tokens_seen": 102432768 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004894082246740221, + "loss": 1.7566, + "theoretical_loss": 4.755956045540676, + "tokens_seen": 102498304 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004893981945837513, + "loss": 1.6919, + "theoretical_loss": 4.755536027786134, + "tokens_seen": 102563840 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004893881644934805, + "loss": 1.8014, + "theoretical_loss": 4.75511635342008, + "tokens_seen": 102629376 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004893781344032096, + "loss": 1.731, + "theoretical_loss": 4.7546970219427696, + "tokens_seen": 102694912 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004893681043129388, + "loss": 1.7362, + "theoretical_loss": 4.754278032855497, + "tokens_seen": 102760448 + }, + { + "epoch": 0.03, + "learning_rate": 0.000489358074222668, + "loss": 1.7819, + "theoretical_loss": 4.753859385660604, + "tokens_seen": 102825984 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004893480441323972, + "loss": 1.8458, + "theoretical_loss": 4.75344107986147, + "tokens_seen": 102891520 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004893380140421264, + "loss": 1.774, + "theoretical_loss": 4.753023114962512, + "tokens_seen": 102957056 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004893279839518556, + "loss": 1.7003, + "theoretical_loss": 4.752605490469182, + "tokens_seen": 103022592 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004893179538615847, + "loss": 1.776, + "theoretical_loss": 4.752188205887961, + "tokens_seen": 103088128 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004893079237713139, + "loss": 1.8348, + "theoretical_loss": 4.751771260726363, + "tokens_seen": 103153664 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0020021332893520594, + "objective/train/docs_used": 69265, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.6259801387786865, + "objective/train/original_loss": 3.6259796619415283, + "objective/train/theoretical_loss": 4.751354654492923, + "objective/train/tokens_used": 123679200, + "objective/train/value_avg": -0.0036602020263671875, + "objective/train/value_loss": 7.886350795160979e-05, + "objective/train/value_max": -0.00013339519500732422, + "objective/train/value_min": -0.10125732421875, + "objective/train/value_reward_corr": 0.001650083964517551, + "objective/train/value_std": 0.0035839080810546875, + "objective/train/weight_avg": 0.9836885333061218, + "objective/train/weighted_lm_loss": 3.5808167457580566, + "objective/train/weights_max": 2.717618227005005, + "objective/train/weights_min": 0.3122745156288147, + "theoretical_loss": 4.751354654492923, + "tokens_seen": 103219200 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004892978936810432, + "loss": 1.7996, + "theoretical_loss": 4.751354654492923, + "tokens_seen": 103219200 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004892878635907723, + "loss": 1.7864, + "theoretical_loss": 4.750938386697204, + "tokens_seen": 103284736 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004892778335005016, + "loss": 1.8137, + "theoretical_loss": 4.750522456849783, + "tokens_seen": 103350272 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004892678034102307, + "loss": 1.918, + "theoretical_loss": 4.7501068644622615, + "tokens_seen": 103415808 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004892577733199599, + "loss": 1.7518, + "theoretical_loss": 4.74969160904725, + "tokens_seen": 103481344 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004892477432296891, + "loss": 1.8226, + "theoretical_loss": 4.749276690118377, + "tokens_seen": 103546880 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004892377131394183, + "loss": 1.7833, + "theoretical_loss": 4.7488621071902735, + "tokens_seen": 103612416 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004892276830491475, + "loss": 1.6744, + "theoretical_loss": 4.748447859778581, + "tokens_seen": 103677952 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004892176529588767, + "loss": 1.8289, + "theoretical_loss": 4.7480339473999456, + "tokens_seen": 103743488 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004892076228686058, + "loss": 1.7003, + "theoretical_loss": 4.747620369572012, + "tokens_seen": 103809024 + }, + { + "epoch": 0.03, + "learning_rate": 0.000489197592778335, + "loss": 1.7644, + "theoretical_loss": 4.747207125813423, + "tokens_seen": 103874560 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004891875626880642, + "loss": 1.8347, + "theoretical_loss": 4.746794215643821, + "tokens_seen": 103940096 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004891775325977934, + "loss": 1.8396, + "theoretical_loss": 4.746381638583839, + "tokens_seen": 104005632 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004891675025075226, + "loss": 1.8646, + "theoretical_loss": 4.745969394155098, + "tokens_seen": 104071168 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004891574724172517, + "loss": 1.6118, + "theoretical_loss": 4.74555748188021, + "tokens_seen": 104136704 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004891474423269809, + "loss": 1.8891, + "theoretical_loss": 4.745145901282772, + "tokens_seen": 104202240 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004891374122367101, + "loss": 1.7616, + "theoretical_loss": 4.744734651887361, + "tokens_seen": 104267776 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004891273821464393, + "loss": 1.8171, + "theoretical_loss": 4.744323733219536, + "tokens_seen": 104333312 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004891173520561686, + "loss": 1.7707, + "theoretical_loss": 4.743913144805833, + "tokens_seen": 104398848 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004891073219658976, + "loss": 1.8061, + "theoretical_loss": 4.74350288617376, + "tokens_seen": 104464384 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004890972918756269, + "loss": 1.7356, + "theoretical_loss": 4.7430929568517985, + "tokens_seen": 104529920 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004890872617853561, + "loss": 1.795, + "theoretical_loss": 4.742683356369401, + "tokens_seen": 104595456 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004890772316950853, + "loss": 1.6517, + "theoretical_loss": 4.742274084256983, + "tokens_seen": 104660992 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004890672016048145, + "loss": 1.8195, + "theoretical_loss": 4.7418651400459275, + "tokens_seen": 104726528 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004890571715145437, + "loss": 1.7975, + "theoretical_loss": 4.741456523268576, + "tokens_seen": 104792064 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0016873390413820744, + "objective/train/docs_used": 70556, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.486652135848999, + "objective/train/original_loss": 3.486652135848999, + "objective/train/theoretical_loss": 4.741048233458233, + "objective/train/tokens_used": 125317600, + "objective/train/value_avg": -0.004413604736328125, + "objective/train/value_loss": 0.00011342013021931052, + "objective/train/value_max": -0.00014543533325195312, + "objective/train/value_min": -0.124755859375, + "objective/train/value_reward_corr": 0.039567412469757396, + "objective/train/value_std": 0.004810333251953125, + "objective/train/weight_avg": 0.9885098934173584, + "objective/train/weighted_lm_loss": 3.45222806930542, + "objective/train/weights_max": 3.408240795135498, + "objective/train/weights_min": 0.49337080121040344, + "theoretical_loss": 4.741048233458233, + "tokens_seen": 104857600 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004890471414242728, + "loss": 1.7463, + "theoretical_loss": 4.741048233458233, + "tokens_seen": 104857600 + }, + { + "epoch": 0.03, + "learning_rate": 0.000489037111334002, + "loss": 1.8411, + "theoretical_loss": 4.7406402701491555, + "tokens_seen": 104923136 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004890270812437312, + "loss": 1.7196, + "theoretical_loss": 4.740232632876554, + "tokens_seen": 104988672 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004890170511534604, + "loss": 1.8317, + "theoretical_loss": 4.739825321176593, + "tokens_seen": 105054208 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004890070210631896, + "loss": 1.712, + "theoretical_loss": 4.739418334586386, + "tokens_seen": 105119744 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004889969909729187, + "loss": 1.6653, + "theoretical_loss": 4.739011672643988, + "tokens_seen": 105185280 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004889869608826479, + "loss": 1.6779, + "theoretical_loss": 4.738605334888403, + "tokens_seen": 105250816 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004889769307923771, + "loss": 1.724, + "theoretical_loss": 4.738199320859575, + "tokens_seen": 105316352 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004889669007021063, + "loss": 1.7363, + "theoretical_loss": 4.737793630098384, + "tokens_seen": 105381888 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004889568706118355, + "loss": 1.6754, + "theoretical_loss": 4.737388262146647, + "tokens_seen": 105447424 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004889468405215647, + "loss": 1.8596, + "theoretical_loss": 4.736983216547117, + "tokens_seen": 105512960 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004889368104312939, + "loss": 1.7533, + "theoretical_loss": 4.736578492843476, + "tokens_seen": 105578496 + }, + { + "epoch": 0.03, + "learning_rate": 0.000488926780341023, + "loss": 1.745, + "theoretical_loss": 4.736174090580336, + "tokens_seen": 105644032 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004889167502507523, + "loss": 1.7824, + "theoretical_loss": 4.735770009303232, + "tokens_seen": 105709568 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004889067201604815, + "loss": 1.653, + "theoretical_loss": 4.735366248558629, + "tokens_seen": 105775104 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004888966900702107, + "loss": 1.7637, + "theoretical_loss": 4.734962807893908, + "tokens_seen": 105840640 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004888866599799398, + "loss": 1.7981, + "theoretical_loss": 4.734559686857368, + "tokens_seen": 105906176 + }, + { + "epoch": 0.03, + "learning_rate": 0.000488876629889669, + "loss": 1.7705, + "theoretical_loss": 4.734156884998232, + "tokens_seen": 105971712 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004888665997993982, + "loss": 1.766, + "theoretical_loss": 4.733754401866626, + "tokens_seen": 106037248 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004888565697091274, + "loss": 1.6952, + "theoretical_loss": 4.7333522370136, + "tokens_seen": 106102784 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004888465396188566, + "loss": 1.7152, + "theoretical_loss": 4.732950389991101, + "tokens_seen": 106168320 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004888365095285858, + "loss": 1.7691, + "theoretical_loss": 4.732548860351992, + "tokens_seen": 106233856 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004888264794383149, + "loss": 1.6061, + "theoretical_loss": 4.732147647650036, + "tokens_seen": 106299392 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004888164493480441, + "loss": 1.7653, + "theoretical_loss": 4.731746751439898, + "tokens_seen": 106364928 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004888064192577733, + "loss": 1.7622, + "theoretical_loss": 4.731346171277146, + "tokens_seen": 106430464 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0030148255173116922, + "objective/train/docs_used": 70950, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.252803325653076, + "objective/train/original_loss": 4.252802848815918, + "objective/train/theoretical_loss": 4.73094590671824, + "objective/train/tokens_used": 126956000, + "objective/train/value_avg": -0.004852294921875, + "objective/train/value_loss": 0.00011471990001155064, + "objective/train/value_max": -0.0002065896987915039, + "objective/train/value_min": -0.1424560546875, + "objective/train/value_reward_corr": 0.04255470821864496, + "objective/train/value_std": 0.004497528076171875, + "objective/train/weight_avg": 0.9751693606376648, + "objective/train/weighted_lm_loss": 4.13572883605957, + "objective/train/weights_max": 4.0156378746032715, + "objective/train/weights_min": 0.24123552441596985, + "theoretical_loss": 4.73094590671824, + "tokens_seen": 106496000 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004887963891675025, + "loss": 1.7238, + "theoretical_loss": 4.73094590671824, + "tokens_seen": 106496000 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004887863590772317, + "loss": 1.7226, + "theoretical_loss": 4.730545957320541, + "tokens_seen": 106561536 + }, + { + "epoch": 0.03, + "learning_rate": 0.000488776328986961, + "loss": 1.8008, + "theoretical_loss": 4.730146322642297, + "tokens_seen": 106627072 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048876629889669, + "loss": 1.7629, + "theoretical_loss": 4.72974700224265, + "tokens_seen": 106692608 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004887562688064193, + "loss": 1.681, + "theoretical_loss": 4.729347995681627, + "tokens_seen": 106758144 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004887462387161484, + "loss": 1.798, + "theoretical_loss": 4.728949302520143, + "tokens_seen": 106823680 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004887362086258777, + "loss": 1.8057, + "theoretical_loss": 4.728550922319995, + "tokens_seen": 106889216 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004887261785356069, + "loss": 1.7104, + "theoretical_loss": 4.728152854643862, + "tokens_seen": 106954752 + }, + { + "epoch": 0.03, + "learning_rate": 0.000488716148445336, + "loss": 1.7322, + "theoretical_loss": 4.727755099055299, + "tokens_seen": 107020288 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004887061183550652, + "loss": 1.7739, + "theoretical_loss": 4.727357655118739, + "tokens_seen": 107085824 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004886960882647944, + "loss": 1.8392, + "theoretical_loss": 4.726960522399492, + "tokens_seen": 107151360 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004886860581745236, + "loss": 1.9112, + "theoretical_loss": 4.726563700463732, + "tokens_seen": 107216896 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004886760280842528, + "loss": 1.6827, + "theoretical_loss": 4.726167188878511, + "tokens_seen": 107282432 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004886659979939819, + "loss": 1.7846, + "theoretical_loss": 4.725770987211742, + "tokens_seen": 107347968 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004886559679037111, + "loss": 1.7786, + "theoretical_loss": 4.725375095032207, + "tokens_seen": 107413504 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004886459378134403, + "loss": 1.7498, + "theoretical_loss": 4.724979511909547, + "tokens_seen": 107479040 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004886359077231695, + "loss": 1.8101, + "theoretical_loss": 4.724584237414265, + "tokens_seen": 107544576 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004886258776328988, + "loss": 1.6643, + "theoretical_loss": 4.724189271117724, + "tokens_seen": 107610112 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004886158475426278, + "loss": 1.6911, + "theoretical_loss": 4.72379461259214, + "tokens_seen": 107675648 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004886058174523571, + "loss": 1.9131, + "theoretical_loss": 4.723400261410584, + "tokens_seen": 107741184 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004885957873620863, + "loss": 1.8349, + "theoretical_loss": 4.723006217146979, + "tokens_seen": 107806720 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004885857572718155, + "loss": 1.7256, + "theoretical_loss": 4.722612479376095, + "tokens_seen": 107872256 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004885757271815447, + "loss": 1.7579, + "theoretical_loss": 4.722219047673552, + "tokens_seen": 107937792 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004885656970912739, + "loss": 1.9905, + "theoretical_loss": 4.721825921615813, + "tokens_seen": 108003328 + }, + { + "epoch": 0.03, + "learning_rate": 0.000488555667001003, + "loss": 1.7359, + "theoretical_loss": 4.7214331007801835, + "tokens_seen": 108068864 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": 0.00024222057254519314, + "objective/train/docs_used": 72030, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.5735690593719482, + "objective/train/original_loss": 3.573568820953369, + "objective/train/theoretical_loss": 4.721040584744811, + "objective/train/tokens_used": 128594400, + "objective/train/value_avg": -0.00423431396484375, + "objective/train/value_loss": 3.9678194298176095e-05, + "objective/train/value_max": -0.00016868114471435547, + "objective/train/value_min": -0.053497314453125, + "objective/train/value_reward_corr": 0.10250441036706731, + "objective/train/value_std": 0.0038318634033203125, + "objective/train/weight_avg": 1.0043721199035645, + "objective/train/weighted_lm_loss": 3.5987274646759033, + "objective/train/weights_max": 1.5887565612792969, + "objective/train/weights_min": 0.5482495427131653, + "theoretical_loss": 4.721040584744811, + "tokens_seen": 108134400 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004885456369107322, + "loss": 1.874, + "theoretical_loss": 4.721040584744811, + "tokens_seen": 108134400 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004885356068204614, + "loss": 1.7648, + "theoretical_loss": 4.720648373088679, + "tokens_seen": 108199936 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004885255767301906, + "loss": 1.8297, + "theoretical_loss": 4.720256465391609, + "tokens_seen": 108265472 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004885155466399198, + "loss": 1.7984, + "theoretical_loss": 4.719864861234255, + "tokens_seen": 108331008 + }, + { + "epoch": 0.03, + "learning_rate": 0.000488505516549649, + "loss": 1.7855, + "theoretical_loss": 4.7194735601981055, + "tokens_seen": 108396544 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004884954864593781, + "loss": 1.7839, + "theoretical_loss": 4.719082561865474, + "tokens_seen": 108462080 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004884854563691073, + "loss": 1.747, + "theoretical_loss": 4.7186918658195065, + "tokens_seen": 108527616 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004884754262788365, + "loss": 1.8412, + "theoretical_loss": 4.718301471644171, + "tokens_seen": 108593152 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004884653961885657, + "loss": 1.7192, + "theoretical_loss": 4.71791137892426, + "tokens_seen": 108658688 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004884553660982949, + "loss": 1.7214, + "theoretical_loss": 4.717521587245386, + "tokens_seen": 108724224 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004884453360080241, + "loss": 1.6421, + "theoretical_loss": 4.717132096193982, + "tokens_seen": 108789760 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004884353059177532, + "loss": 1.767, + "theoretical_loss": 4.716742905357296, + "tokens_seen": 108855296 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004884252758274825, + "loss": 1.7033, + "theoretical_loss": 4.716354014323393, + "tokens_seen": 108920832 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004884152457372117, + "loss": 1.8452, + "theoretical_loss": 4.7159654226811485, + "tokens_seen": 108986368 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004884052156469409, + "loss": 1.6744, + "theoretical_loss": 4.71557713002025, + "tokens_seen": 109051904 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048839518555667, + "loss": 1.907, + "theoretical_loss": 4.715189135931194, + "tokens_seen": 109117440 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004883851554663992, + "loss": 1.769, + "theoretical_loss": 4.714801440005282, + "tokens_seen": 109182976 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004883751253761284, + "loss": 1.7618, + "theoretical_loss": 4.714414041834619, + "tokens_seen": 109248512 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004883650952858576, + "loss": 1.7704, + "theoretical_loss": 4.714026941012115, + "tokens_seen": 109314048 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004883550651955868, + "loss": 1.6664, + "theoretical_loss": 4.713640137131478, + "tokens_seen": 109379584 + }, + { + "epoch": 0.03, + "learning_rate": 0.000488345035105316, + "loss": 1.8391, + "theoretical_loss": 4.713253629787214, + "tokens_seen": 109445120 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004883350050150451, + "loss": 1.8377, + "theoretical_loss": 4.712867418574628, + "tokens_seen": 109510656 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004883249749247743, + "loss": 1.5987, + "theoretical_loss": 4.7124815030898155, + "tokens_seen": 109576192 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004883149448345035, + "loss": 1.712, + "theoretical_loss": 4.712095882929667, + "tokens_seen": 109641728 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004883049147442327, + "loss": 1.6264, + "theoretical_loss": 4.711710557691861, + "tokens_seen": 109707264 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0015746584394946694, + "objective/train/docs_used": 72698, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1268115043640137, + "objective/train/original_loss": 3.1268117427825928, + "objective/train/theoretical_loss": 4.711325526974863, + "objective/train/tokens_used": 130232800, + "objective/train/value_avg": -0.00421905517578125, + "objective/train/value_loss": 5.90582967561204e-05, + "objective/train/value_max": -0.0002359151840209961, + "objective/train/value_min": -0.11798095703125, + "objective/train/value_reward_corr": 0.058824685983334006, + "objective/train/value_std": 0.00421142578125, + "objective/train/weight_avg": 0.9872012138366699, + "objective/train/weighted_lm_loss": 3.085584878921509, + "objective/train/weights_max": 2.889387845993042, + "objective/train/weights_min": 0.6671463847160339, + "theoretical_loss": 4.711325526974863, + "tokens_seen": 109772800 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004882948846539619, + "loss": 1.7111, + "theoretical_loss": 4.711325526974863, + "tokens_seen": 109772800 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004882848545636911, + "loss": 1.7653, + "theoretical_loss": 4.710940790377929, + "tokens_seen": 109838336 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004882748244734203, + "loss": 1.7767, + "theoretical_loss": 4.710556347501095, + "tokens_seen": 109903872 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048826479438314946, + "loss": 1.8028, + "theoretical_loss": 4.7101721979451785, + "tokens_seen": 109969408 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048825476429287864, + "loss": 1.7723, + "theoretical_loss": 4.709788341311781, + "tokens_seen": 110034944 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004882447342026078, + "loss": 1.7464, + "theoretical_loss": 4.709404777203279, + "tokens_seen": 110100480 + }, + { + "epoch": 0.03, + "learning_rate": 0.000488234704112337, + "loss": 1.7462, + "theoretical_loss": 4.7090215052228235, + "tokens_seen": 110166016 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048822467402206624, + "loss": 1.6366, + "theoretical_loss": 4.708638524974344, + "tokens_seen": 110231552 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048821464393179536, + "loss": 1.7662, + "theoretical_loss": 4.708255836062538, + "tokens_seen": 110297088 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004882046138415246, + "loss": 1.9545, + "theoretical_loss": 4.707873438092877, + "tokens_seen": 110362624 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004881945837512537, + "loss": 1.8937, + "theoretical_loss": 4.707491330671596, + "tokens_seen": 110428160 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048818455366098296, + "loss": 1.739, + "theoretical_loss": 4.707109513405701, + "tokens_seen": 110493696 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048817452357071214, + "loss": 1.7945, + "theoretical_loss": 4.706727985902957, + "tokens_seen": 110559232 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004881644934804413, + "loss": 1.7073, + "theoretical_loss": 4.706346747771898, + "tokens_seen": 110624768 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004881544633901705, + "loss": 1.7813, + "theoretical_loss": 4.705965798621811, + "tokens_seen": 110690304 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048814443329989974, + "loss": 1.7482, + "theoretical_loss": 4.705585138062747, + "tokens_seen": 110755840 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048813440320962887, + "loss": 1.7952, + "theoretical_loss": 4.705204765705512, + "tokens_seen": 110821376 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004881243731193581, + "loss": 1.7335, + "theoretical_loss": 4.704824681161664, + "tokens_seen": 110886912 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048811434302908723, + "loss": 1.8335, + "theoretical_loss": 4.704444884043518, + "tokens_seen": 110952448 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048810431293881646, + "loss": 1.5317, + "theoretical_loss": 4.704065373964138, + "tokens_seen": 111017984 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048809428284854564, + "loss": 1.865, + "theoretical_loss": 4.703686150537336, + "tokens_seen": 111083520 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004880842527582748, + "loss": 1.702, + "theoretical_loss": 4.703307213377671, + "tokens_seen": 111149056 + }, + { + "epoch": 0.03, + "learning_rate": 0.000488074222668004, + "loss": 1.8291, + "theoretical_loss": 4.70292856210045, + "tokens_seen": 111214592 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004880641925777332, + "loss": 1.7171, + "theoretical_loss": 4.702550196321718, + "tokens_seen": 111280128 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048805416248746237, + "loss": 1.731, + "theoretical_loss": 4.702172115658267, + "tokens_seen": 111345664 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.0003986795200034976, + "objective/train/docs_used": 74080, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4330945014953613, + "objective/train/original_loss": 3.4330942630767822, + "objective/train/theoretical_loss": 4.701794319727625, + "objective/train/tokens_used": 131871200, + "objective/train/value_avg": -0.0043487548828125, + "objective/train/value_loss": 8.418356446782127e-05, + "objective/train/value_max": -0.00012636184692382812, + "objective/train/value_min": -0.424560546875, + "objective/train/value_reward_corr": 0.06414189295394947, + "objective/train/value_std": 0.006519317626953125, + "objective/train/weight_avg": 1.0041050910949707, + "objective/train/weighted_lm_loss": 3.421333074569702, + "objective/train/weights_max": 65.46415710449219, + "objective/train/weights_min": 0.4073759615421295, + "theoretical_loss": 4.701794319727625, + "tokens_seen": 111411200 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004880441323971916, + "loss": 1.6744, + "theoretical_loss": 4.701794319727625, + "tokens_seen": 111411200 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048803410230692073, + "loss": 1.6957, + "theoretical_loss": 4.70141680814806, + "tokens_seen": 111476736 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048802407221664997, + "loss": 1.8128, + "theoretical_loss": 4.701039580538573, + "tokens_seen": 111542272 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048801404212637915, + "loss": 1.8247, + "theoretical_loss": 4.700662636518901, + "tokens_seen": 111607808 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048800401203610833, + "loss": 1.6172, + "theoretical_loss": 4.700285975709513, + "tokens_seen": 111673344 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004879939819458375, + "loss": 1.7892, + "theoretical_loss": 4.699909597731608, + "tokens_seen": 111738880 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004879839518555667, + "loss": 1.5785, + "theoretical_loss": 4.699533502207114, + "tokens_seen": 111804416 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004879739217652959, + "loss": 1.5896, + "theoretical_loss": 4.699157688758683, + "tokens_seen": 111869952 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004879638916750251, + "loss": 1.8142, + "theoretical_loss": 4.698782157009695, + "tokens_seen": 111935488 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004879538615847543, + "loss": 1.8288, + "theoretical_loss": 4.6984069065842515, + "tokens_seen": 112001024 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048794383149448347, + "loss": 1.7996, + "theoretical_loss": 4.698031937107175, + "tokens_seen": 112066560 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048793380140421265, + "loss": 1.6491, + "theoretical_loss": 4.69765724820401, + "tokens_seen": 112132096 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048792377131394183, + "loss": 1.7153, + "theoretical_loss": 4.697282839501012, + "tokens_seen": 112197632 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048791374122367107, + "loss": 1.7246, + "theoretical_loss": 4.696908710625161, + "tokens_seen": 112263168 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004879037111334002, + "loss": 1.6453, + "theoretical_loss": 4.696534861204146, + "tokens_seen": 112328704 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048789368104312943, + "loss": 1.8954, + "theoretical_loss": 4.696161290866366, + "tokens_seen": 112394240 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048788365095285856, + "loss": 1.514, + "theoretical_loss": 4.695787999240936, + "tokens_seen": 112459776 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004878736208625878, + "loss": 1.6505, + "theoretical_loss": 4.695414985957676, + "tokens_seen": 112525312 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048786359077231697, + "loss": 1.8708, + "theoretical_loss": 4.695042250647113, + "tokens_seen": 112590848 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048785356068204615, + "loss": 1.6399, + "theoretical_loss": 4.694669792940482, + "tokens_seen": 112656384 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048784353059177533, + "loss": 1.7559, + "theoretical_loss": 4.694297612469717, + "tokens_seen": 112721920 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048783350050150457, + "loss": 1.7519, + "theoretical_loss": 4.693925708867458, + "tokens_seen": 112787456 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004878234704112337, + "loss": 1.7645, + "theoretical_loss": 4.693554081767042, + "tokens_seen": 112852992 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048781344032096293, + "loss": 1.7663, + "theoretical_loss": 4.693182730802504, + "tokens_seen": 112918528 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048780341023069206, + "loss": 1.7288, + "theoretical_loss": 4.692811655608578, + "tokens_seen": 112984064 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": -0.001218639430589974, + "objective/train/docs_used": 74817, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.931394577026367, + "objective/train/original_loss": 3.931394100189209, + "objective/train/theoretical_loss": 4.69244085582069, + "objective/train/tokens_used": 133509600, + "objective/train/value_avg": -0.0037479400634765625, + "objective/train/value_loss": 4.995808922103606e-05, + "objective/train/value_max": -0.0002199411392211914, + "objective/train/value_min": -0.060638427734375, + "objective/train/value_reward_corr": 5.536843758906576e-05, + "objective/train/value_std": 0.003078460693359375, + "objective/train/weight_avg": 0.9902520179748535, + "objective/train/weighted_lm_loss": 3.921621799468994, + "objective/train/weights_max": 1.8337888717651367, + "objective/train/weights_min": 0.6463302373886108, + "theoretical_loss": 4.69244085582069, + "tokens_seen": 113049600 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004877933801404213, + "loss": 1.7522, + "theoretical_loss": 4.69244085582069, + "tokens_seen": 113049600 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004877833500501505, + "loss": 1.837, + "theoretical_loss": 4.692070331074961, + "tokens_seen": 113115136 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048777331995987966, + "loss": 1.6568, + "theoretical_loss": 4.691700081008202, + "tokens_seen": 113180672 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048776328986960884, + "loss": 1.7919, + "theoretical_loss": 4.691330105257913, + "tokens_seen": 113246208 + }, + { + "epoch": 0.03, + "learning_rate": 0.000487753259779338, + "loss": 1.7419, + "theoretical_loss": 4.690960403462283, + "tokens_seen": 113311744 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004877432296890672, + "loss": 1.7414, + "theoretical_loss": 4.6905909752601875, + "tokens_seen": 113377280 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048773319959879644, + "loss": 1.8594, + "theoretical_loss": 4.690221820291184, + "tokens_seen": 113442816 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048772316950852556, + "loss": 1.6816, + "theoretical_loss": 4.689852938195516, + "tokens_seen": 113508352 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004877131394182548, + "loss": 1.8176, + "theoretical_loss": 4.689484328614107, + "tokens_seen": 113573888 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004877031093279839, + "loss": 1.7619, + "theoretical_loss": 4.689115991188558, + "tokens_seen": 113639424 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048769307923771316, + "loss": 1.8252, + "theoretical_loss": 4.688747925561149, + "tokens_seen": 113704960 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048768304914744234, + "loss": 1.7571, + "theoretical_loss": 4.688380131374835, + "tokens_seen": 113770496 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004876730190571715, + "loss": 1.7911, + "theoretical_loss": 4.688012608273249, + "tokens_seen": 113836032 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004876629889669007, + "loss": 1.7801, + "theoretical_loss": 4.687645355900694, + "tokens_seen": 113901568 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048765295887662994, + "loss": 1.6689, + "theoretical_loss": 4.687278373902141, + "tokens_seen": 113967104 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048764292878635907, + "loss": 1.791, + "theoretical_loss": 4.6869116619232365, + "tokens_seen": 114032640 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004876328986960883, + "loss": 1.758, + "theoretical_loss": 4.686545219610291, + "tokens_seen": 114098176 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048762286860581743, + "loss": 1.7729, + "theoretical_loss": 4.68617904661028, + "tokens_seen": 114163712 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048761283851554666, + "loss": 1.738, + "theoretical_loss": 4.685813142570847, + "tokens_seen": 114229248 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048760280842527584, + "loss": 1.7095, + "theoretical_loss": 4.685447507140298, + "tokens_seen": 114294784 + }, + { + "epoch": 0.03, + "learning_rate": 0.000487592778335005, + "loss": 1.8215, + "theoretical_loss": 4.685082139967594, + "tokens_seen": 114360320 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004875827482447342, + "loss": 1.8086, + "theoretical_loss": 4.684717040702367, + "tokens_seen": 114425856 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004875727181544634, + "loss": 1.8279, + "theoretical_loss": 4.684352208994895, + "tokens_seen": 114491392 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048756268806419257, + "loss": 1.6549, + "theoretical_loss": 4.683987644496119, + "tokens_seen": 114556928 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004875526579739218, + "loss": 1.7579, + "theoretical_loss": 4.683623346857635, + "tokens_seen": 114622464 + }, + { + "epoch": 0.03, + "objective/train/advantage_avg": 0.00017911549366544932, + "objective/train/docs_used": 76169, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.572613477706909, + "objective/train/original_loss": 3.5726137161254883, + "objective/train/theoretical_loss": 4.683259315731689, + "objective/train/tokens_used": 135148000, + "objective/train/value_avg": -0.0042877197265625, + "objective/train/value_loss": 6.735048373229802e-05, + "objective/train/value_max": -0.0001323223114013672, + "objective/train/value_min": -0.126220703125, + "objective/train/value_reward_corr": 0.09837608292321696, + "objective/train/value_std": 0.00445556640625, + "objective/train/weight_avg": 1.0051283836364746, + "objective/train/weighted_lm_loss": 3.615123987197876, + "objective/train/weights_max": 3.533210515975952, + "objective/train/weights_min": 0.4044919013977051, + "theoretical_loss": 4.683259315731689, + "tokens_seen": 114688000 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048754262788365093, + "loss": 1.7664, + "theoretical_loss": 4.683259315731689, + "tokens_seen": 114688000 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048753259779338017, + "loss": 1.7725, + "theoretical_loss": 4.682895550771182, + "tokens_seen": 114753536 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048752256770310935, + "loss": 1.8046, + "theoretical_loss": 4.682532051629661, + "tokens_seen": 114819072 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048751253761283853, + "loss": 1.6996, + "theoretical_loss": 4.682168817961324, + "tokens_seen": 114884608 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004875025075225677, + "loss": 1.6822, + "theoretical_loss": 4.681805849421017, + "tokens_seen": 114950144 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004874924774322969, + "loss": 1.6944, + "theoretical_loss": 4.681443145664227, + "tokens_seen": 115015680 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048748244734202607, + "loss": 1.5985, + "theoretical_loss": 4.681080706347089, + "tokens_seen": 115081216 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004874724172517553, + "loss": 1.7241, + "theoretical_loss": 4.680718531126377, + "tokens_seen": 115146752 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048746238716148443, + "loss": 1.7764, + "theoretical_loss": 4.6803566196595074, + "tokens_seen": 115212288 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048745235707121367, + "loss": 1.7677, + "theoretical_loss": 4.6799949716045335, + "tokens_seen": 115277824 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004874423269809428, + "loss": 1.5976, + "theoretical_loss": 4.679633586620149, + "tokens_seen": 115343360 + }, + { + "epoch": 0.03, + "learning_rate": 0.00048743229689067203, + "loss": 1.7048, + "theoretical_loss": 4.679272464365681, + "tokens_seen": 115408896 + }, + { + "epoch": 0.03, + "learning_rate": 0.0004874222668004012, + "loss": 1.8638, + "theoretical_loss": 4.678911604501092, + "tokens_seen": 115474432 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004874122367101304, + "loss": 1.683, + "theoretical_loss": 4.678551006686977, + "tokens_seen": 115539968 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004874022066198596, + "loss": 1.7073, + "theoretical_loss": 4.678190670584562, + "tokens_seen": 115605504 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048739217652958876, + "loss": 1.7731, + "theoretical_loss": 4.677830595855703, + "tokens_seen": 115671040 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048738214643931794, + "loss": 1.8451, + "theoretical_loss": 4.6774707821628825, + "tokens_seen": 115736576 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048737211634904717, + "loss": 1.7196, + "theoretical_loss": 4.677111229169213, + "tokens_seen": 115802112 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004873620862587763, + "loss": 1.7519, + "theoretical_loss": 4.676751936538429, + "tokens_seen": 115867648 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048735205616850553, + "loss": 1.7241, + "theoretical_loss": 4.676392903934891, + "tokens_seen": 115933184 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004873420260782347, + "loss": 1.6967, + "theoretical_loss": 4.676034131023581, + "tokens_seen": 115998720 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004873319959879639, + "loss": 1.7274, + "theoretical_loss": 4.675675617470099, + "tokens_seen": 116064256 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004873219658976931, + "loss": 1.655, + "theoretical_loss": 4.6753173629406675, + "tokens_seen": 116129792 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048731193580742226, + "loss": 1.6993, + "theoretical_loss": 4.674959367102126, + "tokens_seen": 116195328 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048730190571715144, + "loss": 1.6491, + "theoretical_loss": 4.674601629621927, + "tokens_seen": 116260864 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0012560490285977721, + "objective/train/docs_used": 76931, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.371119976043701, + "objective/train/original_loss": 3.3711202144622803, + "objective/train/theoretical_loss": 4.674244150168143, + "objective/train/tokens_used": 136786400, + "objective/train/value_avg": -0.004322052001953125, + "objective/train/value_loss": 9.297260839957744e-05, + "objective/train/value_max": -0.00018525123596191406, + "objective/train/value_min": -0.12054443359375, + "objective/train/value_reward_corr": 0.10319234750338975, + "objective/train/value_std": 0.0037746429443359375, + "objective/train/weight_avg": 0.9914646744728088, + "objective/train/weighted_lm_loss": 3.359170436859131, + "objective/train/weights_max": 3.1157777309417725, + "objective/train/weights_min": 0.19344022870063782, + "theoretical_loss": 4.674244150168143, + "tokens_seen": 116326400 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004872918756268807, + "loss": 1.4768, + "theoretical_loss": 4.674244150168143, + "tokens_seen": 116326400 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004872818455366098, + "loss": 1.7307, + "theoretical_loss": 4.673886928409454, + "tokens_seen": 116391936 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048727181544633904, + "loss": 1.8168, + "theoretical_loss": 4.673529964015156, + "tokens_seen": 116457472 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048726178535606816, + "loss": 1.7251, + "theoretical_loss": 4.6731732566551525, + "tokens_seen": 116523008 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004872517552657974, + "loss": 1.6346, + "theoretical_loss": 4.672816805999958, + "tokens_seen": 116588544 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004872417251755266, + "loss": 1.791, + "theoretical_loss": 4.672460611720694, + "tokens_seen": 116654080 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048723169508525576, + "loss": 1.8159, + "theoretical_loss": 4.672104673489084, + "tokens_seen": 116719616 + }, + { + "epoch": 0.04, + "learning_rate": 0.000487221664994985, + "loss": 1.7601, + "theoretical_loss": 4.67174899097746, + "tokens_seen": 116785152 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004872116349047141, + "loss": 1.7567, + "theoretical_loss": 4.671393563858756, + "tokens_seen": 116850688 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048720160481444336, + "loss": 1.7944, + "theoretical_loss": 4.671038391806507, + "tokens_seen": 116916224 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048719157472417254, + "loss": 1.7802, + "theoretical_loss": 4.670683474494848, + "tokens_seen": 116981760 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004871815446339017, + "loss": 1.5865, + "theoretical_loss": 4.670328811598512, + "tokens_seen": 117047296 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004871715145436309, + "loss": 1.8443, + "theoretical_loss": 4.669974402792832, + "tokens_seen": 117112832 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048716148445336014, + "loss": 1.6321, + "theoretical_loss": 4.6696202477537305, + "tokens_seen": 117178368 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048715145436308927, + "loss": 1.6064, + "theoretical_loss": 4.669266346157732, + "tokens_seen": 117243904 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004871414242728185, + "loss": 1.7983, + "theoretical_loss": 4.668912697681949, + "tokens_seen": 117309440 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048713139418254763, + "loss": 1.7783, + "theoretical_loss": 4.668559302004087, + "tokens_seen": 117374976 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048712136409227686, + "loss": 1.7489, + "theoretical_loss": 4.668206158802439, + "tokens_seen": 117440512 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048711133400200604, + "loss": 1.8422, + "theoretical_loss": 4.667853267755892, + "tokens_seen": 117506048 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004871013039117352, + "loss": 1.7012, + "theoretical_loss": 4.667500628543914, + "tokens_seen": 117571584 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004870912738214644, + "loss": 1.687, + "theoretical_loss": 4.667148240846565, + "tokens_seen": 117637120 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004870812437311936, + "loss": 1.782, + "theoretical_loss": 4.666796104344483, + "tokens_seen": 117702656 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048707121364092277, + "loss": 1.7842, + "theoretical_loss": 4.666444218718894, + "tokens_seen": 117768192 + }, + { + "epoch": 0.04, + "learning_rate": 0.000487061183550652, + "loss": 1.7611, + "theoretical_loss": 4.666092583651603, + "tokens_seen": 117833728 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048705115346038113, + "loss": 1.772, + "theoretical_loss": 4.665741198824998, + "tokens_seen": 117899264 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.00038878252962604165, + "objective/train/docs_used": 78354, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.042851448059082, + "objective/train/original_loss": 3.042851209640503, + "objective/train/theoretical_loss": 4.665390063922041, + "objective/train/tokens_used": 138424800, + "objective/train/value_avg": -0.004241943359375, + "objective/train/value_loss": 7.542497041868046e-05, + "objective/train/value_max": -0.00023055076599121094, + "objective/train/value_min": -0.43701171875, + "objective/train/value_reward_corr": 0.05987546765027811, + "objective/train/value_std": 0.00545501708984375, + "objective/train/weight_avg": 1.0035232305526733, + "objective/train/weighted_lm_loss": 3.038206100463867, + "objective/train/weights_max": 75.6312484741211, + "objective/train/weights_min": 0.1927441507577896, + "theoretical_loss": 4.665390063922041, + "tokens_seen": 117964800 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048704112337011037, + "loss": 1.8031, + "theoretical_loss": 4.665390063922041, + "tokens_seen": 117964800 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048703109327983955, + "loss": 1.6233, + "theoretical_loss": 4.665039178626278, + "tokens_seen": 118030336 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048702106318956873, + "loss": 1.7439, + "theoretical_loss": 4.664688542621825, + "tokens_seen": 118095872 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004870110330992979, + "loss": 1.7754, + "theoretical_loss": 4.664338155593377, + "tokens_seen": 118161408 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004870010030090271, + "loss": 1.7974, + "theoretical_loss": 4.6639880172262, + "tokens_seen": 118226944 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048699097291875627, + "loss": 1.7529, + "theoretical_loss": 4.6636381272061325, + "tokens_seen": 118292480 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004869809428284855, + "loss": 1.6736, + "theoretical_loss": 4.663288485219585, + "tokens_seen": 118358016 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048697091273821463, + "loss": 1.8142, + "theoretical_loss": 4.662939090953536, + "tokens_seen": 118423552 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048696088264794387, + "loss": 1.746, + "theoretical_loss": 4.662589944095533, + "tokens_seen": 118489088 + }, + { + "epoch": 0.04, + "learning_rate": 0.000486950852557673, + "loss": 1.7925, + "theoretical_loss": 4.662241044333689, + "tokens_seen": 118554624 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048694082246740223, + "loss": 1.7898, + "theoretical_loss": 4.661892391356682, + "tokens_seen": 118620160 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004869307923771314, + "loss": 1.4929, + "theoretical_loss": 4.661543984853757, + "tokens_seen": 118685696 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004869207622868606, + "loss": 1.6891, + "theoretical_loss": 4.661195824514717, + "tokens_seen": 118751232 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004869107321965898, + "loss": 1.6522, + "theoretical_loss": 4.660847910029929, + "tokens_seen": 118816768 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048690070210631896, + "loss": 1.7419, + "theoretical_loss": 4.660500241090322, + "tokens_seen": 118882304 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048689067201604814, + "loss": 1.6896, + "theoretical_loss": 4.660152817387377, + "tokens_seen": 118947840 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048688064192577737, + "loss": 1.7408, + "theoretical_loss": 4.6598056386131415, + "tokens_seen": 119013376 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004868706118355065, + "loss": 1.7106, + "theoretical_loss": 4.659458704460212, + "tokens_seen": 119078912 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048686058174523573, + "loss": 1.7433, + "theoretical_loss": 4.6591120146217415, + "tokens_seen": 119144448 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004868505516549649, + "loss": 1.7623, + "theoretical_loss": 4.658765568791438, + "tokens_seen": 119209984 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004868405215646941, + "loss": 1.7558, + "theoretical_loss": 4.6584193666635585, + "tokens_seen": 119275520 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004868304914744233, + "loss": 1.7183, + "theoretical_loss": 4.658073407932915, + "tokens_seen": 119341056 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048682046138415246, + "loss": 1.7846, + "theoretical_loss": 4.657727692294865, + "tokens_seen": 119406592 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048681043129388164, + "loss": 1.8436, + "theoretical_loss": 4.657382219445317, + "tokens_seen": 119472128 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004868004012036109, + "loss": 1.7228, + "theoretical_loss": 4.657036989080726, + "tokens_seen": 119537664 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0011901543475687504, + "objective/train/docs_used": 78905, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2844271659851074, + "objective/train/original_loss": 3.2844274044036865, + "objective/train/theoretical_loss": 4.656692000898091, + "objective/train/tokens_used": 140063200, + "objective/train/value_avg": -0.00424957275390625, + "objective/train/value_loss": 8.55424368637614e-05, + "objective/train/value_max": -0.00018668174743652344, + "objective/train/value_min": -0.1728515625, + "objective/train/value_reward_corr": 0.048649634557705386, + "objective/train/value_std": 0.00391387939453125, + "objective/train/weight_avg": 0.992118239402771, + "objective/train/weighted_lm_loss": 3.243591785430908, + "objective/train/weights_max": 4.985652923583984, + "objective/train/weights_min": 0.3720249533653259, + "theoretical_loss": 4.656692000898091, + "tokens_seen": 119603200 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048679037111334, + "loss": 1.692, + "theoretical_loss": 4.656692000898091, + "tokens_seen": 119603200 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048678034102306924, + "loss": 1.7353, + "theoretical_loss": 4.656347254594957, + "tokens_seen": 119668736 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048677031093279836, + "loss": 1.737, + "theoretical_loss": 4.656002749869411, + "tokens_seen": 119734272 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004867602808425276, + "loss": 1.6784, + "theoretical_loss": 4.655658486420084, + "tokens_seen": 119799808 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004867502507522568, + "loss": 1.7895, + "theoretical_loss": 4.655314463946143, + "tokens_seen": 119865344 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048674022066198596, + "loss": 1.7623, + "theoretical_loss": 4.654970682147299, + "tokens_seen": 119930880 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048673019057171514, + "loss": 1.5916, + "theoretical_loss": 4.6546271407238, + "tokens_seen": 119996416 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004867201604814443, + "loss": 1.7639, + "theoretical_loss": 4.654283839376431, + "tokens_seen": 120061952 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004867101303911735, + "loss": 1.6417, + "theoretical_loss": 4.653940777806509, + "tokens_seen": 120127488 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048670010030090274, + "loss": 1.8189, + "theoretical_loss": 4.653597955715888, + "tokens_seen": 120193024 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048669007021063187, + "loss": 1.7441, + "theoretical_loss": 4.653255372806958, + "tokens_seen": 120258560 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004866800401203611, + "loss": 1.8814, + "theoretical_loss": 4.652913028782635, + "tokens_seen": 120324096 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004866700100300903, + "loss": 1.7832, + "theoretical_loss": 4.652570923346367, + "tokens_seen": 120389632 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048665997993981947, + "loss": 1.7, + "theoretical_loss": 4.6522290562021364, + "tokens_seen": 120455168 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048664994984954865, + "loss": 1.7811, + "theoretical_loss": 4.651887427054448, + "tokens_seen": 120520704 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048663991975927783, + "loss": 1.8243, + "theoretical_loss": 4.651546035608336, + "tokens_seen": 120586240 + }, + { + "epoch": 0.04, + "learning_rate": 0.000486629889669007, + "loss": 1.8943, + "theoretical_loss": 4.651204881569357, + "tokens_seen": 120651776 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048661985957873624, + "loss": 1.7692, + "theoretical_loss": 4.650863964643597, + "tokens_seen": 120717312 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048660982948846537, + "loss": 1.7747, + "theoretical_loss": 4.6505232845376625, + "tokens_seen": 120782848 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004865997993981946, + "loss": 1.7918, + "theoretical_loss": 4.650182840958681, + "tokens_seen": 120848384 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048658976930792373, + "loss": 2.1763, + "theoretical_loss": 4.649842633614304, + "tokens_seen": 120913920 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048657973921765297, + "loss": 1.7867, + "theoretical_loss": 4.649502662212699, + "tokens_seen": 120979456 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048656970912738215, + "loss": 1.7286, + "theoretical_loss": 4.649162926462553, + "tokens_seen": 121044992 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048655967903711133, + "loss": 1.8882, + "theoretical_loss": 4.648823426073072, + "tokens_seen": 121110528 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004865496489468405, + "loss": 1.7884, + "theoretical_loss": 4.648484160753975, + "tokens_seen": 121176064 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0017844230169430375, + "objective/train/docs_used": 79692, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1131162643432617, + "objective/train/original_loss": 3.1131160259246826, + "objective/train/theoretical_loss": 4.648145130215498, + "objective/train/tokens_used": 141701600, + "objective/train/value_avg": -0.003437042236328125, + "objective/train/value_loss": 4.977492426405661e-05, + "objective/train/value_max": -0.0001366138458251953, + "objective/train/value_min": -0.05511474609375, + "objective/train/value_reward_corr": 0.012432402428666067, + "objective/train/value_std": 0.0029735565185546875, + "objective/train/weight_avg": 0.9845498204231262, + "objective/train/weighted_lm_loss": 3.075005054473877, + "objective/train/weights_max": 1.7352429628372192, + "objective/train/weights_min": 0.5977539420127869, + "theoretical_loss": 4.648145130215498, + "tokens_seen": 121241600 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048653961885656975, + "loss": 1.7675, + "theoretical_loss": 4.648145130215498, + "tokens_seen": 121241600 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004865295887662989, + "loss": 1.7672, + "theoretical_loss": 4.64780633416839, + "tokens_seen": 121307136 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004865195586760281, + "loss": 1.8741, + "theoretical_loss": 4.647467772323912, + "tokens_seen": 121372672 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048650952858575724, + "loss": 1.5876, + "theoretical_loss": 4.647129444393837, + "tokens_seen": 121438208 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048649949849548647, + "loss": 1.821, + "theoretical_loss": 4.646791350090446, + "tokens_seen": 121503744 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048648946840521565, + "loss": 1.8888, + "theoretical_loss": 4.646453489126531, + "tokens_seen": 121569280 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048647943831494483, + "loss": 1.6709, + "theoretical_loss": 4.646115861215389, + "tokens_seen": 121634816 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048646940822467407, + "loss": 1.6497, + "theoretical_loss": 4.645778466070829, + "tokens_seen": 121700352 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004864593781344032, + "loss": 1.6461, + "theoretical_loss": 4.64544130340716, + "tokens_seen": 121765888 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048644934804413243, + "loss": 1.6965, + "theoretical_loss": 4.645104372939196, + "tokens_seen": 121831424 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004864393179538616, + "loss": 1.6776, + "theoretical_loss": 4.644767674382256, + "tokens_seen": 121896960 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004864292878635908, + "loss": 1.8483, + "theoretical_loss": 4.644431207452159, + "tokens_seen": 121962496 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048641925777332, + "loss": 1.7607, + "theoretical_loss": 4.6440949718652265, + "tokens_seen": 122028032 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048640922768304916, + "loss": 1.6269, + "theoretical_loss": 4.643758967338277, + "tokens_seen": 122093568 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048639919759277834, + "loss": 1.7251, + "theoretical_loss": 4.643423193588631, + "tokens_seen": 122159104 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048638916750250757, + "loss": 1.8421, + "theoretical_loss": 4.643087650334102, + "tokens_seen": 122224640 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004863791374122367, + "loss": 1.6986, + "theoretical_loss": 4.642752337293002, + "tokens_seen": 122290176 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048636910732196593, + "loss": 1.7787, + "theoretical_loss": 4.642417254184139, + "tokens_seen": 122355712 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004863590772316951, + "loss": 1.8139, + "theoretical_loss": 4.642082400726812, + "tokens_seen": 122421248 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004863490471414243, + "loss": 1.8558, + "theoretical_loss": 4.641747776640816, + "tokens_seen": 122486784 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004863390170511535, + "loss": 1.7747, + "theoretical_loss": 4.641413381646435, + "tokens_seen": 122552320 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048632898696088266, + "loss": 1.7553, + "theoretical_loss": 4.641079215464444, + "tokens_seen": 122617856 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048631895687061184, + "loss": 1.7899, + "theoretical_loss": 4.640745277816107, + "tokens_seen": 122683392 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004863089267803411, + "loss": 1.708, + "theoretical_loss": 4.640411568423177, + "tokens_seen": 122748928 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004862988966900702, + "loss": 1.5624, + "theoretical_loss": 4.640078087007897, + "tokens_seen": 122814464 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.003484945045784116, + "objective/train/docs_used": 81051, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3645172119140625, + "objective/train/original_loss": 3.3645172119140625, + "objective/train/theoretical_loss": 4.639744833292989, + "objective/train/tokens_used": 143340000, + "objective/train/value_avg": -0.004138946533203125, + "objective/train/value_loss": 0.00010569865844445303, + "objective/train/value_max": -0.0001926422119140625, + "objective/train/value_min": -0.11163330078125, + "objective/train/value_reward_corr": 0.07021393259969443, + "objective/train/value_std": 0.0038547515869140625, + "objective/train/weight_avg": 0.9701031446456909, + "objective/train/weighted_lm_loss": 3.2553179264068604, + "objective/train/weights_max": 2.591919422149658, + "objective/train/weights_min": 0.46712520718574524, + "theoretical_loss": 4.639744833292989, + "tokens_seen": 122880000 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048628886659979944, + "loss": 1.7668, + "theoretical_loss": 4.639744833292989, + "tokens_seen": 122880000 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048627883650952857, + "loss": 1.759, + "theoretical_loss": 4.639411807001665, + "tokens_seen": 122945536 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004862688064192578, + "loss": 1.7252, + "theoretical_loss": 4.639079007857621, + "tokens_seen": 123011072 + }, + { + "epoch": 0.04, + "learning_rate": 0.000486258776328987, + "loss": 1.7178, + "theoretical_loss": 4.638746435585032, + "tokens_seen": 123076608 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048624874623871616, + "loss": 1.7023, + "theoretical_loss": 4.638414089908558, + "tokens_seen": 123142144 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048623871614844534, + "loss": 1.6779, + "theoretical_loss": 4.63808197055334, + "tokens_seen": 123207680 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004862286860581745, + "loss": 1.6275, + "theoretical_loss": 4.637750077244993, + "tokens_seen": 123273216 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004862186559679037, + "loss": 1.6669, + "theoretical_loss": 4.637418409709617, + "tokens_seen": 123338752 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048620862587763294, + "loss": 1.7289, + "theoretical_loss": 4.637086967673785, + "tokens_seen": 123404288 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048619859578736207, + "loss": 1.6939, + "theoretical_loss": 4.6367557508645465, + "tokens_seen": 123469824 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004861885656970913, + "loss": 1.834, + "theoretical_loss": 4.6364247590094285, + "tokens_seen": 123535360 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004861785356068205, + "loss": 1.7588, + "theoretical_loss": 4.636093991836429, + "tokens_seen": 123600896 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048616850551654967, + "loss": 1.7665, + "theoretical_loss": 4.635763449074021, + "tokens_seen": 123666432 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048615847542627885, + "loss": 1.6387, + "theoretical_loss": 4.635433130451148, + "tokens_seen": 123731968 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048614844533600803, + "loss": 1.7423, + "theoretical_loss": 4.635103035697226, + "tokens_seen": 123797504 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004861384152457372, + "loss": 1.6936, + "theoretical_loss": 4.634773164542136, + "tokens_seen": 123863040 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048612838515546644, + "loss": 1.66, + "theoretical_loss": 4.634443516716235, + "tokens_seen": 123928576 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048611835506519557, + "loss": 1.7017, + "theoretical_loss": 4.634114091950341, + "tokens_seen": 123994112 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004861083249749248, + "loss": 1.6853, + "theoretical_loss": 4.633784889975743, + "tokens_seen": 124059648 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048609829488465393, + "loss": 1.5915, + "theoretical_loss": 4.633455910524191, + "tokens_seen": 124125184 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048608826479438317, + "loss": 1.6247, + "theoretical_loss": 4.633127153327905, + "tokens_seen": 124190720 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048607823470411235, + "loss": 1.8227, + "theoretical_loss": 4.632798618119562, + "tokens_seen": 124256256 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048606820461384153, + "loss": 1.7232, + "theoretical_loss": 4.6324703046323075, + "tokens_seen": 124321792 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004860581745235707, + "loss": 1.7448, + "theoretical_loss": 4.632142212599742, + "tokens_seen": 124387328 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048604814443329995, + "loss": 1.7548, + "theoretical_loss": 4.631814341755933, + "tokens_seen": 124452864 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0005775440949946642, + "objective/train/docs_used": 81846, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.664000988006592, + "objective/train/original_loss": 3.664001226425171, + "objective/train/theoretical_loss": 4.631486691835402, + "objective/train/tokens_used": 144978400, + "objective/train/value_avg": -0.004756927490234375, + "objective/train/value_loss": 7.758277206448838e-05, + "objective/train/value_max": -0.000293731689453125, + "objective/train/value_min": -0.10052490234375, + "objective/train/value_reward_corr": 0.022895426615430085, + "objective/train/value_std": 0.004222869873046875, + "objective/train/weight_avg": 0.9979974627494812, + "objective/train/weighted_lm_loss": 3.666339159011841, + "objective/train/weights_max": 2.6327946186065674, + "objective/train/weights_min": 0.46702274680137634, + "theoretical_loss": 4.631486691835402, + "tokens_seen": 124518400 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004860381143430291, + "loss": 1.7482, + "theoretical_loss": 4.631486691835402, + "tokens_seen": 124518400 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004860280842527583, + "loss": 1.7643, + "theoretical_loss": 4.631159262573131, + "tokens_seen": 124583936 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048601805416248744, + "loss": 1.7851, + "theoretical_loss": 4.630832053704557, + "tokens_seen": 124649472 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048600802407221667, + "loss": 1.6168, + "theoretical_loss": 4.6305050649655755, + "tokens_seen": 124715008 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048599799398194585, + "loss": 1.6776, + "theoretical_loss": 4.630178296092535, + "tokens_seen": 124780544 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048598796389167503, + "loss": 1.7047, + "theoretical_loss": 4.629851746822241, + "tokens_seen": 124846080 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004859779338014042, + "loss": 1.7604, + "theoretical_loss": 4.6295254168919495, + "tokens_seen": 124911616 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004859679037111334, + "loss": 1.7214, + "theoretical_loss": 4.629199306039368, + "tokens_seen": 124977152 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004859578736208626, + "loss": 1.7311, + "theoretical_loss": 4.628873414002657, + "tokens_seen": 125042688 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004859478435305918, + "loss": 1.7825, + "theoretical_loss": 4.6285477405204265, + "tokens_seen": 125108224 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048593781344032094, + "loss": 1.672, + "theoretical_loss": 4.628222285331734, + "tokens_seen": 125173760 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004859277833500502, + "loss": 1.5815, + "theoretical_loss": 4.627897048176086, + "tokens_seen": 125239296 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004859177532597793, + "loss": 1.7871, + "theoretical_loss": 4.627572028793438, + "tokens_seen": 125304832 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048590772316950854, + "loss": 1.6571, + "theoretical_loss": 4.627247226924187, + "tokens_seen": 125370368 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004858976930792377, + "loss": 1.7852, + "theoretical_loss": 4.626922642309179, + "tokens_seen": 125435904 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004858876629889669, + "loss": 1.7809, + "theoretical_loss": 4.6265982746897, + "tokens_seen": 125501440 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004858776328986961, + "loss": 1.7349, + "theoretical_loss": 4.626274123807484, + "tokens_seen": 125566976 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004858676028084253, + "loss": 1.5727, + "theoretical_loss": 4.625950189404703, + "tokens_seen": 125632512 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048585757271815444, + "loss": 1.6873, + "theoretical_loss": 4.625626471223972, + "tokens_seen": 125698048 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004858475426278837, + "loss": 1.7011, + "theoretical_loss": 4.625302969008344, + "tokens_seen": 125763584 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004858375125376128, + "loss": 1.7511, + "theoretical_loss": 4.624979682501314, + "tokens_seen": 125829120 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048582748244734204, + "loss": 1.7266, + "theoretical_loss": 4.624656611446811, + "tokens_seen": 125894656 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004858174523570712, + "loss": 1.7504, + "theoretical_loss": 4.624333755589207, + "tokens_seen": 125960192 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004858074222668004, + "loss": 1.6945, + "theoretical_loss": 4.624011114673303, + "tokens_seen": 126025728 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004857973921765296, + "loss": 1.7696, + "theoretical_loss": 4.623688688444341, + "tokens_seen": 126091264 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0012250370346009731, + "objective/train/docs_used": 83055, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.876255750656128, + "objective/train/original_loss": 3.876255750656128, + "objective/train/theoretical_loss": 4.623366476647994, + "objective/train/tokens_used": 146616800, + "objective/train/value_avg": -0.00445556640625, + "objective/train/value_loss": 7.68098107073456e-05, + "objective/train/value_max": -0.0001838207244873047, + "objective/train/value_min": -0.06707763671875, + "objective/train/value_reward_corr": 0.15087125311595054, + "objective/train/value_std": 0.00411224365234375, + "objective/train/weight_avg": 0.9912324547767639, + "objective/train/weighted_lm_loss": 3.8503661155700684, + "objective/train/weights_max": 1.9557551145553589, + "objective/train/weights_min": 0.13663341104984283, + "theoretical_loss": 4.623366476647994, + "tokens_seen": 126156800 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048578736208625877, + "loss": 1.7515, + "theoretical_loss": 4.623366476647994, + "tokens_seen": 126156800 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048577733199598795, + "loss": 1.7095, + "theoretical_loss": 4.623044479030369, + "tokens_seen": 126222336 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004857673019057172, + "loss": 1.7865, + "theoretical_loss": 4.622722695338005, + "tokens_seen": 126287872 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004857572718154463, + "loss": 1.6009, + "theoretical_loss": 4.622401125317873, + "tokens_seen": 126353408 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048574724172517554, + "loss": 1.6112, + "theoretical_loss": 4.622079768717375, + "tokens_seen": 126418944 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048573721163490467, + "loss": 1.7112, + "theoretical_loss": 4.62175862528434, + "tokens_seen": 126484480 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004857271815446339, + "loss": 1.62, + "theoretical_loss": 4.621437694767026, + "tokens_seen": 126550016 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048571715145436314, + "loss": 1.7409, + "theoretical_loss": 4.621116976914122, + "tokens_seen": 126615552 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048570712136409227, + "loss": 1.6698, + "theoretical_loss": 4.620796471474737, + "tokens_seen": 126681088 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004856970912738215, + "loss": 1.772, + "theoretical_loss": 4.620476178198411, + "tokens_seen": 126746624 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004856870611835507, + "loss": 1.559, + "theoretical_loss": 4.620156096835108, + "tokens_seen": 126812160 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048567703109327987, + "loss": 1.7886, + "theoretical_loss": 4.619836227135212, + "tokens_seen": 126877696 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048566700100300905, + "loss": 1.7905, + "theoretical_loss": 4.619516568849535, + "tokens_seen": 126943232 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048565697091273823, + "loss": 1.7963, + "theoretical_loss": 4.619197121729307, + "tokens_seen": 127008768 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004856469408224674, + "loss": 1.7114, + "theoretical_loss": 4.61887788552618, + "tokens_seen": 127074304 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048563691073219664, + "loss": 1.7311, + "theoretical_loss": 4.618558859992225, + "tokens_seen": 127139840 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048562688064192577, + "loss": 1.6349, + "theoretical_loss": 4.6182400448799354, + "tokens_seen": 127205376 + }, + { + "epoch": 0.04, + "learning_rate": 0.000485616850551655, + "loss": 1.8085, + "theoretical_loss": 4.617921439942219, + "tokens_seen": 127270912 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048560682046138413, + "loss": 1.5135, + "theoretical_loss": 4.617603044932402, + "tokens_seen": 127336448 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048559679037111337, + "loss": 1.8067, + "theoretical_loss": 4.617284859604227, + "tokens_seen": 127401984 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048558676028084255, + "loss": 1.7397, + "theoretical_loss": 4.616966883711853, + "tokens_seen": 127467520 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048557673019057173, + "loss": 1.7011, + "theoretical_loss": 4.616649117009851, + "tokens_seen": 127533056 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004855667001003009, + "loss": 1.752, + "theoretical_loss": 4.616331559253208, + "tokens_seen": 127598592 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048555667001003015, + "loss": 1.7739, + "theoretical_loss": 4.6160142101973225, + "tokens_seen": 127664128 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004855466399197593, + "loss": 1.6681, + "theoretical_loss": 4.6156970695980055, + "tokens_seen": 127729664 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0013656317023560405, + "objective/train/docs_used": 83592, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.134276866912842, + "objective/train/original_loss": 3.13427734375, + "objective/train/theoretical_loss": 4.615380137211477, + "objective/train/tokens_used": 148255200, + "objective/train/value_avg": -0.004425048828125, + "objective/train/value_loss": 0.0001211340568261221, + "objective/train/value_max": -0.0001634359359741211, + "objective/train/value_min": -0.251953125, + "objective/train/value_reward_corr": 0.17757240107278469, + "objective/train/value_std": 0.00481414794921875, + "objective/train/weight_avg": 0.9919333457946777, + "objective/train/weighted_lm_loss": 3.0919599533081055, + "objective/train/weights_max": 11.317649841308594, + "objective/train/weights_min": 0.03579573705792427, + "theoretical_loss": 4.615380137211477, + "tokens_seen": 127795200 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004855366098294885, + "loss": 1.6174, + "theoretical_loss": 4.615380137211477, + "tokens_seen": 127795200 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048552657973921764, + "loss": 1.7283, + "theoretical_loss": 4.615063412794369, + "tokens_seen": 127860736 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048551654964894687, + "loss": 1.8503, + "theoretical_loss": 4.6147468961037195, + "tokens_seen": 127926272 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048550651955867605, + "loss": 1.7097, + "theoretical_loss": 4.6144305868969795, + "tokens_seen": 127991808 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048549648946840523, + "loss": 1.788, + "theoretical_loss": 4.614114484932002, + "tokens_seen": 128057344 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004854864593781344, + "loss": 1.6084, + "theoretical_loss": 4.61379858996705, + "tokens_seen": 128122880 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004854764292878636, + "loss": 1.6316, + "theoretical_loss": 4.613482901760792, + "tokens_seen": 128188416 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004854663991975928, + "loss": 1.8108, + "theoretical_loss": 4.613167420072298, + "tokens_seen": 128253952 + }, + { + "epoch": 0.04, + "learning_rate": 0.000485456369107322, + "loss": 1.6692, + "theoretical_loss": 4.612852144661042, + "tokens_seen": 128319488 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048544633901705114, + "loss": 1.6993, + "theoretical_loss": 4.612537075286906, + "tokens_seen": 128385024 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004854363089267804, + "loss": 1.8459, + "theoretical_loss": 4.612222211710168, + "tokens_seen": 128450560 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004854262788365095, + "loss": 1.6821, + "theoretical_loss": 4.611907553691509, + "tokens_seen": 128516096 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048541624874623874, + "loss": 1.7078, + "theoretical_loss": 4.61159310099201, + "tokens_seen": 128581632 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004854062186559679, + "loss": 1.806, + "theoretical_loss": 4.611278853373152, + "tokens_seen": 128647168 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004853961885656971, + "loss": 1.7263, + "theoretical_loss": 4.6109648105968155, + "tokens_seen": 128712704 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004853861584754263, + "loss": 1.7822, + "theoretical_loss": 4.610650972425274, + "tokens_seen": 128778240 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004853761283851555, + "loss": 1.6148, + "theoretical_loss": 4.610337338621205, + "tokens_seen": 128843776 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048536609829488464, + "loss": 1.7763, + "theoretical_loss": 4.6100239089476736, + "tokens_seen": 128909312 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004853560682046139, + "loss": 1.7041, + "theoretical_loss": 4.609710683168146, + "tokens_seen": 128974848 + }, + { + "epoch": 0.04, + "learning_rate": 0.000485346038114343, + "loss": 1.7548, + "theoretical_loss": 4.6093976610464775, + "tokens_seen": 129040384 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048533600802407224, + "loss": 1.648, + "theoretical_loss": 4.609084842346924, + "tokens_seen": 129105920 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004853259779338014, + "loss": 1.6247, + "theoretical_loss": 4.608772226834126, + "tokens_seen": 129171456 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004853159478435306, + "loss": 1.7666, + "theoretical_loss": 4.60845981427312, + "tokens_seen": 129236992 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004853059177532598, + "loss": 1.7447, + "theoretical_loss": 4.60814760442933, + "tokens_seen": 129302528 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048529588766298897, + "loss": 1.7527, + "theoretical_loss": 4.607835597068576, + "tokens_seen": 129368064 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0014973214128986, + "objective/train/docs_used": 84814, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.6059629917144775, + "objective/train/original_loss": 3.6059629917144775, + "objective/train/theoretical_loss": 4.607523791957059, + "objective/train/tokens_used": 149893600, + "objective/train/value_avg": -0.003955841064453125, + "objective/train/value_loss": 0.00011385645484551787, + "objective/train/value_max": -0.0001647472381591797, + "objective/train/value_min": -0.47265625, + "objective/train/value_reward_corr": 0.13410643142373643, + "objective/train/value_std": 0.007511138916015625, + "objective/train/weight_avg": 0.9972882270812988, + "objective/train/weighted_lm_loss": 3.560007095336914, + "objective/train/weights_max": 86.63135528564453, + "objective/train/weights_min": 0.2501309812068939, + "theoretical_loss": 4.607523791957059, + "tokens_seen": 129433600 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048528585757271815, + "loss": 1.7162, + "theoretical_loss": 4.607523791957059, + "tokens_seen": 129433600 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004852758274824474, + "loss": 1.7857, + "theoretical_loss": 4.607212188861373, + "tokens_seen": 129499136 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004852657973921765, + "loss": 1.756, + "theoretical_loss": 4.606900787548499, + "tokens_seen": 129564672 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048525576730190574, + "loss": 1.7145, + "theoretical_loss": 4.606589587785802, + "tokens_seen": 129630208 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048524573721163487, + "loss": 1.7731, + "theoretical_loss": 4.606278589341037, + "tokens_seen": 129695744 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004852357071213641, + "loss": 1.8037, + "theoretical_loss": 4.605967791982339, + "tokens_seen": 129761280 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004852256770310933, + "loss": 1.7309, + "theoretical_loss": 4.60565719547823, + "tokens_seen": 129826816 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048521564694082247, + "loss": 1.7205, + "theoretical_loss": 4.605346799597614, + "tokens_seen": 129892352 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048520561685055165, + "loss": 1.6968, + "theoretical_loss": 4.605036604109778, + "tokens_seen": 129957888 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004851955867602809, + "loss": 1.7574, + "theoretical_loss": 4.604726608784391, + "tokens_seen": 130023424 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048518555667001, + "loss": 1.7377, + "theoretical_loss": 4.6044168133914996, + "tokens_seen": 130088960 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048517552657973925, + "loss": 1.7236, + "theoretical_loss": 4.604107217701534, + "tokens_seen": 130154496 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004851654964894684, + "loss": 1.6527, + "theoretical_loss": 4.6037978214853, + "tokens_seen": 130220032 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004851554663991976, + "loss": 1.7112, + "theoretical_loss": 4.603488624513987, + "tokens_seen": 130285568 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004851454363089268, + "loss": 1.6074, + "theoretical_loss": 4.603179626559155, + "tokens_seen": 130351104 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048513540621865597, + "loss": 1.7284, + "theoretical_loss": 4.602870827392745, + "tokens_seen": 130416640 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048512537612838515, + "loss": 1.681, + "theoretical_loss": 4.6025622267870725, + "tokens_seen": 130482176 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048511534603811433, + "loss": 1.7458, + "theoretical_loss": 4.602253824514828, + "tokens_seen": 130547712 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004851053159478435, + "loss": 1.6168, + "theoretical_loss": 4.601945620349077, + "tokens_seen": 130613248 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048509528585757275, + "loss": 1.5934, + "theoretical_loss": 4.601637614063257, + "tokens_seen": 130678784 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004850852557673019, + "loss": 1.7282, + "theoretical_loss": 4.601329805431178, + "tokens_seen": 130744320 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004850752256770311, + "loss": 1.6506, + "theoretical_loss": 4.601022194227024, + "tokens_seen": 130809856 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048506519558676024, + "loss": 1.7557, + "theoretical_loss": 4.600714780225347, + "tokens_seen": 130875392 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004850551654964895, + "loss": 1.723, + "theoretical_loss": 4.600407563201072, + "tokens_seen": 130940928 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048504513540621866, + "loss": 1.6983, + "theoretical_loss": 4.60010054292949, + "tokens_seen": 131006464 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0014902562834322453, + "objective/train/docs_used": 85560, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.5203230381011963, + "objective/train/original_loss": 3.5203232765197754, + "objective/train/theoretical_loss": 4.599793719186264, + "objective/train/tokens_used": 151532000, + "objective/train/value_avg": -0.004146575927734375, + "objective/train/value_loss": 5.869594679097645e-05, + "objective/train/value_max": -0.0001609325408935547, + "objective/train/value_min": -0.152099609375, + "objective/train/value_reward_corr": 0.09338081776172305, + "objective/train/value_std": 0.00411224365234375, + "objective/train/weight_avg": 0.9879959225654602, + "objective/train/weighted_lm_loss": 3.4878337383270264, + "objective/train/weights_max": 3.8030784130096436, + "objective/train/weights_min": 0.5955092906951904, + "theoretical_loss": 4.599793719186264, + "tokens_seen": 131072000 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048503510531594784, + "loss": 1.6726, + "theoretical_loss": 4.599793719186264, + "tokens_seen": 131072000 + }, + { + "epoch": 0.04, + "learning_rate": 0.000485025075225677, + "loss": 1.7199, + "theoretical_loss": 4.599487091747424, + "tokens_seen": 131137536 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048501504513540625, + "loss": 1.7869, + "theoretical_loss": 4.599180660389363, + "tokens_seen": 131203072 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004850050150451354, + "loss": 1.7218, + "theoretical_loss": 4.598874424888848, + "tokens_seen": 131268608 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004849949849548646, + "loss": 1.5589, + "theoretical_loss": 4.598568385023004, + "tokens_seen": 131334144 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004849849548645938, + "loss": 1.6774, + "theoretical_loss": 4.598262540569325, + "tokens_seen": 131399680 + }, + { + "epoch": 0.04, + "learning_rate": 0.000484974924774323, + "loss": 1.56, + "theoretical_loss": 4.597956891305667, + "tokens_seen": 131465216 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004849648946840522, + "loss": 1.6192, + "theoretical_loss": 4.59765143701025, + "tokens_seen": 131530752 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048495486459378134, + "loss": 1.8027, + "theoretical_loss": 4.597346177461656, + "tokens_seen": 131596288 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004849448345035106, + "loss": 1.8429, + "theoretical_loss": 4.597041112438829, + "tokens_seen": 131661824 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004849348044132397, + "loss": 1.6547, + "theoretical_loss": 4.5967362417210715, + "tokens_seen": 131727360 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048492477432296894, + "loss": 1.6938, + "theoretical_loss": 4.596431565088048, + "tokens_seen": 131792896 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004849147442326981, + "loss": 1.677, + "theoretical_loss": 4.596127082319786, + "tokens_seen": 131858432 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004849047141424273, + "loss": 1.5699, + "theoretical_loss": 4.595822793196664, + "tokens_seen": 131923968 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004848946840521565, + "loss": 1.7245, + "theoretical_loss": 4.595518697499422, + "tokens_seen": 131989504 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004848846539618857, + "loss": 1.7355, + "theoretical_loss": 4.595214795009159, + "tokens_seen": 132055040 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048487462387161484, + "loss": 1.7169, + "theoretical_loss": 4.594911085507325, + "tokens_seen": 132120576 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004848645937813441, + "loss": 1.6871, + "theoretical_loss": 4.594607568775732, + "tokens_seen": 132186112 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004848545636910732, + "loss": 1.7405, + "theoretical_loss": 4.594304244596541, + "tokens_seen": 132251648 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048484453360080244, + "loss": 1.5688, + "theoretical_loss": 4.59400111275227, + "tokens_seen": 132317184 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004848345035105316, + "loss": 1.6275, + "theoretical_loss": 4.593698173025789, + "tokens_seen": 132382720 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004848244734202608, + "loss": 1.659, + "theoretical_loss": 4.593395425200322, + "tokens_seen": 132448256 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048481444332999, + "loss": 1.6451, + "theoretical_loss": 4.593092869059442, + "tokens_seen": 132513792 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048480441323971917, + "loss": 1.5764, + "theoretical_loss": 4.592790504387073, + "tokens_seen": 132579328 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048479438314944835, + "loss": 1.7486, + "theoretical_loss": 4.592488330967495, + "tokens_seen": 132644864 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0003711888857651502, + "objective/train/docs_used": 86181, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.226935386657715, + "objective/train/original_loss": 3.226935386657715, + "objective/train/theoretical_loss": 4.592186348585331, + "objective/train/tokens_used": 153170400, + "objective/train/value_avg": -0.0044403076171875, + "objective/train/value_loss": 5.364429307519458e-05, + "objective/train/value_max": -0.0002269744873046875, + "objective/train/value_min": -0.1365966796875, + "objective/train/value_reward_corr": 0.07220046486699368, + "objective/train/value_std": 0.004535675048828125, + "objective/train/weight_avg": 0.9990192651748657, + "objective/train/weighted_lm_loss": 3.215496301651001, + "objective/train/weights_max": 3.8410804271698, + "objective/train/weights_min": 0.5186830163002014, + "theoretical_loss": 4.592186348585331, + "tokens_seen": 132710400 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004847843530591776, + "loss": 1.6259, + "theoretical_loss": 4.592186348585331, + "tokens_seen": 132710400 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004847743229689067, + "loss": 1.749, + "theoretical_loss": 4.591884557025555, + "tokens_seen": 132775936 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048476429287863594, + "loss": 1.6088, + "theoretical_loss": 4.591582956073488, + "tokens_seen": 132841472 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048475426278836507, + "loss": 1.6436, + "theoretical_loss": 4.5912815455148, + "tokens_seen": 132907008 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004847442326980943, + "loss": 1.7471, + "theoretical_loss": 4.590980325135508, + "tokens_seen": 132972544 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004847342026078235, + "loss": 1.6609, + "theoretical_loss": 4.590679294721973, + "tokens_seen": 133038080 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048472417251755267, + "loss": 1.7677, + "theoretical_loss": 4.5903784540609, + "tokens_seen": 133103616 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048471414242728185, + "loss": 1.6526, + "theoretical_loss": 4.5900778029393425, + "tokens_seen": 133169152 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004847041123370111, + "loss": 1.5804, + "theoretical_loss": 4.5897773411446945, + "tokens_seen": 133234688 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004846940822467402, + "loss": 1.7239, + "theoretical_loss": 4.589477068464692, + "tokens_seen": 133300224 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048468405215646945, + "loss": 1.6342, + "theoretical_loss": 4.589176984687415, + "tokens_seen": 133365760 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004846740220661986, + "loss": 1.6419, + "theoretical_loss": 4.588877089601287, + "tokens_seen": 133431296 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004846639919759278, + "loss": 1.6957, + "theoretical_loss": 4.5885773829950685, + "tokens_seen": 133496832 + }, + { + "epoch": 0.04, + "learning_rate": 0.000484653961885657, + "loss": 1.6812, + "theoretical_loss": 4.5882778646578615, + "tokens_seen": 133562368 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048464393179538617, + "loss": 1.7265, + "theoretical_loss": 4.587978534379106, + "tokens_seen": 133627904 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048463390170511535, + "loss": 1.602, + "theoretical_loss": 4.587679391948584, + "tokens_seen": 133693440 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048462387161484453, + "loss": 1.8671, + "theoretical_loss": 4.587380437156413, + "tokens_seen": 133758976 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004846138415245737, + "loss": 1.7088, + "theoretical_loss": 4.587081669793047, + "tokens_seen": 133824512 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048460381143430295, + "loss": 1.6822, + "theoretical_loss": 4.586783089649279, + "tokens_seen": 133890048 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004845937813440321, + "loss": 1.5679, + "theoretical_loss": 4.586484696516234, + "tokens_seen": 133955584 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004845837512537613, + "loss": 1.6345, + "theoretical_loss": 4.586186490185376, + "tokens_seen": 134021120 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048457372116349044, + "loss": 1.6552, + "theoretical_loss": 4.585888470448501, + "tokens_seen": 134086656 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004845636910732197, + "loss": 1.6349, + "theoretical_loss": 4.585590637097741, + "tokens_seen": 134152192 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048455366098294886, + "loss": 1.6911, + "theoretical_loss": 4.585292989925557, + "tokens_seen": 134217728 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048454363089267804, + "loss": 1.6548, + "theoretical_loss": 4.584995528724745, + "tokens_seen": 134283264 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.001881853211671114, + "objective/train/docs_used": 87535, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1740124225616455, + "objective/train/original_loss": 3.1740124225616455, + "objective/train/theoretical_loss": 4.584698253288433, + "objective/train/tokens_used": 154808800, + "objective/train/value_avg": -0.0036468505859375, + "objective/train/value_loss": 8.56215920066461e-05, + "objective/train/value_max": -0.00013768672943115234, + "objective/train/value_min": -0.11553955078125, + "objective/train/value_reward_corr": 0.08093028224795444, + "objective/train/value_std": 0.0036067962646484375, + "objective/train/weight_avg": 0.9849144816398621, + "objective/train/weighted_lm_loss": 3.1133203506469727, + "objective/train/weights_max": 2.9991166591644287, + "objective/train/weights_min": 0.1907527595758438, + "theoretical_loss": 4.584698253288433, + "tokens_seen": 134348800 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004845336008024072, + "loss": 1.5881, + "theoretical_loss": 4.584698253288433, + "tokens_seen": 134348800 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048452357071213645, + "loss": 1.7095, + "theoretical_loss": 4.58440116341008, + "tokens_seen": 134414336 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004845135406218656, + "loss": 1.833, + "theoretical_loss": 4.584104258883473, + "tokens_seen": 134479872 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004845035105315948, + "loss": 1.5628, + "theoretical_loss": 4.58380753950273, + "tokens_seen": 134545408 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048449348044132394, + "loss": 1.5964, + "theoretical_loss": 4.583511005062297, + "tokens_seen": 134610944 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004844834503510532, + "loss": 1.8395, + "theoretical_loss": 4.583214655356951, + "tokens_seen": 134676480 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048447342026078236, + "loss": 1.78, + "theoretical_loss": 4.5829184901817905, + "tokens_seen": 134742016 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048446339017051154, + "loss": 1.7715, + "theoretical_loss": 4.582622509332248, + "tokens_seen": 134807552 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004844533600802407, + "loss": 1.7026, + "theoretical_loss": 4.582326712604077, + "tokens_seen": 134873088 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004844433299899699, + "loss": 1.6382, + "theoretical_loss": 4.58203109979336, + "tokens_seen": 134938624 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004844332998996991, + "loss": 1.7192, + "theoretical_loss": 4.581735670696499, + "tokens_seen": 135004160 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004844232698094283, + "loss": 1.594, + "theoretical_loss": 4.581440425110225, + "tokens_seen": 135069696 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048441323971915745, + "loss": 1.7305, + "theoretical_loss": 4.581145362831592, + "tokens_seen": 135135232 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004844032096288867, + "loss": 1.6992, + "theoretical_loss": 4.580850483657974, + "tokens_seen": 135200768 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048439317953861586, + "loss": 1.6582, + "theoretical_loss": 4.580555787387068, + "tokens_seen": 135266304 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048438314944834504, + "loss": 1.7257, + "theoretical_loss": 4.580261273816895, + "tokens_seen": 135331840 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004843731193580742, + "loss": 1.7847, + "theoretical_loss": 4.579966942745792, + "tokens_seen": 135397376 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004843630892678034, + "loss": 1.707, + "theoretical_loss": 4.5796727939724216, + "tokens_seen": 135462912 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004843530591775326, + "loss": 1.6838, + "theoretical_loss": 4.579378827295761, + "tokens_seen": 135528448 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004843430290872618, + "loss": 1.6469, + "theoretical_loss": 4.579085042515108, + "tokens_seen": 135593984 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048433299899699095, + "loss": 1.5703, + "theoretical_loss": 4.578791439430079, + "tokens_seen": 135659520 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004843229689067202, + "loss": 1.5585, + "theoretical_loss": 4.578498017840608, + "tokens_seen": 135725056 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004843129388164493, + "loss": 1.7014, + "theoretical_loss": 4.578204777546944, + "tokens_seen": 135790592 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048430290872617855, + "loss": 1.613, + "theoretical_loss": 4.577911718349654, + "tokens_seen": 135856128 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048429287863590773, + "loss": 1.7328, + "theoretical_loss": 4.5776188400496185, + "tokens_seen": 135921664 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0014367452822625637, + "objective/train/docs_used": 88189, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1207454204559326, + "objective/train/original_loss": 3.1207456588745117, + "objective/train/theoretical_loss": 4.577326142448035, + "objective/train/tokens_used": 156447200, + "objective/train/value_avg": -0.004486083984375, + "objective/train/value_loss": 7.657230889890343e-05, + "objective/train/value_max": -0.00017130374908447266, + "objective/train/value_min": -0.14208984375, + "objective/train/value_reward_corr": 0.18337797529611538, + "objective/train/value_std": 0.00434112548828125, + "objective/train/weight_avg": 0.9893320202827454, + "objective/train/weighted_lm_loss": 3.0562238693237305, + "objective/train/weights_max": 4.140839099884033, + "objective/train/weights_min": 0.37720155715942383, + "theoretical_loss": 4.577326142448035, + "tokens_seen": 135987200 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004842828485456369, + "loss": 1.6084, + "theoretical_loss": 4.577326142448035, + "tokens_seen": 135987200 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004842728184553661, + "loss": 1.5753, + "theoretical_loss": 4.577033625346415, + "tokens_seen": 136052736 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048426278836509527, + "loss": 1.504, + "theoretical_loss": 4.576741288546581, + "tokens_seen": 136118272 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048425275827482445, + "loss": 1.838, + "theoretical_loss": 4.576449131850669, + "tokens_seen": 136183808 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004842427281845537, + "loss": 1.641, + "theoretical_loss": 4.576157155061131, + "tokens_seen": 136249344 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048423269809428287, + "loss": 1.6226, + "theoretical_loss": 4.575865357980726, + "tokens_seen": 136314880 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048422266800401205, + "loss": 1.628, + "theoretical_loss": 4.575573740412527, + "tokens_seen": 136380416 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004842126379137413, + "loss": 1.5225, + "theoretical_loss": 4.575282302159915, + "tokens_seen": 136445952 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004842026078234704, + "loss": 1.64, + "theoretical_loss": 4.574991043026582, + "tokens_seen": 136511488 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048419257773319965, + "loss": 1.6136, + "theoretical_loss": 4.574699962816529, + "tokens_seen": 136577024 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004841825476429288, + "loss": 1.6679, + "theoretical_loss": 4.574409061334064, + "tokens_seen": 136642560 + }, + { + "epoch": 0.04, + "learning_rate": 0.000484172517552658, + "loss": 1.7105, + "theoretical_loss": 4.574118338383805, + "tokens_seen": 136708096 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004841624874623872, + "loss": 1.6396, + "theoretical_loss": 4.5738277937706755, + "tokens_seen": 136773632 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048415245737211637, + "loss": 1.6266, + "theoretical_loss": 4.573537427299906, + "tokens_seen": 136839168 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048414242728184555, + "loss": 1.8165, + "theoretical_loss": 4.573247238777032, + "tokens_seen": 136904704 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048413239719157473, + "loss": 1.5763, + "theoretical_loss": 4.572957228007897, + "tokens_seen": 136970240 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004841223671013039, + "loss": 1.6015, + "theoretical_loss": 4.572667394798646, + "tokens_seen": 137035776 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048411233701103315, + "loss": 1.6445, + "theoretical_loss": 4.572377738955729, + "tokens_seen": 137101312 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004841023069207623, + "loss": 1.7455, + "theoretical_loss": 4.5720882602859, + "tokens_seen": 137166848 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004840922768304915, + "loss": 1.7451, + "theoretical_loss": 4.571798958596215, + "tokens_seen": 137232384 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048408224674022064, + "loss": 1.5776, + "theoretical_loss": 4.571509833694032, + "tokens_seen": 137297920 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004840722166499499, + "loss": 1.6567, + "theoretical_loss": 4.571220885387013, + "tokens_seen": 137363456 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048406218655967906, + "loss": 1.8204, + "theoretical_loss": 4.570932113483116, + "tokens_seen": 137428992 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048405215646940824, + "loss": 1.8214, + "theoretical_loss": 4.570643517790604, + "tokens_seen": 137494528 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004840421263791374, + "loss": 1.6302, + "theoretical_loss": 4.570355098118037, + "tokens_seen": 137560064 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -5.870992026757449e-05, + "objective/train/docs_used": 88907, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1448848247528076, + "objective/train/original_loss": 3.1448850631713867, + "objective/train/theoretical_loss": 4.570066854274275, + "objective/train/tokens_used": 158085600, + "objective/train/value_avg": -0.00441741943359375, + "objective/train/value_loss": 6.887120252940804e-05, + "objective/train/value_max": -0.00019872188568115234, + "objective/train/value_min": -0.10430908203125, + "objective/train/value_reward_corr": 0.07953465510229804, + "objective/train/value_std": 0.00409698486328125, + "objective/train/weight_avg": 1.0026212930679321, + "objective/train/weighted_lm_loss": 3.1487481594085693, + "objective/train/weights_max": 2.6180918216705322, + "objective/train/weights_min": 0.2909766137599945, + "theoretical_loss": 4.570066854274275, + "tokens_seen": 137625600 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048403209628886665, + "loss": 1.6789, + "theoretical_loss": 4.570066854274275, + "tokens_seen": 137625600 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004840220661985958, + "loss": 1.7796, + "theoretical_loss": 4.569778786068477, + "tokens_seen": 137691136 + }, + { + "epoch": 0.04, + "learning_rate": 0.000484012036108325, + "loss": 1.6901, + "theoretical_loss": 4.569490893310098, + "tokens_seen": 137756672 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048400200601805414, + "loss": 1.6, + "theoretical_loss": 4.569203175808893, + "tokens_seen": 137822208 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004839919759277834, + "loss": 1.6573, + "theoretical_loss": 4.5689156333749095, + "tokens_seen": 137887744 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048398194583751256, + "loss": 1.4978, + "theoretical_loss": 4.5686282658184965, + "tokens_seen": 137953280 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048397191574724174, + "loss": 1.6209, + "theoretical_loss": 4.568341072950292, + "tokens_seen": 138018816 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004839618856569709, + "loss": 1.6138, + "theoretical_loss": 4.568054054581236, + "tokens_seen": 138084352 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004839518555667001, + "loss": 1.7132, + "theoretical_loss": 4.5677672105225575, + "tokens_seen": 138149888 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004839418254764293, + "loss": 1.7499, + "theoretical_loss": 4.5674805405857795, + "tokens_seen": 138215424 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004839317953861585, + "loss": 1.7153, + "theoretical_loss": 4.56719404458272, + "tokens_seen": 138280960 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048392176529588765, + "loss": 1.6228, + "theoretical_loss": 4.56690772232549, + "tokens_seen": 138346496 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004839117352056169, + "loss": 1.6519, + "theoretical_loss": 4.566621573626489, + "tokens_seen": 138412032 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048390170511534606, + "loss": 1.7298, + "theoretical_loss": 4.56633559829841, + "tokens_seen": 138477568 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048389167502507524, + "loss": 1.5896, + "theoretical_loss": 4.566049796154239, + "tokens_seen": 138543104 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004838816449348044, + "loss": 1.7408, + "theoretical_loss": 4.565764167007247, + "tokens_seen": 138608640 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004838716148445336, + "loss": 1.796, + "theoretical_loss": 4.565478710670996, + "tokens_seen": 138674176 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004838615847542628, + "loss": 1.712, + "theoretical_loss": 4.565193426959341, + "tokens_seen": 138739712 + }, + { + "epoch": 0.04, + "learning_rate": 0.000483851554663992, + "loss": 1.7661, + "theoretical_loss": 4.56490831568642, + "tokens_seen": 138805248 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048384152457372115, + "loss": 1.7695, + "theoretical_loss": 4.5646233766666615, + "tokens_seen": 138870784 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004838314944834504, + "loss": 1.6435, + "theoretical_loss": 4.564338609714781, + "tokens_seen": 138936320 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004838214643931795, + "loss": 1.721, + "theoretical_loss": 4.56405401464578, + "tokens_seen": 139001856 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048381143430290875, + "loss": 1.6981, + "theoretical_loss": 4.563769591274946, + "tokens_seen": 139067392 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048380140421263793, + "loss": 1.789, + "theoretical_loss": 4.563485339417852, + "tokens_seen": 139132928 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004837913741223671, + "loss": 1.7187, + "theoretical_loss": 4.563201258890357, + "tokens_seen": 139198464 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0024370872415602207, + "objective/train/docs_used": 90119, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 4.033658504486084, + "objective/train/original_loss": 4.033658504486084, + "objective/train/theoretical_loss": 4.562917349508602, + "objective/train/tokens_used": 159724000, + "objective/train/value_avg": -0.0036449432373046875, + "objective/train/value_loss": 9.656514885136858e-05, + "objective/train/value_max": -0.0003101825714111328, + "objective/train/value_min": -0.07171630859375, + "objective/train/value_reward_corr": 0.23375534603326525, + "objective/train/value_std": 0.00307464599609375, + "objective/train/weight_avg": 0.9801108837127686, + "objective/train/weighted_lm_loss": 3.9586222171783447, + "objective/train/weights_max": 1.9713163375854492, + "objective/train/weights_min": 0.423747718334198, + "theoretical_loss": 4.562917349508602, + "tokens_seen": 139264000 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004837813440320963, + "loss": 1.823, + "theoretical_loss": 4.562917349508602, + "tokens_seen": 139264000 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048377131394182547, + "loss": 1.6107, + "theoretical_loss": 4.562633611089014, + "tokens_seen": 139329536 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048376128385155465, + "loss": 1.6586, + "theoretical_loss": 4.562350043448301, + "tokens_seen": 139395072 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004837512537612839, + "loss": 1.7713, + "theoretical_loss": 4.562066646403457, + "tokens_seen": 139460608 + }, + { + "epoch": 0.04, + "learning_rate": 0.000483741223671013, + "loss": 1.761, + "theoretical_loss": 4.561783419771752, + "tokens_seen": 139526144 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048373119358074225, + "loss": 1.6072, + "theoretical_loss": 4.561500363370744, + "tokens_seen": 139591680 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048372116349047143, + "loss": 1.6233, + "theoretical_loss": 4.561217477018268, + "tokens_seen": 139657216 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004837111334002006, + "loss": 1.8276, + "theoretical_loss": 4.560934760532439, + "tokens_seen": 139722752 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004837011033099298, + "loss": 1.6609, + "theoretical_loss": 4.560652213731656, + "tokens_seen": 139788288 + }, + { + "epoch": 0.04, + "learning_rate": 0.000483691073219659, + "loss": 1.7408, + "theoretical_loss": 4.560369836434591, + "tokens_seen": 139853824 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048368104312938816, + "loss": 1.6502, + "theoretical_loss": 4.5600876284602, + "tokens_seen": 139919360 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004836710130391174, + "loss": 1.5777, + "theoretical_loss": 4.559805589627714, + "tokens_seen": 139984896 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004836609829488465, + "loss": 1.5579, + "theoretical_loss": 4.559523719756642, + "tokens_seen": 140050432 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048365095285857575, + "loss": 1.654, + "theoretical_loss": 4.559242018666771, + "tokens_seen": 140115968 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004836409227683049, + "loss": 1.604, + "theoretical_loss": 4.558960486178163, + "tokens_seen": 140181504 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004836308926780341, + "loss": 1.6359, + "theoretical_loss": 4.5586791221111564, + "tokens_seen": 140247040 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004836208625877633, + "loss": 1.7231, + "theoretical_loss": 4.558397926286366, + "tokens_seen": 140312576 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004836108324974925, + "loss": 1.6543, + "theoretical_loss": 4.5581168985246805, + "tokens_seen": 140378112 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048360080240722166, + "loss": 1.8164, + "theoretical_loss": 4.557836038647261, + "tokens_seen": 140443648 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048359077231695084, + "loss": 1.637, + "theoretical_loss": 4.557555346475546, + "tokens_seen": 140509184 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048358074222668, + "loss": 1.6681, + "theoretical_loss": 4.557274821831242, + "tokens_seen": 140574720 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048357071213640926, + "loss": 1.6737, + "theoretical_loss": 4.556994464536334, + "tokens_seen": 140640256 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004835606820461384, + "loss": 1.6459, + "theoretical_loss": 4.556714274413074, + "tokens_seen": 140705792 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004835506519558676, + "loss": 1.6821, + "theoretical_loss": 4.556434251283989, + "tokens_seen": 140771328 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004835406218655968, + "loss": 1.6631, + "theoretical_loss": 4.556154394971874, + "tokens_seen": 140836864 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.003126199124380946, + "objective/train/docs_used": 90792, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.90812611579895, + "objective/train/original_loss": 3.908125400543213, + "objective/train/theoretical_loss": 4.5558747052997965, + "objective/train/tokens_used": 161362400, + "objective/train/value_avg": -0.004940032958984375, + "objective/train/value_loss": 0.00010934489546343684, + "objective/train/value_max": -0.0002694129943847656, + "objective/train/value_min": -0.056854248046875, + "objective/train/value_reward_corr": 0.22924895661197672, + "objective/train/value_std": 0.00414276123046875, + "objective/train/weight_avg": 0.97386634349823, + "objective/train/weighted_lm_loss": 3.804786205291748, + "objective/train/weights_max": 1.7126208543777466, + "objective/train/weights_min": 0.4921513497829437, + "theoretical_loss": 4.5558747052997965, + "tokens_seen": 140902400 + }, + { + "epoch": 0.04, + "learning_rate": 0.000483530591775326, + "loss": 1.6877, + "theoretical_loss": 4.5558747052997965, + "tokens_seen": 140902400 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048352056168505516, + "loss": 1.7848, + "theoretical_loss": 4.555595182091094, + "tokens_seen": 140967936 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048351053159478434, + "loss": 1.6097, + "theoretical_loss": 4.555315825169371, + "tokens_seen": 141033472 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004835005015045135, + "loss": 1.6221, + "theoretical_loss": 4.555036634358502, + "tokens_seen": 141099008 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048349047141424276, + "loss": 1.6534, + "theoretical_loss": 4.554757609482632, + "tokens_seen": 141164544 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048348044132397194, + "loss": 1.6256, + "theoretical_loss": 4.554478750366169, + "tokens_seen": 141230080 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004834704112337011, + "loss": 1.8275, + "theoretical_loss": 4.5542000568337935, + "tokens_seen": 141295616 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004834603811434303, + "loss": 1.7212, + "theoretical_loss": 4.553921528710448, + "tokens_seen": 141361152 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004834503510531595, + "loss": 1.6936, + "theoretical_loss": 4.553643165821343, + "tokens_seen": 141426688 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004834403209628887, + "loss": 1.7475, + "theoretical_loss": 4.553364967991957, + "tokens_seen": 141492224 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048343029087261785, + "loss": 1.7287, + "theoretical_loss": 4.553086935048029, + "tokens_seen": 141557760 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004834202607823471, + "loss": 1.5952, + "theoretical_loss": 4.552809066815566, + "tokens_seen": 141623296 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048341023069207626, + "loss": 1.6425, + "theoretical_loss": 4.552531363120837, + "tokens_seen": 141688832 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048340020060180544, + "loss": 1.6327, + "theoretical_loss": 4.552253823790375, + "tokens_seen": 141754368 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004833901705115346, + "loss": 1.7634, + "theoretical_loss": 4.5519764486509775, + "tokens_seen": 141819904 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004833801404212638, + "loss": 1.7306, + "theoretical_loss": 4.551699237529702, + "tokens_seen": 141885440 + }, + { + "epoch": 0.04, + "learning_rate": 0.000483370110330993, + "loss": 1.7074, + "theoretical_loss": 4.55142219025387, + "tokens_seen": 141950976 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004833600802407222, + "loss": 1.6394, + "theoretical_loss": 4.551145306651064, + "tokens_seen": 142016512 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048335005015045135, + "loss": 1.5773, + "theoretical_loss": 4.550868586549125, + "tokens_seen": 142082048 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004833400200601806, + "loss": 1.8448, + "theoretical_loss": 4.550592029776158, + "tokens_seen": 142147584 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004833299899699097, + "loss": 1.5938, + "theoretical_loss": 4.550315636160526, + "tokens_seen": 142213120 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048331995987963895, + "loss": 1.7092, + "theoretical_loss": 4.550039405530851, + "tokens_seen": 142278656 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048330992978936813, + "loss": 1.77, + "theoretical_loss": 4.549763337716014, + "tokens_seen": 142344192 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004832998996990973, + "loss": 1.6479, + "theoretical_loss": 4.549487432545156, + "tokens_seen": 142409728 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004832898696088265, + "loss": 1.7929, + "theoretical_loss": 4.549211689847674, + "tokens_seen": 142475264 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.00023376510944217443, + "objective/train/docs_used": 92175, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9430620670318604, + "objective/train/original_loss": 2.9430620670318604, + "objective/train/theoretical_loss": 4.548936109453223, + "objective/train/tokens_used": 163000800, + "objective/train/value_avg": -0.0039825439453125, + "objective/train/value_loss": 5.814094765810296e-05, + "objective/train/value_max": -0.0002359151840209961, + "objective/train/value_min": -0.2186279296875, + "objective/train/value_reward_corr": 0.10043027659974987, + "objective/train/value_std": 0.004848480224609375, + "objective/train/weight_avg": 1.00080406665802, + "objective/train/weighted_lm_loss": 2.9502735137939453, + "objective/train/weights_max": 8.345840454101562, + "objective/train/weights_min": 0.13761156797409058, + "theoretical_loss": 4.548936109453223, + "tokens_seen": 142540800 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048327983951855567, + "loss": 1.6619, + "theoretical_loss": 4.548936109453223, + "tokens_seen": 142540800 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048326980942828485, + "loss": 1.7235, + "theoretical_loss": 4.5486606911917145, + "tokens_seen": 142606336 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004832597793380141, + "loss": 1.7245, + "theoretical_loss": 4.548385434893316, + "tokens_seen": 142671872 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004832497492477432, + "loss": 1.7541, + "theoretical_loss": 4.548110340388453, + "tokens_seen": 142737408 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048323971915747245, + "loss": 1.6581, + "theoretical_loss": 4.547835407507803, + "tokens_seen": 142802944 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048322968906720163, + "loss": 1.6029, + "theoretical_loss": 4.5475606360822995, + "tokens_seen": 142868480 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004832196589769308, + "loss": 1.8581, + "theoretical_loss": 4.547286025943132, + "tokens_seen": 142934016 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048320962888666, + "loss": 1.6852, + "theoretical_loss": 4.547011576921742, + "tokens_seen": 142999552 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004831995987963892, + "loss": 1.7312, + "theoretical_loss": 4.546737288849825, + "tokens_seen": 143065088 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048318956870611836, + "loss": 1.7785, + "theoretical_loss": 4.546463161559329, + "tokens_seen": 143130624 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004831795386158476, + "loss": 1.674, + "theoretical_loss": 4.546189194882455, + "tokens_seen": 143196160 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004831695085255767, + "loss": 1.7001, + "theoretical_loss": 4.545915388651654, + "tokens_seen": 143261696 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048315947843530595, + "loss": 1.6633, + "theoretical_loss": 4.54564174269963, + "tokens_seen": 143327232 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004831494483450351, + "loss": 1.8128, + "theoretical_loss": 4.545368256859338, + "tokens_seen": 143392768 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004831394182547643, + "loss": 1.7099, + "theoretical_loss": 4.545094930963982, + "tokens_seen": 143458304 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004831293881644935, + "loss": 1.7269, + "theoretical_loss": 4.5448217648470175, + "tokens_seen": 143523840 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004831193580742227, + "loss": 1.6174, + "theoretical_loss": 4.544548758342148, + "tokens_seen": 143589376 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048310932798395186, + "loss": 1.8343, + "theoretical_loss": 4.544275911283326, + "tokens_seen": 143654912 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048309929789368104, + "loss": 1.7405, + "theoretical_loss": 4.5440032235047525, + "tokens_seen": 143720448 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004830892678034102, + "loss": 1.6824, + "theoretical_loss": 4.543730694840878, + "tokens_seen": 143785984 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048307923771313946, + "loss": 1.6883, + "theoretical_loss": 4.543458325126398, + "tokens_seen": 143851520 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004830692076228686, + "loss": 1.6831, + "theoretical_loss": 4.543186114196256, + "tokens_seen": 143917056 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004830591775325978, + "loss": 1.6102, + "theoretical_loss": 4.542914061885643, + "tokens_seen": 143982592 + }, + { + "epoch": 0.04, + "learning_rate": 0.000483049147442327, + "loss": 1.7479, + "theoretical_loss": 4.542642168029994, + "tokens_seen": 144048128 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004830391173520562, + "loss": 1.7309, + "theoretical_loss": 4.54237043246499, + "tokens_seen": 144113664 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0009486679336987436, + "objective/train/docs_used": 92798, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9092118740081787, + "objective/train/original_loss": 2.9092118740081787, + "objective/train/theoretical_loss": 4.542098855026559, + "objective/train/tokens_used": 164639200, + "objective/train/value_avg": -0.0037784576416015625, + "objective/train/value_loss": 4.9420545110479e-05, + "objective/train/value_max": -0.00014317035675048828, + "objective/train/value_min": -0.10357666015625, + "objective/train/value_reward_corr": 0.14084773533332057, + "objective/train/value_std": 0.003475189208984375, + "objective/train/weight_avg": 0.9928898215293884, + "objective/train/weighted_lm_loss": 2.866554021835327, + "objective/train/weights_max": 2.817265272140503, + "objective/train/weights_min": 0.24196821451187134, + "theoretical_loss": 4.542098855026559, + "tokens_seen": 144179200 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048302908726178536, + "loss": 1.6441, + "theoretical_loss": 4.542098855026559, + "tokens_seen": 144179200 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048301905717151454, + "loss": 1.7623, + "theoretical_loss": 4.54182743555087, + "tokens_seen": 144244736 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004830090270812437, + "loss": 1.7225, + "theoretical_loss": 4.54155617387434, + "tokens_seen": 144310272 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048299899699097296, + "loss": 1.6403, + "theoretical_loss": 4.541285069833626, + "tokens_seen": 144375808 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004829889669007021, + "loss": 1.6373, + "theoretical_loss": 4.541014123265632, + "tokens_seen": 144441344 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004829789368104313, + "loss": 1.6098, + "theoretical_loss": 4.5407433340075, + "tokens_seen": 144506880 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048296890672016045, + "loss": 1.6685, + "theoretical_loss": 4.5404727018966184, + "tokens_seen": 144572416 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004829588766298897, + "loss": 1.8045, + "theoretical_loss": 4.5402022267706155, + "tokens_seen": 144637952 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048294884653961887, + "loss": 1.6942, + "theoretical_loss": 4.539931908467359, + "tokens_seen": 144703488 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048293881644934805, + "loss": 1.6564, + "theoretical_loss": 4.539661746824962, + "tokens_seen": 144769024 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048292878635907723, + "loss": 1.6733, + "theoretical_loss": 4.539391741681774, + "tokens_seen": 144834560 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048291875626880646, + "loss": 1.6406, + "theoretical_loss": 4.539121892876385, + "tokens_seen": 144900096 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004829087261785356, + "loss": 1.7074, + "theoretical_loss": 4.538852200247625, + "tokens_seen": 144965632 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004828986960882648, + "loss": 1.7999, + "theoretical_loss": 4.538582663634564, + "tokens_seen": 145031168 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048288866599799395, + "loss": 1.7908, + "theoretical_loss": 4.5383132828765085, + "tokens_seen": 145096704 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004828786359077232, + "loss": 1.7969, + "theoretical_loss": 4.538044057813003, + "tokens_seen": 145162240 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048286860581745237, + "loss": 1.6786, + "theoretical_loss": 4.5377749882838305, + "tokens_seen": 145227776 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048285857572718155, + "loss": 1.7756, + "theoretical_loss": 4.537506074129013, + "tokens_seen": 145293312 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048284854563691073, + "loss": 1.6807, + "theoretical_loss": 4.537237315188804, + "tokens_seen": 145358848 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004828385155466399, + "loss": 1.6336, + "theoretical_loss": 4.536968711303699, + "tokens_seen": 145424384 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004828284854563691, + "loss": 1.7089, + "theoretical_loss": 4.536700262314425, + "tokens_seen": 145489920 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048281845536609833, + "loss": 1.7157, + "theoretical_loss": 4.536431968061946, + "tokens_seen": 145555456 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048280842527582746, + "loss": 1.6388, + "theoretical_loss": 4.536163828387459, + "tokens_seen": 145620992 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004827983951855567, + "loss": 1.7143, + "theoretical_loss": 4.535895843132399, + "tokens_seen": 145686528 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004827883650952858, + "loss": 1.8023, + "theoretical_loss": 4.53562801213843, + "tokens_seen": 145752064 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": -0.0014763283543288708, + "objective/train/docs_used": 94036, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0775508880615234, + "objective/train/original_loss": 3.0775508880615234, + "objective/train/theoretical_loss": 4.535360335247455, + "objective/train/tokens_used": 166277600, + "objective/train/value_avg": -0.00433349609375, + "objective/train/value_loss": 8.298490865854546e-05, + "objective/train/value_max": -0.0002359151840209961, + "objective/train/value_min": -0.14990234375, + "objective/train/value_reward_corr": 0.10493461355568282, + "objective/train/value_std": 0.0038585662841796875, + "objective/train/weight_avg": 0.988985538482666, + "objective/train/weighted_lm_loss": 3.0320470333099365, + "objective/train/weights_max": 4.242955207824707, + "objective/train/weights_min": 0.1932816356420517, + "theoretical_loss": 4.535360335247455, + "tokens_seen": 145817600 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048277833500501505, + "loss": 1.5763, + "theoretical_loss": 4.535360335247455, + "tokens_seen": 145817600 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048276830491474423, + "loss": 1.7121, + "theoretical_loss": 4.535092812301606, + "tokens_seen": 145883136 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004827582748244734, + "loss": 1.6225, + "theoretical_loss": 4.534825443143248, + "tokens_seen": 145948672 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004827482447342026, + "loss": 1.7207, + "theoretical_loss": 4.53455822761498, + "tokens_seen": 146014208 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048273821464393183, + "loss": 1.6605, + "theoretical_loss": 4.53429116555963, + "tokens_seen": 146079744 + }, + { + "epoch": 0.04, + "learning_rate": 0.000482728184553661, + "loss": 1.6165, + "theoretical_loss": 4.53402425682026, + "tokens_seen": 146145280 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004827181544633902, + "loss": 1.7206, + "theoretical_loss": 4.53375750124016, + "tokens_seen": 146210816 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004827081243731194, + "loss": 1.6719, + "theoretical_loss": 4.533490898662853, + "tokens_seen": 146276352 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048269809428284856, + "loss": 1.6776, + "theoretical_loss": 4.53322444893209, + "tokens_seen": 146341888 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004826880641925778, + "loss": 1.582, + "theoretical_loss": 4.532958151891852, + "tokens_seen": 146407424 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004826780341023069, + "loss": 1.6454, + "theoretical_loss": 4.53269200738635, + "tokens_seen": 146472960 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048266800401203615, + "loss": 1.6585, + "theoretical_loss": 4.532426015260021, + "tokens_seen": 146538496 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004826579739217653, + "loss": 1.6864, + "theoretical_loss": 4.532160175357533, + "tokens_seen": 146604032 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004826479438314945, + "loss": 1.6035, + "theoretical_loss": 4.531894487523781, + "tokens_seen": 146669568 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004826379137412237, + "loss": 1.6368, + "theoretical_loss": 4.531628951603885, + "tokens_seen": 146735104 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004826278836509529, + "loss": 1.6852, + "theoretical_loss": 4.531363567443194, + "tokens_seen": 146800640 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048261785356068206, + "loss": 1.7629, + "theoretical_loss": 4.531098334887284, + "tokens_seen": 146866176 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048260782347041124, + "loss": 1.6465, + "theoretical_loss": 4.530833253781955, + "tokens_seen": 146931712 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004825977933801404, + "loss": 1.6537, + "theoretical_loss": 4.530568323973234, + "tokens_seen": 146997248 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048258776328986966, + "loss": 1.8182, + "theoretical_loss": 4.530303545307371, + "tokens_seen": 147062784 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004825777331995988, + "loss": 1.7244, + "theoretical_loss": 4.530038917630844, + "tokens_seen": 147128320 + }, + { + "epoch": 0.04, + "learning_rate": 0.000482567703109328, + "loss": 1.7627, + "theoretical_loss": 4.529774440790353, + "tokens_seen": 147193856 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004825576730190572, + "loss": 1.788, + "theoretical_loss": 4.529510114632821, + "tokens_seen": 147259392 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004825476429287864, + "loss": 1.6979, + "theoretical_loss": 4.529245939005397, + "tokens_seen": 147324928 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048253761283851556, + "loss": 1.6705, + "theoretical_loss": 4.528981913755452, + "tokens_seen": 147390464 + }, + { + "epoch": 0.04, + "objective/train/advantage_avg": 9.767048322828487e-05, + "objective/train/docs_used": 94726, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0011119842529297, + "objective/train/original_loss": 3.0011119842529297, + "objective/train/theoretical_loss": 4.5287180387305765, + "objective/train/tokens_used": 167916000, + "objective/train/value_avg": -0.00397491455078125, + "objective/train/value_loss": 4.726049519376829e-05, + "objective/train/value_max": -0.00025725364685058594, + "objective/train/value_min": -0.0753173828125, + "objective/train/value_reward_corr": 0.1136570090155205, + "objective/train/value_std": 0.00351715087890625, + "objective/train/weight_avg": 1.0032274723052979, + "objective/train/weighted_lm_loss": 3.007887363433838, + "objective/train/weights_max": 1.8675719499588013, + "objective/train/weights_min": 0.42085304856300354, + "theoretical_loss": 4.5287180387305765, + "tokens_seen": 147456000 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048252758274824474, + "loss": 1.7382, + "theoretical_loss": 4.5287180387305765, + "tokens_seen": 147456000 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004825175526579739, + "loss": 1.8354, + "theoretical_loss": 4.528454313778588, + "tokens_seen": 147521536 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048250752256770316, + "loss": 1.6657, + "theoretical_loss": 4.528190738747523, + "tokens_seen": 147587072 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004824974924774323, + "loss": 1.6687, + "theoretical_loss": 4.527927313485639, + "tokens_seen": 147652608 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004824874623871615, + "loss": 1.6624, + "theoretical_loss": 4.527664037841416, + "tokens_seen": 147718144 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048247743229689065, + "loss": 1.4999, + "theoretical_loss": 4.527400911663553, + "tokens_seen": 147783680 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004824674022066199, + "loss": 1.7878, + "theoretical_loss": 4.527137934800969, + "tokens_seen": 147849216 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048245737211634907, + "loss": 1.7912, + "theoretical_loss": 4.526875107102803, + "tokens_seen": 147914752 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048244734202607825, + "loss": 1.6963, + "theoretical_loss": 4.526612428418414, + "tokens_seen": 147980288 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048243731193580743, + "loss": 1.6893, + "theoretical_loss": 4.5263498985973785, + "tokens_seen": 148045824 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048242728184553666, + "loss": 1.7717, + "theoretical_loss": 4.526087517489492, + "tokens_seen": 148111360 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004824172517552658, + "loss": 1.5747, + "theoretical_loss": 4.5258252849447675, + "tokens_seen": 148176896 + }, + { + "epoch": 0.04, + "learning_rate": 0.000482407221664995, + "loss": 1.6531, + "theoretical_loss": 4.525563200813437, + "tokens_seen": 148242432 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048239719157472415, + "loss": 1.7475, + "theoretical_loss": 4.525301264945947, + "tokens_seen": 148307968 + }, + { + "epoch": 0.04, + "learning_rate": 0.0004823871614844534, + "loss": 1.7346, + "theoretical_loss": 4.525039477192962, + "tokens_seen": 148373504 + }, + { + "epoch": 0.04, + "learning_rate": 0.00048237713139418257, + "loss": 1.7296, + "theoretical_loss": 4.524777837405364, + "tokens_seen": 148439040 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048236710130391175, + "loss": 1.5325, + "theoretical_loss": 4.5245163454342485, + "tokens_seen": 148504576 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048235707121364093, + "loss": 1.6688, + "theoretical_loss": 4.524255001130929, + "tokens_seen": 148570112 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004823470411233701, + "loss": 1.7433, + "theoretical_loss": 4.523993804346933, + "tokens_seen": 148635648 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004823370110330993, + "loss": 1.646, + "theoretical_loss": 4.5237327549340005, + "tokens_seen": 148701184 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048232698094282853, + "loss": 1.7242, + "theoretical_loss": 4.523471852744089, + "tokens_seen": 148766720 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048231695085255766, + "loss": 1.717, + "theoretical_loss": 4.52321109762937, + "tokens_seen": 148832256 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004823069207622869, + "loss": 1.609, + "theoretical_loss": 4.522950489442225, + "tokens_seen": 148897792 + }, + { + "epoch": 0.05, + "learning_rate": 0.000482296890672016, + "loss": 1.6365, + "theoretical_loss": 4.522690028035251, + "tokens_seen": 148963328 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048228686058174525, + "loss": 1.7677, + "theoretical_loss": 4.522429713261258, + "tokens_seen": 149028864 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0017635049298405647, + "objective/train/docs_used": 95869, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.5543582439422607, + "objective/train/original_loss": 3.5543582439422607, + "objective/train/theoretical_loss": 4.522169544973266, + "objective/train/tokens_used": 169554400, + "objective/train/value_avg": -0.0045166015625, + "objective/train/value_loss": 0.00012174026778666303, + "objective/train/value_max": -0.00028014183044433594, + "objective/train/value_min": -0.09619140625, + "objective/train/value_reward_corr": 0.12214163460548841, + "objective/train/value_std": 0.00437164306640625, + "objective/train/weight_avg": 0.9878044128417969, + "objective/train/weighted_lm_loss": 3.49035906791687, + "objective/train/weights_max": 2.594200849533081, + "objective/train/weights_min": 0.1926669478416443, + "theoretical_loss": 4.522169544973266, + "tokens_seen": 149094400 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048227683049147443, + "loss": 1.6665, + "theoretical_loss": 4.522169544973266, + "tokens_seen": 149094400 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004822668004012036, + "loss": 1.6777, + "theoretical_loss": 4.52190952302451, + "tokens_seen": 149159936 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004822567703109328, + "loss": 1.7128, + "theoretical_loss": 4.521649647268434, + "tokens_seen": 149225472 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048224674022066203, + "loss": 1.6631, + "theoretical_loss": 4.521389917558694, + "tokens_seen": 149291008 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048223671013039116, + "loss": 1.7219, + "theoretical_loss": 4.521130333749157, + "tokens_seen": 149356544 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004822266800401204, + "loss": 1.6976, + "theoretical_loss": 4.520870895693898, + "tokens_seen": 149422080 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004822166499498495, + "loss": 1.7452, + "theoretical_loss": 4.520611603247206, + "tokens_seen": 149487616 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048220661985957876, + "loss": 1.6274, + "theoretical_loss": 4.520352456263574, + "tokens_seen": 149553152 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048219658976930794, + "loss": 1.643, + "theoretical_loss": 4.520093454597711, + "tokens_seen": 149618688 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004821865596790371, + "loss": 1.5984, + "theoretical_loss": 4.519834598104527, + "tokens_seen": 149684224 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004821765295887663, + "loss": 1.7, + "theoretical_loss": 4.5195758866391476, + "tokens_seen": 149749760 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004821664994984955, + "loss": 1.5974, + "theoretical_loss": 4.519317320056901, + "tokens_seen": 149815296 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048215646940822466, + "loss": 1.6064, + "theoretical_loss": 4.519058898213325, + "tokens_seen": 149880832 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004821464393179539, + "loss": 1.5924, + "theoretical_loss": 4.518800620964164, + "tokens_seen": 149946368 + }, + { + "epoch": 0.05, + "learning_rate": 0.000482136409227683, + "loss": 1.6996, + "theoretical_loss": 4.51854248816537, + "tokens_seen": 150011904 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048212637913741226, + "loss": 1.6872, + "theoretical_loss": 4.5182844996731015, + "tokens_seen": 150077440 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004821163490471414, + "loss": 1.6972, + "theoretical_loss": 4.5180266553437205, + "tokens_seen": 150142976 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004821063189568706, + "loss": 1.7279, + "theoretical_loss": 4.517768955033798, + "tokens_seen": 150208512 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004820962888665998, + "loss": 1.7285, + "theoretical_loss": 4.5175113986001065, + "tokens_seen": 150274048 + }, + { + "epoch": 0.05, + "learning_rate": 0.000482086258776329, + "loss": 1.6884, + "theoretical_loss": 4.517253985899628, + "tokens_seen": 150339584 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048207622868605816, + "loss": 1.6135, + "theoretical_loss": 4.516996716789544, + "tokens_seen": 150405120 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004820661985957874, + "loss": 1.667, + "theoretical_loss": 4.516739591127244, + "tokens_seen": 150470656 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048205616850551653, + "loss": 1.5239, + "theoretical_loss": 4.516482608770319, + "tokens_seen": 150536192 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048204613841524576, + "loss": 1.7266, + "theoretical_loss": 4.516225769576564, + "tokens_seen": 150601728 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004820361083249749, + "loss": 1.5608, + "theoretical_loss": 4.515969073403976, + "tokens_seen": 150667264 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -4.462402648641728e-05, + "objective/train/docs_used": 96442, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4963104724884033, + "objective/train/original_loss": 3.496309995651245, + "objective/train/theoretical_loss": 4.515712520110756, + "objective/train/tokens_used": 171192800, + "objective/train/value_avg": -0.004100799560546875, + "objective/train/value_loss": 5.421983587439172e-05, + "objective/train/value_max": -0.0001780986785888672, + "objective/train/value_min": -0.07366943359375, + "objective/train/value_reward_corr": 0.13719458064413093, + "objective/train/value_std": 0.0033626556396484375, + "objective/train/weight_avg": 1.0021165609359741, + "objective/train/weighted_lm_loss": 3.5160083770751953, + "objective/train/weights_max": 1.8046764135360718, + "objective/train/weights_min": 0.4660395085811615, + "theoretical_loss": 4.515712520110756, + "tokens_seen": 150732800 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004820260782347041, + "loss": 1.6908, + "theoretical_loss": 4.515712520110756, + "tokens_seen": 150732800 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004820160481444333, + "loss": 1.6788, + "theoretical_loss": 4.515456109555306, + "tokens_seen": 150798336 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004820060180541625, + "loss": 1.6729, + "theoretical_loss": 4.515199841596231, + "tokens_seen": 150863872 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048199598796389167, + "loss": 1.579, + "theoretical_loss": 4.514943716092336, + "tokens_seen": 150929408 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048198595787362085, + "loss": 1.7257, + "theoretical_loss": 4.514687732902628, + "tokens_seen": 150994944 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004819759277833501, + "loss": 1.5578, + "theoretical_loss": 4.514431891886314, + "tokens_seen": 151060480 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048196589769307927, + "loss": 1.7502, + "theoretical_loss": 4.514176192902802, + "tokens_seen": 151126016 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048195586760280845, + "loss": 1.7085, + "theoretical_loss": 4.513920635811699, + "tokens_seen": 151191552 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048194583751253763, + "loss": 1.6351, + "theoretical_loss": 4.513665220472813, + "tokens_seen": 151257088 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048193580742226686, + "loss": 1.6884, + "theoretical_loss": 4.513409946746149, + "tokens_seen": 151322624 + }, + { + "epoch": 0.05, + "learning_rate": 0.000481925777331996, + "loss": 1.828, + "theoretical_loss": 4.513154814491914, + "tokens_seen": 151388160 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004819157472417252, + "loss": 1.6108, + "theoretical_loss": 4.5128998235705104, + "tokens_seen": 151453696 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048190571715145435, + "loss": 1.6684, + "theoretical_loss": 4.512644973842541, + "tokens_seen": 151519232 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004818956870611836, + "loss": 1.602, + "theoretical_loss": 4.512390265168802, + "tokens_seen": 151584768 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048188565697091277, + "loss": 1.5664, + "theoretical_loss": 4.512135697410294, + "tokens_seen": 151650304 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048187562688064195, + "loss": 1.6957, + "theoretical_loss": 4.511881270428207, + "tokens_seen": 151715840 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048186559679037113, + "loss": 1.6321, + "theoretical_loss": 4.511626984083934, + "tokens_seen": 151781376 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004818555667001003, + "loss": 1.6855, + "theoretical_loss": 4.51137283823906, + "tokens_seen": 151846912 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004818455366098295, + "loss": 1.6167, + "theoretical_loss": 4.511118832755368, + "tokens_seen": 151912448 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048183550651955873, + "loss": 1.6285, + "theoretical_loss": 4.510864967494836, + "tokens_seen": 151977984 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048182547642928786, + "loss": 1.5531, + "theoretical_loss": 4.510611242319637, + "tokens_seen": 152043520 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004818154463390171, + "loss": 1.5923, + "theoretical_loss": 4.51035765709214, + "tokens_seen": 152109056 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004818054162487462, + "loss": 1.7163, + "theoretical_loss": 4.510104211674904, + "tokens_seen": 152174592 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048179538615847545, + "loss": 1.568, + "theoretical_loss": 4.509850905930689, + "tokens_seen": 152240128 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048178535606820463, + "loss": 1.6873, + "theoretical_loss": 4.509597739722444, + "tokens_seen": 152305664 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0008836690685711801, + "objective/train/docs_used": 97778, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8549742698669434, + "objective/train/original_loss": 2.8549747467041016, + "objective/train/theoretical_loss": 4.509344712913311, + "objective/train/tokens_used": 172831200, + "objective/train/value_avg": -0.004703521728515625, + "objective/train/value_loss": 9.828743350226432e-05, + "objective/train/value_max": -0.0002378225326538086, + "objective/train/value_min": -0.2132568359375, + "objective/train/value_reward_corr": 0.06578367675293646, + "objective/train/value_std": 0.004802703857421875, + "objective/train/weight_avg": 0.9958447217941284, + "objective/train/weighted_lm_loss": 2.836376667022705, + "objective/train/weights_max": 8.436507225036621, + "objective/train/weights_min": 0.1409144550561905, + "theoretical_loss": 4.509344712913311, + "tokens_seen": 152371200 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004817753259779338, + "loss": 1.6647, + "theoretical_loss": 4.509344712913311, + "tokens_seen": 152371200 + }, + { + "epoch": 0.05, + "learning_rate": 0.000481765295887663, + "loss": 1.7148, + "theoretical_loss": 4.509091825366629, + "tokens_seen": 152436736 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048175526579739223, + "loss": 1.693, + "theoretical_loss": 4.508839076945925, + "tokens_seen": 152502272 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048174523570712136, + "loss": 1.5952, + "theoretical_loss": 4.508586467514924, + "tokens_seen": 152567808 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004817352056168506, + "loss": 1.6869, + "theoretical_loss": 4.508333996937535, + "tokens_seen": 152633344 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004817251755265797, + "loss": 1.6704, + "theoretical_loss": 4.5080816650778655, + "tokens_seen": 152698880 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048171514543630896, + "loss": 1.7872, + "theoretical_loss": 4.5078294718002105, + "tokens_seen": 152764416 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048170511534603814, + "loss": 1.5718, + "theoretical_loss": 4.50757741696906, + "tokens_seen": 152829952 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004816950852557673, + "loss": 1.6109, + "theoretical_loss": 4.507325500449089, + "tokens_seen": 152895488 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004816850551654965, + "loss": 1.7233, + "theoretical_loss": 4.507073722105167, + "tokens_seen": 152961024 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004816750250752257, + "loss": 1.7927, + "theoretical_loss": 4.506822081802351, + "tokens_seen": 153026560 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048166499498495486, + "loss": 1.627, + "theoretical_loss": 4.506570579405888, + "tokens_seen": 153092096 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004816549648946841, + "loss": 1.6175, + "theoretical_loss": 4.506319214781215, + "tokens_seen": 153157632 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004816449348044132, + "loss": 1.6949, + "theoretical_loss": 4.5060679877939585, + "tokens_seen": 153223168 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048163490471414246, + "loss": 1.6152, + "theoretical_loss": 4.505816898309931, + "tokens_seen": 153288704 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004816248746238716, + "loss": 1.7322, + "theoretical_loss": 4.505565946195134, + "tokens_seen": 153354240 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004816148445336008, + "loss": 1.7736, + "theoretical_loss": 4.505315131315757, + "tokens_seen": 153419776 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048160481444333, + "loss": 1.8123, + "theoretical_loss": 4.5050644535381785, + "tokens_seen": 153485312 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004815947843530592, + "loss": 1.7229, + "theoretical_loss": 4.504813912728961, + "tokens_seen": 153550848 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048158475426278837, + "loss": 1.6151, + "theoretical_loss": 4.504563508754857, + "tokens_seen": 153616384 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004815747241725176, + "loss": 1.6856, + "theoretical_loss": 4.504313241482803, + "tokens_seen": 153681920 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048156469408224673, + "loss": 1.6848, + "theoretical_loss": 4.5040631107799225, + "tokens_seen": 153747456 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048155466399197596, + "loss": 1.7605, + "theoretical_loss": 4.503813116513525, + "tokens_seen": 153812992 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004815446339017051, + "loss": 1.5602, + "theoretical_loss": 4.503563258551105, + "tokens_seen": 153878528 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004815346038114343, + "loss": 1.6309, + "theoretical_loss": 4.503313536760341, + "tokens_seen": 153944064 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0019255278166383505, + "objective/train/docs_used": 98099, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4421207904815674, + "objective/train/original_loss": 3.4421207904815674, + "objective/train/theoretical_loss": 4.503063951009098, + "objective/train/tokens_used": 174469600, + "objective/train/value_avg": -0.004314422607421875, + "objective/train/value_loss": 9.987063822336495e-05, + "objective/train/value_max": -0.00015473365783691406, + "objective/train/value_min": -0.1473388671875, + "objective/train/value_reward_corr": 0.160983263022596, + "objective/train/value_std": 0.00537109375, + "objective/train/weight_avg": 0.9854989051818848, + "objective/train/weighted_lm_loss": 3.384446382522583, + "objective/train/weights_max": 3.667006731033325, + "objective/train/weights_min": 0.2407218962907791, + "theoretical_loss": 4.503063951009098, + "tokens_seen": 154009600 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004815245737211635, + "loss": 1.7304, + "theoretical_loss": 4.503063951009098, + "tokens_seen": 154009600 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004815145436308927, + "loss": 1.6864, + "theoretical_loss": 4.502814501165425, + "tokens_seen": 154075136 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048150451354062187, + "loss": 1.726, + "theoretical_loss": 4.502565187097554, + "tokens_seen": 154140672 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048149448345035105, + "loss": 1.6469, + "theoretical_loss": 4.502316008673902, + "tokens_seen": 154206208 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048148445336008023, + "loss": 1.5044, + "theoretical_loss": 4.502066965763069, + "tokens_seen": 154271744 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048147442326980947, + "loss": 1.6555, + "theoretical_loss": 4.5018180582338365, + "tokens_seen": 154337280 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004814643931795386, + "loss": 1.6986, + "theoretical_loss": 4.501569285955171, + "tokens_seen": 154402816 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048145436308926783, + "loss": 1.5253, + "theoretical_loss": 4.50132064879622, + "tokens_seen": 154468352 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048144433299899696, + "loss": 1.7276, + "theoretical_loss": 4.5010721466263135, + "tokens_seen": 154533888 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004814343029087262, + "loss": 1.763, + "theoretical_loss": 4.500823779314965, + "tokens_seen": 154599424 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048142427281845537, + "loss": 1.6669, + "theoretical_loss": 4.500575546731865, + "tokens_seen": 154664960 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048141424272818455, + "loss": 1.7008, + "theoretical_loss": 4.500327448746887, + "tokens_seen": 154730496 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048140421263791373, + "loss": 1.7682, + "theoretical_loss": 4.5000794852300885, + "tokens_seen": 154796032 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048139418254764297, + "loss": 1.6709, + "theoretical_loss": 4.499831656051704, + "tokens_seen": 154861568 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004813841524573721, + "loss": 1.6577, + "theoretical_loss": 4.499583961082147, + "tokens_seen": 154927104 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048137412236710133, + "loss": 1.8027, + "theoretical_loss": 4.499336400192014, + "tokens_seen": 154992640 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048136409227683046, + "loss": 1.6626, + "theoretical_loss": 4.49908897325208, + "tokens_seen": 155058176 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004813540621865597, + "loss": 1.5921, + "theoretical_loss": 4.498841680133298, + "tokens_seen": 155123712 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004813440320962889, + "loss": 1.6788, + "theoretical_loss": 4.498594520706801, + "tokens_seen": 155189248 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048133400200601806, + "loss": 1.6344, + "theoretical_loss": 4.498347494843897, + "tokens_seen": 155254784 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048132397191574724, + "loss": 1.66, + "theoretical_loss": 4.498100602416079, + "tokens_seen": 155320320 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004813139418254764, + "loss": 1.7881, + "theoretical_loss": 4.49785384329501, + "tokens_seen": 155385856 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004813039117352056, + "loss": 1.7896, + "theoretical_loss": 4.497607217352536, + "tokens_seen": 155451392 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048129388164493483, + "loss": 1.6187, + "theoretical_loss": 4.497360724460679, + "tokens_seen": 155516928 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048128385155466396, + "loss": 1.6324, + "theoretical_loss": 4.497114364491635, + "tokens_seen": 155582464 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.002433479530736804, + "objective/train/docs_used": 99496, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4155564308166504, + "objective/train/original_loss": 3.4155561923980713, + "objective/train/theoretical_loss": 4.496868137317781, + "objective/train/tokens_used": 176108000, + "objective/train/value_avg": -0.004024505615234375, + "objective/train/value_loss": 9.586540545569733e-05, + "objective/train/value_max": -0.00020182132720947266, + "objective/train/value_min": -0.2401123046875, + "objective/train/value_reward_corr": 0.20298125794907393, + "objective/train/value_std": 0.004222869873046875, + "objective/train/weight_avg": 0.9803794026374817, + "objective/train/weighted_lm_loss": 3.3570170402526855, + "objective/train/weights_max": 10.57536506652832, + "objective/train/weights_min": 0.32507771253585815, + "theoretical_loss": 4.496868137317781, + "tokens_seen": 155648000 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004812738214643932, + "loss": 1.5096, + "theoretical_loss": 4.496868137317781, + "tokens_seen": 155648000 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004812637913741223, + "loss": 1.6983, + "theoretical_loss": 4.496622042811667, + "tokens_seen": 155713536 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048125376128385156, + "loss": 1.6448, + "theoretical_loss": 4.496376080846021, + "tokens_seen": 155779072 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048124373119358074, + "loss": 1.7165, + "theoretical_loss": 4.496130251293744, + "tokens_seen": 155844608 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004812337011033099, + "loss": 1.713, + "theoretical_loss": 4.4958845540279135, + "tokens_seen": 155910144 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048122367101303916, + "loss": 1.6017, + "theoretical_loss": 4.495638988921783, + "tokens_seen": 155975680 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048121364092276834, + "loss": 1.7017, + "theoretical_loss": 4.495393555848779, + "tokens_seen": 156041216 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004812036108324975, + "loss": 1.5954, + "theoretical_loss": 4.495148254682505, + "tokens_seen": 156106752 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004811935807422267, + "loss": 1.7987, + "theoretical_loss": 4.494903085296733, + "tokens_seen": 156172288 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004811835506519559, + "loss": 1.7109, + "theoretical_loss": 4.494658047565416, + "tokens_seen": 156237824 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048117352056168506, + "loss": 1.784, + "theoretical_loss": 4.494413141362674, + "tokens_seen": 156303360 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004811634904714143, + "loss": 1.674, + "theoretical_loss": 4.494168366562805, + "tokens_seen": 156368896 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004811534603811434, + "loss": 1.717, + "theoretical_loss": 4.493923723040275, + "tokens_seen": 156434432 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048114343029087266, + "loss": 1.5562, + "theoretical_loss": 4.493679210669727, + "tokens_seen": 156499968 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004811334002006018, + "loss": 1.6001, + "theoretical_loss": 4.493434829325974, + "tokens_seen": 156565504 + }, + { + "epoch": 0.05, + "learning_rate": 0.000481123370110331, + "loss": 1.5882, + "theoretical_loss": 4.493190578883999, + "tokens_seen": 156631040 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004811133400200602, + "loss": 1.4818, + "theoretical_loss": 4.49294645921896, + "tokens_seen": 156696576 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004811033099297894, + "loss": 1.6971, + "theoretical_loss": 4.492702470206184, + "tokens_seen": 156762112 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048109327983951857, + "loss": 1.6773, + "theoretical_loss": 4.49245861172117, + "tokens_seen": 156827648 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004810832497492478, + "loss": 1.6191, + "theoretical_loss": 4.492214883639586, + "tokens_seen": 156893184 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048107321965897693, + "loss": 1.6522, + "theoretical_loss": 4.491971285837273, + "tokens_seen": 156958720 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048106318956870616, + "loss": 1.6673, + "theoretical_loss": 4.49172781819024, + "tokens_seen": 157024256 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004810531594784353, + "loss": 1.6602, + "theoretical_loss": 4.491484480574666, + "tokens_seen": 157089792 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004810431293881645, + "loss": 1.7665, + "theoretical_loss": 4.4912412728668984, + "tokens_seen": 157155328 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004810330992978937, + "loss": 1.658, + "theoretical_loss": 4.490998194943456, + "tokens_seen": 157220864 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.001290497020818293, + "objective/train/docs_used": 100000, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.569195508956909, + "objective/train/original_loss": 3.56919527053833, + "objective/train/theoretical_loss": 4.490755246681026, + "objective/train/tokens_used": 177746400, + "objective/train/value_avg": -0.004344940185546875, + "objective/train/value_loss": 5.11783073307015e-05, + "objective/train/value_max": -0.00021314620971679688, + "objective/train/value_min": -0.058013916015625, + "objective/train/value_reward_corr": 0.07540446659477323, + "objective/train/value_std": 0.0033321380615234375, + "objective/train/weight_avg": 0.989546000957489, + "objective/train/weighted_lm_loss": 3.5577542781829834, + "objective/train/weights_max": 1.6375195980072021, + "objective/train/weights_min": 0.37344327569007874, + "theoretical_loss": 4.490755246681026, + "tokens_seen": 157286400 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004810230692076229, + "loss": 1.6076, + "theoretical_loss": 4.490755246681026, + "tokens_seen": 157286400 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048101303911735207, + "loss": 1.7108, + "theoretical_loss": 4.490512427956462, + "tokens_seen": 157351936 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048100300902708125, + "loss": 1.7462, + "theoretical_loss": 4.490269738646786, + "tokens_seen": 157417472 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048099297893681043, + "loss": 1.6953, + "theoretical_loss": 4.490027178629192, + "tokens_seen": 157483008 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048098294884653967, + "loss": 1.6925, + "theoretical_loss": 4.4897847477810355, + "tokens_seen": 157548544 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004809729187562688, + "loss": 1.6427, + "theoretical_loss": 4.489542445979844, + "tokens_seen": 157614080 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048096288866599803, + "loss": 1.6996, + "theoretical_loss": 4.489300273103311, + "tokens_seen": 157679616 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048095285857572716, + "loss": 1.7048, + "theoretical_loss": 4.489058229029293, + "tokens_seen": 157745152 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004809428284854564, + "loss": 1.5989, + "theoretical_loss": 4.4888163136358195, + "tokens_seen": 157810688 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048093279839518557, + "loss": 1.7347, + "theoretical_loss": 4.48857452680108, + "tokens_seen": 157876224 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048092276830491475, + "loss": 1.6197, + "theoretical_loss": 4.4883328684034325, + "tokens_seen": 157941760 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048091273821464393, + "loss": 1.7363, + "theoretical_loss": 4.488091338321402, + "tokens_seen": 158007296 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048090270812437317, + "loss": 1.5928, + "theoretical_loss": 4.487849936433676, + "tokens_seen": 158072832 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004808926780341023, + "loss": 1.5929, + "theoretical_loss": 4.487608662619107, + "tokens_seen": 158138368 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048088264794383153, + "loss": 1.7256, + "theoretical_loss": 4.487367516756715, + "tokens_seen": 158203904 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048087261785356066, + "loss": 1.7422, + "theoretical_loss": 4.487126498725679, + "tokens_seen": 158269440 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004808625877632899, + "loss": 1.7379, + "theoretical_loss": 4.48688560840535, + "tokens_seen": 158334976 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004808525576730191, + "loss": 1.6124, + "theoretical_loss": 4.486644845675235, + "tokens_seen": 158400512 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048084252758274826, + "loss": 1.5926, + "theoretical_loss": 4.48640421041501, + "tokens_seen": 158466048 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048083249749247744, + "loss": 1.6652, + "theoretical_loss": 4.486163702504509, + "tokens_seen": 158531584 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004808224674022066, + "loss": 1.5928, + "theoretical_loss": 4.4859233218237335, + "tokens_seen": 158597120 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004808124373119358, + "loss": 1.6013, + "theoretical_loss": 4.485683068252847, + "tokens_seen": 158662656 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048080240722166503, + "loss": 1.7415, + "theoretical_loss": 4.485442941672172, + "tokens_seen": 158728192 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048079237713139416, + "loss": 1.7016, + "theoretical_loss": 4.485202941962196, + "tokens_seen": 158793728 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004807823470411234, + "loss": 1.6023, + "theoretical_loss": 4.484963069003568, + "tokens_seen": 158859264 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.00018905980687122792, + "objective/train/docs_used": 101476, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.357400894165039, + "objective/train/original_loss": 3.35740065574646, + "objective/train/theoretical_loss": 4.484723322677097, + "objective/train/tokens_used": 179384800, + "objective/train/value_avg": -0.0045166015625, + "objective/train/value_loss": 5.644482735078782e-05, + "objective/train/value_max": -0.0002491474151611328, + "objective/train/value_min": -0.08758544921875, + "objective/train/value_reward_corr": 0.09637617783220483, + "objective/train/value_std": 0.00423431396484375, + "objective/train/weight_avg": 1.000861644744873, + "objective/train/weighted_lm_loss": 3.3825533390045166, + "objective/train/weights_max": 2.400926113128662, + "objective/train/weights_min": 0.45388129353523254, + "theoretical_loss": 4.484723322677097, + "tokens_seen": 158924800 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004807723169508525, + "loss": 1.6609, + "theoretical_loss": 4.484723322677097, + "tokens_seen": 158924800 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048076228686058176, + "loss": 1.7435, + "theoretical_loss": 4.484483702863755, + "tokens_seen": 158990336 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048075225677031094, + "loss": 1.7357, + "theoretical_loss": 4.4842442094446735, + "tokens_seen": 159055872 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004807422266800401, + "loss": 1.6311, + "theoretical_loss": 4.484004842301145, + "tokens_seen": 159121408 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004807321965897693, + "loss": 1.6252, + "theoretical_loss": 4.483765601314621, + "tokens_seen": 159186944 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048072216649949854, + "loss": 1.7421, + "theoretical_loss": 4.483526486366717, + "tokens_seen": 159252480 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048071213640922766, + "loss": 1.6207, + "theoretical_loss": 4.483287497339205, + "tokens_seen": 159318016 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004807021063189569, + "loss": 1.6632, + "theoretical_loss": 4.483048634114016, + "tokens_seen": 159383552 + }, + { + "epoch": 0.05, + "learning_rate": 0.000480692076228686, + "loss": 1.5788, + "theoretical_loss": 4.482809896573242, + "tokens_seen": 159449088 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048068204613841526, + "loss": 1.6004, + "theoretical_loss": 4.482571284599134, + "tokens_seen": 159514624 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048067201604814444, + "loss": 1.7213, + "theoretical_loss": 4.482332798074099, + "tokens_seen": 159580160 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004806619859578736, + "loss": 1.7352, + "theoretical_loss": 4.482094436880706, + "tokens_seen": 159645696 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004806519558676028, + "loss": 1.7674, + "theoretical_loss": 4.481856200901679, + "tokens_seen": 159711232 + }, + { + "epoch": 0.05, + "learning_rate": 0.000480641925777332, + "loss": 1.8641, + "theoretical_loss": 4.481618090019903, + "tokens_seen": 159776768 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048063189568706117, + "loss": 1.7148, + "theoretical_loss": 4.481380104118417, + "tokens_seen": 159842304 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004806218655967904, + "loss": 1.7128, + "theoretical_loss": 4.481142243080419, + "tokens_seen": 159907840 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048061183550651953, + "loss": 1.5394, + "theoretical_loss": 4.480904506789265, + "tokens_seen": 159973376 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048060180541624877, + "loss": 1.665, + "theoretical_loss": 4.480666895128465, + "tokens_seen": 160038912 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048059177532597795, + "loss": 1.5563, + "theoretical_loss": 4.480429407981688, + "tokens_seen": 160104448 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048058174523570713, + "loss": 1.8855, + "theoretical_loss": 4.4801920452327595, + "tokens_seen": 160169984 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004805717151454363, + "loss": 1.6866, + "theoretical_loss": 4.479954806765657, + "tokens_seen": 160235520 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004805616850551655, + "loss": 1.5241, + "theoretical_loss": 4.479717692464517, + "tokens_seen": 160301056 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048055165496489467, + "loss": 1.6626, + "theoretical_loss": 4.479480702213631, + "tokens_seen": 160366592 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004805416248746239, + "loss": 1.6217, + "theoretical_loss": 4.479243835897444, + "tokens_seen": 160432128 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048053159478435303, + "loss": 1.6335, + "theoretical_loss": 4.479007093400558, + "tokens_seen": 160497664 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0009749096352607012, + "objective/train/docs_used": 102054, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3278448581695557, + "objective/train/original_loss": 3.3278448581695557, + "objective/train/theoretical_loss": 4.478770474607726, + "objective/train/tokens_used": 181023200, + "objective/train/value_avg": -0.00411224365234375, + "objective/train/value_loss": 7.985084084793925e-05, + "objective/train/value_max": -0.00017261505126953125, + "objective/train/value_min": -0.254638671875, + "objective/train/value_reward_corr": 0.1440103364824816, + "objective/train/value_std": 0.0046234130859375, + "objective/train/weight_avg": 0.9943360686302185, + "objective/train/weighted_lm_loss": 3.303476572036743, + "objective/train/weights_max": 11.534893989562988, + "objective/train/weights_min": 0.3339805603027344, + "theoretical_loss": 4.478770474607726, + "tokens_seen": 160563200 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048052156469408227, + "loss": 1.6173, + "theoretical_loss": 4.478770474607726, + "tokens_seen": 160563200 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004805115346038114, + "loss": 1.6912, + "theoretical_loss": 4.478533979403862, + "tokens_seen": 160628736 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048050150451354063, + "loss": 1.6583, + "theoretical_loss": 4.478297607674024, + "tokens_seen": 160694272 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004804914744232698, + "loss": 1.6402, + "theoretical_loss": 4.4780613593034335, + "tokens_seen": 160759808 + }, + { + "epoch": 0.05, + "learning_rate": 0.000480481444332999, + "loss": 1.6079, + "theoretical_loss": 4.47782523417746, + "tokens_seen": 160825344 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048047141424272823, + "loss": 1.6015, + "theoretical_loss": 4.477589232181627, + "tokens_seen": 160890880 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048046138415245736, + "loss": 1.5884, + "theoretical_loss": 4.47735335320161, + "tokens_seen": 160956416 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004804513540621866, + "loss": 1.7023, + "theoretical_loss": 4.47711759712324, + "tokens_seen": 161021952 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048044132397191577, + "loss": 1.6, + "theoretical_loss": 4.476881963832499, + "tokens_seen": 161087488 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048043129388164495, + "loss": 1.6114, + "theoretical_loss": 4.476646453215521, + "tokens_seen": 161153024 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048042126379137413, + "loss": 1.5379, + "theoretical_loss": 4.476411065158589, + "tokens_seen": 161218560 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048041123370110337, + "loss": 1.7744, + "theoretical_loss": 4.476175799548144, + "tokens_seen": 161284096 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004804012036108325, + "loss": 1.6591, + "theoretical_loss": 4.475940656270772, + "tokens_seen": 161349632 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048039117352056173, + "loss": 1.6682, + "theoretical_loss": 4.475705635213215, + "tokens_seen": 161415168 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048038114343029086, + "loss": 1.6627, + "theoretical_loss": 4.475470736262361, + "tokens_seen": 161480704 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004803711133400201, + "loss": 1.6234, + "theoretical_loss": 4.475235959305252, + "tokens_seen": 161546240 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004803610832497493, + "loss": 1.5481, + "theoretical_loss": 4.47500130422908, + "tokens_seen": 161611776 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048035105315947846, + "loss": 1.6239, + "theoretical_loss": 4.4747667709211845, + "tokens_seen": 161677312 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048034102306920764, + "loss": 1.7391, + "theoretical_loss": 4.474532359269059, + "tokens_seen": 161742848 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004803309929789368, + "loss": 1.4832, + "theoretical_loss": 4.474298069160341, + "tokens_seen": 161808384 + }, + { + "epoch": 0.05, + "learning_rate": 0.000480320962888666, + "loss": 1.7045, + "theoretical_loss": 4.4740639004828235, + "tokens_seen": 161873920 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048031093279839523, + "loss": 1.524, + "theoretical_loss": 4.4738298531244425, + "tokens_seen": 161939456 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048030090270812436, + "loss": 1.7691, + "theoretical_loss": 4.473595926973286, + "tokens_seen": 162004992 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004802908726178536, + "loss": 1.6501, + "theoretical_loss": 4.473362121917591, + "tokens_seen": 162070528 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004802808425275827, + "loss": 1.6439, + "theoretical_loss": 4.473128437845741, + "tokens_seen": 162136064 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0017178815323859453, + "objective/train/docs_used": 102718, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.66158127784729, + "objective/train/original_loss": 3.66158127784729, + "objective/train/theoretical_loss": 4.4728948746462684, + "objective/train/tokens_used": 182661600, + "objective/train/value_avg": -0.00452423095703125, + "objective/train/value_loss": 8.700842590769753e-05, + "objective/train/value_max": -0.00022172927856445312, + "objective/train/value_min": -0.157470703125, + "objective/train/value_reward_corr": 0.10817499818491717, + "objective/train/value_std": 0.004123687744140625, + "objective/train/weight_avg": 0.9869884848594666, + "objective/train/weighted_lm_loss": 3.6356987953186035, + "objective/train/weights_max": 4.829326629638672, + "objective/train/weights_min": 0.4415709674358368, + "theoretical_loss": 4.4728948746462684, + "tokens_seen": 162201600 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048027081243731196, + "loss": 1.7303, + "theoretical_loss": 4.4728948746462684, + "tokens_seen": 162201600 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048026078234704114, + "loss": 1.5589, + "theoretical_loss": 4.47266143220785, + "tokens_seen": 162267136 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004802507522567703, + "loss": 1.5402, + "theoretical_loss": 4.472428110419317, + "tokens_seen": 162332672 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004802407221664995, + "loss": 1.5173, + "theoretical_loss": 4.472194909169643, + "tokens_seen": 162398208 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048023069207622874, + "loss": 1.7247, + "theoretical_loss": 4.471961828347947, + "tokens_seen": 162463744 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048022066198595786, + "loss": 1.6369, + "theoretical_loss": 4.471728867843497, + "tokens_seen": 162529280 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004802106318956871, + "loss": 1.6011, + "theoretical_loss": 4.47149602754571, + "tokens_seen": 162594816 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048020060180541623, + "loss": 1.7751, + "theoretical_loss": 4.4712633073441435, + "tokens_seen": 162660352 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048019057171514546, + "loss": 1.5578, + "theoretical_loss": 4.471030707128506, + "tokens_seen": 162725888 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048018054162487464, + "loss": 1.5825, + "theoretical_loss": 4.470798226788649, + "tokens_seen": 162791424 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004801705115346038, + "loss": 1.6864, + "theoretical_loss": 4.470565866214568, + "tokens_seen": 162856960 + }, + { + "epoch": 0.05, + "learning_rate": 0.000480160481444333, + "loss": 1.574, + "theoretical_loss": 4.4703336252964085, + "tokens_seen": 162922496 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004801504513540622, + "loss": 1.6385, + "theoretical_loss": 4.470101503924456, + "tokens_seen": 162988032 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048014042126379137, + "loss": 1.7045, + "theoretical_loss": 4.469869501989144, + "tokens_seen": 163053568 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004801303911735206, + "loss": 1.6167, + "theoretical_loss": 4.469637619381047, + "tokens_seen": 163119104 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048012036108324973, + "loss": 1.6928, + "theoretical_loss": 4.469405855990889, + "tokens_seen": 163184640 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048011033099297897, + "loss": 1.6443, + "theoretical_loss": 4.469174211709531, + "tokens_seen": 163250176 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048010030090270815, + "loss": 1.5931, + "theoretical_loss": 4.468942686427985, + "tokens_seen": 163315712 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048009027081243733, + "loss": 1.6165, + "theoretical_loss": 4.4687112800373985, + "tokens_seen": 163381248 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004800802407221665, + "loss": 1.7354, + "theoretical_loss": 4.46847999242907, + "tokens_seen": 163446784 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004800702106318957, + "loss": 1.5379, + "theoretical_loss": 4.468248823494436, + "tokens_seen": 163512320 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048006018054162487, + "loss": 1.667, + "theoretical_loss": 4.4680177731250765, + "tokens_seen": 163577856 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004800501504513541, + "loss": 1.6669, + "theoretical_loss": 4.467786841212715, + "tokens_seen": 163643392 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048004012036108323, + "loss": 1.6028, + "theoretical_loss": 4.467556027649216, + "tokens_seen": 163708928 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048003009027081247, + "loss": 1.5808, + "theoretical_loss": 4.467325332326588, + "tokens_seen": 163774464 + }, + { + "debugging/Self-BLEU-5": 0.4072555680918346, + "debugging/distinct-1-grams": 0.8137531241311615, + "debugging/distinct-2-grams": 0.9866498101723051, + "debugging/entropy-1-grams": 5.4466195562752855, + "debugging/entropy-2-grams": 6.104764435286928, + "debugging/length": 544.5, + "debugging/num_segments": 6, + "debugging/raw_token_scores_avg": 0.003607792081311345, + "debugging/raw_token_scores_std": 0.005498606711626053, + "debugging/score": 0.002729943707102512, + "debugging/score_std": 0.0025596098515178914, + "epoch": 0.05, + "objective/train/advantage_avg": 0.00029724364867433906, + "objective/train/docs_used": 104069, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1673457622528076, + "objective/train/original_loss": 3.1673457622528076, + "objective/train/theoretical_loss": 4.467094755136979, + "objective/train/tokens_used": 184300000, + "objective/train/value_avg": -0.00390625, + "objective/train/value_loss": 3.8953046896494925e-05, + "objective/train/value_max": -0.00015354156494140625, + "objective/train/value_min": -0.18603515625, + "objective/train/value_reward_corr": 0.13354600789152604, + "objective/train/value_std": 0.00376129150390625, + "objective/train/weight_avg": 1.0050071477890015, + "objective/train/weighted_lm_loss": 3.1790030002593994, + "objective/train/weights_max": 6.060983180999756, + "objective/train/weights_min": 0.6616734862327576, + "theoretical_loss": 4.467094755136979, + "tokens_seen": 163840000 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004800200601805416, + "loss": 1.6985, + "theoretical_loss": 4.467094755136979, + "tokens_seen": 163840000 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048001003009027083, + "loss": 1.632, + "theoretical_loss": 4.4668642959726785, + "tokens_seen": 163905536 + }, + { + "epoch": 0.05, + "learning_rate": 0.00048, + "loss": 1.7489, + "theoretical_loss": 4.466633954726119, + "tokens_seen": 163971072 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004799899699097292, + "loss": 1.5603, + "theoretical_loss": 4.466403731289873, + "tokens_seen": 164036608 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004799799398194584, + "loss": 1.5782, + "theoretical_loss": 4.466173625556653, + "tokens_seen": 164102144 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047996990972918756, + "loss": 1.6898, + "theoretical_loss": 4.465943637419313, + "tokens_seen": 164167680 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047995987963891674, + "loss": 1.5189, + "theoretical_loss": 4.465713766770847, + "tokens_seen": 164233216 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047994984954864597, + "loss": 1.6582, + "theoretical_loss": 4.465484013504389, + "tokens_seen": 164298752 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004799398194583751, + "loss": 1.5483, + "theoretical_loss": 4.4652543775132125, + "tokens_seen": 164364288 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047992978936810433, + "loss": 1.6081, + "theoretical_loss": 4.465024858690731, + "tokens_seen": 164429824 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004799197592778335, + "loss": 1.7114, + "theoretical_loss": 4.4647954569304975, + "tokens_seen": 164495360 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004799097291875627, + "loss": 1.5663, + "theoretical_loss": 4.464566172126203, + "tokens_seen": 164560896 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004798996990972919, + "loss": 1.635, + "theoretical_loss": 4.464337004171679, + "tokens_seen": 164626432 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047988966900702106, + "loss": 1.5507, + "theoretical_loss": 4.464107952960893, + "tokens_seen": 164691968 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047987963891675024, + "loss": 1.6147, + "theoretical_loss": 4.463879018387954, + "tokens_seen": 164757504 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004798696088264795, + "loss": 1.6006, + "theoretical_loss": 4.463650200347106, + "tokens_seen": 164823040 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004798595787362086, + "loss": 1.5757, + "theoretical_loss": 4.4634214987327345, + "tokens_seen": 164888576 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047984954864593784, + "loss": 1.6092, + "theoretical_loss": 4.463192913439359, + "tokens_seen": 164954112 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047983951855566696, + "loss": 1.7585, + "theoretical_loss": 4.4629644443616385, + "tokens_seen": 165019648 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004798294884653962, + "loss": 1.6907, + "theoretical_loss": 4.46273609139437, + "tokens_seen": 165085184 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004798194583751254, + "loss": 1.5841, + "theoretical_loss": 4.462507854432483, + "tokens_seen": 165150720 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047980942828485456, + "loss": 1.6847, + "theoretical_loss": 4.46227973337105, + "tokens_seen": 165216256 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047979939819458374, + "loss": 1.7353, + "theoretical_loss": 4.462051728105276, + "tokens_seen": 165281792 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004797893681043129, + "loss": 1.7261, + "theoretical_loss": 4.461823838530503, + "tokens_seen": 165347328 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004797793380140421, + "loss": 1.7238, + "theoretical_loss": 4.461596064542208, + "tokens_seen": 165412864 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0014207016211003065, + "objective/train/docs_used": 105300, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0605790615081787, + "objective/train/original_loss": 3.0605788230895996, + "objective/train/theoretical_loss": 4.461368406036007, + "objective/train/tokens_used": 185938400, + "objective/train/value_avg": -0.00551605224609375, + "objective/train/value_loss": 8.823371899779886e-05, + "objective/train/value_max": -0.00013875961303710938, + "objective/train/value_min": -0.07977294921875, + "objective/train/value_reward_corr": 0.16617162429130944, + "objective/train/value_std": 0.00469970703125, + "objective/train/weight_avg": 0.9899736642837524, + "objective/train/weighted_lm_loss": 3.039191961288452, + "objective/train/weights_max": 1.9433166980743408, + "objective/train/weights_min": 0.24339155852794647, + "theoretical_loss": 4.461368406036007, + "tokens_seen": 165478400 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047976930792377134, + "loss": 1.5506, + "theoretical_loss": 4.461368406036007, + "tokens_seen": 165478400 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047975927783350047, + "loss": 1.6718, + "theoretical_loss": 4.461140862907648, + "tokens_seen": 165543936 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004797492477432297, + "loss": 1.6989, + "theoretical_loss": 4.4609134350530155, + "tokens_seen": 165609472 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004797392176529589, + "loss": 1.8152, + "theoretical_loss": 4.460686122368132, + "tokens_seen": 165675008 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047972918756268807, + "loss": 1.6462, + "theoretical_loss": 4.460458924749149, + "tokens_seen": 165740544 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004797191574724173, + "loss": 1.6661, + "theoretical_loss": 4.460231842092359, + "tokens_seen": 165806080 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047970912738214643, + "loss": 1.6257, + "theoretical_loss": 4.460004874294185, + "tokens_seen": 165871616 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047969909729187566, + "loss": 1.6691, + "theoretical_loss": 4.459778021251183, + "tokens_seen": 165937152 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047968906720160484, + "loss": 1.6683, + "theoretical_loss": 4.459551282860048, + "tokens_seen": 166002688 + }, + { + "epoch": 0.05, + "learning_rate": 0.000479679037111334, + "loss": 1.6697, + "theoretical_loss": 4.459324659017604, + "tokens_seen": 166068224 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004796690070210632, + "loss": 1.6794, + "theoretical_loss": 4.459098149620811, + "tokens_seen": 166133760 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004796589769307924, + "loss": 1.6539, + "theoretical_loss": 4.458871754566761, + "tokens_seen": 166199296 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047964894684052157, + "loss": 1.5601, + "theoretical_loss": 4.458645473752681, + "tokens_seen": 166264832 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004796389167502508, + "loss": 1.6776, + "theoretical_loss": 4.458419307075928, + "tokens_seen": 166330368 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047962888665997993, + "loss": 1.8211, + "theoretical_loss": 4.458193254433994, + "tokens_seen": 166395904 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047961885656970917, + "loss": 1.8058, + "theoretical_loss": 4.4579673157245026, + "tokens_seen": 166461440 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047960882647943835, + "loss": 1.5766, + "theoretical_loss": 4.457741490845209, + "tokens_seen": 166526976 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047959879638916753, + "loss": 1.6941, + "theoretical_loss": 4.457515779694001, + "tokens_seen": 166592512 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004795887662988967, + "loss": 1.5119, + "theoretical_loss": 4.457290182168899, + "tokens_seen": 166658048 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004795787362086259, + "loss": 1.612, + "theoretical_loss": 4.457064698168051, + "tokens_seen": 166723584 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047956870611835507, + "loss": 1.5822, + "theoretical_loss": 4.456839327589742, + "tokens_seen": 166789120 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004795586760280843, + "loss": 1.7486, + "theoretical_loss": 4.456614070332385, + "tokens_seen": 166854656 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047954864593781343, + "loss": 1.5105, + "theoretical_loss": 4.456388926294522, + "tokens_seen": 166920192 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047953861584754267, + "loss": 1.5817, + "theoretical_loss": 4.45616389537483, + "tokens_seen": 166985728 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004795285857572718, + "loss": 1.7405, + "theoretical_loss": 4.455938977472112, + "tokens_seen": 167051264 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0012980670435354114, + "objective/train/docs_used": 106148, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.7933056354522705, + "objective/train/original_loss": 3.7933053970336914, + "objective/train/theoretical_loss": 4.455714172485305, + "objective/train/tokens_used": 187576800, + "objective/train/value_avg": -0.004886627197265625, + "objective/train/value_loss": 0.0001022761789499782, + "objective/train/value_max": -0.0002148151397705078, + "objective/train/value_min": -0.48974609375, + "objective/train/value_reward_corr": 0.10441467828619537, + "objective/train/value_std": 0.006366729736328125, + "objective/train/weight_avg": 0.9987539052963257, + "objective/train/weighted_lm_loss": 3.7621920108795166, + "objective/train/weights_max": 124.1967544555664, + "objective/train/weights_min": 0.3701811134815216, + "theoretical_loss": 4.455714172485305, + "tokens_seen": 167116800 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047951855566700103, + "loss": 1.7375, + "theoretical_loss": 4.455714172485305, + "tokens_seen": 167116800 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004795085255767302, + "loss": 1.6996, + "theoretical_loss": 4.4554894803134735, + "tokens_seen": 167182336 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004794984954864594, + "loss": 1.7223, + "theoretical_loss": 4.455264900855813, + "tokens_seen": 167247872 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004794884653961886, + "loss": 1.7344, + "theoretical_loss": 4.4550404340116465, + "tokens_seen": 167313408 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047947843530591776, + "loss": 1.6813, + "theoretical_loss": 4.454816079680429, + "tokens_seen": 167378944 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047946840521564694, + "loss": 1.6093, + "theoretical_loss": 4.454591837761743, + "tokens_seen": 167444480 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047945837512537617, + "loss": 1.62, + "theoretical_loss": 4.454367708155301, + "tokens_seen": 167510016 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004794483450351053, + "loss": 1.6868, + "theoretical_loss": 4.454143690760943, + "tokens_seen": 167575552 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047943831494483453, + "loss": 1.5641, + "theoretical_loss": 4.453919785478638, + "tokens_seen": 167641088 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004794282848545637, + "loss": 1.5394, + "theoretical_loss": 4.453695992208482, + "tokens_seen": 167706624 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004794182547642929, + "loss": 1.7029, + "theoretical_loss": 4.453472310850701, + "tokens_seen": 167772160 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004794082246740221, + "loss": 1.6511, + "theoretical_loss": 4.453248741305649, + "tokens_seen": 167837696 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047939819458375126, + "loss": 1.75, + "theoretical_loss": 4.4530252834738056, + "tokens_seen": 167903232 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047938816449348044, + "loss": 1.6236, + "theoretical_loss": 4.452801937255778, + "tokens_seen": 167968768 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004793781344032097, + "loss": 1.6682, + "theoretical_loss": 4.4525787025523025, + "tokens_seen": 168034304 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004793681043129388, + "loss": 1.651, + "theoretical_loss": 4.452355579264241, + "tokens_seen": 168099840 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047935807422266804, + "loss": 1.7259, + "theoretical_loss": 4.452132567292581, + "tokens_seen": 168165376 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047934804413239716, + "loss": 1.7794, + "theoretical_loss": 4.451909666538439, + "tokens_seen": 168230912 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004793380140421264, + "loss": 1.6764, + "theoretical_loss": 4.4516868769030555, + "tokens_seen": 168296448 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004793279839518556, + "loss": 1.6654, + "theoretical_loss": 4.4514641982878, + "tokens_seen": 168361984 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047931795386158476, + "loss": 1.6614, + "theoretical_loss": 4.451241630594166, + "tokens_seen": 168427520 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047930792377131394, + "loss": 1.5316, + "theoretical_loss": 4.451019173723771, + "tokens_seen": 168493056 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004792978936810431, + "loss": 1.7355, + "theoretical_loss": 4.450796827578361, + "tokens_seen": 168558592 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004792878635907723, + "loss": 1.7494, + "theoretical_loss": 4.450574592059806, + "tokens_seen": 168624128 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047927783350050154, + "loss": 1.73, + "theoretical_loss": 4.450352467070101, + "tokens_seen": 168689664 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0008069667383097112, + "objective/train/docs_used": 106755, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6916706562042236, + "objective/train/original_loss": 2.6916704177856445, + "objective/train/theoretical_loss": 4.450130452511366, + "objective/train/tokens_used": 189215200, + "objective/train/value_avg": -0.00354766845703125, + "objective/train/value_loss": 4.7904479288263246e-05, + "objective/train/value_max": -0.00018525123596191406, + "objective/train/value_min": -0.06866455078125, + "objective/train/value_reward_corr": 0.15676807842595955, + "objective/train/value_std": 0.0029850006103515625, + "objective/train/weight_avg": 0.994168758392334, + "objective/train/weighted_lm_loss": 2.672523260116577, + "objective/train/weights_max": 1.9870387315750122, + "objective/train/weights_min": 0.3766982853412628, + "theoretical_loss": 4.450130452511366, + "tokens_seen": 168755200 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047926780341023067, + "loss": 1.549, + "theoretical_loss": 4.450130452511366, + "tokens_seen": 168755200 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004792577733199599, + "loss": 1.6546, + "theoretical_loss": 4.449908548285846, + "tokens_seen": 168820736 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004792477432296891, + "loss": 1.5505, + "theoretical_loss": 4.449686754295909, + "tokens_seen": 168886272 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047923771313941827, + "loss": 1.6136, + "theoretical_loss": 4.44946507044405, + "tokens_seen": 168951808 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047922768304914745, + "loss": 1.7572, + "theoretical_loss": 4.449243496632885, + "tokens_seen": 169017344 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047921765295887663, + "loss": 1.5205, + "theoretical_loss": 4.449022032765155, + "tokens_seen": 169082880 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004792076228686058, + "loss": 1.7071, + "theoretical_loss": 4.448800678743726, + "tokens_seen": 169148416 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047919759277833504, + "loss": 1.667, + "theoretical_loss": 4.448579434471585, + "tokens_seen": 169213952 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047918756268806417, + "loss": 1.6545, + "theoretical_loss": 4.4483582998518445, + "tokens_seen": 169279488 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004791775325977934, + "loss": 1.6701, + "theoretical_loss": 4.448137274787738, + "tokens_seen": 169345024 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047916750250752253, + "loss": 1.6245, + "theoretical_loss": 4.447916359182623, + "tokens_seen": 169410560 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047915747241725177, + "loss": 1.8459, + "theoretical_loss": 4.4476955529399795, + "tokens_seen": 169476096 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047914744232698095, + "loss": 1.6755, + "theoretical_loss": 4.44747485596341, + "tokens_seen": 169541632 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047913741223671013, + "loss": 1.731, + "theoretical_loss": 4.447254268156637, + "tokens_seen": 169607168 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004791273821464393, + "loss": 1.689, + "theoretical_loss": 4.44703378942351, + "tokens_seen": 169672704 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047911735205616855, + "loss": 1.7373, + "theoretical_loss": 4.446813419667994, + "tokens_seen": 169738240 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004791073219658977, + "loss": 1.5939, + "theoretical_loss": 4.446593158794181, + "tokens_seen": 169803776 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004790972918756269, + "loss": 1.684, + "theoretical_loss": 4.446373006706281, + "tokens_seen": 169869312 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047908726178535604, + "loss": 1.6339, + "theoretical_loss": 4.446152963308625, + "tokens_seen": 169934848 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047907723169508527, + "loss": 1.648, + "theoretical_loss": 4.445933028505671, + "tokens_seen": 170000384 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047906720160481445, + "loss": 1.6602, + "theoretical_loss": 4.445713202201988, + "tokens_seen": 170065920 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047905717151454363, + "loss": 1.6992, + "theoretical_loss": 4.445493484302272, + "tokens_seen": 170131456 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004790471414242728, + "loss": 1.7234, + "theoretical_loss": 4.44527387471134, + "tokens_seen": 170196992 + }, + { + "epoch": 0.05, + "learning_rate": 0.000479037111334002, + "loss": 1.7275, + "theoretical_loss": 4.445054373334125, + "tokens_seen": 170262528 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004790270812437312, + "loss": 1.7646, + "theoretical_loss": 4.444834980075683, + "tokens_seen": 170328064 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0013016839511692524, + "objective/train/docs_used": 107830, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.572859287261963, + "objective/train/original_loss": 3.572859048843384, + "objective/train/theoretical_loss": 4.44461569484119, + "objective/train/tokens_used": 190853600, + "objective/train/value_avg": -0.004058837890625, + "objective/train/value_loss": 7.10627282387577e-05, + "objective/train/value_max": -0.00019562244415283203, + "objective/train/value_min": -0.059417724609375, + "objective/train/value_reward_corr": 0.2472846564980405, + "objective/train/value_std": 0.00365447998046875, + "objective/train/weight_avg": 0.9900668859481812, + "objective/train/weighted_lm_loss": 3.555338144302368, + "objective/train/weights_max": 1.5902118682861328, + "objective/train/weights_min": 0.08621224761009216, + "theoretical_loss": 4.44461569484119, + "tokens_seen": 170393600 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004790170511534604, + "loss": 1.6594, + "theoretical_loss": 4.44461569484119, + "tokens_seen": 170393600 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047900702106318954, + "loss": 1.5488, + "theoretical_loss": 4.444396517535939, + "tokens_seen": 170459136 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004789969909729188, + "loss": 1.666, + "theoretical_loss": 4.444177448065345, + "tokens_seen": 170524672 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004789869608826479, + "loss": 1.7515, + "theoretical_loss": 4.443958486334941, + "tokens_seen": 170590208 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047897693079237714, + "loss": 1.5054, + "theoretical_loss": 4.443739632250379, + "tokens_seen": 170655744 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047896690070210637, + "loss": 1.7181, + "theoretical_loss": 4.443520885717431, + "tokens_seen": 170721280 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004789568706118355, + "loss": 1.6223, + "theoretical_loss": 4.443302246641985, + "tokens_seen": 170786816 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047894684052156473, + "loss": 1.5551, + "theoretical_loss": 4.443083714930051, + "tokens_seen": 170852352 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004789368104312939, + "loss": 1.6267, + "theoretical_loss": 4.442865290487752, + "tokens_seen": 170917888 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004789267803410231, + "loss": 1.6893, + "theoretical_loss": 4.442646973221336, + "tokens_seen": 170983424 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004789167502507523, + "loss": 1.6541, + "theoretical_loss": 4.442428763037165, + "tokens_seen": 171048960 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047890672016048146, + "loss": 1.6249, + "theoretical_loss": 4.442210659841717, + "tokens_seen": 171114496 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047889669007021064, + "loss": 1.5528, + "theoretical_loss": 4.441992663541589, + "tokens_seen": 171180032 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004788866599799399, + "loss": 1.6583, + "theoretical_loss": 4.4417747740434965, + "tokens_seen": 171245568 + }, + { + "epoch": 0.05, + "learning_rate": 0.000478876629889669, + "loss": 1.6894, + "theoretical_loss": 4.441556991254271, + "tokens_seen": 171311104 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047886659979939824, + "loss": 1.5654, + "theoretical_loss": 4.441339315080861, + "tokens_seen": 171376640 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047885656970912736, + "loss": 1.6304, + "theoretical_loss": 4.441121745430332, + "tokens_seen": 171442176 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004788465396188566, + "loss": 1.6704, + "theoretical_loss": 4.440904282209865, + "tokens_seen": 171507712 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004788365095285858, + "loss": 1.6453, + "theoretical_loss": 4.440686925326759, + "tokens_seen": 171573248 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047882647943831496, + "loss": 1.6477, + "theoretical_loss": 4.440469674688426, + "tokens_seen": 171638784 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047881644934804414, + "loss": 1.5634, + "theoretical_loss": 4.4402525302023985, + "tokens_seen": 171704320 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004788064192577733, + "loss": 1.719, + "theoretical_loss": 4.440035491776321, + "tokens_seen": 171769856 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004787963891675025, + "loss": 1.6816, + "theoretical_loss": 4.439818559317954, + "tokens_seen": 171835392 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047878635907723174, + "loss": 1.5253, + "theoretical_loss": 4.439601732735176, + "tokens_seen": 171900928 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047877632898696087, + "loss": 1.6334, + "theoretical_loss": 4.439385011935977, + "tokens_seen": 171966464 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": 0.0005362321389839053, + "objective/train/docs_used": 108471, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2646145820617676, + "objective/train/original_loss": 3.2646148204803467, + "objective/train/theoretical_loss": 4.439168396828466, + "objective/train/tokens_used": 192492000, + "objective/train/value_avg": -0.00460052490234375, + "objective/train/value_loss": 8.281418558908626e-05, + "objective/train/value_max": -0.0003407001495361328, + "objective/train/value_min": -0.06488037109375, + "objective/train/value_reward_corr": 0.16786030183876174, + "objective/train/value_std": 0.0037822723388671875, + "objective/train/weight_avg": 1.0090911388397217, + "objective/train/weighted_lm_loss": 3.2978157997131348, + "objective/train/weights_max": 1.9132506847381592, + "objective/train/weights_min": 0.3132558763027191, + "theoretical_loss": 4.439168396828466, + "tokens_seen": 172032000 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004787662988966901, + "loss": 1.6304, + "theoretical_loss": 4.439168396828466, + "tokens_seen": 172032000 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004787562688064193, + "loss": 1.6169, + "theoretical_loss": 4.438951887320862, + "tokens_seen": 172097536 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047874623871614847, + "loss": 1.6654, + "theoretical_loss": 4.438735483321503, + "tokens_seen": 172163072 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047873620862587765, + "loss": 1.5454, + "theoretical_loss": 4.438519184738839, + "tokens_seen": 172228608 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047872617853560683, + "loss": 1.6326, + "theoretical_loss": 4.438302991481436, + "tokens_seen": 172294144 + }, + { + "epoch": 0.05, + "learning_rate": 0.000478716148445336, + "loss": 1.6254, + "theoretical_loss": 4.43808690345797, + "tokens_seen": 172359680 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047870611835506524, + "loss": 1.515, + "theoretical_loss": 4.437870920577237, + "tokens_seen": 172425216 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047869608826479437, + "loss": 1.5172, + "theoretical_loss": 4.437655042748141, + "tokens_seen": 172490752 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004786860581745236, + "loss": 1.5998, + "theoretical_loss": 4.437439269879704, + "tokens_seen": 172556288 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047867602808425273, + "loss": 1.6561, + "theoretical_loss": 4.437223601881057, + "tokens_seen": 172621824 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047866599799398197, + "loss": 1.6548, + "theoretical_loss": 4.437008038661449, + "tokens_seen": 172687360 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047865596790371115, + "loss": 1.6823, + "theoretical_loss": 4.436792580130238, + "tokens_seen": 172752896 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047864593781344033, + "loss": 1.5623, + "theoretical_loss": 4.436577226196894, + "tokens_seen": 172818432 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004786359077231695, + "loss": 1.6074, + "theoretical_loss": 4.436361976771005, + "tokens_seen": 172883968 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047862587763289875, + "loss": 1.561, + "theoretical_loss": 4.436146831762267, + "tokens_seen": 172949504 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004786158475426279, + "loss": 1.6342, + "theoretical_loss": 4.435931791080489, + "tokens_seen": 173015040 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004786058174523571, + "loss": 1.624, + "theoretical_loss": 4.435716854635592, + "tokens_seen": 173080576 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047859578736208624, + "loss": 1.5922, + "theoretical_loss": 4.43550202233761, + "tokens_seen": 173146112 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047858575727181547, + "loss": 1.6097, + "theoretical_loss": 4.435287294096685, + "tokens_seen": 173211648 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047857572718154465, + "loss": 1.7632, + "theoretical_loss": 4.435072669823078, + "tokens_seen": 173277184 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047856569709127383, + "loss": 1.6751, + "theoretical_loss": 4.434858149427153, + "tokens_seen": 173342720 + }, + { + "epoch": 0.05, + "learning_rate": 0.000478555667001003, + "loss": 1.6041, + "theoretical_loss": 4.4346437328193895, + "tokens_seen": 173408256 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004785456369107322, + "loss": 1.6629, + "theoretical_loss": 4.434429419910377, + "tokens_seen": 173473792 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004785356068204614, + "loss": 1.5892, + "theoretical_loss": 4.434215210610818, + "tokens_seen": 173539328 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004785255767301906, + "loss": 1.7711, + "theoretical_loss": 4.43400110483152, + "tokens_seen": 173604864 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0005488115712068975, + "objective/train/docs_used": 109645, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3331103324890137, + "objective/train/original_loss": 3.333110809326172, + "objective/train/theoretical_loss": 4.433787102483406, + "objective/train/tokens_used": 194130400, + "objective/train/value_avg": -0.004665374755859375, + "objective/train/value_loss": 6.432956433854997e-05, + "objective/train/value_max": -0.00018525123596191406, + "objective/train/value_min": -0.1402587890625, + "objective/train/value_reward_corr": 0.16826627488135332, + "objective/train/value_std": 0.004486083984375, + "objective/train/weight_avg": 0.9976364970207214, + "objective/train/weighted_lm_loss": 3.327805757522583, + "objective/train/weights_max": 4.065707683563232, + "objective/train/weights_min": 0.38797613978385925, + "theoretical_loss": 4.433787102483406, + "tokens_seen": 173670400 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047851554663991974, + "loss": 1.6569, + "theoretical_loss": 4.433787102483406, + "tokens_seen": 173670400 + }, + { + "epoch": 0.05, + "learning_rate": 0.000478505516549649, + "loss": 1.6701, + "theoretical_loss": 4.4335732034775095, + "tokens_seen": 173735936 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004784954864593781, + "loss": 1.6525, + "theoretical_loss": 4.43335940772497, + "tokens_seen": 173801472 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047848545636910734, + "loss": 1.6001, + "theoretical_loss": 4.433145715137038, + "tokens_seen": 173867008 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004784754262788365, + "loss": 1.7999, + "theoretical_loss": 4.4329321256250775, + "tokens_seen": 173932544 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004784653961885657, + "loss": 1.6769, + "theoretical_loss": 4.432718639100558, + "tokens_seen": 173998080 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004784553660982949, + "loss": 1.7592, + "theoretical_loss": 4.43250525547506, + "tokens_seen": 174063616 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004784453360080241, + "loss": 1.6358, + "theoretical_loss": 4.432291974660272, + "tokens_seen": 174129152 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047843530591775324, + "loss": 1.5835, + "theoretical_loss": 4.432078796567993, + "tokens_seen": 174194688 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004784252758274825, + "loss": 1.6639, + "theoretical_loss": 4.43186572111013, + "tokens_seen": 174260224 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004784152457372116, + "loss": 1.5973, + "theoretical_loss": 4.431652748198697, + "tokens_seen": 174325760 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047840521564694084, + "loss": 1.5643, + "theoretical_loss": 4.431439877745822, + "tokens_seen": 174391296 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047839518555667, + "loss": 1.4386, + "theoretical_loss": 4.431227109663735, + "tokens_seen": 174456832 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004783851554663992, + "loss": 1.7543, + "theoretical_loss": 4.431014443864776, + "tokens_seen": 174522368 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004783751253761284, + "loss": 1.5906, + "theoretical_loss": 4.430801880261395, + "tokens_seen": 174587904 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047836509528585756, + "loss": 1.6556, + "theoretical_loss": 4.430589418766148, + "tokens_seen": 174653440 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047835506519558675, + "loss": 1.6072, + "theoretical_loss": 4.430377059291699, + "tokens_seen": 174718976 + }, + { + "epoch": 0.05, + "learning_rate": 0.000478345035105316, + "loss": 1.7469, + "theoretical_loss": 4.430164801750817, + "tokens_seen": 174784512 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004783350050150451, + "loss": 1.6235, + "theoretical_loss": 4.4299526460563845, + "tokens_seen": 174850048 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047832497492477434, + "loss": 1.5824, + "theoretical_loss": 4.429740592121385, + "tokens_seen": 174915584 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047831494483450347, + "loss": 1.6658, + "theoretical_loss": 4.429528639858911, + "tokens_seen": 174981120 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004783049147442327, + "loss": 1.6635, + "theoretical_loss": 4.429316789182162, + "tokens_seen": 175046656 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004782948846539619, + "loss": 1.6609, + "theoretical_loss": 4.429105040004445, + "tokens_seen": 175112192 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047828485456369107, + "loss": 1.6143, + "theoretical_loss": 4.428893392239171, + "tokens_seen": 175177728 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047827482447342025, + "loss": 1.6467, + "theoretical_loss": 4.428681845799859, + "tokens_seen": 175243264 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0022454136051237583, + "objective/train/docs_used": 110166, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.499192237854004, + "objective/train/original_loss": 3.499192714691162, + "objective/train/theoretical_loss": 4.4284704006001325, + "objective/train/tokens_used": 195768800, + "objective/train/value_avg": -0.005733489990234375, + "objective/train/value_loss": 0.00015903993335086852, + "objective/train/value_max": -0.0002758502960205078, + "objective/train/value_min": -0.169921875, + "objective/train/value_reward_corr": 0.11364452782070504, + "objective/train/value_std": 0.0059356689453125, + "objective/train/weight_avg": 0.9850161075592041, + "objective/train/weighted_lm_loss": 3.4498698711395264, + "objective/train/weights_max": 5.002477169036865, + "objective/train/weights_min": 0.20243151485919952, + "theoretical_loss": 4.4284704006001325, + "tokens_seen": 175308800 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004782647943831495, + "loss": 1.6394, + "theoretical_loss": 4.4284704006001325, + "tokens_seen": 175308800 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004782547642928786, + "loss": 1.7382, + "theoretical_loss": 4.428259056553724, + "tokens_seen": 175374336 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047824473420260785, + "loss": 1.6944, + "theoretical_loss": 4.428047813574468, + "tokens_seen": 175439872 + }, + { + "epoch": 0.05, + "learning_rate": 0.000478234704112337, + "loss": 1.6256, + "theoretical_loss": 4.427836671576307, + "tokens_seen": 175505408 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004782246740220662, + "loss": 1.7421, + "theoretical_loss": 4.427625630473288, + "tokens_seen": 175570944 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047821464393179544, + "loss": 1.6959, + "theoretical_loss": 4.427414690179562, + "tokens_seen": 175636480 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047820461384152457, + "loss": 1.5332, + "theoretical_loss": 4.42720385060939, + "tokens_seen": 175702016 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004781945837512538, + "loss": 1.6216, + "theoretical_loss": 4.426993111677129, + "tokens_seen": 175767552 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047818455366098293, + "loss": 1.7836, + "theoretical_loss": 4.42678247329725, + "tokens_seen": 175833088 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047817452357071217, + "loss": 1.6832, + "theoretical_loss": 4.426571935384323, + "tokens_seen": 175898624 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047816449348044135, + "loss": 1.5587, + "theoretical_loss": 4.426361497853024, + "tokens_seen": 175964160 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047815446339017053, + "loss": 1.6529, + "theoretical_loss": 4.426151160618133, + "tokens_seen": 176029696 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004781444332998997, + "loss": 1.622, + "theoretical_loss": 4.425940923594533, + "tokens_seen": 176095232 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047813440320962895, + "loss": 1.6419, + "theoretical_loss": 4.4257307866972155, + "tokens_seen": 176160768 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004781243731193581, + "loss": 1.6404, + "theoretical_loss": 4.425520749841268, + "tokens_seen": 176226304 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004781143430290873, + "loss": 1.6638, + "theoretical_loss": 4.425310812941888, + "tokens_seen": 176291840 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047810431293881644, + "loss": 1.5286, + "theoretical_loss": 4.425100975914372, + "tokens_seen": 176357376 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047809428284854567, + "loss": 1.6693, + "theoretical_loss": 4.424891238674125, + "tokens_seen": 176422912 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047808425275827485, + "loss": 1.5779, + "theoretical_loss": 4.424681601136649, + "tokens_seen": 176488448 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047807422266800403, + "loss": 1.546, + "theoretical_loss": 4.4244720632175545, + "tokens_seen": 176553984 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004780641925777332, + "loss": 1.6359, + "theoretical_loss": 4.42426262483255, + "tokens_seen": 176619520 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004780541624874624, + "loss": 1.7128, + "theoretical_loss": 4.424053285897449, + "tokens_seen": 176685056 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004780441323971916, + "loss": 1.7171, + "theoretical_loss": 4.423844046328166, + "tokens_seen": 176750592 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004780341023069208, + "loss": 1.5665, + "theoretical_loss": 4.4236349060407205, + "tokens_seen": 176816128 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047802407221664994, + "loss": 1.689, + "theoretical_loss": 4.423425864951231, + "tokens_seen": 176881664 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0003916101122740656, + "objective/train/docs_used": 110915, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2701847553253174, + "objective/train/original_loss": 3.2701847553253174, + "objective/train/theoretical_loss": 4.42321692297592, + "objective/train/tokens_used": 197407200, + "objective/train/value_avg": -0.00493621826171875, + "objective/train/value_loss": 5.3145056881476194e-05, + "objective/train/value_max": -0.00033283233642578125, + "objective/train/value_min": -0.10015869140625, + "objective/train/value_reward_corr": 0.2045033002627345, + "objective/train/value_std": 0.00433349609375, + "objective/train/weight_avg": 0.9986991286277771, + "objective/train/weighted_lm_loss": 3.2671732902526855, + "objective/train/weights_max": 2.7225990295410156, + "objective/train/weights_min": 0.6898055672645569, + "theoretical_loss": 4.42321692297592, + "tokens_seen": 176947200 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004780140421263792, + "loss": 1.5957, + "theoretical_loss": 4.42321692297592, + "tokens_seen": 176947200 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004780040120361083, + "loss": 1.5738, + "theoretical_loss": 4.42300808003111, + "tokens_seen": 177012736 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047799398194583754, + "loss": 1.6797, + "theoretical_loss": 4.422799336033226, + "tokens_seen": 177078272 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004779839518555667, + "loss": 1.6669, + "theoretical_loss": 4.4225906908987955, + "tokens_seen": 177143808 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004779739217652959, + "loss": 1.6742, + "theoretical_loss": 4.422382144544446, + "tokens_seen": 177209344 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004779638916750251, + "loss": 1.5999, + "theoretical_loss": 4.4221736968869045, + "tokens_seen": 177274880 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004779538615847543, + "loss": 1.6061, + "theoretical_loss": 4.421965347843003, + "tokens_seen": 177340416 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047794383149448344, + "loss": 1.5636, + "theoretical_loss": 4.421757097329672, + "tokens_seen": 177405952 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004779338014042127, + "loss": 1.582, + "theoretical_loss": 4.421548945263941, + "tokens_seen": 177471488 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004779237713139418, + "loss": 1.7078, + "theoretical_loss": 4.421340891562943, + "tokens_seen": 177537024 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047791374122367104, + "loss": 1.5826, + "theoretical_loss": 4.421132936143909, + "tokens_seen": 177602560 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004779037111334002, + "loss": 1.5644, + "theoretical_loss": 4.420925078924172, + "tokens_seen": 177668096 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004778936810431294, + "loss": 1.707, + "theoretical_loss": 4.420717319821163, + "tokens_seen": 177733632 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004778836509528586, + "loss": 1.6516, + "theoretical_loss": 4.420509658752415, + "tokens_seen": 177799168 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047787362086258776, + "loss": 1.5247, + "theoretical_loss": 4.420302095635559, + "tokens_seen": 177864704 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047786359077231695, + "loss": 1.6871, + "theoretical_loss": 4.420094630388327, + "tokens_seen": 177930240 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004778535606820462, + "loss": 1.5933, + "theoretical_loss": 4.419887262928549, + "tokens_seen": 177995776 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004778435305917753, + "loss": 1.5948, + "theoretical_loss": 4.419679993174155, + "tokens_seen": 178061312 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047783350050150454, + "loss": 1.5477, + "theoretical_loss": 4.419472821043172, + "tokens_seen": 178126848 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047782347041123367, + "loss": 1.6453, + "theoretical_loss": 4.41926574645373, + "tokens_seen": 178192384 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004778134403209629, + "loss": 1.739, + "theoretical_loss": 4.419058769324055, + "tokens_seen": 178257920 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004778034102306921, + "loss": 1.7296, + "theoretical_loss": 4.41885188957247, + "tokens_seen": 178323456 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047779338014042127, + "loss": 1.7282, + "theoretical_loss": 4.418645107117401, + "tokens_seen": 178388992 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047778335005015045, + "loss": 1.5953, + "theoretical_loss": 4.418438421877369, + "tokens_seen": 178454528 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004777733199598797, + "loss": 1.6345, + "theoretical_loss": 4.418231833770992, + "tokens_seen": 178520064 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": -0.0004318766004871577, + "objective/train/docs_used": 112413, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1470284461975098, + "objective/train/original_loss": 3.147028684616089, + "objective/train/theoretical_loss": 4.418025342716991, + "objective/train/tokens_used": 199045600, + "objective/train/value_avg": -0.004703521728515625, + "objective/train/value_loss": 6.757702067261562e-05, + "objective/train/value_max": -0.0002359151840209961, + "objective/train/value_min": -0.055816650390625, + "objective/train/value_reward_corr": 0.19166889700553238, + "objective/train/value_std": 0.0035800933837890625, + "objective/train/weight_avg": 0.9986901879310608, + "objective/train/weighted_lm_loss": 3.1362743377685547, + "objective/train/weights_max": 1.668764591217041, + "objective/train/weights_min": 0.08696535229682922, + "theoretical_loss": 4.418025342716991, + "tokens_seen": 178585600 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004777632898696088, + "loss": 1.562, + "theoretical_loss": 4.418025342716991, + "tokens_seen": 178585600 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047775325977933805, + "loss": 1.5322, + "theoretical_loss": 4.417818948634178, + "tokens_seen": 178651136 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004777432296890672, + "loss": 1.6716, + "theoretical_loss": 4.41761265144147, + "tokens_seen": 178716672 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004777331995987964, + "loss": 1.6084, + "theoretical_loss": 4.417406451057875, + "tokens_seen": 178782208 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004777231695085256, + "loss": 1.5198, + "theoretical_loss": 4.417200347402503, + "tokens_seen": 178847744 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047771313941825477, + "loss": 1.5743, + "theoretical_loss": 4.416994340394557, + "tokens_seen": 178913280 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047770310932798395, + "loss": 1.6162, + "theoretical_loss": 4.416788429953341, + "tokens_seen": 178978816 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047769307923771313, + "loss": 1.7008, + "theoretical_loss": 4.416582615998255, + "tokens_seen": 179044352 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004776830491474423, + "loss": 1.5722, + "theoretical_loss": 4.416376898448793, + "tokens_seen": 179109888 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047767301905717155, + "loss": 1.5624, + "theoretical_loss": 4.416171277224549, + "tokens_seen": 179175424 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004776629889669007, + "loss": 1.6611, + "theoretical_loss": 4.415965752245213, + "tokens_seen": 179240960 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004776529588766299, + "loss": 1.595, + "theoretical_loss": 4.415760323430568, + "tokens_seen": 179306496 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047764292878635904, + "loss": 1.5458, + "theoretical_loss": 4.415554990700496, + "tokens_seen": 179372032 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004776328986960883, + "loss": 1.5232, + "theoretical_loss": 4.415349753974977, + "tokens_seen": 179437568 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047762286860581746, + "loss": 1.6278, + "theoretical_loss": 4.41514461317408, + "tokens_seen": 179503104 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047761283851554664, + "loss": 3.455, + "theoretical_loss": 4.414939568217978, + "tokens_seen": 179568640 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004776028084252758, + "loss": 1.7385, + "theoretical_loss": 4.414734619026934, + "tokens_seen": 179634176 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047759277833500505, + "loss": 1.6428, + "theoretical_loss": 4.4145297655213085, + "tokens_seen": 179699712 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004775827482447342, + "loss": 1.685, + "theoretical_loss": 4.414325007621555, + "tokens_seen": 179765248 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004775727181544634, + "loss": 1.639, + "theoretical_loss": 4.414120345248225, + "tokens_seen": 179830784 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047756268806419254, + "loss": 1.5568, + "theoretical_loss": 4.4139157783219645, + "tokens_seen": 179896320 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004775526579739218, + "loss": 1.683, + "theoretical_loss": 4.4137113067635125, + "tokens_seen": 179961856 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047754262788365096, + "loss": 1.6588, + "theoretical_loss": 4.413506930493703, + "tokens_seen": 180027392 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047753259779338014, + "loss": 1.6294, + "theoretical_loss": 4.413302649433467, + "tokens_seen": 180092928 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004775225677031093, + "loss": 1.6332, + "theoretical_loss": 4.413098463503827, + "tokens_seen": 180158464 + }, + { + "epoch": 0.05, + "objective/train/advantage_avg": 6.74756447551772e-05, + "objective/train/docs_used": 112982, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1833510398864746, + "objective/train/original_loss": 3.1833508014678955, + "objective/train/theoretical_loss": 4.412894372625901, + "objective/train/tokens_used": 200684000, + "objective/train/value_avg": -0.004970550537109375, + "objective/train/value_loss": 7.567997090518475e-05, + "objective/train/value_max": -0.00033283233642578125, + "objective/train/value_min": -0.1236572265625, + "objective/train/value_reward_corr": 0.1380041409728264, + "objective/train/value_std": 0.004932403564453125, + "objective/train/weight_avg": 1.0043848752975464, + "objective/train/weighted_lm_loss": 3.194762706756592, + "objective/train/weights_max": 3.386185884475708, + "objective/train/weights_min": 0.2937648594379425, + "theoretical_loss": 4.412894372625901, + "tokens_seen": 180224000 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004775125376128385, + "loss": 1.6638, + "theoretical_loss": 4.412894372625901, + "tokens_seen": 180224000 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004775025075225677, + "loss": 1.7565, + "theoretical_loss": 4.412690376720901, + "tokens_seen": 180289536 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004774924774322969, + "loss": 1.7163, + "theoretical_loss": 4.412486475710132, + "tokens_seen": 180355072 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047748244734202605, + "loss": 1.6018, + "theoretical_loss": 4.412282669514994, + "tokens_seen": 180420608 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004774724172517553, + "loss": 1.7087, + "theoretical_loss": 4.412078958056981, + "tokens_seen": 180486144 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004774623871614845, + "loss": 1.5489, + "theoretical_loss": 4.411875341257677, + "tokens_seen": 180551680 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047745235707121364, + "loss": 1.7622, + "theoretical_loss": 4.411671819038765, + "tokens_seen": 180617216 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004774423269809429, + "loss": 1.5967, + "theoretical_loss": 4.411468391322017, + "tokens_seen": 180682752 + }, + { + "epoch": 0.05, + "learning_rate": 0.000477432296890672, + "loss": 1.5351, + "theoretical_loss": 4.411265058029299, + "tokens_seen": 180748288 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047742226680040124, + "loss": 1.6913, + "theoretical_loss": 4.411061819082569, + "tokens_seen": 180813824 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004774122367101304, + "loss": 1.667, + "theoretical_loss": 4.4108586744038805, + "tokens_seen": 180879360 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004774022066198596, + "loss": 1.6682, + "theoretical_loss": 4.410655623915377, + "tokens_seen": 180944896 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004773921765295888, + "loss": 1.6156, + "theoretical_loss": 4.410452667539296, + "tokens_seen": 181010432 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047738214643931797, + "loss": 1.6111, + "theoretical_loss": 4.410249805197967, + "tokens_seen": 181075968 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047737211634904715, + "loss": 1.7229, + "theoretical_loss": 4.410047036813811, + "tokens_seen": 181141504 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004773620862587764, + "loss": 1.5134, + "theoretical_loss": 4.409844362309341, + "tokens_seen": 181207040 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004773520561685055, + "loss": 1.6467, + "theoretical_loss": 4.409641781607164, + "tokens_seen": 181272576 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047734202607823474, + "loss": 1.6643, + "theoretical_loss": 4.4094392946299745, + "tokens_seen": 181338112 + }, + { + "epoch": 0.05, + "learning_rate": 0.00047733199598796387, + "loss": 1.663, + "theoretical_loss": 4.409236901300563, + "tokens_seen": 181403648 + }, + { + "epoch": 0.05, + "learning_rate": 0.0004773219658976931, + "loss": 1.7284, + "theoretical_loss": 4.40903460154181, + "tokens_seen": 181469184 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004773119358074223, + "loss": 1.6074, + "theoretical_loss": 4.408832395276686, + "tokens_seen": 181534720 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047730190571715147, + "loss": 1.6735, + "theoretical_loss": 4.408630282428255, + "tokens_seen": 181600256 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047729187562688065, + "loss": 1.5595, + "theoretical_loss": 4.408428262919669, + "tokens_seen": 181665792 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004772818455366099, + "loss": 1.6317, + "theoretical_loss": 4.408226336674173, + "tokens_seen": 181731328 + }, + { + "epoch": 0.06, + "learning_rate": 0.000477271815446339, + "loss": 1.7024, + "theoretical_loss": 4.4080245036151045, + "tokens_seen": 181796864 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -5.915776273468509e-05, + "objective/train/docs_used": 114216, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.187669515609741, + "objective/train/original_loss": 3.187669038772583, + "objective/train/theoretical_loss": 4.407822763665887, + "objective/train/tokens_used": 202322400, + "objective/train/value_avg": -0.004276275634765625, + "objective/train/value_loss": 6.049642979633063e-05, + "objective/train/value_max": -0.00013768672943115234, + "objective/train/value_min": -0.091064453125, + "objective/train/value_reward_corr": 0.11438448277129878, + "objective/train/value_std": 0.004085540771484375, + "objective/train/weight_avg": 1.002307415008545, + "objective/train/weighted_lm_loss": 3.206420421600342, + "objective/train/weights_max": 2.485924243927002, + "objective/train/weights_min": 0.33174893260002136, + "theoretical_loss": 4.407822763665887, + "tokens_seen": 181862400 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047726178535606825, + "loss": 1.605, + "theoretical_loss": 4.407822763665887, + "tokens_seen": 181862400 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004772517552657974, + "loss": 1.6965, + "theoretical_loss": 4.407621116750039, + "tokens_seen": 181927936 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004772417251755266, + "loss": 1.7027, + "theoretical_loss": 4.4074195627911665, + "tokens_seen": 181993472 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004772316950852558, + "loss": 1.7028, + "theoretical_loss": 4.407218101712966, + "tokens_seen": 182059008 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047722166499498497, + "loss": 1.6763, + "theoretical_loss": 4.407016733439226, + "tokens_seen": 182124544 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047721163490471415, + "loss": 1.6056, + "theoretical_loss": 4.406815457893824, + "tokens_seen": 182190080 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047720160481444333, + "loss": 1.6386, + "theoretical_loss": 4.4066142750007256, + "tokens_seen": 182255616 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004771915747241725, + "loss": 1.5751, + "theoretical_loss": 4.406413184683987, + "tokens_seen": 182321152 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047718154463390175, + "loss": 1.5921, + "theoretical_loss": 4.406212186867756, + "tokens_seen": 182386688 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004771715145436309, + "loss": 1.5995, + "theoretical_loss": 4.406011281476267, + "tokens_seen": 182452224 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004771614844533601, + "loss": 1.579, + "theoretical_loss": 4.405810468433844, + "tokens_seen": 182517760 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047715145436308924, + "loss": 1.6542, + "theoretical_loss": 4.4056097476649025, + "tokens_seen": 182583296 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004771414242728185, + "loss": 1.5997, + "theoretical_loss": 4.405409119093944, + "tokens_seen": 182648832 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047713139418254766, + "loss": 1.5886, + "theoretical_loss": 4.405208582645562, + "tokens_seen": 182714368 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047712136409227684, + "loss": 1.7552, + "theoretical_loss": 4.405008138244434, + "tokens_seen": 182779904 + }, + { + "epoch": 0.06, + "learning_rate": 0.000477111334002006, + "loss": 1.7465, + "theoretical_loss": 4.404807785815331, + "tokens_seen": 182845440 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047710130391173525, + "loss": 1.7056, + "theoretical_loss": 4.40460752528311, + "tokens_seen": 182910976 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004770912738214644, + "loss": 1.717, + "theoretical_loss": 4.404407356572715, + "tokens_seen": 182976512 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004770812437311936, + "loss": 1.5787, + "theoretical_loss": 4.404207279609182, + "tokens_seen": 183042048 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047707121364092274, + "loss": 1.6963, + "theoretical_loss": 4.404007294317632, + "tokens_seen": 183107584 + }, + { + "epoch": 0.06, + "learning_rate": 0.000477061183550652, + "loss": 1.6296, + "theoretical_loss": 4.403807400623273, + "tokens_seen": 183173120 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047705115346038116, + "loss": 1.6514, + "theoretical_loss": 4.403607598451404, + "tokens_seen": 183238656 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047704112337011034, + "loss": 1.6408, + "theoretical_loss": 4.40340788772741, + "tokens_seen": 183304192 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004770310932798395, + "loss": 1.6548, + "theoretical_loss": 4.403208268376764, + "tokens_seen": 183369728 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004770210631895687, + "loss": 1.7054, + "theoretical_loss": 4.403008740325024, + "tokens_seen": 183435264 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.0013485237723216414, + "objective/train/docs_used": 114775, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2710623741149902, + "objective/train/original_loss": 3.2710626125335693, + "objective/train/theoretical_loss": 4.40280930349784, + "objective/train/tokens_used": 203960800, + "objective/train/value_avg": -0.004852294921875, + "objective/train/value_loss": 8.19043125375174e-05, + "objective/train/value_max": -0.0002359151840209961, + "objective/train/value_min": -0.162109375, + "objective/train/value_reward_corr": 0.2294086585482485, + "objective/train/value_std": 0.00485992431640625, + "objective/train/weight_avg": 0.9904411435127258, + "objective/train/weighted_lm_loss": 3.2439463138580322, + "objective/train/weights_max": 4.630191326141357, + "objective/train/weights_min": 0.34162816405296326, + "theoretical_loss": 4.40280930349784, + "tokens_seen": 183500800 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004770110330992979, + "loss": 1.689, + "theoretical_loss": 4.40280930349784, + "tokens_seen": 183500800 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004770010030090271, + "loss": 1.518, + "theoretical_loss": 4.4026099578209426, + "tokens_seen": 183566336 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047699097291875625, + "loss": 1.5089, + "theoretical_loss": 4.402410703220154, + "tokens_seen": 183631872 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004769809428284855, + "loss": 1.7371, + "theoretical_loss": 4.4022115396213835, + "tokens_seen": 183697408 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047697091273821466, + "loss": 1.7285, + "theoretical_loss": 4.402012466950625, + "tokens_seen": 183762944 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047696088264794384, + "loss": 1.6465, + "theoretical_loss": 4.4018134851339585, + "tokens_seen": 183828480 + }, + { + "epoch": 0.06, + "learning_rate": 0.000476950852557673, + "loss": 1.7414, + "theoretical_loss": 4.401614594097554, + "tokens_seen": 183894016 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004769408224674022, + "loss": 1.6987, + "theoretical_loss": 4.401415793767662, + "tokens_seen": 183959552 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004769307923771314, + "loss": 1.619, + "theoretical_loss": 4.401217084070625, + "tokens_seen": 184025088 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004769207622868606, + "loss": 1.6643, + "theoretical_loss": 4.401018464932868, + "tokens_seen": 184090624 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047691073219658975, + "loss": 1.5807, + "theoretical_loss": 4.400819936280903, + "tokens_seen": 184156160 + }, + { + "epoch": 0.06, + "learning_rate": 0.000476900702106319, + "loss": 1.613, + "theoretical_loss": 4.400621498041327, + "tokens_seen": 184221696 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004768906720160481, + "loss": 1.5739, + "theoretical_loss": 4.400423150140825, + "tokens_seen": 184287232 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047688064192577735, + "loss": 1.6504, + "theoretical_loss": 4.400224892506165, + "tokens_seen": 184352768 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047687061183550653, + "loss": 1.7289, + "theoretical_loss": 4.400026725064199, + "tokens_seen": 184418304 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004768605817452357, + "loss": 1.6985, + "theoretical_loss": 4.39982864774187, + "tokens_seen": 184483840 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004768505516549649, + "loss": 1.6823, + "theoretical_loss": 4.3996306604662, + "tokens_seen": 184549376 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047684052156469407, + "loss": 1.7349, + "theoretical_loss": 4.399432763164299, + "tokens_seen": 184614912 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047683049147442325, + "loss": 1.6567, + "theoretical_loss": 4.399234955763362, + "tokens_seen": 184680448 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004768204613841525, + "loss": 1.591, + "theoretical_loss": 4.399037238190667, + "tokens_seen": 184745984 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004768104312938816, + "loss": 1.6313, + "theoretical_loss": 4.398839610373579, + "tokens_seen": 184811520 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047680040120361085, + "loss": 1.6102, + "theoretical_loss": 4.398642072239545, + "tokens_seen": 184877056 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047679037111334003, + "loss": 1.6572, + "theoretical_loss": 4.398444623716097, + "tokens_seen": 184942592 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004767803410230692, + "loss": 1.6653, + "theoretical_loss": 4.398247264730853, + "tokens_seen": 185008128 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004767703109327984, + "loss": 1.7204, + "theoretical_loss": 4.398049995211513, + "tokens_seen": 185073664 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.0007021362544037402, + "objective/train/docs_used": 115939, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.5056755542755127, + "objective/train/original_loss": 3.5056753158569336, + "objective/train/theoretical_loss": 4.397852815085862, + "objective/train/tokens_used": 205599200, + "objective/train/value_avg": -0.004390716552734375, + "objective/train/value_loss": 4.470492058317177e-05, + "objective/train/value_max": -0.0002415180206298828, + "objective/train/value_min": -0.05987548828125, + "objective/train/value_reward_corr": 0.28346733320381295, + "objective/train/value_std": 0.00360107421875, + "objective/train/weight_avg": 0.9951296448707581, + "objective/train/weighted_lm_loss": 3.493630886077881, + "objective/train/weights_max": 1.7922935485839844, + "objective/train/weights_min": 0.5958160161972046, + "theoretical_loss": 4.397852815085862, + "tokens_seen": 185139200 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004767602808425276, + "loss": 1.5574, + "theoretical_loss": 4.397852815085862, + "tokens_seen": 185139200 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047675025075225676, + "loss": 1.6276, + "theoretical_loss": 4.397655724281767, + "tokens_seen": 185204736 + }, + { + "epoch": 0.06, + "learning_rate": 0.000476740220661986, + "loss": 1.7151, + "theoretical_loss": 4.397458722727183, + "tokens_seen": 185270272 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004767301905717151, + "loss": 1.6451, + "theoretical_loss": 4.397261810350145, + "tokens_seen": 185335808 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047672016048144435, + "loss": 1.5987, + "theoretical_loss": 4.397064987078769, + "tokens_seen": 185401344 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047671013039117353, + "loss": 1.5104, + "theoretical_loss": 4.396868252841262, + "tokens_seen": 185466880 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004767001003009027, + "loss": 1.5753, + "theoretical_loss": 4.396671607565907, + "tokens_seen": 185532416 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047669007021063195, + "loss": 1.6585, + "theoretical_loss": 4.396475051181074, + "tokens_seen": 185597952 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004766800401203611, + "loss": 1.6025, + "theoretical_loss": 4.3962785836152145, + "tokens_seen": 185663488 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004766700100300903, + "loss": 1.6091, + "theoretical_loss": 4.396082204796863, + "tokens_seen": 185729024 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047665997993981944, + "loss": 1.7055, + "theoretical_loss": 4.395885914654636, + "tokens_seen": 185794560 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004766499498495487, + "loss": 1.6614, + "theoretical_loss": 4.395689713117235, + "tokens_seen": 185860096 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047663991975927786, + "loss": 1.5762, + "theoretical_loss": 4.395493600113442, + "tokens_seen": 185925632 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047662988966900704, + "loss": 1.6091, + "theoretical_loss": 4.39529757557212, + "tokens_seen": 185991168 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004766198595787362, + "loss": 1.7348, + "theoretical_loss": 4.395101639422217, + "tokens_seen": 186056704 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047660982948846545, + "loss": 1.5866, + "theoretical_loss": 4.3949057915927625, + "tokens_seen": 186122240 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004765997993981946, + "loss": 1.689, + "theoretical_loss": 4.394710032012865, + "tokens_seen": 186187776 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004765897693079238, + "loss": 1.6264, + "theoretical_loss": 4.39451436061172, + "tokens_seen": 186253312 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047657973921765294, + "loss": 1.6043, + "theoretical_loss": 4.3943187773186, + "tokens_seen": 186318848 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004765697091273822, + "loss": 1.5324, + "theoretical_loss": 4.3941232820628615, + "tokens_seen": 186384384 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047655967903711136, + "loss": 1.5366, + "theoretical_loss": 4.393927874773942, + "tokens_seen": 186449920 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047654964894684054, + "loss": 1.4868, + "theoretical_loss": 4.3937325553813595, + "tokens_seen": 186515456 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004765396188565697, + "loss": 1.6733, + "theoretical_loss": 4.393537323814715, + "tokens_seen": 186580992 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004765295887662989, + "loss": 1.6286, + "theoretical_loss": 4.393342180003689, + "tokens_seen": 186646528 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004765195586760281, + "loss": 1.4655, + "theoretical_loss": 4.393147123878043, + "tokens_seen": 186712064 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": 6.765581929357722e-05, + "objective/train/docs_used": 116562, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3133769035339355, + "objective/train/original_loss": 3.3133769035339355, + "objective/train/theoretical_loss": 4.392952155367621, + "objective/train/tokens_used": 207237600, + "objective/train/value_avg": -0.0045928955078125, + "objective/train/value_loss": 8.992018410935998e-05, + "objective/train/value_max": -0.0002065896987915039, + "objective/train/value_min": -0.123779296875, + "objective/train/value_reward_corr": 0.16207444402505292, + "objective/train/value_std": 0.003963470458984375, + "objective/train/weight_avg": 1.0047105550765991, + "objective/train/weighted_lm_loss": 3.326340913772583, + "objective/train/weights_max": 3.1559746265411377, + "objective/train/weights_min": 0.37577974796295166, + "theoretical_loss": 4.392952155367621, + "tokens_seen": 186777600 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004765095285857573, + "loss": 1.6624, + "theoretical_loss": 4.392952155367621, + "tokens_seen": 186777600 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047649949849548645, + "loss": 1.6748, + "theoretical_loss": 4.392757274402346, + "tokens_seen": 186843136 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004764894684052157, + "loss": 1.697, + "theoretical_loss": 4.392562480912222, + "tokens_seen": 186908672 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047647943831494486, + "loss": 1.6627, + "theoretical_loss": 4.392367774827333, + "tokens_seen": 186974208 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047646940822467404, + "loss": 1.5914, + "theoretical_loss": 4.392173156077845, + "tokens_seen": 187039744 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004764593781344032, + "loss": 1.7558, + "theoretical_loss": 4.391978624594003, + "tokens_seen": 187105280 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004764493480441324, + "loss": 1.6609, + "theoretical_loss": 4.391784180306133, + "tokens_seen": 187170816 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004764393179538616, + "loss": 1.6452, + "theoretical_loss": 4.391589823144638, + "tokens_seen": 187236352 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004764292878635908, + "loss": 1.551, + "theoretical_loss": 4.391395553040007, + "tokens_seen": 187301888 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047641925777331995, + "loss": 1.6467, + "theoretical_loss": 4.391201369922801, + "tokens_seen": 187367424 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004764092276830492, + "loss": 1.7188, + "theoretical_loss": 4.391007273723668, + "tokens_seen": 187432960 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004763991975927783, + "loss": 1.743, + "theoretical_loss": 4.39081326437333, + "tokens_seen": 187498496 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047638916750250755, + "loss": 1.7516, + "theoretical_loss": 4.390619341802592, + "tokens_seen": 187564032 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047637913741223673, + "loss": 1.6332, + "theoretical_loss": 4.390425505942336, + "tokens_seen": 187629568 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004763691073219659, + "loss": 1.7044, + "theoretical_loss": 4.390231756723523, + "tokens_seen": 187695104 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004763590772316951, + "loss": 1.6195, + "theoretical_loss": 4.390038094077196, + "tokens_seen": 187760640 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047634904714142427, + "loss": 1.7456, + "theoretical_loss": 4.389844517934475, + "tokens_seen": 187826176 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047633901705115345, + "loss": 1.7042, + "theoretical_loss": 4.389651028226558, + "tokens_seen": 187891712 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004763289869608827, + "loss": 1.6567, + "theoretical_loss": 4.389457624884722, + "tokens_seen": 187957248 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004763189568706118, + "loss": 1.7358, + "theoretical_loss": 4.389264307840324, + "tokens_seen": 188022784 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047630892678034105, + "loss": 1.6617, + "theoretical_loss": 4.389071077024798, + "tokens_seen": 188088320 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047629889669007023, + "loss": 1.685, + "theoretical_loss": 4.388877932369657, + "tokens_seen": 188153856 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004762888665997994, + "loss": 1.5847, + "theoretical_loss": 4.3886848738064925, + "tokens_seen": 188219392 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004762788365095286, + "loss": 1.6159, + "theoretical_loss": 4.388491901266973, + "tokens_seen": 188284928 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004762688064192578, + "loss": 1.6562, + "theoretical_loss": 4.388299014682847, + "tokens_seen": 188350464 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.0015140672912821174, + "objective/train/docs_used": 117858, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2329676151275635, + "objective/train/original_loss": 3.2329680919647217, + "objective/train/theoretical_loss": 4.388106213985938, + "objective/train/tokens_used": 208876000, + "objective/train/value_avg": -0.004001617431640625, + "objective/train/value_loss": 5.142891677678563e-05, + "objective/train/value_max": -0.0002491474151611328, + "objective/train/value_min": -0.08740234375, + "objective/train/value_reward_corr": 0.2535726283204234, + "objective/train/value_std": 0.0036296844482421875, + "objective/train/weight_avg": 0.9872520565986633, + "objective/train/weighted_lm_loss": 3.206974744796753, + "objective/train/weights_max": 2.3080294132232666, + "objective/train/weights_min": 0.19601033627986908, + "theoretical_loss": 4.388106213985938, + "tokens_seen": 188416000 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047625877632898696, + "loss": 1.5481, + "theoretical_loss": 4.388106213985938, + "tokens_seen": 188416000 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004762487462387162, + "loss": 1.5539, + "theoretical_loss": 4.3879134991081505, + "tokens_seen": 188481536 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004762387161484453, + "loss": 1.5497, + "theoretical_loss": 4.387720869981464, + "tokens_seen": 188547072 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047622868605817455, + "loss": 1.7048, + "theoretical_loss": 4.387528326537936, + "tokens_seen": 188612608 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004762186559679037, + "loss": 1.5828, + "theoretical_loss": 4.3873358687097035, + "tokens_seen": 188678144 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004762086258776329, + "loss": 1.6447, + "theoretical_loss": 4.387143496428978, + "tokens_seen": 188743680 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004761985957873621, + "loss": 1.7244, + "theoretical_loss": 4.38695120962805, + "tokens_seen": 188809216 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004761885656970913, + "loss": 1.6666, + "theoretical_loss": 4.386759008239285, + "tokens_seen": 188874752 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047617853560682046, + "loss": 1.6455, + "theoretical_loss": 4.38656689219513, + "tokens_seen": 188940288 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047616850551654964, + "loss": 1.6514, + "theoretical_loss": 4.386374861428104, + "tokens_seen": 189005824 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004761584754262788, + "loss": 1.6935, + "theoretical_loss": 4.386182915870804, + "tokens_seen": 189071360 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047614844533600806, + "loss": 1.6841, + "theoretical_loss": 4.3859910554559045, + "tokens_seen": 189136896 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004761384152457372, + "loss": 1.7274, + "theoretical_loss": 4.385799280116157, + "tokens_seen": 189202432 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004761283851554664, + "loss": 1.6487, + "theoretical_loss": 4.385607589784388, + "tokens_seen": 189267968 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004761183550651956, + "loss": 1.6598, + "theoretical_loss": 4.3854159843935, + "tokens_seen": 189333504 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004761083249749248, + "loss": 1.5333, + "theoretical_loss": 4.385224463876474, + "tokens_seen": 189399040 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047609829488465396, + "loss": 1.6629, + "theoretical_loss": 4.385033028166365, + "tokens_seen": 189464576 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047608826479438314, + "loss": 1.6506, + "theoretical_loss": 4.384841677196304, + "tokens_seen": 189530112 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004760782347041123, + "loss": 1.6211, + "theoretical_loss": 4.384650410899498, + "tokens_seen": 189595648 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047606820461384156, + "loss": 1.7624, + "theoretical_loss": 4.384459229209231, + "tokens_seen": 189661184 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004760581745235707, + "loss": 1.6853, + "theoretical_loss": 4.38426813205886, + "tokens_seen": 189726720 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004760481444332999, + "loss": 1.6981, + "theoretical_loss": 4.384077119381821, + "tokens_seen": 189792256 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047603811434302905, + "loss": 1.5506, + "theoretical_loss": 4.383886191111623, + "tokens_seen": 189857792 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004760280842527583, + "loss": 1.6964, + "theoretical_loss": 4.383695347181849, + "tokens_seen": 189923328 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047601805416248746, + "loss": 1.6432, + "theoretical_loss": 4.383504587526161, + "tokens_seen": 189988864 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.0018566895741969347, + "objective/train/docs_used": 118609, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.217747211456299, + "objective/train/original_loss": 3.217747688293457, + "objective/train/theoretical_loss": 4.383313912078293, + "objective/train/tokens_used": 210514400, + "objective/train/value_avg": -0.00469970703125, + "objective/train/value_loss": 8.192943641915917e-05, + "objective/train/value_max": -0.00021314620971679688, + "objective/train/value_min": -0.12158203125, + "objective/train/value_reward_corr": 0.2790606668560142, + "objective/train/value_std": 0.004364013671875, + "objective/train/weight_avg": 0.9853137731552124, + "objective/train/weighted_lm_loss": 3.187112331390381, + "objective/train/weights_max": 3.301065444946289, + "objective/train/weights_min": 0.3742133378982544, + "theoretical_loss": 4.383313912078293, + "tokens_seen": 190054400 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047600802407221665, + "loss": 1.7356, + "theoretical_loss": 4.383313912078293, + "tokens_seen": 190054400 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004759979939819458, + "loss": 1.4822, + "theoretical_loss": 4.383123320772055, + "tokens_seen": 190119936 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047598796389167506, + "loss": 1.7786, + "theoretical_loss": 4.382932813541331, + "tokens_seen": 190185472 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004759779338014042, + "loss": 1.5812, + "theoretical_loss": 4.382742390320081, + "tokens_seen": 190251008 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004759679037111334, + "loss": 1.6313, + "theoretical_loss": 4.382552051042339, + "tokens_seen": 190316544 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004759578736208626, + "loss": 1.6193, + "theoretical_loss": 4.382361795642213, + "tokens_seen": 190382080 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004759478435305918, + "loss": 1.5875, + "theoretical_loss": 4.3821716240538855, + "tokens_seen": 190447616 + }, + { + "epoch": 0.06, + "learning_rate": 0.000475937813440321, + "loss": 1.5372, + "theoretical_loss": 4.381981536211615, + "tokens_seen": 190513152 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047592778335005015, + "loss": 1.6439, + "theoretical_loss": 4.381791532049733, + "tokens_seen": 190578688 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004759177532597794, + "loss": 1.7422, + "theoretical_loss": 4.381601611502643, + "tokens_seen": 190644224 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004759077231695085, + "loss": 1.5538, + "theoretical_loss": 4.381411774504825, + "tokens_seen": 190709760 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047589769307923775, + "loss": 1.7046, + "theoretical_loss": 4.381222020990832, + "tokens_seen": 190775296 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047588766298896693, + "loss": 1.6143, + "theoretical_loss": 4.381032350895292, + "tokens_seen": 190840832 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004758776328986961, + "loss": 1.6437, + "theoretical_loss": 4.380842764152904, + "tokens_seen": 190906368 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004758676028084253, + "loss": 1.6651, + "theoretical_loss": 4.380653260698443, + "tokens_seen": 190971904 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047585757271815447, + "loss": 1.5235, + "theoretical_loss": 4.380463840466755, + "tokens_seen": 191037440 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047584754262788365, + "loss": 1.6672, + "theoretical_loss": 4.380274503392762, + "tokens_seen": 191102976 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004758375125376129, + "loss": 1.8059, + "theoretical_loss": 4.380085249411458, + "tokens_seen": 191168512 + }, + { + "epoch": 0.06, + "learning_rate": 0.000475827482447342, + "loss": 1.6358, + "theoretical_loss": 4.379896078457908, + "tokens_seen": 191234048 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047581745235707125, + "loss": 1.628, + "theoretical_loss": 4.379706990467254, + "tokens_seen": 191299584 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047580742226680043, + "loss": 1.6441, + "theoretical_loss": 4.379517985374708, + "tokens_seen": 191365120 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004757973921765296, + "loss": 1.5521, + "theoretical_loss": 4.379329063115555, + "tokens_seen": 191430656 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004757873620862588, + "loss": 1.5358, + "theoretical_loss": 4.3791402236251535, + "tokens_seen": 191496192 + }, + { + "epoch": 0.06, + "learning_rate": 0.000475777331995988, + "loss": 1.7267, + "theoretical_loss": 4.378951466838935, + "tokens_seen": 191561728 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047576730190571716, + "loss": 1.6229, + "theoretical_loss": 4.378762792692402, + "tokens_seen": 191627264 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.0016334770480170846, + "objective/train/docs_used": 119339, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2669641971588135, + "objective/train/original_loss": 3.2669639587402344, + "objective/train/theoretical_loss": 4.37857420112113, + "objective/train/tokens_used": 212152800, + "objective/train/value_avg": -0.004482269287109375, + "objective/train/value_loss": 0.00010417697194498032, + "objective/train/value_max": -0.0002779960632324219, + "objective/train/value_min": -0.07794189453125, + "objective/train/value_reward_corr": 0.2931094616673652, + "objective/train/value_std": 0.00408172607421875, + "objective/train/weight_avg": 0.9881911873817444, + "objective/train/weighted_lm_loss": 3.2373690605163574, + "objective/train/weights_max": 1.9856791496276855, + "objective/train/weights_min": 0.2882629632949829, + "theoretical_loss": 4.37857420112113, + "tokens_seen": 191692800 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004757572718154464, + "loss": 1.53, + "theoretical_loss": 4.37857420112113, + "tokens_seen": 191692800 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004757472417251755, + "loss": 1.5736, + "theoretical_loss": 4.378385692060767, + "tokens_seen": 191758336 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047573721163490475, + "loss": 1.6552, + "theoretical_loss": 4.378197265447032, + "tokens_seen": 191823872 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004757271815446339, + "loss": 1.6259, + "theoretical_loss": 4.378008921215717, + "tokens_seen": 191889408 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004757171514543631, + "loss": 1.6293, + "theoretical_loss": 4.377820659302687, + "tokens_seen": 191954944 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004757071213640923, + "loss": 1.6518, + "theoretical_loss": 4.377632479643877, + "tokens_seen": 192020480 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004756970912738215, + "loss": 1.6456, + "theoretical_loss": 4.377444382175294, + "tokens_seen": 192086016 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047568706118355066, + "loss": 1.5645, + "theoretical_loss": 4.377256366833016, + "tokens_seen": 192151552 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047567703109327984, + "loss": 1.6288, + "theoretical_loss": 4.3770684335531955, + "tokens_seen": 192217088 + }, + { + "epoch": 0.06, + "learning_rate": 0.000475667001003009, + "loss": 1.5688, + "theoretical_loss": 4.376880582272053, + "tokens_seen": 192282624 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047565697091273826, + "loss": 1.7446, + "theoretical_loss": 4.3766928129258815, + "tokens_seen": 192348160 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004756469408224674, + "loss": 1.5985, + "theoretical_loss": 4.376505125451045, + "tokens_seen": 192413696 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004756369107321966, + "loss": 1.6657, + "theoretical_loss": 4.37631751978398, + "tokens_seen": 192479232 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004756268806419258, + "loss": 1.7863, + "theoretical_loss": 4.376129995861191, + "tokens_seen": 192544768 + }, + { + "epoch": 0.06, + "learning_rate": 0.000475616850551655, + "loss": 1.5663, + "theoretical_loss": 4.3759425536192555, + "tokens_seen": 192610304 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047560682046138416, + "loss": 1.5984, + "theoretical_loss": 4.3757551929948235, + "tokens_seen": 192675840 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047559679037111334, + "loss": 1.6672, + "theoretical_loss": 4.375567913924611, + "tokens_seen": 192741376 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004755867602808425, + "loss": 1.619, + "theoretical_loss": 4.375380716345409, + "tokens_seen": 192806912 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047557673019057176, + "loss": 1.5443, + "theoretical_loss": 4.375193600194075, + "tokens_seen": 192872448 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004755667001003009, + "loss": 1.5577, + "theoretical_loss": 4.375006565407541, + "tokens_seen": 192937984 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004755566700100301, + "loss": 1.5459, + "theoretical_loss": 4.374819611922806, + "tokens_seen": 193003520 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047554663991975925, + "loss": 1.7788, + "theoretical_loss": 4.374632739676942, + "tokens_seen": 193069056 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004755366098294885, + "loss": 1.6868, + "theoretical_loss": 4.374445948607087, + "tokens_seen": 193134592 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047552657973921766, + "loss": 1.5314, + "theoretical_loss": 4.374259238650452, + "tokens_seen": 193200128 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047551654964894685, + "loss": 1.5927, + "theoretical_loss": 4.374072609744318, + "tokens_seen": 193265664 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.0014554819790646434, + "objective/train/docs_used": 120055, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.418766498565674, + "objective/train/original_loss": 3.418766736984253, + "objective/train/theoretical_loss": 4.373886061826036, + "objective/train/tokens_used": 213791200, + "objective/train/value_avg": -0.00511932373046875, + "objective/train/value_loss": 4.8218356823781505e-05, + "objective/train/value_max": -0.00021660327911376953, + "objective/train/value_min": -0.073974609375, + "objective/train/value_reward_corr": 0.26682541104070745, + "objective/train/value_std": 0.00409698486328125, + "objective/train/weight_avg": 0.9878091812133789, + "objective/train/weighted_lm_loss": 3.375635862350464, + "objective/train/weights_max": 1.8839359283447266, + "objective/train/weights_min": 0.6967576146125793, + "theoretical_loss": 4.373886061826036, + "tokens_seen": 193331200 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047550651955867603, + "loss": 1.7038, + "theoretical_loss": 4.373886061826036, + "tokens_seen": 193331200 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047549648946840526, + "loss": 1.6416, + "theoretical_loss": 4.373699594833022, + "tokens_seen": 193396736 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004754864593781344, + "loss": 1.5874, + "theoretical_loss": 4.373513208702769, + "tokens_seen": 193462272 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004754764292878636, + "loss": 1.6043, + "theoretical_loss": 4.373326903372833, + "tokens_seen": 193527808 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047546639919759275, + "loss": 1.6169, + "theoretical_loss": 4.3731406787808424, + "tokens_seen": 193593344 + }, + { + "epoch": 0.06, + "learning_rate": 0.000475456369107322, + "loss": 1.6343, + "theoretical_loss": 4.372954534864496, + "tokens_seen": 193658880 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047544633901705117, + "loss": 1.6771, + "theoretical_loss": 4.372768471561558, + "tokens_seen": 193724416 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047543630892678035, + "loss": 1.6808, + "theoretical_loss": 4.372582488809863, + "tokens_seen": 193789952 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047542627883650953, + "loss": 1.6864, + "theoretical_loss": 4.372396586547319, + "tokens_seen": 193855488 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004754162487462387, + "loss": 1.6627, + "theoretical_loss": 4.372210764711896, + "tokens_seen": 193921024 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004754062186559679, + "loss": 1.6511, + "theoretical_loss": 4.372025023241637, + "tokens_seen": 193986560 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047539618856569713, + "loss": 1.6979, + "theoretical_loss": 4.371839362074652, + "tokens_seen": 194052096 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047538615847542625, + "loss": 1.6999, + "theoretical_loss": 4.37165378114912, + "tokens_seen": 194117632 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004753761283851555, + "loss": 1.7022, + "theoretical_loss": 4.371468280403288, + "tokens_seen": 194183168 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004753660982948846, + "loss": 1.6476, + "theoretical_loss": 4.371282859775474, + "tokens_seen": 194248704 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047535606820461385, + "loss": 1.5721, + "theoretical_loss": 4.371097519204059, + "tokens_seen": 194314240 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047534603811434303, + "loss": 1.605, + "theoretical_loss": 4.370912258627498, + "tokens_seen": 194379776 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004753360080240722, + "loss": 1.6486, + "theoretical_loss": 4.370727077984309, + "tokens_seen": 194445312 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004753259779338014, + "loss": 1.5761, + "theoretical_loss": 4.370541977213082, + "tokens_seen": 194510848 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047531594784353063, + "loss": 1.6748, + "theoretical_loss": 4.370356956252472, + "tokens_seen": 194576384 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047530591775325976, + "loss": 1.6429, + "theoretical_loss": 4.370172015041203, + "tokens_seen": 194641920 + }, + { + "epoch": 0.06, + "learning_rate": 0.000475295887662989, + "loss": 1.6505, + "theoretical_loss": 4.369987153518067, + "tokens_seen": 194707456 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004752858575727181, + "loss": 1.6961, + "theoretical_loss": 4.369802371621922, + "tokens_seen": 194772992 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047527582748244736, + "loss": 1.5441, + "theoretical_loss": 4.369617669291695, + "tokens_seen": 194838528 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047526579739217654, + "loss": 1.5423, + "theoretical_loss": 4.36943304646638, + "tokens_seen": 194904064 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.0018730128649622202, + "objective/train/docs_used": 121265, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.276707410812378, + "objective/train/original_loss": 3.276707887649536, + "objective/train/theoretical_loss": 4.369248503085039, + "objective/train/tokens_used": 215429600, + "objective/train/value_avg": -0.00457000732421875, + "objective/train/value_loss": 6.284283881541342e-05, + "objective/train/value_max": -0.00022172927856445312, + "objective/train/value_min": -0.07720947265625, + "objective/train/value_reward_corr": 0.20163267528917658, + "objective/train/value_std": 0.003814697265625, + "objective/train/weight_avg": 0.9842771291732788, + "objective/train/weighted_lm_loss": 3.2304389476776123, + "objective/train/weights_max": 1.9728951454162598, + "objective/train/weights_min": 0.44272634387016296, + "theoretical_loss": 4.369248503085039, + "tokens_seen": 194969600 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004752557673019057, + "loss": 1.5361, + "theoretical_loss": 4.369248503085039, + "tokens_seen": 194969600 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004752457372116349, + "loss": 1.7779, + "theoretical_loss": 4.3690640390867985, + "tokens_seen": 195035136 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004752357071213641, + "loss": 1.6039, + "theoretical_loss": 4.368879654410855, + "tokens_seen": 195100672 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047522567703109326, + "loss": 1.6852, + "theoretical_loss": 4.368695348996471, + "tokens_seen": 195166208 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004752156469408225, + "loss": 1.6957, + "theoretical_loss": 4.368511122782975, + "tokens_seen": 195231744 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004752056168505517, + "loss": 1.5286, + "theoretical_loss": 4.3683269757097625, + "tokens_seen": 195297280 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047519558676028086, + "loss": 1.6892, + "theoretical_loss": 4.368142907716297, + "tokens_seen": 195362816 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047518555667001004, + "loss": 1.6807, + "theoretical_loss": 4.36795891874211, + "tokens_seen": 195428352 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004751755265797392, + "loss": 1.6803, + "theoretical_loss": 4.3677750087267935, + "tokens_seen": 195493888 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047516549648946846, + "loss": 1.7983, + "theoretical_loss": 4.367591177610011, + "tokens_seen": 195559424 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004751554663991976, + "loss": 1.572, + "theoretical_loss": 4.367407425331493, + "tokens_seen": 195624960 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004751454363089268, + "loss": 1.6788, + "theoretical_loss": 4.367223751831032, + "tokens_seen": 195690496 + }, + { + "epoch": 0.06, + "learning_rate": 0.000475135406218656, + "loss": 1.6724, + "theoretical_loss": 4.367040157048489, + "tokens_seen": 195756032 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004751253761283852, + "loss": 1.6788, + "theoretical_loss": 4.3668566409237926, + "tokens_seen": 195821568 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047511534603811436, + "loss": 1.5555, + "theoretical_loss": 4.3666732033969335, + "tokens_seen": 195887104 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047510531594784354, + "loss": 1.6067, + "theoretical_loss": 4.366489844407972, + "tokens_seen": 195952640 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004750952858575727, + "loss": 1.6666, + "theoretical_loss": 4.36630656389703, + "tokens_seen": 196018176 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047508525576730196, + "loss": 1.6264, + "theoretical_loss": 4.366123361804301, + "tokens_seen": 196083712 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004750752256770311, + "loss": 1.5584, + "theoretical_loss": 4.3659402380700385, + "tokens_seen": 196149248 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004750651955867603, + "loss": 1.7213, + "theoretical_loss": 4.365757192634564, + "tokens_seen": 196214784 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047505516549648945, + "loss": 1.5392, + "theoretical_loss": 4.365574225438263, + "tokens_seen": 196280320 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004750451354062187, + "loss": 1.6869, + "theoretical_loss": 4.365391336421589, + "tokens_seen": 196345856 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047503510531594787, + "loss": 1.6658, + "theoretical_loss": 4.365208525525057, + "tokens_seen": 196411392 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047502507522567705, + "loss": 1.6124, + "theoretical_loss": 4.36502579268925, + "tokens_seen": 196476928 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047501504513540623, + "loss": 1.5294, + "theoretical_loss": 4.364843137854816, + "tokens_seen": 196542464 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.0014808224514126778, + "objective/train/docs_used": 121711, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4903926849365234, + "objective/train/original_loss": 3.4903924465179443, + "objective/train/theoretical_loss": 4.364660560962464, + "objective/train/tokens_used": 217068000, + "objective/train/value_avg": -0.004962921142578125, + "objective/train/value_loss": 8.954705117503181e-05, + "objective/train/value_max": -0.00031757354736328125, + "objective/train/value_min": -0.080078125, + "objective/train/value_reward_corr": 0.2391219845677354, + "objective/train/value_std": 0.004322052001953125, + "objective/train/weight_avg": 0.9893972277641296, + "objective/train/weighted_lm_loss": 3.4640843868255615, + "objective/train/weights_max": 1.823743224143982, + "objective/train/weights_min": 0.4551805853843689, + "theoretical_loss": 4.364660560962464, + "tokens_seen": 196608000 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047500501504513546, + "loss": 1.6171, + "theoretical_loss": 4.364660560962464, + "tokens_seen": 196608000 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004749949849548646, + "loss": 1.7097, + "theoretical_loss": 4.364478061952973, + "tokens_seen": 196673536 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004749849548645938, + "loss": 1.6446, + "theoretical_loss": 4.364295640767184, + "tokens_seen": 196739072 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047497492477432295, + "loss": 1.5771, + "theoretical_loss": 4.364113297346002, + "tokens_seen": 196804608 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004749648946840522, + "loss": 1.5477, + "theoretical_loss": 4.363931031630398, + "tokens_seen": 196870144 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047495486459378137, + "loss": 1.7147, + "theoretical_loss": 4.363748843561407, + "tokens_seen": 196935680 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047494483450351055, + "loss": 1.6297, + "theoretical_loss": 4.363566733080129, + "tokens_seen": 197001216 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047493480441323973, + "loss": 1.6908, + "theoretical_loss": 4.363384700127726, + "tokens_seen": 197066752 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004749247743229689, + "loss": 1.673, + "theoretical_loss": 4.363202744645427, + "tokens_seen": 197132288 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004749147442326981, + "loss": 1.5052, + "theoretical_loss": 4.363020866574524, + "tokens_seen": 197197824 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047490471414242733, + "loss": 1.7431, + "theoretical_loss": 4.362839065856372, + "tokens_seen": 197263360 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047489468405215646, + "loss": 1.592, + "theoretical_loss": 4.3626573424323905, + "tokens_seen": 197328896 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004748846539618857, + "loss": 1.5786, + "theoretical_loss": 4.362475696244064, + "tokens_seen": 197394432 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004748746238716148, + "loss": 1.7361, + "theoretical_loss": 4.362294127232939, + "tokens_seen": 197459968 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047486459378134405, + "loss": 1.6851, + "theoretical_loss": 4.362112635340626, + "tokens_seen": 197525504 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047485456369107323, + "loss": 1.6718, + "theoretical_loss": 4.3619312205088, + "tokens_seen": 197591040 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004748445336008024, + "loss": 1.773, + "theoretical_loss": 4.361749882679199, + "tokens_seen": 197656576 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004748345035105316, + "loss": 1.7333, + "theoretical_loss": 4.361568621793623, + "tokens_seen": 197722112 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047482447342026083, + "loss": 1.7466, + "theoretical_loss": 4.361387437793937, + "tokens_seen": 197787648 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047481444332998996, + "loss": 1.6061, + "theoretical_loss": 4.361206330622068, + "tokens_seen": 197853184 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004748044132397192, + "loss": 1.7321, + "theoretical_loss": 4.361025300220008, + "tokens_seen": 197918720 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004747943831494483, + "loss": 1.5966, + "theoretical_loss": 4.3608443465298095, + "tokens_seen": 197984256 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047478435305917756, + "loss": 1.5151, + "theoretical_loss": 4.360663469493589, + "tokens_seen": 198049792 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047477432296890674, + "loss": 1.7513, + "theoretical_loss": 4.3604826690535265, + "tokens_seen": 198115328 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004747642928786359, + "loss": 1.6187, + "theoretical_loss": 4.360301945151863, + "tokens_seen": 198180864 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.0010157151846215129, + "objective/train/docs_used": 122248, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.364971399307251, + "objective/train/original_loss": 3.364971160888672, + "objective/train/theoretical_loss": 4.360121297730904, + "objective/train/tokens_used": 218706400, + "objective/train/value_avg": -0.005279541015625, + "objective/train/value_loss": 7.905796519480646e-05, + "objective/train/value_max": -0.00030303001403808594, + "objective/train/value_min": -0.1417236328125, + "objective/train/value_reward_corr": 0.10583331433607056, + "objective/train/value_std": 0.00501251220703125, + "objective/train/weight_avg": 0.9937010407447815, + "objective/train/weighted_lm_loss": 3.3649027347564697, + "objective/train/weights_max": 3.4961888790130615, + "objective/train/weights_min": 0.372824102640152, + "theoretical_loss": 4.360121297730904, + "tokens_seen": 198246400 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004747542627883651, + "loss": 1.673, + "theoretical_loss": 4.360121297730904, + "tokens_seen": 198246400 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004747442326980943, + "loss": 1.7335, + "theoretical_loss": 4.359940726733017, + "tokens_seen": 198311936 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047473420260782346, + "loss": 1.5745, + "theoretical_loss": 4.359760232100632, + "tokens_seen": 198377472 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004747241725175527, + "loss": 1.5838, + "theoretical_loss": 4.359579813776239, + "tokens_seen": 198443008 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004747141424272818, + "loss": 1.7159, + "theoretical_loss": 4.359399471702394, + "tokens_seen": 198508544 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047470411233701106, + "loss": 1.7413, + "theoretical_loss": 4.359219205821713, + "tokens_seen": 198574080 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004746940822467402, + "loss": 1.5525, + "theoretical_loss": 4.359039016076876, + "tokens_seen": 198639616 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004746840521564694, + "loss": 1.5433, + "theoretical_loss": 4.358858902410621, + "tokens_seen": 198705152 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004746740220661986, + "loss": 1.5874, + "theoretical_loss": 4.358678864765753, + "tokens_seen": 198770688 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004746639919759278, + "loss": 1.5606, + "theoretical_loss": 4.358498903085135, + "tokens_seen": 198836224 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047465396188565696, + "loss": 1.5822, + "theoretical_loss": 4.358319017311694, + "tokens_seen": 198901760 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004746439317953862, + "loss": 1.6332, + "theoretical_loss": 4.358139207388417, + "tokens_seen": 198967296 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004746339017051153, + "loss": 1.6731, + "theoretical_loss": 4.357959473258354, + "tokens_seen": 199032832 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047462387161484456, + "loss": 1.7527, + "theoretical_loss": 4.3577798148646165, + "tokens_seen": 199098368 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004746138415245737, + "loss": 1.5744, + "theoretical_loss": 4.357600232150376, + "tokens_seen": 199163904 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004746038114343029, + "loss": 1.6904, + "theoretical_loss": 4.357420725058867, + "tokens_seen": 199229440 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004745937813440321, + "loss": 1.6179, + "theoretical_loss": 4.357241293533383, + "tokens_seen": 199294976 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004745837512537613, + "loss": 1.5934, + "theoretical_loss": 4.3570619375172805, + "tokens_seen": 199360512 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047457372116349047, + "loss": 1.5983, + "theoretical_loss": 4.356882656953977, + "tokens_seen": 199426048 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047456369107321965, + "loss": 1.5823, + "theoretical_loss": 4.3567034517869505, + "tokens_seen": 199491584 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047455366098294883, + "loss": 1.6638, + "theoretical_loss": 4.35652432195974, + "tokens_seen": 199557120 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047454363089267807, + "loss": 1.62, + "theoretical_loss": 4.3563452674159455, + "tokens_seen": 199622656 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004745336008024072, + "loss": 1.6529, + "theoretical_loss": 4.356166288099227, + "tokens_seen": 199688192 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047452357071213643, + "loss": 1.6141, + "theoretical_loss": 4.355987383953305, + "tokens_seen": 199753728 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047451354062186555, + "loss": 1.5059, + "theoretical_loss": 4.355808554921964, + "tokens_seen": 199819264 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": 0.0008952340576797724, + "objective/train/docs_used": 123553, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.298407554626465, + "objective/train/original_loss": 3.2984073162078857, + "objective/train/theoretical_loss": 4.355629800949043, + "objective/train/tokens_used": 220344800, + "objective/train/value_avg": -0.004856109619140625, + "objective/train/value_loss": 4.926410110783763e-05, + "objective/train/value_max": -0.00022339820861816406, + "objective/train/value_min": -0.09136962890625, + "objective/train/value_reward_corr": 0.10699764352571954, + "objective/train/value_std": 0.004058837890625, + "objective/train/weight_avg": 1.0113779306411743, + "objective/train/weighted_lm_loss": 3.3353934288024902, + "objective/train/weights_max": 2.418647527694702, + "objective/train/weights_min": 0.5013248324394226, + "theoretical_loss": 4.355629800949043, + "tokens_seen": 199884800 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004745035105315948, + "loss": 1.5917, + "theoretical_loss": 4.355629800949043, + "tokens_seen": 199884800 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047449348044132397, + "loss": 1.6243, + "theoretical_loss": 4.3554511219784455, + "tokens_seen": 199950336 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047448345035105315, + "loss": 1.5844, + "theoretical_loss": 4.355272517954134, + "tokens_seen": 200015872 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004744734202607824, + "loss": 1.6085, + "theoretical_loss": 4.355093988820133, + "tokens_seen": 200081408 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047446339017051157, + "loss": 1.5629, + "theoretical_loss": 4.354915534520525, + "tokens_seen": 200146944 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047445336008024075, + "loss": 1.7054, + "theoretical_loss": 4.354737154999452, + "tokens_seen": 200212480 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047444332998996993, + "loss": 1.5355, + "theoretical_loss": 4.354558850201118, + "tokens_seen": 200278016 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004744332998996991, + "loss": 1.644, + "theoretical_loss": 4.3543806200697865, + "tokens_seen": 200343552 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004744232698094283, + "loss": 1.4999, + "theoretical_loss": 4.35420246454978, + "tokens_seen": 200409088 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047441323971915753, + "loss": 1.6973, + "theoretical_loss": 4.354024383585481, + "tokens_seen": 200474624 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047440320962888666, + "loss": 1.5718, + "theoretical_loss": 4.3538463771213305, + "tokens_seen": 200540160 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004743931795386159, + "loss": 1.6029, + "theoretical_loss": 4.353668445101833, + "tokens_seen": 200605696 + }, + { + "epoch": 0.06, + "learning_rate": 0.000474383149448345, + "loss": 1.6924, + "theoretical_loss": 4.353490587471546, + "tokens_seen": 200671232 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047437311935807425, + "loss": 1.7458, + "theoretical_loss": 4.353312804175093, + "tokens_seen": 200736768 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047436308926780343, + "loss": 1.5892, + "theoretical_loss": 4.353135095157152, + "tokens_seen": 200802304 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004743530591775326, + "loss": 1.6771, + "theoretical_loss": 4.352957460362464, + "tokens_seen": 200867840 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004743430290872618, + "loss": 1.6793, + "theoretical_loss": 4.352779899735825, + "tokens_seen": 200933376 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047433299899699103, + "loss": 1.6769, + "theoretical_loss": 4.352602413222094, + "tokens_seen": 200998912 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047432296890672016, + "loss": 1.7077, + "theoretical_loss": 4.3524250007661855, + "tokens_seen": 201064448 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004743129388164494, + "loss": 1.6685, + "theoretical_loss": 4.352247662313075, + "tokens_seen": 201129984 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004743029087261785, + "loss": 1.8044, + "theoretical_loss": 4.352070397807797, + "tokens_seen": 201195520 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047429287863590776, + "loss": 1.6937, + "theoretical_loss": 4.3518932071954435, + "tokens_seen": 201261056 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047428284854563694, + "loss": 1.6411, + "theoretical_loss": 4.351716090421165, + "tokens_seen": 201326592 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004742728184553661, + "loss": 1.7, + "theoretical_loss": 4.351539047430173, + "tokens_seen": 201392128 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004742627883650953, + "loss": 1.7109, + "theoretical_loss": 4.351362078167735, + "tokens_seen": 201457664 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": 0.000982465106062591, + "objective/train/docs_used": 124253, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.054840326309204, + "objective/train/original_loss": 3.054840087890625, + "objective/train/theoretical_loss": 4.351185182579177, + "objective/train/tokens_used": 221983200, + "objective/train/value_avg": -0.00382232666015625, + "objective/train/value_loss": 2.966829197248444e-05, + "objective/train/value_max": -0.0002378225326538086, + "objective/train/value_min": -0.047149658203125, + "objective/train/value_reward_corr": 0.19604572787426394, + "objective/train/value_std": 0.002826690673828125, + "objective/train/weight_avg": 1.011224627494812, + "objective/train/weighted_lm_loss": 3.0991663932800293, + "objective/train/weights_max": 1.424397587776184, + "objective/train/weights_min": 0.33705228567123413, + "theoretical_loss": 4.351185182579177, + "tokens_seen": 201523200 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004742527582748245, + "loss": 1.6657, + "theoretical_loss": 4.351185182579177, + "tokens_seen": 201523200 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047424272818455366, + "loss": 1.7038, + "theoretical_loss": 4.3510083606098835, + "tokens_seen": 201588736 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004742326980942829, + "loss": 1.6602, + "theoretical_loss": 4.350831612205299, + "tokens_seen": 201654272 + }, + { + "epoch": 0.06, + "learning_rate": 0.000474222668004012, + "loss": 1.5552, + "theoretical_loss": 4.350654937310924, + "tokens_seen": 201719808 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047421263791374126, + "loss": 1.5393, + "theoretical_loss": 4.350478335872317, + "tokens_seen": 201785344 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004742026078234704, + "loss": 1.607, + "theoretical_loss": 4.350301807835097, + "tokens_seen": 201850880 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004741925777331996, + "loss": 1.6198, + "theoretical_loss": 4.3501253531449375, + "tokens_seen": 201916416 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004741825476429288, + "loss": 1.5947, + "theoretical_loss": 4.349948971747573, + "tokens_seen": 201981952 + }, + { + "epoch": 0.06, + "learning_rate": 0.000474172517552658, + "loss": 1.684, + "theoretical_loss": 4.3497726635887926, + "tokens_seen": 202047488 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047416248746238716, + "loss": 1.6243, + "theoretical_loss": 4.349596428614447, + "tokens_seen": 202113024 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004741524573721164, + "loss": 1.6434, + "theoretical_loss": 4.34942026677044, + "tokens_seen": 202178560 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004741424272818455, + "loss": 1.5938, + "theoretical_loss": 4.349244178002736, + "tokens_seen": 202244096 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047413239719157476, + "loss": 1.6183, + "theoretical_loss": 4.349068162257357, + "tokens_seen": 202309632 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004741223671013039, + "loss": 1.5608, + "theoretical_loss": 4.348892219480378, + "tokens_seen": 202375168 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004741123370110331, + "loss": 1.5725, + "theoretical_loss": 4.348716349617939, + "tokens_seen": 202440704 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004741023069207623, + "loss": 1.6219, + "theoretical_loss": 4.348540552616229, + "tokens_seen": 202506240 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004740922768304915, + "loss": 1.5252, + "theoretical_loss": 4.3483648284214995, + "tokens_seen": 202571776 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047408224674022067, + "loss": 1.6017, + "theoretical_loss": 4.348189176980057, + "tokens_seen": 202637312 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047407221664994985, + "loss": 1.5923, + "theoretical_loss": 4.348013598238266, + "tokens_seen": 202702848 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047406218655967903, + "loss": 1.6279, + "theoretical_loss": 4.347838092142546, + "tokens_seen": 202768384 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047405215646940827, + "loss": 1.5752, + "theoretical_loss": 4.347662658639376, + "tokens_seen": 202833920 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004740421263791374, + "loss": 1.7382, + "theoretical_loss": 4.347487297675288, + "tokens_seen": 202899456 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047403209628886663, + "loss": 1.7008, + "theoretical_loss": 4.347312009196874, + "tokens_seen": 202964992 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047402206619859575, + "loss": 1.717, + "theoretical_loss": 4.347136793150781, + "tokens_seen": 203030528 + }, + { + "epoch": 0.06, + "learning_rate": 0.000474012036108325, + "loss": 1.5946, + "theoretical_loss": 4.346961649483713, + "tokens_seen": 203096064 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": 0.00016722730651963502, + "objective/train/docs_used": 125838, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3937480449676514, + "objective/train/original_loss": 3.3937482833862305, + "objective/train/theoretical_loss": 4.3467865781424315, + "objective/train/tokens_used": 223621600, + "objective/train/value_avg": -0.0053253173828125, + "objective/train/value_loss": 5.692290505976416e-05, + "objective/train/value_max": -0.0002868175506591797, + "objective/train/value_min": -0.06512451171875, + "objective/train/value_reward_corr": 0.29651013961767064, + "objective/train/value_std": 0.00482940673828125, + "objective/train/weight_avg": 1.004485011100769, + "objective/train/weighted_lm_loss": 3.4077415466308594, + "objective/train/weights_max": 1.7859992980957031, + "objective/train/weights_min": 0.5764724612236023, + "theoretical_loss": 4.3467865781424315, + "tokens_seen": 203161600 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047400200601805417, + "loss": 1.5827, + "theoretical_loss": 4.3467865781424315, + "tokens_seen": 203161600 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047399197592778335, + "loss": 1.7114, + "theoretical_loss": 4.346611579073751, + "tokens_seen": 203227136 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047398194583751253, + "loss": 1.6833, + "theoretical_loss": 4.346436652224545, + "tokens_seen": 203292672 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047397191574724177, + "loss": 1.6559, + "theoretical_loss": 4.346261797541742, + "tokens_seen": 203358208 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004739618856569709, + "loss": 1.643, + "theoretical_loss": 4.346087014972328, + "tokens_seen": 203423744 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047395185556670013, + "loss": 1.6785, + "theoretical_loss": 4.345912304463344, + "tokens_seen": 203489280 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047394182547642926, + "loss": 1.6095, + "theoretical_loss": 4.345737665961886, + "tokens_seen": 203554816 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004739317953861585, + "loss": 1.546, + "theoretical_loss": 4.3455630994151075, + "tokens_seen": 203620352 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004739217652958877, + "loss": 1.6271, + "theoretical_loss": 4.345388604770218, + "tokens_seen": 203685888 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047391173520561686, + "loss": 1.622, + "theoretical_loss": 4.3452141819744785, + "tokens_seen": 203751424 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047390170511534604, + "loss": 1.59, + "theoretical_loss": 4.345039830975213, + "tokens_seen": 203816960 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004738916750250752, + "loss": 1.5966, + "theoretical_loss": 4.344865551719795, + "tokens_seen": 203882496 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004738816449348044, + "loss": 1.6492, + "theoretical_loss": 4.344691344155654, + "tokens_seen": 203948032 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047387161484453363, + "loss": 1.6943, + "theoretical_loss": 4.344517208230279, + "tokens_seen": 204013568 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047386158475426276, + "loss": 1.6456, + "theoretical_loss": 4.344343143891211, + "tokens_seen": 204079104 + }, + { + "epoch": 0.06, + "learning_rate": 0.000473851554663992, + "loss": 1.6087, + "theoretical_loss": 4.344169151086046, + "tokens_seen": 204144640 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004738415245737211, + "loss": 1.5916, + "theoretical_loss": 4.343995229762437, + "tokens_seen": 204210176 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047383149448345036, + "loss": 1.6572, + "theoretical_loss": 4.34382137986809, + "tokens_seen": 204275712 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047382146439317954, + "loss": 1.561, + "theoretical_loss": 4.343647601350769, + "tokens_seen": 204341248 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004738114343029087, + "loss": 1.5717, + "theoretical_loss": 4.343473894158289, + "tokens_seen": 204406784 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004738014042126379, + "loss": 1.6351, + "theoretical_loss": 4.343300258238523, + "tokens_seen": 204472320 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047379137412236714, + "loss": 1.5497, + "theoretical_loss": 4.343126693539398, + "tokens_seen": 204537856 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047378134403209626, + "loss": 1.7216, + "theoretical_loss": 4.342953200008895, + "tokens_seen": 204603392 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004737713139418255, + "loss": 1.5459, + "theoretical_loss": 4.342779777595051, + "tokens_seen": 204668928 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004737612838515546, + "loss": 1.5375, + "theoretical_loss": 4.342606426245956, + "tokens_seen": 204734464 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": 0.00046731301699765027, + "objective/train/docs_used": 126453, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.893792152404785, + "objective/train/original_loss": 2.893791913986206, + "objective/train/theoretical_loss": 4.342433145909755, + "objective/train/tokens_used": 225260000, + "objective/train/value_avg": -0.004932403564453125, + "objective/train/value_loss": 4.786020508618094e-05, + "objective/train/value_max": -0.00033283233642578125, + "objective/train/value_min": -0.1485595703125, + "objective/train/value_reward_corr": 0.27032416425755085, + "objective/train/value_std": 0.0046539306640625, + "objective/train/weight_avg": 1.007056713104248, + "objective/train/weighted_lm_loss": 2.9221627712249756, + "objective/train/weights_max": 4.184020042419434, + "objective/train/weights_min": 0.29072141647338867, + "theoretical_loss": 4.342433145909755, + "tokens_seen": 204800000 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047375125376128386, + "loss": 1.5775, + "theoretical_loss": 4.342433145909755, + "tokens_seen": 204800000 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047374122367101304, + "loss": 1.5227, + "theoretical_loss": 4.3422599365346475, + "tokens_seen": 204865536 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004737311935807422, + "loss": 1.5797, + "theoretical_loss": 4.342086798068888, + "tokens_seen": 204931072 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047372116349047146, + "loss": 1.7435, + "theoretical_loss": 4.341913730460785, + "tokens_seen": 204996608 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004737111334002006, + "loss": 1.6716, + "theoretical_loss": 4.341740733658701, + "tokens_seen": 205062144 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004737011033099298, + "loss": 1.6307, + "theoretical_loss": 4.34156780761105, + "tokens_seen": 205127680 + }, + { + "epoch": 0.06, + "learning_rate": 0.000473691073219659, + "loss": 1.6016, + "theoretical_loss": 4.341394952266306, + "tokens_seen": 205193216 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004736810431293882, + "loss": 1.5591, + "theoretical_loss": 4.341222167572992, + "tokens_seen": 205258752 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047367101303911736, + "loss": 1.6779, + "theoretical_loss": 4.341049453479686, + "tokens_seen": 205324288 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004736609829488466, + "loss": 1.6818, + "theoretical_loss": 4.340876809935021, + "tokens_seen": 205389824 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004736509528585757, + "loss": 1.5948, + "theoretical_loss": 4.3407042368876825, + "tokens_seen": 205455360 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047364092276830496, + "loss": 1.6258, + "theoretical_loss": 4.34053173428641, + "tokens_seen": 205520896 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004736308926780341, + "loss": 1.546, + "theoretical_loss": 4.340359302079998, + "tokens_seen": 205586432 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004736208625877633, + "loss": 1.7504, + "theoretical_loss": 4.340186940217292, + "tokens_seen": 205651968 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004736108324974925, + "loss": 1.6844, + "theoretical_loss": 4.340014648647193, + "tokens_seen": 205717504 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004736008024072217, + "loss": 1.4681, + "theoretical_loss": 4.339842427318654, + "tokens_seen": 205783040 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047359077231695087, + "loss": 1.6618, + "theoretical_loss": 4.339670276180684, + "tokens_seen": 205848576 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047358074222668005, + "loss": 1.6861, + "theoretical_loss": 4.3394981951823395, + "tokens_seen": 205914112 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047357071213640923, + "loss": 1.705, + "theoretical_loss": 4.339326184272737, + "tokens_seen": 205979648 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047356068204613847, + "loss": 1.5448, + "theoretical_loss": 4.339154243401042, + "tokens_seen": 206045184 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004735506519558676, + "loss": 1.6104, + "theoretical_loss": 4.338982372516473, + "tokens_seen": 206110720 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047354062186559683, + "loss": 1.6519, + "theoretical_loss": 4.338810571568304, + "tokens_seen": 206176256 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047353059177532595, + "loss": 1.5232, + "theoretical_loss": 4.33863884050586, + "tokens_seen": 206241792 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004735205616850552, + "loss": 1.6235, + "theoretical_loss": 4.338467179278519, + "tokens_seen": 206307328 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047351053159478437, + "loss": 1.4621, + "theoretical_loss": 4.338295587835711, + "tokens_seen": 206372864 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.00047868501860648394, + "objective/train/docs_used": 127848, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.254330635070801, + "objective/train/original_loss": 3.25433087348938, + "objective/train/theoretical_loss": 4.33812406612692, + "objective/train/tokens_used": 226898400, + "objective/train/value_avg": -0.005062103271484375, + "objective/train/value_loss": 7.8853867307771e-05, + "objective/train/value_max": -0.0003528594970703125, + "objective/train/value_min": -0.1236572265625, + "objective/train/value_reward_corr": 0.21219871851866182, + "objective/train/value_std": 0.00406646728515625, + "objective/train/weight_avg": 0.9986884593963623, + "objective/train/weighted_lm_loss": 3.2525360584259033, + "objective/train/weights_max": 2.5045297145843506, + "objective/train/weights_min": 0.08479506522417068, + "theoretical_loss": 4.33812406612692, + "tokens_seen": 206438400 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047350050150451355, + "loss": 1.6121, + "theoretical_loss": 4.33812406612692, + "tokens_seen": 206438400 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047349047141424273, + "loss": 1.6075, + "theoretical_loss": 4.337952614101683, + "tokens_seen": 206503936 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047348044132397197, + "loss": 1.7777, + "theoretical_loss": 4.337781231709587, + "tokens_seen": 206569472 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004734704112337011, + "loss": 1.59, + "theoretical_loss": 4.337609918900275, + "tokens_seen": 206635008 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047346038114343033, + "loss": 1.6468, + "theoretical_loss": 4.337438675623439, + "tokens_seen": 206700544 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047345035105315946, + "loss": 1.4651, + "theoretical_loss": 4.3372675018288245, + "tokens_seen": 206766080 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004734403209628887, + "loss": 1.6023, + "theoretical_loss": 4.337096397466231, + "tokens_seen": 206831616 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004734302908726179, + "loss": 1.6481, + "theoretical_loss": 4.336925362485507, + "tokens_seen": 206897152 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047342026078234706, + "loss": 1.6736, + "theoretical_loss": 4.3367543968365565, + "tokens_seen": 206962688 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047341023069207624, + "loss": 1.6178, + "theoretical_loss": 4.336583500469333, + "tokens_seen": 207028224 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004734002006018054, + "loss": 1.677, + "theoretical_loss": 4.336412673333841, + "tokens_seen": 207093760 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004733901705115346, + "loss": 1.6485, + "theoretical_loss": 4.336241915380142, + "tokens_seen": 207159296 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047338014042126383, + "loss": 1.6312, + "theoretical_loss": 4.336071226558343, + "tokens_seen": 207224832 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047337011033099296, + "loss": 1.4822, + "theoretical_loss": 4.3359006068186075, + "tokens_seen": 207290368 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004733600802407222, + "loss": 1.6881, + "theoretical_loss": 4.335730056111148, + "tokens_seen": 207355904 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004733500501504513, + "loss": 1.6991, + "theoretical_loss": 4.335559574386228, + "tokens_seen": 207421440 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047334002006018056, + "loss": 1.6626, + "theoretical_loss": 4.335389161594167, + "tokens_seen": 207486976 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047332998996990974, + "loss": 1.484, + "theoretical_loss": 4.33521881768533, + "tokens_seen": 207552512 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004733199598796389, + "loss": 1.5587, + "theoretical_loss": 4.3350485426101395, + "tokens_seen": 207618048 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004733099297893681, + "loss": 1.6636, + "theoretical_loss": 4.334878336319063, + "tokens_seen": 207683584 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047329989969909734, + "loss": 1.6303, + "theoretical_loss": 4.334708198762624, + "tokens_seen": 207749120 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047328986960882646, + "loss": 1.5472, + "theoretical_loss": 4.334538129891396, + "tokens_seen": 207814656 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004732798395185557, + "loss": 1.5835, + "theoretical_loss": 4.334368129656002, + "tokens_seen": 207880192 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004732698094282848, + "loss": 1.671, + "theoretical_loss": 4.334198198007117, + "tokens_seen": 207945728 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047325977933801406, + "loss": 1.5103, + "theoretical_loss": 4.33402833489547, + "tokens_seen": 208011264 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": 9.124818461714312e-05, + "objective/train/docs_used": 128338, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3726956844329834, + "objective/train/original_loss": 3.3726956844329834, + "objective/train/theoretical_loss": 4.333858540271834, + "objective/train/tokens_used": 228536800, + "objective/train/value_avg": -0.004375457763671875, + "objective/train/value_loss": 5.316070019034669e-05, + "objective/train/value_max": -0.0003612041473388672, + "objective/train/value_min": -0.054290771484375, + "objective/train/value_reward_corr": 0.19711641388000053, + "objective/train/value_std": 0.0037097930908203125, + "objective/train/weight_avg": 1.0034427642822266, + "objective/train/weighted_lm_loss": 3.379629373550415, + "objective/train/weights_max": 1.721003770828247, + "objective/train/weights_min": 0.44609570503234863, + "theoretical_loss": 4.333858540271834, + "tokens_seen": 208076800 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047324974924774324, + "loss": 1.6756, + "theoretical_loss": 4.333858540271834, + "tokens_seen": 208076800 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004732397191574724, + "loss": 1.5739, + "theoretical_loss": 4.33368881408704, + "tokens_seen": 208142336 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004732296890672016, + "loss": 1.6993, + "theoretical_loss": 4.333519156291965, + "tokens_seen": 208207872 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004732196589769308, + "loss": 1.6698, + "theoretical_loss": 4.333349566837541, + "tokens_seen": 208273408 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047320962888665997, + "loss": 1.6156, + "theoretical_loss": 4.3331800456747445, + "tokens_seen": 208338944 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004731995987963892, + "loss": 1.5583, + "theoretical_loss": 4.333010592754609, + "tokens_seen": 208404480 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047318956870611833, + "loss": 1.5281, + "theoretical_loss": 4.3328412080282135, + "tokens_seen": 208470016 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047317953861584757, + "loss": 1.6042, + "theoretical_loss": 4.332671891446692, + "tokens_seen": 208535552 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047316950852557675, + "loss": 1.5669, + "theoretical_loss": 4.332502642961224, + "tokens_seen": 208601088 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047315947843530593, + "loss": 1.6788, + "theoretical_loss": 4.332333462523044, + "tokens_seen": 208666624 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004731494483450351, + "loss": 1.5474, + "theoretical_loss": 4.332164350083432, + "tokens_seen": 208732160 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004731394182547643, + "loss": 1.4498, + "theoretical_loss": 4.331995305593723, + "tokens_seen": 208797696 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047312938816449347, + "loss": 1.7022, + "theoretical_loss": 4.331826329005298, + "tokens_seen": 208863232 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004731193580742227, + "loss": 1.6659, + "theoretical_loss": 4.3316574202695906, + "tokens_seen": 208928768 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047310932798395183, + "loss": 1.573, + "theoretical_loss": 4.3314885793380835, + "tokens_seen": 208994304 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047309929789368107, + "loss": 1.6564, + "theoretical_loss": 4.331319806162309, + "tokens_seen": 209059840 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004730892678034102, + "loss": 1.685, + "theoretical_loss": 4.33115110069385, + "tokens_seen": 209125376 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047307923771313943, + "loss": 1.6745, + "theoretical_loss": 4.330982462884338, + "tokens_seen": 209190912 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004730692076228686, + "loss": 1.6447, + "theoretical_loss": 4.3308138926854545, + "tokens_seen": 209256448 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004730591775325978, + "loss": 1.6698, + "theoretical_loss": 4.330645390048932, + "tokens_seen": 209321984 + }, + { + "epoch": 0.06, + "learning_rate": 0.000473049147442327, + "loss": 1.7173, + "theoretical_loss": 4.330476954926552, + "tokens_seen": 209387520 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047303911735205615, + "loss": 1.7023, + "theoretical_loss": 4.330308587270143, + "tokens_seen": 209453056 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047302908726178534, + "loss": 1.5309, + "theoretical_loss": 4.330140287031587, + "tokens_seen": 209518592 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047301905717151457, + "loss": 1.6606, + "theoretical_loss": 4.329972054162813, + "tokens_seen": 209584128 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004730090270812437, + "loss": 1.5802, + "theoretical_loss": 4.329803888615798, + "tokens_seen": 209649664 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.0018911852966994047, + "objective/train/docs_used": 129483, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.7283051013946533, + "objective/train/original_loss": 3.7283051013946533, + "objective/train/theoretical_loss": 4.3296357903425715, + "objective/train/tokens_used": 230175200, + "objective/train/value_avg": -0.006046295166015625, + "objective/train/value_loss": 9.95642549241893e-05, + "objective/train/value_max": -0.0003380775451660156, + "objective/train/value_min": -0.07891845703125, + "objective/train/value_reward_corr": 0.24465354227718106, + "objective/train/value_std": 0.004833221435546875, + "objective/train/weight_avg": 0.9857138395309448, + "objective/train/weighted_lm_loss": 3.674426555633545, + "objective/train/weights_max": 1.8038166761398315, + "objective/train/weights_min": 0.3818269968032837, + "theoretical_loss": 4.3296357903425715, + "tokens_seen": 209715200 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047299899699097293, + "loss": 1.7458, + "theoretical_loss": 4.3296357903425715, + "tokens_seen": 209715200 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004729889669007021, + "loss": 1.5737, + "theoretical_loss": 4.329467759295211, + "tokens_seen": 209780736 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004729789368104313, + "loss": 1.6351, + "theoretical_loss": 4.32929979542584, + "tokens_seen": 209846272 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047296890672016053, + "loss": 1.6956, + "theoretical_loss": 4.3291318986866365, + "tokens_seen": 209911808 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047295887662988966, + "loss": 1.6612, + "theoretical_loss": 4.328964069029822, + "tokens_seen": 209977344 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004729488465396189, + "loss": 1.7291, + "theoretical_loss": 4.328796306407672, + "tokens_seen": 210042880 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004729388164493481, + "loss": 1.4823, + "theoretical_loss": 4.3286286107725065, + "tokens_seen": 210108416 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047292878635907726, + "loss": 1.5933, + "theoretical_loss": 4.328460982076697, + "tokens_seen": 210173952 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047291875626880644, + "loss": 1.5588, + "theoretical_loss": 4.328293420272663, + "tokens_seen": 210239488 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004729087261785356, + "loss": 1.6097, + "theoretical_loss": 4.328125925312872, + "tokens_seen": 210305024 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004728986960882648, + "loss": 1.6483, + "theoretical_loss": 4.32795849714984, + "tokens_seen": 210370560 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047288866599799403, + "loss": 1.5991, + "theoretical_loss": 4.327791135736133, + "tokens_seen": 210436096 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047287863590772316, + "loss": 1.613, + "theoretical_loss": 4.327623841024364, + "tokens_seen": 210501632 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004728686058174524, + "loss": 1.5804, + "theoretical_loss": 4.327456612967195, + "tokens_seen": 210567168 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004728585757271815, + "loss": 1.6271, + "theoretical_loss": 4.327289451517336, + "tokens_seen": 210632704 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047284854563691076, + "loss": 1.6133, + "theoretical_loss": 4.327122356627546, + "tokens_seen": 210698240 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047283851554663994, + "loss": 1.6725, + "theoretical_loss": 4.326955328250631, + "tokens_seen": 210763776 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004728284854563691, + "loss": 1.6896, + "theoretical_loss": 4.326788366339445, + "tokens_seen": 210829312 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004728184553660983, + "loss": 1.6901, + "theoretical_loss": 4.326621470846892, + "tokens_seen": 210894848 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047280842527582754, + "loss": 1.714, + "theoretical_loss": 4.326454641725923, + "tokens_seen": 210960384 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047279839518555666, + "loss": 1.7002, + "theoretical_loss": 4.326287878929536, + "tokens_seen": 211025920 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004727883650952859, + "loss": 1.478, + "theoretical_loss": 4.326121182410777, + "tokens_seen": 211091456 + }, + { + "epoch": 0.06, + "learning_rate": 0.000472778335005015, + "loss": 1.6121, + "theoretical_loss": 4.325954552122742, + "tokens_seen": 211156992 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047276830491474426, + "loss": 1.5905, + "theoretical_loss": 4.325787988018572, + "tokens_seen": 211222528 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047275827482447344, + "loss": 1.6698, + "theoretical_loss": 4.325621490051457, + "tokens_seen": 211288064 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": -0.0004364687774796039, + "objective/train/docs_used": 130263, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.608811378479004, + "objective/train/original_loss": 3.608811140060425, + "objective/train/theoretical_loss": 4.325455058174634, + "objective/train/tokens_used": 231813600, + "objective/train/value_avg": -0.005413055419921875, + "objective/train/value_loss": 6.047657734598033e-05, + "objective/train/value_max": -0.0003936290740966797, + "objective/train/value_min": -0.178466796875, + "objective/train/value_reward_corr": 0.18653402548242443, + "objective/train/value_std": 0.004764556884765625, + "objective/train/weight_avg": 0.998710036277771, + "objective/train/weighted_lm_loss": 3.608120918273926, + "objective/train/weights_max": 5.957601547241211, + "objective/train/weights_min": 0.593762218952179, + "theoretical_loss": 4.325455058174634, + "tokens_seen": 211353600 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004727482447342026, + "loss": 1.7753, + "theoretical_loss": 4.325455058174634, + "tokens_seen": 211353600 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004727382146439318, + "loss": 1.7257, + "theoretical_loss": 4.325288692341388, + "tokens_seen": 211419136 + }, + { + "epoch": 0.06, + "learning_rate": 0.000472728184553661, + "loss": 1.6682, + "theoretical_loss": 4.325122392505054, + "tokens_seen": 211484672 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047271815446339017, + "loss": 1.7212, + "theoretical_loss": 4.324956158619008, + "tokens_seen": 211550208 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004727081243731194, + "loss": 1.6877, + "theoretical_loss": 4.324789990636679, + "tokens_seen": 211615744 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047269809428284853, + "loss": 1.6533, + "theoretical_loss": 4.3246238885115424, + "tokens_seen": 211681280 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047268806419257777, + "loss": 1.6174, + "theoretical_loss": 4.324457852197119, + "tokens_seen": 211746816 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047267803410230695, + "loss": 1.6717, + "theoretical_loss": 4.324291881646978, + "tokens_seen": 211812352 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047266800401203613, + "loss": 1.6538, + "theoretical_loss": 4.324125976814734, + "tokens_seen": 211877888 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004726579739217653, + "loss": 1.5311, + "theoretical_loss": 4.323960137654053, + "tokens_seen": 211943424 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004726479438314945, + "loss": 1.6374, + "theoretical_loss": 4.323794364118644, + "tokens_seen": 212008960 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047263791374122367, + "loss": 1.6254, + "theoretical_loss": 4.3236286561622626, + "tokens_seen": 212074496 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004726278836509529, + "loss": 1.5731, + "theoretical_loss": 4.323463013738714, + "tokens_seen": 212140032 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047261785356068203, + "loss": 1.5749, + "theoretical_loss": 4.323297436801848, + "tokens_seen": 212205568 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047260782347041127, + "loss": 1.5484, + "theoretical_loss": 4.323131925305563, + "tokens_seen": 212271104 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004725977933801404, + "loss": 1.7545, + "theoretical_loss": 4.322966479203803, + "tokens_seen": 212336640 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047258776328986963, + "loss": 1.6119, + "theoretical_loss": 4.322801098450558, + "tokens_seen": 212402176 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004725777331995988, + "loss": 1.7016, + "theoretical_loss": 4.322635782999866, + "tokens_seen": 212467712 + }, + { + "epoch": 0.06, + "learning_rate": 0.000472567703109328, + "loss": 1.7635, + "theoretical_loss": 4.32247053280581, + "tokens_seen": 212533248 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004725576730190572, + "loss": 1.7294, + "theoretical_loss": 4.322305347822521, + "tokens_seen": 212598784 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047254764292878636, + "loss": 1.5817, + "theoretical_loss": 4.322140228004175, + "tokens_seen": 212664320 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047253761283851554, + "loss": 1.5734, + "theoretical_loss": 4.321975173304995, + "tokens_seen": 212729856 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047252758274824477, + "loss": 1.6772, + "theoretical_loss": 4.321810183679251, + "tokens_seen": 212795392 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004725175526579739, + "loss": 1.6449, + "theoretical_loss": 4.321645259081256, + "tokens_seen": 212860928 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047250752256770313, + "loss": 1.7207, + "theoretical_loss": 4.321480399465375, + "tokens_seen": 212926464 + }, + { + "epoch": 0.06, + "objective/train/advantage_avg": 7.575751806143671e-05, + "objective/train/docs_used": 131630, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.254868268966675, + "objective/train/original_loss": 2.254868268966675, + "objective/train/theoretical_loss": 4.321315604786012, + "objective/train/tokens_used": 233452000, + "objective/train/value_avg": -0.0043487548828125, + "objective/train/value_loss": 7.144419214455411e-05, + "objective/train/value_max": -0.00033020973205566406, + "objective/train/value_min": -0.06622314453125, + "objective/train/value_reward_corr": 0.11200089434056885, + "objective/train/value_std": 0.0031299591064453125, + "objective/train/weight_avg": 1.0039212703704834, + "objective/train/weighted_lm_loss": 2.271632194519043, + "objective/train/weights_max": 1.490443229675293, + "objective/train/weights_min": 0.29985612630844116, + "theoretical_loss": 4.321315604786012, + "tokens_seen": 212992000 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004724974924774323, + "loss": 1.4675, + "theoretical_loss": 4.321315604786012, + "tokens_seen": 212992000 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004724874623871615, + "loss": 1.5661, + "theoretical_loss": 4.321150874997622, + "tokens_seen": 213057536 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004724774322968907, + "loss": 1.5972, + "theoretical_loss": 4.320986210054705, + "tokens_seen": 213123072 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047246740220661986, + "loss": 1.6151, + "theoretical_loss": 4.320821609911805, + "tokens_seen": 213188608 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047245737211634904, + "loss": 1.5797, + "theoretical_loss": 4.320657074523515, + "tokens_seen": 213254144 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004724473420260783, + "loss": 1.5829, + "theoretical_loss": 4.32049260384447, + "tokens_seen": 213319680 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004724373119358074, + "loss": 1.3988, + "theoretical_loss": 4.320328197829354, + "tokens_seen": 213385216 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047242728184553664, + "loss": 1.6568, + "theoretical_loss": 4.3201638564328935, + "tokens_seen": 213450752 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047241725175526576, + "loss": 1.6262, + "theoretical_loss": 4.319999579609864, + "tokens_seen": 213516288 + }, + { + "epoch": 0.06, + "learning_rate": 0.000472407221664995, + "loss": 1.5327, + "theoretical_loss": 4.319835367315085, + "tokens_seen": 213581824 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004723971915747242, + "loss": 1.6669, + "theoretical_loss": 4.31967121950342, + "tokens_seen": 213647360 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047238716148445336, + "loss": 1.5998, + "theoretical_loss": 4.31950713612978, + "tokens_seen": 213712896 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047237713139418254, + "loss": 1.5711, + "theoretical_loss": 4.3193431171491214, + "tokens_seen": 213778432 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004723671013039117, + "loss": 1.5891, + "theoretical_loss": 4.319179162516443, + "tokens_seen": 213843968 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004723570712136409, + "loss": 1.7, + "theoretical_loss": 4.3190152721867925, + "tokens_seen": 213909504 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047234704112337014, + "loss": 1.6379, + "theoretical_loss": 4.318851446115261, + "tokens_seen": 213975040 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047233701103309927, + "loss": 1.5685, + "theoretical_loss": 4.318687684256984, + "tokens_seen": 214040576 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004723269809428285, + "loss": 1.6173, + "theoretical_loss": 4.318523986567144, + "tokens_seen": 214106112 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004723169508525577, + "loss": 1.6555, + "theoretical_loss": 4.318360353000967, + "tokens_seen": 214171648 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047230692076228686, + "loss": 1.5036, + "theoretical_loss": 4.318196783513724, + "tokens_seen": 214237184 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047229689067201605, + "loss": 1.628, + "theoretical_loss": 4.318033278060732, + "tokens_seen": 214302720 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004722868605817452, + "loss": 1.646, + "theoretical_loss": 4.31786983659735, + "tokens_seen": 214368256 + }, + { + "epoch": 0.06, + "learning_rate": 0.0004722768304914744, + "loss": 1.6748, + "theoretical_loss": 4.317706459078985, + "tokens_seen": 214433792 + }, + { + "epoch": 0.06, + "learning_rate": 0.00047226680040120364, + "loss": 1.6155, + "theoretical_loss": 4.317543145461089, + "tokens_seen": 214499328 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047225677031093277, + "loss": 1.6048, + "theoretical_loss": 4.317379895699155, + "tokens_seen": 214564864 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.00038902819505892694, + "objective/train/docs_used": 132228, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3919694423675537, + "objective/train/original_loss": 3.391969919204712, + "objective/train/theoretical_loss": 4.317216709748722, + "objective/train/tokens_used": 235090400, + "objective/train/value_avg": -0.004779815673828125, + "objective/train/value_loss": 7.788165385136381e-05, + "objective/train/value_max": -0.000263214111328125, + "objective/train/value_min": -0.2098388671875, + "objective/train/value_reward_corr": 0.26989821901529215, + "objective/train/value_std": 0.004718780517578125, + "objective/train/weight_avg": 0.9999018907546997, + "objective/train/weighted_lm_loss": 3.385392904281616, + "objective/train/weights_max": 7.920759677886963, + "objective/train/weights_min": 0.14297199249267578, + "theoretical_loss": 4.317216709748722, + "tokens_seen": 214630400 + }, + { + "epoch": 0.07, + "learning_rate": 0.000472246740220662, + "loss": 1.5977, + "theoretical_loss": 4.317216709748722, + "tokens_seen": 214630400 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047223671013039113, + "loss": 1.4625, + "theoretical_loss": 4.317053587565377, + "tokens_seen": 214695936 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047222668004012037, + "loss": 1.5508, + "theoretical_loss": 4.3168905291047475, + "tokens_seen": 214761472 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004722166499498496, + "loss": 1.6684, + "theoretical_loss": 4.316727534322505, + "tokens_seen": 214827008 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047220661985957873, + "loss": 1.5415, + "theoretical_loss": 4.316564603174369, + "tokens_seen": 214892544 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047219658976930797, + "loss": 1.6202, + "theoretical_loss": 4.3164017356160995, + "tokens_seen": 214958080 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047218655967903715, + "loss": 1.7159, + "theoretical_loss": 4.316238931603504, + "tokens_seen": 215023616 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047217652958876633, + "loss": 1.6162, + "theoretical_loss": 4.316076191092431, + "tokens_seen": 215089152 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004721664994984955, + "loss": 1.6074, + "theoretical_loss": 4.315913514038776, + "tokens_seen": 215154688 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004721564694082247, + "loss": 1.6105, + "theoretical_loss": 4.315750900398476, + "tokens_seen": 215220224 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047214643931795387, + "loss": 1.5926, + "theoretical_loss": 4.3155883501275145, + "tokens_seen": 215285760 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004721364092276831, + "loss": 1.6361, + "theoretical_loss": 4.315425863181917, + "tokens_seen": 215351296 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047212637913741223, + "loss": 1.6152, + "theoretical_loss": 4.315263439517754, + "tokens_seen": 215416832 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047211634904714147, + "loss": 1.6872, + "theoretical_loss": 4.315101079091139, + "tokens_seen": 215482368 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004721063189568706, + "loss": 1.6971, + "theoretical_loss": 4.314938781858229, + "tokens_seen": 215547904 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047209628886659983, + "loss": 1.6618, + "theoretical_loss": 4.314776547775228, + "tokens_seen": 215613440 + }, + { + "epoch": 0.07, + "learning_rate": 0.000472086258776329, + "loss": 1.6269, + "theoretical_loss": 4.314614376798378, + "tokens_seen": 215678976 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004720762286860582, + "loss": 1.4727, + "theoretical_loss": 4.314452268883969, + "tokens_seen": 215744512 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004720661985957874, + "loss": 1.643, + "theoretical_loss": 4.314290223988333, + "tokens_seen": 215810048 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047205616850551656, + "loss": 1.6276, + "theoretical_loss": 4.314128242067846, + "tokens_seen": 215875584 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047204613841524574, + "loss": 1.6252, + "theoretical_loss": 4.313966323078927, + "tokens_seen": 215941120 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047203610832497497, + "loss": 1.7542, + "theoretical_loss": 4.313804466978039, + "tokens_seen": 216006656 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004720260782347041, + "loss": 1.6279, + "theoretical_loss": 4.313642673721686, + "tokens_seen": 216072192 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047201604814443333, + "loss": 1.5142, + "theoretical_loss": 4.313480943266418, + "tokens_seen": 216137728 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004720060180541625, + "loss": 1.6694, + "theoretical_loss": 4.313319275568828, + "tokens_seen": 216203264 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.0005424710107035935, + "objective/train/docs_used": 133529, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.7661995887756348, + "objective/train/original_loss": 3.766199827194214, + "objective/train/theoretical_loss": 4.313157670585552, + "objective/train/tokens_used": 236728800, + "objective/train/value_avg": -0.005023956298828125, + "objective/train/value_loss": 8.630299271317199e-05, + "objective/train/value_max": -0.0003101825714111328, + "objective/train/value_min": -0.089111328125, + "objective/train/value_reward_corr": 0.4217709677016321, + "objective/train/value_std": 0.005039215087890625, + "objective/train/weight_avg": 0.9985507130622864, + "objective/train/weighted_lm_loss": 3.7782645225524902, + "objective/train/weights_max": 1.9959365129470825, + "objective/train/weights_min": 0.3308405876159668, + "theoretical_loss": 4.313157670585552, + "tokens_seen": 216268800 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004719959879638917, + "loss": 1.7513, + "theoretical_loss": 4.313157670585552, + "tokens_seen": 216268800 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004719859578736209, + "loss": 1.6011, + "theoretical_loss": 4.312996128273267, + "tokens_seen": 216334336 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047197592778335006, + "loss": 1.661, + "theoretical_loss": 4.312834648588694, + "tokens_seen": 216399872 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047196589769307924, + "loss": 1.475, + "theoretical_loss": 4.312673231488599, + "tokens_seen": 216465408 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004719558676028085, + "loss": 1.5824, + "theoretical_loss": 4.312511876929789, + "tokens_seen": 216530944 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004719458375125376, + "loss": 1.5382, + "theoretical_loss": 4.312350584869114, + "tokens_seen": 216596480 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047193580742226684, + "loss": 1.604, + "theoretical_loss": 4.3121893552634685, + "tokens_seen": 216662016 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047192577733199596, + "loss": 1.6244, + "theoretical_loss": 4.312028188069787, + "tokens_seen": 216727552 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004719157472417252, + "loss": 1.7238, + "theoretical_loss": 4.311867083245048, + "tokens_seen": 216793088 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004719057171514544, + "loss": 1.629, + "theoretical_loss": 4.311706040746273, + "tokens_seen": 216858624 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047189568706118356, + "loss": 1.4775, + "theoretical_loss": 4.311545060530526, + "tokens_seen": 216924160 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047188565697091274, + "loss": 1.5343, + "theoretical_loss": 4.311384142554914, + "tokens_seen": 216989696 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004718756268806419, + "loss": 1.7139, + "theoretical_loss": 4.311223286776586, + "tokens_seen": 217055232 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004718655967903711, + "loss": 1.5299, + "theoretical_loss": 4.311062493152731, + "tokens_seen": 217120768 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047185556670010034, + "loss": 1.6675, + "theoretical_loss": 4.310901761640586, + "tokens_seen": 217186304 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047184553660982947, + "loss": 1.6422, + "theoretical_loss": 4.310741092197424, + "tokens_seen": 217251840 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004718355065195587, + "loss": 1.6546, + "theoretical_loss": 4.310580484780565, + "tokens_seen": 217317376 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004718254764292879, + "loss": 1.5708, + "theoretical_loss": 4.310419939347369, + "tokens_seen": 217382912 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047181544633901706, + "loss": 1.6765, + "theoretical_loss": 4.310259455855238, + "tokens_seen": 217448448 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047180541624874625, + "loss": 1.4996, + "theoretical_loss": 4.3100990342616186, + "tokens_seen": 217513984 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004717953861584754, + "loss": 1.6936, + "theoretical_loss": 4.309938674523996, + "tokens_seen": 217579520 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004717853560682046, + "loss": 1.5739, + "theoretical_loss": 4.3097783765998985, + "tokens_seen": 217645056 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047177532597793384, + "loss": 1.5969, + "theoretical_loss": 4.309618140446899, + "tokens_seen": 217710592 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047176529588766297, + "loss": 1.7179, + "theoretical_loss": 4.309457966022608, + "tokens_seen": 217776128 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004717552657973922, + "loss": 1.5449, + "theoretical_loss": 4.30929785328468, + "tokens_seen": 217841664 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.0003172562865074724, + "objective/train/docs_used": 134228, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.493424654006958, + "objective/train/original_loss": 3.493425130844116, + "objective/train/theoretical_loss": 4.309137802190812, + "objective/train/tokens_used": 238367200, + "objective/train/value_avg": -0.004245758056640625, + "objective/train/value_loss": 7.838869350962341e-05, + "objective/train/value_max": -0.00023233890533447266, + "objective/train/value_min": -0.1041259765625, + "objective/train/value_reward_corr": 0.19572999208491607, + "objective/train/value_std": 0.003841400146484375, + "objective/train/weight_avg": 1.000328540802002, + "objective/train/weighted_lm_loss": 3.5164566040039062, + "objective/train/weights_max": 2.3146064281463623, + "objective/train/weights_min": 0.24589259922504425, + "theoretical_loss": 4.309137802190812, + "tokens_seen": 217907200 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047174523570712133, + "loss": 1.5723, + "theoretical_loss": 4.309137802190812, + "tokens_seen": 217907200 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047173520561685057, + "loss": 1.6348, + "theoretical_loss": 4.308977812698743, + "tokens_seen": 217972736 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047172517552657975, + "loss": 1.6762, + "theoretical_loss": 4.30881788476625, + "tokens_seen": 218038272 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047171514543630893, + "loss": 1.5431, + "theoretical_loss": 4.3086580183511565, + "tokens_seen": 218103808 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004717051153460381, + "loss": 1.6782, + "theoretical_loss": 4.308498213411323, + "tokens_seen": 218169344 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047169508525576735, + "loss": 1.5805, + "theoretical_loss": 4.3083384699046565, + "tokens_seen": 218234880 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004716850551654965, + "loss": 1.6244, + "theoretical_loss": 4.308178787789101, + "tokens_seen": 218300416 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004716750250752257, + "loss": 1.5677, + "theoretical_loss": 4.3080191670226435, + "tokens_seen": 218365952 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047166499498495484, + "loss": 1.7359, + "theoretical_loss": 4.307859607563313, + "tokens_seen": 218431488 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047165496489468407, + "loss": 1.5463, + "theoretical_loss": 4.30770010936918, + "tokens_seen": 218497024 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047164493480441325, + "loss": 1.5974, + "theoretical_loss": 4.307540672398354, + "tokens_seen": 218562560 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047163490471414243, + "loss": 1.6791, + "theoretical_loss": 4.307381296608988, + "tokens_seen": 218628096 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004716248746238716, + "loss": 1.5557, + "theoretical_loss": 4.307221981959276, + "tokens_seen": 218693632 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004716148445336008, + "loss": 1.5826, + "theoretical_loss": 4.307062728407451, + "tokens_seen": 218759168 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047160481444333, + "loss": 1.6831, + "theoretical_loss": 4.30690353591179, + "tokens_seen": 218824704 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004715947843530592, + "loss": 1.5606, + "theoretical_loss": 4.306744404430608, + "tokens_seen": 218890240 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047158475426278834, + "loss": 1.6234, + "theoretical_loss": 4.306585333922263, + "tokens_seen": 218955776 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004715747241725176, + "loss": 1.641, + "theoretical_loss": 4.3064263243451535, + "tokens_seen": 219021312 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004715646940822467, + "loss": 1.6406, + "theoretical_loss": 4.306267375657719, + "tokens_seen": 219086848 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047155466399197594, + "loss": 1.7053, + "theoretical_loss": 4.306108487818438, + "tokens_seen": 219152384 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004715446339017051, + "loss": 1.5369, + "theoretical_loss": 4.305949660785832, + "tokens_seen": 219217920 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004715346038114343, + "loss": 1.6612, + "theoretical_loss": 4.305790894518463, + "tokens_seen": 219283456 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004715245737211635, + "loss": 1.541, + "theoretical_loss": 4.30563218897493, + "tokens_seen": 219348992 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004715145436308927, + "loss": 1.62, + "theoretical_loss": 4.3054735441138785, + "tokens_seen": 219414528 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047150451354062184, + "loss": 1.6528, + "theoretical_loss": 4.305314959893989, + "tokens_seen": 219480064 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.0007514945464208722, + "objective/train/docs_used": 135676, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.773653745651245, + "objective/train/original_loss": 3.773653745651245, + "objective/train/theoretical_loss": 4.305156436273988, + "objective/train/tokens_used": 240005600, + "objective/train/value_avg": -0.004276275634765625, + "objective/train/value_loss": 8.571583748562261e-05, + "objective/train/value_max": -0.0003101825714111328, + "objective/train/value_min": -0.09466552734375, + "objective/train/value_reward_corr": 0.26634238178579894, + "objective/train/value_std": 0.0038280487060546875, + "objective/train/weight_avg": 0.9963048696517944, + "objective/train/weighted_lm_loss": 3.7683732509613037, + "objective/train/weights_max": 2.370694398880005, + "objective/train/weights_min": 0.19405366480350494, + "theoretical_loss": 4.305156436273988, + "tokens_seen": 219545600 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004714944834503511, + "loss": 1.5385, + "theoretical_loss": 4.305156436273988, + "tokens_seen": 219545600 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004714844533600802, + "loss": 1.6542, + "theoretical_loss": 4.304997973212636, + "tokens_seen": 219611136 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047147442326980944, + "loss": 1.6738, + "theoretical_loss": 4.304839570668738, + "tokens_seen": 219676672 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004714643931795387, + "loss": 1.5949, + "theoretical_loss": 4.30468122860114, + "tokens_seen": 219742208 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004714543630892678, + "loss": 1.6657, + "theoretical_loss": 4.304522946968725, + "tokens_seen": 219807744 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047144433299899704, + "loss": 1.6038, + "theoretical_loss": 4.3043647257304185, + "tokens_seen": 219873280 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047143430290872616, + "loss": 1.5696, + "theoretical_loss": 4.3042065648451855, + "tokens_seen": 219938816 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004714242728184554, + "loss": 1.6298, + "theoretical_loss": 4.304048464272031, + "tokens_seen": 220004352 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004714142427281846, + "loss": 1.6508, + "theoretical_loss": 4.30389042397, + "tokens_seen": 220069888 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047140421263791376, + "loss": 1.5559, + "theoretical_loss": 4.303732443898179, + "tokens_seen": 220135424 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047139418254764294, + "loss": 1.6801, + "theoretical_loss": 4.3035745240156915, + "tokens_seen": 220200960 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004713841524573721, + "loss": 1.6699, + "theoretical_loss": 4.303416664281703, + "tokens_seen": 220266496 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004713741223671013, + "loss": 1.713, + "theoretical_loss": 4.303258864655419, + "tokens_seen": 220332032 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047136409227683054, + "loss": 1.6243, + "theoretical_loss": 4.303101125096083, + "tokens_seen": 220397568 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047135406218655967, + "loss": 1.6617, + "theoretical_loss": 4.302943445562979, + "tokens_seen": 220463104 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004713440320962889, + "loss": 1.7007, + "theoretical_loss": 4.302785826015434, + "tokens_seen": 220528640 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004713340020060181, + "loss": 1.7609, + "theoretical_loss": 4.302628266412809, + "tokens_seen": 220594176 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047132397191574726, + "loss": 1.602, + "theoretical_loss": 4.302470766714508, + "tokens_seen": 220659712 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047131394182547645, + "loss": 1.6063, + "theoretical_loss": 4.3023133268799745, + "tokens_seen": 220725248 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047130391173520563, + "loss": 1.6371, + "theoretical_loss": 4.302155946868689, + "tokens_seen": 220790784 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004712938816449348, + "loss": 1.5857, + "theoretical_loss": 4.301998626640176, + "tokens_seen": 220856320 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047128385155466404, + "loss": 1.6183, + "theoretical_loss": 4.301841366153996, + "tokens_seen": 220921856 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047127382146439317, + "loss": 1.6388, + "theoretical_loss": 4.301684165369747, + "tokens_seen": 220987392 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004712637913741224, + "loss": 1.702, + "theoretical_loss": 4.301527024247071, + "tokens_seen": 221052928 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047125376128385153, + "loss": 1.6648, + "theoretical_loss": 4.301369942745647, + "tokens_seen": 221118464 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.00138198328204453, + "objective/train/docs_used": 136371, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7412490844726562, + "objective/train/original_loss": 2.7412493228912354, + "objective/train/theoretical_loss": 4.3012129208251935, + "objective/train/tokens_used": 241644000, + "objective/train/value_avg": -0.00513458251953125, + "objective/train/value_loss": 0.00010170343739446253, + "objective/train/value_max": -0.00020825862884521484, + "objective/train/value_min": -0.140380859375, + "objective/train/value_reward_corr": 0.19829483321119415, + "objective/train/value_std": 0.004566192626953125, + "objective/train/weight_avg": 0.9906851053237915, + "objective/train/weighted_lm_loss": 2.7222704887390137, + "objective/train/weights_max": 3.0794124603271484, + "objective/train/weights_min": 0.14361153542995453, + "theoretical_loss": 4.3012129208251935, + "tokens_seen": 221184000 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047124373119358077, + "loss": 1.5881, + "theoretical_loss": 4.3012129208251935, + "tokens_seen": 221184000 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047123370110330995, + "loss": 1.685, + "theoretical_loss": 4.301055958445467, + "tokens_seen": 221249536 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047122367101303913, + "loss": 1.5842, + "theoretical_loss": 4.300899055566264, + "tokens_seen": 221315072 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004712136409227683, + "loss": 1.603, + "theoretical_loss": 4.300742212147419, + "tokens_seen": 221380608 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047120361083249755, + "loss": 1.5785, + "theoretical_loss": 4.300585428148809, + "tokens_seen": 221446144 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004711935807422267, + "loss": 1.5704, + "theoretical_loss": 4.300428703530346, + "tokens_seen": 221511680 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004711835506519559, + "loss": 1.6079, + "theoretical_loss": 4.300272038251983, + "tokens_seen": 221577216 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047117352056168504, + "loss": 1.6084, + "theoretical_loss": 4.30011543227371, + "tokens_seen": 221642752 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047116349047141427, + "loss": 1.5583, + "theoretical_loss": 4.299958885555558, + "tokens_seen": 221708288 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047115346038114345, + "loss": 1.7428, + "theoretical_loss": 4.2998023980575955, + "tokens_seen": 221773824 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047114343029087263, + "loss": 1.6515, + "theoretical_loss": 4.299645969739929, + "tokens_seen": 221839360 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004711334002006018, + "loss": 1.4985, + "theoretical_loss": 4.299489600562707, + "tokens_seen": 221904896 + }, + { + "epoch": 0.07, + "learning_rate": 0.000471123370110331, + "loss": 1.6693, + "theoretical_loss": 4.299333290486111, + "tokens_seen": 221970432 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004711133400200602, + "loss": 1.6521, + "theoretical_loss": 4.299177039470367, + "tokens_seen": 222035968 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004711033099297894, + "loss": 1.5706, + "theoretical_loss": 4.2990208474757345, + "tokens_seen": 222101504 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047109327983951854, + "loss": 1.5728, + "theoretical_loss": 4.298864714462514, + "tokens_seen": 222167040 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004710832497492478, + "loss": 1.6011, + "theoretical_loss": 4.298708640391046, + "tokens_seen": 222232576 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004710732196589769, + "loss": 1.6232, + "theoretical_loss": 4.2985526252217054, + "tokens_seen": 222298112 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047106318956870614, + "loss": 1.5977, + "theoretical_loss": 4.298396668914909, + "tokens_seen": 222363648 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004710531594784353, + "loss": 1.5753, + "theoretical_loss": 4.298240771431109, + "tokens_seen": 222429184 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004710431293881645, + "loss": 1.536, + "theoretical_loss": 4.298084932730797, + "tokens_seen": 222494720 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004710330992978937, + "loss": 1.5279, + "theoretical_loss": 4.297929152774504, + "tokens_seen": 222560256 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004710230692076229, + "loss": 1.6934, + "theoretical_loss": 4.297773431522797, + "tokens_seen": 222625792 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047101303911735204, + "loss": 1.6163, + "theoretical_loss": 4.297617768936283, + "tokens_seen": 222691328 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004710030090270813, + "loss": 1.6191, + "theoretical_loss": 4.297462164975606, + "tokens_seen": 222756864 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.001985639100894332, + "objective/train/docs_used": 137650, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8959412574768066, + "objective/train/original_loss": 2.8959412574768066, + "objective/train/theoretical_loss": 4.297306619601446, + "objective/train/tokens_used": 243282400, + "objective/train/value_avg": -0.005558013916015625, + "objective/train/value_loss": 9.902730380417779e-05, + "objective/train/value_max": -0.0003859996795654297, + "objective/train/value_min": -0.186279296875, + "objective/train/value_reward_corr": 0.2007513448467842, + "objective/train/value_std": 0.005558013916015625, + "objective/train/weight_avg": 0.9849157929420471, + "objective/train/weighted_lm_loss": 2.8455116748809814, + "objective/train/weights_max": 5.657164573669434, + "objective/train/weights_min": 0.19911527633666992, + "theoretical_loss": 4.297306619601446, + "tokens_seen": 222822400 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004709929789368104, + "loss": 1.5041, + "theoretical_loss": 4.297306619601446, + "tokens_seen": 222822400 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047098294884653964, + "loss": 1.5501, + "theoretical_loss": 4.297151132774527, + "tokens_seen": 222887936 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004709729187562688, + "loss": 1.6083, + "theoretical_loss": 4.296995704455602, + "tokens_seen": 222953472 + }, + { + "epoch": 0.07, + "learning_rate": 0.000470962888665998, + "loss": 1.7397, + "theoretical_loss": 4.29684033460547, + "tokens_seen": 223019008 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004709528585757272, + "loss": 1.6344, + "theoretical_loss": 4.296685023184963, + "tokens_seen": 223084544 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047094282848545636, + "loss": 1.665, + "theoretical_loss": 4.296529770154954, + "tokens_seen": 223150080 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047093279839518555, + "loss": 1.4904, + "theoretical_loss": 4.296374575476349, + "tokens_seen": 223215616 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004709227683049148, + "loss": 1.6659, + "theoretical_loss": 4.2962194391100965, + "tokens_seen": 223281152 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004709127382146439, + "loss": 1.7324, + "theoretical_loss": 4.296064361017181, + "tokens_seen": 223346688 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047090270812437314, + "loss": 1.6793, + "theoretical_loss": 4.295909341158622, + "tokens_seen": 223412224 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047089267803410227, + "loss": 1.5648, + "theoretical_loss": 4.295754379495481, + "tokens_seen": 223477760 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004708826479438315, + "loss": 1.5421, + "theoretical_loss": 4.295599475988853, + "tokens_seen": 223543296 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004708726178535607, + "loss": 1.6429, + "theoretical_loss": 4.295444630599873, + "tokens_seen": 223608832 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047086258776328987, + "loss": 1.5878, + "theoretical_loss": 4.29528984328971, + "tokens_seen": 223674368 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047085255767301905, + "loss": 1.598, + "theoretical_loss": 4.295135114019576, + "tokens_seen": 223739904 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004708425275827483, + "loss": 1.5808, + "theoretical_loss": 4.294980442750715, + "tokens_seen": 223805440 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004708324974924774, + "loss": 1.6373, + "theoretical_loss": 4.294825829444411, + "tokens_seen": 223870976 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047082246740220665, + "loss": 1.6948, + "theoretical_loss": 4.294671274061983, + "tokens_seen": 223936512 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004708124373119358, + "loss": 1.6368, + "theoretical_loss": 4.29451677656479, + "tokens_seen": 224002048 + }, + { + "epoch": 0.07, + "learning_rate": 0.000470802407221665, + "loss": 1.5552, + "theoretical_loss": 4.294362336914225, + "tokens_seen": 224067584 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004707923771313942, + "loss": 1.5813, + "theoretical_loss": 4.2942079550717205, + "tokens_seen": 224133120 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047078234704112337, + "loss": 1.6168, + "theoretical_loss": 4.294053630998745, + "tokens_seen": 224198656 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047077231695085255, + "loss": 1.6162, + "theoretical_loss": 4.293899364656803, + "tokens_seen": 224264192 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047076228686058173, + "loss": 1.5819, + "theoretical_loss": 4.293745156007438, + "tokens_seen": 224329728 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004707522567703109, + "loss": 1.6674, + "theoretical_loss": 4.293591005012228, + "tokens_seen": 224395264 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": 8.608900679973885e-05, + "objective/train/docs_used": 138458, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2689714431762695, + "objective/train/original_loss": 3.2689709663391113, + "objective/train/theoretical_loss": 4.29343691163279, + "objective/train/tokens_used": 244920800, + "objective/train/value_avg": -0.005268096923828125, + "objective/train/value_loss": 8.354801684617996e-05, + "objective/train/value_max": -0.00022876262664794922, + "objective/train/value_min": -0.09039306640625, + "objective/train/value_reward_corr": 0.25621065969290635, + "objective/train/value_std": 0.0046234130859375, + "objective/train/weight_avg": 1.0046730041503906, + "objective/train/weighted_lm_loss": 3.292328119277954, + "objective/train/weights_max": 2.159123659133911, + "objective/train/weights_min": 0.3340187668800354, + "theoretical_loss": 4.29343691163279, + "tokens_seen": 224460800 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047074222668004015, + "loss": 1.676, + "theoretical_loss": 4.29343691163279, + "tokens_seen": 224460800 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004707321965897693, + "loss": 1.538, + "theoretical_loss": 4.293282875830776, + "tokens_seen": 224526336 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004707221664994985, + "loss": 1.6252, + "theoretical_loss": 4.293128897567877, + "tokens_seen": 224591872 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047071213640922775, + "loss": 1.5653, + "theoretical_loss": 4.292974976805818, + "tokens_seen": 224657408 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004707021063189569, + "loss": 1.5109, + "theoretical_loss": 4.29282111350636, + "tokens_seen": 224722944 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004706920762286861, + "loss": 1.6498, + "theoretical_loss": 4.292667307631305, + "tokens_seen": 224788480 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047068204613841524, + "loss": 1.5604, + "theoretical_loss": 4.292513559142487, + "tokens_seen": 224854016 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047067201604814447, + "loss": 1.5748, + "theoretical_loss": 4.292359868001779, + "tokens_seen": 224919552 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047066198595787365, + "loss": 1.6967, + "theoretical_loss": 4.29220623417109, + "tokens_seen": 224985088 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047065195586760283, + "loss": 1.6154, + "theoretical_loss": 4.2920526576123645, + "tokens_seen": 225050624 + }, + { + "epoch": 0.07, + "learning_rate": 0.000470641925777332, + "loss": 1.6827, + "theoretical_loss": 4.291899138287585, + "tokens_seen": 225116160 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004706318956870612, + "loss": 1.6948, + "theoretical_loss": 4.291745676158767, + "tokens_seen": 225181696 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004706218655967904, + "loss": 1.6415, + "theoretical_loss": 4.291592271187967, + "tokens_seen": 225247232 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004706118355065196, + "loss": 1.6999, + "theoretical_loss": 4.291438923337274, + "tokens_seen": 225312768 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047060180541624874, + "loss": 1.5954, + "theoretical_loss": 4.291285632568814, + "tokens_seen": 225378304 + }, + { + "epoch": 0.07, + "learning_rate": 0.000470591775325978, + "loss": 1.591, + "theoretical_loss": 4.291132398844749, + "tokens_seen": 225443840 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004705817452357071, + "loss": 1.6223, + "theoretical_loss": 4.29097922212728, + "tokens_seen": 225509376 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047057171514543634, + "loss": 1.5327, + "theoretical_loss": 4.290826102378639, + "tokens_seen": 225574912 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004705616850551655, + "loss": 1.6033, + "theoretical_loss": 4.290673039561098, + "tokens_seen": 225640448 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004705516549648947, + "loss": 1.6652, + "theoretical_loss": 4.290520033636963, + "tokens_seen": 225705984 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004705416248746239, + "loss": 1.5623, + "theoretical_loss": 4.290367084568576, + "tokens_seen": 225771520 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004705315947843531, + "loss": 1.6041, + "theoretical_loss": 4.290214192318317, + "tokens_seen": 225837056 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047052156469408224, + "loss": 1.7106, + "theoretical_loss": 4.290061356848597, + "tokens_seen": 225902592 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004705115346038115, + "loss": 1.4311, + "theoretical_loss": 4.289908578121868, + "tokens_seen": 225968128 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004705015045135406, + "loss": 1.6733, + "theoretical_loss": 4.289755856100615, + "tokens_seen": 226033664 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.001394487451761961, + "objective/train/docs_used": 139641, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3454904556274414, + "objective/train/original_loss": 3.3454909324645996, + "objective/train/theoretical_loss": 4.289603190747359, + "objective/train/tokens_used": 246559200, + "objective/train/value_avg": -0.00469207763671875, + "objective/train/value_loss": 7.450778502970934e-05, + "objective/train/value_max": -0.00030541419982910156, + "objective/train/value_min": -0.048675537109375, + "objective/train/value_reward_corr": 0.3522624492690106, + "objective/train/value_std": 0.0038909912109375, + "objective/train/weight_avg": 0.9895340800285339, + "objective/train/weighted_lm_loss": 3.317887306213379, + "objective/train/weights_max": 1.57668137550354, + "objective/train/weights_min": 0.40886664390563965, + "theoretical_loss": 4.289603190747359, + "tokens_seen": 226099200 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047049147442326984, + "loss": 1.5803, + "theoretical_loss": 4.289603190747359, + "tokens_seen": 226099200 + }, + { + "epoch": 0.07, + "learning_rate": 0.000470481444332999, + "loss": 1.4468, + "theoretical_loss": 4.2894505820246565, + "tokens_seen": 226164736 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004704714142427282, + "loss": 1.5586, + "theoretical_loss": 4.2892980298951, + "tokens_seen": 226230272 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004704613841524574, + "loss": 1.564, + "theoretical_loss": 4.289145534321319, + "tokens_seen": 226295808 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047045135406218656, + "loss": 1.7414, + "theoretical_loss": 4.288993095265975, + "tokens_seen": 226361344 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047044132397191575, + "loss": 1.5567, + "theoretical_loss": 4.288840712691767, + "tokens_seen": 226426880 + }, + { + "epoch": 0.07, + "learning_rate": 0.000470431293881645, + "loss": 1.566, + "theoretical_loss": 4.2886883865614305, + "tokens_seen": 226492416 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004704212637913741, + "loss": 1.6637, + "theoretical_loss": 4.288536116837733, + "tokens_seen": 226557952 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047041123370110334, + "loss": 1.5893, + "theoretical_loss": 4.288383903483481, + "tokens_seen": 226623488 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047040120361083247, + "loss": 1.6371, + "theoretical_loss": 4.288231746461515, + "tokens_seen": 226689024 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004703911735205617, + "loss": 1.5388, + "theoretical_loss": 4.28807964573471, + "tokens_seen": 226754560 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004703811434302909, + "loss": 1.7181, + "theoretical_loss": 4.287927601265976, + "tokens_seen": 226820096 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047037111334002007, + "loss": 1.6386, + "theoretical_loss": 4.28777561301826, + "tokens_seen": 226885632 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047036108324974925, + "loss": 1.7152, + "theoretical_loss": 4.287623680954542, + "tokens_seen": 226951168 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004703510531594785, + "loss": 1.6589, + "theoretical_loss": 4.287471805037837, + "tokens_seen": 227016704 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004703410230692076, + "loss": 1.6609, + "theoretical_loss": 4.287319985231198, + "tokens_seen": 227082240 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047033099297893685, + "loss": 1.7099, + "theoretical_loss": 4.287168221497709, + "tokens_seen": 227147776 + }, + { + "epoch": 0.07, + "learning_rate": 0.000470320962888666, + "loss": 1.6481, + "theoretical_loss": 4.287016513800492, + "tokens_seen": 227213312 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004703109327983952, + "loss": 1.6592, + "theoretical_loss": 4.2868648621027035, + "tokens_seen": 227278848 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004703009027081244, + "loss": 1.6483, + "theoretical_loss": 4.286713266367532, + "tokens_seen": 227344384 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047029087261785357, + "loss": 1.5763, + "theoretical_loss": 4.2865617265582046, + "tokens_seen": 227409920 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047028084252758275, + "loss": 1.5721, + "theoretical_loss": 4.28641024263798, + "tokens_seen": 227475456 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047027081243731193, + "loss": 1.652, + "theoretical_loss": 4.286258814570154, + "tokens_seen": 227540992 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004702607823470411, + "loss": 1.6077, + "theoretical_loss": 4.286107442318055, + "tokens_seen": 227606528 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047025075225677035, + "loss": 1.6059, + "theoretical_loss": 4.28595612584505, + "tokens_seen": 227672064 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.002631810260936618, + "objective/train/docs_used": 140376, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.365994691848755, + "objective/train/original_loss": 3.365995168685913, + "objective/train/theoretical_loss": 4.2858048651145335, + "objective/train/tokens_used": 248197600, + "objective/train/value_avg": -0.00716400146484375, + "objective/train/value_loss": 0.00012186620733700693, + "objective/train/value_max": -0.0003814697265625, + "objective/train/value_min": -0.1455078125, + "objective/train/value_reward_corr": 0.7235012391400263, + "objective/train/value_std": 0.00858306884765625, + "objective/train/weight_avg": 0.9793472290039062, + "objective/train/weighted_lm_loss": 3.28609561920166, + "objective/train/weights_max": 3.7466039657592773, + "objective/train/weights_min": 0.37764790654182434, + "theoretical_loss": 4.2858048651145335, + "tokens_seen": 227737600 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004702407221664995, + "loss": 1.5391, + "theoretical_loss": 4.2858048651145335, + "tokens_seen": 227737600 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004702306920762287, + "loss": 1.7197, + "theoretical_loss": 4.285653660089942, + "tokens_seen": 227803136 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047022066198595784, + "loss": 1.5896, + "theoretical_loss": 4.285502510734742, + "tokens_seen": 227868672 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004702106318956871, + "loss": 1.6475, + "theoretical_loss": 4.285351417012436, + "tokens_seen": 227934208 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047020060180541626, + "loss": 1.6073, + "theoretical_loss": 4.285200378886559, + "tokens_seen": 227999744 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047019057171514544, + "loss": 1.6372, + "theoretical_loss": 4.2850493963206855, + "tokens_seen": 228065280 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004701805416248746, + "loss": 1.7209, + "theoretical_loss": 4.284898469278417, + "tokens_seen": 228130816 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047017051153460385, + "loss": 1.6357, + "theoretical_loss": 4.2847475977233955, + "tokens_seen": 228196352 + }, + { + "epoch": 0.07, + "learning_rate": 0.000470160481444333, + "loss": 1.5677, + "theoretical_loss": 4.284596781619294, + "tokens_seen": 228261888 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004701504513540622, + "loss": 1.6652, + "theoretical_loss": 4.284446020929821, + "tokens_seen": 228327424 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047014042126379134, + "loss": 1.5604, + "theoretical_loss": 4.284295315618719, + "tokens_seen": 228392960 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004701303911735206, + "loss": 1.6796, + "theoretical_loss": 4.284144665649762, + "tokens_seen": 228458496 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047012036108324976, + "loss": 1.5503, + "theoretical_loss": 4.283994070986764, + "tokens_seen": 228524032 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047011033099297894, + "loss": 1.6418, + "theoretical_loss": 4.283843531593567, + "tokens_seen": 228589568 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004701003009027081, + "loss": 1.5724, + "theoretical_loss": 4.283693047434049, + "tokens_seen": 228655104 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004700902708124373, + "loss": 1.602, + "theoretical_loss": 4.283542618472125, + "tokens_seen": 228720640 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004700802407221665, + "loss": 1.5485, + "theoretical_loss": 4.283392244671739, + "tokens_seen": 228786176 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004700702106318957, + "loss": 1.5809, + "theoretical_loss": 4.2832419259968715, + "tokens_seen": 228851712 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047006018054162484, + "loss": 1.6703, + "theoretical_loss": 4.283091662411537, + "tokens_seen": 228917248 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004700501504513541, + "loss": 1.6668, + "theoretical_loss": 4.282941453879784, + "tokens_seen": 228982784 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047004012036108326, + "loss": 1.6005, + "theoretical_loss": 4.282791300365692, + "tokens_seen": 229048320 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047003009027081244, + "loss": 1.6113, + "theoretical_loss": 4.2826412018333775, + "tokens_seen": 229113856 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004700200601805416, + "loss": 1.5361, + "theoretical_loss": 4.282491158246989, + "tokens_seen": 229179392 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004700100300902708, + "loss": 1.6971, + "theoretical_loss": 4.28234116957071, + "tokens_seen": 229244928 + }, + { + "epoch": 0.07, + "learning_rate": 0.00047, + "loss": 1.5677, + "theoretical_loss": 4.282191235768756, + "tokens_seen": 229310464 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.0005236627184785903, + "objective/train/docs_used": 141693, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.252042770385742, + "objective/train/original_loss": 3.252042770385742, + "objective/train/theoretical_loss": 4.282041356805376, + "objective/train/tokens_used": 249836000, + "objective/train/value_avg": -0.004802703857421875, + "objective/train/value_loss": 6.927771755727008e-05, + "objective/train/value_max": -0.0003418922424316406, + "objective/train/value_min": -0.0615234375, + "objective/train/value_reward_corr": 0.21685299167615976, + "objective/train/value_std": 0.00420379638671875, + "objective/train/weight_avg": 0.9979325532913208, + "objective/train/weighted_lm_loss": 3.2556893825531006, + "objective/train/weights_max": 1.727613925933838, + "objective/train/weights_min": 0.19515977799892426, + "theoretical_loss": 4.282041356805376, + "tokens_seen": 229376000 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004699899699097292, + "loss": 1.6531, + "theoretical_loss": 4.282041356805376, + "tokens_seen": 229376000 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046997993981945835, + "loss": 1.6562, + "theoretical_loss": 4.2818915326448534, + "tokens_seen": 229441536 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004699699097291876, + "loss": 1.622, + "theoretical_loss": 4.281741763251505, + "tokens_seen": 229507072 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046995987963891676, + "loss": 1.6779, + "theoretical_loss": 4.281592048589681, + "tokens_seen": 229572608 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046994984954864595, + "loss": 1.5398, + "theoretical_loss": 4.281442388623764, + "tokens_seen": 229638144 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004699398194583752, + "loss": 1.8542, + "theoretical_loss": 4.281292783318172, + "tokens_seen": 229703680 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004699297893681043, + "loss": 1.6337, + "theoretical_loss": 4.281143232637353, + "tokens_seen": 229769216 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046991975927783354, + "loss": 1.829, + "theoretical_loss": 4.280993736545792, + "tokens_seen": 229834752 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046990972918756267, + "loss": 1.5564, + "theoretical_loss": 4.280844295008004, + "tokens_seen": 229900288 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004698996990972919, + "loss": 1.5409, + "theoretical_loss": 4.280694907988541, + "tokens_seen": 229965824 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004698896690070211, + "loss": 1.8137, + "theoretical_loss": 4.280545575451983, + "tokens_seen": 230031360 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046987963891675027, + "loss": 1.5793, + "theoretical_loss": 4.280396297362948, + "tokens_seen": 230096896 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046986960882647945, + "loss": 1.5715, + "theoretical_loss": 4.280247073686083, + "tokens_seen": 230162432 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004698595787362087, + "loss": 1.7353, + "theoretical_loss": 4.280097904386071, + "tokens_seen": 230227968 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004698495486459378, + "loss": 1.7296, + "theoretical_loss": 4.279948789427628, + "tokens_seen": 230293504 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046983951855566705, + "loss": 1.6759, + "theoretical_loss": 4.2797997287755, + "tokens_seen": 230359040 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004698294884653962, + "loss": 1.5604, + "theoretical_loss": 4.279650722394468, + "tokens_seen": 230424576 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004698194583751254, + "loss": 1.5682, + "theoretical_loss": 4.279501770249346, + "tokens_seen": 230490112 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004698094282848546, + "loss": 1.5512, + "theoretical_loss": 4.27935287230498, + "tokens_seen": 230555648 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046979939819458377, + "loss": 1.7138, + "theoretical_loss": 4.279204028526249, + "tokens_seen": 230621184 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046978936810431295, + "loss": 1.8138, + "theoretical_loss": 4.279055238878065, + "tokens_seen": 230686720 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046977933801404213, + "loss": 1.6035, + "theoretical_loss": 4.2789065033253735, + "tokens_seen": 230752256 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004697693079237713, + "loss": 1.7235, + "theoretical_loss": 4.278757821833151, + "tokens_seen": 230817792 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046975927783350055, + "loss": 1.6009, + "theoretical_loss": 4.278609194366406, + "tokens_seen": 230883328 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004697492477432297, + "loss": 1.5885, + "theoretical_loss": 4.278460620890183, + "tokens_seen": 230948864 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.0011405792320147157, + "objective/train/docs_used": 142407, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9171128273010254, + "objective/train/original_loss": 2.9171130657196045, + "objective/train/theoretical_loss": 4.278312101369555, + "objective/train/tokens_used": 251474400, + "objective/train/value_avg": -0.00553131103515625, + "objective/train/value_loss": 9.605884406482801e-05, + "objective/train/value_max": -0.0003800392150878906, + "objective/train/value_min": -0.17431640625, + "objective/train/value_reward_corr": 0.18644700504969483, + "objective/train/value_std": 0.0050506591796875, + "objective/train/weight_avg": 0.9932889938354492, + "objective/train/weighted_lm_loss": 2.922639846801758, + "objective/train/weights_max": 5.3712053298950195, + "objective/train/weights_min": 0.306576132774353, + "theoretical_loss": 4.278312101369555, + "tokens_seen": 231014400 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004697392176529589, + "loss": 1.5522, + "theoretical_loss": 4.278312101369555, + "tokens_seen": 231014400 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046972918756268804, + "loss": 1.5612, + "theoretical_loss": 4.278163635769631, + "tokens_seen": 231079936 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004697191574724173, + "loss": 1.5578, + "theoretical_loss": 4.27801522405555, + "tokens_seen": 231145472 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046970912738214646, + "loss": 1.6025, + "theoretical_loss": 4.277866866192484, + "tokens_seen": 231211008 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046969909729187564, + "loss": 1.5071, + "theoretical_loss": 4.277718562145639, + "tokens_seen": 231276544 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004696890672016048, + "loss": 1.4882, + "theoretical_loss": 4.27757031188025, + "tokens_seen": 231342080 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046967903711133405, + "loss": 1.6108, + "theoretical_loss": 4.277422115361588, + "tokens_seen": 231407616 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004696690070210632, + "loss": 1.5816, + "theoretical_loss": 4.277273972554954, + "tokens_seen": 231473152 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004696589769307924, + "loss": 1.5253, + "theoretical_loss": 4.277125883425682, + "tokens_seen": 231538688 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046964894684052154, + "loss": 1.5431, + "theoretical_loss": 4.276977847939136, + "tokens_seen": 231604224 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004696389167502508, + "loss": 1.5863, + "theoretical_loss": 4.276829866060717, + "tokens_seen": 231669760 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046962888665997996, + "loss": 1.565, + "theoretical_loss": 4.276681937755853, + "tokens_seen": 231735296 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046961885656970914, + "loss": 1.5526, + "theoretical_loss": 4.276534062990007, + "tokens_seen": 231800832 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004696088264794383, + "loss": 1.5739, + "theoretical_loss": 4.276386241728673, + "tokens_seen": 231866368 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004695987963891675, + "loss": 1.6328, + "theoretical_loss": 4.2762384739373775, + "tokens_seen": 231931904 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004695887662988967, + "loss": 1.5862, + "theoretical_loss": 4.276090759581678, + "tokens_seen": 231997440 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004695787362086259, + "loss": 1.5075, + "theoretical_loss": 4.275943098627166, + "tokens_seen": 232062976 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046956870611835505, + "loss": 1.6665, + "theoretical_loss": 4.275795491039462, + "tokens_seen": 232128512 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004695586760280843, + "loss": 1.6654, + "theoretical_loss": 4.27564793678422, + "tokens_seen": 232194048 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046954864593781346, + "loss": 1.6367, + "theoretical_loss": 4.2755004358271265, + "tokens_seen": 232259584 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046953861584754264, + "loss": 1.4055, + "theoretical_loss": 4.275352988133898, + "tokens_seen": 232325120 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004695285857572718, + "loss": 1.6382, + "theoretical_loss": 4.275205593670284, + "tokens_seen": 232390656 + }, + { + "epoch": 0.07, + "learning_rate": 0.000469518555667001, + "loss": 1.6557, + "theoretical_loss": 4.275058252402065, + "tokens_seen": 232456192 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004695085255767302, + "loss": 1.4723, + "theoretical_loss": 4.274910964295053, + "tokens_seen": 232521728 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004694984954864594, + "loss": 1.5602, + "theoretical_loss": 4.274763729315092, + "tokens_seen": 232587264 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.0006264677504077554, + "objective/train/docs_used": 143093, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2246336936950684, + "objective/train/original_loss": 3.2246334552764893, + "objective/train/theoretical_loss": 4.274616547428058, + "objective/train/tokens_used": 253112800, + "objective/train/value_avg": -0.004962921142578125, + "objective/train/value_loss": 6.604345253435895e-05, + "objective/train/value_max": -0.0002472400665283203, + "objective/train/value_min": -0.1683349609375, + "objective/train/value_reward_corr": 0.271221667946698, + "objective/train/value_std": 0.004924774169921875, + "objective/train/weight_avg": 0.9968739151954651, + "objective/train/weighted_lm_loss": 3.2226834297180176, + "objective/train/weights_max": 4.7793378829956055, + "objective/train/weights_min": 0.24906934797763824, + "theoretical_loss": 4.274616547428058, + "tokens_seen": 232652800 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046948846539618855, + "loss": 1.6545, + "theoretical_loss": 4.274616547428058, + "tokens_seen": 232652800 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004694784353059178, + "loss": 1.6283, + "theoretical_loss": 4.274469418599858, + "tokens_seen": 232718336 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004694684052156469, + "loss": 1.5723, + "theoretical_loss": 4.274322342796429, + "tokens_seen": 232783872 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046945837512537615, + "loss": 1.6238, + "theoretical_loss": 4.274175319983744, + "tokens_seen": 232849408 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004694483450351053, + "loss": 1.4825, + "theoretical_loss": 4.274028350127802, + "tokens_seen": 232914944 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004694383149448345, + "loss": 1.7066, + "theoretical_loss": 4.273881433194635, + "tokens_seen": 232980480 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004694282848545637, + "loss": 1.4691, + "theoretical_loss": 4.27373456915031, + "tokens_seen": 233046016 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046941825476429287, + "loss": 1.635, + "theoretical_loss": 4.27358775796092, + "tokens_seen": 233111552 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046940822467402205, + "loss": 1.4736, + "theoretical_loss": 4.273440999592593, + "tokens_seen": 233177088 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004693981945837513, + "loss": 1.5779, + "theoretical_loss": 4.273294294011485, + "tokens_seen": 233242624 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004693881644934804, + "loss": 1.5771, + "theoretical_loss": 4.273147641183788, + "tokens_seen": 233308160 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046937813440320965, + "loss": 1.652, + "theoretical_loss": 4.2730010410757195, + "tokens_seen": 233373696 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046936810431293883, + "loss": 1.6267, + "theoretical_loss": 4.272854493653532, + "tokens_seen": 233439232 + }, + { + "epoch": 0.07, + "learning_rate": 0.000469358074222668, + "loss": 1.6478, + "theoretical_loss": 4.272707998883508, + "tokens_seen": 233504768 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004693480441323972, + "loss": 1.4793, + "theoretical_loss": 4.27256155673196, + "tokens_seen": 233570304 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004693380140421264, + "loss": 1.4994, + "theoretical_loss": 4.272415167165232, + "tokens_seen": 233635840 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046932798395185555, + "loss": 1.6407, + "theoretical_loss": 4.272268830149702, + "tokens_seen": 233701376 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004693179538615848, + "loss": 1.6244, + "theoretical_loss": 4.272122545651774, + "tokens_seen": 233766912 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004693079237713139, + "loss": 1.723, + "theoretical_loss": 4.271976313637885, + "tokens_seen": 233832448 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046929789368104315, + "loss": 1.6017, + "theoretical_loss": 4.271830134074504, + "tokens_seen": 233897984 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004692878635907723, + "loss": 1.5658, + "theoretical_loss": 4.27168400692813, + "tokens_seen": 233963520 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004692778335005015, + "loss": 1.6261, + "theoretical_loss": 4.271537932165291, + "tokens_seen": 234029056 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004692678034102307, + "loss": 1.5271, + "theoretical_loss": 4.27139190975255, + "tokens_seen": 234094592 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004692577733199599, + "loss": 1.6679, + "theoretical_loss": 4.271245939656495, + "tokens_seen": 234160128 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046924774322968906, + "loss": 1.7534, + "theoretical_loss": 4.2711000218437505, + "tokens_seen": 234225664 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": 0.000233753293287009, + "objective/train/docs_used": 144323, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.286902904510498, + "objective/train/original_loss": 3.286902904510498, + "objective/train/theoretical_loss": 4.2709541562809665, + "objective/train/tokens_used": 254751200, + "objective/train/value_avg": -0.005451202392578125, + "objective/train/value_loss": 7.120460941223428e-05, + "objective/train/value_max": -0.00018966197967529297, + "objective/train/value_min": -0.058441162109375, + "objective/train/value_reward_corr": 0.32925786213870806, + "objective/train/value_std": 0.00446319580078125, + "objective/train/weight_avg": 1.0056533813476562, + "objective/train/weighted_lm_loss": 3.296312093734741, + "objective/train/weights_max": 1.7846015691757202, + "objective/train/weights_min": 0.24807843565940857, + "theoretical_loss": 4.2709541562809665, + "tokens_seen": 234291200 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046923771313941824, + "loss": 1.688, + "theoretical_loss": 4.2709541562809665, + "tokens_seen": 234291200 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004692276830491474, + "loss": 1.6565, + "theoretical_loss": 4.270808342934827, + "tokens_seen": 234356736 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046921765295887666, + "loss": 1.6162, + "theoretical_loss": 4.270662581772045, + "tokens_seen": 234422272 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046920762286860584, + "loss": 1.6343, + "theoretical_loss": 4.270516872759366, + "tokens_seen": 234487808 + }, + { + "epoch": 0.07, + "learning_rate": 0.000469197592778335, + "loss": 1.7799, + "theoretical_loss": 4.2703712158635625, + "tokens_seen": 234553344 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046918756268806425, + "loss": 1.6576, + "theoretical_loss": 4.2702256110514405, + "tokens_seen": 234618880 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004691775325977934, + "loss": 1.5625, + "theoretical_loss": 4.270080058289835, + "tokens_seen": 234684416 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004691675025075226, + "loss": 1.6034, + "theoretical_loss": 4.269934557545612, + "tokens_seen": 234749952 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046915747241725174, + "loss": 1.5988, + "theoretical_loss": 4.269789108785667, + "tokens_seen": 234815488 + }, + { + "epoch": 0.07, + "learning_rate": 0.000469147442326981, + "loss": 1.6779, + "theoretical_loss": 4.269643711976926, + "tokens_seen": 234881024 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046913741223671016, + "loss": 1.5775, + "theoretical_loss": 4.269498367086347, + "tokens_seen": 234946560 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046912738214643934, + "loss": 1.5746, + "theoretical_loss": 4.269353074080915, + "tokens_seen": 235012096 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004691173520561685, + "loss": 1.6428, + "theoretical_loss": 4.269207832927649, + "tokens_seen": 235077632 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004691073219658977, + "loss": 1.5448, + "theoretical_loss": 4.269062643593594, + "tokens_seen": 235143168 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004690972918756269, + "loss": 1.5512, + "theoretical_loss": 4.268917506045829, + "tokens_seen": 235208704 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004690872617853561, + "loss": 1.5673, + "theoretical_loss": 4.268772420251461, + "tokens_seen": 235274240 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046907723169508525, + "loss": 1.6478, + "theoretical_loss": 4.268627386177627, + "tokens_seen": 235339776 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004690672016048145, + "loss": 1.6644, + "theoretical_loss": 4.268482403791495, + "tokens_seen": 235405312 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046905717151454366, + "loss": 1.609, + "theoretical_loss": 4.268337473060262, + "tokens_seen": 235470848 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046904714142427284, + "loss": 1.6493, + "theoretical_loss": 4.268192593951154, + "tokens_seen": 235536384 + }, + { + "epoch": 0.07, + "learning_rate": 0.000469037111334002, + "loss": 1.6232, + "theoretical_loss": 4.2680477664314305, + "tokens_seen": 235601920 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004690270812437312, + "loss": 1.5183, + "theoretical_loss": 4.267902990468377, + "tokens_seen": 235667456 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004690170511534604, + "loss": 1.517, + "theoretical_loss": 4.267758266029312, + "tokens_seen": 235732992 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004690070210631896, + "loss": 1.576, + "theoretical_loss": 4.267613593081581, + "tokens_seen": 235798528 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046899699097291875, + "loss": 1.6642, + "theoretical_loss": 4.26746897159256, + "tokens_seen": 235864064 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.0003513921983540058, + "objective/train/docs_used": 145084, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.374095916748047, + "objective/train/original_loss": 3.374095916748047, + "objective/train/theoretical_loss": 4.267324401529657, + "objective/train/tokens_used": 256389600, + "objective/train/value_avg": -0.00502777099609375, + "objective/train/value_loss": 3.7510893889702857e-05, + "objective/train/value_max": -0.00026726722717285156, + "objective/train/value_min": -0.056243896484375, + "objective/train/value_reward_corr": 0.3188127872850686, + "objective/train/value_std": 0.004108428955078125, + "objective/train/weight_avg": 0.9983444809913635, + "objective/train/weighted_lm_loss": 3.3815503120422363, + "objective/train/weights_max": 1.754947543144226, + "objective/train/weights_min": 0.6028777956962585, + "theoretical_loss": 4.267324401529657, + "tokens_seen": 235929600 + }, + { + "epoch": 0.07, + "learning_rate": 0.000468986960882648, + "loss": 1.7024, + "theoretical_loss": 4.267324401529657, + "tokens_seen": 235929600 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004689769307923771, + "loss": 1.6149, + "theoretical_loss": 4.267179882860306, + "tokens_seen": 235995136 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046896690070210635, + "loss": 1.4749, + "theoretical_loss": 4.267035415551975, + "tokens_seen": 236060672 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046895687061183553, + "loss": 1.5112, + "theoretical_loss": 4.266890999572157, + "tokens_seen": 236126208 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004689468405215647, + "loss": 1.6064, + "theoretical_loss": 4.266746634888377, + "tokens_seen": 236191744 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004689368104312939, + "loss": 1.5507, + "theoretical_loss": 4.26660232146819, + "tokens_seen": 236257280 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046892678034102307, + "loss": 1.6317, + "theoretical_loss": 4.26645805927918, + "tokens_seen": 236322816 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046891675025075225, + "loss": 1.6588, + "theoretical_loss": 4.266313848288959, + "tokens_seen": 236388352 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004689067201604815, + "loss": 1.5517, + "theoretical_loss": 4.266169688465172, + "tokens_seen": 236453888 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004688966900702106, + "loss": 1.5019, + "theoretical_loss": 4.266025579775489, + "tokens_seen": 236519424 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046888665997993985, + "loss": 1.6021, + "theoretical_loss": 4.265881522187613, + "tokens_seen": 236584960 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046887662988966903, + "loss": 1.5862, + "theoretical_loss": 4.265737515669274, + "tokens_seen": 236650496 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004688665997993982, + "loss": 1.5154, + "theoretical_loss": 4.2655935601882335, + "tokens_seen": 236716032 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004688565697091274, + "loss": 1.4732, + "theoretical_loss": 4.265449655712279, + "tokens_seen": 236781568 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004688465396188566, + "loss": 1.5962, + "theoretical_loss": 4.265305802209231, + "tokens_seen": 236847104 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046883650952858575, + "loss": 1.6555, + "theoretical_loss": 4.265161999646936, + "tokens_seen": 236912640 + }, + { + "epoch": 0.07, + "learning_rate": 0.000468826479438315, + "loss": 1.4415, + "theoretical_loss": 4.265018247993272, + "tokens_seen": 236978176 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004688164493480441, + "loss": 1.4313, + "theoretical_loss": 4.264874547216145, + "tokens_seen": 237043712 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046880641925777335, + "loss": 1.561, + "theoretical_loss": 4.26473089728349, + "tokens_seen": 237109248 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004687963891675025, + "loss": 1.552, + "theoretical_loss": 4.264587298163272, + "tokens_seen": 237174784 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004687863590772317, + "loss": 1.5182, + "theoretical_loss": 4.264443749823484, + "tokens_seen": 237240320 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004687763289869609, + "loss": 1.5377, + "theoretical_loss": 4.2643002522321485, + "tokens_seen": 237305856 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004687662988966901, + "loss": 1.4708, + "theoretical_loss": 4.264156805357317, + "tokens_seen": 237371392 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046875626880641926, + "loss": 1.5792, + "theoretical_loss": 4.264013409167069, + "tokens_seen": 237436928 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046874623871614844, + "loss": 1.4759, + "theoretical_loss": 4.263870063629515, + "tokens_seen": 237502464 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -7.047795224934816e-05, + "objective/train/docs_used": 146518, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.436687707901001, + "objective/train/original_loss": 3.436687707901001, + "objective/train/theoretical_loss": 4.263726768712791, + "objective/train/tokens_used": 258028000, + "objective/train/value_avg": -0.0045013427734375, + "objective/train/value_loss": 7.912515138741583e-05, + "objective/train/value_max": -0.00030541419982910156, + "objective/train/value_min": -0.06536865234375, + "objective/train/value_reward_corr": 0.30617061481476915, + "objective/train/value_std": 0.00370025634765625, + "objective/train/weight_avg": 1.0027203559875488, + "objective/train/weighted_lm_loss": 3.444139242172241, + "objective/train/weights_max": 1.7314765453338623, + "objective/train/weights_min": 0.0842919647693634, + "theoretical_loss": 4.263726768712791, + "tokens_seen": 237568000 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004687362086258776, + "loss": 1.5601, + "theoretical_loss": 4.263726768712791, + "tokens_seen": 237568000 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046872617853560686, + "loss": 1.637, + "theoretical_loss": 4.263583524385067, + "tokens_seen": 237633536 + }, + { + "epoch": 0.07, + "learning_rate": 0.000468716148445336, + "loss": 1.686, + "theoretical_loss": 4.263440330614536, + "tokens_seen": 237699072 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004687061183550652, + "loss": 1.4477, + "theoretical_loss": 4.263297187369423, + "tokens_seen": 237764608 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004686960882647944, + "loss": 1.6426, + "theoretical_loss": 4.263154094617981, + "tokens_seen": 237830144 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004686860581745236, + "loss": 1.624, + "theoretical_loss": 4.263011052328492, + "tokens_seen": 237895680 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046867602808425276, + "loss": 1.6087, + "theoretical_loss": 4.262868060469268, + "tokens_seen": 237961216 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046866599799398194, + "loss": 1.593, + "theoretical_loss": 4.262725119008646, + "tokens_seen": 238026752 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004686559679037111, + "loss": 1.6482, + "theoretical_loss": 4.262582227914994, + "tokens_seen": 238092288 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046864593781344036, + "loss": 1.6144, + "theoretical_loss": 4.262439387156709, + "tokens_seen": 238157824 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004686359077231695, + "loss": 1.6591, + "theoretical_loss": 4.262296596702216, + "tokens_seen": 238223360 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004686258776328987, + "loss": 1.6034, + "theoretical_loss": 4.2621538565199675, + "tokens_seen": 238288896 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046861584754262785, + "loss": 1.5273, + "theoretical_loss": 4.262011166578445, + "tokens_seen": 238354432 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004686058174523571, + "loss": 1.6466, + "theoretical_loss": 4.261868526846159, + "tokens_seen": 238419968 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046859578736208626, + "loss": 1.445, + "theoretical_loss": 4.261725937291648, + "tokens_seen": 238485504 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046858575727181545, + "loss": 1.7199, + "theoretical_loss": 4.261583397883479, + "tokens_seen": 238551040 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004685757271815446, + "loss": 1.598, + "theoretical_loss": 4.261440908590247, + "tokens_seen": 238616576 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004685656970912738, + "loss": 1.4631, + "theoretical_loss": 4.261298469380575, + "tokens_seen": 238682112 + }, + { + "epoch": 0.07, + "learning_rate": 0.000468555667001003, + "loss": 1.6509, + "theoretical_loss": 4.261156080223116, + "tokens_seen": 238747648 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004685456369107322, + "loss": 1.6469, + "theoretical_loss": 4.261013741086549, + "tokens_seen": 238813184 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046853560682046135, + "loss": 1.624, + "theoretical_loss": 4.260871451939582, + "tokens_seen": 238878720 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004685255767301906, + "loss": 1.5012, + "theoretical_loss": 4.2607292127509515, + "tokens_seen": 238944256 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046851554663991977, + "loss": 1.6186, + "theoretical_loss": 4.260587023489422, + "tokens_seen": 239009792 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046850551654964895, + "loss": 1.5795, + "theoretical_loss": 4.260444884123785, + "tokens_seen": 239075328 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046849548645937813, + "loss": 1.6804, + "theoretical_loss": 4.260302794622864, + "tokens_seen": 239140864 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.0015173207502812147, + "objective/train/docs_used": 147039, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.919320583343506, + "objective/train/original_loss": 2.9193201065063477, + "objective/train/theoretical_loss": 4.260160754955504, + "objective/train/tokens_used": 259666400, + "objective/train/value_avg": -0.00495147705078125, + "objective/train/value_loss": 7.551752787549049e-05, + "objective/train/value_max": -0.0001971721649169922, + "objective/train/value_min": -0.089599609375, + "objective/train/value_reward_corr": 0.33329001263626773, + "objective/train/value_std": 0.004398345947265625, + "objective/train/weight_avg": 0.9883362650871277, + "objective/train/weighted_lm_loss": 2.8869214057922363, + "objective/train/weights_max": 2.0587172508239746, + "objective/train/weights_min": 0.34302544593811035, + "theoretical_loss": 4.260160754955504, + "tokens_seen": 239206400 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004684854563691073, + "loss": 1.5615, + "theoretical_loss": 4.260160754955504, + "tokens_seen": 239206400 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004684754262788365, + "loss": 1.6412, + "theoretical_loss": 4.260018765090583, + "tokens_seen": 239271936 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046846539618856573, + "loss": 1.6725, + "theoretical_loss": 4.259876824997007, + "tokens_seen": 239337472 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004684553660982949, + "loss": 1.5548, + "theoretical_loss": 4.259734934643706, + "tokens_seen": 239403008 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004684453360080241, + "loss": 1.6531, + "theoretical_loss": 4.259593093999641, + "tokens_seen": 239468544 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046843530591775327, + "loss": 1.5731, + "theoretical_loss": 4.259451303033801, + "tokens_seen": 239534080 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046842527582748245, + "loss": 1.6143, + "theoretical_loss": 4.259309561715201, + "tokens_seen": 239599616 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004684152457372117, + "loss": 1.5568, + "theoretical_loss": 4.259167870012885, + "tokens_seen": 239665152 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004684052156469408, + "loss": 1.5527, + "theoretical_loss": 4.259026227895926, + "tokens_seen": 239730688 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046839518555667005, + "loss": 1.6386, + "theoretical_loss": 4.258884635333421, + "tokens_seen": 239796224 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046838515546639923, + "loss": 1.6225, + "theoretical_loss": 4.258743092294498, + "tokens_seen": 239861760 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004683751253761284, + "loss": 1.6958, + "theoretical_loss": 4.258601598748312, + "tokens_seen": 239927296 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004683650952858576, + "loss": 1.6253, + "theoretical_loss": 4.2584601546640455, + "tokens_seen": 239992832 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004683550651955868, + "loss": 1.4865, + "theoretical_loss": 4.258318760010908, + "tokens_seen": 240058368 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046834503510531595, + "loss": 1.5562, + "theoretical_loss": 4.258177414758135, + "tokens_seen": 240123904 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004683350050150452, + "loss": 1.7195, + "theoretical_loss": 4.258036118874995, + "tokens_seen": 240189440 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004683249749247743, + "loss": 1.648, + "theoretical_loss": 4.257894872330778, + "tokens_seen": 240254976 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046831494483450355, + "loss": 1.5567, + "theoretical_loss": 4.257753675094804, + "tokens_seen": 240320512 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004683049147442327, + "loss": 1.5902, + "theoretical_loss": 4.2576125271364225, + "tokens_seen": 240386048 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004682948846539619, + "loss": 1.5372, + "theoretical_loss": 4.257471428425006, + "tokens_seen": 240451584 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004682848545636911, + "loss": 1.6845, + "theoretical_loss": 4.257330378929957, + "tokens_seen": 240517120 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004682748244734203, + "loss": 1.6338, + "theoretical_loss": 4.257189378620707, + "tokens_seen": 240582656 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046826479438314946, + "loss": 1.6434, + "theoretical_loss": 4.257048427466709, + "tokens_seen": 240648192 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046825476429287864, + "loss": 1.6164, + "theoretical_loss": 4.256907525437451, + "tokens_seen": 240713728 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004682447342026078, + "loss": 1.6395, + "theoretical_loss": 4.256766672502442, + "tokens_seen": 240779264 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.0011522077256813645, + "objective/train/docs_used": 148118, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3183250427246094, + "objective/train/original_loss": 3.3183255195617676, + "objective/train/theoretical_loss": 4.256625868631222, + "objective/train/tokens_used": 261304800, + "objective/train/value_avg": -0.00536346435546875, + "objective/train/value_loss": 7.961366645758972e-05, + "objective/train/value_max": -0.0003006458282470703, + "objective/train/value_min": -0.07684326171875, + "objective/train/value_reward_corr": 0.191484197353201, + "objective/train/value_std": 0.0044097900390625, + "objective/train/weight_avg": 0.9921500086784363, + "objective/train/weighted_lm_loss": 3.3045260906219482, + "objective/train/weights_max": 1.997511386871338, + "objective/train/weights_min": 0.24925468862056732, + "theoretical_loss": 4.256625868631222, + "tokens_seen": 240844800 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046823470411233706, + "loss": 1.5472, + "theoretical_loss": 4.256625868631222, + "tokens_seen": 240844800 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004682246740220662, + "loss": 1.6296, + "theoretical_loss": 4.256485113793355, + "tokens_seen": 240910336 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004682146439317954, + "loss": 1.5567, + "theoretical_loss": 4.2563444079584345, + "tokens_seen": 240975872 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004682046138415246, + "loss": 1.5751, + "theoretical_loss": 4.2562037510960815, + "tokens_seen": 241041408 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004681945837512538, + "loss": 1.6184, + "theoretical_loss": 4.2560631431759415, + "tokens_seen": 241106944 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046818455366098296, + "loss": 1.7308, + "theoretical_loss": 4.25592258416769, + "tokens_seen": 241172480 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046817452357071214, + "loss": 1.6597, + "theoretical_loss": 4.255782074041028, + "tokens_seen": 241238016 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004681644934804413, + "loss": 1.6203, + "theoretical_loss": 4.255641612765683, + "tokens_seen": 241303552 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046815446339017056, + "loss": 1.6624, + "theoretical_loss": 4.255501200311411, + "tokens_seen": 241369088 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004681444332998997, + "loss": 1.6457, + "theoretical_loss": 4.255360836647993, + "tokens_seen": 241434624 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004681344032096289, + "loss": 1.5448, + "theoretical_loss": 4.255220521745239, + "tokens_seen": 241500160 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046812437311935805, + "loss": 1.6522, + "theoretical_loss": 4.255080255572984, + "tokens_seen": 241565696 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004681143430290873, + "loss": 1.7152, + "theoretical_loss": 4.254940038101091, + "tokens_seen": 241631232 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046810431293881646, + "loss": 1.5942, + "theoretical_loss": 4.254799869299449, + "tokens_seen": 241696768 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046809428284854565, + "loss": 1.5943, + "theoretical_loss": 4.254659749137975, + "tokens_seen": 241762304 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004680842527582748, + "loss": 1.6058, + "theoretical_loss": 4.254519677586611, + "tokens_seen": 241827840 + }, + { + "epoch": 0.07, + "learning_rate": 0.000468074222668004, + "loss": 1.6734, + "theoretical_loss": 4.254379654615327, + "tokens_seen": 241893376 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004680641925777332, + "loss": 1.742, + "theoretical_loss": 4.254239680194121, + "tokens_seen": 241958912 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004680541624874624, + "loss": 1.5659, + "theoretical_loss": 4.254099754293014, + "tokens_seen": 242024448 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046804413239719155, + "loss": 1.6056, + "theoretical_loss": 4.253959876882057, + "tokens_seen": 242089984 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004680341023069208, + "loss": 1.7034, + "theoretical_loss": 4.253820047931324, + "tokens_seen": 242155520 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046802407221664997, + "loss": 1.5452, + "theoretical_loss": 4.253680267410921, + "tokens_seen": 242221056 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046801404212637915, + "loss": 1.6164, + "theoretical_loss": 4.253540535290975, + "tokens_seen": 242286592 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046800401203610833, + "loss": 1.5931, + "theoretical_loss": 4.253400851541643, + "tokens_seen": 242352128 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004679939819458375, + "loss": 1.5772, + "theoretical_loss": 4.253261216133106, + "tokens_seen": 242417664 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": 0.0005841829697601497, + "objective/train/docs_used": 148826, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2828218936920166, + "objective/train/original_loss": 3.2828214168548584, + "objective/train/theoretical_loss": 4.253121629035574, + "objective/train/tokens_used": 262943200, + "objective/train/value_avg": -0.00518798828125, + "objective/train/value_loss": 5.273719216347672e-05, + "objective/train/value_max": -0.0002551078796386719, + "objective/train/value_min": -0.0849609375, + "objective/train/value_reward_corr": 0.39579306799364555, + "objective/train/value_std": 0.00469207763671875, + "objective/train/weight_avg": 1.008391261100769, + "objective/train/weighted_lm_loss": 3.301391363143921, + "objective/train/weights_max": 2.255216121673584, + "objective/train/weights_min": 0.44679397344589233, + "theoretical_loss": 4.253121629035574, + "tokens_seen": 242483200 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004679839518555667, + "loss": 1.706, + "theoretical_loss": 4.253121629035574, + "tokens_seen": 242483200 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046797392176529593, + "loss": 1.424, + "theoretical_loss": 4.252982090219281, + "tokens_seen": 242548736 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046796389167502505, + "loss": 1.6464, + "theoretical_loss": 4.25284259965449, + "tokens_seen": 242614272 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004679538615847543, + "loss": 1.6141, + "theoretical_loss": 4.252703157311487, + "tokens_seen": 242679808 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004679438314944834, + "loss": 1.6013, + "theoretical_loss": 4.2525637631605875, + "tokens_seen": 242745344 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046793380140421265, + "loss": 1.6368, + "theoretical_loss": 4.252424417172132, + "tokens_seen": 242810880 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046792377131394183, + "loss": 1.5793, + "theoretical_loss": 4.252285119316486, + "tokens_seen": 242876416 + }, + { + "epoch": 0.07, + "learning_rate": 0.000467913741223671, + "loss": 1.6285, + "theoretical_loss": 4.252145869564044, + "tokens_seen": 242941952 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004679037111334002, + "loss": 1.6011, + "theoretical_loss": 4.252006667885224, + "tokens_seen": 243007488 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046789368104312943, + "loss": 1.5892, + "theoretical_loss": 4.251867514250472, + "tokens_seen": 243073024 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046788365095285856, + "loss": 1.5601, + "theoretical_loss": 4.2517284086302585, + "tokens_seen": 243138560 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004678736208625878, + "loss": 1.5242, + "theoretical_loss": 4.251589350995081, + "tokens_seen": 243204096 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004678635907723169, + "loss": 1.5776, + "theoretical_loss": 4.251450341315464, + "tokens_seen": 243269632 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046785356068204616, + "loss": 1.5529, + "theoretical_loss": 4.251311379561956, + "tokens_seen": 243335168 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046784353059177534, + "loss": 1.4964, + "theoretical_loss": 4.251172465705134, + "tokens_seen": 243400704 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004678335005015045, + "loss": 1.5469, + "theoretical_loss": 4.2510335997155995, + "tokens_seen": 243466240 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004678234704112337, + "loss": 1.4795, + "theoretical_loss": 4.250894781563979, + "tokens_seen": 243531776 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004678134403209629, + "loss": 1.5074, + "theoretical_loss": 4.2507560112209255, + "tokens_seen": 243597312 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046780341023069206, + "loss": 1.5512, + "theoretical_loss": 4.25061728865712, + "tokens_seen": 243662848 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004677933801404213, + "loss": 1.5598, + "theoretical_loss": 4.250478613843266, + "tokens_seen": 243728384 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004677833500501504, + "loss": 1.5167, + "theoretical_loss": 4.250339986750097, + "tokens_seen": 243793920 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046777331995987966, + "loss": 1.6374, + "theoretical_loss": 4.250201407348368, + "tokens_seen": 243859456 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004677632898696088, + "loss": 1.6028, + "theoretical_loss": 4.250062875608862, + "tokens_seen": 243924992 + }, + { + "epoch": 0.07, + "learning_rate": 0.000467753259779338, + "loss": 1.5522, + "theoretical_loss": 4.249924391502387, + "tokens_seen": 243990528 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004677432296890672, + "loss": 1.5541, + "theoretical_loss": 4.249785954999778, + "tokens_seen": 244056064 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.0014816877664998174, + "objective/train/docs_used": 150000, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0205459594726562, + "objective/train/original_loss": 3.0205461978912354, + "objective/train/theoretical_loss": 4.249647566071895, + "objective/train/tokens_used": 264581600, + "objective/train/value_avg": -0.00620269775390625, + "objective/train/value_loss": 0.00013895663141738623, + "objective/train/value_max": -0.00025916099548339844, + "objective/train/value_min": -0.050994873046875, + "objective/train/value_reward_corr": 0.41897098858208665, + "objective/train/value_std": 0.004581451416015625, + "objective/train/weight_avg": 0.9915401339530945, + "objective/train/weighted_lm_loss": 2.98543643951416, + "objective/train/weights_max": 1.493505597114563, + "objective/train/weights_min": 0.34956425428390503, + "theoretical_loss": 4.249647566071895, + "tokens_seen": 244121600 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004677331995987964, + "loss": 1.5187, + "theoretical_loss": 4.249647566071895, + "tokens_seen": 244121600 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046772316950852556, + "loss": 1.6228, + "theoretical_loss": 4.249509224689621, + "tokens_seen": 244187136 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004677131394182548, + "loss": 1.6518, + "theoretical_loss": 4.24937093082387, + "tokens_seen": 244252672 + }, + { + "epoch": 0.07, + "learning_rate": 0.000467703109327984, + "loss": 1.5401, + "theoretical_loss": 4.249232684445579, + "tokens_seen": 244318208 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046769307923771316, + "loss": 1.4553, + "theoretical_loss": 4.249094485525706, + "tokens_seen": 244383744 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046768304914744234, + "loss": 1.6269, + "theoretical_loss": 4.248956334035244, + "tokens_seen": 244449280 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004676730190571715, + "loss": 1.5803, + "theoretical_loss": 4.248818229945202, + "tokens_seen": 244514816 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046766298896690076, + "loss": 1.5874, + "theoretical_loss": 4.248680173226621, + "tokens_seen": 244580352 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004676529588766299, + "loss": 1.5586, + "theoretical_loss": 4.248542163850566, + "tokens_seen": 244645888 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004676429287863591, + "loss": 1.5444, + "theoretical_loss": 4.248404201788124, + "tokens_seen": 244711424 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046763289869608825, + "loss": 1.6578, + "theoretical_loss": 4.248266287010411, + "tokens_seen": 244776960 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004676228686058175, + "loss": 1.4797, + "theoretical_loss": 4.248128419488569, + "tokens_seen": 244842496 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046761283851554666, + "loss": 1.682, + "theoretical_loss": 4.247990599193761, + "tokens_seen": 244908032 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046760280842527585, + "loss": 1.5814, + "theoretical_loss": 4.247852826097181, + "tokens_seen": 244973568 + }, + { + "epoch": 0.07, + "learning_rate": 0.000467592778335005, + "loss": 1.5981, + "theoretical_loss": 4.247715100170042, + "tokens_seen": 245039104 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004675827482447342, + "loss": 1.631, + "theoretical_loss": 4.247577421383589, + "tokens_seen": 245104640 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004675727181544634, + "loss": 1.579, + "theoretical_loss": 4.247439789709086, + "tokens_seen": 245170176 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004675626880641926, + "loss": 1.6035, + "theoretical_loss": 4.247302205117826, + "tokens_seen": 245235712 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046755265797392175, + "loss": 1.5037, + "theoretical_loss": 4.247164667581126, + "tokens_seen": 245301248 + }, + { + "epoch": 0.07, + "learning_rate": 0.000467542627883651, + "loss": 1.5891, + "theoretical_loss": 4.247027177070329, + "tokens_seen": 245366784 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046753259779338017, + "loss": 1.54, + "theoretical_loss": 4.2468897335568006, + "tokens_seen": 245432320 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046752256770310935, + "loss": 1.4595, + "theoretical_loss": 4.246752337011935, + "tokens_seen": 245497856 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046751253761283853, + "loss": 1.6064, + "theoretical_loss": 4.246614987407148, + "tokens_seen": 245563392 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004675025075225677, + "loss": 1.6706, + "theoretical_loss": 4.246477684713883, + "tokens_seen": 245628928 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004674924774322969, + "loss": 1.5135, + "theoretical_loss": 4.246340428903608, + "tokens_seen": 245694464 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.00026855801115743816, + "objective/train/docs_used": 150657, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.478126287460327, + "objective/train/original_loss": 3.4781265258789062, + "objective/train/theoretical_loss": 4.246203219947814, + "objective/train/tokens_used": 266220000, + "objective/train/value_avg": -0.0039825439453125, + "objective/train/value_loss": 5.522449646377936e-05, + "objective/train/value_max": -0.0002397298812866211, + "objective/train/value_min": -0.095703125, + "objective/train/value_reward_corr": 0.2194336755746952, + "objective/train/value_std": 0.0035419464111328125, + "objective/train/weight_avg": 0.9997106194496155, + "objective/train/weighted_lm_loss": 3.4783589839935303, + "objective/train/weights_max": 2.463991403579712, + "objective/train/weights_min": 0.1378861367702484, + "theoretical_loss": 4.246203219947814, + "tokens_seen": 245760000 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046748244734202613, + "loss": 1.6674, + "theoretical_loss": 4.246203219947814, + "tokens_seen": 245760000 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046747241725175525, + "loss": 1.659, + "theoretical_loss": 4.24606605781802, + "tokens_seen": 245825536 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004674623871614845, + "loss": 1.6642, + "theoretical_loss": 4.245928942485766, + "tokens_seen": 245891072 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004674523570712136, + "loss": 1.4886, + "theoretical_loss": 4.245791873922621, + "tokens_seen": 245956608 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046744232698094285, + "loss": 1.4802, + "theoretical_loss": 4.245654852100177, + "tokens_seen": 246022144 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046743229689067203, + "loss": 1.5956, + "theoretical_loss": 4.245517876990049, + "tokens_seen": 246087680 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004674222668004012, + "loss": 1.6084, + "theoretical_loss": 4.245380948563881, + "tokens_seen": 246153216 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004674122367101304, + "loss": 1.5476, + "theoretical_loss": 4.245244066793337, + "tokens_seen": 246218752 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046740220661985963, + "loss": 1.5418, + "theoretical_loss": 4.24510723165011, + "tokens_seen": 246284288 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046739217652958876, + "loss": 1.5534, + "theoretical_loss": 4.244970443105913, + "tokens_seen": 246349824 + }, + { + "epoch": 0.07, + "learning_rate": 0.000467382146439318, + "loss": 1.6306, + "theoretical_loss": 4.24483370113249, + "tokens_seen": 246415360 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004673721163490471, + "loss": 1.5661, + "theoretical_loss": 4.244697005701604, + "tokens_seen": 246480896 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046736208625877636, + "loss": 1.5984, + "theoretical_loss": 4.244560356785045, + "tokens_seen": 246546432 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046735205616850554, + "loss": 1.6207, + "theoretical_loss": 4.244423754354627, + "tokens_seen": 246611968 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004673420260782347, + "loss": 1.6431, + "theoretical_loss": 4.24428719838219, + "tokens_seen": 246677504 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004673319959879639, + "loss": 1.6212, + "theoretical_loss": 4.2441506888395955, + "tokens_seen": 246743040 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004673219658976931, + "loss": 1.4733, + "theoretical_loss": 4.2440142256987325, + "tokens_seen": 246808576 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046731193580742226, + "loss": 1.6933, + "theoretical_loss": 4.243877808931515, + "tokens_seen": 246874112 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004673019057171515, + "loss": 1.5764, + "theoretical_loss": 4.243741438509876, + "tokens_seen": 246939648 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004672918756268806, + "loss": 1.5803, + "theoretical_loss": 4.243605114405781, + "tokens_seen": 247005184 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046728184553660986, + "loss": 1.5202, + "theoretical_loss": 4.243468836591212, + "tokens_seen": 247070720 + }, + { + "epoch": 0.07, + "learning_rate": 0.000467271815446339, + "loss": 1.5578, + "theoretical_loss": 4.243332605038182, + "tokens_seen": 247136256 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004672617853560682, + "loss": 1.492, + "theoretical_loss": 4.243196419718723, + "tokens_seen": 247201792 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004672517552657974, + "loss": 1.6684, + "theoretical_loss": 4.243060280604896, + "tokens_seen": 247267328 + }, + { + "epoch": 0.07, + "learning_rate": 0.0004672417251755266, + "loss": 1.8248, + "theoretical_loss": 4.242924187668781, + "tokens_seen": 247332864 + }, + { + "epoch": 0.07, + "objective/train/advantage_avg": -0.0006330063915811479, + "objective/train/docs_used": 151862, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.396022319793701, + "objective/train/original_loss": 3.396022319793701, + "objective/train/theoretical_loss": 4.242788140882488, + "objective/train/tokens_used": 267858400, + "objective/train/value_avg": -0.004589080810546875, + "objective/train/value_loss": 5.8797475503524765e-05, + "objective/train/value_max": -0.00019407272338867188, + "objective/train/value_min": -0.05865478515625, + "objective/train/value_reward_corr": 0.24393054070217196, + "objective/train/value_std": 0.003997802734375, + "objective/train/weight_avg": 0.9963887929916382, + "objective/train/weighted_lm_loss": 3.40251088142395, + "objective/train/weights_max": 1.4959681034088135, + "objective/train/weights_min": 0.41442546248435974, + "theoretical_loss": 4.242788140882488, + "tokens_seen": 247398400 + }, + { + "epoch": 0.07, + "learning_rate": 0.00046723169508525576, + "loss": 1.5678, + "theoretical_loss": 4.242788140882488, + "tokens_seen": 247398400 + }, + { + "epoch": 0.07, + "learning_rate": 0.000467221664994985, + "loss": 1.5645, + "theoretical_loss": 4.242652140218147, + "tokens_seen": 247463936 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004672116349047141, + "loss": 1.4727, + "theoretical_loss": 4.242516185647914, + "tokens_seen": 247529472 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046720160481444336, + "loss": 1.6939, + "theoretical_loss": 4.242380277143969, + "tokens_seen": 247595008 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004671915747241725, + "loss": 1.6762, + "theoretical_loss": 4.242244414678515, + "tokens_seen": 247660544 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004671815446339017, + "loss": 1.656, + "theoretical_loss": 4.242108598223782, + "tokens_seen": 247726080 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004671715145436309, + "loss": 1.6033, + "theoretical_loss": 4.24197282775202, + "tokens_seen": 247791616 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004671614844533601, + "loss": 1.6121, + "theoretical_loss": 4.2418371032355076, + "tokens_seen": 247857152 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046715145436308927, + "loss": 1.7267, + "theoretical_loss": 4.241701424646544, + "tokens_seen": 247922688 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046714142427281845, + "loss": 1.5927, + "theoretical_loss": 4.241565791957452, + "tokens_seen": 247988224 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046713139418254763, + "loss": 1.6019, + "theoretical_loss": 4.241430205140583, + "tokens_seen": 248053760 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046712136409227686, + "loss": 1.6572, + "theoretical_loss": 4.241294664168306, + "tokens_seen": 248119296 + }, + { + "epoch": 0.08, + "learning_rate": 0.000467111334002006, + "loss": 1.6221, + "theoretical_loss": 4.24115916901302, + "tokens_seen": 248184832 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004671013039117352, + "loss": 1.644, + "theoretical_loss": 4.241023719647143, + "tokens_seen": 248250368 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046709127382146435, + "loss": 1.5783, + "theoretical_loss": 4.2408883160431206, + "tokens_seen": 248315904 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004670812437311936, + "loss": 1.5214, + "theoretical_loss": 4.240752958173419, + "tokens_seen": 248381440 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046707121364092277, + "loss": 1.6188, + "theoretical_loss": 4.240617646010532, + "tokens_seen": 248446976 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046706118355065195, + "loss": 1.5629, + "theoretical_loss": 4.240482379526973, + "tokens_seen": 248512512 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046705115346038113, + "loss": 1.5055, + "theoretical_loss": 4.240347158695281, + "tokens_seen": 248578048 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046704112337011037, + "loss": 1.5546, + "theoretical_loss": 4.240211983488021, + "tokens_seen": 248643584 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004670310932798395, + "loss": 1.7687, + "theoretical_loss": 4.240076853877778, + "tokens_seen": 248709120 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046702106318956873, + "loss": 1.5994, + "theoretical_loss": 4.239941769837165, + "tokens_seen": 248774656 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046701103309929786, + "loss": 1.6298, + "theoretical_loss": 4.2398067313388115, + "tokens_seen": 248840192 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004670010030090271, + "loss": 1.6293, + "theoretical_loss": 4.239671738355379, + "tokens_seen": 248905728 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004669909729187563, + "loss": 1.5338, + "theoretical_loss": 4.239536790859547, + "tokens_seen": 248971264 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": 0.0010563047835603356, + "objective/train/docs_used": 152583, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1286203861236572, + "objective/train/original_loss": 3.128620147705078, + "objective/train/theoretical_loss": 4.2394018888240215, + "objective/train/tokens_used": 269496800, + "objective/train/value_avg": -0.0050048828125, + "objective/train/value_loss": 7.66580633353442e-05, + "objective/train/value_max": -0.0004372596740722656, + "objective/train/value_min": -0.054901123046875, + "objective/train/value_reward_corr": 0.25034497125568567, + "objective/train/value_std": 0.003963470458984375, + "objective/train/weight_avg": 1.0138503313064575, + "objective/train/weighted_lm_loss": 3.1635947227478027, + "objective/train/weights_max": 1.7315400838851929, + "objective/train/weights_min": 0.19603650271892548, + "theoretical_loss": 4.2394018888240215, + "tokens_seen": 249036800 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046698094282848545, + "loss": 1.5911, + "theoretical_loss": 4.2394018888240215, + "tokens_seen": 249036800 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046697091273821464, + "loss": 1.6224, + "theoretical_loss": 4.239267032221529, + "tokens_seen": 249102336 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004669608826479438, + "loss": 1.5941, + "theoretical_loss": 4.239132221024823, + "tokens_seen": 249167872 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046695085255767305, + "loss": 1.6314, + "theoretical_loss": 4.238997455206679, + "tokens_seen": 249233408 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046694082246740223, + "loss": 1.5469, + "theoretical_loss": 4.238862734739896, + "tokens_seen": 249298944 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004669307923771314, + "loss": 1.5937, + "theoretical_loss": 4.238728059597294, + "tokens_seen": 249364480 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004669207622868606, + "loss": 1.6939, + "theoretical_loss": 4.238593429751722, + "tokens_seen": 249430016 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046691073219658983, + "loss": 1.4945, + "theoretical_loss": 4.238458845176048, + "tokens_seen": 249495552 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046690070210631896, + "loss": 1.5648, + "theoretical_loss": 4.2383243058431646, + "tokens_seen": 249561088 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004668906720160482, + "loss": 1.6973, + "theoretical_loss": 4.238189811725987, + "tokens_seen": 249626624 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004668806419257773, + "loss": 1.5797, + "theoretical_loss": 4.238055362797455, + "tokens_seen": 249692160 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046687061183550656, + "loss": 1.611, + "theoretical_loss": 4.2379209590305305, + "tokens_seen": 249757696 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046686058174523574, + "loss": 1.7537, + "theoretical_loss": 4.2377866003982, + "tokens_seen": 249823232 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004668505516549649, + "loss": 1.7391, + "theoretical_loss": 4.237652286873473, + "tokens_seen": 249888768 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004668405215646941, + "loss": 1.6716, + "theoretical_loss": 4.237518018429381, + "tokens_seen": 249954304 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004668304914744233, + "loss": 1.4797, + "theoretical_loss": 4.23738379503898, + "tokens_seen": 250019840 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046682046138415246, + "loss": 1.6167, + "theoretical_loss": 4.2372496166753475, + "tokens_seen": 250085376 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004668104312938817, + "loss": 1.4992, + "theoretical_loss": 4.237115483311586, + "tokens_seen": 250150912 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004668004012036108, + "loss": 1.6426, + "theoretical_loss": 4.23698139492082, + "tokens_seen": 250216448 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046679037111334006, + "loss": 1.5402, + "theoretical_loss": 4.2368473514761975, + "tokens_seen": 250281984 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004667803410230692, + "loss": 1.5273, + "theoretical_loss": 4.23671335295089, + "tokens_seen": 250347520 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004667703109327984, + "loss": 1.723, + "theoretical_loss": 4.2365793993180905, + "tokens_seen": 250413056 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004667602808425276, + "loss": 1.4434, + "theoretical_loss": 4.2364454905510165, + "tokens_seen": 250478592 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004667502507522568, + "loss": 1.6294, + "theoretical_loss": 4.236311626622909, + "tokens_seen": 250544128 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046674022066198596, + "loss": 1.5951, + "theoretical_loss": 4.23617780750703, + "tokens_seen": 250609664 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": 0.0009797430830076337, + "objective/train/docs_used": 153242, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.135709047317505, + "objective/train/original_loss": 3.135709285736084, + "objective/train/theoretical_loss": 4.236044033176665, + "objective/train/tokens_used": 271135200, + "objective/train/value_avg": -0.004596710205078125, + "objective/train/value_loss": 3.242048478568904e-05, + "objective/train/value_max": -0.0002359151840209961, + "objective/train/value_min": -0.06597900390625, + "objective/train/value_reward_corr": 0.1549496738297232, + "objective/train/value_std": 0.0035037994384765625, + "objective/train/weight_avg": 1.011414885520935, + "objective/train/weighted_lm_loss": 3.1743392944335938, + "objective/train/weights_max": 1.8476572036743164, + "objective/train/weights_min": 0.6915006637573242, + "theoretical_loss": 4.236044033176665, + "tokens_seen": 250675200 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004667301905717152, + "loss": 1.6072, + "theoretical_loss": 4.236044033176665, + "tokens_seen": 250675200 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004667201604814443, + "loss": 1.5058, + "theoretical_loss": 4.235910303605123, + "tokens_seen": 250740736 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046671013039117356, + "loss": 1.5809, + "theoretical_loss": 4.235776618765736, + "tokens_seen": 250806272 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004667001003009027, + "loss": 1.564, + "theoretical_loss": 4.235642978631859, + "tokens_seen": 250871808 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004666900702106319, + "loss": 1.499, + "theoretical_loss": 4.2355093831768675, + "tokens_seen": 250937344 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004666800401203611, + "loss": 1.653, + "theoretical_loss": 4.235375832374164, + "tokens_seen": 251002880 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004666700100300903, + "loss": 1.6148, + "theoretical_loss": 4.23524232619717, + "tokens_seen": 251068416 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046665997993981947, + "loss": 1.7161, + "theoretical_loss": 4.235108864619331, + "tokens_seen": 251133952 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046664994984954865, + "loss": 1.5342, + "theoretical_loss": 4.234975447614117, + "tokens_seen": 251199488 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046663991975927783, + "loss": 1.4945, + "theoretical_loss": 4.234842075155017, + "tokens_seen": 251265024 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046662988966900706, + "loss": 1.5712, + "theoretical_loss": 4.234708747215546, + "tokens_seen": 251330560 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004666198595787362, + "loss": 1.5101, + "theoretical_loss": 4.23457546376924, + "tokens_seen": 251396096 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046660982948846543, + "loss": 1.6629, + "theoretical_loss": 4.2344422247896585, + "tokens_seen": 251461632 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046659979939819455, + "loss": 1.5776, + "theoretical_loss": 4.234309030250383, + "tokens_seen": 251527168 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004665897693079238, + "loss": 1.6464, + "theoretical_loss": 4.234175880125018, + "tokens_seen": 251592704 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046657973921765297, + "loss": 1.7555, + "theoretical_loss": 4.23404277438719, + "tokens_seen": 251658240 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046656970912738215, + "loss": 1.5932, + "theoretical_loss": 4.233909713010547, + "tokens_seen": 251723776 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046655967903711133, + "loss": 1.7103, + "theoretical_loss": 4.233776695968762, + "tokens_seen": 251789312 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046654964894684057, + "loss": 1.7649, + "theoretical_loss": 4.23364372323553, + "tokens_seen": 251854848 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004665396188565697, + "loss": 1.5712, + "theoretical_loss": 4.233510794784567, + "tokens_seen": 251920384 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046652958876629893, + "loss": 1.6125, + "theoretical_loss": 4.233377910589612, + "tokens_seen": 251985920 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046651955867602806, + "loss": 1.5307, + "theoretical_loss": 4.233245070624426, + "tokens_seen": 252051456 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004665095285857573, + "loss": 1.8415, + "theoretical_loss": 4.233112274862793, + "tokens_seen": 252116992 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004664994984954865, + "loss": 1.7538, + "theoretical_loss": 4.232979523278521, + "tokens_seen": 252182528 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046648946840521565, + "loss": 1.7573, + "theoretical_loss": 4.232846815845436, + "tokens_seen": 252248064 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.001624198746867478, + "objective/train/docs_used": 154565, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3259193897247314, + "objective/train/original_loss": 3.325918674468994, + "objective/train/theoretical_loss": 4.232714152537391, + "objective/train/tokens_used": 272773600, + "objective/train/value_avg": -0.005435943603515625, + "objective/train/value_loss": 0.00011724854266503826, + "objective/train/value_max": -0.00022518634796142578, + "objective/train/value_min": -0.1300048828125, + "objective/train/value_reward_corr": 0.2784036715163959, + "objective/train/value_std": 0.00466156005859375, + "objective/train/weight_avg": 0.9889499545097351, + "objective/train/weighted_lm_loss": 3.2925877571105957, + "objective/train/weights_max": 3.167649984359741, + "objective/train/weights_min": 0.1446286141872406, + "theoretical_loss": 4.232714152537391, + "tokens_seen": 252313600 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046647943831494484, + "loss": 1.7932, + "theoretical_loss": 4.232714152537391, + "tokens_seen": 252313600 + }, + { + "epoch": 0.08, + "learning_rate": 0.000466469408224674, + "loss": 1.7703, + "theoretical_loss": 4.232581533328259, + "tokens_seen": 252379136 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004664593781344032, + "loss": 1.6368, + "theoretical_loss": 4.2324489581919345, + "tokens_seen": 252444672 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046644934804413243, + "loss": 1.8097, + "theoretical_loss": 4.232316427102335, + "tokens_seen": 252510208 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046643931795386156, + "loss": 1.7439, + "theoretical_loss": 4.2321839400334005, + "tokens_seen": 252575744 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004664292878635908, + "loss": 1.8694, + "theoretical_loss": 4.232051496959093, + "tokens_seen": 252641280 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004664192577733199, + "loss": 1.6915, + "theoretical_loss": 4.231919097853398, + "tokens_seen": 252706816 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046640922768304916, + "loss": 1.6563, + "theoretical_loss": 4.2317867426903195, + "tokens_seen": 252772352 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046639919759277834, + "loss": 1.7187, + "theoretical_loss": 4.2316544314438875, + "tokens_seen": 252837888 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004663891675025075, + "loss": 1.6029, + "theoretical_loss": 4.231522164088151, + "tokens_seen": 252903424 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004663791374122367, + "loss": 1.5901, + "theoretical_loss": 4.231389940597185, + "tokens_seen": 252968960 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046636910732196594, + "loss": 1.6708, + "theoretical_loss": 4.231257760945081, + "tokens_seen": 253034496 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046635907723169506, + "loss": 1.6639, + "theoretical_loss": 4.231125625105959, + "tokens_seen": 253100032 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004663490471414243, + "loss": 1.5665, + "theoretical_loss": 4.230993533053954, + "tokens_seen": 253165568 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004663390170511534, + "loss": 1.6368, + "theoretical_loss": 4.230861484763229, + "tokens_seen": 253231104 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046632898696088266, + "loss": 1.6358, + "theoretical_loss": 4.230729480207964, + "tokens_seen": 253296640 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046631895687061184, + "loss": 1.545, + "theoretical_loss": 4.230597519362366, + "tokens_seen": 253362176 + }, + { + "epoch": 0.08, + "learning_rate": 0.000466308926780341, + "loss": 1.6223, + "theoretical_loss": 4.23046560220066, + "tokens_seen": 253427712 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004662988966900702, + "loss": 1.542, + "theoretical_loss": 4.230333728697093, + "tokens_seen": 253493248 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004662888665997994, + "loss": 1.5479, + "theoretical_loss": 4.2302018988259364, + "tokens_seen": 253558784 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046627883650952857, + "loss": 1.5618, + "theoretical_loss": 4.2300701125614815, + "tokens_seen": 253624320 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004662688064192578, + "loss": 1.6204, + "theoretical_loss": 4.229938369878042, + "tokens_seen": 253689856 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046625877632898693, + "loss": 1.5579, + "theoretical_loss": 4.2298066707499515, + "tokens_seen": 253755392 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046624874623871616, + "loss": 1.6887, + "theoretical_loss": 4.229675015151569, + "tokens_seen": 253820928 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046623871614844535, + "loss": 1.6227, + "theoretical_loss": 4.2295434030572725, + "tokens_seen": 253886464 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.0024841567501425743, + "objective/train/docs_used": 156106, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.388544797897339, + "objective/train/original_loss": 3.3885445594787598, + "objective/train/theoretical_loss": 4.229411834441462, + "objective/train/tokens_used": 274412000, + "objective/train/value_avg": -0.004657745361328125, + "objective/train/value_loss": 7.463053043466061e-05, + "objective/train/value_max": -0.00027370452880859375, + "objective/train/value_min": -0.08251953125, + "objective/train/value_reward_corr": 0.40008330411164145, + "objective/train/value_std": 0.00450897216796875, + "objective/train/weight_avg": 0.9786760210990906, + "objective/train/weighted_lm_loss": 3.3079473972320557, + "objective/train/weights_max": 2.224255323410034, + "objective/train/weights_min": 0.4701083302497864, + "theoretical_loss": 4.229411834441462, + "tokens_seen": 253952000 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004662286860581745, + "loss": 1.6937, + "theoretical_loss": 4.229411834441462, + "tokens_seen": 253952000 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004662186559679037, + "loss": 1.591, + "theoretical_loss": 4.229280309278561, + "tokens_seen": 254017536 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004662086258776329, + "loss": 1.5924, + "theoretical_loss": 4.2291488275430105, + "tokens_seen": 254083072 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004661985957873621, + "loss": 1.682, + "theoretical_loss": 4.229017389209279, + "tokens_seen": 254148608 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004661885656970913, + "loss": 1.639, + "theoretical_loss": 4.228885994251851, + "tokens_seen": 254214144 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004661785356068205, + "loss": 1.7244, + "theoretical_loss": 4.228754642645235, + "tokens_seen": 254279680 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046616850551654967, + "loss": 1.5995, + "theoretical_loss": 4.228623334363962, + "tokens_seen": 254345216 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046615847542627885, + "loss": 1.5819, + "theoretical_loss": 4.228492069382583, + "tokens_seen": 254410752 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046614844533600803, + "loss": 1.4766, + "theoretical_loss": 4.228360847675672, + "tokens_seen": 254476288 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046613841524573727, + "loss": 1.6033, + "theoretical_loss": 4.228229669217823, + "tokens_seen": 254541824 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004661283851554664, + "loss": 1.655, + "theoretical_loss": 4.228098533983651, + "tokens_seen": 254607360 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046611835506519563, + "loss": 1.7153, + "theoretical_loss": 4.227967441947795, + "tokens_seen": 254672896 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046610832497492475, + "loss": 1.5667, + "theoretical_loss": 4.227836393084912, + "tokens_seen": 254738432 + }, + { + "epoch": 0.08, + "learning_rate": 0.000466098294884654, + "loss": 1.5491, + "theoretical_loss": 4.227705387369683, + "tokens_seen": 254803968 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046608826479438317, + "loss": 1.5874, + "theoretical_loss": 4.227574424776811, + "tokens_seen": 254869504 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046607823470411235, + "loss": 1.6225, + "theoretical_loss": 4.227443505281016, + "tokens_seen": 254935040 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046606820461384153, + "loss": 1.5815, + "theoretical_loss": 4.227312628857045, + "tokens_seen": 255000576 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046605817452357077, + "loss": 1.6674, + "theoretical_loss": 4.2271817954796616, + "tokens_seen": 255066112 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004660481444332999, + "loss": 1.5796, + "theoretical_loss": 4.227051005123653, + "tokens_seen": 255131648 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046603811434302913, + "loss": 1.5787, + "theoretical_loss": 4.226920257763827, + "tokens_seen": 255197184 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046602808425275826, + "loss": 1.5485, + "theoretical_loss": 4.226789553375013, + "tokens_seen": 255262720 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004660180541624875, + "loss": 1.5469, + "theoretical_loss": 4.226658891932061, + "tokens_seen": 255328256 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004660080240722167, + "loss": 1.5788, + "theoretical_loss": 4.226528273409844, + "tokens_seen": 255393792 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046599799398194586, + "loss": 1.5858, + "theoretical_loss": 4.226397697783252, + "tokens_seen": 255459328 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046598796389167504, + "loss": 1.5115, + "theoretical_loss": 4.226267165027202, + "tokens_seen": 255524864 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -9.361851698486134e-05, + "objective/train/docs_used": 156630, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.603975772857666, + "objective/train/original_loss": 2.603975772857666, + "objective/train/theoretical_loss": 4.226136675116626, + "objective/train/tokens_used": 276050400, + "objective/train/value_avg": -0.004230499267578125, + "objective/train/value_loss": 4.4958032958675176e-05, + "objective/train/value_max": -0.0002269744873046875, + "objective/train/value_min": -0.10687255859375, + "objective/train/value_reward_corr": 0.23408691984356833, + "objective/train/value_std": 0.003612518310546875, + "objective/train/weight_avg": 1.0011894702911377, + "objective/train/weighted_lm_loss": 2.612729549407959, + "objective/train/weights_max": 2.862401008605957, + "objective/train/weights_min": 0.35550495982170105, + "theoretical_loss": 4.226136675116626, + "tokens_seen": 255590400 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004659779338014042, + "loss": 1.5206, + "theoretical_loss": 4.226136675116626, + "tokens_seen": 255590400 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004659679037111334, + "loss": 1.5552, + "theoretical_loss": 4.226006228026482, + "tokens_seen": 255655936 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046595787362086263, + "loss": 1.6234, + "theoretical_loss": 4.225875823731746, + "tokens_seen": 255721472 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046594784353059176, + "loss": 1.4978, + "theoretical_loss": 4.225745462207417, + "tokens_seen": 255787008 + }, + { + "epoch": 0.08, + "learning_rate": 0.000465937813440321, + "loss": 1.7061, + "theoretical_loss": 4.225615143428513, + "tokens_seen": 255852544 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004659277833500501, + "loss": 1.7375, + "theoretical_loss": 4.225484867370074, + "tokens_seen": 255918080 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046591775325977936, + "loss": 1.5673, + "theoretical_loss": 4.225354634007163, + "tokens_seen": 255983616 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046590772316950854, + "loss": 1.4751, + "theoretical_loss": 4.225224443314859, + "tokens_seen": 256049152 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004658976930792377, + "loss": 1.6619, + "theoretical_loss": 4.225094295268267, + "tokens_seen": 256114688 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004658876629889669, + "loss": 1.5729, + "theoretical_loss": 4.22496418984251, + "tokens_seen": 256180224 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046587763289869614, + "loss": 1.6094, + "theoretical_loss": 4.224834127012732, + "tokens_seen": 256245760 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046586760280842526, + "loss": 1.4766, + "theoretical_loss": 4.224704106754101, + "tokens_seen": 256311296 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004658575727181545, + "loss": 1.4901, + "theoretical_loss": 4.224574129041801, + "tokens_seen": 256376832 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004658475426278836, + "loss": 1.4327, + "theoretical_loss": 4.224444193851039, + "tokens_seen": 256442368 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046583751253761286, + "loss": 1.5386, + "theoretical_loss": 4.224314301157044, + "tokens_seen": 256507904 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046582748244734204, + "loss": 1.4503, + "theoretical_loss": 4.224184450935065, + "tokens_seen": 256573440 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004658174523570712, + "loss": 1.5855, + "theoretical_loss": 4.22405464316037, + "tokens_seen": 256638976 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004658074222668004, + "loss": 1.5897, + "theoretical_loss": 4.223924877808249, + "tokens_seen": 256704512 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004657973921765296, + "loss": 1.6001, + "theoretical_loss": 4.223795154854015, + "tokens_seen": 256770048 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046578736208625877, + "loss": 1.6814, + "theoretical_loss": 4.223665474272996, + "tokens_seen": 256835584 + }, + { + "epoch": 0.08, + "learning_rate": 0.000465777331995988, + "loss": 1.6798, + "theoretical_loss": 4.223535836040548, + "tokens_seen": 256901120 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046576730190571713, + "loss": 1.6483, + "theoretical_loss": 4.223406240132041, + "tokens_seen": 256966656 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046575727181544636, + "loss": 1.5655, + "theoretical_loss": 4.223276686522869, + "tokens_seen": 257032192 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046574724172517555, + "loss": 1.456, + "theoretical_loss": 4.223147175188446, + "tokens_seen": 257097728 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004657372116349047, + "loss": 1.579, + "theoretical_loss": 4.223017706104207, + "tokens_seen": 257163264 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.0018058911664411426, + "objective/train/docs_used": 157258, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4619832038879395, + "objective/train/original_loss": 3.4619834423065186, + "objective/train/theoretical_loss": 4.2228882792456055, + "objective/train/tokens_used": 277688800, + "objective/train/value_avg": -0.004730224609375, + "objective/train/value_loss": 6.986790685914457e-05, + "objective/train/value_max": -0.00032258033752441406, + "objective/train/value_min": -0.055511474609375, + "objective/train/value_reward_corr": 0.23421314830923656, + "objective/train/value_std": 0.003925323486328125, + "objective/train/weight_avg": 0.9851995706558228, + "objective/train/weighted_lm_loss": 3.425145387649536, + "objective/train/weights_max": 1.5824897289276123, + "objective/train/weights_min": 0.29303178191185, + "theoretical_loss": 4.2228882792456055, + "tokens_seen": 257228800 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004657271815446339, + "loss": 1.5078, + "theoretical_loss": 4.2228882792456055, + "tokens_seen": 257228800 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004657171514543631, + "loss": 1.7033, + "theoretical_loss": 4.222758894588118, + "tokens_seen": 257294336 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046570712136409227, + "loss": 1.633, + "theoretical_loss": 4.22262955210724, + "tokens_seen": 257359872 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004656970912738215, + "loss": 1.6262, + "theoretical_loss": 4.222500251778489, + "tokens_seen": 257425408 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046568706118355063, + "loss": 1.5795, + "theoretical_loss": 4.2223709935774005, + "tokens_seen": 257490944 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046567703109327987, + "loss": 1.6319, + "theoretical_loss": 4.222241777479533, + "tokens_seen": 257556480 + }, + { + "epoch": 0.08, + "learning_rate": 0.000465667001003009, + "loss": 1.448, + "theoretical_loss": 4.222112603460462, + "tokens_seen": 257622016 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046565697091273823, + "loss": 1.5644, + "theoretical_loss": 4.221983471495789, + "tokens_seen": 257687552 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004656469408224674, + "loss": 1.5357, + "theoretical_loss": 4.22185438156113, + "tokens_seen": 257753088 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004656369107321966, + "loss": 1.4619, + "theoretical_loss": 4.221725333632123, + "tokens_seen": 257818624 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004656268806419258, + "loss": 1.5693, + "theoretical_loss": 4.221596327684429, + "tokens_seen": 257884160 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046561685055165495, + "loss": 1.4378, + "theoretical_loss": 4.221467363693727, + "tokens_seen": 257949696 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046560682046138414, + "loss": 1.572, + "theoretical_loss": 4.221338441635716, + "tokens_seen": 258015232 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046559679037111337, + "loss": 1.6003, + "theoretical_loss": 4.221209561486116, + "tokens_seen": 258080768 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004655867602808425, + "loss": 1.5257, + "theoretical_loss": 4.221080723220668, + "tokens_seen": 258146304 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046557673019057173, + "loss": 1.6004, + "theoretical_loss": 4.22095192681513, + "tokens_seen": 258211840 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004655667001003009, + "loss": 1.6334, + "theoretical_loss": 4.220823172245284, + "tokens_seen": 258277376 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004655566700100301, + "loss": 1.4935, + "theoretical_loss": 4.220694459486931, + "tokens_seen": 258342912 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004655466399197593, + "loss": 1.5991, + "theoretical_loss": 4.220565788515891, + "tokens_seen": 258408448 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046553660982948846, + "loss": 1.5714, + "theoretical_loss": 4.220437159308005, + "tokens_seen": 258473984 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046552657973921764, + "loss": 1.5416, + "theoretical_loss": 4.2203085718391335, + "tokens_seen": 258539520 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004655165496489469, + "loss": 1.6001, + "theoretical_loss": 4.220180026085158, + "tokens_seen": 258605056 + }, + { + "epoch": 0.08, + "learning_rate": 0.000465506519558676, + "loss": 1.6501, + "theoretical_loss": 4.220051522021979, + "tokens_seen": 258670592 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046549648946840524, + "loss": 1.5788, + "theoretical_loss": 4.219923059625518, + "tokens_seen": 258736128 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046548645937813436, + "loss": 1.5534, + "theoretical_loss": 4.219794638871717, + "tokens_seen": 258801664 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.0003851059009321034, + "objective/train/docs_used": 158510, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.698704957962036, + "objective/train/original_loss": 2.698704719543457, + "objective/train/theoretical_loss": 4.219666259736535, + "objective/train/tokens_used": 279327200, + "objective/train/value_avg": -0.004222869873046875, + "objective/train/value_loss": 3.92514411942102e-05, + "objective/train/value_max": -0.00022518634796142578, + "objective/train/value_min": -0.047149658203125, + "objective/train/value_reward_corr": 0.3580544728525609, + "objective/train/value_std": 0.0035762786865234375, + "objective/train/weight_avg": 0.9980652332305908, + "objective/train/weighted_lm_loss": 2.701373815536499, + "objective/train/weights_max": 1.4639885425567627, + "objective/train/weights_min": 0.6810554265975952, + "theoretical_loss": 4.219666259736535, + "tokens_seen": 258867200 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004654764292878636, + "loss": 1.621, + "theoretical_loss": 4.219666259736535, + "tokens_seen": 258867200 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004654663991975928, + "loss": 1.6176, + "theoretical_loss": 4.219537922195954, + "tokens_seen": 258932736 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046545636910732196, + "loss": 1.4905, + "theoretical_loss": 4.219409626225975, + "tokens_seen": 258998272 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004654463390170512, + "loss": 1.6272, + "theoretical_loss": 4.219281371802619, + "tokens_seen": 259063808 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004654363089267803, + "loss": 1.6235, + "theoretical_loss": 4.219153158901927, + "tokens_seen": 259129344 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046542627883650956, + "loss": 1.5855, + "theoretical_loss": 4.219024987499958, + "tokens_seen": 259194880 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046541624874623874, + "loss": 1.5631, + "theoretical_loss": 4.218896857572794, + "tokens_seen": 259260416 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004654062186559679, + "loss": 1.6556, + "theoretical_loss": 4.218768769096535, + "tokens_seen": 259325952 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004653961885656971, + "loss": 1.5299, + "theoretical_loss": 4.218640722047301, + "tokens_seen": 259391488 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046538615847542634, + "loss": 1.6026, + "theoretical_loss": 4.218512716401232, + "tokens_seen": 259457024 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046537612838515546, + "loss": 1.5776, + "theoretical_loss": 4.218384752134487, + "tokens_seen": 259522560 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004653660982948847, + "loss": 1.6489, + "theoretical_loss": 4.218256829223247, + "tokens_seen": 259588096 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004653560682046138, + "loss": 1.619, + "theoretical_loss": 4.2181289476437085, + "tokens_seen": 259653632 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046534603811434306, + "loss": 1.6432, + "theoretical_loss": 4.218001107372092, + "tokens_seen": 259719168 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046533600802407224, + "loss": 1.5323, + "theoretical_loss": 4.217873308384636, + "tokens_seen": 259784704 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004653259779338014, + "loss": 1.525, + "theoretical_loss": 4.2177455506575985, + "tokens_seen": 259850240 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004653159478435306, + "loss": 1.6356, + "theoretical_loss": 4.217617834167258, + "tokens_seen": 259915776 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004653059177532598, + "loss": 1.5159, + "theoretical_loss": 4.217490158889911, + "tokens_seen": 259981312 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046529588766298897, + "loss": 1.5841, + "theoretical_loss": 4.217362524801874, + "tokens_seen": 260046848 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004652858575727182, + "loss": 1.5196, + "theoretical_loss": 4.217234931879483, + "tokens_seen": 260112384 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046527582748244733, + "loss": 1.6409, + "theoretical_loss": 4.2171073800990975, + "tokens_seen": 260177920 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046526579739217656, + "loss": 1.5923, + "theoretical_loss": 4.21697986943709, + "tokens_seen": 260243456 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046525576730190575, + "loss": 1.4699, + "theoretical_loss": 4.216852399869857, + "tokens_seen": 260308992 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004652457372116349, + "loss": 1.5628, + "theoretical_loss": 4.216724971373813, + "tokens_seen": 260374528 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004652357071213641, + "loss": 1.5005, + "theoretical_loss": 4.216597583925391, + "tokens_seen": 260440064 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -6.928863876964897e-05, + "objective/train/docs_used": 159098, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.230801820755005, + "objective/train/original_loss": 3.230802059173584, + "objective/train/theoretical_loss": 4.216470237501046, + "objective/train/tokens_used": 280965600, + "objective/train/value_avg": -0.0051116943359375, + "objective/train/value_loss": 6.349608884193003e-05, + "objective/train/value_max": -0.0005769729614257812, + "objective/train/value_min": -0.07989501953125, + "objective/train/value_reward_corr": 0.30332606398493644, + "objective/train/value_std": 0.004302978515625, + "objective/train/weight_avg": 1.0023417472839355, + "objective/train/weighted_lm_loss": 3.226764440536499, + "objective/train/weights_max": 2.1239726543426514, + "objective/train/weights_min": 0.2548407018184662, + "theoretical_loss": 4.216470237501046, + "tokens_seen": 260505600 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004652256770310933, + "loss": 1.5017, + "theoretical_loss": 4.216470237501046, + "tokens_seen": 260505600 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046521564694082247, + "loss": 1.5931, + "theoretical_loss": 4.21634293207725, + "tokens_seen": 260571136 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004652056168505517, + "loss": 1.5952, + "theoretical_loss": 4.216215667630497, + "tokens_seen": 260636672 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046519558676028083, + "loss": 1.5548, + "theoretical_loss": 4.216088444137297, + "tokens_seen": 260702208 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046518555667001007, + "loss": 1.5434, + "theoretical_loss": 4.215961261574181, + "tokens_seen": 260767744 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004651755265797392, + "loss": 1.5938, + "theoretical_loss": 4.215834119917702, + "tokens_seen": 260833280 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046516549648946843, + "loss": 1.5735, + "theoretical_loss": 4.215707019144427, + "tokens_seen": 260898816 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004651554663991976, + "loss": 1.5414, + "theoretical_loss": 4.215579959230946, + "tokens_seen": 260964352 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004651454363089268, + "loss": 1.5027, + "theoretical_loss": 4.2154529401538685, + "tokens_seen": 261029888 + }, + { + "epoch": 0.08, + "learning_rate": 0.000465135406218656, + "loss": 1.7147, + "theoretical_loss": 4.215325961889821, + "tokens_seen": 261095424 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046512537612838515, + "loss": 1.667, + "theoretical_loss": 4.215199024415451, + "tokens_seen": 261160960 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046511534603811434, + "loss": 1.6691, + "theoretical_loss": 4.215072127707424, + "tokens_seen": 261226496 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046510531594784357, + "loss": 1.6427, + "theoretical_loss": 4.214945271742426, + "tokens_seen": 261292032 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004650952858575727, + "loss": 1.4482, + "theoretical_loss": 4.214818456497161, + "tokens_seen": 261357568 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046508525576730193, + "loss": 1.5018, + "theoretical_loss": 4.214691681948352, + "tokens_seen": 261423104 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004650752256770311, + "loss": 1.6611, + "theoretical_loss": 4.214564948072743, + "tokens_seen": 261488640 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004650651955867603, + "loss": 1.571, + "theoretical_loss": 4.214438254847096, + "tokens_seen": 261554176 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004650551654964895, + "loss": 1.6058, + "theoretical_loss": 4.214311602248191, + "tokens_seen": 261619712 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046504513540621866, + "loss": 1.5198, + "theoretical_loss": 4.2141849902528286, + "tokens_seen": 261685248 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046503510531594784, + "loss": 1.5584, + "theoretical_loss": 4.214058418837828, + "tokens_seen": 261750784 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004650250752256771, + "loss": 1.6028, + "theoretical_loss": 4.213931887980027, + "tokens_seen": 261816320 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004650150451354062, + "loss": 1.4941, + "theoretical_loss": 4.213805397656283, + "tokens_seen": 261881856 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046500501504513544, + "loss": 1.6899, + "theoretical_loss": 4.213678947843472, + "tokens_seen": 261947392 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046499498495486456, + "loss": 1.6206, + "theoretical_loss": 4.213552538518489, + "tokens_seen": 262012928 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004649849548645938, + "loss": 1.6822, + "theoretical_loss": 4.2134261696582485, + "tokens_seen": 262078464 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.0013451252598315477, + "objective/train/docs_used": 160106, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0790252685546875, + "objective/train/original_loss": 3.0790255069732666, + "objective/train/theoretical_loss": 4.213299841239684, + "objective/train/tokens_used": 282604000, + "objective/train/value_avg": -0.00490570068359375, + "objective/train/value_loss": 7.244500739034265e-05, + "objective/train/value_max": -0.0002892017364501953, + "objective/train/value_min": -0.0526123046875, + "objective/train/value_reward_corr": 0.3063743511181274, + "objective/train/value_std": 0.003711700439453125, + "objective/train/weight_avg": 0.9897674322128296, + "objective/train/weighted_lm_loss": 3.045382261276245, + "objective/train/weights_max": 1.554847240447998, + "objective/train/weights_min": 0.1439460963010788, + "theoretical_loss": 4.213299841239684, + "tokens_seen": 262144000 + }, + { + "epoch": 0.08, + "learning_rate": 0.000464974924774323, + "loss": 1.6259, + "theoretical_loss": 4.213299841239684, + "tokens_seen": 262144000 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046496489468405216, + "loss": 1.5709, + "theoretical_loss": 4.213173553239747, + "tokens_seen": 262209536 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046495486459378134, + "loss": 1.6436, + "theoretical_loss": 4.213047305635407, + "tokens_seen": 262275072 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004649448345035105, + "loss": 1.6645, + "theoretical_loss": 4.212921098403656, + "tokens_seen": 262340608 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004649348044132397, + "loss": 1.6693, + "theoretical_loss": 4.2127949315215005, + "tokens_seen": 262406144 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046492477432296894, + "loss": 1.6412, + "theoretical_loss": 4.212668804965968, + "tokens_seen": 262471680 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046491474423269807, + "loss": 1.5581, + "theoretical_loss": 4.212542718714107, + "tokens_seen": 262537216 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004649047141424273, + "loss": 1.4651, + "theoretical_loss": 4.212416672742981, + "tokens_seen": 262602752 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004648946840521565, + "loss": 1.671, + "theoretical_loss": 4.212290667029672, + "tokens_seen": 262668288 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046488465396188566, + "loss": 1.6103, + "theoretical_loss": 4.212164701551285, + "tokens_seen": 262733824 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046487462387161485, + "loss": 1.6796, + "theoretical_loss": 4.212038776284941, + "tokens_seen": 262799360 + }, + { + "epoch": 0.08, + "learning_rate": 0.000464864593781344, + "loss": 1.5428, + "theoretical_loss": 4.211912891207779, + "tokens_seen": 262864896 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004648545636910732, + "loss": 1.4958, + "theoretical_loss": 4.2117870462969575, + "tokens_seen": 262930432 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046484453360080244, + "loss": 1.5451, + "theoretical_loss": 4.211661241529654, + "tokens_seen": 262995968 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046483450351053157, + "loss": 1.4646, + "theoretical_loss": 4.211535476883066, + "tokens_seen": 263061504 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004648244734202608, + "loss": 1.6156, + "theoretical_loss": 4.211409752334407, + "tokens_seen": 263127040 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046481444332998993, + "loss": 1.5043, + "theoretical_loss": 4.211284067860909, + "tokens_seen": 263192576 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046480441323971917, + "loss": 1.5657, + "theoretical_loss": 4.211158423439825, + "tokens_seen": 263258112 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046479438314944835, + "loss": 1.5807, + "theoretical_loss": 4.211032819048425, + "tokens_seen": 263323648 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046478435305917753, + "loss": 1.5551, + "theoretical_loss": 4.210907254663999, + "tokens_seen": 263389184 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004647743229689067, + "loss": 1.5649, + "theoretical_loss": 4.210781730263853, + "tokens_seen": 263454720 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046476429287863595, + "loss": 1.5975, + "theoretical_loss": 4.210656245825313, + "tokens_seen": 263520256 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004647542627883651, + "loss": 1.5807, + "theoretical_loss": 4.210530801325724, + "tokens_seen": 263585792 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004647442326980943, + "loss": 1.4979, + "theoretical_loss": 4.21040539674245, + "tokens_seen": 263651328 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046473420260782344, + "loss": 1.6118, + "theoretical_loss": 4.21028003205287, + "tokens_seen": 263716864 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.001008928520604968, + "objective/train/docs_used": 160649, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9749510288238525, + "objective/train/original_loss": 2.9749512672424316, + "objective/train/theoretical_loss": 4.210154707234386, + "objective/train/tokens_used": 284242400, + "objective/train/value_avg": -0.004566192626953125, + "objective/train/value_loss": 9.512479300610721e-05, + "objective/train/value_max": -0.0001926422119140625, + "objective/train/value_min": -0.0831298828125, + "objective/train/value_reward_corr": 0.21837199650030706, + "objective/train/value_std": 0.0044708251953125, + "objective/train/weight_avg": 0.9942463636398315, + "objective/train/weighted_lm_loss": 2.981112241744995, + "objective/train/weights_max": 2.296299457550049, + "objective/train/weights_min": 0.35143613815307617, + "theoretical_loss": 4.210154707234386, + "tokens_seen": 263782400 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046472417251755267, + "loss": 1.5206, + "theoretical_loss": 4.210154707234386, + "tokens_seen": 263782400 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004647141424272819, + "loss": 1.5868, + "theoretical_loss": 4.210029422264414, + "tokens_seen": 263847936 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046470411233701103, + "loss": 1.5356, + "theoretical_loss": 4.209904177120393, + "tokens_seen": 263913472 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046469408224674027, + "loss": 1.524, + "theoretical_loss": 4.209778971779777, + "tokens_seen": 263979008 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004646840521564694, + "loss": 1.6421, + "theoretical_loss": 4.209653806220039, + "tokens_seen": 264044544 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046467402206619863, + "loss": 1.5071, + "theoretical_loss": 4.2095286804186705, + "tokens_seen": 264110080 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004646639919759278, + "loss": 1.5745, + "theoretical_loss": 4.209403594353183, + "tokens_seen": 264175616 + }, + { + "epoch": 0.08, + "learning_rate": 0.000464653961885657, + "loss": 1.4754, + "theoretical_loss": 4.209278548001103, + "tokens_seen": 264241152 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004646439317953862, + "loss": 1.5645, + "theoretical_loss": 4.209153541339977, + "tokens_seen": 264306688 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046463390170511535, + "loss": 1.5299, + "theoretical_loss": 4.209028574347371, + "tokens_seen": 264372224 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046462387161484454, + "loss": 1.5752, + "theoretical_loss": 4.208903647000868, + "tokens_seen": 264437760 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046461384152457377, + "loss": 1.5914, + "theoretical_loss": 4.20877875927807, + "tokens_seen": 264503296 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004646038114343029, + "loss": 1.6628, + "theoretical_loss": 4.208653911156595, + "tokens_seen": 264568832 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046459378134403213, + "loss": 1.5136, + "theoretical_loss": 4.20852910261408, + "tokens_seen": 264634368 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004645837512537613, + "loss": 1.5711, + "theoretical_loss": 4.208404333628184, + "tokens_seen": 264699904 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004645737211634905, + "loss": 1.4165, + "theoretical_loss": 4.208279604176577, + "tokens_seen": 264765440 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004645636910732197, + "loss": 1.6383, + "theoretical_loss": 4.208154914236954, + "tokens_seen": 264830976 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046455366098294886, + "loss": 1.4598, + "theoretical_loss": 4.208030263787025, + "tokens_seen": 264896512 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046454363089267804, + "loss": 1.5482, + "theoretical_loss": 4.2079056528045164, + "tokens_seen": 264962048 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004645336008024073, + "loss": 1.6154, + "theoretical_loss": 4.207781081267177, + "tokens_seen": 265027584 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004645235707121364, + "loss": 1.4512, + "theoretical_loss": 4.207656549152769, + "tokens_seen": 265093120 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046451354062186564, + "loss": 1.6545, + "theoretical_loss": 4.207532056439075, + "tokens_seen": 265158656 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046450351053159476, + "loss": 1.5859, + "theoretical_loss": 4.207407603103897, + "tokens_seen": 265224192 + }, + { + "epoch": 0.08, + "learning_rate": 0.000464493480441324, + "loss": 1.6993, + "theoretical_loss": 4.207283189125054, + "tokens_seen": 265289728 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004644834503510532, + "loss": 1.5693, + "theoretical_loss": 4.20715881448038, + "tokens_seen": 265355264 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": 0.0008920444524846971, + "objective/train/docs_used": 161970, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8885281085968018, + "objective/train/original_loss": 2.888528347015381, + "objective/train/theoretical_loss": 4.20703447914773, + "objective/train/tokens_used": 285880800, + "objective/train/value_avg": -0.005092620849609375, + "objective/train/value_loss": 6.700577068841085e-05, + "objective/train/value_max": -0.00037407875061035156, + "objective/train/value_min": -0.07891845703125, + "objective/train/value_reward_corr": 0.24993333301695156, + "objective/train/value_std": 0.0044403076171875, + "objective/train/weight_avg": 1.0120338201522827, + "objective/train/weighted_lm_loss": 2.9115891456604004, + "objective/train/weights_max": 2.2016005516052246, + "objective/train/weights_min": 0.20103861391544342, + "theoretical_loss": 4.20703447914773, + "tokens_seen": 265420800 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046447342026078236, + "loss": 1.4665, + "theoretical_loss": 4.20703447914773, + "tokens_seen": 265420800 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046446339017051154, + "loss": 1.5452, + "theoretical_loss": 4.206910183104977, + "tokens_seen": 265486336 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004644533600802407, + "loss": 1.697, + "theoretical_loss": 4.20678592633001, + "tokens_seen": 265551872 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004644433299899699, + "loss": 1.5092, + "theoretical_loss": 4.206661708800739, + "tokens_seen": 265617408 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046443329989969914, + "loss": 1.574, + "theoretical_loss": 4.206537530495089, + "tokens_seen": 265682944 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046442326980942827, + "loss": 1.5478, + "theoretical_loss": 4.2064133913910045, + "tokens_seen": 265748480 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004644132397191575, + "loss": 1.5116, + "theoretical_loss": 4.206289291466445, + "tokens_seen": 265814016 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004644032096288867, + "loss": 1.7096, + "theoretical_loss": 4.206165230699392, + "tokens_seen": 265879552 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046439317953861586, + "loss": 1.4522, + "theoretical_loss": 4.206041209067842, + "tokens_seen": 265945088 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046438314944834505, + "loss": 1.4868, + "theoretical_loss": 4.20591722654981, + "tokens_seen": 266010624 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004643731193580742, + "loss": 1.4729, + "theoretical_loss": 4.20579328312333, + "tokens_seen": 266076160 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004643630892678034, + "loss": 1.6823, + "theoretical_loss": 4.2056693787664505, + "tokens_seen": 266141696 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046435305917753264, + "loss": 1.5122, + "theoretical_loss": 4.205545513457242, + "tokens_seen": 266207232 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046434302908726177, + "loss": 1.4735, + "theoretical_loss": 4.205421687173789, + "tokens_seen": 266272768 + }, + { + "epoch": 0.08, + "learning_rate": 0.000464332998996991, + "loss": 1.5777, + "theoretical_loss": 4.2052978998941954, + "tokens_seen": 266338304 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046432296890672013, + "loss": 1.4862, + "theoretical_loss": 4.205174151596583, + "tokens_seen": 266403840 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046431293881644937, + "loss": 1.525, + "theoretical_loss": 4.205050442259091, + "tokens_seen": 266469376 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046430290872617855, + "loss": 1.5975, + "theoretical_loss": 4.204926771859874, + "tokens_seen": 266534912 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046429287863590773, + "loss": 1.6129, + "theoretical_loss": 4.204803140377109, + "tokens_seen": 266600448 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004642828485456369, + "loss": 1.5983, + "theoretical_loss": 4.204679547788987, + "tokens_seen": 266665984 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046427281845536615, + "loss": 1.5685, + "theoretical_loss": 4.204555994073717, + "tokens_seen": 266731520 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004642627883650953, + "loss": 1.381, + "theoretical_loss": 4.204432479209526, + "tokens_seen": 266797056 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004642527582748245, + "loss": 1.5493, + "theoretical_loss": 4.204309003174659, + "tokens_seen": 266862592 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046424272818455364, + "loss": 1.4842, + "theoretical_loss": 4.204185565947378, + "tokens_seen": 266928128 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046423269809428287, + "loss": 1.5514, + "theoretical_loss": 4.2040621675059615, + "tokens_seen": 266993664 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": 5.243722989689559e-05, + "objective/train/docs_used": 162606, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9739370346069336, + "objective/train/original_loss": 2.9739372730255127, + "objective/train/theoretical_loss": 4.203938807828708, + "objective/train/tokens_used": 287519200, + "objective/train/value_avg": -0.004638671875, + "objective/train/value_loss": 3.761557672987692e-05, + "objective/train/value_max": -0.00021314620971679688, + "objective/train/value_min": -0.06622314453125, + "objective/train/value_reward_corr": 0.31746057272661704, + "objective/train/value_std": 0.003955841064453125, + "objective/train/weight_avg": 1.002349853515625, + "objective/train/weighted_lm_loss": 2.9886200428009033, + "objective/train/weights_max": 1.9391144514083862, + "objective/train/weights_min": 0.4736725389957428, + "theoretical_loss": 4.203938807828708, + "tokens_seen": 267059200 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046422266800401205, + "loss": 1.4302, + "theoretical_loss": 4.203938807828708, + "tokens_seen": 267059200 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046421263791374123, + "loss": 1.4717, + "theoretical_loss": 4.20381548689393, + "tokens_seen": 267124736 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004642026078234704, + "loss": 1.4967, + "theoretical_loss": 4.203692204679962, + "tokens_seen": 267190272 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004641925777331996, + "loss": 1.5679, + "theoretical_loss": 4.203568961165152, + "tokens_seen": 267255808 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004641825476429288, + "loss": 1.4696, + "theoretical_loss": 4.203445756327866, + "tokens_seen": 267321344 + }, + { + "epoch": 0.08, + "learning_rate": 0.000464172517552658, + "loss": 1.6602, + "theoretical_loss": 4.203322590146491, + "tokens_seen": 267386880 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046416248746238714, + "loss": 1.533, + "theoretical_loss": 4.203199462599425, + "tokens_seen": 267452416 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004641524573721164, + "loss": 1.5473, + "theoretical_loss": 4.203076373665089, + "tokens_seen": 267517952 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004641424272818455, + "loss": 1.4985, + "theoretical_loss": 4.202953323321919, + "tokens_seen": 267583488 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046413239719157474, + "loss": 1.4758, + "theoretical_loss": 4.202830311548369, + "tokens_seen": 267649024 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004641223671013039, + "loss": 1.5556, + "theoretical_loss": 4.2027073383229085, + "tokens_seen": 267714560 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004641123370110331, + "loss": 1.578, + "theoretical_loss": 4.2025844036240265, + "tokens_seen": 267780096 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004641023069207623, + "loss": 1.6226, + "theoretical_loss": 4.202461507430228, + "tokens_seen": 267845632 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004640922768304915, + "loss": 1.5124, + "theoretical_loss": 4.202338649720037, + "tokens_seen": 267911168 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046408224674022064, + "loss": 1.4916, + "theoretical_loss": 4.202215830471991, + "tokens_seen": 267976704 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004640722166499499, + "loss": 1.6945, + "theoretical_loss": 4.20209304966465, + "tokens_seen": 268042240 + }, + { + "epoch": 0.08, + "learning_rate": 0.000464062186559679, + "loss": 1.6196, + "theoretical_loss": 4.201970307276586, + "tokens_seen": 268107776 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046405215646940824, + "loss": 1.5688, + "theoretical_loss": 4.201847603286391, + "tokens_seen": 268173312 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004640421263791374, + "loss": 1.5336, + "theoretical_loss": 4.201724937672674, + "tokens_seen": 268238848 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004640320962888666, + "loss": 1.5105, + "theoretical_loss": 4.201602310414062, + "tokens_seen": 268304384 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004640220661985958, + "loss": 1.545, + "theoretical_loss": 4.2014797214891955, + "tokens_seen": 268369920 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046401203610832496, + "loss": 1.5193, + "theoretical_loss": 4.2013571708767365, + "tokens_seen": 268435456 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046400200601805414, + "loss": 1.5461, + "theoretical_loss": 4.20123465855536, + "tokens_seen": 268500992 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004639919759277834, + "loss": 1.4339, + "theoretical_loss": 4.201112184503762, + "tokens_seen": 268566528 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004639819458375125, + "loss": 1.3977, + "theoretical_loss": 4.200989748700653, + "tokens_seen": 268632064 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": 0.0004237874527461827, + "objective/train/docs_used": 163909, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3710663318634033, + "objective/train/original_loss": 3.3710663318634033, + "objective/train/theoretical_loss": 4.200867351124762, + "objective/train/tokens_used": 289157600, + "objective/train/value_avg": -0.005023956298828125, + "objective/train/value_loss": 6.233029125723988e-05, + "objective/train/value_max": -0.0002868175506591797, + "objective/train/value_min": -0.06719970703125, + "objective/train/value_reward_corr": 0.3944315731727345, + "objective/train/value_std": 0.00482940673828125, + "objective/train/weight_avg": 1.007068157196045, + "objective/train/weighted_lm_loss": 3.4082014560699463, + "objective/train/weights_max": 1.6374890804290771, + "objective/train/weights_min": 0.40586361289024353, + "theoretical_loss": 4.200867351124762, + "tokens_seen": 268697600 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046397191574724174, + "loss": 1.5896, + "theoretical_loss": 4.200867351124762, + "tokens_seen": 268697600 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004639618856569709, + "loss": 1.6472, + "theoretical_loss": 4.200744991754833, + "tokens_seen": 268763136 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004639518555667001, + "loss": 1.5676, + "theoretical_loss": 4.200622670569628, + "tokens_seen": 268828672 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046394182547642934, + "loss": 1.6763, + "theoretical_loss": 4.200500387547929, + "tokens_seen": 268894208 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046393179538615847, + "loss": 1.4581, + "theoretical_loss": 4.200378142668528, + "tokens_seen": 268959744 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004639217652958877, + "loss": 1.5866, + "theoretical_loss": 4.200255935910242, + "tokens_seen": 269025280 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004639117352056169, + "loss": 1.5242, + "theoretical_loss": 4.2001337672518995, + "tokens_seen": 269090816 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046390170511534606, + "loss": 1.5612, + "theoretical_loss": 4.200011636672347, + "tokens_seen": 269156352 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046389167502507525, + "loss": 1.5716, + "theoretical_loss": 4.199889544150449, + "tokens_seen": 269221888 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004638816449348044, + "loss": 1.5021, + "theoretical_loss": 4.199767489665086, + "tokens_seen": 269287424 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004638716148445336, + "loss": 1.5794, + "theoretical_loss": 4.199645473195155, + "tokens_seen": 269352960 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046386158475426284, + "loss": 1.6077, + "theoretical_loss": 4.199523494719571, + "tokens_seen": 269418496 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046385155466399197, + "loss": 1.6412, + "theoretical_loss": 4.199401554217266, + "tokens_seen": 269484032 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004638415245737212, + "loss": 1.5264, + "theoretical_loss": 4.1992796516671875, + "tokens_seen": 269549568 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046383149448345033, + "loss": 1.5218, + "theoretical_loss": 4.1991577870483, + "tokens_seen": 269615104 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046382146439317957, + "loss": 1.6732, + "theoretical_loss": 4.199035960339585, + "tokens_seen": 269680640 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046381143430290875, + "loss": 1.4938, + "theoretical_loss": 4.198914171520041, + "tokens_seen": 269746176 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046380140421263793, + "loss": 1.5463, + "theoretical_loss": 4.198792420568684, + "tokens_seen": 269811712 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004637913741223671, + "loss": 1.5762, + "theoretical_loss": 4.198670707464545, + "tokens_seen": 269877248 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046378134403209635, + "loss": 1.657, + "theoretical_loss": 4.198549032186672, + "tokens_seen": 269942784 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004637713139418255, + "loss": 1.4976, + "theoretical_loss": 4.19842739471413, + "tokens_seen": 270008320 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004637612838515547, + "loss": 1.4577, + "theoretical_loss": 4.198305795026004, + "tokens_seen": 270073856 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046375125376128384, + "loss": 1.5544, + "theoretical_loss": 4.198184233101388, + "tokens_seen": 270139392 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046374122367101307, + "loss": 1.6366, + "theoretical_loss": 4.1980627089194, + "tokens_seen": 270204928 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046373119358074225, + "loss": 1.6503, + "theoretical_loss": 4.197941222459171, + "tokens_seen": 270270464 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.0006233596941456199, + "objective/train/docs_used": 164570, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.5490896701812744, + "objective/train/original_loss": 3.5490896701812744, + "objective/train/theoretical_loss": 4.197819773699849, + "objective/train/tokens_used": 290796000, + "objective/train/value_avg": -0.005191802978515625, + "objective/train/value_loss": 6.625676905969158e-05, + "objective/train/value_max": -0.00042724609375, + "objective/train/value_min": -0.1632080078125, + "objective/train/value_reward_corr": 0.2462283254546004, + "objective/train/value_std": 0.004352569580078125, + "objective/train/weight_avg": 0.996935248374939, + "objective/train/weighted_lm_loss": 3.5486605167388916, + "objective/train/weights_max": 4.731579303741455, + "objective/train/weights_min": 0.344914972782135, + "theoretical_loss": 4.197819773699849, + "tokens_seen": 270336000 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046372116349047143, + "loss": 1.5548, + "theoretical_loss": 4.197819773699849, + "tokens_seen": 270336000 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004637111334002006, + "loss": 1.5924, + "theoretical_loss": 4.197698362620599, + "tokens_seen": 270401536 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004637011033099298, + "loss": 1.584, + "theoretical_loss": 4.197576989200604, + "tokens_seen": 270467072 + }, + { + "epoch": 0.08, + "learning_rate": 0.000463691073219659, + "loss": 1.6073, + "theoretical_loss": 4.19745565341906, + "tokens_seen": 270532608 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004636810431293882, + "loss": 1.5968, + "theoretical_loss": 4.197334355255182, + "tokens_seen": 270598144 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046367101303911734, + "loss": 1.6085, + "theoretical_loss": 4.197213094688202, + "tokens_seen": 270663680 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004636609829488466, + "loss": 1.5392, + "theoretical_loss": 4.197091871697366, + "tokens_seen": 270729216 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004636509528585757, + "loss": 1.6194, + "theoretical_loss": 4.196970686261939, + "tokens_seen": 270794752 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046364092276830494, + "loss": 1.588, + "theoretical_loss": 4.196849538361201, + "tokens_seen": 270860288 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004636308926780341, + "loss": 1.5422, + "theoretical_loss": 4.196728427974451, + "tokens_seen": 270925824 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004636208625877633, + "loss": 1.6758, + "theoretical_loss": 4.196607355080999, + "tokens_seen": 270991360 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004636108324974925, + "loss": 1.6254, + "theoretical_loss": 4.196486319660177, + "tokens_seen": 271056896 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004636008024072217, + "loss": 1.5406, + "theoretical_loss": 4.19636532169133, + "tokens_seen": 271122432 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046359077231695084, + "loss": 1.5202, + "theoretical_loss": 4.196244361153821, + "tokens_seen": 271187968 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004635807422266801, + "loss": 1.6678, + "theoretical_loss": 4.196123438027029, + "tokens_seen": 271253504 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004635707121364092, + "loss": 1.64, + "theoretical_loss": 4.1960025522903495, + "tokens_seen": 271319040 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046356068204613844, + "loss": 1.5705, + "theoretical_loss": 4.1958817039231935, + "tokens_seen": 271384576 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004635506519558676, + "loss": 1.5445, + "theoretical_loss": 4.195760892904988, + "tokens_seen": 271450112 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004635406218655968, + "loss": 1.5173, + "theoretical_loss": 4.195640119215179, + "tokens_seen": 271515648 + }, + { + "epoch": 0.08, + "learning_rate": 0.000463530591775326, + "loss": 1.5866, + "theoretical_loss": 4.195519382833226, + "tokens_seen": 271581184 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046352056168505516, + "loss": 1.6819, + "theoretical_loss": 4.195398683738605, + "tokens_seen": 271646720 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046351053159478434, + "loss": 1.5551, + "theoretical_loss": 4.19527802191081, + "tokens_seen": 271712256 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004635005015045136, + "loss": 1.4848, + "theoretical_loss": 4.195157397329349, + "tokens_seen": 271777792 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004634904714142427, + "loss": 1.4847, + "theoretical_loss": 4.1950368099737485, + "tokens_seen": 271843328 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046348044132397194, + "loss": 1.5558, + "theoretical_loss": 4.1949162598235485, + "tokens_seen": 271908864 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": 0.0005287969252094626, + "objective/train/docs_used": 165961, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4372849464416504, + "objective/train/original_loss": 3.4372851848602295, + "objective/train/theoretical_loss": 4.194795746858309, + "objective/train/tokens_used": 292434400, + "objective/train/value_avg": -0.00556182861328125, + "objective/train/value_loss": 7.154197373893112e-05, + "objective/train/value_max": -0.0004222393035888672, + "objective/train/value_min": -0.061431884765625, + "objective/train/value_reward_corr": 0.22851876451495773, + "objective/train/value_std": 0.004405975341796875, + "objective/train/weight_avg": 1.0084182024002075, + "objective/train/weighted_lm_loss": 3.474635124206543, + "objective/train/weights_max": 1.7694674730300903, + "objective/train/weights_min": 0.08528491854667664, + "theoretical_loss": 4.194795746858309, + "tokens_seen": 271974400 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046347041123370107, + "loss": 1.5501, + "theoretical_loss": 4.194795746858309, + "tokens_seen": 271974400 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004634603811434303, + "loss": 1.5802, + "theoretical_loss": 4.194675271057601, + "tokens_seen": 272039936 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004634503510531595, + "loss": 1.6696, + "theoretical_loss": 4.194554832401017, + "tokens_seen": 272105472 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046344032096288867, + "loss": 1.6214, + "theoretical_loss": 4.194434430868161, + "tokens_seen": 272171008 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046343029087261785, + "loss": 1.6561, + "theoretical_loss": 4.194314066438656, + "tokens_seen": 272236544 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004634202607823471, + "loss": 1.5502, + "theoretical_loss": 4.194193739092141, + "tokens_seen": 272302080 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004634102306920762, + "loss": 1.4982, + "theoretical_loss": 4.194073448808269, + "tokens_seen": 272367616 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046340020060180545, + "loss": 1.5686, + "theoretical_loss": 4.193953195566712, + "tokens_seen": 272433152 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046339017051153457, + "loss": 1.4416, + "theoretical_loss": 4.193832979347155, + "tokens_seen": 272498688 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004633801404212638, + "loss": 1.4929, + "theoretical_loss": 4.193712800129301, + "tokens_seen": 272564224 + }, + { + "epoch": 0.08, + "learning_rate": 0.000463370110330993, + "loss": 1.5458, + "theoretical_loss": 4.193592657892869, + "tokens_seen": 272629760 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046336008024072217, + "loss": 1.6166, + "theoretical_loss": 4.193472552617592, + "tokens_seen": 272695296 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046335005015045135, + "loss": 1.5572, + "theoretical_loss": 4.193352484283222, + "tokens_seen": 272760832 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046334002006018053, + "loss": 1.5771, + "theoretical_loss": 4.193232452869525, + "tokens_seen": 272826368 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004633299899699097, + "loss": 1.5457, + "theoretical_loss": 4.193112458356283, + "tokens_seen": 272891904 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046331995987963895, + "loss": 1.6965, + "theoretical_loss": 4.192992500723294, + "tokens_seen": 272957440 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004633099297893681, + "loss": 1.461, + "theoretical_loss": 4.192872579950373, + "tokens_seen": 273022976 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004632998996990973, + "loss": 1.5627, + "theoretical_loss": 4.192752696017349, + "tokens_seen": 273088512 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046328986960882644, + "loss": 1.563, + "theoretical_loss": 4.1926328489040685, + "tokens_seen": 273154048 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004632798395185557, + "loss": 1.5194, + "theoretical_loss": 4.192513038590393, + "tokens_seen": 273219584 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046326980942828485, + "loss": 1.5676, + "theoretical_loss": 4.192393265056201, + "tokens_seen": 273285120 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046325977933801404, + "loss": 1.5868, + "theoretical_loss": 4.192273528281385, + "tokens_seen": 273350656 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004632497492477432, + "loss": 1.4889, + "theoretical_loss": 4.1921538282458535, + "tokens_seen": 273416192 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046323971915747245, + "loss": 1.5598, + "theoretical_loss": 4.192034164929534, + "tokens_seen": 273481728 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004632296890672016, + "loss": 1.6009, + "theoretical_loss": 4.191914538312365, + "tokens_seen": 273547264 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.0011514191282913089, + "objective/train/docs_used": 166474, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.464843988418579, + "objective/train/original_loss": 3.464843988418579, + "objective/train/theoretical_loss": 4.191794948374304, + "objective/train/tokens_used": 294072800, + "objective/train/value_avg": -0.005741119384765625, + "objective/train/value_loss": 7.605633436469361e-05, + "objective/train/value_max": -0.0004458427429199219, + "objective/train/value_min": -0.1387939453125, + "objective/train/value_reward_corr": 0.36453326848666395, + "objective/train/value_std": 0.00498199462890625, + "objective/train/weight_avg": 0.9921438694000244, + "objective/train/weighted_lm_loss": 3.4331793785095215, + "objective/train/weights_max": 3.593920946121216, + "objective/train/weights_min": 0.40881985425949097, + "theoretical_loss": 4.191794948374304, + "tokens_seen": 273612800 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004632196589769308, + "loss": 1.5548, + "theoretical_loss": 4.191794948374304, + "tokens_seen": 273612800 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046320962888666, + "loss": 1.5273, + "theoretical_loss": 4.191675395095324, + "tokens_seen": 273678336 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004631995987963892, + "loss": 1.5207, + "theoretical_loss": 4.191555878455411, + "tokens_seen": 273743872 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004631895687061184, + "loss": 1.5699, + "theoretical_loss": 4.191436398434571, + "tokens_seen": 273809408 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046317953861584754, + "loss": 1.6154, + "theoretical_loss": 4.191316955012822, + "tokens_seen": 273874944 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004631695085255768, + "loss": 1.6123, + "theoretical_loss": 4.1911975481702, + "tokens_seen": 273940480 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004631594784353059, + "loss": 1.5339, + "theoretical_loss": 4.191078177886755, + "tokens_seen": 274006016 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046314944834503514, + "loss": 1.5798, + "theoretical_loss": 4.190958844142555, + "tokens_seen": 274071552 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004631394182547643, + "loss": 1.5444, + "theoretical_loss": 4.1908395469176805, + "tokens_seen": 274137088 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004631293881644935, + "loss": 1.5764, + "theoretical_loss": 4.19072028619223, + "tokens_seen": 274202624 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004631193580742227, + "loss": 1.5393, + "theoretical_loss": 4.190601061946316, + "tokens_seen": 274268160 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004631093279839519, + "loss": 1.5641, + "theoretical_loss": 4.190481874160068, + "tokens_seen": 274333696 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046309929789368104, + "loss": 1.6653, + "theoretical_loss": 4.19036272281363, + "tokens_seen": 274399232 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004630892678034103, + "loss": 1.5484, + "theoretical_loss": 4.190243607887163, + "tokens_seen": 274464768 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004630792377131394, + "loss": 1.6263, + "theoretical_loss": 4.190124529360842, + "tokens_seen": 274530304 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046306920762286864, + "loss": 1.6148, + "theoretical_loss": 4.190005487214857, + "tokens_seen": 274595840 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004630591775325978, + "loss": 1.6259, + "theoretical_loss": 4.189886481429417, + "tokens_seen": 274661376 + }, + { + "epoch": 0.08, + "learning_rate": 0.000463049147442327, + "loss": 1.6045, + "theoretical_loss": 4.189767511984741, + "tokens_seen": 274726912 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004630391173520562, + "loss": 1.5139, + "theoretical_loss": 4.189648578861069, + "tokens_seen": 274792448 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046302908726178536, + "loss": 1.6311, + "theoretical_loss": 4.1895296820386525, + "tokens_seen": 274857984 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046301905717151455, + "loss": 1.4737, + "theoretical_loss": 4.189410821497761, + "tokens_seen": 274923520 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004630090270812438, + "loss": 1.5417, + "theoretical_loss": 4.189291997218677, + "tokens_seen": 274989056 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004629989969909729, + "loss": 1.7162, + "theoretical_loss": 4.1891732091817016, + "tokens_seen": 275054592 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046298896690070214, + "loss": 1.5958, + "theoretical_loss": 4.189054457367148, + "tokens_seen": 275120128 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046297893681043127, + "loss": 1.5551, + "theoretical_loss": 4.188935741755347, + "tokens_seen": 275185664 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.0008209240040741861, + "objective/train/docs_used": 167916, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.5386571884155273, + "objective/train/original_loss": 3.5386569499969482, + "objective/train/theoretical_loss": 4.188817062326644, + "objective/train/tokens_used": 295711200, + "objective/train/value_avg": -0.00554656982421875, + "objective/train/value_loss": 6.72917885822244e-05, + "objective/train/value_max": -0.0002269744873046875, + "objective/train/value_min": -0.07806396484375, + "objective/train/value_reward_corr": 0.3297603560755826, + "objective/train/value_std": 0.005107879638671875, + "objective/train/weight_avg": 0.9949719905853271, + "objective/train/weighted_lm_loss": 3.539189100265503, + "objective/train/weights_max": 1.893621563911438, + "objective/train/weights_min": 0.24572381377220154, + "theoretical_loss": 4.188817062326644, + "tokens_seen": 275251200 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004629689067201605, + "loss": 1.6554, + "theoretical_loss": 4.188817062326644, + "tokens_seen": 275251200 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004629588766298897, + "loss": 1.623, + "theoretical_loss": 4.188698419061398, + "tokens_seen": 275316736 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046294884653961887, + "loss": 1.6324, + "theoretical_loss": 4.188579811939988, + "tokens_seen": 275382272 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046293881644934805, + "loss": 1.6178, + "theoretical_loss": 4.188461240942803, + "tokens_seen": 275447808 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004629287863590773, + "loss": 1.539, + "theoretical_loss": 4.18834270605025, + "tokens_seen": 275513344 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004629187562688064, + "loss": 1.5221, + "theoretical_loss": 4.188224207242753, + "tokens_seen": 275578880 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046290872617853565, + "loss": 1.4679, + "theoretical_loss": 4.188105744500747, + "tokens_seen": 275644416 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046289869608826477, + "loss": 1.5572, + "theoretical_loss": 4.187987317804685, + "tokens_seen": 275709952 + }, + { + "epoch": 0.08, + "learning_rate": 0.000462888665997994, + "loss": 1.6401, + "theoretical_loss": 4.187868927135035, + "tokens_seen": 275775488 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004628786359077232, + "loss": 1.6694, + "theoretical_loss": 4.1877505724722806, + "tokens_seen": 275841024 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046286860581745237, + "loss": 1.5845, + "theoretical_loss": 4.187632253796918, + "tokens_seen": 275906560 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046285857572718155, + "loss": 1.6355, + "theoretical_loss": 4.187513971089463, + "tokens_seen": 275972096 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046284854563691073, + "loss": 1.6141, + "theoretical_loss": 4.187395724330443, + "tokens_seen": 276037632 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004628385155466399, + "loss": 1.5423, + "theoretical_loss": 4.187277513500401, + "tokens_seen": 276103168 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046282848545636915, + "loss": 1.6056, + "theoretical_loss": 4.187159338579897, + "tokens_seen": 276168704 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004628184553660983, + "loss": 1.5784, + "theoretical_loss": 4.187041199549505, + "tokens_seen": 276234240 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004628084252758275, + "loss": 1.4761, + "theoretical_loss": 4.186923096389812, + "tokens_seen": 276299776 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046279839518555664, + "loss": 1.6144, + "theoretical_loss": 4.186805029081425, + "tokens_seen": 276365312 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004627883650952859, + "loss": 1.5538, + "theoretical_loss": 4.186686997604962, + "tokens_seen": 276430848 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046277833500501505, + "loss": 1.5506, + "theoretical_loss": 4.186569001941057, + "tokens_seen": 276496384 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046276830491474424, + "loss": 1.4938, + "theoretical_loss": 4.18645104207036, + "tokens_seen": 276561920 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004627582748244734, + "loss": 1.5878, + "theoretical_loss": 4.186333117973534, + "tokens_seen": 276627456 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046274824473420265, + "loss": 1.6284, + "theoretical_loss": 4.1862152296312605, + "tokens_seen": 276692992 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004627382146439318, + "loss": 1.6295, + "theoretical_loss": 4.1860973770242325, + "tokens_seen": 276758528 + }, + { + "epoch": 0.08, + "learning_rate": 0.000462728184553661, + "loss": 1.6017, + "theoretical_loss": 4.185979560133161, + "tokens_seen": 276824064 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.0018113775877282023, + "objective/train/docs_used": 168545, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0363528728485107, + "objective/train/original_loss": 3.0363528728485107, + "objective/train/theoretical_loss": 4.185861778938769, + "objective/train/tokens_used": 297349600, + "objective/train/value_avg": -0.0057373046875, + "objective/train/value_loss": 0.00010671783093130216, + "objective/train/value_max": -0.0003514289855957031, + "objective/train/value_min": -0.11395263671875, + "objective/train/value_reward_corr": 0.4132379000915515, + "objective/train/value_std": 0.006008148193359375, + "objective/train/weight_avg": 0.9867966175079346, + "objective/train/weighted_lm_loss": 3.0045855045318604, + "objective/train/weights_max": 2.616267442703247, + "objective/train/weights_min": 0.19872069358825684, + "theoretical_loss": 4.185861778938769, + "tokens_seen": 276889600 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046271815446339014, + "loss": 1.6298, + "theoretical_loss": 4.185861778938769, + "tokens_seen": 276889600 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004627081243731194, + "loss": 1.5171, + "theoretical_loss": 4.185744033421795, + "tokens_seen": 276955136 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046269809428284856, + "loss": 1.6409, + "theoretical_loss": 4.185626323562996, + "tokens_seen": 277020672 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046268806419257774, + "loss": 1.593, + "theoretical_loss": 4.18550864934314, + "tokens_seen": 277086208 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004626780341023069, + "loss": 1.631, + "theoretical_loss": 4.18539101074301, + "tokens_seen": 277151744 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004626680040120361, + "loss": 1.5841, + "theoretical_loss": 4.185273407743407, + "tokens_seen": 277217280 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004626579739217653, + "loss": 1.6, + "theoretical_loss": 4.185155840325143, + "tokens_seen": 277282816 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004626479438314945, + "loss": 1.5601, + "theoretical_loss": 4.185038308469049, + "tokens_seen": 277348352 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046263791374122364, + "loss": 1.6514, + "theoretical_loss": 4.184920812155967, + "tokens_seen": 277413888 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004626278836509529, + "loss": 1.6368, + "theoretical_loss": 4.184803351366757, + "tokens_seen": 277479424 + }, + { + "epoch": 0.08, + "learning_rate": 0.000462617853560682, + "loss": 1.6662, + "theoretical_loss": 4.184685926082291, + "tokens_seen": 277544960 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046260782347041124, + "loss": 1.646, + "theoretical_loss": 4.184568536283457, + "tokens_seen": 277610496 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004625977933801404, + "loss": 1.4554, + "theoretical_loss": 4.184451181951159, + "tokens_seen": 277676032 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004625877632898696, + "loss": 1.5676, + "theoretical_loss": 4.184333863066316, + "tokens_seen": 277741568 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004625777331995988, + "loss": 1.4265, + "theoretical_loss": 4.184216579609857, + "tokens_seen": 277807104 + }, + { + "epoch": 0.08, + "learning_rate": 0.000462567703109328, + "loss": 1.4726, + "theoretical_loss": 4.184099331562732, + "tokens_seen": 277872640 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046255767301905715, + "loss": 1.5301, + "theoretical_loss": 4.183982118905902, + "tokens_seen": 277938176 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004625476429287864, + "loss": 1.5684, + "theoretical_loss": 4.183864941620344, + "tokens_seen": 278003712 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004625376128385155, + "loss": 1.4634, + "theoretical_loss": 4.183747799687049, + "tokens_seen": 278069248 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046252758274824475, + "loss": 1.4757, + "theoretical_loss": 4.183630693087024, + "tokens_seen": 278134784 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004625175526579739, + "loss": 1.6815, + "theoretical_loss": 4.183513621801289, + "tokens_seen": 278200320 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004625075225677031, + "loss": 1.6493, + "theoretical_loss": 4.183396585810881, + "tokens_seen": 278265856 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004624974924774323, + "loss": 1.5585, + "theoretical_loss": 4.183279585096847, + "tokens_seen": 278331392 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046248746238716147, + "loss": 1.4742, + "theoretical_loss": 4.183162619640255, + "tokens_seen": 278396928 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046247743229689065, + "loss": 1.5331, + "theoretical_loss": 4.1830456894221815, + "tokens_seen": 278462464 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.0007582520483992994, + "objective/train/docs_used": 169743, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.971870183944702, + "objective/train/original_loss": 2.971870183944702, + "objective/train/theoretical_loss": 4.182928794423724, + "objective/train/tokens_used": 298988000, + "objective/train/value_avg": -0.00554656982421875, + "objective/train/value_loss": 5.722793866880238e-05, + "objective/train/value_max": -0.00033283233642578125, + "objective/train/value_min": -0.10321044921875, + "objective/train/value_reward_corr": 0.40857252614543565, + "objective/train/value_std": 0.005527496337890625, + "objective/train/weight_avg": 0.9951887726783752, + "objective/train/weighted_lm_loss": 2.964979410171509, + "objective/train/weights_max": 2.4510374069213867, + "objective/train/weights_min": 0.4494607448577881, + "theoretical_loss": 4.182928794423724, + "tokens_seen": 278528000 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004624674022066199, + "loss": 1.5225, + "theoretical_loss": 4.182928794423724, + "tokens_seen": 278528000 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046245737211634907, + "loss": 1.5062, + "theoretical_loss": 4.182811934625987, + "tokens_seen": 278593536 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046244734202607825, + "loss": 1.5733, + "theoretical_loss": 4.182695110010097, + "tokens_seen": 278659072 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004624373119358075, + "loss": 1.527, + "theoretical_loss": 4.182578320557189, + "tokens_seen": 278724608 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004624272818455366, + "loss": 1.5271, + "theoretical_loss": 4.182461566248418, + "tokens_seen": 278790144 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046241725175526585, + "loss": 1.5674, + "theoretical_loss": 4.1823448470649485, + "tokens_seen": 278855680 + }, + { + "epoch": 0.08, + "learning_rate": 0.000462407221664995, + "loss": 1.537, + "theoretical_loss": 4.182228162987963, + "tokens_seen": 278921216 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004623971915747242, + "loss": 1.4454, + "theoretical_loss": 4.182111513998657, + "tokens_seen": 278986752 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004623871614844534, + "loss": 1.5138, + "theoretical_loss": 4.181994900078241, + "tokens_seen": 279052288 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046237713139418257, + "loss": 1.4816, + "theoretical_loss": 4.181878321207939, + "tokens_seen": 279117824 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046236710130391175, + "loss": 1.6007, + "theoretical_loss": 4.18176177736899, + "tokens_seen": 279183360 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046235707121364093, + "loss": 1.5148, + "theoretical_loss": 4.181645268542649, + "tokens_seen": 279248896 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004623470411233701, + "loss": 1.4969, + "theoretical_loss": 4.181528794710183, + "tokens_seen": 279314432 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046233701103309935, + "loss": 1.621, + "theoretical_loss": 4.181412355852875, + "tokens_seen": 279379968 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004623269809428285, + "loss": 1.5873, + "theoretical_loss": 4.181295951952022, + "tokens_seen": 279445504 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004623169508525577, + "loss": 1.5854, + "theoretical_loss": 4.181179582988934, + "tokens_seen": 279511040 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046230692076228684, + "loss": 1.5502, + "theoretical_loss": 4.181063248944938, + "tokens_seen": 279576576 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004622968906720161, + "loss": 1.6575, + "theoretical_loss": 4.180946949801374, + "tokens_seen": 279642112 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046228686058174525, + "loss": 1.6122, + "theoretical_loss": 4.180830685539596, + "tokens_seen": 279707648 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046227683049147444, + "loss": 1.625, + "theoretical_loss": 4.180714456140972, + "tokens_seen": 279773184 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004622668004012036, + "loss": 1.5091, + "theoretical_loss": 4.180598261586886, + "tokens_seen": 279838720 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046225677031093285, + "loss": 1.5291, + "theoretical_loss": 4.180482101858734, + "tokens_seen": 279904256 + }, + { + "epoch": 0.08, + "learning_rate": 0.000462246740220662, + "loss": 1.5802, + "theoretical_loss": 4.18036597693793, + "tokens_seen": 279969792 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004622367101303912, + "loss": 1.49, + "theoretical_loss": 4.180249886805897, + "tokens_seen": 280035328 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046222668004012034, + "loss": 1.6253, + "theoretical_loss": 4.180133831444077, + "tokens_seen": 280100864 + }, + { + "epoch": 0.08, + "objective/train/advantage_avg": -0.0009453926468268037, + "objective/train/docs_used": 170343, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.723726272583008, + "objective/train/original_loss": 2.723726272583008, + "objective/train/theoretical_loss": 4.180017810833923, + "objective/train/tokens_used": 300626400, + "objective/train/value_avg": -0.00518798828125, + "objective/train/value_loss": 6.934170232852921e-05, + "objective/train/value_max": -0.00022876262664794922, + "objective/train/value_min": -0.12176513671875, + "objective/train/value_reward_corr": 0.2579636956825722, + "objective/train/value_std": 0.004383087158203125, + "objective/train/weight_avg": 0.993744432926178, + "objective/train/weighted_lm_loss": 2.725592613220215, + "objective/train/weights_max": 3.128117084503174, + "objective/train/weights_min": 0.3026140034198761, + "theoretical_loss": 4.180017810833923, + "tokens_seen": 280166400 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004622166499498496, + "loss": 1.5668, + "theoretical_loss": 4.180017810833923, + "tokens_seen": 280166400 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046220661985957876, + "loss": 1.5311, + "theoretical_loss": 4.179901824956904, + "tokens_seen": 280231936 + }, + { + "epoch": 0.08, + "learning_rate": 0.00046219658976930794, + "loss": 1.6728, + "theoretical_loss": 4.179785873794503, + "tokens_seen": 280297472 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004621865596790371, + "loss": 1.5607, + "theoretical_loss": 4.179669957328218, + "tokens_seen": 280363008 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004621765295887663, + "loss": 1.666, + "theoretical_loss": 4.179554075539558, + "tokens_seen": 280428544 + }, + { + "epoch": 0.08, + "learning_rate": 0.0004621664994984955, + "loss": 1.6178, + "theoretical_loss": 4.17943822841005, + "tokens_seen": 280494080 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004621564694082247, + "loss": 1.4985, + "theoretical_loss": 4.1793224159212325, + "tokens_seen": 280559616 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046214643931795384, + "loss": 1.4872, + "theoretical_loss": 4.1792066380546595, + "tokens_seen": 280625152 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004621364092276831, + "loss": 1.5238, + "theoretical_loss": 4.179090894791899, + "tokens_seen": 280690688 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004621263791374122, + "loss": 1.5339, + "theoretical_loss": 4.178975186114532, + "tokens_seen": 280756224 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046211634904714144, + "loss": 1.6039, + "theoretical_loss": 4.178859512004156, + "tokens_seen": 280821760 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004621063189568706, + "loss": 1.3987, + "theoretical_loss": 4.178743872442379, + "tokens_seen": 280887296 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004620962888665998, + "loss": 1.4232, + "theoretical_loss": 4.178628267410826, + "tokens_seen": 280952832 + }, + { + "epoch": 0.09, + "learning_rate": 0.000462086258776329, + "loss": 1.7161, + "theoretical_loss": 4.178512696891136, + "tokens_seen": 281018368 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004620762286860582, + "loss": 1.6617, + "theoretical_loss": 4.178397160864961, + "tokens_seen": 281083904 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046206619859578735, + "loss": 1.67, + "theoretical_loss": 4.178281659313965, + "tokens_seen": 281149440 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004620561685055166, + "loss": 1.5608, + "theoretical_loss": 4.178166192219832, + "tokens_seen": 281214976 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004620461384152457, + "loss": 1.4543, + "theoretical_loss": 4.178050759564254, + "tokens_seen": 281280512 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046203610832497495, + "loss": 1.609, + "theoretical_loss": 4.177935361328939, + "tokens_seen": 281346048 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004620260782347041, + "loss": 1.564, + "theoretical_loss": 4.17781999749561, + "tokens_seen": 281411584 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004620160481444333, + "loss": 1.6459, + "theoretical_loss": 4.177704668046003, + "tokens_seen": 281477120 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004620060180541625, + "loss": 1.5774, + "theoretical_loss": 4.177589372961869, + "tokens_seen": 281542656 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046199598796389167, + "loss": 1.474, + "theoretical_loss": 4.177474112224969, + "tokens_seen": 281608192 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046198595787362085, + "loss": 1.6051, + "theoretical_loss": 4.177358885817085, + "tokens_seen": 281673728 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004619759277833501, + "loss": 1.5202, + "theoretical_loss": 4.177243693720007, + "tokens_seen": 281739264 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0002689921238925308, + "objective/train/docs_used": 171008, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9972167015075684, + "objective/train/original_loss": 2.9972164630889893, + "objective/train/theoretical_loss": 4.177128535915539, + "objective/train/tokens_used": 302264800, + "objective/train/value_avg": -0.0046234130859375, + "objective/train/value_loss": 8.776275353739038e-05, + "objective/train/value_max": -0.0002613067626953125, + "objective/train/value_min": -0.05572509765625, + "objective/train/value_reward_corr": 0.37918976946671756, + "objective/train/value_std": 0.003849029541015625, + "objective/train/weight_avg": 1.001031756401062, + "objective/train/weighted_lm_loss": 3.006185293197632, + "objective/train/weights_max": 1.513967752456665, + "objective/train/weights_min": 0.2096860110759735, + "theoretical_loss": 4.177128535915539, + "tokens_seen": 281804800 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004619658976930792, + "loss": 1.5073, + "theoretical_loss": 4.177128535915539, + "tokens_seen": 281804800 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046195586760280845, + "loss": 1.4969, + "theoretical_loss": 4.177013412385504, + "tokens_seen": 281870336 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046194583751253763, + "loss": 1.5492, + "theoretical_loss": 4.176898323111734, + "tokens_seen": 281935872 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004619358074222668, + "loss": 1.5501, + "theoretical_loss": 4.1767832680760755, + "tokens_seen": 282001408 + }, + { + "epoch": 0.09, + "learning_rate": 0.000461925777331996, + "loss": 1.6274, + "theoretical_loss": 4.176668247260391, + "tokens_seen": 282066944 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004619157472417252, + "loss": 1.5835, + "theoretical_loss": 4.176553260646555, + "tokens_seen": 282132480 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046190571715145435, + "loss": 1.5967, + "theoretical_loss": 4.176438308216456, + "tokens_seen": 282198016 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004618956870611836, + "loss": 1.4904, + "theoretical_loss": 4.176323389951998, + "tokens_seen": 282263552 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004618856569709127, + "loss": 1.5626, + "theoretical_loss": 4.1762085058350955, + "tokens_seen": 282329088 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046187562688064195, + "loss": 1.5776, + "theoretical_loss": 4.176093655847679, + "tokens_seen": 282394624 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004618655967903711, + "loss": 1.6545, + "theoretical_loss": 4.175978839971694, + "tokens_seen": 282460160 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004618555667001003, + "loss": 1.4813, + "theoretical_loss": 4.175864058189096, + "tokens_seen": 282525696 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004618455366098295, + "loss": 1.6785, + "theoretical_loss": 4.175749310481859, + "tokens_seen": 282591232 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004618355065195587, + "loss": 1.5782, + "theoretical_loss": 4.175634596831965, + "tokens_seen": 282656768 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046182547642928786, + "loss": 1.5313, + "theoretical_loss": 4.175519917221415, + "tokens_seen": 282722304 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046181544633901704, + "loss": 1.6231, + "theoretical_loss": 4.17540527163222, + "tokens_seen": 282787840 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004618054162487462, + "loss": 1.4209, + "theoretical_loss": 4.175290660046407, + "tokens_seen": 282853376 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046179538615847545, + "loss": 1.5877, + "theoretical_loss": 4.175176082446017, + "tokens_seen": 282918912 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004617853560682046, + "loss": 1.6354, + "theoretical_loss": 4.175061538813101, + "tokens_seen": 282984448 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004617753259779338, + "loss": 1.611, + "theoretical_loss": 4.174947029129728, + "tokens_seen": 283049984 + }, + { + "epoch": 0.09, + "learning_rate": 0.000461765295887663, + "loss": 1.5124, + "theoretical_loss": 4.174832553377978, + "tokens_seen": 283115520 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004617552657973922, + "loss": 1.6705, + "theoretical_loss": 4.174718111539945, + "tokens_seen": 283181056 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046174523570712136, + "loss": 1.4982, + "theoretical_loss": 4.174603703597737, + "tokens_seen": 283246592 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046173520561685054, + "loss": 1.6181, + "theoretical_loss": 4.1744893295334755, + "tokens_seen": 283312128 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004617251755265797, + "loss": 1.4832, + "theoretical_loss": 4.174374989329296, + "tokens_seen": 283377664 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.001453412463888526, + "objective/train/docs_used": 172282, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2372913360595703, + "objective/train/original_loss": 3.2372915744781494, + "objective/train/theoretical_loss": 4.174260682967347, + "objective/train/tokens_used": 303903200, + "objective/train/value_avg": -0.005306243896484375, + "objective/train/value_loss": 8.233919652411714e-05, + "objective/train/value_max": -0.0002472400665283203, + "objective/train/value_min": -0.06768798828125, + "objective/train/value_reward_corr": 0.3471498754614506, + "objective/train/value_std": 0.004425048828125, + "objective/train/weight_avg": 0.9891772866249084, + "objective/train/weighted_lm_loss": 3.2144041061401367, + "objective/train/weights_max": 1.5756338834762573, + "objective/train/weights_min": 0.24524155259132385, + "theoretical_loss": 4.174260682967347, + "tokens_seen": 283443200 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046171514543630896, + "loss": 1.5175, + "theoretical_loss": 4.174260682967347, + "tokens_seen": 283443200 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046170511534603814, + "loss": 1.4546, + "theoretical_loss": 4.17414641042979, + "tokens_seen": 283508736 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004616950852557673, + "loss": 1.5765, + "theoretical_loss": 4.1740321716988005, + "tokens_seen": 283574272 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004616850551654965, + "loss": 1.5945, + "theoretical_loss": 4.1739179667565685, + "tokens_seen": 283639808 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004616750250752257, + "loss": 1.4963, + "theoretical_loss": 4.173803795585296, + "tokens_seen": 283705344 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004616649949849549, + "loss": 1.4388, + "theoretical_loss": 4.173689658167199, + "tokens_seen": 283770880 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046165496489468404, + "loss": 1.6239, + "theoretical_loss": 4.173575554484509, + "tokens_seen": 283836416 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004616449348044133, + "loss": 1.4203, + "theoretical_loss": 4.173461484519466, + "tokens_seen": 283901952 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004616349047141424, + "loss": 1.5474, + "theoretical_loss": 4.1733474482543285, + "tokens_seen": 283967488 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046162487462387164, + "loss": 1.6186, + "theoretical_loss": 4.173233445671365, + "tokens_seen": 284033024 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004616148445336008, + "loss": 1.5725, + "theoretical_loss": 4.173119476752861, + "tokens_seen": 284098560 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046160481444333, + "loss": 1.5099, + "theoretical_loss": 4.173005541481111, + "tokens_seen": 284164096 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004615947843530592, + "loss": 1.5083, + "theoretical_loss": 4.1728916398384275, + "tokens_seen": 284229632 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004615847542627884, + "loss": 1.6313, + "theoretical_loss": 4.172777771807132, + "tokens_seen": 284295168 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046157472417251755, + "loss": 1.6072, + "theoretical_loss": 4.1726639373695615, + "tokens_seen": 284360704 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004615646940822468, + "loss": 1.4148, + "theoretical_loss": 4.172550136508066, + "tokens_seen": 284426240 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004615546639919759, + "loss": 1.6558, + "theoretical_loss": 4.172436369205011, + "tokens_seen": 284491776 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046154463390170515, + "loss": 1.6076, + "theoretical_loss": 4.172322635442772, + "tokens_seen": 284557312 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004615346038114343, + "loss": 1.7259, + "theoretical_loss": 4.172208935203739, + "tokens_seen": 284622848 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004615245737211635, + "loss": 1.5315, + "theoretical_loss": 4.172095268470316, + "tokens_seen": 284688384 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004615145436308927, + "loss": 1.4795, + "theoretical_loss": 4.1719816352249195, + "tokens_seen": 284753920 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046150451354062187, + "loss": 1.4138, + "theoretical_loss": 4.17186803544998, + "tokens_seen": 284819456 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046149448345035105, + "loss": 1.5078, + "theoretical_loss": 4.17175446912794, + "tokens_seen": 284884992 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004614844533600803, + "loss": 1.5541, + "theoretical_loss": 4.171640936241256, + "tokens_seen": 284950528 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004614744232698094, + "loss": 1.4969, + "theoretical_loss": 4.171527436772399, + "tokens_seen": 285016064 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0020911998581141233, + "objective/train/docs_used": 172985, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.256631851196289, + "objective/train/original_loss": 3.25663161277771, + "objective/train/theoretical_loss": 4.171413970703851, + "objective/train/tokens_used": 305541600, + "objective/train/value_avg": -0.00612640380859375, + "objective/train/value_loss": 7.648356404388323e-05, + "objective/train/value_max": -0.00035691261291503906, + "objective/train/value_min": -0.0635986328125, + "objective/train/value_reward_corr": 0.47586931863943244, + "objective/train/value_std": 0.005649566650390625, + "objective/train/weight_avg": 0.9826661348342896, + "objective/train/weighted_lm_loss": 3.206834077835083, + "objective/train/weights_max": 1.6156482696533203, + "objective/train/weights_min": 0.40956130623817444, + "theoretical_loss": 4.171413970703851, + "tokens_seen": 285081600 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046146439317953865, + "loss": 1.5378, + "theoretical_loss": 4.171413970703851, + "tokens_seen": 285081600 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046145436308926783, + "loss": 1.4182, + "theoretical_loss": 4.171300538018109, + "tokens_seen": 285147136 + }, + { + "epoch": 0.09, + "learning_rate": 0.000461444332998997, + "loss": 1.668, + "theoretical_loss": 4.1711871386976815, + "tokens_seen": 285212672 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004614343029087262, + "loss": 1.5334, + "theoretical_loss": 4.171073772725091, + "tokens_seen": 285278208 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004614242728184554, + "loss": 1.694, + "theoretical_loss": 4.170960440082874, + "tokens_seen": 285343744 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046141424272818455, + "loss": 1.6221, + "theoretical_loss": 4.17084714075358, + "tokens_seen": 285409280 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004614042126379138, + "loss": 1.5162, + "theoretical_loss": 4.170733874719769, + "tokens_seen": 285474816 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004613941825476429, + "loss": 1.5718, + "theoretical_loss": 4.1706206419640175, + "tokens_seen": 285540352 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046138415245737215, + "loss": 1.6468, + "theoretical_loss": 4.170507442468914, + "tokens_seen": 285605888 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004613741223671013, + "loss": 1.545, + "theoretical_loss": 4.170394276217059, + "tokens_seen": 285671424 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004613640922768305, + "loss": 1.6107, + "theoretical_loss": 4.170281143191067, + "tokens_seen": 285736960 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004613540621865597, + "loss": 1.4315, + "theoretical_loss": 4.170168043373567, + "tokens_seen": 285802496 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004613440320962889, + "loss": 1.6151, + "theoretical_loss": 4.170054976747199, + "tokens_seen": 285868032 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046133400200601806, + "loss": 1.6373, + "theoretical_loss": 4.169941943294615, + "tokens_seen": 285933568 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046132397191574724, + "loss": 1.525, + "theoretical_loss": 4.1698289429984845, + "tokens_seen": 285999104 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004613139418254764, + "loss": 1.4783, + "theoretical_loss": 4.169715975841485, + "tokens_seen": 286064640 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046130391173520566, + "loss": 1.6155, + "theoretical_loss": 4.16960304180631, + "tokens_seen": 286130176 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004612938816449348, + "loss": 1.6027, + "theoretical_loss": 4.169490140875665, + "tokens_seen": 286195712 + }, + { + "epoch": 0.09, + "learning_rate": 0.000461283851554664, + "loss": 1.5111, + "theoretical_loss": 4.16937727303227, + "tokens_seen": 286261248 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004612738214643932, + "loss": 1.4829, + "theoretical_loss": 4.169264438258854, + "tokens_seen": 286326784 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004612637913741224, + "loss": 1.5044, + "theoretical_loss": 4.169151636538165, + "tokens_seen": 286392320 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046125376128385156, + "loss": 1.6684, + "theoretical_loss": 4.169038867852958, + "tokens_seen": 286457856 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046124373119358074, + "loss": 1.4582, + "theoretical_loss": 4.168926132186005, + "tokens_seen": 286523392 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004612337011033099, + "loss": 1.5756, + "theoretical_loss": 4.16881342952009, + "tokens_seen": 286588928 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046122367101303916, + "loss": 1.5621, + "theoretical_loss": 4.168700759838007, + "tokens_seen": 286654464 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": 0.0004032111610285938, + "objective/train/docs_used": 174327, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8161866664886475, + "objective/train/original_loss": 2.8161864280700684, + "objective/train/theoretical_loss": 4.168588123122568, + "objective/train/tokens_used": 307180000, + "objective/train/value_avg": -0.0050048828125, + "objective/train/value_loss": 4.9631100409897044e-05, + "objective/train/value_max": -0.0002913475036621094, + "objective/train/value_min": -0.076416015625, + "objective/train/value_reward_corr": 0.25768760931020374, + "objective/train/value_std": 0.0039005279541015625, + "objective/train/weight_avg": 1.0064133405685425, + "objective/train/weighted_lm_loss": 2.829594612121582, + "objective/train/weights_max": 1.9062013626098633, + "objective/train/weights_min": 0.4297851026058197, + "theoretical_loss": 4.168588123122568, + "tokens_seen": 286720000 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004612136409227683, + "loss": 1.6106, + "theoretical_loss": 4.168588123122568, + "tokens_seen": 286720000 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004612036108324975, + "loss": 1.495, + "theoretical_loss": 4.168475519356592, + "tokens_seen": 286785536 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046119358074222665, + "loss": 1.626, + "theoretical_loss": 4.168362948522918, + "tokens_seen": 286851072 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004611835506519559, + "loss": 1.6345, + "theoretical_loss": 4.168250410604391, + "tokens_seen": 286916608 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046117352056168506, + "loss": 1.5052, + "theoretical_loss": 4.1681379055838725, + "tokens_seen": 286982144 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046116349047141425, + "loss": 1.6319, + "theoretical_loss": 4.168025433444237, + "tokens_seen": 287047680 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004611534603811434, + "loss": 1.6112, + "theoretical_loss": 4.16791299416837, + "tokens_seen": 287113216 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004611434302908726, + "loss": 1.5238, + "theoretical_loss": 4.167800587739171, + "tokens_seen": 287178752 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004611334002006018, + "loss": 1.5152, + "theoretical_loss": 4.167688214139552, + "tokens_seen": 287244288 + }, + { + "epoch": 0.09, + "learning_rate": 0.000461123370110331, + "loss": 1.6503, + "theoretical_loss": 4.167575873352437, + "tokens_seen": 287309824 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046111334002006015, + "loss": 1.5004, + "theoretical_loss": 4.167463565360766, + "tokens_seen": 287375360 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004611033099297894, + "loss": 1.467, + "theoretical_loss": 4.1673512901474865, + "tokens_seen": 287440896 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046109327983951857, + "loss": 1.4915, + "theoretical_loss": 4.1672390476955625, + "tokens_seen": 287506432 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046108324974924775, + "loss": 1.6139, + "theoretical_loss": 4.167126837987971, + "tokens_seen": 287571968 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046107321965897693, + "loss": 1.6477, + "theoretical_loss": 4.167014661007699, + "tokens_seen": 287637504 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004610631895687061, + "loss": 1.5497, + "theoretical_loss": 4.16690251673775, + "tokens_seen": 287703040 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004610531594784353, + "loss": 1.5324, + "theoretical_loss": 4.1667904051611355, + "tokens_seen": 287768576 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004610431293881645, + "loss": 1.4639, + "theoretical_loss": 4.166678326260882, + "tokens_seen": 287834112 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046103309929789365, + "loss": 1.5909, + "theoretical_loss": 4.166566280020032, + "tokens_seen": 287899648 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004610230692076229, + "loss": 1.703, + "theoretical_loss": 4.166454266421635, + "tokens_seen": 287965184 + }, + { + "epoch": 0.09, + "learning_rate": 0.000461013039117352, + "loss": 1.539, + "theoretical_loss": 4.1663422854487555, + "tokens_seen": 288030720 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046100300902708125, + "loss": 1.4314, + "theoretical_loss": 4.166230337084471, + "tokens_seen": 288096256 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046099297893681043, + "loss": 1.5851, + "theoretical_loss": 4.166118421311873, + "tokens_seen": 288161792 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004609829488465396, + "loss": 1.6975, + "theoretical_loss": 4.166006538114062, + "tokens_seen": 288227328 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004609729187562688, + "loss": 1.482, + "theoretical_loss": 4.165894687474155, + "tokens_seen": 288292864 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": 0.00021786089928355068, + "objective/train/docs_used": 174865, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.906122922897339, + "objective/train/original_loss": 2.906123161315918, + "objective/train/theoretical_loss": 4.165782869375278, + "objective/train/tokens_used": 308818400, + "objective/train/value_avg": -0.005035400390625, + "objective/train/value_loss": 6.380959530360997e-05, + "objective/train/value_max": -0.0003077983856201172, + "objective/train/value_min": -0.050994873046875, + "objective/train/value_reward_corr": 0.2502568910199997, + "objective/train/value_std": 0.0037078857421875, + "objective/train/weight_avg": 1.0051978826522827, + "objective/train/weighted_lm_loss": 2.9202516078948975, + "objective/train/weights_max": 1.6652058362960815, + "objective/train/weights_min": 0.5322730541229248, + "theoretical_loss": 4.165782869375278, + "tokens_seen": 288358400 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046096288866599803, + "loss": 1.4949, + "theoretical_loss": 4.165782869375278, + "tokens_seen": 288358400 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004609528585757272, + "loss": 1.4781, + "theoretical_loss": 4.1656710838005715, + "tokens_seen": 288423936 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004609428284854564, + "loss": 1.6696, + "theoretical_loss": 4.165559330733188, + "tokens_seen": 288489472 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004609327983951856, + "loss": 1.456, + "theoretical_loss": 4.165447610156296, + "tokens_seen": 288555008 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046092276830491475, + "loss": 1.5716, + "theoretical_loss": 4.165335922053069, + "tokens_seen": 288620544 + }, + { + "epoch": 0.09, + "learning_rate": 0.000460912738214644, + "loss": 1.5875, + "theoretical_loss": 4.165224266406701, + "tokens_seen": 288686080 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004609027081243731, + "loss": 1.5455, + "theoretical_loss": 4.165112643200393, + "tokens_seen": 288751616 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046089267803410235, + "loss": 1.5894, + "theoretical_loss": 4.165001052417361, + "tokens_seen": 288817152 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004608826479438315, + "loss": 1.5293, + "theoretical_loss": 4.164889494040834, + "tokens_seen": 288882688 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004608726178535607, + "loss": 1.4825, + "theoretical_loss": 4.16477796805405, + "tokens_seen": 288948224 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004608625877632899, + "loss": 1.528, + "theoretical_loss": 4.1646664744402635, + "tokens_seen": 289013760 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004608525576730191, + "loss": 1.4208, + "theoretical_loss": 4.16455501318274, + "tokens_seen": 289079296 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046084252758274826, + "loss": 1.5774, + "theoretical_loss": 4.164443584264758, + "tokens_seen": 289144832 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046083249749247744, + "loss": 1.4647, + "theoretical_loss": 4.164332187669605, + "tokens_seen": 289210368 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004608224674022066, + "loss": 1.5769, + "theoretical_loss": 4.164220823380587, + "tokens_seen": 289275904 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046081243731193586, + "loss": 1.5152, + "theoretical_loss": 4.164109491381017, + "tokens_seen": 289341440 + }, + { + "epoch": 0.09, + "learning_rate": 0.000460802407221665, + "loss": 1.4816, + "theoretical_loss": 4.163998191654223, + "tokens_seen": 289406976 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004607923771313942, + "loss": 1.571, + "theoretical_loss": 4.163886924183544, + "tokens_seen": 289472512 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004607823470411234, + "loss": 1.6017, + "theoretical_loss": 4.163775688952334, + "tokens_seen": 289538048 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004607723169508526, + "loss": 1.391, + "theoretical_loss": 4.163664485943956, + "tokens_seen": 289603584 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046076228686058176, + "loss": 1.4782, + "theoretical_loss": 4.163553315141788, + "tokens_seen": 289669120 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046075225677031094, + "loss": 1.615, + "theoretical_loss": 4.163442176529218, + "tokens_seen": 289734656 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004607422266800401, + "loss": 1.5142, + "theoretical_loss": 4.163331070089649, + "tokens_seen": 289800192 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046073219658976936, + "loss": 1.4543, + "theoretical_loss": 4.163219995806494, + "tokens_seen": 289865728 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004607221664994985, + "loss": 1.6229, + "theoretical_loss": 4.163108953663179, + "tokens_seen": 289931264 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0016043048817664385, + "objective/train/docs_used": 176137, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.972346305847168, + "objective/train/original_loss": 2.972346544265747, + "objective/train/theoretical_loss": 4.162997943643143, + "objective/train/tokens_used": 310456800, + "objective/train/value_avg": -0.005786895751953125, + "objective/train/value_loss": 6.52651215204969e-05, + "objective/train/value_max": -0.00021314620971679688, + "objective/train/value_min": -0.07354736328125, + "objective/train/value_reward_corr": 0.28443725402255576, + "objective/train/value_std": 0.0047760009765625, + "objective/train/weight_avg": 0.9871335625648499, + "objective/train/weighted_lm_loss": 2.945119619369507, + "objective/train/weights_max": 1.8565813302993774, + "objective/train/weights_min": 0.49660438299179077, + "theoretical_loss": 4.162997943643143, + "tokens_seen": 289996800 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004607121364092277, + "loss": 1.5398, + "theoretical_loss": 4.162997943643143, + "tokens_seen": 289996800 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046070210631895685, + "loss": 1.6332, + "theoretical_loss": 4.162886965729837, + "tokens_seen": 290062336 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004606920762286861, + "loss": 1.5521, + "theoretical_loss": 4.162776019906723, + "tokens_seen": 290127872 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046068204613841526, + "loss": 1.6337, + "theoretical_loss": 4.162665106157277, + "tokens_seen": 290193408 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046067201604814445, + "loss": 1.5921, + "theoretical_loss": 4.162554224464987, + "tokens_seen": 290258944 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004606619859578736, + "loss": 1.5696, + "theoretical_loss": 4.162443374813352, + "tokens_seen": 290324480 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004606519558676028, + "loss": 1.5388, + "theoretical_loss": 4.1623325571858825, + "tokens_seen": 290390016 + }, + { + "epoch": 0.09, + "learning_rate": 0.000460641925777332, + "loss": 1.4178, + "theoretical_loss": 4.162221771566105, + "tokens_seen": 290455552 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004606318956870612, + "loss": 1.5399, + "theoretical_loss": 4.162111017937555, + "tokens_seen": 290521088 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046062186559679035, + "loss": 1.5162, + "theoretical_loss": 4.1620002962837805, + "tokens_seen": 290586624 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004606118355065196, + "loss": 1.5103, + "theoretical_loss": 4.161889606588343, + "tokens_seen": 290652160 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046060180541624877, + "loss": 1.4952, + "theoretical_loss": 4.161778948834815, + "tokens_seen": 290717696 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046059177532597795, + "loss": 1.5589, + "theoretical_loss": 4.161668323006781, + "tokens_seen": 290783232 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046058174523570713, + "loss": 1.4821, + "theoretical_loss": 4.161557729087839, + "tokens_seen": 290848768 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004605717151454363, + "loss": 1.5397, + "theoretical_loss": 4.161447167061597, + "tokens_seen": 290914304 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004605616850551655, + "loss": 1.3999, + "theoretical_loss": 4.1613366369116775, + "tokens_seen": 290979840 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004605516549648947, + "loss": 1.4561, + "theoretical_loss": 4.161226138621714, + "tokens_seen": 291045376 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046054162487462385, + "loss": 1.455, + "theoretical_loss": 4.1611156721753515, + "tokens_seen": 291110912 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004605315947843531, + "loss": 1.5308, + "theoretical_loss": 4.161005237556247, + "tokens_seen": 291176448 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004605215646940822, + "loss": 1.5354, + "theoretical_loss": 4.160894834748071, + "tokens_seen": 291241984 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046051153460381145, + "loss": 1.5346, + "theoretical_loss": 4.1607844637345055, + "tokens_seen": 291307520 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046050150451354063, + "loss": 1.633, + "theoretical_loss": 4.160674124499243, + "tokens_seen": 291373056 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004604914744232698, + "loss": 1.5913, + "theoretical_loss": 4.160563817025991, + "tokens_seen": 291438592 + }, + { + "epoch": 0.09, + "learning_rate": 0.000460481444332999, + "loss": 1.6131, + "theoretical_loss": 4.160453541298465, + "tokens_seen": 291504128 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046047141424272823, + "loss": 1.652, + "theoretical_loss": 4.160343297300397, + "tokens_seen": 291569664 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0018563736230134964, + "objective/train/docs_used": 176478, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7597336769104004, + "objective/train/original_loss": 2.7597336769104004, + "objective/train/theoretical_loss": 4.160233085015529, + "objective/train/tokens_used": 312095200, + "objective/train/value_avg": -0.006069183349609375, + "objective/train/value_loss": 0.00010078938794322312, + "objective/train/value_max": -0.00046372413635253906, + "objective/train/value_min": -0.07421875, + "objective/train/value_reward_corr": 0.4281425120627959, + "objective/train/value_std": 0.005779266357421875, + "objective/train/weight_avg": 0.9860097765922546, + "objective/train/weighted_lm_loss": 2.7229602336883545, + "objective/train/weights_max": 1.5817255973815918, + "objective/train/weights_min": 0.3569692075252533, + "theoretical_loss": 4.160233085015529, + "tokens_seen": 291635200 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046046138415245736, + "loss": 1.5146, + "theoretical_loss": 4.160233085015529, + "tokens_seen": 291635200 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004604513540621866, + "loss": 1.5878, + "theoretical_loss": 4.160122904427611, + "tokens_seen": 291700736 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004604413239719157, + "loss": 1.6322, + "theoretical_loss": 4.160012755520413, + "tokens_seen": 291766272 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046043129388164495, + "loss": 1.5317, + "theoretical_loss": 4.159902638277712, + "tokens_seen": 291831808 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046042126379137414, + "loss": 1.6151, + "theoretical_loss": 4.159792552683296, + "tokens_seen": 291897344 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004604112337011033, + "loss": 1.5997, + "theoretical_loss": 4.159682498720967, + "tokens_seen": 291962880 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004604012036108325, + "loss": 1.5469, + "theoretical_loss": 4.15957247637454, + "tokens_seen": 292028416 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004603911735205617, + "loss": 1.5452, + "theoretical_loss": 4.159462485627838, + "tokens_seen": 292093952 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046038114343029086, + "loss": 1.5121, + "theoretical_loss": 4.1593525264647, + "tokens_seen": 292159488 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004603711133400201, + "loss": 1.6098, + "theoretical_loss": 4.159242598868976, + "tokens_seen": 292225024 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004603610832497492, + "loss": 1.7086, + "theoretical_loss": 4.159132702824524, + "tokens_seen": 292290560 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046035105315947846, + "loss": 1.6439, + "theoretical_loss": 4.15902283831522, + "tokens_seen": 292356096 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004603410230692076, + "loss": 1.5408, + "theoretical_loss": 4.158913005324949, + "tokens_seen": 292421632 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004603309929789368, + "loss": 1.5727, + "theoretical_loss": 4.158803203837605, + "tokens_seen": 292487168 + }, + { + "epoch": 0.09, + "learning_rate": 0.000460320962888666, + "loss": 1.5471, + "theoretical_loss": 4.158693433837098, + "tokens_seen": 292552704 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004603109327983952, + "loss": 1.5923, + "theoretical_loss": 4.158583695307349, + "tokens_seen": 292618240 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046030090270812436, + "loss": 1.6772, + "theoretical_loss": 4.158473988232288, + "tokens_seen": 292683776 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004602908726178536, + "loss": 1.5145, + "theoretical_loss": 4.158364312595862, + "tokens_seen": 292749312 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004602808425275827, + "loss": 1.5278, + "theoretical_loss": 4.158254668382024, + "tokens_seen": 292814848 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046027081243731196, + "loss": 1.6936, + "theoretical_loss": 4.158145055574742, + "tokens_seen": 292880384 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004602607823470411, + "loss": 1.4407, + "theoretical_loss": 4.158035474157996, + "tokens_seen": 292945920 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004602507522567703, + "loss": 1.6251, + "theoretical_loss": 4.157925924115778, + "tokens_seen": 293011456 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004602407221664995, + "loss": 1.5544, + "theoretical_loss": 4.157816405432088, + "tokens_seen": 293076992 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004602306920762287, + "loss": 1.5422, + "theoretical_loss": 4.157706918090943, + "tokens_seen": 293142528 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046022066198595787, + "loss": 1.572, + "theoretical_loss": 4.157597462076368, + "tokens_seen": 293208064 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0007445771479979157, + "objective/train/docs_used": 177825, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.940498113632202, + "objective/train/original_loss": 2.940497875213623, + "objective/train/theoretical_loss": 4.157488037372401, + "objective/train/tokens_used": 313733600, + "objective/train/value_avg": -0.00531005859375, + "objective/train/value_loss": 0.0001416540762875229, + "objective/train/value_max": -0.0003612041473388672, + "objective/train/value_min": -0.05499267578125, + "objective/train/value_reward_corr": 0.24279629484567106, + "objective/train/value_std": 0.004810333251953125, + "objective/train/weight_avg": 0.9988784790039062, + "objective/train/weighted_lm_loss": 2.9702019691467285, + "objective/train/weights_max": 1.6959775686264038, + "objective/train/weights_min": 0.576576292514801, + "theoretical_loss": 4.157488037372401, + "tokens_seen": 293273600 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046021063189568705, + "loss": 1.4621, + "theoretical_loss": 4.157488037372401, + "tokens_seen": 293273600 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004602006018054163, + "loss": 1.5189, + "theoretical_loss": 4.157378643963092, + "tokens_seen": 293339136 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046019057171514546, + "loss": 1.6124, + "theoretical_loss": 4.157269281832502, + "tokens_seen": 293404672 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046018054162487465, + "loss": 1.6169, + "theoretical_loss": 4.157159950964704, + "tokens_seen": 293470208 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004601705115346038, + "loss": 1.5135, + "theoretical_loss": 4.157050651343782, + "tokens_seen": 293535744 + }, + { + "epoch": 0.09, + "learning_rate": 0.000460160481444333, + "loss": 1.643, + "theoretical_loss": 4.156941382953835, + "tokens_seen": 293601280 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004601504513540622, + "loss": 1.6357, + "theoretical_loss": 4.156832145778967, + "tokens_seen": 293666816 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004601404212637914, + "loss": 1.4399, + "theoretical_loss": 4.1567229398033, + "tokens_seen": 293732352 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046013039117352055, + "loss": 1.6031, + "theoretical_loss": 4.156613765010965, + "tokens_seen": 293797888 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004601203610832498, + "loss": 1.4705, + "theoretical_loss": 4.156504621386103, + "tokens_seen": 293863424 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046011033099297897, + "loss": 1.5656, + "theoretical_loss": 4.156395508912872, + "tokens_seen": 293928960 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046010030090270815, + "loss": 1.4166, + "theoretical_loss": 4.156286427575434, + "tokens_seen": 293994496 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046009027081243733, + "loss": 1.3814, + "theoretical_loss": 4.156177377357969, + "tokens_seen": 294060032 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004600802407221665, + "loss": 1.6517, + "theoretical_loss": 4.156068358244665, + "tokens_seen": 294125568 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004600702106318957, + "loss": 1.6131, + "theoretical_loss": 4.155959370219724, + "tokens_seen": 294191104 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046006018054162493, + "loss": 1.6623, + "theoretical_loss": 4.155850413267357, + "tokens_seen": 294256640 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046005015045135405, + "loss": 1.558, + "theoretical_loss": 4.155741487371789, + "tokens_seen": 294322176 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004600401203610833, + "loss": 1.4439, + "theoretical_loss": 4.155632592517254, + "tokens_seen": 294387712 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004600300902708124, + "loss": 1.5951, + "theoretical_loss": 4.1555237286879985, + "tokens_seen": 294453248 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046002006018054165, + "loss": 1.6521, + "theoretical_loss": 4.155414895868282, + "tokens_seen": 294518784 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046001003009027083, + "loss": 1.6095, + "theoretical_loss": 4.155306094042373, + "tokens_seen": 294584320 + }, + { + "epoch": 0.09, + "learning_rate": 0.00046, + "loss": 1.597, + "theoretical_loss": 4.155197323194555, + "tokens_seen": 294649856 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004599899699097292, + "loss": 1.4715, + "theoretical_loss": 4.155088583309118, + "tokens_seen": 294715392 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045997993981945843, + "loss": 1.4991, + "theoretical_loss": 4.154979874370367, + "tokens_seen": 294780928 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045996990972918756, + "loss": 1.6302, + "theoretical_loss": 4.154871196362619, + "tokens_seen": 294846464 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0006301368121057749, + "objective/train/docs_used": 179027, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.38765811920166, + "objective/train/original_loss": 3.38765811920166, + "objective/train/theoretical_loss": 4.154762549270199, + "objective/train/tokens_used": 315372000, + "objective/train/value_avg": -0.005222320556640625, + "objective/train/value_loss": 5.285854058456607e-05, + "objective/train/value_max": -0.0003712177276611328, + "objective/train/value_min": -0.07366943359375, + "objective/train/value_reward_corr": 0.3310514099543329, + "objective/train/value_std": 0.004184722900390625, + "objective/train/weight_avg": 0.9962450265884399, + "objective/train/weighted_lm_loss": 3.3746256828308105, + "objective/train/weights_max": 1.9690202474594116, + "objective/train/weights_min": 0.4357612729072571, + "theoretical_loss": 4.154762549270199, + "tokens_seen": 294912000 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004599598796389168, + "loss": 1.554, + "theoretical_loss": 4.154762549270199, + "tokens_seen": 294912000 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004599498495486459, + "loss": 1.5292, + "theoretical_loss": 4.154653933077447, + "tokens_seen": 294977536 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045993981945837515, + "loss": 1.6494, + "theoretical_loss": 4.154545347768712, + "tokens_seen": 295043072 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045992978936810434, + "loss": 1.6011, + "theoretical_loss": 4.1544367933283555, + "tokens_seen": 295108608 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004599197592778335, + "loss": 1.4503, + "theoretical_loss": 4.154328269740751, + "tokens_seen": 295174144 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004599097291875627, + "loss": 1.5452, + "theoretical_loss": 4.154219776990281, + "tokens_seen": 295239680 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004598996990972919, + "loss": 1.5551, + "theoretical_loss": 4.154111315061342, + "tokens_seen": 295305216 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045988966900702106, + "loss": 1.5598, + "theoretical_loss": 4.1540028839383405, + "tokens_seen": 295370752 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004598796389167503, + "loss": 1.6139, + "theoretical_loss": 4.153894483605694, + "tokens_seen": 295436288 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004598696088264794, + "loss": 1.6887, + "theoretical_loss": 4.153786114047834, + "tokens_seen": 295501824 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045985957873620866, + "loss": 1.5244, + "theoretical_loss": 4.153677775249198, + "tokens_seen": 295567360 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004598495486459378, + "loss": 1.6635, + "theoretical_loss": 4.153569467194241, + "tokens_seen": 295632896 + }, + { + "epoch": 0.09, + "learning_rate": 0.000459839518555667, + "loss": 1.6279, + "theoretical_loss": 4.153461189867425, + "tokens_seen": 295698432 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004598294884653962, + "loss": 1.4832, + "theoretical_loss": 4.153352943253225, + "tokens_seen": 295763968 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004598194583751254, + "loss": 1.5679, + "theoretical_loss": 4.153244727336126, + "tokens_seen": 295829504 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045980942828485456, + "loss": 1.6175, + "theoretical_loss": 4.153136542100627, + "tokens_seen": 295895040 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004597993981945838, + "loss": 1.4254, + "theoretical_loss": 4.153028387531236, + "tokens_seen": 295960576 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004597893681043129, + "loss": 1.5946, + "theoretical_loss": 4.152920263612472, + "tokens_seen": 296026112 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045977933801404216, + "loss": 1.5398, + "theoretical_loss": 4.1528121703288665, + "tokens_seen": 296091648 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004597693079237713, + "loss": 1.5073, + "theoretical_loss": 4.152704107664961, + "tokens_seen": 296157184 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004597592778335005, + "loss": 1.5443, + "theoretical_loss": 4.15259607560531, + "tokens_seen": 296222720 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004597492477432297, + "loss": 1.4834, + "theoretical_loss": 4.152488074134478, + "tokens_seen": 296288256 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004597392176529589, + "loss": 1.5727, + "theoretical_loss": 4.15238010323704, + "tokens_seen": 296353792 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045972918756268807, + "loss": 1.5757, + "theoretical_loss": 4.152272162897584, + "tokens_seen": 296419328 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045971915747241725, + "loss": 1.5476, + "theoretical_loss": 4.1521642531007075, + "tokens_seen": 296484864 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0008237991132773459, + "objective/train/docs_used": 179648, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.522160768508911, + "objective/train/original_loss": 3.522160768508911, + "objective/train/theoretical_loss": 4.15205637383102, + "objective/train/tokens_used": 317010400, + "objective/train/value_avg": -0.0045318603515625, + "objective/train/value_loss": 7.343762990785763e-05, + "objective/train/value_max": -0.0003814697265625, + "objective/train/value_min": -0.04986572265625, + "objective/train/value_reward_corr": 0.20597697964111317, + "objective/train/value_std": 0.0032558441162109375, + "objective/train/weight_avg": 0.9950143098831177, + "objective/train/weighted_lm_loss": 3.507246971130371, + "objective/train/weights_max": 1.5455286502838135, + "objective/train/weights_min": 0.19632835686206818, + "theoretical_loss": 4.15205637383102, + "tokens_seen": 296550400 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045970912738214643, + "loss": 1.5541, + "theoretical_loss": 4.15205637383102, + "tokens_seen": 296550400 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045969909729187566, + "loss": 1.5858, + "theoretical_loss": 4.151948525073142, + "tokens_seen": 296615936 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004596890672016048, + "loss": 1.4268, + "theoretical_loss": 4.151840706811706, + "tokens_seen": 296681472 + }, + { + "epoch": 0.09, + "learning_rate": 0.000459679037111334, + "loss": 1.4848, + "theoretical_loss": 4.151732919031354, + "tokens_seen": 296747008 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045966900702106315, + "loss": 1.5767, + "theoretical_loss": 4.15162516171674, + "tokens_seen": 296812544 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004596589769307924, + "loss": 1.5425, + "theoretical_loss": 4.151517434852528, + "tokens_seen": 296878080 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045964894684052157, + "loss": 1.5809, + "theoretical_loss": 4.151409738423396, + "tokens_seen": 296943616 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045963891675025075, + "loss": 1.5846, + "theoretical_loss": 4.151302072414031, + "tokens_seen": 297009152 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045962888665997993, + "loss": 1.5818, + "theoretical_loss": 4.15119443680913, + "tokens_seen": 297074688 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045961885656970917, + "loss": 1.6456, + "theoretical_loss": 4.151086831593402, + "tokens_seen": 297140224 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004596088264794383, + "loss": 1.5178, + "theoretical_loss": 4.15097925675157, + "tokens_seen": 297205760 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045959879638916753, + "loss": 1.6039, + "theoretical_loss": 4.150871712268364, + "tokens_seen": 297271296 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045958876629889666, + "loss": 1.5413, + "theoretical_loss": 4.150764198128526, + "tokens_seen": 297336832 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004595787362086259, + "loss": 1.6696, + "theoretical_loss": 4.15065671431681, + "tokens_seen": 297402368 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004595687061183551, + "loss": 1.5826, + "theoretical_loss": 4.15054926081798, + "tokens_seen": 297467904 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045955867602808425, + "loss": 1.4989, + "theoretical_loss": 4.150441837616814, + "tokens_seen": 297533440 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045954864593781344, + "loss": 1.4177, + "theoretical_loss": 4.150334444698096, + "tokens_seen": 297598976 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004595386158475426, + "loss": 1.5013, + "theoretical_loss": 4.150227082046624, + "tokens_seen": 297664512 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004595285857572718, + "loss": 1.4151, + "theoretical_loss": 4.150119749647207, + "tokens_seen": 297730048 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045951855566700103, + "loss": 1.5105, + "theoretical_loss": 4.150012447484665, + "tokens_seen": 297795584 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045950852557673016, + "loss": 1.5883, + "theoretical_loss": 4.149905175543827, + "tokens_seen": 297861120 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004594984954864594, + "loss": 1.5261, + "theoretical_loss": 4.149797933809536, + "tokens_seen": 297926656 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004594884653961885, + "loss": 1.5689, + "theoretical_loss": 4.149690722266644, + "tokens_seen": 297992192 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045947843530591776, + "loss": 1.5412, + "theoretical_loss": 4.1495835409000135, + "tokens_seen": 298057728 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045946840521564694, + "loss": 1.535, + "theoretical_loss": 4.149476389694518, + "tokens_seen": 298123264 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0014343935763463378, + "objective/train/docs_used": 180307, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.368480920791626, + "objective/train/original_loss": 3.368480682373047, + "objective/train/theoretical_loss": 4.149369268635046, + "objective/train/tokens_used": 318648800, + "objective/train/value_avg": -0.00555419921875, + "objective/train/value_loss": 9.708144352771342e-05, + "objective/train/value_max": -0.00036978721618652344, + "objective/train/value_min": -0.10687255859375, + "objective/train/value_reward_corr": 0.28793158878694924, + "objective/train/value_std": 0.0045166015625, + "objective/train/weight_avg": 0.9901039600372314, + "objective/train/weighted_lm_loss": 3.3476951122283936, + "objective/train/weights_max": 2.6038570404052734, + "objective/train/weights_min": 0.37465256452560425, + "theoretical_loss": 4.149369268635046, + "tokens_seen": 298188800 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004594583751253761, + "loss": 1.4911, + "theoretical_loss": 4.149369268635046, + "tokens_seen": 298188800 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045944834503510536, + "loss": 1.5197, + "theoretical_loss": 4.14926217770649, + "tokens_seen": 298254336 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045943831494483454, + "loss": 1.6268, + "theoretical_loss": 4.149155116893757, + "tokens_seen": 298319872 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004594282848545637, + "loss": 1.4542, + "theoretical_loss": 4.149048086181768, + "tokens_seen": 298385408 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004594182547642929, + "loss": 1.3778, + "theoretical_loss": 4.148941085555448, + "tokens_seen": 298450944 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004594082246740221, + "loss": 1.4631, + "theoretical_loss": 4.1488341149997385, + "tokens_seen": 298516480 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045939819458375126, + "loss": 1.6057, + "theoretical_loss": 4.148727174499589, + "tokens_seen": 298582016 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004593881644934805, + "loss": 1.5911, + "theoretical_loss": 4.148620264039961, + "tokens_seen": 298647552 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004593781344032096, + "loss": 1.5624, + "theoretical_loss": 4.148513383605826, + "tokens_seen": 298713088 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045936810431293886, + "loss": 1.5165, + "theoretical_loss": 4.1484065331821665, + "tokens_seen": 298778624 + }, + { + "epoch": 0.09, + "learning_rate": 0.000459358074222668, + "loss": 1.5142, + "theoretical_loss": 4.148299712753977, + "tokens_seen": 298844160 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004593480441323972, + "loss": 1.6168, + "theoretical_loss": 4.1481929223062615, + "tokens_seen": 298909696 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004593380140421264, + "loss": 1.5344, + "theoretical_loss": 4.148086161824035, + "tokens_seen": 298975232 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004593279839518556, + "loss": 1.5565, + "theoretical_loss": 4.147979431292324, + "tokens_seen": 299040768 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045931795386158476, + "loss": 1.4859, + "theoretical_loss": 4.147872730696165, + "tokens_seen": 299106304 + }, + { + "epoch": 0.09, + "learning_rate": 0.000459307923771314, + "loss": 1.502, + "theoretical_loss": 4.147766060020605, + "tokens_seen": 299171840 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004592978936810431, + "loss": 1.4898, + "theoretical_loss": 4.147659419250703, + "tokens_seen": 299237376 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045928786359077236, + "loss": 1.6332, + "theoretical_loss": 4.147552808371526, + "tokens_seen": 299302912 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004592778335005015, + "loss": 1.5316, + "theoretical_loss": 4.147446227368155, + "tokens_seen": 299368448 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004592678034102307, + "loss": 1.5681, + "theoretical_loss": 4.147339676225681, + "tokens_seen": 299433984 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004592577733199599, + "loss": 1.6629, + "theoretical_loss": 4.1472331549292045, + "tokens_seen": 299499520 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004592477432296891, + "loss": 1.5835, + "theoretical_loss": 4.147126663463837, + "tokens_seen": 299565056 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045923771313941827, + "loss": 1.5054, + "theoretical_loss": 4.1470202018147, + "tokens_seen": 299630592 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045922768304914745, + "loss": 1.5481, + "theoretical_loss": 4.146913769966929, + "tokens_seen": 299696128 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045921765295887663, + "loss": 1.5964, + "theoretical_loss": 4.146807367905666, + "tokens_seen": 299761664 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0010667219758033752, + "objective/train/docs_used": 181680, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8599600791931152, + "objective/train/original_loss": 2.8599600791931152, + "objective/train/theoretical_loss": 4.146700995616065, + "objective/train/tokens_used": 320287200, + "objective/train/value_avg": -0.004474639892578125, + "objective/train/value_loss": 5.703698843717575e-05, + "objective/train/value_max": -0.0003654956817626953, + "objective/train/value_min": -0.05560302734375, + "objective/train/value_reward_corr": 0.23961405523519758, + "objective/train/value_std": 0.003734588623046875, + "objective/train/weight_avg": 0.9920263290405273, + "objective/train/weighted_lm_loss": 2.8595850467681885, + "objective/train/weights_max": 1.6155805587768555, + "objective/train/weights_min": 0.3474769592285156, + "theoretical_loss": 4.146700995616065, + "tokens_seen": 299827200 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045920762286860586, + "loss": 1.5318, + "theoretical_loss": 4.146700995616065, + "tokens_seen": 299827200 + }, + { + "epoch": 0.09, + "learning_rate": 0.000459197592778335, + "loss": 1.4604, + "theoretical_loss": 4.146594653083293, + "tokens_seen": 299892736 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004591875626880642, + "loss": 1.4025, + "theoretical_loss": 4.146488340292523, + "tokens_seen": 299958272 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045917753259779335, + "loss": 1.4826, + "theoretical_loss": 4.146382057228943, + "tokens_seen": 300023808 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004591675025075226, + "loss": 1.591, + "theoretical_loss": 4.146275803877751, + "tokens_seen": 300089344 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045915747241725177, + "loss": 1.5431, + "theoretical_loss": 4.146169580224152, + "tokens_seen": 300154880 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045914744232698095, + "loss": 1.5921, + "theoretical_loss": 4.146063386253366, + "tokens_seen": 300220416 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045913741223671013, + "loss": 1.4232, + "theoretical_loss": 4.145957221950621, + "tokens_seen": 300285952 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045912738214643937, + "loss": 1.5201, + "theoretical_loss": 4.145851087301156, + "tokens_seen": 300351488 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004591173520561685, + "loss": 1.4749, + "theoretical_loss": 4.145744982290221, + "tokens_seen": 300417024 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045910732196589773, + "loss": 1.4949, + "theoretical_loss": 4.145638906903077, + "tokens_seen": 300482560 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045909729187562686, + "loss": 1.578, + "theoretical_loss": 4.145532861124994, + "tokens_seen": 300548096 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004590872617853561, + "loss": 1.4588, + "theoretical_loss": 4.1454268449412535, + "tokens_seen": 300613632 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004590772316950853, + "loss": 1.5198, + "theoretical_loss": 4.145320858337149, + "tokens_seen": 300679168 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045906720160481445, + "loss": 1.4263, + "theoretical_loss": 4.145214901297981, + "tokens_seen": 300744704 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045905717151454364, + "loss": 1.4787, + "theoretical_loss": 4.145108973809064, + "tokens_seen": 300810240 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004590471414242728, + "loss": 1.4909, + "theoretical_loss": 4.145003075855721, + "tokens_seen": 300875776 + }, + { + "epoch": 0.09, + "learning_rate": 0.000459037111334002, + "loss": 1.3304, + "theoretical_loss": 4.144897207423284, + "tokens_seen": 300941312 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045902708124373123, + "loss": 1.4995, + "theoretical_loss": 4.144791368497101, + "tokens_seen": 301006848 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045901705115346036, + "loss": 1.6309, + "theoretical_loss": 4.144685559062525, + "tokens_seen": 301072384 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004590070210631896, + "loss": 1.414, + "theoretical_loss": 4.144579779104921, + "tokens_seen": 301137920 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004589969909729187, + "loss": 1.56, + "theoretical_loss": 4.1444740286096655, + "tokens_seen": 301203456 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045898696088264796, + "loss": 1.4089, + "theoretical_loss": 4.144368307562145, + "tokens_seen": 301268992 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045897693079237714, + "loss": 1.6185, + "theoretical_loss": 4.144262615947755, + "tokens_seen": 301334528 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004589669007021063, + "loss": 1.6037, + "theoretical_loss": 4.144156953751904, + "tokens_seen": 301400064 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.001217499258928001, + "objective/train/docs_used": 182385, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.6162219047546387, + "objective/train/original_loss": 3.6162219047546387, + "objective/train/theoretical_loss": 4.144051320960009, + "objective/train/tokens_used": 321925600, + "objective/train/value_avg": -0.0051116943359375, + "objective/train/value_loss": 7.139430090319365e-05, + "objective/train/value_max": -0.00028252601623535156, + "objective/train/value_min": -0.09075927734375, + "objective/train/value_reward_corr": 0.35250839852620613, + "objective/train/value_std": 0.004482269287109375, + "objective/train/weight_avg": 0.9911404252052307, + "objective/train/weighted_lm_loss": 3.5865681171417236, + "objective/train/weights_max": 2.4023325443267822, + "objective/train/weights_min": 0.2510486841201782, + "theoretical_loss": 4.144051320960009, + "tokens_seen": 301465600 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004589568706118355, + "loss": 1.6178, + "theoretical_loss": 4.144051320960009, + "tokens_seen": 301465600 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045894684052156474, + "loss": 1.4608, + "theoretical_loss": 4.143945717557497, + "tokens_seen": 301531136 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045893681043129386, + "loss": 1.4785, + "theoretical_loss": 4.143840143529808, + "tokens_seen": 301596672 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004589267803410231, + "loss": 1.5795, + "theoretical_loss": 4.143734598862389, + "tokens_seen": 301662208 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004589167502507522, + "loss": 1.5796, + "theoretical_loss": 4.1436290835407, + "tokens_seen": 301727744 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045890672016048146, + "loss": 1.4582, + "theoretical_loss": 4.14352359755021, + "tokens_seen": 301793280 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045889669007021064, + "loss": 1.5165, + "theoretical_loss": 4.143418140876399, + "tokens_seen": 301858816 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004588866599799398, + "loss": 1.5415, + "theoretical_loss": 4.1433127135047565, + "tokens_seen": 301924352 + }, + { + "epoch": 0.09, + "learning_rate": 0.000458876629889669, + "loss": 1.5361, + "theoretical_loss": 4.143207315420783, + "tokens_seen": 301989888 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004588665997993982, + "loss": 1.4726, + "theoretical_loss": 4.143101946609991, + "tokens_seen": 302055424 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045885656970912737, + "loss": 1.5699, + "theoretical_loss": 4.142996607057899, + "tokens_seen": 302120960 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004588465396188566, + "loss": 1.5466, + "theoretical_loss": 4.142891296750039, + "tokens_seen": 302186496 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045883650952858573, + "loss": 1.6906, + "theoretical_loss": 4.142786015671954, + "tokens_seen": 302252032 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045882647943831496, + "loss": 1.5816, + "theoretical_loss": 4.142680763809194, + "tokens_seen": 302317568 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045881644934804415, + "loss": 1.629, + "theoretical_loss": 4.142575541147323, + "tokens_seen": 302383104 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004588064192577733, + "loss": 1.5665, + "theoretical_loss": 4.142470347671912, + "tokens_seen": 302448640 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004587963891675025, + "loss": 1.6255, + "theoretical_loss": 4.142365183368544, + "tokens_seen": 302514176 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004587863590772317, + "loss": 1.5647, + "theoretical_loss": 4.142260048222812, + "tokens_seen": 302579712 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045877632898696087, + "loss": 1.5388, + "theoretical_loss": 4.142154942220319, + "tokens_seen": 302645248 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004587662988966901, + "loss": 1.5468, + "theoretical_loss": 4.14204986534668, + "tokens_seen": 302710784 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045875626880641923, + "loss": 1.5096, + "theoretical_loss": 4.141944817587516, + "tokens_seen": 302776320 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045874623871614847, + "loss": 1.527, + "theoretical_loss": 4.141839798928462, + "tokens_seen": 302841856 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004587362086258776, + "loss": 1.5532, + "theoretical_loss": 4.141734809355163, + "tokens_seen": 302907392 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045872617853560683, + "loss": 1.6143, + "theoretical_loss": 4.141629848853272, + "tokens_seen": 302972928 + }, + { + "epoch": 0.09, + "learning_rate": 0.000458716148445336, + "loss": 1.5166, + "theoretical_loss": 4.141524917408454, + "tokens_seen": 303038464 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": 0.0005927731399424374, + "objective/train/docs_used": 183646, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.123976230621338, + "objective/train/original_loss": 3.123975992202759, + "objective/train/theoretical_loss": 4.141420015006382, + "objective/train/tokens_used": 323564000, + "objective/train/value_avg": -0.00528717041015625, + "objective/train/value_loss": 6.0399128415156156e-05, + "objective/train/value_max": -0.00026535987854003906, + "objective/train/value_min": -0.06768798828125, + "objective/train/value_reward_corr": 0.2281788722176918, + "objective/train/value_std": 0.0042877197265625, + "objective/train/weight_avg": 1.0086904764175415, + "objective/train/weighted_lm_loss": 3.16738224029541, + "objective/train/weights_max": 1.9677284955978394, + "objective/train/weights_min": 0.08473040163516998, + "theoretical_loss": 4.141420015006382, + "tokens_seen": 303104000 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004587061183550652, + "loss": 1.5999, + "theoretical_loss": 4.141420015006382, + "tokens_seen": 303104000 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004586960882647944, + "loss": 1.5875, + "theoretical_loss": 4.1413151416327425, + "tokens_seen": 303169536 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045868605817452355, + "loss": 1.5466, + "theoretical_loss": 4.14121029727323, + "tokens_seen": 303235072 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004586760280842528, + "loss": 1.5457, + "theoretical_loss": 4.141105481913549, + "tokens_seen": 303300608 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045866599799398197, + "loss": 1.619, + "theoretical_loss": 4.1410006955394145, + "tokens_seen": 303366144 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045865596790371115, + "loss": 1.4476, + "theoretical_loss": 4.140895938136552, + "tokens_seen": 303431680 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045864593781344033, + "loss": 1.6524, + "theoretical_loss": 4.140791209690697, + "tokens_seen": 303497216 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045863590772316957, + "loss": 1.5308, + "theoretical_loss": 4.140686510187595, + "tokens_seen": 303562752 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004586258776328987, + "loss": 1.6556, + "theoretical_loss": 4.1405818396130005, + "tokens_seen": 303628288 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045861584754262793, + "loss": 1.5402, + "theoretical_loss": 4.140477197952681, + "tokens_seen": 303693824 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045860581745235706, + "loss": 1.5556, + "theoretical_loss": 4.14037258519241, + "tokens_seen": 303759360 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004585957873620863, + "loss": 1.542, + "theoretical_loss": 4.140268001317976, + "tokens_seen": 303824896 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004585857572718155, + "loss": 1.6296, + "theoretical_loss": 4.140163446315172, + "tokens_seen": 303890432 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045857572718154465, + "loss": 1.6179, + "theoretical_loss": 4.140058920169805, + "tokens_seen": 303955968 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045856569709127384, + "loss": 1.5221, + "theoretical_loss": 4.139954422867691, + "tokens_seen": 304021504 + }, + { + "epoch": 0.09, + "learning_rate": 0.000458555667001003, + "loss": 1.6287, + "theoretical_loss": 4.1398499543946565, + "tokens_seen": 304087040 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004585456369107322, + "loss": 1.5956, + "theoretical_loss": 4.139745514736537, + "tokens_seen": 304152576 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045853560682046143, + "loss": 1.4579, + "theoretical_loss": 4.139641103879177, + "tokens_seen": 304218112 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045852557673019056, + "loss": 1.6104, + "theoretical_loss": 4.139536721808435, + "tokens_seen": 304283648 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004585155466399198, + "loss": 1.5427, + "theoretical_loss": 4.1394323685101755, + "tokens_seen": 304349184 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004585055165496489, + "loss": 1.4851, + "theoretical_loss": 4.139328043970274, + "tokens_seen": 304414720 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045849548645937816, + "loss": 1.6082, + "theoretical_loss": 4.139223748174618, + "tokens_seen": 304480256 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045848545636910734, + "loss": 1.5393, + "theoretical_loss": 4.139119481109101, + "tokens_seen": 304545792 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004584754262788365, + "loss": 1.469, + "theoretical_loss": 4.139015242759631, + "tokens_seen": 304611328 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004584653961885657, + "loss": 1.5457, + "theoretical_loss": 4.138911033112123, + "tokens_seen": 304676864 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": 0.000655351032037288, + "objective/train/docs_used": 184075, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6766202449798584, + "objective/train/original_loss": 2.6766204833984375, + "objective/train/theoretical_loss": 4.138806852152502, + "objective/train/tokens_used": 325202400, + "objective/train/value_avg": -0.004726409912109375, + "objective/train/value_loss": 5.846222484251484e-05, + "objective/train/value_max": -0.0004935264587402344, + "objective/train/value_min": -0.045440673828125, + "objective/train/value_reward_corr": 0.2678873451056279, + "objective/train/value_std": 0.0036373138427734375, + "objective/train/weight_avg": 1.009247899055481, + "objective/train/weighted_lm_loss": 2.7024953365325928, + "objective/train/weights_max": 1.4808696508407593, + "objective/train/weights_min": 0.29267987608909607, + "theoretical_loss": 4.138806852152502, + "tokens_seen": 304742400 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045845536609829494, + "loss": 1.4652, + "theoretical_loss": 4.138806852152502, + "tokens_seen": 304742400 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045844533600802406, + "loss": 1.5572, + "theoretical_loss": 4.1387026998667045, + "tokens_seen": 304807936 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004584353059177533, + "loss": 1.5512, + "theoretical_loss": 4.138598576240675, + "tokens_seen": 304873472 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004584252758274824, + "loss": 1.4503, + "theoretical_loss": 4.138494481260371, + "tokens_seen": 304939008 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045841524573721166, + "loss": 1.5364, + "theoretical_loss": 4.138390414911756, + "tokens_seen": 305004544 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045840521564694084, + "loss": 1.5996, + "theoretical_loss": 4.138286377180805, + "tokens_seen": 305070080 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045839518555667, + "loss": 1.5888, + "theoretical_loss": 4.138182368053505, + "tokens_seen": 305135616 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004583851554663992, + "loss": 1.4439, + "theoretical_loss": 4.138078387515849, + "tokens_seen": 305201152 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004583751253761284, + "loss": 1.5927, + "theoretical_loss": 4.1379744355538435, + "tokens_seen": 305266688 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045836509528585757, + "loss": 1.4484, + "theoretical_loss": 4.137870512153501, + "tokens_seen": 305332224 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004583550651955868, + "loss": 1.6667, + "theoretical_loss": 4.137766617300849, + "tokens_seen": 305397760 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045834503510531593, + "loss": 1.5343, + "theoretical_loss": 4.137662750981919, + "tokens_seen": 305463296 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045833500501504516, + "loss": 1.567, + "theoretical_loss": 4.137558913182758, + "tokens_seen": 305528832 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045832497492477435, + "loss": 1.5407, + "theoretical_loss": 4.137455103889417, + "tokens_seen": 305594368 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004583149448345035, + "loss": 1.5117, + "theoretical_loss": 4.137351323087962, + "tokens_seen": 305659904 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004583049147442327, + "loss": 1.5372, + "theoretical_loss": 4.137247570764467, + "tokens_seen": 305725440 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004582948846539619, + "loss": 1.5589, + "theoretical_loss": 4.137143846905014, + "tokens_seen": 305790976 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045828485456369107, + "loss": 1.5309, + "theoretical_loss": 4.137040151495696, + "tokens_seen": 305856512 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004582748244734203, + "loss": 1.537, + "theoretical_loss": 4.136936484522616, + "tokens_seen": 305922048 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045826479438314943, + "loss": 1.4577, + "theoretical_loss": 4.136832845971888, + "tokens_seen": 305987584 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045825476429287867, + "loss": 1.534, + "theoretical_loss": 4.136729235829634, + "tokens_seen": 306053120 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004582447342026078, + "loss": 1.5326, + "theoretical_loss": 4.136625654081986, + "tokens_seen": 306118656 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045823470411233703, + "loss": 1.6718, + "theoretical_loss": 4.136522100715087, + "tokens_seen": 306184192 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004582246740220662, + "loss": 1.4695, + "theoretical_loss": 4.136418575715086, + "tokens_seen": 306249728 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004582146439317954, + "loss": 1.599, + "theoretical_loss": 4.136315079068147, + "tokens_seen": 306315264 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.00030646490631625056, + "objective/train/docs_used": 185377, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.445709466934204, + "objective/train/original_loss": 3.445709466934204, + "objective/train/theoretical_loss": 4.136211610760441, + "objective/train/tokens_used": 326840800, + "objective/train/value_avg": -0.006084442138671875, + "objective/train/value_loss": 5.714117287425324e-05, + "objective/train/value_max": -0.0002758502960205078, + "objective/train/value_min": -0.0469970703125, + "objective/train/value_reward_corr": 0.37569914768674884, + "objective/train/value_std": 0.00482177734375, + "objective/train/weight_avg": 0.9997056126594543, + "objective/train/weighted_lm_loss": 3.455996513366699, + "objective/train/weights_max": 1.5197746753692627, + "objective/train/weights_min": 0.5575826168060303, + "theoretical_loss": 4.136211610760441, + "tokens_seen": 306380800 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045820461384152457, + "loss": 1.6599, + "theoretical_loss": 4.136211610760441, + "tokens_seen": 306380800 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045819458375125375, + "loss": 1.412, + "theoretical_loss": 4.136108170778147, + "tokens_seen": 306446336 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045818455366098294, + "loss": 1.5267, + "theoretical_loss": 4.136004759107458, + "tokens_seen": 306511872 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045817452357071217, + "loss": 1.5585, + "theoretical_loss": 4.135901375734571, + "tokens_seen": 306577408 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004581644934804413, + "loss": 1.5096, + "theoretical_loss": 4.1357980206457, + "tokens_seen": 306642944 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045815446339017053, + "loss": 1.4925, + "theoretical_loss": 4.135694693827061, + "tokens_seen": 306708480 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004581444332998997, + "loss": 1.5877, + "theoretical_loss": 4.135591395264884, + "tokens_seen": 306774016 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004581344032096289, + "loss": 1.5545, + "theoretical_loss": 4.135488124945409, + "tokens_seen": 306839552 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004581243731193581, + "loss": 1.554, + "theoretical_loss": 4.135384882854884, + "tokens_seen": 306905088 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045811434302908726, + "loss": 1.4685, + "theoretical_loss": 4.135281668979566, + "tokens_seen": 306970624 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045810431293881644, + "loss": 1.525, + "theoretical_loss": 4.135178483305725, + "tokens_seen": 307036160 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004580942828485457, + "loss": 1.5178, + "theoretical_loss": 4.135075325819637, + "tokens_seen": 307101696 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004580842527582748, + "loss": 1.6113, + "theoretical_loss": 4.134972196507588, + "tokens_seen": 307167232 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045807422266800404, + "loss": 1.4209, + "theoretical_loss": 4.134869095355876, + "tokens_seen": 307232768 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045806419257773316, + "loss": 1.5233, + "theoretical_loss": 4.134766022350807, + "tokens_seen": 307298304 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004580541624874624, + "loss": 1.5126, + "theoretical_loss": 4.134662977478697, + "tokens_seen": 307363840 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004580441323971916, + "loss": 1.589, + "theoretical_loss": 4.13455996072587, + "tokens_seen": 307429376 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045803410230692076, + "loss": 1.4424, + "theoretical_loss": 4.134456972078661, + "tokens_seen": 307494912 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045802407221664994, + "loss": 1.5817, + "theoretical_loss": 4.134354011523417, + "tokens_seen": 307560448 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004580140421263791, + "loss": 1.5464, + "theoretical_loss": 4.134251079046488, + "tokens_seen": 307625984 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004580040120361083, + "loss": 1.6068, + "theoretical_loss": 4.13414817463424, + "tokens_seen": 307691520 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045799398194583754, + "loss": 1.5386, + "theoretical_loss": 4.134045298273046, + "tokens_seen": 307757056 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045798395185556667, + "loss": 1.5899, + "theoretical_loss": 4.133942449949287, + "tokens_seen": 307822592 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004579739217652959, + "loss": 1.4562, + "theoretical_loss": 4.133839629649357, + "tokens_seen": 307888128 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004579638916750251, + "loss": 1.5075, + "theoretical_loss": 4.133736837359656, + "tokens_seen": 307953664 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": 1.8867878679884598e-05, + "objective/train/docs_used": 186163, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.088027238845825, + "objective/train/original_loss": 3.088027000427246, + "objective/train/theoretical_loss": 4.133634073066595, + "objective/train/tokens_used": 328479200, + "objective/train/value_avg": -0.00536346435546875, + "objective/train/value_loss": 5.620209776679985e-05, + "objective/train/value_max": -0.0003554821014404297, + "objective/train/value_min": -0.0726318359375, + "objective/train/value_reward_corr": 0.43898151239113115, + "objective/train/value_std": 0.004520416259765625, + "objective/train/weight_avg": 1.0029020309448242, + "objective/train/weighted_lm_loss": 3.1078362464904785, + "objective/train/weights_max": 1.754947543144226, + "objective/train/weights_min": 0.5918911099433899, + "theoretical_loss": 4.133634073066595, + "tokens_seen": 308019200 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045795386158475426, + "loss": 1.4791, + "theoretical_loss": 4.133634073066595, + "tokens_seen": 308019200 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004579438314944835, + "loss": 1.6484, + "theoretical_loss": 4.133531336756596, + "tokens_seen": 308084736 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004579338014042126, + "loss": 1.6555, + "theoretical_loss": 4.1334286284160875, + "tokens_seen": 308150272 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045792377131394186, + "loss": 1.5289, + "theoretical_loss": 4.1333259480315085, + "tokens_seen": 308215808 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045791374122367104, + "loss": 1.486, + "theoretical_loss": 4.1332232955893105, + "tokens_seen": 308281344 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004579037111334002, + "loss": 1.5563, + "theoretical_loss": 4.133120671075949, + "tokens_seen": 308346880 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004578936810431294, + "loss": 1.7262, + "theoretical_loss": 4.133018074477894, + "tokens_seen": 308412416 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004578836509528586, + "loss": 1.5871, + "theoretical_loss": 4.132915505781621, + "tokens_seen": 308477952 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045787362086258777, + "loss": 1.4569, + "theoretical_loss": 4.132812964973617, + "tokens_seen": 308543488 + }, + { + "epoch": 0.09, + "learning_rate": 0.000457863590772317, + "loss": 1.569, + "theoretical_loss": 4.132710452040378, + "tokens_seen": 308609024 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045785356068204613, + "loss": 1.441, + "theoretical_loss": 4.132607966968411, + "tokens_seen": 308674560 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045784353059177536, + "loss": 1.6295, + "theoretical_loss": 4.132505509744229, + "tokens_seen": 308740096 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045783350050150455, + "loss": 1.4682, + "theoretical_loss": 4.132403080354357, + "tokens_seen": 308805632 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004578234704112337, + "loss": 1.466, + "theoretical_loss": 4.132300678785328, + "tokens_seen": 308871168 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004578134403209629, + "loss": 1.5616, + "theoretical_loss": 4.132198305023686, + "tokens_seen": 308936704 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004578034102306921, + "loss": 1.4189, + "theoretical_loss": 4.132095959055983, + "tokens_seen": 309002240 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045779338014042127, + "loss": 1.5444, + "theoretical_loss": 4.13199364086878, + "tokens_seen": 309067776 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004577833500501505, + "loss": 1.6924, + "theoretical_loss": 4.131891350448649, + "tokens_seen": 309133312 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045777331995987963, + "loss": 1.562, + "theoretical_loss": 4.1317890877821695, + "tokens_seen": 309198848 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045776328986960887, + "loss": 1.4792, + "theoretical_loss": 4.131686852855932, + "tokens_seen": 309264384 + }, + { + "epoch": 0.09, + "learning_rate": 0.000457753259779338, + "loss": 1.5005, + "theoretical_loss": 4.131584645656535, + "tokens_seen": 309329920 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045774322968906723, + "loss": 1.6515, + "theoretical_loss": 4.131482466170587, + "tokens_seen": 309395456 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004577331995987964, + "loss": 1.6146, + "theoretical_loss": 4.131380314384706, + "tokens_seen": 309460992 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004577231695085256, + "loss": 1.5503, + "theoretical_loss": 4.131278190285518, + "tokens_seen": 309526528 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004577131394182548, + "loss": 1.532, + "theoretical_loss": 4.1311760938596604, + "tokens_seen": 309592064 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0020533830393105745, + "objective/train/docs_used": 187126, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9980552196502686, + "objective/train/original_loss": 2.9980549812316895, + "objective/train/theoretical_loss": 4.131074025093778, + "objective/train/tokens_used": 330117600, + "objective/train/value_avg": -0.0048980712890625, + "objective/train/value_loss": 0.00013585116539616138, + "objective/train/value_max": -0.0003249645233154297, + "objective/train/value_min": -0.466064453125, + "objective/train/value_reward_corr": 0.24131453962290178, + "objective/train/value_std": 0.006168365478515625, + "objective/train/weight_avg": 0.990337610244751, + "objective/train/weighted_lm_loss": 3.00663685798645, + "objective/train/weights_max": 90.07518768310547, + "objective/train/weights_min": 0.296698659658432, + "theoretical_loss": 4.131074025093778, + "tokens_seen": 309657600 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045770310932798395, + "loss": 1.5684, + "theoretical_loss": 4.131074025093778, + "tokens_seen": 309657600 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045769307923771314, + "loss": 1.5646, + "theoretical_loss": 4.130971983974526, + "tokens_seen": 309723136 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045768304914744237, + "loss": 1.6356, + "theoretical_loss": 4.130869970488567, + "tokens_seen": 309788672 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004576730190571715, + "loss": 1.4889, + "theoretical_loss": 4.130767984622577, + "tokens_seen": 309854208 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045766298896690073, + "loss": 1.5647, + "theoretical_loss": 4.130666026363236, + "tokens_seen": 309919744 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004576529588766299, + "loss": 1.3731, + "theoretical_loss": 4.1305640956972365, + "tokens_seen": 309985280 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004576429287863591, + "loss": 1.4562, + "theoretical_loss": 4.13046219261128, + "tokens_seen": 310050816 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004576328986960883, + "loss": 1.6086, + "theoretical_loss": 4.130360317092077, + "tokens_seen": 310116352 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045762286860581746, + "loss": 1.5602, + "theoretical_loss": 4.130258469126346, + "tokens_seen": 310181888 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045761283851554664, + "loss": 1.5758, + "theoretical_loss": 4.130156648700816, + "tokens_seen": 310247424 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004576028084252759, + "loss": 1.5912, + "theoretical_loss": 4.130054855802225, + "tokens_seen": 310312960 + }, + { + "epoch": 0.09, + "learning_rate": 0.000457592778335005, + "loss": 1.5646, + "theoretical_loss": 4.129953090417319, + "tokens_seen": 310378496 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045758274824473424, + "loss": 1.5533, + "theoretical_loss": 4.129851352532856, + "tokens_seen": 310444032 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045757271815446336, + "loss": 1.5263, + "theoretical_loss": 4.129749642135599, + "tokens_seen": 310509568 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004575626880641926, + "loss": 1.4797, + "theoretical_loss": 4.129647959212325, + "tokens_seen": 310575104 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004575526579739218, + "loss": 1.569, + "theoretical_loss": 4.129546303749816, + "tokens_seen": 310640640 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045754262788365096, + "loss": 1.4513, + "theoretical_loss": 4.129444675734866, + "tokens_seen": 310706176 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045753259779338014, + "loss": 1.5166, + "theoretical_loss": 4.129343075154275, + "tokens_seen": 310771712 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004575225677031093, + "loss": 1.4572, + "theoretical_loss": 4.129241501994857, + "tokens_seen": 310837248 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004575125376128385, + "loss": 1.5697, + "theoretical_loss": 4.129139956243429, + "tokens_seen": 310902784 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045750250752256774, + "loss": 1.5862, + "theoretical_loss": 4.129038437886822, + "tokens_seen": 310968320 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045749247743229687, + "loss": 1.5641, + "theoretical_loss": 4.128936946911875, + "tokens_seen": 311033856 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004574824473420261, + "loss": 1.685, + "theoretical_loss": 4.1288354833054335, + "tokens_seen": 311099392 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004574724172517553, + "loss": 1.61, + "theoretical_loss": 4.1287340470543565, + "tokens_seen": 311164928 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045746238716148446, + "loss": 1.6163, + "theoretical_loss": 4.128632638145508, + "tokens_seen": 311230464 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0019108790438622236, + "objective/train/docs_used": 187875, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.958167314529419, + "objective/train/original_loss": 2.958167314529419, + "objective/train/theoretical_loss": 4.128531256565763, + "objective/train/tokens_used": 331756000, + "objective/train/value_avg": -0.005878448486328125, + "objective/train/value_loss": 6.783901335438713e-05, + "objective/train/value_max": -0.0003814697265625, + "objective/train/value_min": -0.08343505859375, + "objective/train/value_reward_corr": 0.442782633297184, + "objective/train/value_std": 0.004947662353515625, + "objective/train/weight_avg": 0.9841439723968506, + "objective/train/weighted_lm_loss": 2.9012856483459473, + "objective/train/weights_max": 2.3033177852630615, + "objective/train/weights_min": 0.614088773727417, + "theoretical_loss": 4.128531256565763, + "tokens_seen": 311296000 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045745235707121364, + "loss": 1.4796, + "theoretical_loss": 4.128531256565763, + "tokens_seen": 311296000 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004574423269809428, + "loss": 1.4804, + "theoretical_loss": 4.128429902302006, + "tokens_seen": 311361536 + }, + { + "epoch": 0.09, + "learning_rate": 0.000457432296890672, + "loss": 1.6134, + "theoretical_loss": 4.128328575341129, + "tokens_seen": 311427072 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045742226680040124, + "loss": 1.6138, + "theoretical_loss": 4.128227275670034, + "tokens_seen": 311492608 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045741223671013037, + "loss": 1.5558, + "theoretical_loss": 4.128126003275633, + "tokens_seen": 311558144 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004574022066198596, + "loss": 1.613, + "theoretical_loss": 4.128024758144845, + "tokens_seen": 311623680 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045739217652958873, + "loss": 1.6107, + "theoretical_loss": 4.127923540264598, + "tokens_seen": 311689216 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045738214643931797, + "loss": 1.6196, + "theoretical_loss": 4.127822349621832, + "tokens_seen": 311754752 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045737211634904715, + "loss": 1.6168, + "theoretical_loss": 4.127721186203494, + "tokens_seen": 311820288 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045736208625877633, + "loss": 1.5625, + "theoretical_loss": 4.127620049996538, + "tokens_seen": 311885824 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004573520561685055, + "loss": 1.6361, + "theoretical_loss": 4.12751894098793, + "tokens_seen": 311951360 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045734202607823475, + "loss": 1.7018, + "theoretical_loss": 4.127417859164645, + "tokens_seen": 312016896 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045733199598796387, + "loss": 1.5341, + "theoretical_loss": 4.127316804513664, + "tokens_seen": 312082432 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004573219658976931, + "loss": 1.5511, + "theoretical_loss": 4.127215777021981, + "tokens_seen": 312147968 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045731193580742223, + "loss": 1.4236, + "theoretical_loss": 4.127114776676595, + "tokens_seen": 312213504 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045730190571715147, + "loss": 1.6484, + "theoretical_loss": 4.127013803464516, + "tokens_seen": 312279040 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045729187562688065, + "loss": 1.4543, + "theoretical_loss": 4.126912857372765, + "tokens_seen": 312344576 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045728184553660983, + "loss": 1.6077, + "theoretical_loss": 4.126811938388367, + "tokens_seen": 312410112 + }, + { + "epoch": 0.09, + "learning_rate": 0.000457271815446339, + "loss": 1.5992, + "theoretical_loss": 4.12671104649836, + "tokens_seen": 312475648 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004572617853560682, + "loss": 1.6403, + "theoretical_loss": 4.12661018168979, + "tokens_seen": 312541184 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004572517552657974, + "loss": 1.4591, + "theoretical_loss": 4.126509343949709, + "tokens_seen": 312606720 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004572417251755266, + "loss": 1.6549, + "theoretical_loss": 4.126408533265183, + "tokens_seen": 312672256 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045723169508525574, + "loss": 1.4931, + "theoretical_loss": 4.126307749623283, + "tokens_seen": 312737792 + }, + { + "epoch": 0.09, + "learning_rate": 0.000457221664994985, + "loss": 1.5394, + "theoretical_loss": 4.12620699301109, + "tokens_seen": 312803328 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004572116349047141, + "loss": 1.6583, + "theoretical_loss": 4.1261062634156955, + "tokens_seen": 312868864 + }, + { + "epoch": 0.09, + "objective/train/advantage_avg": -0.0004307777271606028, + "objective/train/docs_used": 188927, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3356974124908447, + "objective/train/original_loss": 3.3356971740722656, + "objective/train/theoretical_loss": 4.126005560824196, + "objective/train/tokens_used": 333394400, + "objective/train/value_avg": -0.005352020263671875, + "objective/train/value_loss": 6.015393591951579e-05, + "objective/train/value_max": -0.0006537437438964844, + "objective/train/value_min": -0.115966796875, + "objective/train/value_reward_corr": 0.43111152565993194, + "objective/train/value_std": 0.0049896240234375, + "objective/train/weight_avg": 0.9986037611961365, + "objective/train/weighted_lm_loss": 3.3340134620666504, + "objective/train/weights_max": 2.598384141921997, + "objective/train/weights_min": 0.3411398231983185, + "theoretical_loss": 4.126005560824196, + "tokens_seen": 312934400 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045720160481444334, + "loss": 1.6832, + "theoretical_loss": 4.126005560824196, + "tokens_seen": 312934400 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045719157472417257, + "loss": 1.6201, + "theoretical_loss": 4.125904885223701, + "tokens_seen": 312999936 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004571815446339017, + "loss": 1.7141, + "theoretical_loss": 4.125804236601327, + "tokens_seen": 313065472 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045717151454363093, + "loss": 1.5905, + "theoretical_loss": 4.1257036149441975, + "tokens_seen": 313131008 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004571614844533601, + "loss": 1.4913, + "theoretical_loss": 4.125603020239449, + "tokens_seen": 313196544 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004571514543630893, + "loss": 1.6039, + "theoretical_loss": 4.125502452474223, + "tokens_seen": 313262080 + }, + { + "epoch": 0.09, + "learning_rate": 0.0004571414242728185, + "loss": 1.5537, + "theoretical_loss": 4.125401911635674, + "tokens_seen": 313327616 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045713139418254766, + "loss": 1.6527, + "theoretical_loss": 4.125301397710959, + "tokens_seen": 313393152 + }, + { + "epoch": 0.09, + "learning_rate": 0.00045712136409227684, + "loss": 1.6407, + "theoretical_loss": 4.12520091068725, + "tokens_seen": 313458688 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004571113340020061, + "loss": 1.5822, + "theoretical_loss": 4.125100450551725, + "tokens_seen": 313524224 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004571013039117352, + "loss": 1.4627, + "theoretical_loss": 4.12500001729157, + "tokens_seen": 313589760 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045709127382146444, + "loss": 1.598, + "theoretical_loss": 4.124899610893983, + "tokens_seen": 313655296 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045708124373119356, + "loss": 1.5687, + "theoretical_loss": 4.1247992313461666, + "tokens_seen": 313720832 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004570712136409228, + "loss": 1.6464, + "theoretical_loss": 4.124698878635335, + "tokens_seen": 313786368 + }, + { + "epoch": 0.1, + "learning_rate": 0.000457061183550652, + "loss": 1.617, + "theoretical_loss": 4.12459855274871, + "tokens_seen": 313851904 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045705115346038116, + "loss": 1.5188, + "theoretical_loss": 4.124498253673523, + "tokens_seen": 313917440 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045704112337011034, + "loss": 1.7302, + "theoretical_loss": 4.124397981397014, + "tokens_seen": 313982976 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004570310932798395, + "loss": 1.6035, + "theoretical_loss": 4.124297735906431, + "tokens_seen": 314048512 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004570210631895687, + "loss": 1.5882, + "theoretical_loss": 4.124197517189032, + "tokens_seen": 314114048 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045701103309929794, + "loss": 1.6746, + "theoretical_loss": 4.124097325232081, + "tokens_seen": 314179584 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045700100300902707, + "loss": 1.6512, + "theoretical_loss": 4.123997160022854, + "tokens_seen": 314245120 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004569909729187563, + "loss": 1.647, + "theoretical_loss": 4.1238970215486335, + "tokens_seen": 314310656 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004569809428284855, + "loss": 1.5819, + "theoretical_loss": 4.123796909796713, + "tokens_seen": 314376192 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045697091273821466, + "loss": 1.4952, + "theoretical_loss": 4.123696824754391, + "tokens_seen": 314441728 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045696088264794384, + "loss": 1.5325, + "theoretical_loss": 4.123596766408978, + "tokens_seen": 314507264 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": 0.00014038987865205854, + "objective/train/docs_used": 189547, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.135436773300171, + "objective/train/original_loss": 3.135436534881592, + "objective/train/theoretical_loss": 4.123496734747793, + "objective/train/tokens_used": 335032800, + "objective/train/value_avg": -0.005126953125, + "objective/train/value_loss": 6.233378371689469e-05, + "objective/train/value_max": -0.0003669261932373047, + "objective/train/value_min": -0.0552978515625, + "objective/train/value_reward_corr": 0.2709507883195218, + "objective/train/value_std": 0.0037136077880859375, + "objective/train/weight_avg": 1.0042648315429688, + "objective/train/weighted_lm_loss": 3.1551175117492676, + "objective/train/weights_max": 1.716831922531128, + "objective/train/weights_min": 0.3770145773887634, + "theoretical_loss": 4.123496734747793, + "tokens_seen": 314572800 + }, + { + "epoch": 0.1, + "learning_rate": 0.000456950852557673, + "loss": 1.628, + "theoretical_loss": 4.123496734747793, + "tokens_seen": 314572800 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004569408224674022, + "loss": 1.6281, + "theoretical_loss": 4.12339672975816, + "tokens_seen": 314638336 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045693079237713144, + "loss": 1.5058, + "theoretical_loss": 4.123296751427417, + "tokens_seen": 314703872 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045692076228686057, + "loss": 1.5974, + "theoretical_loss": 4.123196799742907, + "tokens_seen": 314769408 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004569107321965898, + "loss": 1.6267, + "theoretical_loss": 4.123096874691981, + "tokens_seen": 314834944 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045690070210631893, + "loss": 1.4577, + "theoretical_loss": 4.122996976262003, + "tokens_seen": 314900480 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045689067201604817, + "loss": 1.6211, + "theoretical_loss": 4.122897104440343, + "tokens_seen": 314966016 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045688064192577735, + "loss": 1.4988, + "theoretical_loss": 4.1227972592143765, + "tokens_seen": 315031552 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045687061183550653, + "loss": 1.5984, + "theoretical_loss": 4.122697440571493, + "tokens_seen": 315097088 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004568605817452357, + "loss": 1.5914, + "theoretical_loss": 4.122597648499087, + "tokens_seen": 315162624 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045685055165496495, + "loss": 1.6466, + "theoretical_loss": 4.122497882984564, + "tokens_seen": 315228160 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045684052156469407, + "loss": 1.5211, + "theoretical_loss": 4.122398144015335, + "tokens_seen": 315293696 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004568304914744233, + "loss": 1.477, + "theoretical_loss": 4.122298431578824, + "tokens_seen": 315359232 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045682046138415243, + "loss": 1.557, + "theoretical_loss": 4.12219874566246, + "tokens_seen": 315424768 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045681043129388167, + "loss": 1.5218, + "theoretical_loss": 4.122099086253681, + "tokens_seen": 315490304 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045680040120361085, + "loss": 1.6124, + "theoretical_loss": 4.121999453339935, + "tokens_seen": 315555840 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045679037111334003, + "loss": 1.5825, + "theoretical_loss": 4.121899846908677, + "tokens_seen": 315621376 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004567803410230692, + "loss": 1.6647, + "theoretical_loss": 4.121800266947372, + "tokens_seen": 315686912 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004567703109327984, + "loss": 1.612, + "theoretical_loss": 4.121700713443492, + "tokens_seen": 315752448 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004567602808425276, + "loss": 1.5701, + "theoretical_loss": 4.121601186384519, + "tokens_seen": 315817984 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004567502507522568, + "loss": 1.7369, + "theoretical_loss": 4.121501685757943, + "tokens_seen": 315883520 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045674022066198594, + "loss": 1.4756, + "theoretical_loss": 4.121402211551262, + "tokens_seen": 315949056 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004567301905717152, + "loss": 1.5657, + "theoretical_loss": 4.121302763751983, + "tokens_seen": 316014592 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004567201604814443, + "loss": 1.6003, + "theoretical_loss": 4.12120334234762, + "tokens_seen": 316080128 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045671013039117354, + "loss": 1.5166, + "theoretical_loss": 4.121103947325699, + "tokens_seen": 316145664 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.0009667151025496423, + "objective/train/docs_used": 190733, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1063966751098633, + "objective/train/original_loss": 3.1063966751098633, + "objective/train/theoretical_loss": 4.121004578673752, + "objective/train/tokens_used": 336671200, + "objective/train/value_avg": -0.004482269287109375, + "objective/train/value_loss": 6.787029269617051e-05, + "objective/train/value_max": -0.0003829002380371094, + "objective/train/value_min": -0.0604248046875, + "objective/train/value_reward_corr": 0.22998116445215014, + "objective/train/value_std": 0.0039005279541015625, + "objective/train/weight_avg": 0.9935060739517212, + "objective/train/weighted_lm_loss": 3.1133065223693848, + "objective/train/weights_max": 1.8298757076263428, + "objective/train/weights_min": 0.19988392293453217, + "theoretical_loss": 4.121004578673752, + "tokens_seen": 316211200 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004567001003009027, + "loss": 1.5523, + "theoretical_loss": 4.121004578673752, + "tokens_seen": 316211200 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004566900702106319, + "loss": 1.506, + "theoretical_loss": 4.120905236379318, + "tokens_seen": 316276736 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004566800401203611, + "loss": 1.6049, + "theoretical_loss": 4.120805920429949, + "tokens_seen": 316342272 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004566700100300903, + "loss": 1.618, + "theoretical_loss": 4.120706630813201, + "tokens_seen": 316407808 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045665997993981944, + "loss": 1.6121, + "theoretical_loss": 4.12060736751664, + "tokens_seen": 316473344 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004566499498495487, + "loss": 1.5755, + "theoretical_loss": 4.120508130527841, + "tokens_seen": 316538880 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004566399197592778, + "loss": 1.6931, + "theoretical_loss": 4.120408919834388, + "tokens_seen": 316604416 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045662988966900704, + "loss": 1.6387, + "theoretical_loss": 4.120309735423871, + "tokens_seen": 316669952 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004566198595787362, + "loss": 1.6085, + "theoretical_loss": 4.120210577283891, + "tokens_seen": 316735488 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004566098294884654, + "loss": 1.5332, + "theoretical_loss": 4.120111445402056, + "tokens_seen": 316801024 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004565997993981946, + "loss": 1.5493, + "theoretical_loss": 4.1200123397659825, + "tokens_seen": 316866560 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045658976930792376, + "loss": 1.6081, + "theoretical_loss": 4.119913260363296, + "tokens_seen": 316932096 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045657973921765294, + "loss": 1.6284, + "theoretical_loss": 4.11981420718163, + "tokens_seen": 316997632 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004565697091273822, + "loss": 1.5646, + "theoretical_loss": 4.119715180208626, + "tokens_seen": 317063168 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004565596790371113, + "loss": 1.548, + "theoretical_loss": 4.119616179431935, + "tokens_seen": 317128704 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045654964894684054, + "loss": 1.628, + "theoretical_loss": 4.119517204839215, + "tokens_seen": 317194240 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045653961885656967, + "loss": 1.4617, + "theoretical_loss": 4.119418256418133, + "tokens_seen": 317259776 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004565295887662989, + "loss": 1.6586, + "theoretical_loss": 4.119319334156366, + "tokens_seen": 317325312 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004565195586760281, + "loss": 1.5281, + "theoretical_loss": 4.119220438041595, + "tokens_seen": 317390848 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045650952858575727, + "loss": 1.5991, + "theoretical_loss": 4.119121568061515, + "tokens_seen": 317456384 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045649949849548645, + "loss": 1.5048, + "theoretical_loss": 4.119022724203824, + "tokens_seen": 317521920 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004564894684052157, + "loss": 1.6376, + "theoretical_loss": 4.1189239064562315, + "tokens_seen": 317587456 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004564794383149448, + "loss": 1.7098, + "theoretical_loss": 4.118825114806455, + "tokens_seen": 317652992 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045646940822467405, + "loss": 1.6133, + "theoretical_loss": 4.118726349242221, + "tokens_seen": 317718528 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045645937813440317, + "loss": 1.5333, + "theoretical_loss": 4.118627609751259, + "tokens_seen": 317784064 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.0015202882932499051, + "objective/train/docs_used": 191330, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.282801866531372, + "objective/train/original_loss": 3.282802104949951, + "objective/train/theoretical_loss": 4.118528896321316, + "objective/train/tokens_used": 338309600, + "objective/train/value_avg": -0.0055999755859375, + "objective/train/value_loss": 8.663135668030009e-05, + "objective/train/value_max": -0.0004673004150390625, + "objective/train/value_min": -0.060760498046875, + "objective/train/value_reward_corr": 0.492179463056939, + "objective/train/value_std": 0.005382537841796875, + "objective/train/weight_avg": 0.9888144135475159, + "objective/train/weighted_lm_loss": 3.261019468307495, + "objective/train/weights_max": 1.6082628965377808, + "objective/train/weights_min": 0.38593459129333496, + "theoretical_loss": 4.118528896321316, + "tokens_seen": 317849600 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004564493480441324, + "loss": 1.5657, + "theoretical_loss": 4.118528896321316, + "tokens_seen": 317849600 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045643931795386164, + "loss": 1.4811, + "theoretical_loss": 4.118430208940139, + "tokens_seen": 317915136 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045642928786359077, + "loss": 1.6121, + "theoretical_loss": 4.1183315475954885, + "tokens_seen": 317980672 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045641925777332, + "loss": 1.5246, + "theoretical_loss": 4.118232912275129, + "tokens_seen": 318046208 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045640922768304913, + "loss": 1.5934, + "theoretical_loss": 4.118134302966838, + "tokens_seen": 318111744 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045639919759277837, + "loss": 1.5687, + "theoretical_loss": 4.118035719658397, + "tokens_seen": 318177280 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045638916750250755, + "loss": 1.5854, + "theoretical_loss": 4.117937162337598, + "tokens_seen": 318242816 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045637913741223673, + "loss": 1.6577, + "theoretical_loss": 4.117838630992242, + "tokens_seen": 318308352 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004563691073219659, + "loss": 1.6119, + "theoretical_loss": 4.117740125610136, + "tokens_seen": 318373888 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045635907723169515, + "loss": 1.6397, + "theoretical_loss": 4.117641646179096, + "tokens_seen": 318439424 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045634904714142427, + "loss": 1.5034, + "theoretical_loss": 4.117543192686947, + "tokens_seen": 318504960 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004563390170511535, + "loss": 1.6295, + "theoretical_loss": 4.1174447651215225, + "tokens_seen": 318570496 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045632898696088263, + "loss": 1.5993, + "theoretical_loss": 4.117346363470663, + "tokens_seen": 318636032 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045631895687061187, + "loss": 1.5149, + "theoretical_loss": 4.117247987722216, + "tokens_seen": 318701568 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045630892678034105, + "loss": 1.548, + "theoretical_loss": 4.117149637864041, + "tokens_seen": 318767104 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045629889669007023, + "loss": 1.6219, + "theoretical_loss": 4.117051313884003, + "tokens_seen": 318832640 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004562888665997994, + "loss": 1.7525, + "theoretical_loss": 4.116953015769975, + "tokens_seen": 318898176 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004562788365095286, + "loss": 1.4661, + "theoretical_loss": 4.11685474350984, + "tokens_seen": 318963712 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004562688064192578, + "loss": 1.6135, + "theoretical_loss": 4.116756497091487, + "tokens_seen": 319029248 + }, + { + "epoch": 0.1, + "learning_rate": 0.000456258776328987, + "loss": 1.591, + "theoretical_loss": 4.116658276502815, + "tokens_seen": 319094784 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045624874623871614, + "loss": 1.6442, + "theoretical_loss": 4.116560081731729, + "tokens_seen": 319160320 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004562387161484454, + "loss": 1.6125, + "theoretical_loss": 4.1164619127661455, + "tokens_seen": 319225856 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004562286860581745, + "loss": 1.5578, + "theoretical_loss": 4.116363769593986, + "tokens_seen": 319291392 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045621865596790374, + "loss": 1.6464, + "theoretical_loss": 4.116265652203181, + "tokens_seen": 319356928 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004562086258776329, + "loss": 1.5772, + "theoretical_loss": 4.116167560581669, + "tokens_seen": 319422464 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.0013914998853579164, + "objective/train/docs_used": 192764, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0408318042755127, + "objective/train/original_loss": 3.0408313274383545, + "objective/train/theoretical_loss": 4.116069494717399, + "objective/train/tokens_used": 339948000, + "objective/train/value_avg": -0.004535675048828125, + "objective/train/value_loss": 4.7242978325812146e-05, + "objective/train/value_max": -0.0003769397735595703, + "objective/train/value_min": -0.03704833984375, + "objective/train/value_reward_corr": 0.2571219658464712, + "objective/train/value_std": 0.003326416015625, + "objective/train/weight_avg": 0.9883155226707458, + "objective/train/weighted_lm_loss": 3.0222179889678955, + "objective/train/weights_max": 1.358499526977539, + "objective/train/weights_min": 0.34019118547439575, + "theoretical_loss": 4.116069494717399, + "tokens_seen": 319488000 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004561985957873621, + "loss": 1.5432, + "theoretical_loss": 4.116069494717399, + "tokens_seen": 319488000 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004561885656970913, + "loss": 1.655, + "theoretical_loss": 4.115971454598324, + "tokens_seen": 319553536 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004561785356068205, + "loss": 1.6994, + "theoretical_loss": 4.115873440212409, + "tokens_seen": 319619072 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045616850551654964, + "loss": 1.5639, + "theoretical_loss": 4.115775451547624, + "tokens_seen": 319684608 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004561584754262789, + "loss": 1.6262, + "theoretical_loss": 4.11567748859195, + "tokens_seen": 319750144 + }, + { + "epoch": 0.1, + "learning_rate": 0.000456148445336008, + "loss": 1.4918, + "theoretical_loss": 4.115579551333372, + "tokens_seen": 319815680 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045613841524573724, + "loss": 1.6026, + "theoretical_loss": 4.115481639759887, + "tokens_seen": 319881216 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004561283851554664, + "loss": 1.6679, + "theoretical_loss": 4.115383753859499, + "tokens_seen": 319946752 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004561183550651956, + "loss": 1.4781, + "theoretical_loss": 4.115285893620219, + "tokens_seen": 320012288 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004561083249749248, + "loss": 1.6218, + "theoretical_loss": 4.115188059030066, + "tokens_seen": 320077824 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045609829488465396, + "loss": 1.678, + "theoretical_loss": 4.11509025007707, + "tokens_seen": 320143360 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045608826479438314, + "loss": 1.4996, + "theoretical_loss": 4.114992466749264, + "tokens_seen": 320208896 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004560782347041124, + "loss": 1.5266, + "theoretical_loss": 4.114894709034695, + "tokens_seen": 320274432 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004560682046138415, + "loss": 1.6768, + "theoretical_loss": 4.114796976921412, + "tokens_seen": 320339968 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045605817452357074, + "loss": 1.5492, + "theoretical_loss": 4.114699270397477, + "tokens_seen": 320405504 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045604814443329987, + "loss": 1.6532, + "theoretical_loss": 4.114601589450956, + "tokens_seen": 320471040 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004560381143430291, + "loss": 1.6191, + "theoretical_loss": 4.114503934069926, + "tokens_seen": 320536576 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004560280842527583, + "loss": 1.4917, + "theoretical_loss": 4.114406304242472, + "tokens_seen": 320602112 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045601805416248747, + "loss": 1.5938, + "theoretical_loss": 4.114308699956684, + "tokens_seen": 320667648 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045600802407221665, + "loss": 1.5803, + "theoretical_loss": 4.114211121200662, + "tokens_seen": 320733184 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004559979939819459, + "loss": 1.6433, + "theoretical_loss": 4.114113567962514, + "tokens_seen": 320798720 + }, + { + "epoch": 0.1, + "learning_rate": 0.000455987963891675, + "loss": 1.4425, + "theoretical_loss": 4.114016040230357, + "tokens_seen": 320864256 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045597793380140425, + "loss": 1.4787, + "theoretical_loss": 4.113918537992314, + "tokens_seen": 320929792 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045596790371113337, + "loss": 1.6592, + "theoretical_loss": 4.113821061236516, + "tokens_seen": 320995328 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004559578736208626, + "loss": 1.5427, + "theoretical_loss": 4.113723609951103, + "tokens_seen": 321060864 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.000465029472252354, + "objective/train/docs_used": 193449, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5969622135162354, + "objective/train/original_loss": 2.5969622135162354, + "objective/train/theoretical_loss": 4.113626184124224, + "objective/train/tokens_used": 341586400, + "objective/train/value_avg": -0.0059051513671875, + "objective/train/value_loss": 7.666564488317817e-05, + "objective/train/value_max": -0.0004858970642089844, + "objective/train/value_min": -0.057708740234375, + "objective/train/value_reward_corr": 0.42802707593445644, + "objective/train/value_std": 0.005168914794921875, + "objective/train/weight_avg": 0.9989637136459351, + "objective/train/weighted_lm_loss": 2.5625948905944824, + "objective/train/weights_max": 1.575238585472107, + "objective/train/weights_min": 0.4560464024543762, + "theoretical_loss": 4.113626184124224, + "tokens_seen": 321126400 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004559478435305918, + "loss": 1.5258, + "theoretical_loss": 4.113626184124224, + "tokens_seen": 321126400 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045593781344032097, + "loss": 1.4936, + "theoretical_loss": 4.113528783744033, + "tokens_seen": 321191936 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045592778335005015, + "loss": 1.5458, + "theoretical_loss": 4.113431408798694, + "tokens_seen": 321257472 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045591775325977933, + "loss": 1.584, + "theoretical_loss": 4.113334059276378, + "tokens_seen": 321323008 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004559077231695085, + "loss": 1.5642, + "theoretical_loss": 4.113236735165264, + "tokens_seen": 321388544 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045589769307923775, + "loss": 1.5398, + "theoretical_loss": 4.11313943645354, + "tokens_seen": 321454080 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004558876629889669, + "loss": 1.6288, + "theoretical_loss": 4.113042163129402, + "tokens_seen": 321519616 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004558776328986961, + "loss": 1.4926, + "theoretical_loss": 4.112944915181051, + "tokens_seen": 321585152 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045586760280842524, + "loss": 1.4887, + "theoretical_loss": 4.112847692596698, + "tokens_seen": 321650688 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004558575727181545, + "loss": 1.6725, + "theoretical_loss": 4.112750495364563, + "tokens_seen": 321716224 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045584754262788365, + "loss": 1.5432, + "theoretical_loss": 4.112653323472872, + "tokens_seen": 321781760 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045583751253761284, + "loss": 1.5506, + "theoretical_loss": 4.11255617690986, + "tokens_seen": 321847296 + }, + { + "epoch": 0.1, + "learning_rate": 0.000455827482447342, + "loss": 1.5068, + "theoretical_loss": 4.112459055663768, + "tokens_seen": 321912832 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045581745235707125, + "loss": 1.5785, + "theoretical_loss": 4.112361959722849, + "tokens_seen": 321978368 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004558074222668004, + "loss": 1.5061, + "theoretical_loss": 4.112264889075357, + "tokens_seen": 322043904 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004557973921765296, + "loss": 1.549, + "theoretical_loss": 4.11216784370956, + "tokens_seen": 322109440 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045578736208625874, + "loss": 1.5567, + "theoretical_loss": 4.112070823613734, + "tokens_seen": 322174976 + }, + { + "epoch": 0.1, + "learning_rate": 0.000455777331995988, + "loss": 1.457, + "theoretical_loss": 4.111973828776156, + "tokens_seen": 322240512 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045576730190571716, + "loss": 1.5904, + "theoretical_loss": 4.111876859185118, + "tokens_seen": 322306048 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045575727181544634, + "loss": 1.6108, + "theoretical_loss": 4.111779914828917, + "tokens_seen": 322371584 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004557472417251755, + "loss": 1.6921, + "theoretical_loss": 4.111682995695858, + "tokens_seen": 322437120 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004557372116349047, + "loss": 1.5262, + "theoretical_loss": 4.111586101774253, + "tokens_seen": 322502656 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004557271815446339, + "loss": 1.5532, + "theoretical_loss": 4.111489233052423, + "tokens_seen": 322568192 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004557171514543631, + "loss": 1.611, + "theoretical_loss": 4.111392389518697, + "tokens_seen": 322633728 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045570712136409224, + "loss": 1.6038, + "theoretical_loss": 4.111295571161409, + "tokens_seen": 322699264 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": 0.0002044005668722093, + "objective/train/docs_used": 194727, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4373886585235596, + "objective/train/original_loss": 3.4373886585235596, + "objective/train/theoretical_loss": 4.111198777968905, + "objective/train/tokens_used": 343224800, + "objective/train/value_avg": -0.005947113037109375, + "objective/train/value_loss": 5.361830335459672e-05, + "objective/train/value_max": -0.0005130767822265625, + "objective/train/value_min": -0.055816650390625, + "objective/train/value_reward_corr": 0.23668425243897379, + "objective/train/value_std": 0.004398345947265625, + "objective/train/weight_avg": 1.0046494007110596, + "objective/train/weighted_lm_loss": 3.449214458465576, + "objective/train/weights_max": 1.7141895294189453, + "objective/train/weights_min": 0.4159776568412781, + "theoretical_loss": 4.111198777968905, + "tokens_seen": 322764800 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004556970912738215, + "loss": 1.4816, + "theoretical_loss": 4.111198777968905, + "tokens_seen": 322764800 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004556870611835507, + "loss": 1.5684, + "theoretical_loss": 4.111102009929535, + "tokens_seen": 322830336 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045567703109327984, + "loss": 1.4788, + "theoretical_loss": 4.11100526703166, + "tokens_seen": 322895872 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004556670010030091, + "loss": 1.531, + "theoretical_loss": 4.110908549263647, + "tokens_seen": 322961408 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004556569709127382, + "loss": 1.4626, + "theoretical_loss": 4.11081185661387, + "tokens_seen": 323026944 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045564694082246744, + "loss": 1.5333, + "theoretical_loss": 4.110715189070711, + "tokens_seen": 323092480 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004556369107321966, + "loss": 1.6159, + "theoretical_loss": 4.110618546622561, + "tokens_seen": 323158016 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004556268806419258, + "loss": 1.4761, + "theoretical_loss": 4.110521929257819, + "tokens_seen": 323223552 + }, + { + "epoch": 0.1, + "learning_rate": 0.000455616850551655, + "loss": 1.511, + "theoretical_loss": 4.110425336964889, + "tokens_seen": 323289088 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045560682046138416, + "loss": 1.5941, + "theoretical_loss": 4.110328769732186, + "tokens_seen": 323354624 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045559679037111334, + "loss": 1.4799, + "theoretical_loss": 4.110232227548131, + "tokens_seen": 323420160 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004555867602808426, + "loss": 1.5525, + "theoretical_loss": 4.110135710401153, + "tokens_seen": 323485696 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004555767301905717, + "loss": 1.6316, + "theoretical_loss": 4.110039218279687, + "tokens_seen": 323551232 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045556670010030094, + "loss": 1.6521, + "theoretical_loss": 4.109942751172179, + "tokens_seen": 323616768 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045555667001003007, + "loss": 1.6205, + "theoretical_loss": 4.10984630906708, + "tokens_seen": 323682304 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004555466399197593, + "loss": 1.5539, + "theoretical_loss": 4.109749891952849, + "tokens_seen": 323747840 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004555366098294885, + "loss": 1.6163, + "theoretical_loss": 4.109653499817955, + "tokens_seen": 323813376 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045552657973921767, + "loss": 1.6792, + "theoretical_loss": 4.109557132650871, + "tokens_seen": 323878912 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045551654964894685, + "loss": 1.3861, + "theoretical_loss": 4.109460790440082, + "tokens_seen": 323944448 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004555065195586761, + "loss": 1.5928, + "theoretical_loss": 4.109364473174075, + "tokens_seen": 324009984 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004554964894684052, + "loss": 1.5418, + "theoretical_loss": 4.109268180841351, + "tokens_seen": 324075520 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045548645937813445, + "loss": 1.3405, + "theoretical_loss": 4.109171913430414, + "tokens_seen": 324141056 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045547642928786357, + "loss": 1.5085, + "theoretical_loss": 4.109075670929776, + "tokens_seen": 324206592 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004554663991975928, + "loss": 1.5663, + "theoretical_loss": 4.108979453327959, + "tokens_seen": 324272128 + }, + { + "epoch": 0.1, + "learning_rate": 0.000455456369107322, + "loss": 1.5731, + "theoretical_loss": 4.108883260613491, + "tokens_seen": 324337664 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.0003778735117521137, + "objective/train/docs_used": 195463, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.350701093673706, + "objective/train/original_loss": 3.350701332092285, + "objective/train/theoretical_loss": 4.108787092774909, + "objective/train/tokens_used": 344863200, + "objective/train/value_avg": -0.005374908447265625, + "objective/train/value_loss": 5.827749555464834e-05, + "objective/train/value_max": -0.0004546642303466797, + "objective/train/value_min": -0.06878662109375, + "objective/train/value_reward_corr": 0.22570675087289874, + "objective/train/value_std": 0.0038967132568359375, + "objective/train/weight_avg": 0.9990272521972656, + "objective/train/weighted_lm_loss": 3.3494606018066406, + "objective/train/weights_max": 1.8703728914260864, + "objective/train/weights_min": 0.6420565843582153, + "theoretical_loss": 4.108787092774909, + "tokens_seen": 324403200 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045544633901705117, + "loss": 1.5465, + "theoretical_loss": 4.108787092774909, + "tokens_seen": 324403200 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045543630892678035, + "loss": 1.5516, + "theoretical_loss": 4.108690949800755, + "tokens_seen": 324468736 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045542627883650953, + "loss": 1.4882, + "theoretical_loss": 4.1085948316795795, + "tokens_seen": 324534272 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004554162487462387, + "loss": 1.5273, + "theoretical_loss": 4.108498738399943, + "tokens_seen": 324599808 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045540621865596795, + "loss": 1.5131, + "theoretical_loss": 4.108402669950411, + "tokens_seen": 324665344 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004553961885656971, + "loss": 1.5458, + "theoretical_loss": 4.108306626319557, + "tokens_seen": 324730880 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004553861584754263, + "loss": 1.4374, + "theoretical_loss": 4.108210607495963, + "tokens_seen": 324796416 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045537612838515544, + "loss": 1.543, + "theoretical_loss": 4.108114613468217, + "tokens_seen": 324861952 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004553660982948847, + "loss": 1.5538, + "theoretical_loss": 4.1080186442249165, + "tokens_seen": 324927488 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045535606820461385, + "loss": 1.5599, + "theoretical_loss": 4.107922699754665, + "tokens_seen": 324993024 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045534603811434304, + "loss": 1.49, + "theoretical_loss": 4.107826780046074, + "tokens_seen": 325058560 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004553360080240722, + "loss": 1.5503, + "theoretical_loss": 4.107730885087763, + "tokens_seen": 325124096 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045532597793380145, + "loss": 1.5883, + "theoretical_loss": 4.107635014868358, + "tokens_seen": 325189632 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004553159478435306, + "loss": 1.4511, + "theoretical_loss": 4.107539169376494, + "tokens_seen": 325255168 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004553059177532598, + "loss": 1.5601, + "theoretical_loss": 4.107443348600812, + "tokens_seen": 325320704 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045529588766298894, + "loss": 1.4834, + "theoretical_loss": 4.107347552529961, + "tokens_seen": 325386240 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004552858575727182, + "loss": 1.4438, + "theoretical_loss": 4.107251781152598, + "tokens_seen": 325451776 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045527582748244736, + "loss": 1.5638, + "theoretical_loss": 4.107156034457387, + "tokens_seen": 325517312 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045526579739217654, + "loss": 1.4904, + "theoretical_loss": 4.107060312433, + "tokens_seen": 325582848 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004552557673019057, + "loss": 1.5362, + "theoretical_loss": 4.106964615068115, + "tokens_seen": 325648384 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004552457372116349, + "loss": 1.4713, + "theoretical_loss": 4.10686894235142, + "tokens_seen": 325713920 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004552357071213641, + "loss": 1.5191, + "theoretical_loss": 4.106773294271608, + "tokens_seen": 325779456 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004552256770310933, + "loss": 1.3925, + "theoretical_loss": 4.1066776708173816, + "tokens_seen": 325844992 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045521564694082244, + "loss": 1.4588, + "theoretical_loss": 4.106582071977449, + "tokens_seen": 325910528 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004552056168505517, + "loss": 1.5632, + "theoretical_loss": 4.106486497740526, + "tokens_seen": 325976064 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.0003522943297866732, + "objective/train/docs_used": 196731, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.344179630279541, + "objective/train/original_loss": 3.344179391860962, + "objective/train/theoretical_loss": 4.106390948095338, + "objective/train/tokens_used": 346501600, + "objective/train/value_avg": -0.004909515380859375, + "objective/train/value_loss": 7.431336416630074e-05, + "objective/train/value_max": -0.0004992485046386719, + "objective/train/value_min": -0.06549072265625, + "objective/train/value_reward_corr": 0.24712982696721128, + "objective/train/value_std": 0.0035457611083984375, + "objective/train/weight_avg": 0.9997643828392029, + "objective/train/weighted_lm_loss": 3.3503143787384033, + "objective/train/weights_max": 1.6255396604537964, + "objective/train/weights_min": 0.1413990557193756, + "theoretical_loss": 4.106390948095338, + "tokens_seen": 326041600 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004551955867602808, + "loss": 1.5621, + "theoretical_loss": 4.106390948095338, + "tokens_seen": 326041600 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045518555667001004, + "loss": 1.5909, + "theoretical_loss": 4.106295423030614, + "tokens_seen": 326107136 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004551755265797392, + "loss": 1.4934, + "theoretical_loss": 4.106199922535095, + "tokens_seen": 326172672 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004551654964894684, + "loss": 1.5187, + "theoretical_loss": 4.106104446597525, + "tokens_seen": 326238208 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004551554663991976, + "loss": 1.5606, + "theoretical_loss": 4.10600899520666, + "tokens_seen": 326303744 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004551454363089268, + "loss": 1.5793, + "theoretical_loss": 4.1059135683512595, + "tokens_seen": 326369280 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045513540621865595, + "loss": 1.5755, + "theoretical_loss": 4.105818166020091, + "tokens_seen": 326434816 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004551253761283852, + "loss": 1.6065, + "theoretical_loss": 4.105722788201932, + "tokens_seen": 326500352 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004551153460381143, + "loss": 1.568, + "theoretical_loss": 4.105627434885564, + "tokens_seen": 326565888 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045510531594784354, + "loss": 1.5214, + "theoretical_loss": 4.10553210605978, + "tokens_seen": 326631424 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004550952858575727, + "loss": 1.5585, + "theoretical_loss": 4.105436801713375, + "tokens_seen": 326696960 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004550852557673019, + "loss": 1.5256, + "theoretical_loss": 4.105341521835155, + "tokens_seen": 326762496 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004550752256770311, + "loss": 1.5446, + "theoretical_loss": 4.105246266413934, + "tokens_seen": 326828032 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045506519558676027, + "loss": 1.4807, + "theoretical_loss": 4.10515103543853, + "tokens_seen": 326893568 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045505516549648945, + "loss": 1.4412, + "theoretical_loss": 4.105055828897772, + "tokens_seen": 326959104 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004550451354062187, + "loss": 1.5187, + "theoretical_loss": 4.104960646780494, + "tokens_seen": 327024640 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004550351053159478, + "loss": 1.4887, + "theoretical_loss": 4.104865489075538, + "tokens_seen": 327090176 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045502507522567705, + "loss": 1.5749, + "theoretical_loss": 4.104770355771754, + "tokens_seen": 327155712 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045501504513540623, + "loss": 1.4906, + "theoretical_loss": 4.104675246857997, + "tokens_seen": 327221248 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004550050150451354, + "loss": 1.5294, + "theoretical_loss": 4.104580162323132, + "tokens_seen": 327286784 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004549949849548646, + "loss": 1.5722, + "theoretical_loss": 4.104485102156031, + "tokens_seen": 327352320 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045498495486459377, + "loss": 1.6003, + "theoretical_loss": 4.104390066345571, + "tokens_seen": 327417856 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045497492477432295, + "loss": 1.5286, + "theoretical_loss": 4.104295054880639, + "tokens_seen": 327483392 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004549648946840522, + "loss": 1.4427, + "theoretical_loss": 4.10420006775013, + "tokens_seen": 327548928 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045495486459378137, + "loss": 1.5833, + "theoretical_loss": 4.10410510494294, + "tokens_seen": 327614464 + }, + { + "debugging/Self-BLEU-5": 0.48445945981218463, + "debugging/distinct-1-grams": 0.7948268571619983, + "debugging/distinct-2-grams": 0.9739697199580383, + "debugging/entropy-1-grams": 6.01135557249139, + "debugging/entropy-2-grams": 6.827639403579008, + "debugging/length": 512.9230769230769, + "debugging/num_segments": 13, + "debugging/raw_token_scores_avg": 0.00634933914989233, + "debugging/raw_token_scores_std": 0.008881147019565105, + "debugging/score": 0.008299263930001547, + "debugging/score_std": 0.00590965264802656, + "epoch": 0.1, + "objective/train/advantage_avg": -0.0005666748038493097, + "objective/train/docs_used": 197513, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.163179397583008, + "objective/train/original_loss": 3.1631791591644287, + "objective/train/theoretical_loss": 4.10401016644798, + "objective/train/tokens_used": 348140000, + "objective/train/value_avg": -0.0057830810546875, + "objective/train/value_loss": 7.582655234728009e-05, + "objective/train/value_max": -0.0006589889526367188, + "objective/train/value_min": -0.0762939453125, + "objective/train/value_reward_corr": 0.2896677230658109, + "objective/train/value_std": 0.00438690185546875, + "objective/train/weight_avg": 0.9978270530700684, + "objective/train/weighted_lm_loss": 3.1571900844573975, + "objective/train/weights_max": 1.6560677289962769, + "objective/train/weights_min": 0.2966194450855255, + "theoretical_loss": 4.10401016644798, + "tokens_seen": 327680000 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045494483450351055, + "loss": 1.582, + "theoretical_loss": 4.10401016644798, + "tokens_seen": 327680000 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045493480441323973, + "loss": 1.4914, + "theoretical_loss": 4.103915252254165, + "tokens_seen": 327745536 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004549247743229689, + "loss": 1.4984, + "theoretical_loss": 4.103820362350415, + "tokens_seen": 327811072 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045491474423269815, + "loss": 1.5331, + "theoretical_loss": 4.103725496725662, + "tokens_seen": 327876608 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004549047141424273, + "loss": 1.516, + "theoretical_loss": 4.103630655368843, + "tokens_seen": 327942144 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004548946840521565, + "loss": 1.5391, + "theoretical_loss": 4.1035358382689004, + "tokens_seen": 328007680 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045488465396188564, + "loss": 1.5395, + "theoretical_loss": 4.103441045414786, + "tokens_seen": 328073216 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004548746238716149, + "loss": 1.5079, + "theoretical_loss": 4.103346276795459, + "tokens_seen": 328138752 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045486459378134405, + "loss": 1.3312, + "theoretical_loss": 4.103251532399884, + "tokens_seen": 328204288 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045485456369107324, + "loss": 1.4132, + "theoretical_loss": 4.103156812217035, + "tokens_seen": 328269824 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004548445336008024, + "loss": 1.6346, + "theoretical_loss": 4.1030621162358925, + "tokens_seen": 328335360 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045483450351053165, + "loss": 1.5008, + "theoretical_loss": 4.102967444445444, + "tokens_seen": 328400896 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004548244734202608, + "loss": 1.5415, + "theoretical_loss": 4.102872796834683, + "tokens_seen": 328466432 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045481444332999, + "loss": 1.6515, + "theoretical_loss": 4.102778173392611, + "tokens_seen": 328531968 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045480441323971914, + "loss": 1.6041, + "theoretical_loss": 4.10268357410824, + "tokens_seen": 328597504 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004547943831494484, + "loss": 1.5102, + "theoretical_loss": 4.1025889989705835, + "tokens_seen": 328663040 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045478435305917756, + "loss": 1.4956, + "theoretical_loss": 4.102494447968666, + "tokens_seen": 328728576 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045477432296890674, + "loss": 1.5943, + "theoretical_loss": 4.102399921091518, + "tokens_seen": 328794112 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004547642928786359, + "loss": 1.5436, + "theoretical_loss": 4.102305418328176, + "tokens_seen": 328859648 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004547542627883651, + "loss": 1.5682, + "theoretical_loss": 4.102210939667687, + "tokens_seen": 328925184 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004547442326980943, + "loss": 1.5383, + "theoretical_loss": 4.102116485099103, + "tokens_seen": 328990720 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004547342026078235, + "loss": 1.5537, + "theoretical_loss": 4.102022054611481, + "tokens_seen": 329056256 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045472417251755264, + "loss": 1.5791, + "theoretical_loss": 4.1019276481938896, + "tokens_seen": 329121792 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004547141424272819, + "loss": 1.5373, + "theoretical_loss": 4.101833265835402, + "tokens_seen": 329187328 + }, + { + "epoch": 0.1, + "learning_rate": 0.000454704112337011, + "loss": 1.5272, + "theoretical_loss": 4.101738907525098, + "tokens_seen": 329252864 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": 0.00011983455624431372, + "objective/train/docs_used": 198138, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4587395191192627, + "objective/train/original_loss": 3.458739757537842, + "objective/train/theoretical_loss": 4.101644573252066, + "objective/train/tokens_used": 349778400, + "objective/train/value_avg": -0.00559234619140625, + "objective/train/value_loss": 7.924599049147218e-05, + "objective/train/value_max": -0.0005483627319335938, + "objective/train/value_min": -0.0855712890625, + "objective/train/value_reward_corr": 0.3667247696709051, + "objective/train/value_std": 0.004413604736328125, + "objective/train/weight_avg": 1.0045686960220337, + "objective/train/weighted_lm_loss": 3.474865198135376, + "objective/train/weights_max": 2.1512022018432617, + "objective/train/weights_min": 0.08952353149652481, + "theoretical_loss": 4.101644573252066, + "tokens_seen": 329318400 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045469408224674024, + "loss": 1.4402, + "theoretical_loss": 4.101644573252066, + "tokens_seen": 329318400 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004546840521564694, + "loss": 1.5249, + "theoretical_loss": 4.1015502630054, + "tokens_seen": 329383936 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004546740220661986, + "loss": 1.5176, + "theoretical_loss": 4.101455976774204, + "tokens_seen": 329449472 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004546639919759278, + "loss": 1.518, + "theoretical_loss": 4.101361714547585, + "tokens_seen": 329515008 + }, + { + "epoch": 0.1, + "learning_rate": 0.000454653961885657, + "loss": 1.6259, + "theoretical_loss": 4.1012674763146615, + "tokens_seen": 329580544 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045464393179538615, + "loss": 1.5121, + "theoretical_loss": 4.101173262064555, + "tokens_seen": 329646080 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004546339017051154, + "loss": 1.6768, + "theoretical_loss": 4.101079071786397, + "tokens_seen": 329711616 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004546238716148445, + "loss": 1.6338, + "theoretical_loss": 4.100984905469324, + "tokens_seen": 329777152 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045461384152457374, + "loss": 1.4858, + "theoretical_loss": 4.100890763102481, + "tokens_seen": 329842688 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004546038114343029, + "loss": 1.6891, + "theoretical_loss": 4.10079664467502, + "tokens_seen": 329908224 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004545937813440321, + "loss": 1.425, + "theoretical_loss": 4.1007025501761, + "tokens_seen": 329973760 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004545837512537613, + "loss": 1.623, + "theoretical_loss": 4.100608479594887, + "tokens_seen": 330039296 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045457372116349047, + "loss": 1.6224, + "theoretical_loss": 4.100514432920552, + "tokens_seen": 330104832 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045456369107321965, + "loss": 1.5414, + "theoretical_loss": 4.100420410142277, + "tokens_seen": 330170368 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004545536609829489, + "loss": 1.5302, + "theoretical_loss": 4.100326411249248, + "tokens_seen": 330235904 + }, + { + "epoch": 0.1, + "learning_rate": 0.000454543630892678, + "loss": 1.5106, + "theoretical_loss": 4.100232436230659, + "tokens_seen": 330301440 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045453360080240725, + "loss": 1.5135, + "theoretical_loss": 4.100138485075711, + "tokens_seen": 330366976 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045452357071213643, + "loss": 1.3825, + "theoretical_loss": 4.1000445577736135, + "tokens_seen": 330432512 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004545135406218656, + "loss": 1.3529, + "theoretical_loss": 4.099950654313579, + "tokens_seen": 330498048 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004545035105315948, + "loss": 1.5292, + "theoretical_loss": 4.099856774684832, + "tokens_seen": 330563584 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045449348044132397, + "loss": 1.4188, + "theoretical_loss": 4.0997629188766, + "tokens_seen": 330629120 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045448345035105315, + "loss": 1.5381, + "theoretical_loss": 4.099669086878119, + "tokens_seen": 330694656 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004544734202607824, + "loss": 1.4658, + "theoretical_loss": 4.099575278678634, + "tokens_seen": 330760192 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004544633901705115, + "loss": 1.6444, + "theoretical_loss": 4.099481494267393, + "tokens_seen": 330825728 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045445336008024075, + "loss": 1.4747, + "theoretical_loss": 4.0993877336336535, + "tokens_seen": 330891264 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.0020747294183820486, + "objective/train/docs_used": 199416, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.029064416885376, + "objective/train/original_loss": 3.029064416885376, + "objective/train/theoretical_loss": 4.099293996766681, + "objective/train/tokens_used": 351416800, + "objective/train/value_avg": -0.004993438720703125, + "objective/train/value_loss": 8.146175241563469e-05, + "objective/train/value_max": -0.0003418922424316406, + "objective/train/value_min": -0.050994873046875, + "objective/train/value_reward_corr": 0.40130884226136915, + "objective/train/value_std": 0.00421905517578125, + "objective/train/weight_avg": 0.9829933047294617, + "objective/train/weighted_lm_loss": 2.994530200958252, + "objective/train/weights_max": 1.5916725397109985, + "objective/train/weights_min": 0.3014388084411621, + "theoretical_loss": 4.099293996766681, + "tokens_seen": 330956800 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004544433299899699, + "loss": 1.4304, + "theoretical_loss": 4.099293996766681, + "tokens_seen": 330956800 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004544332998996991, + "loss": 1.4288, + "theoretical_loss": 4.099200283655745, + "tokens_seen": 331022336 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004544232698094283, + "loss": 1.4585, + "theoretical_loss": 4.099106594290124, + "tokens_seen": 331087872 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004544132397191575, + "loss": 1.6893, + "theoretical_loss": 4.099012928659102, + "tokens_seen": 331153408 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045440320962888666, + "loss": 1.6354, + "theoretical_loss": 4.098919286751972, + "tokens_seen": 331218944 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045439317953861584, + "loss": 1.4216, + "theoretical_loss": 4.0988256685580335, + "tokens_seen": 331284480 + }, + { + "epoch": 0.1, + "learning_rate": 0.000454383149448345, + "loss": 1.616, + "theoretical_loss": 4.098732074066591, + "tokens_seen": 331350016 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045437311935807425, + "loss": 1.4911, + "theoretical_loss": 4.098638503266957, + "tokens_seen": 331415552 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004543630892678034, + "loss": 1.4474, + "theoretical_loss": 4.098544956148452, + "tokens_seen": 331481088 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004543530591775326, + "loss": 1.5996, + "theoretical_loss": 4.098451432700403, + "tokens_seen": 331546624 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004543430290872618, + "loss": 1.703, + "theoretical_loss": 4.098357932912142, + "tokens_seen": 331612160 + }, + { + "epoch": 0.1, + "learning_rate": 0.000454332998996991, + "loss": 1.6196, + "theoretical_loss": 4.098264456773012, + "tokens_seen": 331677696 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045432296890672016, + "loss": 1.529, + "theoretical_loss": 4.098171004272357, + "tokens_seen": 331743232 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045431293881644934, + "loss": 1.6048, + "theoretical_loss": 4.098077575399533, + "tokens_seen": 331808768 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004543029087261785, + "loss": 1.575, + "theoretical_loss": 4.097984170143901, + "tokens_seen": 331874304 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045429287863590776, + "loss": 1.5914, + "theoretical_loss": 4.09789078849483, + "tokens_seen": 331939840 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004542828485456369, + "loss": 1.4687, + "theoretical_loss": 4.097797430441693, + "tokens_seen": 332005376 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004542728184553661, + "loss": 1.4862, + "theoretical_loss": 4.097704095973873, + "tokens_seen": 332070912 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045426278836509525, + "loss": 1.421, + "theoretical_loss": 4.097610785080757, + "tokens_seen": 332136448 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004542527582748245, + "loss": 1.4749, + "theoretical_loss": 4.097517497751743, + "tokens_seen": 332201984 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045424272818455366, + "loss": 1.4852, + "theoretical_loss": 4.0974242339762315, + "tokens_seen": 332267520 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045423269809428284, + "loss": 1.6235, + "theoretical_loss": 4.097330993743633, + "tokens_seen": 332333056 + }, + { + "epoch": 0.1, + "learning_rate": 0.000454222668004012, + "loss": 1.4494, + "theoretical_loss": 4.097237777043363, + "tokens_seen": 332398592 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004542126379137412, + "loss": 1.494, + "theoretical_loss": 4.097144583864844, + "tokens_seen": 332464128 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045420260782347044, + "loss": 1.5318, + "theoretical_loss": 4.097051414197506, + "tokens_seen": 332529664 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.00031865545315667987, + "objective/train/docs_used": 200524, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.151939868927002, + "objective/train/original_loss": 3.151939868927002, + "objective/train/theoretical_loss": 4.096958268030786, + "objective/train/tokens_used": 353055200, + "objective/train/value_avg": -0.005451202392578125, + "objective/train/value_loss": 6.0989317717030644e-05, + "objective/train/value_max": -0.0005254745483398438, + "objective/train/value_min": -0.0640869140625, + "objective/train/value_reward_corr": 0.46747108218663186, + "objective/train/value_std": 0.0049591064453125, + "objective/train/weight_avg": 0.9997064471244812, + "objective/train/weighted_lm_loss": 3.1539158821105957, + "objective/train/weights_max": 1.8981298208236694, + "objective/train/weights_min": 0.4375602602958679, + "theoretical_loss": 4.096958268030786, + "tokens_seen": 332595200 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004541925777331996, + "loss": 1.6146, + "theoretical_loss": 4.096958268030786, + "tokens_seen": 332595200 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004541825476429288, + "loss": 1.4312, + "theoretical_loss": 4.096865145354127, + "tokens_seen": 332660736 + }, + { + "epoch": 0.1, + "learning_rate": 0.000454172517552658, + "loss": 1.5713, + "theoretical_loss": 4.09677204615698, + "tokens_seen": 332726272 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004541624874623872, + "loss": 1.4827, + "theoretical_loss": 4.096678970428801, + "tokens_seen": 332791808 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045415245737211635, + "loss": 1.4167, + "theoretical_loss": 4.096585918159054, + "tokens_seen": 332857344 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004541424272818456, + "loss": 1.5903, + "theoretical_loss": 4.096492889337211, + "tokens_seen": 332922880 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004541323971915747, + "loss": 1.6729, + "theoretical_loss": 4.096399883952747, + "tokens_seen": 332988416 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045412236710130395, + "loss": 1.4582, + "theoretical_loss": 4.096306901995149, + "tokens_seen": 333053952 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004541123370110331, + "loss": 1.5849, + "theoretical_loss": 4.096213943453906, + "tokens_seen": 333119488 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004541023069207623, + "loss": 1.4558, + "theoretical_loss": 4.096121008318517, + "tokens_seen": 333185024 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004540922768304915, + "loss": 1.5714, + "theoretical_loss": 4.096028096578486, + "tokens_seen": 333250560 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045408224674022067, + "loss": 1.4333, + "theoretical_loss": 4.095935208223324, + "tokens_seen": 333316096 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045407221664994985, + "loss": 1.4651, + "theoretical_loss": 4.09584234324255, + "tokens_seen": 333381632 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004540621865596791, + "loss": 1.4555, + "theoretical_loss": 4.095749501625689, + "tokens_seen": 333447168 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004540521564694082, + "loss": 1.611, + "theoretical_loss": 4.095656683362272, + "tokens_seen": 333512704 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045404212637913745, + "loss": 1.4778, + "theoretical_loss": 4.095563888441838, + "tokens_seen": 333578240 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045403209628886663, + "loss": 1.5889, + "theoretical_loss": 4.095471116853931, + "tokens_seen": 333643776 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004540220661985958, + "loss": 1.4717, + "theoretical_loss": 4.095378368588103, + "tokens_seen": 333709312 + }, + { + "epoch": 0.1, + "learning_rate": 0.000454012036108325, + "loss": 1.5794, + "theoretical_loss": 4.095285643633915, + "tokens_seen": 333774848 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045400200601805417, + "loss": 1.4909, + "theoretical_loss": 4.095192941980929, + "tokens_seen": 333840384 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045399197592778335, + "loss": 1.5927, + "theoretical_loss": 4.095100263618718, + "tokens_seen": 333905920 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004539819458375126, + "loss": 1.6063, + "theoretical_loss": 4.0950076085368625, + "tokens_seen": 333971456 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004539719157472417, + "loss": 1.4685, + "theoretical_loss": 4.0949149767249455, + "tokens_seen": 334036992 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045396188565697095, + "loss": 1.5455, + "theoretical_loss": 4.0948223681725615, + "tokens_seen": 334102528 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004539518555667001, + "loss": 1.6556, + "theoretical_loss": 4.094729782869308, + "tokens_seen": 334168064 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.000764830328989774, + "objective/train/docs_used": 201196, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3282346725463867, + "objective/train/original_loss": 3.3282346725463867, + "objective/train/theoretical_loss": 4.09463722080479, + "objective/train/tokens_used": 354693600, + "objective/train/value_avg": -0.00530242919921875, + "objective/train/value_loss": 0.00012256567424628884, + "objective/train/value_max": -0.00039505958557128906, + "objective/train/value_min": -0.091552734375, + "objective/train/value_reward_corr": 0.20812589487317357, + "objective/train/value_std": 0.00402069091796875, + "objective/train/weight_avg": 0.9972164630889893, + "objective/train/weighted_lm_loss": 3.3286244869232178, + "objective/train/weights_max": 2.4980924129486084, + "objective/train/weights_min": 0.1374148577451706, + "theoretical_loss": 4.09463722080479, + "tokens_seen": 334233600 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004539418254764293, + "loss": 1.6629, + "theoretical_loss": 4.09463722080479, + "tokens_seen": 334233600 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004539317953861585, + "loss": 1.5646, + "theoretical_loss": 4.094544681968621, + "tokens_seen": 334299136 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004539217652958877, + "loss": 1.5027, + "theoretical_loss": 4.094452166350419, + "tokens_seen": 334364672 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045391173520561686, + "loss": 1.5113, + "theoretical_loss": 4.0943596739398105, + "tokens_seen": 334430208 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045390170511534604, + "loss": 1.4677, + "theoretical_loss": 4.094267204726426, + "tokens_seen": 334495744 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004538916750250752, + "loss": 1.5607, + "theoretical_loss": 4.094174758699907, + "tokens_seen": 334561280 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045388164493480445, + "loss": 1.4885, + "theoretical_loss": 4.0940823358498974, + "tokens_seen": 334626816 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004538716148445336, + "loss": 1.7283, + "theoretical_loss": 4.093989936166049, + "tokens_seen": 334692352 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004538615847542628, + "loss": 1.5661, + "theoretical_loss": 4.093897559638021, + "tokens_seen": 334757888 + }, + { + "epoch": 0.1, + "learning_rate": 0.000453851554663992, + "loss": 1.5323, + "theoretical_loss": 4.09380520625548, + "tokens_seen": 334823424 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004538415245737212, + "loss": 1.5038, + "theoretical_loss": 4.093712876008097, + "tokens_seen": 334888960 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045383149448345036, + "loss": 1.4965, + "theoretical_loss": 4.093620568885552, + "tokens_seen": 334954496 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045382146439317954, + "loss": 1.6149, + "theoretical_loss": 4.093528284877528, + "tokens_seen": 335020032 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004538114343029087, + "loss": 1.4608, + "theoretical_loss": 4.093436023973719, + "tokens_seen": 335085568 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045380140421263796, + "loss": 1.6454, + "theoretical_loss": 4.093343786163823, + "tokens_seen": 335151104 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004537913741223671, + "loss": 1.4857, + "theoretical_loss": 4.093251571437545, + "tokens_seen": 335216640 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004537813440320963, + "loss": 1.5427, + "theoretical_loss": 4.093159379784597, + "tokens_seen": 335282176 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045377131394182545, + "loss": 1.6047, + "theoretical_loss": 4.093067211194698, + "tokens_seen": 335347712 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004537612838515547, + "loss": 1.5874, + "theoretical_loss": 4.092975065657572, + "tokens_seen": 335413248 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045375125376128386, + "loss": 1.6034, + "theoretical_loss": 4.092882943162952, + "tokens_seen": 335478784 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045374122367101304, + "loss": 1.5737, + "theoretical_loss": 4.092790843700574, + "tokens_seen": 335544320 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004537311935807422, + "loss": 1.6363, + "theoretical_loss": 4.092698767260185, + "tokens_seen": 335609856 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004537211634904714, + "loss": 1.6389, + "theoretical_loss": 4.092606713831534, + "tokens_seen": 335675392 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004537111334002006, + "loss": 1.5515, + "theoretical_loss": 4.0925146834043815, + "tokens_seen": 335740928 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004537011033099298, + "loss": 1.6168, + "theoretical_loss": 4.092422675968489, + "tokens_seen": 335806464 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.00042241389746777713, + "objective/train/docs_used": 202460, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4075212478637695, + "objective/train/original_loss": 3.4075212478637695, + "objective/train/theoretical_loss": 4.092330691513631, + "objective/train/tokens_used": 356332000, + "objective/train/value_avg": -0.00479888916015625, + "objective/train/value_loss": 5.585145845543593e-05, + "objective/train/value_max": -0.0005526542663574219, + "objective/train/value_min": -0.07586669921875, + "objective/train/value_reward_corr": 0.37536424580239125, + "objective/train/value_std": 0.003780364990234375, + "objective/train/weight_avg": 0.9984095096588135, + "objective/train/weighted_lm_loss": 3.4001150131225586, + "objective/train/weights_max": 1.9402567148208618, + "objective/train/weights_min": 0.37612393498420715, + "theoretical_loss": 4.092330691513631, + "tokens_seen": 335872000 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045369107321965895, + "loss": 1.5309, + "theoretical_loss": 4.092330691513631, + "tokens_seen": 335872000 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004536810431293882, + "loss": 1.5904, + "theoretical_loss": 4.092238730029582, + "tokens_seen": 335937536 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045367101303911737, + "loss": 1.5323, + "theoretical_loss": 4.092146791506127, + "tokens_seen": 336003072 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045366098294884655, + "loss": 1.5869, + "theoretical_loss": 4.092054875933057, + "tokens_seen": 336068608 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045365095285857573, + "loss": 1.6112, + "theoretical_loss": 4.091962983300168, + "tokens_seen": 336134144 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004536409227683049, + "loss": 1.5023, + "theoretical_loss": 4.091871113597266, + "tokens_seen": 336199680 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004536308926780341, + "loss": 1.5249, + "theoretical_loss": 4.091779266814158, + "tokens_seen": 336265216 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004536208625877633, + "loss": 1.6354, + "theoretical_loss": 4.091687442940664, + "tokens_seen": 336330752 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045361083249749245, + "loss": 1.6336, + "theoretical_loss": 4.091595641966604, + "tokens_seen": 336396288 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004536008024072217, + "loss": 1.4388, + "theoretical_loss": 4.09150386388181, + "tokens_seen": 336461824 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004535907723169508, + "loss": 1.4657, + "theoretical_loss": 4.091412108676117, + "tokens_seen": 336527360 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045358074222668005, + "loss": 1.587, + "theoretical_loss": 4.091320376339368, + "tokens_seen": 336592896 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045357071213640923, + "loss": 1.6142, + "theoretical_loss": 4.091228666861413, + "tokens_seen": 336658432 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004535606820461384, + "loss": 1.5349, + "theoretical_loss": 4.091136980232107, + "tokens_seen": 336723968 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004535506519558676, + "loss": 1.6583, + "theoretical_loss": 4.091045316441311, + "tokens_seen": 336789504 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045354062186559683, + "loss": 1.5132, + "theoretical_loss": 4.090953675478895, + "tokens_seen": 336855040 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045353059177532596, + "loss": 1.4935, + "theoretical_loss": 4.090862057334734, + "tokens_seen": 336920576 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004535205616850552, + "loss": 1.6545, + "theoretical_loss": 4.090770461998709, + "tokens_seen": 336986112 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004535105315947843, + "loss": 1.5883, + "theoretical_loss": 4.090678889460707, + "tokens_seen": 337051648 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045350050150451355, + "loss": 1.4791, + "theoretical_loss": 4.090587339710625, + "tokens_seen": 337117184 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045349047141424274, + "loss": 1.5873, + "theoretical_loss": 4.090495812738362, + "tokens_seen": 337182720 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004534804413239719, + "loss": 1.6753, + "theoretical_loss": 4.090404308533825, + "tokens_seen": 337248256 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004534704112337011, + "loss": 1.3776, + "theoretical_loss": 4.090312827086929, + "tokens_seen": 337313792 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004534603811434303, + "loss": 1.4019, + "theoretical_loss": 4.090221368387593, + "tokens_seen": 337379328 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004534503510531595, + "loss": 1.5496, + "theoretical_loss": 4.090129932425745, + "tokens_seen": 337444864 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": 0.00011293027637293562, + "objective/train/docs_used": 203033, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.524519920349121, + "objective/train/original_loss": 3.524519443511963, + "objective/train/theoretical_loss": 4.0900385191913164, + "objective/train/tokens_used": 357970400, + "objective/train/value_avg": -0.0049285888671875, + "objective/train/value_loss": 5.1093404181301594e-05, + "objective/train/value_max": -0.0005011558532714844, + "objective/train/value_min": -0.06634521484375, + "objective/train/value_reward_corr": 0.31780553499978886, + "objective/train/value_std": 0.004619598388671875, + "objective/train/weight_avg": 1.0034894943237305, + "objective/train/weighted_lm_loss": 3.535531520843506, + "objective/train/weights_max": 1.7087702751159668, + "objective/train/weights_min": 0.13705755770206451, + "theoretical_loss": 4.0900385191913164, + "tokens_seen": 337510400 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004534403209628887, + "loss": 1.7073, + "theoretical_loss": 4.0900385191913164, + "tokens_seen": 337510400 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004534302908726179, + "loss": 1.5169, + "theoretical_loss": 4.089947128674248, + "tokens_seen": 337575936 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045342026078234706, + "loss": 1.5376, + "theoretical_loss": 4.089855760864484, + "tokens_seen": 337641472 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045341023069207624, + "loss": 1.4673, + "theoretical_loss": 4.089764415751978, + "tokens_seen": 337707008 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004534002006018054, + "loss": 1.5867, + "theoretical_loss": 4.089673093326688, + "tokens_seen": 337772544 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045339017051153465, + "loss": 1.6806, + "theoretical_loss": 4.08958179357858, + "tokens_seen": 337838080 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004533801404212638, + "loss": 1.4107, + "theoretical_loss": 4.089490516497624, + "tokens_seen": 337903616 + }, + { + "epoch": 0.1, + "learning_rate": 0.000453370110330993, + "loss": 1.3901, + "theoretical_loss": 4.089399262073798, + "tokens_seen": 337969152 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004533600802407222, + "loss": 1.506, + "theoretical_loss": 4.089308030297087, + "tokens_seen": 338034688 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004533500501504514, + "loss": 1.4836, + "theoretical_loss": 4.08921682115748, + "tokens_seen": 338100224 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045334002006018056, + "loss": 1.4825, + "theoretical_loss": 4.089125634644975, + "tokens_seen": 338165760 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045332998996990974, + "loss": 1.5108, + "theoretical_loss": 4.089034470749575, + "tokens_seen": 338231296 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004533199598796389, + "loss": 1.5781, + "theoretical_loss": 4.08894332946129, + "tokens_seen": 338296832 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045330992978936816, + "loss": 1.6508, + "theoretical_loss": 4.088852210770135, + "tokens_seen": 338362368 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004532998996990973, + "loss": 1.5131, + "theoretical_loss": 4.088761114666133, + "tokens_seen": 338427904 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004532898696088265, + "loss": 1.4644, + "theoretical_loss": 4.088670041139312, + "tokens_seen": 338493440 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045327983951855565, + "loss": 1.533, + "theoretical_loss": 4.088578990179708, + "tokens_seen": 338558976 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004532698094282849, + "loss": 1.3454, + "theoretical_loss": 4.088487961777361, + "tokens_seen": 338624512 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045325977933801406, + "loss": 1.5538, + "theoretical_loss": 4.0883969559223186, + "tokens_seen": 338690048 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045324974924774324, + "loss": 1.4899, + "theoretical_loss": 4.088305972604636, + "tokens_seen": 338755584 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004532397191574724, + "loss": 1.5443, + "theoretical_loss": 4.088215011814372, + "tokens_seen": 338821120 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004532296890672016, + "loss": 1.429, + "theoretical_loss": 4.088124073541594, + "tokens_seen": 338886656 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004532196589769308, + "loss": 1.6116, + "theoretical_loss": 4.088033157776374, + "tokens_seen": 338952192 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045320962888666, + "loss": 1.5087, + "theoretical_loss": 4.087942264508792, + "tokens_seen": 339017728 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045319959879638915, + "loss": 1.6436, + "theoretical_loss": 4.087851393728933, + "tokens_seen": 339083264 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.0006365600856952369, + "objective/train/docs_used": 204232, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.701159954071045, + "objective/train/original_loss": 2.701160192489624, + "objective/train/theoretical_loss": 4.087760545426889, + "objective/train/tokens_used": 359608800, + "objective/train/value_avg": -0.00475311279296875, + "objective/train/value_loss": 4.972262468072586e-05, + "objective/train/value_max": -0.0004897117614746094, + "objective/train/value_min": -0.07891845703125, + "objective/train/value_reward_corr": 0.25324046174242826, + "objective/train/value_std": 0.0036792755126953125, + "objective/train/weight_avg": 0.9958691596984863, + "objective/train/weighted_lm_loss": 2.6842172145843506, + "objective/train/weights_max": 2.1226985454559326, + "objective/train/weights_min": 0.08537442982196808, + "theoretical_loss": 4.087760545426889, + "tokens_seen": 339148800 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004531895687061184, + "loss": 1.4443, + "theoretical_loss": 4.087760545426889, + "tokens_seen": 339148800 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045317953861584757, + "loss": 1.5022, + "theoretical_loss": 4.087669719592757, + "tokens_seen": 339214336 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045316950852557675, + "loss": 1.6183, + "theoretical_loss": 4.087578916216642, + "tokens_seen": 339279872 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045315947843530593, + "loss": 1.6562, + "theoretical_loss": 4.087488135288654, + "tokens_seen": 339345408 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004531494483450351, + "loss": 1.5854, + "theoretical_loss": 4.0873973767989105, + "tokens_seen": 339410944 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004531394182547643, + "loss": 1.4354, + "theoretical_loss": 4.087306640737534, + "tokens_seen": 339476480 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004531293881644935, + "loss": 1.56, + "theoretical_loss": 4.087215927094654, + "tokens_seen": 339542016 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045311935807422265, + "loss": 1.5285, + "theoretical_loss": 4.087125235860405, + "tokens_seen": 339607552 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004531093279839519, + "loss": 1.5429, + "theoretical_loss": 4.087034567024931, + "tokens_seen": 339673088 + }, + { + "epoch": 0.1, + "learning_rate": 0.000453099297893681, + "loss": 1.4711, + "theoretical_loss": 4.086943920578378, + "tokens_seen": 339738624 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045308926780341025, + "loss": 1.5027, + "theoretical_loss": 4.086853296510902, + "tokens_seen": 339804160 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045307923771313943, + "loss": 1.4114, + "theoretical_loss": 4.086762694812661, + "tokens_seen": 339869696 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004530692076228686, + "loss": 1.5141, + "theoretical_loss": 4.0866721154738235, + "tokens_seen": 339935232 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004530591775325978, + "loss": 1.5755, + "theoretical_loss": 4.086581558484562, + "tokens_seen": 340000768 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045304914744232703, + "loss": 1.4296, + "theoretical_loss": 4.086491023835055, + "tokens_seen": 340066304 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045303911735205616, + "loss": 1.5528, + "theoretical_loss": 4.08640051151549, + "tokens_seen": 340131840 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004530290872617854, + "loss": 1.653, + "theoretical_loss": 4.086310021516056, + "tokens_seen": 340197376 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004530190571715145, + "loss": 1.4999, + "theoretical_loss": 4.086219553826952, + "tokens_seen": 340262912 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045300902708124375, + "loss": 1.5508, + "theoretical_loss": 4.086129108438381, + "tokens_seen": 340328448 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045299899699097294, + "loss": 1.4524, + "theoretical_loss": 4.0860386853405535, + "tokens_seen": 340393984 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004529889669007021, + "loss": 1.4587, + "theoretical_loss": 4.0859482845236865, + "tokens_seen": 340459520 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004529789368104313, + "loss": 1.4728, + "theoretical_loss": 4.085857905978002, + "tokens_seen": 340525056 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004529689067201605, + "loss": 1.6202, + "theoretical_loss": 4.085767549693728, + "tokens_seen": 340590592 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045295887662988966, + "loss": 1.4493, + "theoretical_loss": 4.0856772156611, + "tokens_seen": 340656128 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004529488465396189, + "loss": 1.5061, + "theoretical_loss": 4.085586903870359, + "tokens_seen": 340721664 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": 7.94726875028573e-05, + "objective/train/docs_used": 205032, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3741796016693115, + "objective/train/original_loss": 3.3741791248321533, + "objective/train/theoretical_loss": 4.085496614311752, + "objective/train/tokens_used": 361247200, + "objective/train/value_avg": -0.005878448486328125, + "objective/train/value_loss": 5.7224755437346175e-05, + "objective/train/value_max": -0.0006694793701171875, + "objective/train/value_min": -0.0537109375, + "objective/train/value_reward_corr": 0.302335219359007, + "objective/train/value_std": 0.00423431396484375, + "objective/train/weight_avg": 1.0035232305526733, + "objective/train/weighted_lm_loss": 3.384261131286621, + "objective/train/weights_max": 1.5559977293014526, + "objective/train/weights_min": 0.35189223289489746, + "theoretical_loss": 4.085496614311752, + "tokens_seen": 340787200 + }, + { + "epoch": 0.1, + "learning_rate": 0.000452938816449348, + "loss": 1.6144, + "theoretical_loss": 4.085496614311752, + "tokens_seen": 340787200 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045292878635907726, + "loss": 1.6556, + "theoretical_loss": 4.085406346975532, + "tokens_seen": 340852736 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004529187562688064, + "loss": 1.7103, + "theoretical_loss": 4.085316101851959, + "tokens_seen": 340918272 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004529087261785356, + "loss": 1.4434, + "theoretical_loss": 4.085225878931297, + "tokens_seen": 340983808 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004528986960882648, + "loss": 1.5902, + "theoretical_loss": 4.0851356782038195, + "tokens_seen": 341049344 + }, + { + "epoch": 0.1, + "learning_rate": 0.000452888665997994, + "loss": 1.6113, + "theoretical_loss": 4.085045499659804, + "tokens_seen": 341114880 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045287863590772316, + "loss": 1.4857, + "theoretical_loss": 4.084955343289534, + "tokens_seen": 341180416 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004528686058174524, + "loss": 1.5109, + "theoretical_loss": 4.084865209083299, + "tokens_seen": 341245952 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004528585757271815, + "loss": 1.5953, + "theoretical_loss": 4.084775097031397, + "tokens_seen": 341311488 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045284854563691076, + "loss": 1.3534, + "theoretical_loss": 4.084685007124127, + "tokens_seen": 341377024 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004528385155466399, + "loss": 1.5614, + "theoretical_loss": 4.084594939351801, + "tokens_seen": 341442560 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004528284854563691, + "loss": 1.4961, + "theoretical_loss": 4.084504893704732, + "tokens_seen": 341508096 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004528184553660983, + "loss": 1.5194, + "theoretical_loss": 4.084414870173241, + "tokens_seen": 341573632 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004528084252758275, + "loss": 1.5389, + "theoretical_loss": 4.084324868747653, + "tokens_seen": 341639168 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045279839518555667, + "loss": 1.5876, + "theoretical_loss": 4.084234889418303, + "tokens_seen": 341704704 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045278836509528585, + "loss": 1.5049, + "theoretical_loss": 4.084144932175528, + "tokens_seen": 341770240 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045277833500501503, + "loss": 1.5673, + "theoretical_loss": 4.084054997009675, + "tokens_seen": 341835776 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045276830491474426, + "loss": 1.6139, + "theoretical_loss": 4.083965083911093, + "tokens_seen": 341901312 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004527582748244734, + "loss": 1.6571, + "theoretical_loss": 4.08387519287014, + "tokens_seen": 341966848 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004527482447342026, + "loss": 1.6443, + "theoretical_loss": 4.08378532387718, + "tokens_seen": 342032384 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045273821464393175, + "loss": 1.5345, + "theoretical_loss": 4.08369547692258, + "tokens_seen": 342097920 + }, + { + "epoch": 0.1, + "learning_rate": 0.000452728184553661, + "loss": 1.4454, + "theoretical_loss": 4.0836056519967165, + "tokens_seen": 342163456 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045271815446339017, + "loss": 1.4925, + "theoretical_loss": 4.083515849089971, + "tokens_seen": 342228992 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045270812437311935, + "loss": 1.6074, + "theoretical_loss": 4.083426068192731, + "tokens_seen": 342294528 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004526980942828486, + "loss": 1.494, + "theoretical_loss": 4.0833363092953885, + "tokens_seen": 342360064 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.0006433430244214833, + "objective/train/docs_used": 206109, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.224832057952881, + "objective/train/original_loss": 3.22483229637146, + "objective/train/theoretical_loss": 4.083246572388344, + "objective/train/tokens_used": 362885600, + "objective/train/value_avg": -0.005706787109375, + "objective/train/value_loss": 5.7340665080118924e-05, + "objective/train/value_max": -0.00039196014404296875, + "objective/train/value_min": -0.08331298828125, + "objective/train/value_reward_corr": 0.3306582823025273, + "objective/train/value_std": 0.004505157470703125, + "objective/train/weight_avg": 0.996360182762146, + "objective/train/weighted_lm_loss": 3.213338851928711, + "objective/train/weights_max": 1.7583938837051392, + "objective/train/weights_min": 0.5146464705467224, + "theoretical_loss": 4.083246572388344, + "tokens_seen": 342425600 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045268806419257777, + "loss": 1.6064, + "theoretical_loss": 4.083246572388344, + "tokens_seen": 342425600 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045267803410230695, + "loss": 1.4131, + "theoretical_loss": 4.0831568574620025, + "tokens_seen": 342491136 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045266800401203613, + "loss": 1.5059, + "theoretical_loss": 4.083067164506776, + "tokens_seen": 342556672 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004526579739217653, + "loss": 1.6302, + "theoretical_loss": 4.082977493513082, + "tokens_seen": 342622208 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004526479438314945, + "loss": 1.4552, + "theoretical_loss": 4.082887844471342, + "tokens_seen": 342687744 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004526379137412237, + "loss": 1.5681, + "theoretical_loss": 4.0827982173719874, + "tokens_seen": 342753280 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045262788365095285, + "loss": 1.52, + "theoretical_loss": 4.082708612205454, + "tokens_seen": 342818816 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004526178535606821, + "loss": 1.4615, + "theoretical_loss": 4.082619028962182, + "tokens_seen": 342884352 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004526078234704112, + "loss": 1.466, + "theoretical_loss": 4.08252946763262, + "tokens_seen": 342949888 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045259779338014045, + "loss": 1.4581, + "theoretical_loss": 4.0824399282072195, + "tokens_seen": 343015424 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045258776328986963, + "loss": 1.5951, + "theoretical_loss": 4.082350410676442, + "tokens_seen": 343080960 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004525777331995988, + "loss": 1.6051, + "theoretical_loss": 4.082260915030751, + "tokens_seen": 343146496 + }, + { + "epoch": 0.1, + "learning_rate": 0.000452567703109328, + "loss": 1.3892, + "theoretical_loss": 4.082171441260621, + "tokens_seen": 343212032 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045255767301905723, + "loss": 1.474, + "theoretical_loss": 4.082081989356525, + "tokens_seen": 343277568 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045254764292878636, + "loss": 1.5378, + "theoretical_loss": 4.08199255930895, + "tokens_seen": 343343104 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004525376128385156, + "loss": 1.4062, + "theoretical_loss": 4.081903151108384, + "tokens_seen": 343408640 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004525275827482447, + "loss": 1.457, + "theoretical_loss": 4.081813764745322, + "tokens_seen": 343474176 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045251755265797395, + "loss": 1.5957, + "theoretical_loss": 4.081724400210264, + "tokens_seen": 343539712 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045250752256770314, + "loss": 1.5636, + "theoretical_loss": 4.0816350574937195, + "tokens_seen": 343605248 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004524974924774323, + "loss": 1.4286, + "theoretical_loss": 4.0815457365862, + "tokens_seen": 343670784 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004524874623871615, + "loss": 1.5794, + "theoretical_loss": 4.081456437478225, + "tokens_seen": 343736320 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004524774322968907, + "loss": 1.4994, + "theoretical_loss": 4.081367160160319, + "tokens_seen": 343801856 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045246740220661986, + "loss": 1.4644, + "theoretical_loss": 4.081277904623013, + "tokens_seen": 343867392 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004524573721163491, + "loss": 1.5472, + "theoretical_loss": 4.081188670856844, + "tokens_seen": 343932928 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004524473420260782, + "loss": 1.3457, + "theoretical_loss": 4.081099458852354, + "tokens_seen": 343998464 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.0006123662460595369, + "objective/train/docs_used": 206760, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4187471866607666, + "objective/train/original_loss": 3.418747663497925, + "objective/train/theoretical_loss": 4.081010268600093, + "objective/train/tokens_used": 364524000, + "objective/train/value_avg": -0.005359649658203125, + "objective/train/value_loss": 7.02514880686067e-05, + "objective/train/value_max": -0.000614166259765625, + "objective/train/value_min": -0.056243896484375, + "objective/train/value_reward_corr": 0.3724014368493018, + "objective/train/value_std": 0.004001617431640625, + "objective/train/weight_avg": 0.997022271156311, + "objective/train/weighted_lm_loss": 3.423274517059326, + "objective/train/weights_max": 1.4401122331619263, + "objective/train/weights_min": 0.08566150814294815, + "theoretical_loss": 4.081010268600093, + "tokens_seen": 344064000 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045243731193580746, + "loss": 1.5181, + "theoretical_loss": 4.081010268600093, + "tokens_seen": 344064000 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004524272818455366, + "loss": 1.6519, + "theoretical_loss": 4.080921100090614, + "tokens_seen": 344129536 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004524172517552658, + "loss": 1.5006, + "theoretical_loss": 4.080831953314478, + "tokens_seen": 344195072 + }, + { + "epoch": 0.1, + "learning_rate": 0.000452407221664995, + "loss": 1.5037, + "theoretical_loss": 4.080742828262251, + "tokens_seen": 344260608 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004523971915747242, + "loss": 1.565, + "theoretical_loss": 4.080653724924504, + "tokens_seen": 344326144 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045238716148445336, + "loss": 1.4398, + "theoretical_loss": 4.080564643291818, + "tokens_seen": 344391680 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004523771313941826, + "loss": 1.4851, + "theoretical_loss": 4.080475583354775, + "tokens_seen": 344457216 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004523671013039117, + "loss": 1.5286, + "theoretical_loss": 4.0803865451039645, + "tokens_seen": 344522752 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045235707121364096, + "loss": 1.556, + "theoretical_loss": 4.080297528529983, + "tokens_seen": 344588288 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004523470411233701, + "loss": 1.4681, + "theoretical_loss": 4.080208533623432, + "tokens_seen": 344653824 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004523370110330993, + "loss": 1.5661, + "theoretical_loss": 4.080119560374918, + "tokens_seen": 344719360 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004523269809428285, + "loss": 1.547, + "theoretical_loss": 4.080030608775055, + "tokens_seen": 344784896 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004523169508525577, + "loss": 1.5631, + "theoretical_loss": 4.079941678814462, + "tokens_seen": 344850432 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045230692076228687, + "loss": 1.4258, + "theoretical_loss": 4.079852770483765, + "tokens_seen": 344915968 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045229689067201605, + "loss": 1.5176, + "theoretical_loss": 4.079763883773593, + "tokens_seen": 344981504 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045228686058174523, + "loss": 1.5815, + "theoretical_loss": 4.079675018674582, + "tokens_seen": 345047040 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045227683049147446, + "loss": 1.5723, + "theoretical_loss": 4.079586175177377, + "tokens_seen": 345112576 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004522668004012036, + "loss": 1.3421, + "theoretical_loss": 4.079497353272625, + "tokens_seen": 345178112 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004522567703109328, + "loss": 1.4513, + "theoretical_loss": 4.07940855295098, + "tokens_seen": 345243648 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045224674022066195, + "loss": 1.3933, + "theoretical_loss": 4.079319774203102, + "tokens_seen": 345309184 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004522367101303912, + "loss": 1.4912, + "theoretical_loss": 4.079231017019657, + "tokens_seen": 345374720 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045222668004012037, + "loss": 1.6145, + "theoretical_loss": 4.0791422813913165, + "tokens_seen": 345440256 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045221664994984955, + "loss": 1.4715, + "theoretical_loss": 4.079053567308757, + "tokens_seen": 345505792 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045220661985957873, + "loss": 1.3768, + "theoretical_loss": 4.078964874762663, + "tokens_seen": 345571328 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045219658976930797, + "loss": 1.618, + "theoretical_loss": 4.078876203743723, + "tokens_seen": 345636864 + }, + { + "epoch": 0.1, + "objective/train/advantage_avg": -0.00044803437776863575, + "objective/train/docs_used": 207336, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1335790157318115, + "objective/train/original_loss": 3.1335787773132324, + "objective/train/theoretical_loss": 4.0787875542426315, + "objective/train/tokens_used": 366162400, + "objective/train/value_avg": -0.00583648681640625, + "objective/train/value_loss": 8.741112833376974e-05, + "objective/train/value_max": -0.00048041343688964844, + "objective/train/value_min": -0.06561279296875, + "objective/train/value_reward_corr": 0.2906312689519084, + "objective/train/value_std": 0.0048828125, + "objective/train/weight_avg": 0.9995080232620239, + "objective/train/weighted_lm_loss": 3.1292059421539307, + "objective/train/weights_max": 1.6314369440078735, + "objective/train/weights_min": 0.19377626478672028, + "theoretical_loss": 4.0787875542426315, + "tokens_seen": 345702400 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004521865596790371, + "loss": 1.5232, + "theoretical_loss": 4.0787875542426315, + "tokens_seen": 345702400 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045217652958876633, + "loss": 1.3574, + "theoretical_loss": 4.078698926250089, + "tokens_seen": 345767936 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045216649949849546, + "loss": 1.5111, + "theoretical_loss": 4.078610319756802, + "tokens_seen": 345833472 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004521564694082247, + "loss": 1.4327, + "theoretical_loss": 4.078521734753482, + "tokens_seen": 345899008 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045214643931795387, + "loss": 1.5246, + "theoretical_loss": 4.078433171230849, + "tokens_seen": 345964544 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045213640922768305, + "loss": 1.3936, + "theoretical_loss": 4.078344629179623, + "tokens_seen": 346030080 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045212637913741223, + "loss": 1.6217, + "theoretical_loss": 4.078256108590535, + "tokens_seen": 346095616 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004521163490471414, + "loss": 1.5043, + "theoretical_loss": 4.078167609454322, + "tokens_seen": 346161152 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004521063189568706, + "loss": 1.5127, + "theoretical_loss": 4.078079131761723, + "tokens_seen": 346226688 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045209628886659983, + "loss": 1.5888, + "theoretical_loss": 4.077990675503484, + "tokens_seen": 346292224 + }, + { + "epoch": 0.1, + "learning_rate": 0.00045208625877632896, + "loss": 1.5328, + "theoretical_loss": 4.07790224067036, + "tokens_seen": 346357760 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004520762286860582, + "loss": 1.3434, + "theoretical_loss": 4.077813827253106, + "tokens_seen": 346423296 + }, + { + "epoch": 0.1, + "learning_rate": 0.0004520661985957873, + "loss": 1.4736, + "theoretical_loss": 4.077725435242488, + "tokens_seen": 346488832 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045205616850551656, + "loss": 1.3941, + "theoretical_loss": 4.077637064629275, + "tokens_seen": 346554368 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045204613841524574, + "loss": 1.4418, + "theoretical_loss": 4.077548715404243, + "tokens_seen": 346619904 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004520361083249749, + "loss": 1.528, + "theoretical_loss": 4.077460387558171, + "tokens_seen": 346685440 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004520260782347041, + "loss": 1.608, + "theoretical_loss": 4.077372081081847, + "tokens_seen": 346750976 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045201604814443334, + "loss": 1.5951, + "theoretical_loss": 4.077283795966063, + "tokens_seen": 346816512 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045200601805416246, + "loss": 1.536, + "theoretical_loss": 4.077195532201619, + "tokens_seen": 346882048 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004519959879638917, + "loss": 1.5299, + "theoretical_loss": 4.077107289779317, + "tokens_seen": 346947584 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004519859578736208, + "loss": 1.472, + "theoretical_loss": 4.0770190686899666, + "tokens_seen": 347013120 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045197592778335006, + "loss": 1.43, + "theoretical_loss": 4.076930868924384, + "tokens_seen": 347078656 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045196589769307924, + "loss": 1.5102, + "theoretical_loss": 4.07684269047339, + "tokens_seen": 347144192 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004519558676028084, + "loss": 1.4919, + "theoretical_loss": 4.07675453332781, + "tokens_seen": 347209728 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045194583751253766, + "loss": 1.477, + "theoretical_loss": 4.076666397478476, + "tokens_seen": 347275264 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0010245086159557104, + "objective/train/docs_used": 208796, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.826882839202881, + "objective/train/original_loss": 2.8268826007843018, + "objective/train/theoretical_loss": 4.076578282916229, + "objective/train/tokens_used": 367800800, + "objective/train/value_avg": -0.006565093994140625, + "objective/train/value_loss": 9.115591092268005e-05, + "objective/train/value_max": -0.0003275871276855469, + "objective/train/value_min": -0.0894775390625, + "objective/train/value_reward_corr": 0.435832723385585, + "objective/train/value_std": 0.006023406982421875, + "objective/train/weight_avg": 0.9938446879386902, + "objective/train/weighted_lm_loss": 2.8004066944122314, + "objective/train/weights_max": 1.9310410022735596, + "objective/train/weights_min": 0.1931268572807312, + "theoretical_loss": 4.076578282916229, + "tokens_seen": 347340800 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004519358074222668, + "loss": 1.4493, + "theoretical_loss": 4.076578282916229, + "tokens_seen": 347340800 + }, + { + "epoch": 0.11, + "learning_rate": 0.000451925777331996, + "loss": 1.5594, + "theoretical_loss": 4.07649018963191, + "tokens_seen": 347406336 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004519157472417252, + "loss": 1.5917, + "theoretical_loss": 4.07640211761637, + "tokens_seen": 347471872 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004519057171514544, + "loss": 1.5413, + "theoretical_loss": 4.076314066860462, + "tokens_seen": 347537408 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045189568706118356, + "loss": 1.4714, + "theoretical_loss": 4.076226037355047, + "tokens_seen": 347602944 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004518856569709128, + "loss": 1.4337, + "theoretical_loss": 4.076138029090993, + "tokens_seen": 347668480 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004518756268806419, + "loss": 1.4962, + "theoretical_loss": 4.076050042059171, + "tokens_seen": 347734016 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045186559679037116, + "loss": 1.4286, + "theoretical_loss": 4.075962076250458, + "tokens_seen": 347799552 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004518555667001003, + "loss": 1.4616, + "theoretical_loss": 4.075874131655738, + "tokens_seen": 347865088 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004518455366098295, + "loss": 1.4529, + "theoretical_loss": 4.075786208265899, + "tokens_seen": 347930624 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004518355065195587, + "loss": 1.5012, + "theoretical_loss": 4.0756983060718355, + "tokens_seen": 347996160 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004518254764292879, + "loss": 1.5365, + "theoretical_loss": 4.075610425064449, + "tokens_seen": 348061696 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045181544633901707, + "loss": 1.6015, + "theoretical_loss": 4.075522565234643, + "tokens_seen": 348127232 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045180541624874625, + "loss": 1.4424, + "theoretical_loss": 4.075434726573331, + "tokens_seen": 348192768 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045179538615847543, + "loss": 1.4357, + "theoretical_loss": 4.0753469090714285, + "tokens_seen": 348258304 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045178535606820466, + "loss": 1.4044, + "theoretical_loss": 4.075259112719859, + "tokens_seen": 348323840 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004517753259779338, + "loss": 1.5242, + "theoretical_loss": 4.075171337509549, + "tokens_seen": 348389376 + }, + { + "epoch": 0.11, + "learning_rate": 0.000451765295887663, + "loss": 1.3968, + "theoretical_loss": 4.075083583431434, + "tokens_seen": 348454912 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045175526579739215, + "loss": 1.4269, + "theoretical_loss": 4.074995850476453, + "tokens_seen": 348520448 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004517452357071214, + "loss": 1.4946, + "theoretical_loss": 4.074908138635551, + "tokens_seen": 348585984 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045173520561685057, + "loss": 1.4729, + "theoretical_loss": 4.074820447899677, + "tokens_seen": 348651520 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045172517552657975, + "loss": 1.4316, + "theoretical_loss": 4.074732778259788, + "tokens_seen": 348717056 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045171514543630893, + "loss": 1.5543, + "theoretical_loss": 4.0746451297068464, + "tokens_seen": 348782592 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045170511534603817, + "loss": 1.5398, + "theoretical_loss": 4.074557502231819, + "tokens_seen": 348848128 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004516950852557673, + "loss": 1.5031, + "theoretical_loss": 4.074469895825678, + "tokens_seen": 348913664 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0019778357818722725, + "objective/train/docs_used": 209597, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.621542453765869, + "objective/train/original_loss": 2.621542453765869, + "objective/train/theoretical_loss": 4.074382310479402, + "objective/train/tokens_used": 369439200, + "objective/train/value_avg": -0.006175994873046875, + "objective/train/value_loss": 0.0001100477747968398, + "objective/train/value_max": -0.00032258033752441406, + "objective/train/value_min": -0.062103271484375, + "objective/train/value_reward_corr": 0.5045369472700849, + "objective/train/value_std": 0.006435394287109375, + "objective/train/weight_avg": 0.9850577116012573, + "objective/train/weighted_lm_loss": 2.572658061981201, + "objective/train/weights_max": 1.7770438194274902, + "objective/train/weights_min": 0.0835094302892685, + "theoretical_loss": 4.074382310479402, + "tokens_seen": 348979200 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045168505516549653, + "loss": 1.3204, + "theoretical_loss": 4.074382310479402, + "tokens_seen": 348979200 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045167502507522566, + "loss": 1.4641, + "theoretical_loss": 4.074294746183976, + "tokens_seen": 349044736 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004516649949849549, + "loss": 1.5308, + "theoretical_loss": 4.0742072029303875, + "tokens_seen": 349110272 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045165496489468407, + "loss": 1.514, + "theoretical_loss": 4.074119680709633, + "tokens_seen": 349175808 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045164493480441325, + "loss": 1.5502, + "theoretical_loss": 4.074032179512713, + "tokens_seen": 349241344 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045163490471414244, + "loss": 1.4759, + "theoretical_loss": 4.073944699330632, + "tokens_seen": 349306880 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004516248746238716, + "loss": 1.5032, + "theoretical_loss": 4.073857240154403, + "tokens_seen": 349372416 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004516148445336008, + "loss": 1.479, + "theoretical_loss": 4.073769801975043, + "tokens_seen": 349437952 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045160481444333003, + "loss": 1.525, + "theoretical_loss": 4.0736823847835755, + "tokens_seen": 349503488 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045159478435305916, + "loss": 1.4293, + "theoretical_loss": 4.073594988571027, + "tokens_seen": 349569024 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004515847542627884, + "loss": 1.4673, + "theoretical_loss": 4.073507613328433, + "tokens_seen": 349634560 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004515747241725175, + "loss": 1.5574, + "theoretical_loss": 4.07342025904683, + "tokens_seen": 349700096 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045156469408224676, + "loss": 1.479, + "theoretical_loss": 4.073332925717265, + "tokens_seen": 349765632 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045155466399197594, + "loss": 1.3974, + "theoretical_loss": 4.073245613330788, + "tokens_seen": 349831168 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004515446339017051, + "loss": 1.4223, + "theoretical_loss": 4.073158321878454, + "tokens_seen": 349896704 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004515346038114343, + "loss": 1.4661, + "theoretical_loss": 4.073071051351325, + "tokens_seen": 349962240 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045152457372116354, + "loss": 1.4813, + "theoretical_loss": 4.072983801740467, + "tokens_seen": 350027776 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045151454363089266, + "loss": 1.5765, + "theoretical_loss": 4.072896573036953, + "tokens_seen": 350093312 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004515045135406219, + "loss": 1.4321, + "theoretical_loss": 4.072809365231859, + "tokens_seen": 350158848 + }, + { + "epoch": 0.11, + "learning_rate": 0.000451494483450351, + "loss": 1.4995, + "theoretical_loss": 4.072722178316271, + "tokens_seen": 350224384 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045148445336008026, + "loss": 1.4254, + "theoretical_loss": 4.072635012281275, + "tokens_seen": 350289920 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045147442326980944, + "loss": 1.4446, + "theoretical_loss": 4.072547867117966, + "tokens_seen": 350355456 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004514643931795386, + "loss": 1.4827, + "theoretical_loss": 4.072460742817444, + "tokens_seen": 350420992 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004514543630892678, + "loss": 1.4816, + "theoretical_loss": 4.072373639370814, + "tokens_seen": 350486528 + }, + { + "epoch": 0.11, + "learning_rate": 0.000451444332998997, + "loss": 1.3984, + "theoretical_loss": 4.0722865567691855, + "tokens_seen": 350552064 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0013166374992579222, + "objective/train/docs_used": 210243, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.691506862640381, + "objective/train/original_loss": 2.69150710105896, + "objective/train/theoretical_loss": 4.072199495003675, + "objective/train/tokens_used": 371077600, + "objective/train/value_avg": -0.00489044189453125, + "objective/train/value_loss": 7.329282379942015e-05, + "objective/train/value_max": -0.0004513263702392578, + "objective/train/value_min": -0.09088134765625, + "objective/train/value_reward_corr": 0.2789733108859025, + "objective/train/value_std": 0.0035247802734375, + "objective/train/weight_avg": 0.9903168082237244, + "objective/train/weighted_lm_loss": 2.6873018741607666, + "objective/train/weights_max": 2.2524967193603516, + "objective/train/weights_min": 0.7070522308349609, + "theoretical_loss": 4.072199495003675, + "tokens_seen": 350617600 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045143430290872617, + "loss": 1.3954, + "theoretical_loss": 4.072199495003675, + "tokens_seen": 350617600 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004514242728184554, + "loss": 1.4938, + "theoretical_loss": 4.072112454065405, + "tokens_seen": 350683136 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045141424272818453, + "loss": 1.4309, + "theoretical_loss": 4.0720254339455, + "tokens_seen": 350748672 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045140421263791376, + "loss": 1.3804, + "theoretical_loss": 4.0719384346350935, + "tokens_seen": 350814208 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045139418254764294, + "loss": 1.3456, + "theoretical_loss": 4.071851456125324, + "tokens_seen": 350879744 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004513841524573721, + "loss": 1.5424, + "theoretical_loss": 4.071764498407333, + "tokens_seen": 350945280 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004513741223671013, + "loss": 1.5247, + "theoretical_loss": 4.071677561472271, + "tokens_seen": 351010816 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004513640922768305, + "loss": 1.5308, + "theoretical_loss": 4.071590645311289, + "tokens_seen": 351076352 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045135406218655967, + "loss": 1.5833, + "theoretical_loss": 4.07150374991555, + "tokens_seen": 351141888 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004513440320962889, + "loss": 1.3648, + "theoretical_loss": 4.071416875276216, + "tokens_seen": 351207424 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045133400200601803, + "loss": 1.4134, + "theoretical_loss": 4.071330021384458, + "tokens_seen": 351272960 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045132397191574727, + "loss": 1.5308, + "theoretical_loss": 4.071243188231452, + "tokens_seen": 351338496 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004513139418254764, + "loss": 1.4805, + "theoretical_loss": 4.071156375808378, + "tokens_seen": 351404032 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045130391173520563, + "loss": 1.4628, + "theoretical_loss": 4.071069584106424, + "tokens_seen": 351469568 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004512938816449348, + "loss": 1.5298, + "theoretical_loss": 4.070982813116781, + "tokens_seen": 351535104 + }, + { + "epoch": 0.11, + "learning_rate": 0.000451283851554664, + "loss": 1.431, + "theoretical_loss": 4.070896062830645, + "tokens_seen": 351600640 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045127382146439317, + "loss": 1.3952, + "theoretical_loss": 4.070809333239221, + "tokens_seen": 351666176 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045126379137412235, + "loss": 1.4417, + "theoretical_loss": 4.0707226243337145, + "tokens_seen": 351731712 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045125376128385153, + "loss": 1.4734, + "theoretical_loss": 4.070635936105339, + "tokens_seen": 351797248 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045124373119358077, + "loss": 1.4324, + "theoretical_loss": 4.0705492685453155, + "tokens_seen": 351862784 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004512337011033099, + "loss": 1.5592, + "theoretical_loss": 4.070462621644865, + "tokens_seen": 351928320 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045122367101303913, + "loss": 1.3975, + "theoretical_loss": 4.07037599539522, + "tokens_seen": 351993856 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004512136409227683, + "loss": 1.5937, + "theoretical_loss": 4.070289389787613, + "tokens_seen": 352059392 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004512036108324975, + "loss": 1.5934, + "theoretical_loss": 4.070202804813284, + "tokens_seen": 352124928 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045119358074222673, + "loss": 1.4807, + "theoretical_loss": 4.070116240463479, + "tokens_seen": 352190464 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.00013559049693867564, + "objective/train/docs_used": 211560, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.18184494972229, + "objective/train/original_loss": 3.1818454265594482, + "objective/train/theoretical_loss": 4.070029696729449, + "objective/train/tokens_used": 372716000, + "objective/train/value_avg": -0.005954742431640625, + "objective/train/value_loss": 8.581460133427754e-05, + "objective/train/value_max": -0.0003712177276611328, + "objective/train/value_min": -0.07342529296875, + "objective/train/value_reward_corr": 0.5691572690102488, + "objective/train/value_std": 0.005596160888671875, + "objective/train/weight_avg": 1.0026335716247559, + "objective/train/weighted_lm_loss": 3.192401647567749, + "objective/train/weights_max": 2.053203821182251, + "objective/train/weights_min": 0.4187878370285034, + "theoretical_loss": 4.070029696729449, + "tokens_seen": 352256000 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045118355065195586, + "loss": 1.4694, + "theoretical_loss": 4.070029696729449, + "tokens_seen": 352256000 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004511735205616851, + "loss": 1.4532, + "theoretical_loss": 4.069943173602451, + "tokens_seen": 352321536 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004511634904714143, + "loss": 1.5643, + "theoretical_loss": 4.069856671073745, + "tokens_seen": 352387072 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045115346038114345, + "loss": 1.3901, + "theoretical_loss": 4.069770189134598, + "tokens_seen": 352452608 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045114343029087264, + "loss": 1.5026, + "theoretical_loss": 4.069683727776281, + "tokens_seen": 352518144 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004511334002006018, + "loss": 1.4764, + "theoretical_loss": 4.069597286990075, + "tokens_seen": 352583680 + }, + { + "epoch": 0.11, + "learning_rate": 0.000451123370110331, + "loss": 1.3794, + "theoretical_loss": 4.06951086676726, + "tokens_seen": 352649216 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045111334002006023, + "loss": 1.4311, + "theoretical_loss": 4.0694244670991235, + "tokens_seen": 352714752 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045110330992978936, + "loss": 1.3854, + "theoretical_loss": 4.069338087976961, + "tokens_seen": 352780288 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004510932798395186, + "loss": 1.4925, + "theoretical_loss": 4.06925172939207, + "tokens_seen": 352845824 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004510832497492477, + "loss": 1.5105, + "theoretical_loss": 4.069165391335753, + "tokens_seen": 352911360 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045107321965897696, + "loss": 1.5231, + "theoretical_loss": 4.0690790737993225, + "tokens_seen": 352976896 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045106318956870614, + "loss": 1.4607, + "theoretical_loss": 4.068992776774091, + "tokens_seen": 353042432 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004510531594784353, + "loss": 1.4427, + "theoretical_loss": 4.068906500251378, + "tokens_seen": 353107968 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004510431293881645, + "loss": 1.539, + "theoretical_loss": 4.068820244222511, + "tokens_seen": 353173504 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045103309929789374, + "loss": 1.4765, + "theoretical_loss": 4.068734008678818, + "tokens_seen": 353239040 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045102306920762286, + "loss": 1.5075, + "theoretical_loss": 4.068647793611635, + "tokens_seen": 353304576 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004510130391173521, + "loss": 1.4783, + "theoretical_loss": 4.068561599012304, + "tokens_seen": 353370112 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004510030090270812, + "loss": 1.5394, + "theoretical_loss": 4.068475424872171, + "tokens_seen": 353435648 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045099297893681046, + "loss": 1.3932, + "theoretical_loss": 4.068389271182586, + "tokens_seen": 353501184 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045098294884653964, + "loss": 1.501, + "theoretical_loss": 4.068303137934908, + "tokens_seen": 353566720 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004509729187562688, + "loss": 1.6032, + "theoretical_loss": 4.0682170251204965, + "tokens_seen": 353632256 + }, + { + "epoch": 0.11, + "learning_rate": 0.000450962888665998, + "loss": 1.4478, + "theoretical_loss": 4.068130932730722, + "tokens_seen": 353697792 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004509528585757272, + "loss": 1.4354, + "theoretical_loss": 4.068044860756954, + "tokens_seen": 353763328 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045094282848545637, + "loss": 1.5188, + "theoretical_loss": 4.067958809190571, + "tokens_seen": 353828864 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0003648845595307648, + "objective/train/docs_used": 212189, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6738290786743164, + "objective/train/original_loss": 2.6738290786743164, + "objective/train/theoretical_loss": 4.0678727780229575, + "objective/train/tokens_used": 374354400, + "objective/train/value_avg": -0.00519561767578125, + "objective/train/value_loss": 5.6290784414159134e-05, + "objective/train/value_max": -0.00033545494079589844, + "objective/train/value_min": -0.04876708984375, + "objective/train/value_reward_corr": 0.3402517698038429, + "objective/train/value_std": 0.004352569580078125, + "objective/train/weight_avg": 0.9989864826202393, + "objective/train/weighted_lm_loss": 2.6720874309539795, + "objective/train/weights_max": 1.5699594020843506, + "objective/train/weights_min": 0.396958589553833, + "theoretical_loss": 4.0678727780229575, + "tokens_seen": 353894400 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004509327983951856, + "loss": 1.4729, + "theoretical_loss": 4.0678727780229575, + "tokens_seen": 353894400 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045092276830491473, + "loss": 1.5983, + "theoretical_loss": 4.0677867672455, + "tokens_seen": 353959936 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045091273821464396, + "loss": 1.4505, + "theoretical_loss": 4.067700776849592, + "tokens_seen": 354025472 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045090270812437314, + "loss": 1.5059, + "theoretical_loss": 4.067614806826633, + "tokens_seen": 354091008 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004508926780341023, + "loss": 1.4031, + "theoretical_loss": 4.067528857168027, + "tokens_seen": 354156544 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004508826479438315, + "loss": 1.4748, + "theoretical_loss": 4.067442927865182, + "tokens_seen": 354222080 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004508726178535607, + "loss": 1.386, + "theoretical_loss": 4.067357018909513, + "tokens_seen": 354287616 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045086258776328987, + "loss": 1.4724, + "theoretical_loss": 4.06727113029244, + "tokens_seen": 354353152 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004508525576730191, + "loss": 1.5344, + "theoretical_loss": 4.0671852620053865, + "tokens_seen": 354418688 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045084252758274823, + "loss": 1.3461, + "theoretical_loss": 4.067099414039784, + "tokens_seen": 354484224 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045083249749247747, + "loss": 1.5724, + "theoretical_loss": 4.067013586387065, + "tokens_seen": 354549760 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004508224674022066, + "loss": 1.5184, + "theoretical_loss": 4.066927779038673, + "tokens_seen": 354615296 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045081243731193583, + "loss": 1.3765, + "theoretical_loss": 4.066841991986053, + "tokens_seen": 354680832 + }, + { + "epoch": 0.11, + "learning_rate": 0.000450802407221665, + "loss": 1.4701, + "theoretical_loss": 4.066756225220653, + "tokens_seen": 354746368 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004507923771313942, + "loss": 1.4733, + "theoretical_loss": 4.066670478733931, + "tokens_seen": 354811904 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045078234704112337, + "loss": 1.4031, + "theoretical_loss": 4.066584752517348, + "tokens_seen": 354877440 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045077231695085255, + "loss": 1.4647, + "theoretical_loss": 4.066499046562369, + "tokens_seen": 354942976 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045076228686058173, + "loss": 1.5035, + "theoretical_loss": 4.066413360860468, + "tokens_seen": 355008512 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045075225677031097, + "loss": 1.526, + "theoretical_loss": 4.066327695403118, + "tokens_seen": 355074048 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004507422266800401, + "loss": 1.5332, + "theoretical_loss": 4.066242050181803, + "tokens_seen": 355139584 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045073219658976933, + "loss": 1.5448, + "theoretical_loss": 4.06615642518801, + "tokens_seen": 355205120 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004507221664994985, + "loss": 1.4781, + "theoretical_loss": 4.066070820413229, + "tokens_seen": 355270656 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004507121364092277, + "loss": 1.5072, + "theoretical_loss": 4.065985235848959, + "tokens_seen": 355336192 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004507021063189569, + "loss": 1.498, + "theoretical_loss": 4.065899671486703, + "tokens_seen": 355401728 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045069207622868606, + "loss": 1.4695, + "theoretical_loss": 4.0658141273179655, + "tokens_seen": 355467264 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0004946683184243739, + "objective/train/docs_used": 213521, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5360188484191895, + "objective/train/original_loss": 2.5360186100006104, + "objective/train/theoretical_loss": 4.065728603334263, + "objective/train/tokens_used": 375992800, + "objective/train/value_avg": -0.005619049072265625, + "objective/train/value_loss": 5.870456516277045e-05, + "objective/train/value_max": -0.00037550926208496094, + "objective/train/value_min": -0.05706787109375, + "objective/train/value_reward_corr": 0.43789969720757743, + "objective/train/value_std": 0.00560760498046875, + "objective/train/weight_avg": 0.9979245066642761, + "objective/train/weighted_lm_loss": 2.536473512649536, + "objective/train/weights_max": 1.7694674730300903, + "objective/train/weights_min": 0.4790712296962738, + "theoretical_loss": 4.065728603334263, + "tokens_seen": 355532800 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045068204613841524, + "loss": 1.4196, + "theoretical_loss": 4.065728603334263, + "tokens_seen": 355532800 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004506720160481445, + "loss": 1.4852, + "theoretical_loss": 4.065643099527109, + "tokens_seen": 355598336 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004506619859578736, + "loss": 1.5545, + "theoretical_loss": 4.065557615888029, + "tokens_seen": 355663872 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045065195586760284, + "loss": 1.432, + "theoretical_loss": 4.06547215240855, + "tokens_seen": 355729408 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045064192577733196, + "loss": 1.4662, + "theoretical_loss": 4.065386709080206, + "tokens_seen": 355794944 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004506318956870612, + "loss": 1.5857, + "theoretical_loss": 4.0653012858945345, + "tokens_seen": 355860480 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004506218655967904, + "loss": 1.4817, + "theoretical_loss": 4.065215882843079, + "tokens_seen": 355926016 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045061183550651956, + "loss": 1.632, + "theoretical_loss": 4.065130499917388, + "tokens_seen": 355991552 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045060180541624874, + "loss": 1.5973, + "theoretical_loss": 4.065045137109014, + "tokens_seen": 356057088 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004505917753259779, + "loss": 1.5795, + "theoretical_loss": 4.064959794409518, + "tokens_seen": 356122624 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004505817452357071, + "loss": 1.3617, + "theoretical_loss": 4.064874471810461, + "tokens_seen": 356188160 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045057171514543634, + "loss": 1.4484, + "theoretical_loss": 4.064789169303412, + "tokens_seen": 356253696 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045056168505516547, + "loss": 1.5154, + "theoretical_loss": 4.064703886879947, + "tokens_seen": 356319232 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004505516549648947, + "loss": 1.4312, + "theoretical_loss": 4.064618624531644, + "tokens_seen": 356384768 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004505416248746239, + "loss": 1.5096, + "theoretical_loss": 4.064533382250086, + "tokens_seen": 356450304 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045053159478435306, + "loss": 1.5009, + "theoretical_loss": 4.0644481600268625, + "tokens_seen": 356515840 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045052156469408224, + "loss": 1.4128, + "theoretical_loss": 4.064362957853569, + "tokens_seen": 356581376 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004505115346038114, + "loss": 1.5874, + "theoretical_loss": 4.064277775721804, + "tokens_seen": 356646912 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004505015045135406, + "loss": 1.5192, + "theoretical_loss": 4.06419261362317, + "tokens_seen": 356712448 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045049147442326984, + "loss": 1.5059, + "theoretical_loss": 4.0641074715492795, + "tokens_seen": 356777984 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045048144433299897, + "loss": 1.424, + "theoretical_loss": 4.064022349491745, + "tokens_seen": 356843520 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004504714142427282, + "loss": 1.3844, + "theoretical_loss": 4.063937247442187, + "tokens_seen": 356909056 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045046138415245733, + "loss": 1.5259, + "theoretical_loss": 4.0638521653922295, + "tokens_seen": 356974592 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045045135406218657, + "loss": 1.521, + "theoretical_loss": 4.063767103333502, + "tokens_seen": 357040128 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004504413239719158, + "loss": 1.619, + "theoretical_loss": 4.063682061257638, + "tokens_seen": 357105664 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": 0.00040533297578804195, + "objective/train/docs_used": 214188, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.159883499145508, + "objective/train/original_loss": 3.159883499145508, + "objective/train/theoretical_loss": 4.06359703915628, + "objective/train/tokens_used": 377631200, + "objective/train/value_avg": -0.005458831787109375, + "objective/train/value_loss": 7.526182889705524e-05, + "objective/train/value_max": -0.00025916099548339844, + "objective/train/value_min": -0.06085205078125, + "objective/train/value_reward_corr": 0.30837439666670263, + "objective/train/value_std": 0.004840850830078125, + "objective/train/weight_avg": 1.0075849294662476, + "objective/train/weighted_lm_loss": 3.1832282543182373, + "objective/train/weights_max": 1.692358374595642, + "objective/train/weights_min": 0.406004935503006, + "theoretical_loss": 4.06359703915628, + "tokens_seen": 357171200 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045043129388164493, + "loss": 1.4623, + "theoretical_loss": 4.06359703915628, + "tokens_seen": 357171200 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045042126379137416, + "loss": 1.5509, + "theoretical_loss": 4.06351203702107, + "tokens_seen": 357236736 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045041123370110334, + "loss": 1.5237, + "theoretical_loss": 4.063427054843658, + "tokens_seen": 357302272 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004504012036108325, + "loss": 1.3432, + "theoretical_loss": 4.0633420926157005, + "tokens_seen": 357367808 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004503911735205617, + "loss": 1.4122, + "theoretical_loss": 4.063257150328856, + "tokens_seen": 357433344 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004503811434302909, + "loss": 1.5002, + "theoretical_loss": 4.063172227974789, + "tokens_seen": 357498880 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045037111334002007, + "loss": 1.572, + "theoretical_loss": 4.06308732554517, + "tokens_seen": 357564416 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004503610832497493, + "loss": 1.3487, + "theoretical_loss": 4.063002443031674, + "tokens_seen": 357629952 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045035105315947843, + "loss": 1.6128, + "theoretical_loss": 4.062917580425979, + "tokens_seen": 357695488 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045034102306920767, + "loss": 1.4836, + "theoretical_loss": 4.062832737719772, + "tokens_seen": 357761024 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004503309929789368, + "loss": 1.4456, + "theoretical_loss": 4.062747914904741, + "tokens_seen": 357826560 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045032096288866603, + "loss": 1.5385, + "theoretical_loss": 4.062663111972582, + "tokens_seen": 357892096 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004503109327983952, + "loss": 1.4865, + "theoretical_loss": 4.062578328914994, + "tokens_seen": 357957632 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004503009027081244, + "loss": 1.4391, + "theoretical_loss": 4.062493565723683, + "tokens_seen": 358023168 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045029087261785357, + "loss": 1.4151, + "theoretical_loss": 4.062408822390356, + "tokens_seen": 358088704 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045028084252758275, + "loss": 1.4623, + "theoretical_loss": 4.062324098906732, + "tokens_seen": 358154240 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045027081243731193, + "loss": 1.5254, + "theoretical_loss": 4.062239395264526, + "tokens_seen": 358219776 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045026078234704117, + "loss": 1.5637, + "theoretical_loss": 4.062154711455466, + "tokens_seen": 358285312 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004502507522567703, + "loss": 1.4308, + "theoretical_loss": 4.06207004747128, + "tokens_seen": 358350848 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045024072216649953, + "loss": 1.4725, + "theoretical_loss": 4.061985403303702, + "tokens_seen": 358416384 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004502306920762287, + "loss": 1.589, + "theoretical_loss": 4.061900778944473, + "tokens_seen": 358481920 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004502206619859579, + "loss": 1.5644, + "theoretical_loss": 4.061816174385338, + "tokens_seen": 358547456 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004502106318956871, + "loss": 1.5285, + "theoretical_loss": 4.061731589618044, + "tokens_seen": 358612992 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045020060180541626, + "loss": 1.4881, + "theoretical_loss": 4.061647024634347, + "tokens_seen": 358678528 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045019057171514544, + "loss": 1.4645, + "theoretical_loss": 4.061562479426007, + "tokens_seen": 358744064 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0012930217199027538, + "objective/train/docs_used": 215284, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9867944717407227, + "objective/train/original_loss": 2.9867944717407227, + "objective/train/theoretical_loss": 4.061477953984786, + "objective/train/tokens_used": 379269600, + "objective/train/value_avg": -0.0061187744140625, + "objective/train/value_loss": 8.005131530808285e-05, + "objective/train/value_max": -0.00042557716369628906, + "objective/train/value_min": -0.0738525390625, + "objective/train/value_reward_corr": 0.4047717407946021, + "objective/train/value_std": 0.00591278076171875, + "objective/train/weight_avg": 0.9908571243286133, + "objective/train/weighted_lm_loss": 2.969407320022583, + "objective/train/weights_max": 1.8631640672683716, + "objective/train/weights_min": 0.3614153563976288, + "theoretical_loss": 4.061477953984786, + "tokens_seen": 358809600 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004501805416248747, + "loss": 1.4785, + "theoretical_loss": 4.061477953984786, + "tokens_seen": 358809600 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004501705115346038, + "loss": 1.5251, + "theoretical_loss": 4.0613934483024545, + "tokens_seen": 358875136 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045016048144433304, + "loss": 1.5417, + "theoretical_loss": 4.061308962370787, + "tokens_seen": 358940672 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045015045135406216, + "loss": 1.4955, + "theoretical_loss": 4.061224496181562, + "tokens_seen": 359006208 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004501404212637914, + "loss": 1.4468, + "theoretical_loss": 4.061140049726563, + "tokens_seen": 359071744 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004501303911735206, + "loss": 1.3733, + "theoretical_loss": 4.06105562299758, + "tokens_seen": 359137280 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045012036108324976, + "loss": 1.4786, + "theoretical_loss": 4.060971215986406, + "tokens_seen": 359202816 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045011033099297894, + "loss": 1.5303, + "theoretical_loss": 4.06088682868484, + "tokens_seen": 359268352 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004501003009027081, + "loss": 1.5482, + "theoretical_loss": 4.060802461084685, + "tokens_seen": 359333888 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004500902708124373, + "loss": 1.4366, + "theoretical_loss": 4.06071811317775, + "tokens_seen": 359399424 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045008024072216654, + "loss": 1.4308, + "theoretical_loss": 4.060633784955849, + "tokens_seen": 359464960 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045007021063189567, + "loss": 1.4151, + "theoretical_loss": 4.060549476410799, + "tokens_seen": 359530496 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004500601805416249, + "loss": 1.4612, + "theoretical_loss": 4.060465187534424, + "tokens_seen": 359596032 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004500501504513541, + "loss": 1.5057, + "theoretical_loss": 4.060380918318552, + "tokens_seen": 359661568 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045004012036108326, + "loss": 1.4042, + "theoretical_loss": 4.060296668755016, + "tokens_seen": 359727104 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045003009027081244, + "loss": 1.4611, + "theoretical_loss": 4.060212438835654, + "tokens_seen": 359792640 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004500200601805416, + "loss": 1.5815, + "theoretical_loss": 4.0601282285523075, + "tokens_seen": 359858176 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004500100300902708, + "loss": 1.3647, + "theoretical_loss": 4.060044037896826, + "tokens_seen": 359923712 + }, + { + "epoch": 0.11, + "learning_rate": 0.00045000000000000004, + "loss": 1.5364, + "theoretical_loss": 4.05995986686106, + "tokens_seen": 359989248 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044998996990972917, + "loss": 1.4981, + "theoretical_loss": 4.059875715436869, + "tokens_seen": 360054784 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004499799398194584, + "loss": 1.4453, + "theoretical_loss": 4.0597915836161125, + "tokens_seen": 360120320 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044996990972918753, + "loss": 1.4668, + "theoretical_loss": 4.059707471390659, + "tokens_seen": 360185856 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044995987963891677, + "loss": 1.5716, + "theoretical_loss": 4.0596233787523825, + "tokens_seen": 360251392 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044994984954864595, + "loss": 1.6304, + "theoretical_loss": 4.059539305693156, + "tokens_seen": 360316928 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044993981945837513, + "loss": 1.4807, + "theoretical_loss": 4.059455252204863, + "tokens_seen": 360382464 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0013121353695169091, + "objective/train/docs_used": 215704, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.6058855056762695, + "objective/train/original_loss": 3.6058857440948486, + "objective/train/theoretical_loss": 4.05937121827939, + "objective/train/tokens_used": 380908000, + "objective/train/value_avg": -0.00559234619140625, + "objective/train/value_loss": 7.572945469291881e-05, + "objective/train/value_max": -0.0004897117614746094, + "objective/train/value_min": -0.09417724609375, + "objective/train/value_reward_corr": 0.5930943612756638, + "objective/train/value_std": 0.00576019287109375, + "objective/train/weight_avg": 0.9904159307479858, + "objective/train/weighted_lm_loss": 3.5757949352264404, + "objective/train/weights_max": 1.7457025051116943, + "objective/train/weights_min": 0.4964386820793152, + "theoretical_loss": 4.05937121827939, + "tokens_seen": 360448000 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004499297893681043, + "loss": 1.5239, + "theoretical_loss": 4.05937121827939, + "tokens_seen": 360448000 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044991975927783355, + "loss": 1.5613, + "theoretical_loss": 4.059287203908627, + "tokens_seen": 360513536 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044990972918756267, + "loss": 1.5053, + "theoretical_loss": 4.0592032090844725, + "tokens_seen": 360579072 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004498996990972919, + "loss": 1.5603, + "theoretical_loss": 4.059119233798825, + "tokens_seen": 360644608 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044988966900702103, + "loss": 1.4307, + "theoretical_loss": 4.059035278043591, + "tokens_seen": 360710144 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044987963891675027, + "loss": 1.5475, + "theoretical_loss": 4.0589513418106815, + "tokens_seen": 360775680 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044986960882647945, + "loss": 1.5595, + "theoretical_loss": 4.0588674250920125, + "tokens_seen": 360841216 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044985957873620863, + "loss": 1.4516, + "theoretical_loss": 4.058783527879502, + "tokens_seen": 360906752 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004498495486459378, + "loss": 1.5747, + "theoretical_loss": 4.058699650165078, + "tokens_seen": 360972288 + }, + { + "epoch": 0.11, + "learning_rate": 0.000449839518555667, + "loss": 1.454, + "theoretical_loss": 4.058615791940667, + "tokens_seen": 361037824 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004498294884653962, + "loss": 1.5073, + "theoretical_loss": 4.058531953198207, + "tokens_seen": 361103360 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004498194583751254, + "loss": 1.5277, + "theoretical_loss": 4.058448133929635, + "tokens_seen": 361168896 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044980942828485454, + "loss": 1.4508, + "theoretical_loss": 4.058364334126896, + "tokens_seen": 361234432 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044979939819458377, + "loss": 1.5973, + "theoretical_loss": 4.058280553781939, + "tokens_seen": 361299968 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004497893681043129, + "loss": 1.5685, + "theoretical_loss": 4.058196792886719, + "tokens_seen": 361365504 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044977933801404213, + "loss": 1.5318, + "theoretical_loss": 4.0581130514331925, + "tokens_seen": 361431040 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004497693079237713, + "loss": 1.5078, + "theoretical_loss": 4.0580293294133245, + "tokens_seen": 361496576 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004497592778335005, + "loss": 1.4646, + "theoretical_loss": 4.057945626819082, + "tokens_seen": 361562112 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004497492477432297, + "loss": 1.4616, + "theoretical_loss": 4.05786194364244, + "tokens_seen": 361627648 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004497392176529589, + "loss": 1.467, + "theoretical_loss": 4.057778279875373, + "tokens_seen": 361693184 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044972918756268804, + "loss": 1.4684, + "theoretical_loss": 4.057694635509866, + "tokens_seen": 361758720 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004497191574724173, + "loss": 1.4044, + "theoretical_loss": 4.057611010537905, + "tokens_seen": 361824256 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004497091273821464, + "loss": 1.555, + "theoretical_loss": 4.057527404951483, + "tokens_seen": 361889792 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044969909729187564, + "loss": 1.45, + "theoretical_loss": 4.0574438187425965, + "tokens_seen": 361955328 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004496890672016049, + "loss": 1.496, + "theoretical_loss": 4.057360251903246, + "tokens_seen": 362020864 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0017414262983947992, + "objective/train/docs_used": 216313, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1330654621124268, + "objective/train/original_loss": 3.1330654621124268, + "objective/train/theoretical_loss": 4.057276704425439, + "objective/train/tokens_used": 382546400, + "objective/train/value_avg": -0.0055694580078125, + "objective/train/value_loss": 9.4661969342269e-05, + "objective/train/value_max": -0.0008039474487304688, + "objective/train/value_min": -0.06842041015625, + "objective/train/value_reward_corr": 0.36768801806134715, + "objective/train/value_std": 0.004451751708984375, + "objective/train/weight_avg": 0.9870510101318359, + "objective/train/weighted_lm_loss": 3.1005539894104004, + "objective/train/weights_max": 1.9821935892105103, + "objective/train/weights_min": 0.5004755258560181, + "theoretical_loss": 4.057276704425439, + "tokens_seen": 362086400 + }, + { + "epoch": 0.11, + "learning_rate": 0.000449679037111334, + "loss": 1.4605, + "theoretical_loss": 4.057276704425439, + "tokens_seen": 362086400 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044966900702106324, + "loss": 1.5677, + "theoretical_loss": 4.057193176301185, + "tokens_seen": 362151936 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044965897693079236, + "loss": 1.4937, + "theoretical_loss": 4.057109667522502, + "tokens_seen": 362217472 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004496489468405216, + "loss": 1.4503, + "theoretical_loss": 4.057026178081408, + "tokens_seen": 362283008 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004496389167502508, + "loss": 1.408, + "theoretical_loss": 4.056942707969931, + "tokens_seen": 362348544 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044962888665997996, + "loss": 1.495, + "theoretical_loss": 4.056859257180098, + "tokens_seen": 362414080 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044961885656970914, + "loss": 1.4703, + "theoretical_loss": 4.056775825703945, + "tokens_seen": 362479616 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004496088264794383, + "loss": 1.616, + "theoretical_loss": 4.056692413533512, + "tokens_seen": 362545152 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004495987963891675, + "loss": 1.6603, + "theoretical_loss": 4.056609020660843, + "tokens_seen": 362610688 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044958876629889674, + "loss": 1.4066, + "theoretical_loss": 4.0565256470779865, + "tokens_seen": 362676224 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044957873620862587, + "loss": 1.5918, + "theoretical_loss": 4.056442292776994, + "tokens_seen": 362741760 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004495687061183551, + "loss": 1.4395, + "theoretical_loss": 4.056358957749928, + "tokens_seen": 362807296 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004495586760280843, + "loss": 1.4368, + "theoretical_loss": 4.056275641988848, + "tokens_seen": 362872832 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044954864593781346, + "loss": 1.4592, + "theoretical_loss": 4.056192345485823, + "tokens_seen": 362938368 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044953861584754264, + "loss": 1.5054, + "theoretical_loss": 4.056109068232925, + "tokens_seen": 363003904 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004495285857572718, + "loss": 1.5162, + "theoretical_loss": 4.056025810222232, + "tokens_seen": 363069440 + }, + { + "epoch": 0.11, + "learning_rate": 0.000449518555667001, + "loss": 1.5343, + "theoretical_loss": 4.055942571445824, + "tokens_seen": 363134976 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044950852557673024, + "loss": 1.4911, + "theoretical_loss": 4.055859351895789, + "tokens_seen": 363200512 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044949849548645937, + "loss": 1.6824, + "theoretical_loss": 4.055776151564217, + "tokens_seen": 363266048 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004494884653961886, + "loss": 1.4438, + "theoretical_loss": 4.055692970443205, + "tokens_seen": 363331584 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044947843530591773, + "loss": 1.5491, + "theoretical_loss": 4.055609808524853, + "tokens_seen": 363397120 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044946840521564697, + "loss": 1.559, + "theoretical_loss": 4.055526665801265, + "tokens_seen": 363462656 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044945837512537615, + "loss": 1.6053, + "theoretical_loss": 4.055443542264552, + "tokens_seen": 363528192 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044944834503510533, + "loss": 1.5387, + "theoretical_loss": 4.0553604379068275, + "tokens_seen": 363593728 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004494383149448345, + "loss": 1.5051, + "theoretical_loss": 4.055277352720212, + "tokens_seen": 363659264 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0004273501690477133, + "objective/train/docs_used": 217479, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0255792140960693, + "objective/train/original_loss": 3.0255792140960693, + "objective/train/theoretical_loss": 4.055194286696828, + "objective/train/tokens_used": 384184800, + "objective/train/value_avg": -0.004913330078125, + "objective/train/value_loss": 6.548959936480969e-05, + "objective/train/value_max": -0.0005211830139160156, + "objective/train/value_min": -0.06695556640625, + "objective/train/value_reward_corr": 0.21474422809428625, + "objective/train/value_std": 0.00347900390625, + "objective/train/weight_avg": 0.9986748099327087, + "objective/train/weighted_lm_loss": 3.0275354385375977, + "objective/train/weights_max": 1.848397135734558, + "objective/train/weights_min": 0.19782064855098724, + "theoretical_loss": 4.055194286696828, + "tokens_seen": 363724800 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044942828485456375, + "loss": 1.451, + "theoretical_loss": 4.055194286696828, + "tokens_seen": 363724800 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044941825476429287, + "loss": 1.4514, + "theoretical_loss": 4.055111239828806, + "tokens_seen": 363790336 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004494082246740221, + "loss": 1.4498, + "theoretical_loss": 4.055028212108276, + "tokens_seen": 363855872 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044939819458375123, + "loss": 1.5172, + "theoretical_loss": 4.054945203527378, + "tokens_seen": 363921408 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044938816449348047, + "loss": 1.514, + "theoretical_loss": 4.054862214078254, + "tokens_seen": 363986944 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044937813440320965, + "loss": 1.5407, + "theoretical_loss": 4.0547792437530505, + "tokens_seen": 364052480 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044936810431293883, + "loss": 1.4476, + "theoretical_loss": 4.05469629254392, + "tokens_seen": 364118016 + }, + { + "epoch": 0.11, + "learning_rate": 0.000449358074222668, + "loss": 1.5403, + "theoretical_loss": 4.0546133604430175, + "tokens_seen": 364183552 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004493480441323972, + "loss": 1.5336, + "theoretical_loss": 4.054530447442505, + "tokens_seen": 364249088 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004493380140421264, + "loss": 1.5956, + "theoretical_loss": 4.054447553534548, + "tokens_seen": 364314624 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004493279839518556, + "loss": 1.5046, + "theoretical_loss": 4.054364678711315, + "tokens_seen": 364380160 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044931795386158474, + "loss": 1.5086, + "theoretical_loss": 4.054281822964984, + "tokens_seen": 364445696 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044930792377131397, + "loss": 1.5796, + "theoretical_loss": 4.054198986287732, + "tokens_seen": 364511232 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004492978936810431, + "loss": 1.5215, + "theoretical_loss": 4.054116168671743, + "tokens_seen": 364576768 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044928786359077234, + "loss": 1.4886, + "theoretical_loss": 4.054033370109206, + "tokens_seen": 364642304 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004492778335005015, + "loss": 1.5306, + "theoretical_loss": 4.053950590592316, + "tokens_seen": 364707840 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004492678034102307, + "loss": 1.6908, + "theoretical_loss": 4.053867830113268, + "tokens_seen": 364773376 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004492577733199599, + "loss": 1.5755, + "theoretical_loss": 4.053785088664266, + "tokens_seen": 364838912 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004492477432296891, + "loss": 1.4914, + "theoretical_loss": 4.053702366237517, + "tokens_seen": 364904448 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044923771313941824, + "loss": 1.4266, + "theoretical_loss": 4.053619662825232, + "tokens_seen": 364969984 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004492276830491475, + "loss": 1.5039, + "theoretical_loss": 4.053536978419627, + "tokens_seen": 365035520 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004492176529588766, + "loss": 1.499, + "theoretical_loss": 4.053454313012924, + "tokens_seen": 365101056 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044920762286860584, + "loss": 1.5069, + "theoretical_loss": 4.053371666597348, + "tokens_seen": 365166592 + }, + { + "epoch": 0.11, + "learning_rate": 0.000449197592778335, + "loss": 1.6308, + "theoretical_loss": 4.053289039165128, + "tokens_seen": 365232128 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004491875626880642, + "loss": 1.5585, + "theoretical_loss": 4.0532064307085, + "tokens_seen": 365297664 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0007260242127813399, + "objective/train/docs_used": 218231, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.686818838119507, + "objective/train/original_loss": 2.686818838119507, + "objective/train/theoretical_loss": 4.053123841219701, + "objective/train/tokens_used": 385823200, + "objective/train/value_avg": -0.00518035888671875, + "objective/train/value_loss": 6.986282824072987e-05, + "objective/train/value_max": -0.00036835670471191406, + "objective/train/value_min": -0.07464599609375, + "objective/train/value_reward_corr": 0.43507638535153015, + "objective/train/value_std": 0.004467010498046875, + "objective/train/weight_avg": 0.9957458972930908, + "objective/train/weighted_lm_loss": 2.6604411602020264, + "objective/train/weights_max": 1.5952308177947998, + "objective/train/weights_min": 0.13883088529109955, + "theoretical_loss": 4.053123841219701, + "tokens_seen": 365363200 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004491775325977934, + "loss": 1.4453, + "theoretical_loss": 4.053123841219701, + "tokens_seen": 365363200 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044916750250752256, + "loss": 1.5801, + "theoretical_loss": 4.053041270690977, + "tokens_seen": 365428736 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044915747241725174, + "loss": 1.5567, + "theoretical_loss": 4.052958719114574, + "tokens_seen": 365494272 + }, + { + "epoch": 0.11, + "learning_rate": 0.000449147442326981, + "loss": 1.59, + "theoretical_loss": 4.052876186482747, + "tokens_seen": 365559808 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004491374122367101, + "loss": 1.5003, + "theoretical_loss": 4.052793672787751, + "tokens_seen": 365625344 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044912738214643934, + "loss": 1.6109, + "theoretical_loss": 4.05271117802185, + "tokens_seen": 365690880 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044911735205616847, + "loss": 1.5284, + "theoretical_loss": 4.052628702177309, + "tokens_seen": 365756416 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004491073219658977, + "loss": 1.5659, + "theoretical_loss": 4.052546245246399, + "tokens_seen": 365821952 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004490972918756269, + "loss": 1.3922, + "theoretical_loss": 4.052463807221396, + "tokens_seen": 365887488 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044908726178535607, + "loss": 1.4393, + "theoretical_loss": 4.05238138809458, + "tokens_seen": 365953024 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044907723169508525, + "loss": 1.5409, + "theoretical_loss": 4.052298987858236, + "tokens_seen": 366018560 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004490672016048145, + "loss": 1.5367, + "theoretical_loss": 4.052216606504652, + "tokens_seen": 366084096 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004490571715145436, + "loss": 1.5216, + "theoretical_loss": 4.0521342440261225, + "tokens_seen": 366149632 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044904714142427284, + "loss": 1.4905, + "theoretical_loss": 4.052051900414945, + "tokens_seen": 366215168 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044903711133400197, + "loss": 1.4472, + "theoretical_loss": 4.051969575663423, + "tokens_seen": 366280704 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004490270812437312, + "loss": 1.4849, + "theoretical_loss": 4.0518872697638635, + "tokens_seen": 366346240 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004490170511534604, + "loss": 1.5739, + "theoretical_loss": 4.051804982708577, + "tokens_seen": 366411776 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044900702106318957, + "loss": 1.5365, + "theoretical_loss": 4.051722714489882, + "tokens_seen": 366477312 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044899699097291875, + "loss": 1.4514, + "theoretical_loss": 4.051640465100097, + "tokens_seen": 366542848 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044898696088264793, + "loss": 1.4781, + "theoretical_loss": 4.051558234531548, + "tokens_seen": 366608384 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004489769307923771, + "loss": 1.5855, + "theoretical_loss": 4.051476022776566, + "tokens_seen": 366673920 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044896690070210635, + "loss": 1.651, + "theoretical_loss": 4.051393829827483, + "tokens_seen": 366739456 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004489568706118355, + "loss": 1.4392, + "theoretical_loss": 4.051311655676639, + "tokens_seen": 366804992 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004489468405215647, + "loss": 1.4174, + "theoretical_loss": 4.051229500316378, + "tokens_seen": 366870528 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044893681043129395, + "loss": 1.4921, + "theoretical_loss": 4.051147363739045, + "tokens_seen": 366936064 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0005152435624040663, + "objective/train/docs_used": 219467, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0141515731811523, + "objective/train/original_loss": 3.0141518115997314, + "objective/train/theoretical_loss": 4.051065245936996, + "objective/train/tokens_used": 387461600, + "objective/train/value_avg": -0.004787445068359375, + "objective/train/value_loss": 5.696868902305141e-05, + "objective/train/value_max": -0.0003612041473388672, + "objective/train/value_min": -0.085693359375, + "objective/train/value_reward_corr": 0.3046837186335934, + "objective/train/value_std": 0.004238128662109375, + "objective/train/weight_avg": 0.9974380731582642, + "objective/train/weighted_lm_loss": 3.00386905670166, + "objective/train/weights_max": 2.3559253215789795, + "objective/train/weights_min": 0.29524219036102295, + "theoretical_loss": 4.051065245936996, + "tokens_seen": 367001600 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044892678034102307, + "loss": 1.4516, + "theoretical_loss": 4.051065245936996, + "tokens_seen": 367001600 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004489167502507523, + "loss": 1.5186, + "theoretical_loss": 4.050983146902584, + "tokens_seen": 367067136 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044890672016048143, + "loss": 1.5853, + "theoretical_loss": 4.050901066628172, + "tokens_seen": 367132672 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044889669007021067, + "loss": 1.517, + "theoretical_loss": 4.050819005106125, + "tokens_seen": 367198208 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044888665997993985, + "loss": 1.4797, + "theoretical_loss": 4.050736962328814, + "tokens_seen": 367263744 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044887662988966903, + "loss": 1.4878, + "theoretical_loss": 4.050654938288613, + "tokens_seen": 367329280 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004488665997993982, + "loss": 1.3863, + "theoretical_loss": 4.0505729329779, + "tokens_seen": 367394816 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004488565697091274, + "loss": 1.3939, + "theoretical_loss": 4.0504909463890595, + "tokens_seen": 367460352 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004488465396188566, + "loss": 1.5889, + "theoretical_loss": 4.050408978514479, + "tokens_seen": 367525888 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004488365095285858, + "loss": 1.584, + "theoretical_loss": 4.050327029346551, + "tokens_seen": 367591424 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044882647943831494, + "loss": 1.539, + "theoretical_loss": 4.0502450988776735, + "tokens_seen": 367656960 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004488164493480442, + "loss": 1.4864, + "theoretical_loss": 4.050163187100246, + "tokens_seen": 367722496 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004488064192577733, + "loss": 1.4886, + "theoretical_loss": 4.050081294006675, + "tokens_seen": 367788032 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044879638916750254, + "loss": 1.4033, + "theoretical_loss": 4.04999941958937, + "tokens_seen": 367853568 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004487863590772317, + "loss": 1.609, + "theoretical_loss": 4.049917563840747, + "tokens_seen": 367919104 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004487763289869609, + "loss": 1.4746, + "theoretical_loss": 4.049835726753223, + "tokens_seen": 367984640 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004487662988966901, + "loss": 1.4455, + "theoretical_loss": 4.049753908319223, + "tokens_seen": 368050176 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004487562688064193, + "loss": 1.5075, + "theoretical_loss": 4.049672108531174, + "tokens_seen": 368115712 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044874623871614844, + "loss": 1.5232, + "theoretical_loss": 4.049590327381509, + "tokens_seen": 368181248 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004487362086258777, + "loss": 1.4966, + "theoretical_loss": 4.049508564862664, + "tokens_seen": 368246784 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004487261785356068, + "loss": 1.5322, + "theoretical_loss": 4.04942682096708, + "tokens_seen": 368312320 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044871614844533604, + "loss": 1.5634, + "theoretical_loss": 4.049345095687203, + "tokens_seen": 368377856 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004487061183550652, + "loss": 1.4738, + "theoretical_loss": 4.049263389015484, + "tokens_seen": 368443392 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004486960882647944, + "loss": 1.5711, + "theoretical_loss": 4.0491817009443745, + "tokens_seen": 368508928 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004486860581745236, + "loss": 1.3896, + "theoretical_loss": 4.0491000314663355, + "tokens_seen": 368574464 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0006319410749711096, + "objective/train/docs_used": 220098, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2455763816833496, + "objective/train/original_loss": 3.2455761432647705, + "objective/train/theoretical_loss": 4.04901838057383, + "objective/train/tokens_used": 389100000, + "objective/train/value_avg": -0.004428863525390625, + "objective/train/value_loss": 6.680423393845558e-05, + "objective/train/value_max": -0.0004305839538574219, + "objective/train/value_min": -0.06939697265625, + "objective/train/value_reward_corr": 0.26511367581237183, + "objective/train/value_std": 0.0031490325927734375, + "objective/train/weight_avg": 0.996660590171814, + "objective/train/weighted_lm_loss": 3.2451119422912598, + "objective/train/weights_max": 2.001645803451538, + "objective/train/weights_min": 0.1951858401298523, + "theoretical_loss": 4.04901838057383, + "tokens_seen": 368640000 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044867602808425276, + "loss": 1.5615, + "theoretical_loss": 4.04901838057383, + "tokens_seen": 368640000 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044866599799398194, + "loss": 1.4846, + "theoretical_loss": 4.048936748259324, + "tokens_seen": 368705536 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004486559679037112, + "loss": 1.4851, + "theoretical_loss": 4.048855134515291, + "tokens_seen": 368771072 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004486459378134403, + "loss": 1.5448, + "theoretical_loss": 4.048773539334207, + "tokens_seen": 368836608 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044863590772316954, + "loss": 1.6089, + "theoretical_loss": 4.048691962708553, + "tokens_seen": 368902144 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044862587763289867, + "loss": 1.593, + "theoretical_loss": 4.0486104046308125, + "tokens_seen": 368967680 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004486158475426279, + "loss": 1.5608, + "theoretical_loss": 4.048528865093476, + "tokens_seen": 369033216 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004486058174523571, + "loss": 1.4492, + "theoretical_loss": 4.048447344089038, + "tokens_seen": 369098752 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044859578736208627, + "loss": 1.4791, + "theoretical_loss": 4.048365841609995, + "tokens_seen": 369164288 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044858575727181545, + "loss": 1.585, + "theoretical_loss": 4.04828435764885, + "tokens_seen": 369229824 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004485757271815447, + "loss": 1.3924, + "theoretical_loss": 4.048202892198112, + "tokens_seen": 369295360 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004485656970912738, + "loss": 1.5489, + "theoretical_loss": 4.048121445250289, + "tokens_seen": 369360896 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044855566700100304, + "loss": 1.5241, + "theoretical_loss": 4.048040016797899, + "tokens_seen": 369426432 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044854563691073217, + "loss": 1.5533, + "theoretical_loss": 4.047958606833462, + "tokens_seen": 369491968 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004485356068204614, + "loss": 1.4953, + "theoretical_loss": 4.0478772153495015, + "tokens_seen": 369557504 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004485255767301906, + "loss": 1.4899, + "theoretical_loss": 4.047795842338545, + "tokens_seen": 369623040 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044851554663991977, + "loss": 1.5512, + "theoretical_loss": 4.047714487793129, + "tokens_seen": 369688576 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044850551654964895, + "loss": 1.4775, + "theoretical_loss": 4.047633151705787, + "tokens_seen": 369754112 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044849548645937813, + "loss": 1.5169, + "theoretical_loss": 4.047551834069064, + "tokens_seen": 369819648 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004484854563691073, + "loss": 1.6405, + "theoretical_loss": 4.047470534875504, + "tokens_seen": 369885184 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044847542627883655, + "loss": 1.4735, + "theoretical_loss": 4.047389254117657, + "tokens_seen": 369950720 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004484653961885657, + "loss": 1.5275, + "theoretical_loss": 4.047307991788079, + "tokens_seen": 370016256 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004484553660982949, + "loss": 1.4746, + "theoretical_loss": 4.047226747879329, + "tokens_seen": 370081792 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044844533600802404, + "loss": 1.52, + "theoretical_loss": 4.04714552238397, + "tokens_seen": 370147328 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044843530591775327, + "loss": 1.5002, + "theoretical_loss": 4.04706431529457, + "tokens_seen": 370212864 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0010755874682217836, + "objective/train/docs_used": 221122, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.722578525543213, + "objective/train/original_loss": 2.722578287124634, + "objective/train/theoretical_loss": 4.0469831266037, + "objective/train/tokens_used": 390738400, + "objective/train/value_avg": -0.005161285400390625, + "objective/train/value_loss": 5.969602716504596e-05, + "objective/train/value_max": -0.00047469139099121094, + "objective/train/value_min": -0.09124755859375, + "objective/train/value_reward_corr": 0.35998551538548074, + "objective/train/value_std": 0.00464630126953125, + "objective/train/weight_avg": 0.9921076893806458, + "objective/train/weighted_lm_loss": 2.722787618637085, + "objective/train/weights_max": 1.9970574378967285, + "objective/train/weights_min": 0.38522857427597046, + "theoretical_loss": 4.0469831266037, + "tokens_seen": 370278400 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044842527582748245, + "loss": 1.5311, + "theoretical_loss": 4.0469831266037, + "tokens_seen": 370278400 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044841524573721163, + "loss": 1.5231, + "theoretical_loss": 4.0469019563039375, + "tokens_seen": 370343936 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004484052156469408, + "loss": 1.4987, + "theoretical_loss": 4.046820804387862, + "tokens_seen": 370409472 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044839518555667005, + "loss": 1.4937, + "theoretical_loss": 4.046739670848059, + "tokens_seen": 370475008 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004483851554663992, + "loss": 1.5533, + "theoretical_loss": 4.046658555677118, + "tokens_seen": 370540544 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004483751253761284, + "loss": 1.4681, + "theoretical_loss": 4.046577458867633, + "tokens_seen": 370606080 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044836509528585754, + "loss": 1.5785, + "theoretical_loss": 4.046496380412201, + "tokens_seen": 370671616 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004483550651955868, + "loss": 1.4792, + "theoretical_loss": 4.0464153203034225, + "tokens_seen": 370737152 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044834503510531596, + "loss": 1.4251, + "theoretical_loss": 4.046334278533907, + "tokens_seen": 370802688 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044833500501504514, + "loss": 1.5814, + "theoretical_loss": 4.046253255096263, + "tokens_seen": 370868224 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004483249749247743, + "loss": 1.4848, + "theoretical_loss": 4.0461722499831065, + "tokens_seen": 370933760 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004483149448345035, + "loss": 1.4853, + "theoretical_loss": 4.046091263187057, + "tokens_seen": 370999296 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004483049147442327, + "loss": 1.4488, + "theoretical_loss": 4.046010294700737, + "tokens_seen": 371064832 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004482948846539619, + "loss": 1.4217, + "theoretical_loss": 4.045929344516774, + "tokens_seen": 371130368 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044828485456369104, + "loss": 1.6089, + "theoretical_loss": 4.045848412627802, + "tokens_seen": 371195904 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004482748244734203, + "loss": 1.4527, + "theoretical_loss": 4.045767499026455, + "tokens_seen": 371261440 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004482647943831494, + "loss": 1.4636, + "theoretical_loss": 4.045686603705375, + "tokens_seen": 371326976 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044825476429287864, + "loss": 1.4509, + "theoretical_loss": 4.045605726657207, + "tokens_seen": 371392512 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004482447342026078, + "loss": 1.5756, + "theoretical_loss": 4.045524867874599, + "tokens_seen": 371458048 + }, + { + "epoch": 0.11, + "learning_rate": 0.000448234704112337, + "loss": 1.6529, + "theoretical_loss": 4.0454440273502055, + "tokens_seen": 371523584 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004482246740220662, + "loss": 1.5305, + "theoretical_loss": 4.045363205076683, + "tokens_seen": 371589120 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004482146439317954, + "loss": 1.6061, + "theoretical_loss": 4.045282401046695, + "tokens_seen": 371654656 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044820461384152455, + "loss": 1.5595, + "theoretical_loss": 4.045201615252906, + "tokens_seen": 371720192 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004481945837512538, + "loss": 1.4736, + "theoretical_loss": 4.045120847687986, + "tokens_seen": 371785728 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044818455366098296, + "loss": 1.6165, + "theoretical_loss": 4.045040098344612, + "tokens_seen": 371851264 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0005906596779823303, + "objective/train/docs_used": 221664, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.125898838043213, + "objective/train/original_loss": 3.125898838043213, + "objective/train/theoretical_loss": 4.04495936721546, + "objective/train/tokens_used": 392376800, + "objective/train/value_avg": -0.004016876220703125, + "objective/train/value_loss": 6.58667559036985e-05, + "objective/train/value_max": -0.00030303001403808594, + "objective/train/value_min": -0.044677734375, + "objective/train/value_reward_corr": 0.2498620108209604, + "objective/train/value_std": 0.00301361083984375, + "objective/train/weight_avg": 0.9970210790634155, + "objective/train/weighted_lm_loss": 3.12341570854187, + "objective/train/weights_max": 1.4126380681991577, + "objective/train/weights_min": 0.37181925773620605, + "theoretical_loss": 4.04495936721546, + "tokens_seen": 371916800 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044817452357071214, + "loss": 1.5366, + "theoretical_loss": 4.04495936721546, + "tokens_seen": 371916800 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004481644934804414, + "loss": 1.4839, + "theoretical_loss": 4.044878654293215, + "tokens_seen": 371982336 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004481544633901705, + "loss": 1.6058, + "theoretical_loss": 4.044797959570563, + "tokens_seen": 372047872 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044814443329989974, + "loss": 1.5599, + "theoretical_loss": 4.044717283040197, + "tokens_seen": 372113408 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044813440320962887, + "loss": 1.5441, + "theoretical_loss": 4.044636624694812, + "tokens_seen": 372178944 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004481243731193581, + "loss": 1.5686, + "theoretical_loss": 4.044555984527107, + "tokens_seen": 372244480 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004481143430290873, + "loss": 1.6089, + "theoretical_loss": 4.044475362529786, + "tokens_seen": 372310016 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044810431293881647, + "loss": 1.5379, + "theoretical_loss": 4.044394758695559, + "tokens_seen": 372375552 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044809428284854565, + "loss": 1.4959, + "theoretical_loss": 4.044314173017138, + "tokens_seen": 372441088 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004480842527582749, + "loss": 1.5575, + "theoretical_loss": 4.04423360548724, + "tokens_seen": 372506624 + }, + { + "epoch": 0.11, + "learning_rate": 0.000448074222668004, + "loss": 1.5403, + "theoretical_loss": 4.044153056098585, + "tokens_seen": 372572160 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044806419257773324, + "loss": 1.4993, + "theoretical_loss": 4.044072524843899, + "tokens_seen": 372637696 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044805416248746237, + "loss": 1.5199, + "theoretical_loss": 4.04399201171591, + "tokens_seen": 372703232 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004480441323971916, + "loss": 1.5525, + "theoretical_loss": 4.043911516707354, + "tokens_seen": 372768768 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004480341023069208, + "loss": 1.5537, + "theoretical_loss": 4.043831039810967, + "tokens_seen": 372834304 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044802407221664997, + "loss": 1.4713, + "theoretical_loss": 4.043750581019491, + "tokens_seen": 372899840 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044801404212637915, + "loss": 1.4359, + "theoretical_loss": 4.043670140325673, + "tokens_seen": 372965376 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044800401203610833, + "loss": 1.5137, + "theoretical_loss": 4.043589717722263, + "tokens_seen": 373030912 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004479939819458375, + "loss": 1.5934, + "theoretical_loss": 4.043509313202014, + "tokens_seen": 373096448 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044798395185556675, + "loss": 1.4563, + "theoretical_loss": 4.043428926757687, + "tokens_seen": 373161984 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004479739217652959, + "loss": 1.5305, + "theoretical_loss": 4.043348558382044, + "tokens_seen": 373227520 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004479638916750251, + "loss": 1.5462, + "theoretical_loss": 4.04326820806785, + "tokens_seen": 373293056 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044795386158475424, + "loss": 1.5387, + "theoretical_loss": 4.043187875807879, + "tokens_seen": 373358592 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044794383149448347, + "loss": 1.4315, + "theoretical_loss": 4.043107561594906, + "tokens_seen": 373424128 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044793380140421265, + "loss": 1.3302, + "theoretical_loss": 4.043027265421709, + "tokens_seen": 373489664 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": 0.0003906151105184108, + "objective/train/docs_used": 222906, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.783755302429199, + "objective/train/original_loss": 2.7837555408477783, + "objective/train/theoretical_loss": 4.042946987281072, + "objective/train/tokens_used": 394015200, + "objective/train/value_avg": -0.005092620849609375, + "objective/train/value_loss": 5.987756230751984e-05, + "objective/train/value_max": -0.0004093647003173828, + "objective/train/value_min": -0.0489501953125, + "objective/train/value_reward_corr": 0.3358164932041251, + "objective/train/value_std": 0.004314422607421875, + "objective/train/weight_avg": 1.006726622581482, + "objective/train/weighted_lm_loss": 2.8180081844329834, + "objective/train/weights_max": 1.523364543914795, + "objective/train/weights_min": 0.4101358652114868, + "theoretical_loss": 4.042946987281072, + "tokens_seen": 373555200 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044792377131394183, + "loss": 1.5241, + "theoretical_loss": 4.042946987281072, + "tokens_seen": 373555200 + }, + { + "epoch": 0.11, + "learning_rate": 0.000447913741223671, + "loss": 1.5094, + "theoretical_loss": 4.042866727165784, + "tokens_seen": 373620736 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044790371113340025, + "loss": 1.4172, + "theoretical_loss": 4.0427864850686355, + "tokens_seen": 373686272 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004478936810431294, + "loss": 1.4902, + "theoretical_loss": 4.042706260982422, + "tokens_seen": 373751808 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004478836509528586, + "loss": 1.3833, + "theoretical_loss": 4.042626054899946, + "tokens_seen": 373817344 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044787362086258774, + "loss": 1.5207, + "theoretical_loss": 4.042545866814011, + "tokens_seen": 373882880 + }, + { + "epoch": 0.11, + "learning_rate": 0.000447863590772317, + "loss": 1.3796, + "theoretical_loss": 4.042465696717425, + "tokens_seen": 373948416 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044785356068204616, + "loss": 1.4658, + "theoretical_loss": 4.042385544603, + "tokens_seen": 374013952 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044784353059177534, + "loss": 1.4597, + "theoretical_loss": 4.0423054104635545, + "tokens_seen": 374079488 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004478335005015045, + "loss": 1.6308, + "theoretical_loss": 4.042225294291908, + "tokens_seen": 374145024 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004478234704112337, + "loss": 1.4986, + "theoretical_loss": 4.042145196080887, + "tokens_seen": 374210560 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004478134403209629, + "loss": 1.6045, + "theoretical_loss": 4.042065115823319, + "tokens_seen": 374276096 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004478034102306921, + "loss": 1.4669, + "theoretical_loss": 4.041985053512038, + "tokens_seen": 374341632 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044779338014042124, + "loss": 1.5124, + "theoretical_loss": 4.041905009139881, + "tokens_seen": 374407168 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004477833500501505, + "loss": 1.5826, + "theoretical_loss": 4.04182498269969, + "tokens_seen": 374472704 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004477733199598796, + "loss": 1.4024, + "theoretical_loss": 4.04174497418431, + "tokens_seen": 374538240 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044776328986960884, + "loss": 1.5123, + "theoretical_loss": 4.041664983586591, + "tokens_seen": 374603776 + }, + { + "epoch": 0.11, + "learning_rate": 0.000447753259779338, + "loss": 1.5401, + "theoretical_loss": 4.041585010899388, + "tokens_seen": 374669312 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004477432296890672, + "loss": 1.438, + "theoretical_loss": 4.041505056115556, + "tokens_seen": 374734848 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004477331995987964, + "loss": 1.5708, + "theoretical_loss": 4.041425119227959, + "tokens_seen": 374800384 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004477231695085256, + "loss": 1.5901, + "theoretical_loss": 4.041345200229464, + "tokens_seen": 374865920 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044771313941825475, + "loss": 1.4617, + "theoretical_loss": 4.041265299112939, + "tokens_seen": 374931456 + }, + { + "epoch": 0.11, + "learning_rate": 0.000447703109327984, + "loss": 1.5252, + "theoretical_loss": 4.04118541587126, + "tokens_seen": 374996992 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004476930792377131, + "loss": 1.4917, + "theoretical_loss": 4.041105550497304, + "tokens_seen": 375062528 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044768304914744234, + "loss": 1.5734, + "theoretical_loss": 4.041025702983954, + "tokens_seen": 375128064 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.00241080136038363, + "objective/train/docs_used": 223164, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.140078067779541, + "objective/train/original_loss": 3.140077829360962, + "objective/train/theoretical_loss": 4.040945873324096, + "objective/train/tokens_used": 395653600, + "objective/train/value_avg": -0.004547119140625, + "objective/train/value_loss": 7.94610459706746e-05, + "objective/train/value_max": -0.00048041343688964844, + "objective/train/value_min": -0.056243896484375, + "objective/train/value_reward_corr": 0.21117400976713574, + "objective/train/value_std": 0.0033245086669921875, + "objective/train/weight_avg": 0.9795449376106262, + "objective/train/weighted_lm_loss": 3.0971767902374268, + "objective/train/weights_max": 1.7285579442977905, + "objective/train/weights_min": 0.29373684525489807, + "theoretical_loss": 4.040945873324096, + "tokens_seen": 375193600 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004476730190571715, + "loss": 1.5825, + "theoretical_loss": 4.040945873324096, + "tokens_seen": 375193600 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004476629889669007, + "loss": 1.5649, + "theoretical_loss": 4.040866061510623, + "tokens_seen": 375259136 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004476529588766299, + "loss": 1.4352, + "theoretical_loss": 4.040786267536426, + "tokens_seen": 375324672 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044764292878635907, + "loss": 1.5572, + "theoretical_loss": 4.040706491394406, + "tokens_seen": 375390208 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044763289869608825, + "loss": 1.5686, + "theoretical_loss": 4.040626733077466, + "tokens_seen": 375455744 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004476228686058175, + "loss": 1.533, + "theoretical_loss": 4.040546992578511, + "tokens_seen": 375521280 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004476128385155466, + "loss": 1.5036, + "theoretical_loss": 4.040467269890454, + "tokens_seen": 375586816 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044760280842527585, + "loss": 1.4927, + "theoretical_loss": 4.040387565006208, + "tokens_seen": 375652352 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044759277833500503, + "loss": 1.5902, + "theoretical_loss": 4.0403078779186945, + "tokens_seen": 375717888 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004475827482447342, + "loss": 1.5166, + "theoretical_loss": 4.040228208620833, + "tokens_seen": 375783424 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004475727181544634, + "loss": 1.5062, + "theoretical_loss": 4.040148557105555, + "tokens_seen": 375848960 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044756268806419257, + "loss": 1.5565, + "theoretical_loss": 4.040068923365788, + "tokens_seen": 375914496 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044755265797392175, + "loss": 1.4056, + "theoretical_loss": 4.039989307394469, + "tokens_seen": 375980032 + }, + { + "epoch": 0.11, + "learning_rate": 0.000447542627883651, + "loss": 1.5036, + "theoretical_loss": 4.0399097091845375, + "tokens_seen": 376045568 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004475325977933801, + "loss": 1.5008, + "theoretical_loss": 4.039830128728935, + "tokens_seen": 376111104 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044752256770310935, + "loss": 1.563, + "theoretical_loss": 4.039750566020611, + "tokens_seen": 376176640 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004475125376128385, + "loss": 1.4633, + "theoretical_loss": 4.039671021052515, + "tokens_seen": 376242176 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004475025075225677, + "loss": 1.5509, + "theoretical_loss": 4.039591493817603, + "tokens_seen": 376307712 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004474924774322969, + "loss": 1.4037, + "theoretical_loss": 4.039511984308835, + "tokens_seen": 376373248 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004474824473420261, + "loss": 1.5605, + "theoretical_loss": 4.0394324925191745, + "tokens_seen": 376438784 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044747241725175526, + "loss": 1.4702, + "theoretical_loss": 4.039353018441588, + "tokens_seen": 376504320 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044746238716148444, + "loss": 1.6264, + "theoretical_loss": 4.039273562069047, + "tokens_seen": 376569856 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004474523570712136, + "loss": 1.4414, + "theoretical_loss": 4.0391941233945285, + "tokens_seen": 376635392 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044744232698094285, + "loss": 1.4956, + "theoretical_loss": 4.039114702411009, + "tokens_seen": 376700928 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044743229689067204, + "loss": 1.5794, + "theoretical_loss": 4.039035299111475, + "tokens_seen": 376766464 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": -0.0003659981884993613, + "objective/train/docs_used": 224842, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.117237091064453, + "objective/train/original_loss": 3.117236614227295, + "objective/train/theoretical_loss": 4.038955913488913, + "objective/train/tokens_used": 397292000, + "objective/train/value_avg": -0.0054931640625, + "objective/train/value_loss": 4.6120007027639076e-05, + "objective/train/value_max": -0.0005235671997070312, + "objective/train/value_min": -0.08599853515625, + "objective/train/value_reward_corr": 0.4296001625661325, + "objective/train/value_std": 0.0045166015625, + "objective/train/weight_avg": 0.9985966086387634, + "objective/train/weighted_lm_loss": 3.1065616607666016, + "objective/train/weights_max": 2.363126039505005, + "objective/train/weights_min": 0.6257588863372803, + "theoretical_loss": 4.038955913488913, + "tokens_seen": 376832000 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004474222668004012, + "loss": 1.5859, + "theoretical_loss": 4.038955913488913, + "tokens_seen": 376832000 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044741223671013045, + "loss": 1.6522, + "theoretical_loss": 4.038876545536313, + "tokens_seen": 376897536 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004474022066198596, + "loss": 1.4556, + "theoretical_loss": 4.038797195246673, + "tokens_seen": 376963072 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004473921765295888, + "loss": 1.5089, + "theoretical_loss": 4.03871786261299, + "tokens_seen": 377028608 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044738214643931794, + "loss": 1.4955, + "theoretical_loss": 4.03863854762827, + "tokens_seen": 377094144 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004473721163490472, + "loss": 1.4252, + "theoretical_loss": 4.038559250285518, + "tokens_seen": 377159680 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044736208625877636, + "loss": 1.4949, + "theoretical_loss": 4.038479970577747, + "tokens_seen": 377225216 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044735205616850554, + "loss": 1.5478, + "theoretical_loss": 4.038400708497971, + "tokens_seen": 377290752 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004473420260782347, + "loss": 1.5498, + "theoretical_loss": 4.03832146403921, + "tokens_seen": 377356288 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004473319959879639, + "loss": 1.5353, + "theoretical_loss": 4.03824223719449, + "tokens_seen": 377421824 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004473219658976931, + "loss": 1.4898, + "theoretical_loss": 4.038163027956834, + "tokens_seen": 377487360 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004473119358074223, + "loss": 1.5532, + "theoretical_loss": 4.038083836319276, + "tokens_seen": 377552896 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044730190571715144, + "loss": 1.5624, + "theoretical_loss": 4.038004662274851, + "tokens_seen": 377618432 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004472918756268807, + "loss": 1.5106, + "theoretical_loss": 4.037925505816597, + "tokens_seen": 377683968 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004472818455366098, + "loss": 1.4685, + "theoretical_loss": 4.037846366937559, + "tokens_seen": 377749504 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044727181544633904, + "loss": 1.5997, + "theoretical_loss": 4.037767245630782, + "tokens_seen": 377815040 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004472617853560682, + "loss": 1.5287, + "theoretical_loss": 4.037688141889319, + "tokens_seen": 377880576 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004472517552657974, + "loss": 1.456, + "theoretical_loss": 4.037609055706224, + "tokens_seen": 377946112 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004472417251755266, + "loss": 1.5822, + "theoretical_loss": 4.0375299870745565, + "tokens_seen": 378011648 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004472316950852558, + "loss": 1.4676, + "theoretical_loss": 4.03745093598738, + "tokens_seen": 378077184 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044722166499498495, + "loss": 1.4512, + "theoretical_loss": 4.03737190243776, + "tokens_seen": 378142720 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004472116349047142, + "loss": 1.4196, + "theoretical_loss": 4.0372928864187685, + "tokens_seen": 378208256 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004472016048144433, + "loss": 1.5684, + "theoretical_loss": 4.03721388792348, + "tokens_seen": 378273792 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044719157472417254, + "loss": 1.532, + "theoretical_loss": 4.037134906944973, + "tokens_seen": 378339328 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004471815446339017, + "loss": 1.5138, + "theoretical_loss": 4.0370559434763305, + "tokens_seen": 378404864 + }, + { + "epoch": 0.11, + "objective/train/advantage_avg": 0.0004913489101454616, + "objective/train/docs_used": 225570, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2771575450897217, + "objective/train/original_loss": 3.27715802192688, + "objective/train/theoretical_loss": 4.036976997510639, + "objective/train/tokens_used": 398930400, + "objective/train/value_avg": -0.005519866943359375, + "objective/train/value_loss": 5.089551268611103e-05, + "objective/train/value_max": -0.0005335807800292969, + "objective/train/value_min": -0.05072021484375, + "objective/train/value_reward_corr": 0.28920025496413604, + "objective/train/value_std": 0.004352569580078125, + "objective/train/weight_avg": 1.0074063539505005, + "objective/train/weighted_lm_loss": 3.298820972442627, + "objective/train/weights_max": 1.660638451576233, + "objective/train/weights_min": 0.5687808990478516, + "theoretical_loss": 4.036976997510639, + "tokens_seen": 378470400 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004471715145436309, + "loss": 1.5244, + "theoretical_loss": 4.036976997510639, + "tokens_seen": 378470400 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004471614844533601, + "loss": 1.4495, + "theoretical_loss": 4.036898069040989, + "tokens_seen": 378535936 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044715145436308927, + "loss": 1.5245, + "theoretical_loss": 4.036819158060474, + "tokens_seen": 378601472 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044714142427281845, + "loss": 1.5532, + "theoretical_loss": 4.0367402645621935, + "tokens_seen": 378667008 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004471313941825477, + "loss": 1.5043, + "theoretical_loss": 4.036661388539249, + "tokens_seen": 378732544 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004471213640922768, + "loss": 1.6574, + "theoretical_loss": 4.036582529984749, + "tokens_seen": 378798080 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044711133400200605, + "loss": 1.5402, + "theoretical_loss": 4.0365036888918, + "tokens_seen": 378863616 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044710130391173523, + "loss": 1.5101, + "theoretical_loss": 4.036424865253518, + "tokens_seen": 378929152 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004470912738214644, + "loss": 1.4959, + "theoretical_loss": 4.036346059063021, + "tokens_seen": 378994688 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004470812437311936, + "loss": 1.6333, + "theoretical_loss": 4.036267270313431, + "tokens_seen": 379060224 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044707121364092277, + "loss": 1.5828, + "theoretical_loss": 4.036188498997872, + "tokens_seen": 379125760 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044706118355065195, + "loss": 1.6257, + "theoretical_loss": 4.036109745109475, + "tokens_seen": 379191296 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004470511534603812, + "loss": 1.5154, + "theoretical_loss": 4.036031008641373, + "tokens_seen": 379256832 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004470411233701103, + "loss": 1.681, + "theoretical_loss": 4.035952289586705, + "tokens_seen": 379322368 + }, + { + "epoch": 0.11, + "learning_rate": 0.00044703109327983955, + "loss": 1.5269, + "theoretical_loss": 4.03587358793861, + "tokens_seen": 379387904 + }, + { + "epoch": 0.11, + "learning_rate": 0.0004470210631895687, + "loss": 1.5255, + "theoretical_loss": 4.035794903690235, + "tokens_seen": 379453440 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004470110330992979, + "loss": 1.467, + "theoretical_loss": 4.035716236834727, + "tokens_seen": 379518976 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004470010030090271, + "loss": 1.552, + "theoretical_loss": 4.03563758736524, + "tokens_seen": 379584512 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004469909729187563, + "loss": 1.6329, + "theoretical_loss": 4.035558955274933, + "tokens_seen": 379650048 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044698094282848546, + "loss": 1.4365, + "theoretical_loss": 4.035480340556963, + "tokens_seen": 379715584 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044697091273821464, + "loss": 1.4873, + "theoretical_loss": 4.035401743204496, + "tokens_seen": 379781120 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004469608826479438, + "loss": 1.6195, + "theoretical_loss": 4.035323163210702, + "tokens_seen": 379846656 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044695085255767305, + "loss": 1.4857, + "theoretical_loss": 4.035244600568751, + "tokens_seen": 379912192 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004469408224674022, + "loss": 1.4708, + "theoretical_loss": 4.035166055271819, + "tokens_seen": 379977728 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004469307923771314, + "loss": 1.437, + "theoretical_loss": 4.035087527313088, + "tokens_seen": 380043264 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -0.0004517233755905181, + "objective/train/docs_used": 226798, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2873694896698, + "objective/train/original_loss": 3.2873694896698, + "objective/train/theoretical_loss": 4.035009016685741, + "objective/train/tokens_used": 400568800, + "objective/train/value_avg": -0.004974365234375, + "objective/train/value_loss": 5.6121349189197645e-05, + "objective/train/value_max": -0.0004673004150390625, + "objective/train/value_min": -0.052520751953125, + "objective/train/value_reward_corr": 0.4726165273673275, + "objective/train/value_std": 0.00418853759765625, + "objective/train/weight_avg": 0.99814373254776, + "objective/train/weighted_lm_loss": 3.277534246444702, + "objective/train/weights_max": 1.6229150295257568, + "objective/train/weights_min": 0.38652390241622925, + "theoretical_loss": 4.035009016685741, + "tokens_seen": 380108800 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004469207622868606, + "loss": 1.7024, + "theoretical_loss": 4.035009016685741, + "tokens_seen": 380108800 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004469107321965898, + "loss": 1.5546, + "theoretical_loss": 4.034930523382966, + "tokens_seen": 380174336 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044690070210631896, + "loss": 1.5859, + "theoretical_loss": 4.034852047397955, + "tokens_seen": 380239872 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044689067201604814, + "loss": 1.4525, + "theoretical_loss": 4.034773588723901, + "tokens_seen": 380305408 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004468806419257773, + "loss": 1.5407, + "theoretical_loss": 4.034695147354006, + "tokens_seen": 380370944 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044687061183550656, + "loss": 1.5191, + "theoretical_loss": 4.034616723281472, + "tokens_seen": 380436480 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004468605817452357, + "loss": 1.5732, + "theoretical_loss": 4.034538316499505, + "tokens_seen": 380502016 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004468505516549649, + "loss": 1.4974, + "theoretical_loss": 4.0344599270013175, + "tokens_seen": 380567552 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044684052156469405, + "loss": 1.46, + "theoretical_loss": 4.034381554780124, + "tokens_seen": 380633088 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004468304914744233, + "loss": 1.5322, + "theoretical_loss": 4.034303199829141, + "tokens_seen": 380698624 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044682046138415246, + "loss": 1.4667, + "theoretical_loss": 4.034224862141594, + "tokens_seen": 380764160 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044681043129388164, + "loss": 1.5131, + "theoretical_loss": 4.034146541710705, + "tokens_seen": 380829696 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004468004012036108, + "loss": 1.4945, + "theoretical_loss": 4.0340682385297075, + "tokens_seen": 380895232 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044679037111334, + "loss": 1.5107, + "theoretical_loss": 4.033989952591835, + "tokens_seen": 380960768 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004467803410230692, + "loss": 1.3533, + "theoretical_loss": 4.033911683890322, + "tokens_seen": 381026304 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004467703109327984, + "loss": 1.4567, + "theoretical_loss": 4.0338334324184135, + "tokens_seen": 381091840 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044676028084252755, + "loss": 1.3875, + "theoretical_loss": 4.033755198169352, + "tokens_seen": 381157376 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004467502507522568, + "loss": 1.4579, + "theoretical_loss": 4.033676981136388, + "tokens_seen": 381222912 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044674022066198597, + "loss": 1.4896, + "theoretical_loss": 4.033598781312774, + "tokens_seen": 381288448 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044673019057171515, + "loss": 1.5645, + "theoretical_loss": 4.033520598691767, + "tokens_seen": 381353984 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044672016048144433, + "loss": 1.4301, + "theoretical_loss": 4.033442433266627, + "tokens_seen": 381419520 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004467101303911735, + "loss": 1.4039, + "theoretical_loss": 4.0333642850306175, + "tokens_seen": 381485056 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004467001003009027, + "loss": 1.5126, + "theoretical_loss": 4.0332861539770075, + "tokens_seen": 381550592 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004466900702106319, + "loss": 1.5226, + "theoretical_loss": 4.033208040099068, + "tokens_seen": 381616128 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004466800401203611, + "loss": 1.4409, + "theoretical_loss": 4.033129943390076, + "tokens_seen": 381681664 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -0.00021803785057272762, + "objective/train/docs_used": 227456, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.577603340148926, + "objective/train/original_loss": 3.577603578567505, + "objective/train/theoretical_loss": 4.033051863843309, + "objective/train/tokens_used": 402207200, + "objective/train/value_avg": -0.004802703857421875, + "objective/train/value_loss": 5.994086677674204e-05, + "objective/train/value_max": -0.0004565715789794922, + "objective/train/value_min": -0.06683349609375, + "objective/train/value_reward_corr": 0.29137103859680236, + "objective/train/value_std": 0.003772735595703125, + "objective/train/weight_avg": 1.0005382299423218, + "objective/train/weighted_lm_loss": 3.581559181213379, + "objective/train/weights_max": 1.6979916095733643, + "objective/train/weights_min": 0.24534916877746582, + "theoretical_loss": 4.033051863843309, + "tokens_seen": 381747200 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004466700100300903, + "loss": 1.5701, + "theoretical_loss": 4.033051863843309, + "tokens_seen": 381747200 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044665997993981947, + "loss": 1.5637, + "theoretical_loss": 4.032973801452052, + "tokens_seen": 381812736 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044664994984954865, + "loss": 1.5031, + "theoretical_loss": 4.032895756209591, + "tokens_seen": 381878272 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004466399197592779, + "loss": 1.5247, + "theoretical_loss": 4.032817728109217, + "tokens_seen": 381943808 + }, + { + "epoch": 0.12, + "learning_rate": 0.000446629889669007, + "loss": 1.6018, + "theoretical_loss": 4.032739717144223, + "tokens_seen": 382009344 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044661985957873625, + "loss": 1.5026, + "theoretical_loss": 4.03266172330791, + "tokens_seen": 382074880 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044660982948846543, + "loss": 1.6278, + "theoretical_loss": 4.032583746593578, + "tokens_seen": 382140416 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004465997993981946, + "loss": 1.4914, + "theoretical_loss": 4.032505786994534, + "tokens_seen": 382205952 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004465897693079238, + "loss": 1.4258, + "theoretical_loss": 4.032427844504086, + "tokens_seen": 382271488 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044657973921765297, + "loss": 1.4539, + "theoretical_loss": 4.032349919115549, + "tokens_seen": 382337024 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044656970912738215, + "loss": 1.6705, + "theoretical_loss": 4.032272010822239, + "tokens_seen": 382402560 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004465596790371114, + "loss": 1.4814, + "theoretical_loss": 4.032194119617477, + "tokens_seen": 382468096 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004465496489468405, + "loss": 1.5578, + "theoretical_loss": 4.032116245494589, + "tokens_seen": 382533632 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044653961885656975, + "loss": 1.532, + "theoretical_loss": 4.032038388446901, + "tokens_seen": 382599168 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004465295887662989, + "loss": 1.4639, + "theoretical_loss": 4.031960548467747, + "tokens_seen": 382664704 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004465195586760281, + "loss": 1.5271, + "theoretical_loss": 4.031882725550463, + "tokens_seen": 382730240 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004465095285857573, + "loss": 1.5559, + "theoretical_loss": 4.031804919688387, + "tokens_seen": 382795776 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004464994984954865, + "loss": 1.4673, + "theoretical_loss": 4.031727130874863, + "tokens_seen": 382861312 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044648946840521566, + "loss": 1.4665, + "theoretical_loss": 4.031649359103238, + "tokens_seen": 382926848 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044647943831494484, + "loss": 1.5972, + "theoretical_loss": 4.031571604366864, + "tokens_seen": 382992384 + }, + { + "epoch": 0.12, + "learning_rate": 0.000446469408224674, + "loss": 1.4224, + "theoretical_loss": 4.031493866659092, + "tokens_seen": 383057920 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044645937813440325, + "loss": 1.5232, + "theoretical_loss": 4.031416145973285, + "tokens_seen": 383123456 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004464493480441324, + "loss": 1.5492, + "theoretical_loss": 4.031338442302802, + "tokens_seen": 383188992 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004464393179538616, + "loss": 1.5498, + "theoretical_loss": 4.03126075564101, + "tokens_seen": 383254528 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004464292878635908, + "loss": 1.5097, + "theoretical_loss": 4.031183085981278, + "tokens_seen": 383320064 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -0.0008161737932823598, + "objective/train/docs_used": 228876, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2257659435272217, + "objective/train/original_loss": 3.2257659435272217, + "objective/train/theoretical_loss": 4.031105433316977, + "objective/train/tokens_used": 403845600, + "objective/train/value_avg": -0.00502777099609375, + "objective/train/value_loss": 5.0959679356310517e-05, + "objective/train/value_max": -0.0004355907440185547, + "objective/train/value_min": -0.060638427734375, + "objective/train/value_reward_corr": 0.33370273436122033, + "objective/train/value_std": 0.00389862060546875, + "objective/train/weight_avg": 0.9941828846931458, + "objective/train/weighted_lm_loss": 3.2107648849487305, + "objective/train/weights_max": 1.5822914838790894, + "objective/train/weights_min": 0.29798606038093567, + "theoretical_loss": 4.031105433316977, + "tokens_seen": 383385600 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044641925777332, + "loss": 1.5408, + "theoretical_loss": 4.031105433316977, + "tokens_seen": 383385600 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044640922768304916, + "loss": 1.5245, + "theoretical_loss": 4.031027797641488, + "tokens_seen": 383451136 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044639919759277834, + "loss": 1.5288, + "theoretical_loss": 4.0309501789481885, + "tokens_seen": 383516672 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004463891675025075, + "loss": 1.5693, + "theoretical_loss": 4.030872577230465, + "tokens_seen": 383582208 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044637913741223676, + "loss": 1.6014, + "theoretical_loss": 4.030794992481702, + "tokens_seen": 383647744 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004463691073219659, + "loss": 1.481, + "theoretical_loss": 4.030717424695295, + "tokens_seen": 383713280 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004463590772316951, + "loss": 1.5363, + "theoretical_loss": 4.030639873864638, + "tokens_seen": 383778816 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044634904714142425, + "loss": 1.5453, + "theoretical_loss": 4.030562339983129, + "tokens_seen": 383844352 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004463390170511535, + "loss": 1.5595, + "theoretical_loss": 4.030484823044173, + "tokens_seen": 383909888 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044632898696088266, + "loss": 1.5049, + "theoretical_loss": 4.0304073230411745, + "tokens_seen": 383975424 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044631895687061184, + "loss": 1.4087, + "theoretical_loss": 4.0303298399675445, + "tokens_seen": 384040960 + }, + { + "epoch": 0.12, + "learning_rate": 0.000446308926780341, + "loss": 1.4188, + "theoretical_loss": 4.030252373816697, + "tokens_seen": 384106496 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004462988966900702, + "loss": 1.4852, + "theoretical_loss": 4.030174924582049, + "tokens_seen": 384172032 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004462888665997994, + "loss": 1.5387, + "theoretical_loss": 4.030097492257021, + "tokens_seen": 384237568 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004462788365095286, + "loss": 1.4197, + "theoretical_loss": 4.03002007683504, + "tokens_seen": 384303104 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044626880641925775, + "loss": 1.5274, + "theoretical_loss": 4.029942678309534, + "tokens_seen": 384368640 + }, + { + "epoch": 0.12, + "learning_rate": 0.000446258776328987, + "loss": 1.5471, + "theoretical_loss": 4.029865296673934, + "tokens_seen": 384434176 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044624874623871617, + "loss": 1.5632, + "theoretical_loss": 4.029787931921677, + "tokens_seen": 384499712 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044623871614844535, + "loss": 1.5843, + "theoretical_loss": 4.029710584046203, + "tokens_seen": 384565248 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044622868605817453, + "loss": 1.583, + "theoretical_loss": 4.029633253040954, + "tokens_seen": 384630784 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004462186559679037, + "loss": 1.5408, + "theoretical_loss": 4.029555938899378, + "tokens_seen": 384696320 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004462086258776329, + "loss": 1.5486, + "theoretical_loss": 4.029478641614924, + "tokens_seen": 384761856 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004461985957873621, + "loss": 1.405, + "theoretical_loss": 4.029401361181049, + "tokens_seen": 384827392 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044618856569709125, + "loss": 1.4291, + "theoretical_loss": 4.029324097591209, + "tokens_seen": 384892928 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004461785356068205, + "loss": 1.5844, + "theoretical_loss": 4.029246850838865, + "tokens_seen": 384958464 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -5.6192398915300146e-05, + "objective/train/docs_used": 229454, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6609604358673096, + "objective/train/original_loss": 2.6609604358673096, + "objective/train/theoretical_loss": 4.029169620917484, + "objective/train/tokens_used": 405484000, + "objective/train/value_avg": -0.004550933837890625, + "objective/train/value_loss": 9.009215864352882e-05, + "objective/train/value_max": -0.0005769729614257812, + "objective/train/value_min": -0.0830078125, + "objective/train/value_reward_corr": 0.30171062282131345, + "objective/train/value_std": 0.0036754608154296875, + "objective/train/weight_avg": 1.00286066532135, + "objective/train/weighted_lm_loss": 2.66163969039917, + "objective/train/weights_max": 2.202498197555542, + "objective/train/weights_min": 0.08357236534357071, + "theoretical_loss": 4.029169620917484, + "tokens_seen": 385024000 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004461685055165496, + "loss": 1.5843, + "theoretical_loss": 4.029169620917484, + "tokens_seen": 385024000 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044615847542627885, + "loss": 1.585, + "theoretical_loss": 4.029092407820533, + "tokens_seen": 385089536 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044614844533600803, + "loss": 1.5162, + "theoretical_loss": 4.029015211541487, + "tokens_seen": 385155072 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004461384152457372, + "loss": 1.4403, + "theoretical_loss": 4.028938032073819, + "tokens_seen": 385220608 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004461283851554664, + "loss": 1.5395, + "theoretical_loss": 4.0288608694110115, + "tokens_seen": 385286144 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044611835506519563, + "loss": 1.5407, + "theoretical_loss": 4.028783723546547, + "tokens_seen": 385351680 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044610832497492476, + "loss": 1.5751, + "theoretical_loss": 4.028706594473912, + "tokens_seen": 385417216 + }, + { + "epoch": 0.12, + "learning_rate": 0.000446098294884654, + "loss": 1.569, + "theoretical_loss": 4.028629482186598, + "tokens_seen": 385482752 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004460882647943831, + "loss": 1.462, + "theoretical_loss": 4.028552386678099, + "tokens_seen": 385548288 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044607823470411235, + "loss": 1.5056, + "theoretical_loss": 4.028475307941912, + "tokens_seen": 385613824 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044606820461384153, + "loss": 1.4616, + "theoretical_loss": 4.028398245971541, + "tokens_seen": 385679360 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004460581745235707, + "loss": 1.4231, + "theoretical_loss": 4.028321200760489, + "tokens_seen": 385744896 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004460481444332999, + "loss": 1.3917, + "theoretical_loss": 4.028244172302266, + "tokens_seen": 385810432 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004460381143430291, + "loss": 1.5593, + "theoretical_loss": 4.028167160590383, + "tokens_seen": 385875968 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044602808425275826, + "loss": 1.4674, + "theoretical_loss": 4.028090165618357, + "tokens_seen": 385941504 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004460180541624875, + "loss": 1.4904, + "theoretical_loss": 4.028013187379709, + "tokens_seen": 386007040 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004460080240722166, + "loss": 1.5046, + "theoretical_loss": 4.027936225867959, + "tokens_seen": 386072576 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044599799398194586, + "loss": 1.5409, + "theoretical_loss": 4.027859281076638, + "tokens_seen": 386138112 + }, + { + "epoch": 0.12, + "learning_rate": 0.000445987963891675, + "loss": 1.4327, + "theoretical_loss": 4.027782352999273, + "tokens_seen": 386203648 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004459779338014042, + "loss": 1.6407, + "theoretical_loss": 4.027705441629399, + "tokens_seen": 386269184 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004459679037111334, + "loss": 1.3837, + "theoretical_loss": 4.027628546960555, + "tokens_seen": 386334720 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004459578736208626, + "loss": 1.5356, + "theoretical_loss": 4.0275516689862805, + "tokens_seen": 386400256 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044594784353059176, + "loss": 1.4022, + "theoretical_loss": 4.027474807700122, + "tokens_seen": 386465792 + }, + { + "epoch": 0.12, + "learning_rate": 0.000445937813440321, + "loss": 1.553, + "theoretical_loss": 4.027397963095627, + "tokens_seen": 386531328 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004459277833500502, + "loss": 1.4562, + "theoretical_loss": 4.027321135166347, + "tokens_seen": 386596864 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": 7.238032412715256e-05, + "objective/train/docs_used": 230121, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.018120288848877, + "objective/train/original_loss": 3.018120288848877, + "objective/train/theoretical_loss": 4.027244323905839, + "objective/train/tokens_used": 407122400, + "objective/train/value_avg": -0.004825592041015625, + "objective/train/value_loss": 5.094015796203166e-05, + "objective/train/value_max": -0.0004897117614746094, + "objective/train/value_min": -0.049041748046875, + "objective/train/value_reward_corr": 0.30165774050521926, + "objective/train/value_std": 0.0038890838623046875, + "objective/train/weight_avg": 1.0031205415725708, + "objective/train/weighted_lm_loss": 3.0368735790252686, + "objective/train/weights_max": 1.6329978704452515, + "objective/train/weights_min": 0.3055603802204132, + "theoretical_loss": 4.027244323905839, + "tokens_seen": 386662400 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044591775325977936, + "loss": 1.4987, + "theoretical_loss": 4.027244323905839, + "tokens_seen": 386662400 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044590772316950854, + "loss": 1.5193, + "theoretical_loss": 4.027167529307661, + "tokens_seen": 386727936 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004458976930792377, + "loss": 1.6109, + "theoretical_loss": 4.027090751365376, + "tokens_seen": 386793472 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044588766298896696, + "loss": 1.4211, + "theoretical_loss": 4.027013990072551, + "tokens_seen": 386859008 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004458776328986961, + "loss": 1.545, + "theoretical_loss": 4.026937245422756, + "tokens_seen": 386924544 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004458676028084253, + "loss": 1.5584, + "theoretical_loss": 4.026860517409563, + "tokens_seen": 386990080 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044585757271815445, + "loss": 1.4439, + "theoretical_loss": 4.026783806026552, + "tokens_seen": 387055616 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004458475426278837, + "loss": 1.5357, + "theoretical_loss": 4.0267071112673, + "tokens_seen": 387121152 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044583751253761286, + "loss": 1.4847, + "theoretical_loss": 4.026630433125394, + "tokens_seen": 387186688 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044582748244734204, + "loss": 1.5078, + "theoretical_loss": 4.026553771594421, + "tokens_seen": 387252224 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004458174523570712, + "loss": 1.4663, + "theoretical_loss": 4.026477126667972, + "tokens_seen": 387317760 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004458074222668004, + "loss": 1.5353, + "theoretical_loss": 4.026400498339642, + "tokens_seen": 387383296 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004457973921765296, + "loss": 1.4542, + "theoretical_loss": 4.026323886603029, + "tokens_seen": 387448832 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004457873620862588, + "loss": 1.4616, + "theoretical_loss": 4.026247291451735, + "tokens_seen": 387514368 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044577733199598795, + "loss": 1.4363, + "theoretical_loss": 4.026170712879368, + "tokens_seen": 387579904 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004457673019057172, + "loss": 1.4251, + "theoretical_loss": 4.026094150879533, + "tokens_seen": 387645440 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044575727181544637, + "loss": 1.5909, + "theoretical_loss": 4.026017605445846, + "tokens_seen": 387710976 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044574724172517555, + "loss": 1.5615, + "theoretical_loss": 4.025941076571922, + "tokens_seen": 387776512 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044573721163490473, + "loss": 1.4838, + "theoretical_loss": 4.025864564251379, + "tokens_seen": 387842048 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004457271815446339, + "loss": 1.4398, + "theoretical_loss": 4.025788068477843, + "tokens_seen": 387907584 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004457171514543631, + "loss": 1.5158, + "theoretical_loss": 4.025711589244939, + "tokens_seen": 387973120 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004457071213640923, + "loss": 1.5418, + "theoretical_loss": 4.025635126546299, + "tokens_seen": 388038656 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044569709127382145, + "loss": 1.4561, + "theoretical_loss": 4.025558680375555, + "tokens_seen": 388104192 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004456870611835507, + "loss": 1.5417, + "theoretical_loss": 4.025482250726346, + "tokens_seen": 388169728 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004456770310932798, + "loss": 1.4635, + "theoretical_loss": 4.025405837592311, + "tokens_seen": 388235264 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": 0.0006219978095032275, + "objective/train/docs_used": 231531, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.213958740234375, + "objective/train/original_loss": 3.213958740234375, + "objective/train/theoretical_loss": 4.025329440967096, + "objective/train/tokens_used": 408760800, + "objective/train/value_avg": -0.00592041015625, + "objective/train/value_loss": 6.017303894623183e-05, + "objective/train/value_max": -0.0007581710815429688, + "objective/train/value_min": -0.060638427734375, + "objective/train/value_reward_corr": 0.43298464998392466, + "objective/train/value_std": 0.0047454833984375, + "objective/train/weight_avg": 1.00910484790802, + "objective/train/weighted_lm_loss": 3.2373695373535156, + "objective/train/weights_max": 1.6166348457336426, + "objective/train/weights_min": 0.5692585110664368, + "theoretical_loss": 4.025329440967096, + "tokens_seen": 388300800 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044566700100300905, + "loss": 1.52, + "theoretical_loss": 4.025329440967096, + "tokens_seen": 388300800 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044565697091273823, + "loss": 1.4645, + "theoretical_loss": 4.0252530608443475, + "tokens_seen": 388366336 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004456469408224674, + "loss": 1.621, + "theoretical_loss": 4.025176697217718, + "tokens_seen": 388431872 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004456369107321966, + "loss": 1.4831, + "theoretical_loss": 4.025100350080864, + "tokens_seen": 388497408 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044562688064192583, + "loss": 1.5599, + "theoretical_loss": 4.02502401942744, + "tokens_seen": 388562944 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044561685055165496, + "loss": 1.652, + "theoretical_loss": 4.024947705251112, + "tokens_seen": 388628480 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004456068204613842, + "loss": 1.5099, + "theoretical_loss": 4.024871407545542, + "tokens_seen": 388694016 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004455967903711133, + "loss": 1.5216, + "theoretical_loss": 4.024795126304402, + "tokens_seen": 388759552 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044558676028084255, + "loss": 1.4336, + "theoretical_loss": 4.024718861521362, + "tokens_seen": 388825088 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044557673019057173, + "loss": 1.4604, + "theoretical_loss": 4.0246426131901, + "tokens_seen": 388890624 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004455667001003009, + "loss": 1.5139, + "theoretical_loss": 4.024566381304294, + "tokens_seen": 388956160 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004455566700100301, + "loss": 1.4379, + "theoretical_loss": 4.024490165857627, + "tokens_seen": 389021696 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004455466399197593, + "loss": 1.4979, + "theoretical_loss": 4.024413966843787, + "tokens_seen": 389087232 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044553660982948846, + "loss": 1.5285, + "theoretical_loss": 4.024337784256463, + "tokens_seen": 389152768 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004455265797392177, + "loss": 1.4924, + "theoretical_loss": 4.024261618089348, + "tokens_seen": 389218304 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004455165496489468, + "loss": 1.5034, + "theoretical_loss": 4.024185468336139, + "tokens_seen": 389283840 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044550651955867606, + "loss": 1.419, + "theoretical_loss": 4.024109334990538, + "tokens_seen": 389349376 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004454964894684052, + "loss": 1.5427, + "theoretical_loss": 4.024033218046245, + "tokens_seen": 389414912 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004454864593781344, + "loss": 1.4846, + "theoretical_loss": 4.023957117496972, + "tokens_seen": 389480448 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004454764292878636, + "loss": 1.4932, + "theoretical_loss": 4.023881033336427, + "tokens_seen": 389545984 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004454663991975928, + "loss": 1.654, + "theoretical_loss": 4.023804965558325, + "tokens_seen": 389611520 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044545636910732196, + "loss": 1.5368, + "theoretical_loss": 4.023728914156385, + "tokens_seen": 389677056 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004454463390170512, + "loss": 1.5836, + "theoretical_loss": 4.023652879124325, + "tokens_seen": 389742592 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004454363089267803, + "loss": 1.4972, + "theoretical_loss": 4.023576860455872, + "tokens_seen": 389808128 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044542627883650956, + "loss": 1.5322, + "theoretical_loss": 4.023500858144755, + "tokens_seen": 389873664 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": 0.00034032127587124705, + "objective/train/docs_used": 232298, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8616058826446533, + "objective/train/original_loss": 2.8616063594818115, + "objective/train/theoretical_loss": 4.0234248721847035, + "objective/train/tokens_used": 410399200, + "objective/train/value_avg": -0.004344940185546875, + "objective/train/value_loss": 3.394051600480452e-05, + "objective/train/value_max": -0.00042557716369628906, + "objective/train/value_min": -0.039642333984375, + "objective/train/value_reward_corr": 0.31566131027683625, + "objective/train/value_std": 0.002887725830078125, + "objective/train/weight_avg": 1.005057692527771, + "objective/train/weighted_lm_loss": 2.8694674968719482, + "objective/train/weights_max": 1.4278076887130737, + "objective/train/weights_min": 0.587395429611206, + "theoretical_loss": 4.0234248721847035, + "tokens_seen": 389939200 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004454162487462387, + "loss": 1.6017, + "theoretical_loss": 4.0234248721847035, + "tokens_seen": 389939200 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004454062186559679, + "loss": 1.486, + "theoretical_loss": 4.023348902569453, + "tokens_seen": 390004736 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004453961885656971, + "loss": 1.5545, + "theoretical_loss": 4.023272949292743, + "tokens_seen": 390070272 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004453861584754263, + "loss": 1.4346, + "theoretical_loss": 4.023197012348315, + "tokens_seen": 390135808 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044537612838515547, + "loss": 1.5254, + "theoretical_loss": 4.023121091729914, + "tokens_seen": 390201344 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044536609829488465, + "loss": 1.5649, + "theoretical_loss": 4.02304518743129, + "tokens_seen": 390266880 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044535606820461383, + "loss": 1.4775, + "theoretical_loss": 4.022969299446194, + "tokens_seen": 390332416 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044534603811434306, + "loss": 1.6335, + "theoretical_loss": 4.022893427768381, + "tokens_seen": 390397952 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004453360080240722, + "loss": 1.5491, + "theoretical_loss": 4.022817572391613, + "tokens_seen": 390463488 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004453259779338014, + "loss": 1.4026, + "theoretical_loss": 4.02274173330965, + "tokens_seen": 390529024 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044531594784353055, + "loss": 1.4556, + "theoretical_loss": 4.02266591051626, + "tokens_seen": 390594560 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004453059177532598, + "loss": 1.6121, + "theoretical_loss": 4.022590104005211, + "tokens_seen": 390660096 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044529588766298897, + "loss": 1.4825, + "theoretical_loss": 4.022514313770276, + "tokens_seen": 390725632 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044528585757271815, + "loss": 1.4441, + "theoretical_loss": 4.022438539805231, + "tokens_seen": 390791168 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044527582748244733, + "loss": 1.4612, + "theoretical_loss": 4.0223627821038574, + "tokens_seen": 390856704 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044526579739217657, + "loss": 1.5778, + "theoretical_loss": 4.022287040659938, + "tokens_seen": 390922240 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004452557673019057, + "loss": 1.4469, + "theoretical_loss": 4.022211315467258, + "tokens_seen": 390987776 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044524573721163493, + "loss": 1.5442, + "theoretical_loss": 4.022135606519608, + "tokens_seen": 391053312 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044523570712136406, + "loss": 1.5117, + "theoretical_loss": 4.022059913810782, + "tokens_seen": 391118848 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004452256770310933, + "loss": 1.4869, + "theoretical_loss": 4.021984237334577, + "tokens_seen": 391184384 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044521564694082247, + "loss": 1.452, + "theoretical_loss": 4.021908577084791, + "tokens_seen": 391249920 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044520561685055165, + "loss": 1.4574, + "theoretical_loss": 4.021832933055231, + "tokens_seen": 391315456 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044519558676028083, + "loss": 1.4498, + "theoretical_loss": 4.021757305239702, + "tokens_seen": 391380992 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044518555667001, + "loss": 1.4566, + "theoretical_loss": 4.021681693632015, + "tokens_seen": 391446528 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044517552657973925, + "loss": 1.3893, + "theoretical_loss": 4.021606098225982, + "tokens_seen": 391512064 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -0.00028277584351599216, + "objective/train/docs_used": 233436, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0145487785339355, + "objective/train/original_loss": 3.0145487785339355, + "objective/train/theoretical_loss": 4.021530519015424, + "objective/train/tokens_used": 412037600, + "objective/train/value_avg": -0.005329132080078125, + "objective/train/value_loss": 5.7168865168932825e-05, + "objective/train/value_max": -0.0003726482391357422, + "objective/train/value_min": -0.07861328125, + "objective/train/value_reward_corr": 0.2934280284731387, + "objective/train/value_std": 0.0040283203125, + "objective/train/weight_avg": 0.9999294877052307, + "objective/train/weighted_lm_loss": 3.0292720794677734, + "objective/train/weights_max": 2.110806941986084, + "objective/train/weights_min": 0.48324528336524963, + "theoretical_loss": 4.021530519015424, + "tokens_seen": 391577600 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044516549648946843, + "loss": 1.5043, + "theoretical_loss": 4.021530519015424, + "tokens_seen": 391577600 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004451554663991976, + "loss": 1.4269, + "theoretical_loss": 4.021454955994159, + "tokens_seen": 391643136 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004451454363089268, + "loss": 1.4531, + "theoretical_loss": 4.02137940915601, + "tokens_seen": 391708672 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044513540621865603, + "loss": 1.55, + "theoretical_loss": 4.021303878494807, + "tokens_seen": 391774208 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044512537612838516, + "loss": 1.4297, + "theoretical_loss": 4.02122836400438, + "tokens_seen": 391839744 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004451153460381144, + "loss": 1.4576, + "theoretical_loss": 4.021152865678563, + "tokens_seen": 391905280 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004451053159478435, + "loss": 1.554, + "theoretical_loss": 4.0210773835111935, + "tokens_seen": 391970816 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044509528585757275, + "loss": 1.5472, + "theoretical_loss": 4.021001917496112, + "tokens_seen": 392036352 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044508525576730194, + "loss": 1.4, + "theoretical_loss": 4.020926467627164, + "tokens_seen": 392101888 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004450752256770311, + "loss": 1.4362, + "theoretical_loss": 4.020851033898196, + "tokens_seen": 392167424 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004450651955867603, + "loss": 1.5547, + "theoretical_loss": 4.020775616303059, + "tokens_seen": 392232960 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004450551654964895, + "loss": 1.5329, + "theoretical_loss": 4.020700214835609, + "tokens_seen": 392298496 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044504513540621866, + "loss": 1.5646, + "theoretical_loss": 4.020624829489702, + "tokens_seen": 392364032 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004450351053159479, + "loss": 1.5668, + "theoretical_loss": 4.0205494602592, + "tokens_seen": 392429568 + }, + { + "epoch": 0.12, + "learning_rate": 0.000445025075225677, + "loss": 1.3801, + "theoretical_loss": 4.020474107137967, + "tokens_seen": 392495104 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044501504513540626, + "loss": 1.3216, + "theoretical_loss": 4.020398770119872, + "tokens_seen": 392560640 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004450050150451354, + "loss": 1.5442, + "theoretical_loss": 4.020323449198785, + "tokens_seen": 392626176 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004449949849548646, + "loss": 1.4751, + "theoretical_loss": 4.020248144368582, + "tokens_seen": 392691712 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004449849548645938, + "loss": 1.5647, + "theoretical_loss": 4.020172855623139, + "tokens_seen": 392757248 + }, + { + "epoch": 0.12, + "learning_rate": 0.000444974924774323, + "loss": 1.4892, + "theoretical_loss": 4.02009758295634, + "tokens_seen": 392822784 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044496489468405216, + "loss": 1.4949, + "theoretical_loss": 4.020022326362067, + "tokens_seen": 392888320 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004449548645937814, + "loss": 1.3826, + "theoretical_loss": 4.019947085834209, + "tokens_seen": 392953856 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004449448345035105, + "loss": 1.6047, + "theoretical_loss": 4.019871861366658, + "tokens_seen": 393019392 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044493480441323976, + "loss": 1.4608, + "theoretical_loss": 4.019796652953309, + "tokens_seen": 393084928 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004449247743229689, + "loss": 1.4877, + "theoretical_loss": 4.019721460588057, + "tokens_seen": 393150464 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -0.0003070256789214909, + "objective/train/docs_used": 234056, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0160040855407715, + "objective/train/original_loss": 3.0160043239593506, + "objective/train/theoretical_loss": 4.019646284264807, + "objective/train/tokens_used": 413676000, + "objective/train/value_avg": -0.00559234619140625, + "objective/train/value_loss": 4.756533962790854e-05, + "objective/train/value_max": -0.0004973411560058594, + "objective/train/value_min": -0.2440185546875, + "objective/train/value_reward_corr": 0.3651812706169777, + "objective/train/value_std": 0.004467010498046875, + "objective/train/weight_avg": 0.9995595812797546, + "objective/train/weighted_lm_loss": 3.0158543586730957, + "objective/train/weights_max": 11.475170135498047, + "objective/train/weights_min": 0.5619175434112549, + "theoretical_loss": 4.019646284264807, + "tokens_seen": 393216000 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004449147442326981, + "loss": 1.5008, + "theoretical_loss": 4.019646284264807, + "tokens_seen": 393216000 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004449047141424273, + "loss": 1.5102, + "theoretical_loss": 4.019571123977462, + "tokens_seen": 393281536 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004448946840521565, + "loss": 1.3592, + "theoretical_loss": 4.01949597971993, + "tokens_seen": 393347072 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044488465396188567, + "loss": 1.4859, + "theoretical_loss": 4.019420851486123, + "tokens_seen": 393412608 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044487462387161485, + "loss": 1.4327, + "theoretical_loss": 4.019345739269955, + "tokens_seen": 393478144 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044486459378134403, + "loss": 1.4153, + "theoretical_loss": 4.019270643065344, + "tokens_seen": 393543680 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044485456369107326, + "loss": 1.465, + "theoretical_loss": 4.019195562866211, + "tokens_seen": 393609216 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004448445336008024, + "loss": 1.5437, + "theoretical_loss": 4.0191204986664815, + "tokens_seen": 393674752 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004448345035105316, + "loss": 1.4708, + "theoretical_loss": 4.019045450460084, + "tokens_seen": 393740288 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044482447342026075, + "loss": 1.4827, + "theoretical_loss": 4.018970418240949, + "tokens_seen": 393805824 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044481444332999, + "loss": 1.4045, + "theoretical_loss": 4.01889540200301, + "tokens_seen": 393871360 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044480441323971917, + "loss": 1.4439, + "theoretical_loss": 4.018820401740206, + "tokens_seen": 393936896 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044479438314944835, + "loss": 1.5599, + "theoretical_loss": 4.018745417446478, + "tokens_seen": 394002432 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044478435305917753, + "loss": 1.4254, + "theoretical_loss": 4.018670449115772, + "tokens_seen": 394067968 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044477432296890677, + "loss": 1.4593, + "theoretical_loss": 4.018595496742034, + "tokens_seen": 394133504 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004447642928786359, + "loss": 1.4769, + "theoretical_loss": 4.018520560319215, + "tokens_seen": 394199040 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044475426278836513, + "loss": 1.4717, + "theoretical_loss": 4.01844563984127, + "tokens_seen": 394264576 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044474423269809426, + "loss": 1.6131, + "theoretical_loss": 4.018370735302156, + "tokens_seen": 394330112 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004447342026078235, + "loss": 1.4752, + "theoretical_loss": 4.018295846695835, + "tokens_seen": 394395648 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044472417251755267, + "loss": 1.5782, + "theoretical_loss": 4.0182209740162715, + "tokens_seen": 394461184 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044471414242728185, + "loss": 1.5696, + "theoretical_loss": 4.018146117257433, + "tokens_seen": 394526720 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044470411233701103, + "loss": 1.4763, + "theoretical_loss": 4.0180712764132895, + "tokens_seen": 394592256 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004446940822467402, + "loss": 1.5038, + "theoretical_loss": 4.017996451477815, + "tokens_seen": 394657792 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004446840521564694, + "loss": 1.5389, + "theoretical_loss": 4.017921642444987, + "tokens_seen": 394723328 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044467402206619863, + "loss": 1.4263, + "theoretical_loss": 4.017846849308788, + "tokens_seen": 394788864 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -0.0015141032636165619, + "objective/train/docs_used": 235431, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1798226833343506, + "objective/train/original_loss": 3.1798222064971924, + "objective/train/theoretical_loss": 4.017772072063201, + "objective/train/tokens_used": 415314400, + "objective/train/value_avg": -0.005336761474609375, + "objective/train/value_loss": 5.587581472354941e-05, + "objective/train/value_max": -0.0004673004150390625, + "objective/train/value_min": -0.052032470703125, + "objective/train/value_reward_corr": 0.36083978528443117, + "objective/train/value_std": 0.004131317138671875, + "objective/train/weight_avg": 0.9875410199165344, + "objective/train/weighted_lm_loss": 3.162485122680664, + "objective/train/weights_max": 1.6344935894012451, + "objective/train/weights_min": 0.5157076120376587, + "theoretical_loss": 4.017772072063201, + "tokens_seen": 394854400 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044466399197592776, + "loss": 1.4992, + "theoretical_loss": 4.017772072063201, + "tokens_seen": 394854400 + }, + { + "epoch": 0.12, + "learning_rate": 0.000444653961885657, + "loss": 1.5036, + "theoretical_loss": 4.017697310702211, + "tokens_seen": 394919936 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004446439317953861, + "loss": 1.3564, + "theoretical_loss": 4.017622565219812, + "tokens_seen": 394985472 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044463390170511536, + "loss": 1.4479, + "theoretical_loss": 4.017547835609996, + "tokens_seen": 395051008 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044462387161484454, + "loss": 1.4801, + "theoretical_loss": 4.017473121866759, + "tokens_seen": 395116544 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004446138415245737, + "loss": 1.4574, + "theoretical_loss": 4.017398423984105, + "tokens_seen": 395182080 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004446038114343029, + "loss": 1.3471, + "theoretical_loss": 4.017323741956034, + "tokens_seen": 395247616 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044459378134403214, + "loss": 1.444, + "theoretical_loss": 4.0172490757765535, + "tokens_seen": 395313152 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044458375125376126, + "loss": 1.6017, + "theoretical_loss": 4.017174425439675, + "tokens_seen": 395378688 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004445737211634905, + "loss": 1.3898, + "theoretical_loss": 4.017099790939412, + "tokens_seen": 395444224 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004445636910732196, + "loss": 1.5295, + "theoretical_loss": 4.017025172269779, + "tokens_seen": 395509760 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044455366098294886, + "loss": 1.6001, + "theoretical_loss": 4.016950569424798, + "tokens_seen": 395575296 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044454363089267804, + "loss": 1.5389, + "theoretical_loss": 4.01687598239849, + "tokens_seen": 395640832 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004445336008024072, + "loss": 1.4604, + "theoretical_loss": 4.016801411184884, + "tokens_seen": 395706368 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004445235707121364, + "loss": 1.4437, + "theoretical_loss": 4.016726855778008, + "tokens_seen": 395771904 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004445135406218656, + "loss": 1.4051, + "theoretical_loss": 4.0166523161718946, + "tokens_seen": 395837440 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044450351053159477, + "loss": 1.3516, + "theoretical_loss": 4.0165777923605805, + "tokens_seen": 395902976 + }, + { + "epoch": 0.12, + "learning_rate": 0.000444493480441324, + "loss": 1.4623, + "theoretical_loss": 4.016503284338105, + "tokens_seen": 395968512 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044448345035105313, + "loss": 1.4625, + "theoretical_loss": 4.016428792098512, + "tokens_seen": 396034048 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044447342026078236, + "loss": 1.4024, + "theoretical_loss": 4.016354315635844, + "tokens_seen": 396099584 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004444633901705115, + "loss": 1.2771, + "theoretical_loss": 4.016279854944154, + "tokens_seen": 396165120 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004444533600802407, + "loss": 1.4488, + "theoretical_loss": 4.016205410017491, + "tokens_seen": 396230656 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044444332998996996, + "loss": 1.505, + "theoretical_loss": 4.016130980849912, + "tokens_seen": 396296192 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004444332998996991, + "loss": 1.4072, + "theoretical_loss": 4.016056567435475, + "tokens_seen": 396361728 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004444232698094283, + "loss": 1.4511, + "theoretical_loss": 4.015982169768244, + "tokens_seen": 396427264 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -0.0017678500153124332, + "objective/train/docs_used": 235840, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.620478630065918, + "objective/train/original_loss": 2.620478630065918, + "objective/train/theoretical_loss": 4.0159077878422815, + "objective/train/tokens_used": 416952800, + "objective/train/value_avg": -0.005252838134765625, + "objective/train/value_loss": 0.00011120009730802849, + "objective/train/value_max": -0.0004172325134277344, + "objective/train/value_min": -0.07989501953125, + "objective/train/value_reward_corr": 0.527790222486125, + "objective/train/value_std": 0.0053863525390625, + "objective/train/weight_avg": 0.9871116280555725, + "objective/train/weighted_lm_loss": 2.5856919288635254, + "objective/train/weights_max": 1.7392908334732056, + "objective/train/weights_min": 0.13876470923423767, + "theoretical_loss": 4.0159077878422815, + "tokens_seen": 396492800 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004444132397191575, + "loss": 1.3609, + "theoretical_loss": 4.0159077878422815, + "tokens_seen": 396492800 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004444032096288867, + "loss": 1.4996, + "theoretical_loss": 4.015833421651657, + "tokens_seen": 396558336 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044439317953861587, + "loss": 1.397, + "theoretical_loss": 4.015759071190443, + "tokens_seen": 396623872 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044438314944834505, + "loss": 1.4684, + "theoretical_loss": 4.015684736452712, + "tokens_seen": 396689408 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044437311935807423, + "loss": 1.6272, + "theoretical_loss": 4.015610417432544, + "tokens_seen": 396754944 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044436308926780346, + "loss": 1.3573, + "theoretical_loss": 4.0155361141240205, + "tokens_seen": 396820480 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004443530591775326, + "loss": 1.4175, + "theoretical_loss": 4.015461826521225, + "tokens_seen": 396886016 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004443430290872618, + "loss": 1.57, + "theoretical_loss": 4.015387554618244, + "tokens_seen": 396951552 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044433299899699095, + "loss": 1.4745, + "theoretical_loss": 4.01531329840917, + "tokens_seen": 397017088 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004443229689067202, + "loss": 1.5254, + "theoretical_loss": 4.015239057888097, + "tokens_seen": 397082624 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044431293881644937, + "loss": 1.5522, + "theoretical_loss": 4.015164833049122, + "tokens_seen": 397148160 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044430290872617855, + "loss": 1.4579, + "theoretical_loss": 4.015090623886345, + "tokens_seen": 397213696 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044429287863590773, + "loss": 1.3495, + "theoretical_loss": 4.01501643039387, + "tokens_seen": 397279232 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044428284854563697, + "loss": 1.4839, + "theoretical_loss": 4.014942252565803, + "tokens_seen": 397344768 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004442728184553661, + "loss": 1.5372, + "theoretical_loss": 4.014868090396256, + "tokens_seen": 397410304 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044426278836509533, + "loss": 1.4043, + "theoretical_loss": 4.01479394387934, + "tokens_seen": 397475840 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044425275827482446, + "loss": 1.4848, + "theoretical_loss": 4.014719813009172, + "tokens_seen": 397541376 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004442427281845537, + "loss": 1.4539, + "theoretical_loss": 4.014645697779872, + "tokens_seen": 397606912 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044423269809428287, + "loss": 1.4819, + "theoretical_loss": 4.014571598185563, + "tokens_seen": 397672448 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044422266800401205, + "loss": 1.426, + "theoretical_loss": 4.01449751422037, + "tokens_seen": 397737984 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044421263791374123, + "loss": 1.3869, + "theoretical_loss": 4.014423445878423, + "tokens_seen": 397803520 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004442026078234704, + "loss": 1.4963, + "theoretical_loss": 4.014349393153853, + "tokens_seen": 397869056 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004441925777331996, + "loss": 1.4296, + "theoretical_loss": 4.014275356040796, + "tokens_seen": 397934592 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044418254764292883, + "loss": 1.5145, + "theoretical_loss": 4.014201334533391, + "tokens_seen": 398000128 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044417251755265796, + "loss": 1.4155, + "theoretical_loss": 4.01412732862578, + "tokens_seen": 398065664 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -0.0003527200606185943, + "objective/train/docs_used": 236468, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.713815689086914, + "objective/train/original_loss": 2.713815450668335, + "objective/train/theoretical_loss": 4.014053338312108, + "objective/train/tokens_used": 418591200, + "objective/train/value_avg": -0.00417327880859375, + "objective/train/value_loss": 4.621346670319326e-05, + "objective/train/value_max": -0.00020992755889892578, + "objective/train/value_min": -0.0455322265625, + "objective/train/value_reward_corr": 0.32374795591880434, + "objective/train/value_std": 0.0031681060791015625, + "objective/train/weight_avg": 0.9986777305603027, + "objective/train/weighted_lm_loss": 2.6969707012176514, + "objective/train/weights_max": 1.4672352075576782, + "objective/train/weights_min": 0.5757007002830505, + "theoretical_loss": 4.014053338312108, + "tokens_seen": 398131200 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004441624874623872, + "loss": 1.4587, + "theoretical_loss": 4.014053338312108, + "tokens_seen": 398131200 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004441524573721163, + "loss": 1.3578, + "theoretical_loss": 4.013979363586522, + "tokens_seen": 398196736 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044414242728184556, + "loss": 1.4939, + "theoretical_loss": 4.013905404443173, + "tokens_seen": 398262272 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044413239719157474, + "loss": 1.5424, + "theoretical_loss": 4.013831460876216, + "tokens_seen": 398327808 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004441223671013039, + "loss": 1.4185, + "theoretical_loss": 4.01375753287981, + "tokens_seen": 398393344 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004441123370110331, + "loss": 1.5747, + "theoretical_loss": 4.013683620448113, + "tokens_seen": 398458880 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044410230692076234, + "loss": 1.4939, + "theoretical_loss": 4.013609723575291, + "tokens_seen": 398524416 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044409227683049146, + "loss": 1.5267, + "theoretical_loss": 4.01353584225551, + "tokens_seen": 398589952 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004440822467402207, + "loss": 1.5809, + "theoretical_loss": 4.01346197648294, + "tokens_seen": 398655488 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004440722166499498, + "loss": 1.4378, + "theoretical_loss": 4.013388126251755, + "tokens_seen": 398721024 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044406218655967906, + "loss": 1.4892, + "theoretical_loss": 4.013314291556131, + "tokens_seen": 398786560 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044405215646940824, + "loss": 1.5828, + "theoretical_loss": 4.0132404723902475, + "tokens_seen": 398852096 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004440421263791374, + "loss": 1.5405, + "theoretical_loss": 4.013166668748288, + "tokens_seen": 398917632 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004440320962888666, + "loss": 1.5096, + "theoretical_loss": 4.013092880624437, + "tokens_seen": 398983168 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004440220661985958, + "loss": 1.4716, + "theoretical_loss": 4.013019108012884, + "tokens_seen": 399048704 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044401203610832497, + "loss": 1.5513, + "theoretical_loss": 4.012945350907822, + "tokens_seen": 399114240 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004440020060180542, + "loss": 1.5498, + "theoretical_loss": 4.012871609303445, + "tokens_seen": 399179776 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044399197592778333, + "loss": 1.4754, + "theoretical_loss": 4.012797883193952, + "tokens_seen": 399245312 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044398194583751256, + "loss": 1.5539, + "theoretical_loss": 4.012724172573543, + "tokens_seen": 399310848 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004439719157472417, + "loss": 1.547, + "theoretical_loss": 4.012650477436425, + "tokens_seen": 399376384 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004439618856569709, + "loss": 1.5386, + "theoretical_loss": 4.012576797776805, + "tokens_seen": 399441920 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004439518555667001, + "loss": 1.5573, + "theoretical_loss": 4.0125031335888925, + "tokens_seen": 399507456 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004439418254764293, + "loss": 1.519, + "theoretical_loss": 4.012429484866903, + "tokens_seen": 399572992 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044393179538615847, + "loss": 1.5127, + "theoretical_loss": 4.012355851605053, + "tokens_seen": 399638528 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004439217652958877, + "loss": 1.522, + "theoretical_loss": 4.012282233797562, + "tokens_seen": 399704064 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -6.486782513093203e-05, + "objective/train/docs_used": 237759, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6512551307678223, + "objective/train/original_loss": 2.651254653930664, + "objective/train/theoretical_loss": 4.0122086314386545, + "objective/train/tokens_used": 420229600, + "objective/train/value_avg": -0.00534820556640625, + "objective/train/value_loss": 5.744080408476293e-05, + "objective/train/value_max": -0.00046181678771972656, + "objective/train/value_min": -0.046295166015625, + "objective/train/value_reward_corr": 0.30281319423724906, + "objective/train/value_std": 0.0037212371826171875, + "objective/train/weight_avg": 1.002036452293396, + "objective/train/weighted_lm_loss": 2.671934127807617, + "objective/train/weights_max": 1.395751714706421, + "objective/train/weights_min": 0.35512202978134155, + "theoretical_loss": 4.0122086314386545, + "tokens_seen": 399769600 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044391173520561683, + "loss": 1.5058, + "theoretical_loss": 4.0122086314386545, + "tokens_seen": 399769600 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044390170511534607, + "loss": 1.474, + "theoretical_loss": 4.012135044522555, + "tokens_seen": 399835136 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004438916750250752, + "loss": 1.4802, + "theoretical_loss": 4.012061473043495, + "tokens_seen": 399900672 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044388164493480443, + "loss": 1.4051, + "theoretical_loss": 4.0119879169957064, + "tokens_seen": 399966208 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004438716148445336, + "loss": 1.4605, + "theoretical_loss": 4.011914376373426, + "tokens_seen": 400031744 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004438615847542628, + "loss": 1.5125, + "theoretical_loss": 4.01184085117089, + "tokens_seen": 400097280 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044385155466399197, + "loss": 1.5036, + "theoretical_loss": 4.011767341382341, + "tokens_seen": 400162816 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044384152457372115, + "loss": 1.4767, + "theoretical_loss": 4.011693847002025, + "tokens_seen": 400228352 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044383149448345033, + "loss": 1.5164, + "theoretical_loss": 4.011620368024191, + "tokens_seen": 400293888 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044382146439317957, + "loss": 1.3591, + "theoretical_loss": 4.011546904443088, + "tokens_seen": 400359424 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004438114343029087, + "loss": 1.405, + "theoretical_loss": 4.011473456252971, + "tokens_seen": 400424960 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044380140421263793, + "loss": 1.4162, + "theoretical_loss": 4.011400023448097, + "tokens_seen": 400490496 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004437913741223671, + "loss": 1.4185, + "theoretical_loss": 4.0113266060227275, + "tokens_seen": 400556032 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004437813440320963, + "loss": 1.4183, + "theoretical_loss": 4.011253203971124, + "tokens_seen": 400621568 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004437713139418255, + "loss": 1.4594, + "theoretical_loss": 4.011179817287556, + "tokens_seen": 400687104 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044376128385155466, + "loss": 1.4066, + "theoretical_loss": 4.011106445966291, + "tokens_seen": 400752640 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044375125376128384, + "loss": 1.5074, + "theoretical_loss": 4.011033090001602, + "tokens_seen": 400818176 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044374122367101307, + "loss": 1.5238, + "theoretical_loss": 4.010959749387765, + "tokens_seen": 400883712 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004437311935807422, + "loss": 1.5762, + "theoretical_loss": 4.010886424119059, + "tokens_seen": 400949248 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044372116349047143, + "loss": 1.4356, + "theoretical_loss": 4.010813114189765, + "tokens_seen": 401014784 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044371113340020056, + "loss": 1.2872, + "theoretical_loss": 4.0107398195941695, + "tokens_seen": 401080320 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004437011033099298, + "loss": 1.523, + "theoretical_loss": 4.010666540326559, + "tokens_seen": 401145856 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044369107321965903, + "loss": 1.4935, + "theoretical_loss": 4.010593276381226, + "tokens_seen": 401211392 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044368104312938816, + "loss": 1.4595, + "theoretical_loss": 4.010520027752464, + "tokens_seen": 401276928 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004436710130391174, + "loss": 1.5634, + "theoretical_loss": 4.01044679443457, + "tokens_seen": 401342464 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": 0.0006775629590265453, + "objective/train/docs_used": 238664, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.628634452819824, + "objective/train/original_loss": 2.628633975982666, + "objective/train/theoretical_loss": 4.010373576421845, + "objective/train/tokens_used": 421868000, + "objective/train/value_avg": -0.00461578369140625, + "objective/train/value_loss": 3.588988329283893e-05, + "objective/train/value_max": -0.0002758502960205078, + "objective/train/value_min": -0.056976318359375, + "objective/train/value_reward_corr": 0.26017803181238686, + "objective/train/value_std": 0.0037860870361328125, + "objective/train/weight_avg": 1.0085453987121582, + "objective/train/weighted_lm_loss": 2.6612331867218018, + "objective/train/weights_max": 1.579966425895691, + "objective/train/weights_min": 0.5493696331977844, + "theoretical_loss": 4.010373576421845, + "tokens_seen": 401408000 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004436609829488465, + "loss": 1.4812, + "theoretical_loss": 4.010373576421845, + "tokens_seen": 401408000 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044365095285857576, + "loss": 1.5446, + "theoretical_loss": 4.010300373708591, + "tokens_seen": 401473536 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044364092276830494, + "loss": 1.4687, + "theoretical_loss": 4.010227186289116, + "tokens_seen": 401539072 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004436308926780341, + "loss": 1.3273, + "theoretical_loss": 4.010154014157727, + "tokens_seen": 401604608 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004436208625877633, + "loss": 1.4832, + "theoretical_loss": 4.010080857308741, + "tokens_seen": 401670144 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044361083249749254, + "loss": 1.53, + "theoretical_loss": 4.010007715736468, + "tokens_seen": 401735680 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044360080240722166, + "loss": 1.4213, + "theoretical_loss": 4.009934589435231, + "tokens_seen": 401801216 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004435907723169509, + "loss": 1.4884, + "theoretical_loss": 4.009861478399348, + "tokens_seen": 401866752 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044358074222668, + "loss": 1.4817, + "theoretical_loss": 4.009788382623146, + "tokens_seen": 401932288 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044357071213640926, + "loss": 1.3995, + "theoretical_loss": 4.009715302100952, + "tokens_seen": 401997824 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044356068204613844, + "loss": 1.3284, + "theoretical_loss": 4.009642236827098, + "tokens_seen": 402063360 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004435506519558676, + "loss": 1.5042, + "theoretical_loss": 4.009569186795915, + "tokens_seen": 402128896 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004435406218655968, + "loss": 1.6535, + "theoretical_loss": 4.009496152001741, + "tokens_seen": 402194432 + }, + { + "epoch": 0.12, + "learning_rate": 0.000443530591775326, + "loss": 1.3842, + "theoretical_loss": 4.009423132438918, + "tokens_seen": 402259968 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044352056168505517, + "loss": 1.5378, + "theoretical_loss": 4.009350128101785, + "tokens_seen": 402325504 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004435105315947844, + "loss": 1.4667, + "theoretical_loss": 4.00927713898469, + "tokens_seen": 402391040 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044350050150451353, + "loss": 1.5721, + "theoretical_loss": 4.009204165081981, + "tokens_seen": 402456576 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044349047141424276, + "loss": 1.5237, + "theoretical_loss": 4.009131206388012, + "tokens_seen": 402522112 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004434804413239719, + "loss": 1.5108, + "theoretical_loss": 4.009058262897135, + "tokens_seen": 402587648 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004434704112337011, + "loss": 1.517, + "theoretical_loss": 4.008985334603709, + "tokens_seen": 402653184 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004434603811434303, + "loss": 1.4908, + "theoretical_loss": 4.008912421502095, + "tokens_seen": 402718720 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004434503510531595, + "loss": 1.4062, + "theoretical_loss": 4.008839523586658, + "tokens_seen": 402784256 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044344032096288867, + "loss": 1.4944, + "theoretical_loss": 4.008766640851763, + "tokens_seen": 402849792 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004434302908726179, + "loss": 1.4439, + "theoretical_loss": 4.008693773291781, + "tokens_seen": 402915328 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044342026078234703, + "loss": 1.4008, + "theoretical_loss": 4.008620920901085, + "tokens_seen": 402980864 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": 0.0005252551054582, + "objective/train/docs_used": 240054, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.535659074783325, + "objective/train/original_loss": 3.535659074783325, + "objective/train/theoretical_loss": 4.00854808367405, + "objective/train/tokens_used": 423506400, + "objective/train/value_avg": -0.005161285400390625, + "objective/train/value_loss": 7.518201164202765e-05, + "objective/train/value_max": -0.0003006458282470703, + "objective/train/value_min": -0.06927490234375, + "objective/train/value_reward_corr": 0.22905216673565526, + "objective/train/value_std": 0.004352569580078125, + "objective/train/weight_avg": 1.0086153745651245, + "objective/train/weighted_lm_loss": 3.5756375789642334, + "objective/train/weights_max": 1.8239939212799072, + "objective/train/weights_min": 0.13821791112422943, + "theoretical_loss": 4.00854808367405, + "tokens_seen": 403046400 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044341023069207627, + "loss": 1.5338, + "theoretical_loss": 4.00854808367405, + "tokens_seen": 403046400 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004434002006018054, + "loss": 1.4182, + "theoretical_loss": 4.0084752616050565, + "tokens_seen": 403111936 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044339017051153463, + "loss": 1.3607, + "theoretical_loss": 4.008402454688485, + "tokens_seen": 403177472 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004433801404212638, + "loss": 1.5411, + "theoretical_loss": 4.008329662918721, + "tokens_seen": 403243008 + }, + { + "epoch": 0.12, + "learning_rate": 0.000443370110330993, + "loss": 1.5844, + "theoretical_loss": 4.008256886290153, + "tokens_seen": 403308544 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044336008024072217, + "loss": 1.5086, + "theoretical_loss": 4.00818412479717, + "tokens_seen": 403374080 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044335005015045135, + "loss": 1.4865, + "theoretical_loss": 4.008111378434169, + "tokens_seen": 403439616 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044334002006018053, + "loss": 1.4343, + "theoretical_loss": 4.008038647195543, + "tokens_seen": 403505152 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044332998996990977, + "loss": 1.3989, + "theoretical_loss": 4.007965931075695, + "tokens_seen": 403570688 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004433199598796389, + "loss": 1.472, + "theoretical_loss": 4.007893230069027, + "tokens_seen": 403636224 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044330992978936813, + "loss": 1.5315, + "theoretical_loss": 4.007820544169944, + "tokens_seen": 403701760 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004432998996990973, + "loss": 1.5934, + "theoretical_loss": 4.007747873372857, + "tokens_seen": 403767296 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004432898696088265, + "loss": 1.5, + "theoretical_loss": 4.007675217672174, + "tokens_seen": 403832832 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004432798395185557, + "loss": 1.4836, + "theoretical_loss": 4.007602577062314, + "tokens_seen": 403898368 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044326980942828486, + "loss": 1.5811, + "theoretical_loss": 4.007529951537692, + "tokens_seen": 403963904 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044325977933801404, + "loss": 1.3709, + "theoretical_loss": 4.00745734109273, + "tokens_seen": 404029440 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044324974924774327, + "loss": 1.4177, + "theoretical_loss": 4.0073847457218505, + "tokens_seen": 404094976 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004432397191574724, + "loss": 1.5147, + "theoretical_loss": 4.0073121654194805, + "tokens_seen": 404160512 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044322968906720163, + "loss": 1.4777, + "theoretical_loss": 4.00723960018005, + "tokens_seen": 404226048 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044321965897693076, + "loss": 1.5082, + "theoretical_loss": 4.007167049997992, + "tokens_seen": 404291584 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044320962888666, + "loss": 1.4213, + "theoretical_loss": 4.007094514867741, + "tokens_seen": 404357120 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004431995987963892, + "loss": 1.427, + "theoretical_loss": 4.007021994783736, + "tokens_seen": 404422656 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044318956870611836, + "loss": 1.5249, + "theoretical_loss": 4.0069494897404185, + "tokens_seen": 404488192 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044317953861584754, + "loss": 1.2577, + "theoretical_loss": 4.006876999732232, + "tokens_seen": 404553728 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004431695085255767, + "loss": 1.5347, + "theoretical_loss": 4.006804524753626, + "tokens_seen": 404619264 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": 0.000570613075979054, + "objective/train/docs_used": 240788, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6897971630096436, + "objective/train/original_loss": 2.6897976398468018, + "objective/train/theoretical_loss": 4.006732064799048, + "objective/train/tokens_used": 425144800, + "objective/train/value_avg": -0.00501251220703125, + "objective/train/value_loss": 5.652771142194979e-05, + "objective/train/value_max": -0.0003981590270996094, + "objective/train/value_min": -0.0548095703125, + "objective/train/value_reward_corr": 0.45145978307273876, + "objective/train/value_std": 0.004207611083984375, + "objective/train/weight_avg": 1.0083426237106323, + "objective/train/weighted_lm_loss": 2.7098772525787354, + "objective/train/weights_max": 1.7299555540084839, + "objective/train/weights_min": 0.44477033615112305, + "theoretical_loss": 4.006732064799048, + "tokens_seen": 404684800 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004431594784353059, + "loss": 1.4811, + "theoretical_loss": 4.006732064799048, + "tokens_seen": 404684800 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044314944834503514, + "loss": 1.544, + "theoretical_loss": 4.006659619862954, + "tokens_seen": 404750336 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044313941825476427, + "loss": 1.4183, + "theoretical_loss": 4.006587189939798, + "tokens_seen": 404815872 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004431293881644935, + "loss": 1.4634, + "theoretical_loss": 4.00651477502404, + "tokens_seen": 404881408 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004431193580742227, + "loss": 1.3195, + "theoretical_loss": 4.006442375110142, + "tokens_seen": 404946944 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044310932798395186, + "loss": 1.4239, + "theoretical_loss": 4.006369990192569, + "tokens_seen": 405012480 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044309929789368104, + "loss": 1.4157, + "theoretical_loss": 4.00629762026579, + "tokens_seen": 405078016 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004430892678034102, + "loss": 1.5284, + "theoretical_loss": 4.006225265324274, + "tokens_seen": 405143552 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004430792377131394, + "loss": 1.4656, + "theoretical_loss": 4.006152925362496, + "tokens_seen": 405209088 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044306920762286864, + "loss": 1.5027, + "theoretical_loss": 4.006080600374932, + "tokens_seen": 405274624 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044305917753259777, + "loss": 1.5017, + "theoretical_loss": 4.006008290356062, + "tokens_seen": 405340160 + }, + { + "epoch": 0.12, + "learning_rate": 0.000443049147442327, + "loss": 1.4814, + "theoretical_loss": 4.005935995300369, + "tokens_seen": 405405696 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044303911735205613, + "loss": 1.4771, + "theoretical_loss": 4.005863715202339, + "tokens_seen": 405471232 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044302908726178537, + "loss": 1.6085, + "theoretical_loss": 4.005791450056459, + "tokens_seen": 405536768 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044301905717151455, + "loss": 1.4844, + "theoretical_loss": 4.0057191998572215, + "tokens_seen": 405602304 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044300902708124373, + "loss": 1.4688, + "theoretical_loss": 4.005646964599119, + "tokens_seen": 405667840 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004429989969909729, + "loss": 1.555, + "theoretical_loss": 4.005574744276652, + "tokens_seen": 405733376 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004429889669007021, + "loss": 1.4378, + "theoretical_loss": 4.0055025388843175, + "tokens_seen": 405798912 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044297893681043127, + "loss": 1.4779, + "theoretical_loss": 4.0054303484166205, + "tokens_seen": 405864448 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004429689067201605, + "loss": 1.4433, + "theoretical_loss": 4.005358172868066, + "tokens_seen": 405929984 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044295887662988963, + "loss": 1.3939, + "theoretical_loss": 4.005286012233162, + "tokens_seen": 405995520 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044294884653961887, + "loss": 1.6191, + "theoretical_loss": 4.0052138665064225, + "tokens_seen": 406061056 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004429388164493481, + "loss": 1.4812, + "theoretical_loss": 4.00514173568236, + "tokens_seen": 406126592 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044292878635907723, + "loss": 1.4608, + "theoretical_loss": 4.005069619755494, + "tokens_seen": 406192128 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044291875626880647, + "loss": 1.35, + "theoretical_loss": 4.004997518720344, + "tokens_seen": 406257664 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -0.00021204986842349172, + "objective/train/docs_used": 241796, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0945961475372314, + "objective/train/original_loss": 3.0945956707000732, + "objective/train/theoretical_loss": 4.004925432571433, + "objective/train/tokens_used": 426783200, + "objective/train/value_avg": -0.004871368408203125, + "objective/train/value_loss": 4.94729210913647e-05, + "objective/train/value_max": -0.00047659873962402344, + "objective/train/value_min": -0.04779052734375, + "objective/train/value_reward_corr": 0.22430146335702664, + "objective/train/value_std": 0.0037097930908203125, + "objective/train/weight_avg": 1.0001972913742065, + "objective/train/weighted_lm_loss": 3.103844165802002, + "objective/train/weights_max": 1.5591561794281006, + "objective/train/weights_min": 0.14896823465824127, + "theoretical_loss": 4.004925432571433, + "tokens_seen": 406323200 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004429087261785356, + "loss": 1.3961, + "theoretical_loss": 4.004925432571433, + "tokens_seen": 406323200 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044289869608826483, + "loss": 1.3177, + "theoretical_loss": 4.004853361303288, + "tokens_seen": 406388736 + }, + { + "epoch": 0.12, + "learning_rate": 0.000442888665997994, + "loss": 1.3951, + "theoretical_loss": 4.004781304910438, + "tokens_seen": 406454272 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004428786359077232, + "loss": 1.3695, + "theoretical_loss": 4.004709263387414, + "tokens_seen": 406519808 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044286860581745237, + "loss": 1.4156, + "theoretical_loss": 4.004637236728753, + "tokens_seen": 406585344 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044285857572718155, + "loss": 1.4961, + "theoretical_loss": 4.0045652249289905, + "tokens_seen": 406650880 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044284854563691073, + "loss": 1.3931, + "theoretical_loss": 4.004493227982669, + "tokens_seen": 406716416 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044283851554663997, + "loss": 1.537, + "theoretical_loss": 4.004421245884332, + "tokens_seen": 406781952 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004428284854563691, + "loss": 1.4813, + "theoretical_loss": 4.004349278628525, + "tokens_seen": 406847488 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044281845536609833, + "loss": 1.5286, + "theoretical_loss": 4.004277326209799, + "tokens_seen": 406913024 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004428084252758275, + "loss": 1.599, + "theoretical_loss": 4.004205388622704, + "tokens_seen": 406978560 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004427983951855567, + "loss": 1.5118, + "theoretical_loss": 4.004133465861796, + "tokens_seen": 407044096 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004427883650952859, + "loss": 1.5107, + "theoretical_loss": 4.004061557921634, + "tokens_seen": 407109632 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044277833500501506, + "loss": 1.5245, + "theoretical_loss": 4.003989664796777, + "tokens_seen": 407175168 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044276830491474424, + "loss": 1.5343, + "theoretical_loss": 4.003917786481791, + "tokens_seen": 407240704 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044275827482447347, + "loss": 1.4003, + "theoretical_loss": 4.003845922971243, + "tokens_seen": 407306240 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004427482447342026, + "loss": 1.4971, + "theoretical_loss": 4.003774074259699, + "tokens_seen": 407371776 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044273821464393184, + "loss": 1.5958, + "theoretical_loss": 4.003702240341733, + "tokens_seen": 407437312 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044272818455366096, + "loss": 1.4494, + "theoretical_loss": 4.003630421211922, + "tokens_seen": 407502848 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004427181544633902, + "loss": 1.4502, + "theoretical_loss": 4.003558616864842, + "tokens_seen": 407568384 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004427081243731194, + "loss": 1.5009, + "theoretical_loss": 4.003486827295075, + "tokens_seen": 407633920 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044269809428284856, + "loss": 1.5054, + "theoretical_loss": 4.003415052497203, + "tokens_seen": 407699456 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044268806419257774, + "loss": 1.5229, + "theoretical_loss": 4.003343292465816, + "tokens_seen": 407764992 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004426780341023069, + "loss": 1.5041, + "theoretical_loss": 4.0032715471955, + "tokens_seen": 407830528 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004426680040120361, + "loss": 1.464, + "theoretical_loss": 4.00319981668085, + "tokens_seen": 407896064 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": -0.0014560339041054249, + "objective/train/docs_used": 242638, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.885563611984253, + "objective/train/original_loss": 2.885564088821411, + "objective/train/theoretical_loss": 4.003128100916459, + "objective/train/tokens_used": 428421600, + "objective/train/value_avg": -0.004669189453125, + "objective/train/value_loss": 7.802848267601803e-05, + "objective/train/value_max": -0.000537872314453125, + "objective/train/value_min": -0.0738525390625, + "objective/train/value_reward_corr": 0.31417825630343593, + "objective/train/value_std": 0.004001617431640625, + "objective/train/weight_avg": 0.9890123009681702, + "objective/train/weighted_lm_loss": 2.8594555854797363, + "objective/train/weights_max": 1.9697749614715576, + "objective/train/weights_min": 0.31361156702041626, + "theoretical_loss": 4.003128100916459, + "tokens_seen": 407961600 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044265797392176534, + "loss": 1.4494, + "theoretical_loss": 4.003128100916459, + "tokens_seen": 407961600 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044264794383149447, + "loss": 1.4712, + "theoretical_loss": 4.003056399896928, + "tokens_seen": 408027136 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004426379137412237, + "loss": 1.4721, + "theoretical_loss": 4.002984713616855, + "tokens_seen": 408092672 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004426278836509529, + "loss": 1.4427, + "theoretical_loss": 4.002913042070844, + "tokens_seen": 408158208 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044261785356068206, + "loss": 1.5054, + "theoretical_loss": 4.002841385253504, + "tokens_seen": 408223744 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044260782347041124, + "loss": 1.5579, + "theoretical_loss": 4.002769743159441, + "tokens_seen": 408289280 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004425977933801404, + "loss": 1.4906, + "theoretical_loss": 4.00269811578327, + "tokens_seen": 408354816 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004425877632898696, + "loss": 1.4899, + "theoretical_loss": 4.002626503119606, + "tokens_seen": 408420352 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044257773319959884, + "loss": 1.3913, + "theoretical_loss": 4.0025549051630644, + "tokens_seen": 408485888 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044256770310932797, + "loss": 1.4087, + "theoretical_loss": 4.002483321908269, + "tokens_seen": 408551424 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004425576730190572, + "loss": 1.6613, + "theoretical_loss": 4.002411753349842, + "tokens_seen": 408616960 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044254764292878633, + "loss": 1.5636, + "theoretical_loss": 4.002340199482409, + "tokens_seen": 408682496 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044253761283851557, + "loss": 1.4201, + "theoretical_loss": 4.0022686603006, + "tokens_seen": 408748032 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044252758274824475, + "loss": 1.5524, + "theoretical_loss": 4.002197135799047, + "tokens_seen": 408813568 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044251755265797393, + "loss": 1.4946, + "theoretical_loss": 4.002125625972386, + "tokens_seen": 408879104 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004425075225677031, + "loss": 1.4735, + "theoretical_loss": 4.002054130815253, + "tokens_seen": 408944640 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004424974924774323, + "loss": 1.5509, + "theoretical_loss": 4.00198265032229, + "tokens_seen": 409010176 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044248746238716147, + "loss": 1.5833, + "theoretical_loss": 4.001911184488139, + "tokens_seen": 409075712 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004424774322968907, + "loss": 1.4719, + "theoretical_loss": 4.001839733307447, + "tokens_seen": 409141248 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044246740220661983, + "loss": 1.4915, + "theoretical_loss": 4.001768296774863, + "tokens_seen": 409206784 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044245737211634907, + "loss": 1.4806, + "theoretical_loss": 4.001696874885038, + "tokens_seen": 409272320 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044244734202607825, + "loss": 1.4396, + "theoretical_loss": 4.001625467632628, + "tokens_seen": 409337856 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044243731193580743, + "loss": 1.5005, + "theoretical_loss": 4.001554075012289, + "tokens_seen": 409403392 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004424272818455366, + "loss": 1.4159, + "theoretical_loss": 4.001482697018682, + "tokens_seen": 409468928 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004424172517552658, + "loss": 1.4333, + "theoretical_loss": 4.00141133364647, + "tokens_seen": 409534464 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": 0.00048485121806152165, + "objective/train/docs_used": 244086, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6581313610076904, + "objective/train/original_loss": 2.6581311225891113, + "objective/train/theoretical_loss": 4.0013399848903175, + "objective/train/tokens_used": 430060000, + "objective/train/value_avg": -0.004207611083984375, + "objective/train/value_loss": 3.565804217942059e-05, + "objective/train/value_max": -0.0004239082336425781, + "objective/train/value_min": -0.03662109375, + "objective/train/value_reward_corr": 0.1515100942118567, + "objective/train/value_std": 0.0027866363525390625, + "objective/train/weight_avg": 1.0065507888793945, + "objective/train/weighted_lm_loss": 2.6922719478607178, + "objective/train/weights_max": 1.337540626525879, + "objective/train/weights_min": 0.41346222162246704, + "theoretical_loss": 4.0013399848903175, + "tokens_seen": 409600000 + }, + { + "epoch": 0.12, + "learning_rate": 0.000442407221664995, + "loss": 1.4963, + "theoretical_loss": 4.0013399848903175, + "tokens_seen": 409600000 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004423971915747242, + "loss": 1.5484, + "theoretical_loss": 4.001268650744895, + "tokens_seen": 409665536 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044238716148445334, + "loss": 1.4762, + "theoretical_loss": 4.001197331204873, + "tokens_seen": 409731072 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044237713139418257, + "loss": 1.4875, + "theoretical_loss": 4.0011260262649255, + "tokens_seen": 409796608 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004423671013039117, + "loss": 1.5269, + "theoretical_loss": 4.00105473591973, + "tokens_seen": 409862144 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044235707121364093, + "loss": 1.5436, + "theoretical_loss": 4.000983460163966, + "tokens_seen": 409927680 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004423470411233701, + "loss": 1.5266, + "theoretical_loss": 4.000912198992316, + "tokens_seen": 409993216 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004423370110330993, + "loss": 1.528, + "theoretical_loss": 4.000840952399466, + "tokens_seen": 410058752 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004423269809428285, + "loss": 1.515, + "theoretical_loss": 4.000769720380103, + "tokens_seen": 410124288 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004423169508525577, + "loss": 1.5302, + "theoretical_loss": 4.000698502928919, + "tokens_seen": 410189824 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044230692076228684, + "loss": 1.5389, + "theoretical_loss": 4.000627300040607, + "tokens_seen": 410255360 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004422968906720161, + "loss": 1.5494, + "theoretical_loss": 4.000556111709864, + "tokens_seen": 410320896 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004422868605817452, + "loss": 1.5354, + "theoretical_loss": 4.00048493793139, + "tokens_seen": 410386432 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044227683049147444, + "loss": 1.6353, + "theoretical_loss": 4.000413778699884, + "tokens_seen": 410451968 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004422668004012036, + "loss": 1.5595, + "theoretical_loss": 4.000342634010054, + "tokens_seen": 410517504 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004422567703109328, + "loss": 1.6162, + "theoretical_loss": 4.000271503856607, + "tokens_seen": 410583040 + }, + { + "epoch": 0.12, + "learning_rate": 0.000442246740220662, + "loss": 1.5176, + "theoretical_loss": 4.000200388234251, + "tokens_seen": 410648576 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044223671013039116, + "loss": 1.4522, + "theoretical_loss": 4.000129287137703, + "tokens_seen": 410714112 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044222668004012034, + "loss": 1.3912, + "theoretical_loss": 4.000058200561677, + "tokens_seen": 410779648 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004422166499498496, + "loss": 1.4177, + "theoretical_loss": 3.99998712850089, + "tokens_seen": 410845184 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004422066198595787, + "loss": 1.5526, + "theoretical_loss": 3.999916070950066, + "tokens_seen": 410910720 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044219658976930794, + "loss": 1.4177, + "theoretical_loss": 3.9998450279039277, + "tokens_seen": 410976256 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004421865596790371, + "loss": 1.5635, + "theoretical_loss": 3.9997739993572035, + "tokens_seen": 411041792 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004421765295887663, + "loss": 1.5204, + "theoretical_loss": 3.999702985304621, + "tokens_seen": 411107328 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044216649949849554, + "loss": 1.5253, + "theoretical_loss": 3.999631985740914, + "tokens_seen": 411172864 + }, + { + "epoch": 0.12, + "objective/train/advantage_avg": 0.0010110235307365656, + "objective/train/docs_used": 244630, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2667436599731445, + "objective/train/original_loss": 3.2667438983917236, + "objective/train/theoretical_loss": 3.999561000660818, + "objective/train/tokens_used": 431698400, + "objective/train/value_avg": -0.00469970703125, + "objective/train/value_loss": 4.1160972614306957e-05, + "objective/train/value_max": -0.0003654956817626953, + "objective/train/value_min": -0.043853759765625, + "objective/train/value_reward_corr": 0.27760228403332676, + "objective/train/value_std": 0.003170013427734375, + "objective/train/weight_avg": 1.0121179819107056, + "objective/train/weighted_lm_loss": 3.302006721496582, + "objective/train/weights_max": 1.4617563486099243, + "objective/train/weights_min": 0.6111368536949158, + "theoretical_loss": 3.999561000660818, + "tokens_seen": 411238400 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044215646940822467, + "loss": 1.4465, + "theoretical_loss": 3.999561000660818, + "tokens_seen": 411238400 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004421464393179539, + "loss": 1.4473, + "theoretical_loss": 3.99949003005907, + "tokens_seen": 411303936 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004421364092276831, + "loss": 1.4792, + "theoretical_loss": 3.9994190739304116, + "tokens_seen": 411369472 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044212637913741226, + "loss": 1.4532, + "theoretical_loss": 3.999348132269586, + "tokens_seen": 411435008 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044211634904714144, + "loss": 1.5141, + "theoretical_loss": 3.9992772050713388, + "tokens_seen": 411500544 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004421063189568706, + "loss": 1.5379, + "theoretical_loss": 3.9992062923304195, + "tokens_seen": 411566080 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004420962888665998, + "loss": 1.4108, + "theoretical_loss": 3.9991353940415797, + "tokens_seen": 411631616 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044208625877632904, + "loss": 1.3266, + "theoretical_loss": 3.999064510199574, + "tokens_seen": 411697152 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044207622868605817, + "loss": 1.4586, + "theoretical_loss": 3.998993640799159, + "tokens_seen": 411762688 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004420661985957874, + "loss": 1.4004, + "theoretical_loss": 3.9989227858350955, + "tokens_seen": 411828224 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044205616850551653, + "loss": 1.5034, + "theoretical_loss": 3.998851945302146, + "tokens_seen": 411893760 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044204613841524577, + "loss": 1.6559, + "theoretical_loss": 3.998781119195075, + "tokens_seen": 411959296 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044203610832497495, + "loss": 1.4473, + "theoretical_loss": 3.998710307508651, + "tokens_seen": 412024832 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044202607823470413, + "loss": 1.3844, + "theoretical_loss": 3.9986395102376453, + "tokens_seen": 412090368 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004420160481444333, + "loss": 1.4845, + "theoretical_loss": 3.9985687273768313, + "tokens_seen": 412155904 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004420060180541625, + "loss": 1.5345, + "theoretical_loss": 3.998497958920985, + "tokens_seen": 412221440 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044199598796389167, + "loss": 1.4301, + "theoretical_loss": 3.9984272048648855, + "tokens_seen": 412286976 + }, + { + "epoch": 0.12, + "learning_rate": 0.0004419859578736209, + "loss": 1.3844, + "theoretical_loss": 3.9983564652033152, + "tokens_seen": 412352512 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044197592778335003, + "loss": 1.5333, + "theoretical_loss": 3.998285739931058, + "tokens_seen": 412418048 + }, + { + "epoch": 0.12, + "learning_rate": 0.00044196589769307927, + "loss": 1.5174, + "theoretical_loss": 3.9982150290429, + "tokens_seen": 412483584 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044195586760280845, + "loss": 1.3865, + "theoretical_loss": 3.998144332533633, + "tokens_seen": 412549120 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044194583751253763, + "loss": 1.5336, + "theoretical_loss": 3.9980736503980485, + "tokens_seen": 412614656 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004419358074222668, + "loss": 1.468, + "theoretical_loss": 3.9980029826309424, + "tokens_seen": 412680192 + }, + { + "epoch": 0.13, + "learning_rate": 0.000441925777331996, + "loss": 1.4991, + "theoretical_loss": 3.9979323292271123, + "tokens_seen": 412745728 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004419157472417252, + "loss": 1.3445, + "theoretical_loss": 3.9978616901813586, + "tokens_seen": 412811264 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": 0.0002631339884828776, + "objective/train/docs_used": 245807, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.337102174758911, + "objective/train/original_loss": 2.337102174758911, + "objective/train/theoretical_loss": 3.997791065488486, + "objective/train/tokens_used": 433336800, + "objective/train/value_avg": -0.004669189453125, + "objective/train/value_loss": 4.769657607539557e-05, + "objective/train/value_max": -0.0004189014434814453, + "objective/train/value_min": -0.1865234375, + "objective/train/value_reward_corr": 0.12397797838931811, + "objective/train/value_std": 0.003448486328125, + "objective/train/weight_avg": 1.0049933195114136, + "objective/train/weighted_lm_loss": 2.3363404273986816, + "objective/train/weights_max": 5.960977077484131, + "objective/train/weights_min": 0.41635072231292725, + "theoretical_loss": 3.997791065488486, + "tokens_seen": 412876800 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004419057171514544, + "loss": 1.3255, + "theoretical_loss": 3.997791065488486, + "tokens_seen": 412876800 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044189568706118354, + "loss": 1.4368, + "theoretical_loss": 3.997720455143299, + "tokens_seen": 412942336 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044188565697091277, + "loss": 1.4794, + "theoretical_loss": 3.9976498591406076, + "tokens_seen": 413007872 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004418756268806419, + "loss": 1.5191, + "theoretical_loss": 3.9975792774752232, + "tokens_seen": 413073408 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044186559679037113, + "loss": 1.4162, + "theoretical_loss": 3.99750871014196, + "tokens_seen": 413138944 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004418555667001003, + "loss": 1.4844, + "theoretical_loss": 3.997438157135635, + "tokens_seen": 413204480 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004418455366098295, + "loss": 1.4587, + "theoretical_loss": 3.997367618451068, + "tokens_seen": 413270016 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004418355065195587, + "loss": 1.5596, + "theoretical_loss": 3.99729709408308, + "tokens_seen": 413335552 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004418254764292879, + "loss": 1.5941, + "theoretical_loss": 3.9972265840264978, + "tokens_seen": 413401088 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044181544633901704, + "loss": 1.5024, + "theoretical_loss": 3.9971560882761485, + "tokens_seen": 413466624 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004418054162487463, + "loss": 1.5382, + "theoretical_loss": 3.9970856068268623, + "tokens_seen": 413532160 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004417953861584754, + "loss": 1.4993, + "theoretical_loss": 3.9970151396734726, + "tokens_seen": 413597696 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044178535606820464, + "loss": 1.5548, + "theoretical_loss": 3.9969446868108145, + "tokens_seen": 413663232 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004417753259779338, + "loss": 1.4475, + "theoretical_loss": 3.996874248233728, + "tokens_seen": 413728768 + }, + { + "epoch": 0.13, + "learning_rate": 0.000441765295887663, + "loss": 1.4721, + "theoretical_loss": 3.9968038239370527, + "tokens_seen": 413794304 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004417552657973922, + "loss": 1.5384, + "theoretical_loss": 3.9967334139156327, + "tokens_seen": 413859840 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044174523570712136, + "loss": 1.5047, + "theoretical_loss": 3.9966630181643152, + "tokens_seen": 413925376 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044173520561685054, + "loss": 1.5144, + "theoretical_loss": 3.996592636677949, + "tokens_seen": 413990912 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004417251755265798, + "loss": 1.5411, + "theoretical_loss": 3.9965222694513862, + "tokens_seen": 414056448 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004417151454363089, + "loss": 1.3239, + "theoretical_loss": 3.9964519164794803, + "tokens_seen": 414121984 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044170511534603814, + "loss": 1.4912, + "theoretical_loss": 3.9963815777570897, + "tokens_seen": 414187520 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044169508525576727, + "loss": 1.5676, + "theoretical_loss": 3.996311253279074, + "tokens_seen": 414253056 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004416850551654965, + "loss": 1.2881, + "theoretical_loss": 3.996240943040296, + "tokens_seen": 414318592 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004416750250752257, + "loss": 1.5457, + "theoretical_loss": 3.99617064703562, + "tokens_seen": 414384128 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044166499498495487, + "loss": 1.4344, + "theoretical_loss": 3.9961003652599145, + "tokens_seen": 414449664 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": 0.00010633603960741311, + "objective/train/docs_used": 246540, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9288761615753174, + "objective/train/original_loss": 2.9288759231567383, + "objective/train/theoretical_loss": 3.99603009770805, + "objective/train/tokens_used": 434975200, + "objective/train/value_avg": -0.00441741943359375, + "objective/train/value_loss": 4.7588840971002355e-05, + "objective/train/value_max": -0.0005397796630859375, + "objective/train/value_min": -0.054412841796875, + "objective/train/value_reward_corr": 0.1811657930635213, + "objective/train/value_std": 0.0028057098388671875, + "objective/train/weight_avg": 1.0032374858856201, + "objective/train/weighted_lm_loss": 2.937318801879883, + "objective/train/weights_max": 1.6604546308517456, + "objective/train/weights_min": 0.2448582947254181, + "theoretical_loss": 3.99603009770805, + "tokens_seen": 414515200 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044165496489468405, + "loss": 1.4852, + "theoretical_loss": 3.99603009770805, + "tokens_seen": 414515200 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004416449348044133, + "loss": 1.5115, + "theoretical_loss": 3.9959598443748994, + "tokens_seen": 414580736 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004416349047141424, + "loss": 1.5889, + "theoretical_loss": 3.9958896052553383, + "tokens_seen": 414646272 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044162487462387164, + "loss": 1.5081, + "theoretical_loss": 3.9958193803442468, + "tokens_seen": 414711808 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044161484453360077, + "loss": 1.6203, + "theoretical_loss": 3.995749169636504, + "tokens_seen": 414777344 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044160481444333, + "loss": 1.4334, + "theoretical_loss": 3.995678973126995, + "tokens_seen": 414842880 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004415947843530592, + "loss": 1.4395, + "theoretical_loss": 3.995608790810606, + "tokens_seen": 414908416 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044158475426278837, + "loss": 1.4214, + "theoretical_loss": 3.995538622682226, + "tokens_seen": 414973952 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044157472417251755, + "loss": 1.4431, + "theoretical_loss": 3.9954684687367465, + "tokens_seen": 415039488 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044156469408224673, + "loss": 1.5162, + "theoretical_loss": 3.9953983289690624, + "tokens_seen": 415105024 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004415546639919759, + "loss": 1.5648, + "theoretical_loss": 3.9953282033740702, + "tokens_seen": 415170560 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044154463390170515, + "loss": 1.3894, + "theoretical_loss": 3.99525809194667, + "tokens_seen": 415236096 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004415346038114343, + "loss": 1.4082, + "theoretical_loss": 3.995187994681764, + "tokens_seen": 415301632 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004415245737211635, + "loss": 1.4274, + "theoretical_loss": 3.9951179115742574, + "tokens_seen": 415367168 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044151454363089264, + "loss": 1.498, + "theoretical_loss": 3.995047842619057, + "tokens_seen": 415432704 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044150451354062187, + "loss": 1.4846, + "theoretical_loss": 3.9949777878110746, + "tokens_seen": 415498240 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044149448345035105, + "loss": 1.5368, + "theoretical_loss": 3.9949077471452212, + "tokens_seen": 415563776 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044148445336008023, + "loss": 1.4732, + "theoretical_loss": 3.994837720616414, + "tokens_seen": 415629312 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004414744232698094, + "loss": 1.4658, + "theoretical_loss": 3.9947677082195696, + "tokens_seen": 415694848 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044146439317953865, + "loss": 1.5275, + "theoretical_loss": 3.9946977099496097, + "tokens_seen": 415760384 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004414543630892678, + "loss": 1.5648, + "theoretical_loss": 3.9946277258014575, + "tokens_seen": 415825920 + }, + { + "epoch": 0.13, + "learning_rate": 0.000441444332998997, + "loss": 1.4861, + "theoretical_loss": 3.9945577557700394, + "tokens_seen": 415891456 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004414343029087262, + "loss": 1.4255, + "theoretical_loss": 3.9944877998502832, + "tokens_seen": 415956992 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004414242728184554, + "loss": 1.4379, + "theoretical_loss": 3.994417858037121, + "tokens_seen": 416022528 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004414142427281846, + "loss": 1.4428, + "theoretical_loss": 3.9943479303254854, + "tokens_seen": 416088064 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": 0.0004690277564805001, + "objective/train/docs_used": 247824, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.899172067642212, + "objective/train/original_loss": 2.89917254447937, + "objective/train/theoretical_loss": 3.9942780167103145, + "objective/train/tokens_used": 436613600, + "objective/train/value_avg": -0.004947662353515625, + "objective/train/value_loss": 5.957171742920764e-05, + "objective/train/value_max": -0.0005297660827636719, + "objective/train/value_min": -0.08819580078125, + "objective/train/value_reward_corr": 0.3099365111177818, + "objective/train/value_std": 0.003971099853515625, + "objective/train/weight_avg": 1.0074721574783325, + "objective/train/weighted_lm_loss": 2.9122955799102783, + "objective/train/weights_max": 2.1489920616149902, + "objective/train/weights_min": 0.30192214250564575, + "theoretical_loss": 3.9942780167103145, + "tokens_seen": 416153600 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044140421263791374, + "loss": 1.4922, + "theoretical_loss": 3.9942780167103145, + "tokens_seen": 416153600 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044139418254764297, + "loss": 1.5524, + "theoretical_loss": 3.9942081171865462, + "tokens_seen": 416219136 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004413841524573721, + "loss": 1.5841, + "theoretical_loss": 3.9941382317491225, + "tokens_seen": 416284672 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044137412236710133, + "loss": 1.4106, + "theoretical_loss": 3.9940683603929883, + "tokens_seen": 416350208 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004413640922768305, + "loss": 1.4615, + "theoretical_loss": 3.9939985031130902, + "tokens_seen": 416415744 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004413540621865597, + "loss": 1.4409, + "theoretical_loss": 3.993928659904377, + "tokens_seen": 416481280 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004413440320962889, + "loss": 1.5278, + "theoretical_loss": 3.993858830761802, + "tokens_seen": 416546816 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004413340020060181, + "loss": 1.4731, + "theoretical_loss": 3.9937890156803197, + "tokens_seen": 416612352 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044132397191574724, + "loss": 1.5557, + "theoretical_loss": 3.9937192146548863, + "tokens_seen": 416677888 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004413139418254765, + "loss": 1.4753, + "theoretical_loss": 3.993649427680464, + "tokens_seen": 416743424 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004413039117352056, + "loss": 1.5282, + "theoretical_loss": 3.993579654752013, + "tokens_seen": 416808960 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044129388164493484, + "loss": 1.5002, + "theoretical_loss": 3.9935098958645, + "tokens_seen": 416874496 + }, + { + "epoch": 0.13, + "learning_rate": 0.000441283851554664, + "loss": 1.5032, + "theoretical_loss": 3.993440151012892, + "tokens_seen": 416940032 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004412738214643932, + "loss": 1.4541, + "theoretical_loss": 3.99337042019216, + "tokens_seen": 417005568 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004412637913741224, + "loss": 1.5204, + "theoretical_loss": 3.993300703397276, + "tokens_seen": 417071104 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044125376128385156, + "loss": 1.4765, + "theoretical_loss": 3.9932310006232163, + "tokens_seen": 417136640 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044124373119358074, + "loss": 1.5268, + "theoretical_loss": 3.993161311864959, + "tokens_seen": 417202176 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044123370110331, + "loss": 1.4493, + "theoretical_loss": 3.9930916371174843, + "tokens_seen": 417267712 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004412236710130391, + "loss": 1.5821, + "theoretical_loss": 3.9930219763757755, + "tokens_seen": 417333248 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044121364092276834, + "loss": 1.5089, + "theoretical_loss": 3.9929523296348197, + "tokens_seen": 417398784 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044120361083249747, + "loss": 1.4058, + "theoretical_loss": 3.9928826968896036, + "tokens_seen": 417464320 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004411935807422267, + "loss": 1.5026, + "theoretical_loss": 3.9928130781351197, + "tokens_seen": 417529856 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004411835506519559, + "loss": 1.5328, + "theoretical_loss": 3.9927434733663603, + "tokens_seen": 417595392 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044117352056168507, + "loss": 1.5297, + "theoretical_loss": 3.992673882578323, + "tokens_seen": 417660928 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044116349047141425, + "loss": 1.3813, + "theoretical_loss": 3.9926043057660054, + "tokens_seen": 417726464 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -0.00014550569176208228, + "objective/train/docs_used": 248559, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.936624765396118, + "objective/train/original_loss": 2.93662428855896, + "objective/train/theoretical_loss": 3.992534742924409, + "objective/train/tokens_used": 438252000, + "objective/train/value_avg": -0.003932952880859375, + "objective/train/value_loss": 4.5411135943140835e-05, + "objective/train/value_max": -0.00037407875061035156, + "objective/train/value_min": -0.04302978515625, + "objective/train/value_reward_corr": 0.16269159032945799, + "objective/train/value_std": 0.002643585205078125, + "objective/train/weight_avg": 1.0006778240203857, + "objective/train/weighted_lm_loss": 2.9460673332214355, + "objective/train/weights_max": 1.477904200553894, + "objective/train/weights_min": 0.417002409696579, + "theoretical_loss": 3.992534742924409, + "tokens_seen": 417792000 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004411534603811435, + "loss": 1.3544, + "theoretical_loss": 3.992534742924409, + "tokens_seen": 417792000 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004411434302908726, + "loss": 1.4467, + "theoretical_loss": 3.9924651940485387, + "tokens_seen": 417857536 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044113340020060184, + "loss": 1.445, + "theoretical_loss": 3.9923956591334004, + "tokens_seen": 417923072 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044112337011033097, + "loss": 1.4501, + "theoretical_loss": 3.9923261381740023, + "tokens_seen": 417988608 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004411133400200602, + "loss": 1.4421, + "theoretical_loss": 3.9922566311653576, + "tokens_seen": 418054144 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004411033099297894, + "loss": 1.4092, + "theoretical_loss": 3.9921871381024787, + "tokens_seen": 418119680 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044109327983951857, + "loss": 1.4399, + "theoretical_loss": 3.9921176589803835, + "tokens_seen": 418185216 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044108324974924775, + "loss": 1.3632, + "theoretical_loss": 3.9920481937940915, + "tokens_seen": 418250752 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044107321965897693, + "loss": 1.485, + "theoretical_loss": 3.991978742538624, + "tokens_seen": 418316288 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004410631895687061, + "loss": 1.5285, + "theoretical_loss": 3.9919093052090058, + "tokens_seen": 418381824 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044105315947843535, + "loss": 1.5975, + "theoretical_loss": 3.9918398818002636, + "tokens_seen": 418447360 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004410431293881645, + "loss": 1.7507, + "theoretical_loss": 3.9917704723074268, + "tokens_seen": 418512896 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004410330992978937, + "loss": 1.4859, + "theoretical_loss": 3.9917010767255277, + "tokens_seen": 418578432 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044102306920762284, + "loss": 1.5928, + "theoretical_loss": 3.991631695049601, + "tokens_seen": 418643968 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044101303911735207, + "loss": 1.4297, + "theoretical_loss": 3.991562327274684, + "tokens_seen": 418709504 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044100300902708125, + "loss": 1.4933, + "theoretical_loss": 3.9914929733958164, + "tokens_seen": 418775040 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044099297893681043, + "loss": 1.4435, + "theoretical_loss": 3.99142363340804, + "tokens_seen": 418840576 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004409829488465396, + "loss": 1.4461, + "theoretical_loss": 3.9913543073063997, + "tokens_seen": 418906112 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044097291875626885, + "loss": 1.5208, + "theoretical_loss": 3.9912849950859437, + "tokens_seen": 418971648 + }, + { + "epoch": 0.13, + "learning_rate": 0.000440962888665998, + "loss": 1.5215, + "theoretical_loss": 3.9912156967417207, + "tokens_seen": 419037184 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004409528585757272, + "loss": 1.4025, + "theoretical_loss": 3.9911464122687845, + "tokens_seen": 419102720 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044094282848545634, + "loss": 1.3687, + "theoretical_loss": 3.9910771416621893, + "tokens_seen": 419168256 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004409327983951856, + "loss": 1.5598, + "theoretical_loss": 3.9910078849169923, + "tokens_seen": 419233792 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044092276830491476, + "loss": 1.4853, + "theoretical_loss": 3.9909386420282544, + "tokens_seen": 419299328 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044091273821464394, + "loss": 1.3359, + "theoretical_loss": 3.990869412991037, + "tokens_seen": 419364864 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -0.0006983123021200299, + "objective/train/docs_used": 249785, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.640608072280884, + "objective/train/original_loss": 2.640608072280884, + "objective/train/theoretical_loss": 3.9908001978004064, + "objective/train/tokens_used": 439890400, + "objective/train/value_avg": -0.005001068115234375, + "objective/train/value_loss": 6.452847446780652e-05, + "objective/train/value_max": -0.00043392181396484375, + "objective/train/value_min": -0.0816650390625, + "objective/train/value_reward_corr": 0.1579311386591273, + "objective/train/value_std": 0.003940582275390625, + "objective/train/weight_avg": 0.9960026741027832, + "objective/train/weighted_lm_loss": 2.6385746002197266, + "objective/train/weights_max": 2.1068882942199707, + "objective/train/weights_min": 0.34997791051864624, + "theoretical_loss": 3.9908001978004064, + "tokens_seen": 419430400 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004409027081243731, + "loss": 1.4072, + "theoretical_loss": 3.9908001978004064, + "tokens_seen": 419430400 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004408926780341023, + "loss": 1.4899, + "theoretical_loss": 3.99073099645143, + "tokens_seen": 419495936 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004408826479438315, + "loss": 1.5049, + "theoretical_loss": 3.990661808939178, + "tokens_seen": 419561472 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004408726178535607, + "loss": 1.4127, + "theoretical_loss": 3.9905926352587224, + "tokens_seen": 419627008 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044086258776328984, + "loss": 1.4481, + "theoretical_loss": 3.990523475405139, + "tokens_seen": 419692544 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004408525576730191, + "loss": 1.4005, + "theoretical_loss": 3.990454329373505, + "tokens_seen": 419758080 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004408425275827482, + "loss": 1.4482, + "theoretical_loss": 3.9903851971589015, + "tokens_seen": 419823616 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044083249749247744, + "loss": 1.5353, + "theoretical_loss": 3.990316078756411, + "tokens_seen": 419889152 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004408224674022066, + "loss": 1.4928, + "theoretical_loss": 3.990246974161119, + "tokens_seen": 419954688 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004408124373119358, + "loss": 1.4866, + "theoretical_loss": 3.9901778833681125, + "tokens_seen": 420020224 + }, + { + "epoch": 0.13, + "learning_rate": 0.000440802407221665, + "loss": 1.4393, + "theoretical_loss": 3.9901088063724828, + "tokens_seen": 420085760 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004407923771313942, + "loss": 1.4702, + "theoretical_loss": 3.990039743169322, + "tokens_seen": 420151296 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044078234704112335, + "loss": 1.4355, + "theoretical_loss": 3.989970693753726, + "tokens_seen": 420216832 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004407723169508526, + "loss": 1.5252, + "theoretical_loss": 3.989901658120792, + "tokens_seen": 420282368 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004407622868605817, + "loss": 1.5073, + "theoretical_loss": 3.9898326362656205, + "tokens_seen": 420347904 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044075225677031094, + "loss": 1.4998, + "theoretical_loss": 3.9897636281833155, + "tokens_seen": 420413440 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004407422266800401, + "loss": 1.4329, + "theoretical_loss": 3.989694633868981, + "tokens_seen": 420478976 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004407321965897693, + "loss": 1.484, + "theoretical_loss": 3.9896256533177255, + "tokens_seen": 420544512 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004407221664994985, + "loss": 1.6037, + "theoretical_loss": 3.9895566865246597, + "tokens_seen": 420610048 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044071213640922767, + "loss": 1.5408, + "theoretical_loss": 3.9894877334848955, + "tokens_seen": 420675584 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044070210631895685, + "loss": 1.3533, + "theoretical_loss": 3.989418794193549, + "tokens_seen": 420741120 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004406920762286861, + "loss": 1.5289, + "theoretical_loss": 3.9893498686457383, + "tokens_seen": 420806656 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044068204613841527, + "loss": 1.4289, + "theoretical_loss": 3.9892809568365823, + "tokens_seen": 420872192 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044067201604814445, + "loss": 1.5064, + "theoretical_loss": 3.989212058761206, + "tokens_seen": 420937728 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004406619859578737, + "loss": 1.4923, + "theoretical_loss": 3.989143174414733, + "tokens_seen": 421003264 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -0.00192836404312402, + "objective/train/docs_used": 250392, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9362990856170654, + "objective/train/original_loss": 2.9362993240356445, + "objective/train/theoretical_loss": 3.989074303792292, + "objective/train/tokens_used": 441528800, + "objective/train/value_avg": -0.00623321533203125, + "objective/train/value_loss": 0.00010906876559602097, + "objective/train/value_max": -0.0003542900085449219, + "objective/train/value_min": -0.08404541015625, + "objective/train/value_reward_corr": 0.5531534337632157, + "objective/train/value_std": 0.006938934326171875, + "objective/train/weight_avg": 0.9854475259780884, + "objective/train/weighted_lm_loss": 2.903167486190796, + "objective/train/weights_max": 2.172175645828247, + "objective/train/weights_min": 0.1975039690732956, + "theoretical_loss": 3.989074303792292, + "tokens_seen": 421068800 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004406519558676028, + "loss": 1.4797, + "theoretical_loss": 3.989074303792292, + "tokens_seen": 421068800 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044064192577733204, + "loss": 1.4546, + "theoretical_loss": 3.9890054468890135, + "tokens_seen": 421134336 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044063189568706117, + "loss": 1.4694, + "theoretical_loss": 3.9889366037000302, + "tokens_seen": 421199872 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004406218655967904, + "loss": 1.5549, + "theoretical_loss": 3.9888677742204766, + "tokens_seen": 421265408 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004406118355065196, + "loss": 1.4769, + "theoretical_loss": 3.988798958445491, + "tokens_seen": 421330944 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044060180541624877, + "loss": 1.4733, + "theoretical_loss": 3.988730156370214, + "tokens_seen": 421396480 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044059177532597795, + "loss": 1.4544, + "theoretical_loss": 3.9886613679897875, + "tokens_seen": 421462016 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044058174523570713, + "loss": 1.5501, + "theoretical_loss": 3.988592593299358, + "tokens_seen": 421527552 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004405717151454363, + "loss": 1.4414, + "theoretical_loss": 3.9885238322940717, + "tokens_seen": 421593088 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044056168505516555, + "loss": 1.4358, + "theoretical_loss": 3.9884550849690794, + "tokens_seen": 421658624 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004405516549648947, + "loss": 1.5525, + "theoretical_loss": 3.988386351319534, + "tokens_seen": 421724160 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004405416248746239, + "loss": 1.3643, + "theoretical_loss": 3.9883176313405904, + "tokens_seen": 421789696 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044053159478435304, + "loss": 1.5429, + "theoretical_loss": 3.9882489250274062, + "tokens_seen": 421855232 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044052156469408227, + "loss": 1.5054, + "theoretical_loss": 3.9881802323751416, + "tokens_seen": 421920768 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044051153460381145, + "loss": 1.5284, + "theoretical_loss": 3.9881115533789586, + "tokens_seen": 421986304 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044050150451354063, + "loss": 1.4405, + "theoretical_loss": 3.9880428880340224, + "tokens_seen": 422051840 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004404914744232698, + "loss": 1.4744, + "theoretical_loss": 3.987974236335501, + "tokens_seen": 422117376 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044048144433299905, + "loss": 1.693, + "theoretical_loss": 3.987905598278563, + "tokens_seen": 422182912 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004404714142427282, + "loss": 1.5454, + "theoretical_loss": 3.987836973858382, + "tokens_seen": 422248448 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004404613841524574, + "loss": 1.5196, + "theoretical_loss": 3.9877683630701326, + "tokens_seen": 422313984 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044045135406218654, + "loss": 1.3889, + "theoretical_loss": 3.9876997659089914, + "tokens_seen": 422379520 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004404413239719158, + "loss": 1.4512, + "theoretical_loss": 3.987631182370139, + "tokens_seen": 422445056 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044043129388164496, + "loss": 1.4201, + "theoretical_loss": 3.987562612448756, + "tokens_seen": 422510592 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044042126379137414, + "loss": 1.4523, + "theoretical_loss": 3.9874940561400294, + "tokens_seen": 422576128 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004404112337011033, + "loss": 1.5474, + "theoretical_loss": 3.9874255134391445, + "tokens_seen": 422641664 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -0.00043547601671889424, + "objective/train/docs_used": 251199, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.452394723892212, + "objective/train/original_loss": 3.45239520072937, + "objective/train/theoretical_loss": 3.9873569843412913, + "objective/train/tokens_used": 443167200, + "objective/train/value_avg": -0.004878997802734375, + "objective/train/value_loss": 7.244363223435357e-05, + "objective/train/value_max": -0.0005054473876953125, + "objective/train/value_min": -0.097900390625, + "objective/train/value_reward_corr": 0.4175493770882808, + "objective/train/value_std": 0.00434112548828125, + "objective/train/weight_avg": 0.998956024646759, + "objective/train/weighted_lm_loss": 3.461677312850952, + "objective/train/weights_max": 1.798309326171875, + "objective/train/weights_min": 0.32136645913124084, + "theoretical_loss": 3.9873569843412913, + "tokens_seen": 422707200 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004404012036108325, + "loss": 1.5996, + "theoretical_loss": 3.9873569843412913, + "tokens_seen": 422707200 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004403911735205617, + "loss": 1.4386, + "theoretical_loss": 3.987288468841662, + "tokens_seen": 422772736 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004403811434302909, + "loss": 1.5641, + "theoretical_loss": 3.9872199669354504, + "tokens_seen": 422838272 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044037111334002004, + "loss": 1.4069, + "theoretical_loss": 3.987151478617854, + "tokens_seen": 422903808 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004403610832497493, + "loss": 1.5343, + "theoretical_loss": 3.987083003884072, + "tokens_seen": 422969344 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004403510531594784, + "loss": 1.3358, + "theoretical_loss": 3.9870145427293053, + "tokens_seen": 423034880 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044034102306920764, + "loss": 1.5732, + "theoretical_loss": 3.9869460951487588, + "tokens_seen": 423100416 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004403309929789368, + "loss": 1.5384, + "theoretical_loss": 3.9868776611376395, + "tokens_seen": 423165952 + }, + { + "epoch": 0.13, + "learning_rate": 0.000440320962888666, + "loss": 1.2934, + "theoretical_loss": 3.986809240691156, + "tokens_seen": 423231488 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004403109327983952, + "loss": 1.4565, + "theoretical_loss": 3.9867408338045194, + "tokens_seen": 423297024 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004403009027081244, + "loss": 1.4324, + "theoretical_loss": 3.9866724404729443, + "tokens_seen": 423362560 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044029087261785355, + "loss": 1.3525, + "theoretical_loss": 3.986604060691646, + "tokens_seen": 423428096 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004402808425275828, + "loss": 1.454, + "theoretical_loss": 3.9865356944558448, + "tokens_seen": 423493632 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004402708124373119, + "loss": 1.5311, + "theoretical_loss": 3.9864673417607603, + "tokens_seen": 423559168 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044026078234704114, + "loss": 1.4348, + "theoretical_loss": 3.986399002601617, + "tokens_seen": 423624704 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004402507522567703, + "loss": 1.4827, + "theoretical_loss": 3.9863306769736413, + "tokens_seen": 423690240 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004402407221664995, + "loss": 1.4329, + "theoretical_loss": 3.9862623648720605, + "tokens_seen": 423755776 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004402306920762287, + "loss": 1.4915, + "theoretical_loss": 3.986194066292107, + "tokens_seen": 423821312 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044022066198595787, + "loss": 1.476, + "theoretical_loss": 3.9861257812290125, + "tokens_seen": 423886848 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044021063189568705, + "loss": 1.5012, + "theoretical_loss": 3.9860575096780133, + "tokens_seen": 423952384 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004402006018054163, + "loss": 1.5392, + "theoretical_loss": 3.985989251634348, + "tokens_seen": 424017920 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004401905717151454, + "loss": 1.4809, + "theoretical_loss": 3.985921007093257, + "tokens_seen": 424083456 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044018054162487465, + "loss": 1.453, + "theoretical_loss": 3.985852776049983, + "tokens_seen": 424148992 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044017051153460383, + "loss": 1.5259, + "theoretical_loss": 3.985784558499771, + "tokens_seen": 424214528 + }, + { + "epoch": 0.13, + "learning_rate": 0.000440160481444333, + "loss": 1.3386, + "theoretical_loss": 3.9857163544378693, + "tokens_seen": 424280064 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -0.0008398073841817677, + "objective/train/docs_used": 252247, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.703382968902588, + "objective/train/original_loss": 2.703382968902588, + "objective/train/theoretical_loss": 3.985648163859529, + "objective/train/tokens_used": 444805600, + "objective/train/value_avg": -0.005489349365234375, + "objective/train/value_loss": 5.0339884182903916e-05, + "objective/train/value_max": -0.0005130767822265625, + "objective/train/value_min": -0.0548095703125, + "objective/train/value_reward_corr": 0.21447559222144352, + "objective/train/value_std": 0.00408172607421875, + "objective/train/weight_avg": 0.994053840637207, + "objective/train/weighted_lm_loss": 2.70361065864563, + "objective/train/weights_max": 1.6644349098205566, + "objective/train/weights_min": 0.47972798347473145, + "theoretical_loss": 3.985648163859529, + "tokens_seen": 424345600 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004401504513540622, + "loss": 1.3288, + "theoretical_loss": 3.985648163859529, + "tokens_seen": 424345600 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044014042126379137, + "loss": 1.4565, + "theoretical_loss": 3.9855799867600004, + "tokens_seen": 424411136 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044013039117352055, + "loss": 1.3562, + "theoretical_loss": 3.9855118231345403, + "tokens_seen": 424476672 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004401203610832498, + "loss": 1.4974, + "theoretical_loss": 3.9854436729784055, + "tokens_seen": 424542208 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004401103309929789, + "loss": 1.5691, + "theoretical_loss": 3.985375536286856, + "tokens_seen": 424607744 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044010030090270815, + "loss": 1.3058, + "theoretical_loss": 3.9853074130551542, + "tokens_seen": 424673280 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004400902708124373, + "loss": 1.4179, + "theoretical_loss": 3.9852393032785636, + "tokens_seen": 424738816 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004400802407221665, + "loss": 1.4772, + "theoretical_loss": 3.9851712069523524, + "tokens_seen": 424804352 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004400702106318957, + "loss": 1.435, + "theoretical_loss": 3.98510312407179, + "tokens_seen": 424869888 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004400601805416249, + "loss": 1.4543, + "theoretical_loss": 3.9850350546321467, + "tokens_seen": 424935424 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044005015045135406, + "loss": 1.4056, + "theoretical_loss": 3.9849669986286984, + "tokens_seen": 425000960 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044004012036108324, + "loss": 1.3769, + "theoretical_loss": 3.9848989560567207, + "tokens_seen": 425066496 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004400300902708124, + "loss": 1.5538, + "theoretical_loss": 3.984830926911493, + "tokens_seen": 425132032 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044002006018054165, + "loss": 1.3316, + "theoretical_loss": 3.9847629111882963, + "tokens_seen": 425197568 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004400100300902708, + "loss": 1.4857, + "theoretical_loss": 3.984694908882415, + "tokens_seen": 425263104 + }, + { + "epoch": 0.13, + "learning_rate": 0.00044, + "loss": 1.4459, + "theoretical_loss": 3.984626919989134, + "tokens_seen": 425328640 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004399899699097292, + "loss": 1.5708, + "theoretical_loss": 3.9845589445037426, + "tokens_seen": 425394176 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004399799398194584, + "loss": 1.3933, + "theoretical_loss": 3.9844909824215318, + "tokens_seen": 425459712 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043996990972918756, + "loss": 1.3831, + "theoretical_loss": 3.9844230337377944, + "tokens_seen": 425525248 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043995987963891674, + "loss": 1.5433, + "theoretical_loss": 3.984355098447826, + "tokens_seen": 425590784 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004399498495486459, + "loss": 1.3444, + "theoretical_loss": 3.9842871765469248, + "tokens_seen": 425656320 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043993981945837516, + "loss": 1.3954, + "theoretical_loss": 3.984219268030392, + "tokens_seen": 425721856 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043992978936810434, + "loss": 1.4199, + "theoretical_loss": 3.9841513728935287, + "tokens_seen": 425787392 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004399197592778335, + "loss": 1.4856, + "theoretical_loss": 3.984083491131641, + "tokens_seen": 425852928 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004399097291875627, + "loss": 1.4748, + "theoretical_loss": 3.9840156227400367, + "tokens_seen": 425918464 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -9.915436385199428e-05, + "objective/train/docs_used": 253004, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3565986156463623, + "objective/train/original_loss": 3.3565988540649414, + "objective/train/theoretical_loss": 3.9839477677140245, + "objective/train/tokens_used": 446444000, + "objective/train/value_avg": -0.005535125732421875, + "objective/train/value_loss": 5.587530540651642e-05, + "objective/train/value_max": -0.00046372413635253906, + "objective/train/value_min": -0.0816650390625, + "objective/train/value_reward_corr": 0.5183964833240698, + "objective/train/value_std": 0.005496978759765625, + "objective/train/weight_avg": 1.0016849040985107, + "objective/train/weighted_lm_loss": 3.37027645111084, + "objective/train/weights_max": 1.613629698753357, + "objective/train/weights_min": 0.4121001660823822, + "theoretical_loss": 3.9839477677140245, + "tokens_seen": 425984000 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004398996990972919, + "loss": 1.6016, + "theoretical_loss": 3.9839477677140245, + "tokens_seen": 425984000 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004398896690070211, + "loss": 1.4024, + "theoretical_loss": 3.9838799260489175, + "tokens_seen": 426049536 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043987963891675024, + "loss": 1.6081, + "theoretical_loss": 3.9838120977400306, + "tokens_seen": 426115072 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004398696088264795, + "loss": 1.5466, + "theoretical_loss": 3.9837442827826797, + "tokens_seen": 426180608 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004398595787362086, + "loss": 1.464, + "theoretical_loss": 3.9836764811721848, + "tokens_seen": 426246144 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043984954864593784, + "loss": 1.5491, + "theoretical_loss": 3.9836086929038674, + "tokens_seen": 426311680 + }, + { + "epoch": 0.13, + "learning_rate": 0.000439839518555667, + "loss": 1.5149, + "theoretical_loss": 3.983540917973052, + "tokens_seen": 426377216 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004398294884653962, + "loss": 1.3718, + "theoretical_loss": 3.983473156375064, + "tokens_seen": 426442752 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004398194583751254, + "loss": 1.4594, + "theoretical_loss": 3.983405408105233, + "tokens_seen": 426508288 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004398094282848546, + "loss": 1.4875, + "theoretical_loss": 3.98333767315889, + "tokens_seen": 426573824 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043979939819458375, + "loss": 1.4577, + "theoretical_loss": 3.983269951531368, + "tokens_seen": 426639360 + }, + { + "epoch": 0.13, + "learning_rate": 0.000439789368104313, + "loss": 1.5407, + "theoretical_loss": 3.9832022432180025, + "tokens_seen": 426704896 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004397793380140421, + "loss": 1.4544, + "theoretical_loss": 3.983134548214133, + "tokens_seen": 426770432 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043976930792377134, + "loss": 1.4878, + "theoretical_loss": 3.983066866515099, + "tokens_seen": 426835968 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004397592778335005, + "loss": 1.5333, + "theoretical_loss": 3.982999198116244, + "tokens_seen": 426901504 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004397492477432297, + "loss": 1.4884, + "theoretical_loss": 3.9829315430129117, + "tokens_seen": 426967040 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004397392176529589, + "loss": 1.4813, + "theoretical_loss": 3.9828639012004516, + "tokens_seen": 427032576 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043972918756268807, + "loss": 1.4936, + "theoretical_loss": 3.9827962726742117, + "tokens_seen": 427098112 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043971915747241725, + "loss": 1.5732, + "theoretical_loss": 3.982728657429546, + "tokens_seen": 427163648 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004397091273821465, + "loss": 1.5229, + "theoretical_loss": 3.9826610554618083, + "tokens_seen": 427229184 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004396990972918756, + "loss": 1.5414, + "theoretical_loss": 3.982593466766355, + "tokens_seen": 427294720 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043968906720160485, + "loss": 1.5033, + "theoretical_loss": 3.982525891338546, + "tokens_seen": 427360256 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043967903711133403, + "loss": 1.4509, + "theoretical_loss": 3.9824583291737423, + "tokens_seen": 427425792 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004396690070210632, + "loss": 1.4237, + "theoretical_loss": 3.982390780267308, + "tokens_seen": 427491328 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004396589769307924, + "loss": 1.5908, + "theoretical_loss": 3.98232324461461, + "tokens_seen": 427556864 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -0.0012613513972610235, + "objective/train/docs_used": 254260, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0084757804870605, + "objective/train/original_loss": 3.0084760189056396, + "objective/train/theoretical_loss": 3.982255722211016, + "objective/train/tokens_used": 448082400, + "objective/train/value_avg": -0.005626678466796875, + "objective/train/value_loss": 6.726784340571612e-05, + "objective/train/value_max": -0.00044083595275878906, + "objective/train/value_min": -0.06719970703125, + "objective/train/value_reward_corr": 0.5050055231751508, + "objective/train/value_std": 0.00472259521484375, + "objective/train/weight_avg": 0.9905730485916138, + "objective/train/weighted_lm_loss": 2.9752490520477295, + "objective/train/weights_max": 1.535370945930481, + "objective/train/weights_min": 0.38667139410972595, + "theoretical_loss": 3.982255722211016, + "tokens_seen": 427622400 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043964894684052157, + "loss": 1.5209, + "theoretical_loss": 3.982255722211016, + "tokens_seen": 427622400 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043963891675025075, + "loss": 1.5371, + "theoretical_loss": 3.982188213051897, + "tokens_seen": 427687936 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043962888665998, + "loss": 1.505, + "theoretical_loss": 3.9821207171326267, + "tokens_seen": 427753472 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004396188565697091, + "loss": 1.519, + "theoretical_loss": 3.98205323444858, + "tokens_seen": 427819008 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043960882647943835, + "loss": 1.4398, + "theoretical_loss": 3.9819857649951347, + "tokens_seen": 427884544 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004395987963891675, + "loss": 1.4548, + "theoretical_loss": 3.981918308767672, + "tokens_seen": 427950080 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004395887662988967, + "loss": 1.4157, + "theoretical_loss": 3.9818508657615737, + "tokens_seen": 428015616 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004395787362086259, + "loss": 1.3754, + "theoretical_loss": 3.9817834359722246, + "tokens_seen": 428081152 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004395687061183551, + "loss": 1.5973, + "theoretical_loss": 3.981716019395012, + "tokens_seen": 428146688 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043955867602808426, + "loss": 1.5131, + "theoretical_loss": 3.981648616025325, + "tokens_seen": 428212224 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043954864593781344, + "loss": 1.4295, + "theoretical_loss": 3.9815812258585552, + "tokens_seen": 428277760 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004395386158475426, + "loss": 1.4598, + "theoretical_loss": 3.9815138488900974, + "tokens_seen": 428343296 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043952858575727185, + "loss": 1.439, + "theoretical_loss": 3.981446485115348, + "tokens_seen": 428408832 + }, + { + "epoch": 0.13, + "learning_rate": 0.000439518555667001, + "loss": 1.4018, + "theoretical_loss": 3.981379134529705, + "tokens_seen": 428474368 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004395085255767302, + "loss": 1.4763, + "theoretical_loss": 3.98131179712857, + "tokens_seen": 428539904 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004394984954864594, + "loss": 1.4946, + "theoretical_loss": 3.981244472907346, + "tokens_seen": 428605440 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004394884653961886, + "loss": 1.3963, + "theoretical_loss": 3.9811771618614387, + "tokens_seen": 428670976 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043947843530591776, + "loss": 1.5598, + "theoretical_loss": 3.981109863986256, + "tokens_seen": 428736512 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043946840521564694, + "loss": 1.4355, + "theoretical_loss": 3.981042579277208, + "tokens_seen": 428802048 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004394583751253761, + "loss": 1.4123, + "theoretical_loss": 3.9809753077297074, + "tokens_seen": 428867584 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043944834503510536, + "loss": 1.387, + "theoretical_loss": 3.980908049339169, + "tokens_seen": 428933120 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004394383149448345, + "loss": 1.4474, + "theoretical_loss": 3.9808408041010104, + "tokens_seen": 428998656 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004394282848545637, + "loss": 1.4672, + "theoretical_loss": 3.98077357201065, + "tokens_seen": 429064192 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043941825476429285, + "loss": 1.5693, + "theoretical_loss": 3.98070635306351, + "tokens_seen": 429129728 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004394082246740221, + "loss": 1.2953, + "theoretical_loss": 3.980639147255015, + "tokens_seen": 429195264 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -0.00014323210052680224, + "objective/train/docs_used": 255066, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.847367525100708, + "objective/train/original_loss": 2.847367525100708, + "objective/train/theoretical_loss": 3.98057195458059, + "objective/train/tokens_used": 449720800, + "objective/train/value_avg": -0.005329132080078125, + "objective/train/value_loss": 5.3357896831585094e-05, + "objective/train/value_max": -0.00031495094299316406, + "objective/train/value_min": -0.06805419921875, + "objective/train/value_reward_corr": 0.36475847739406914, + "objective/train/value_std": 0.0045928955078125, + "objective/train/weight_avg": 1.0011169910430908, + "objective/train/weighted_lm_loss": 2.8531954288482666, + "objective/train/weights_max": 1.5292916297912598, + "objective/train/weights_min": 0.4124697744846344, + "theoretical_loss": 3.98057195458059, + "tokens_seen": 429260800 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043939819458375126, + "loss": 1.3964, + "theoretical_loss": 3.98057195458059, + "tokens_seen": 429260800 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043938816449348044, + "loss": 1.3947, + "theoretical_loss": 3.980504775035665, + "tokens_seen": 429326336 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004393781344032096, + "loss": 1.3901, + "theoretical_loss": 3.98043760861567, + "tokens_seen": 429391872 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004393681043129388, + "loss": 1.4109, + "theoretical_loss": 3.980370455316039, + "tokens_seen": 429457408 + }, + { + "epoch": 0.13, + "learning_rate": 0.000439358074222668, + "loss": 1.4348, + "theoretical_loss": 3.980303315132206, + "tokens_seen": 429522944 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004393480441323972, + "loss": 1.4935, + "theoretical_loss": 3.9802361880596093, + "tokens_seen": 429588480 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043933801404212635, + "loss": 1.5566, + "theoretical_loss": 3.9801690740936895, + "tokens_seen": 429654016 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004393279839518556, + "loss": 1.5144, + "theoretical_loss": 3.980101973229889, + "tokens_seen": 429719552 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043931795386158477, + "loss": 1.5772, + "theoretical_loss": 3.9800348854636516, + "tokens_seen": 429785088 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043930792377131395, + "loss": 1.4683, + "theoretical_loss": 3.9799678107904244, + "tokens_seen": 429850624 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043929789368104313, + "loss": 1.5011, + "theoretical_loss": 3.979900749205657, + "tokens_seen": 429916160 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004392878635907723, + "loss": 1.6269, + "theoretical_loss": 3.9798337007048, + "tokens_seen": 429981696 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004392778335005015, + "loss": 1.3725, + "theoretical_loss": 3.9797666652833072, + "tokens_seen": 430047232 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004392678034102307, + "loss": 1.4631, + "theoretical_loss": 3.9796996429366356, + "tokens_seen": 430112768 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043925777331995985, + "loss": 1.4135, + "theoretical_loss": 3.9796326336602426, + "tokens_seen": 430178304 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004392477432296891, + "loss": 1.3447, + "theoretical_loss": 3.9795656374495882, + "tokens_seen": 430243840 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004392377131394182, + "loss": 1.4038, + "theoretical_loss": 3.9794986543001363, + "tokens_seen": 430309376 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043922768304914745, + "loss": 1.4806, + "theoretical_loss": 3.9794316842073516, + "tokens_seen": 430374912 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043921765295887663, + "loss": 1.5642, + "theoretical_loss": 3.9793647271667005, + "tokens_seen": 430440448 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004392076228686058, + "loss": 1.5469, + "theoretical_loss": 3.979297783173654, + "tokens_seen": 430505984 + }, + { + "epoch": 0.13, + "learning_rate": 0.000439197592778335, + "loss": 1.5843, + "theoretical_loss": 3.979230852223682, + "tokens_seen": 430571520 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043918756268806423, + "loss": 1.4281, + "theoretical_loss": 3.9791639343122602, + "tokens_seen": 430637056 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004391775325977934, + "loss": 1.5607, + "theoretical_loss": 3.979097029434865, + "tokens_seen": 430702592 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004391675025075226, + "loss": 1.4495, + "theoretical_loss": 3.979030137586974, + "tokens_seen": 430768128 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043915747241725177, + "loss": 1.6223, + "theoretical_loss": 3.978963258764068, + "tokens_seen": 430833664 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -0.0001039931521518156, + "objective/train/docs_used": 256338, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.834935188293457, + "objective/train/original_loss": 2.834935188293457, + "objective/train/theoretical_loss": 3.9788963929616314, + "objective/train/tokens_used": 451359200, + "objective/train/value_avg": -0.0065155029296875, + "objective/train/value_loss": 7.271887443494052e-05, + "objective/train/value_max": -0.00033926963806152344, + "objective/train/value_min": -0.09185791015625, + "objective/train/value_reward_corr": 0.4552049878079449, + "objective/train/value_std": 0.005977630615234375, + "objective/train/weight_avg": 1.002196192741394, + "objective/train/weighted_lm_loss": 2.840906858444214, + "objective/train/weights_max": 1.7853171825408936, + "objective/train/weights_min": 0.20189937949180603, + "theoretical_loss": 3.9788963929616314, + "tokens_seen": 430899200 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043914744232698095, + "loss": 1.4967, + "theoretical_loss": 3.9788963929616314, + "tokens_seen": 430899200 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004391374122367102, + "loss": 1.6257, + "theoretical_loss": 3.9788295401751483, + "tokens_seen": 430964736 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004391273821464393, + "loss": 1.4352, + "theoretical_loss": 3.9787627004001074, + "tokens_seen": 431030272 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043911735205616855, + "loss": 1.3984, + "theoretical_loss": 3.978695873631997, + "tokens_seen": 431095808 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004391073219658977, + "loss": 1.5347, + "theoretical_loss": 3.9786290598663108, + "tokens_seen": 431161344 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004390972918756269, + "loss": 1.4524, + "theoretical_loss": 3.978562259098542, + "tokens_seen": 431226880 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004390872617853561, + "loss": 1.5355, + "theoretical_loss": 3.9784954713241882, + "tokens_seen": 431292416 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004390772316950853, + "loss": 1.4467, + "theoretical_loss": 3.9784286965387468, + "tokens_seen": 431357952 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043906720160481446, + "loss": 1.4578, + "theoretical_loss": 3.9783619347377206, + "tokens_seen": 431423488 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043905717151454364, + "loss": 1.5617, + "theoretical_loss": 3.978295185916612, + "tokens_seen": 431489024 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004390471414242728, + "loss": 1.3964, + "theoretical_loss": 3.9782284500709264, + "tokens_seen": 431554560 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043903711133400205, + "loss": 1.5229, + "theoretical_loss": 3.978161727196172, + "tokens_seen": 431620096 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004390270812437312, + "loss": 1.3124, + "theoretical_loss": 3.978095017287859, + "tokens_seen": 431685632 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004390170511534604, + "loss": 1.3334, + "theoretical_loss": 3.9780283203414992, + "tokens_seen": 431751168 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004390070210631896, + "loss": 1.477, + "theoretical_loss": 3.9779616363526076, + "tokens_seen": 431816704 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004389969909729188, + "loss": 1.5225, + "theoretical_loss": 3.9778949653167004, + "tokens_seen": 431882240 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043898696088264796, + "loss": 1.3588, + "theoretical_loss": 3.977828307229297, + "tokens_seen": 431947776 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043897693079237714, + "loss": 1.5176, + "theoretical_loss": 3.9777616620859186, + "tokens_seen": 432013312 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004389669007021063, + "loss": 1.4264, + "theoretical_loss": 3.9776950298820886, + "tokens_seen": 432078848 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043895687061183556, + "loss": 1.5664, + "theoretical_loss": 3.977628410613332, + "tokens_seen": 432144384 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004389468405215647, + "loss": 1.3689, + "theoretical_loss": 3.9775618042751777, + "tokens_seen": 432209920 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004389368104312939, + "loss": 1.4368, + "theoretical_loss": 3.9774952108631556, + "tokens_seen": 432275456 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043892678034102305, + "loss": 1.2639, + "theoretical_loss": 3.9774286303727973, + "tokens_seen": 432340992 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004389167502507523, + "loss": 1.442, + "theoretical_loss": 3.977362062799638, + "tokens_seen": 432406528 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043890672016048146, + "loss": 1.5517, + "theoretical_loss": 3.977295508139215, + "tokens_seen": 432472064 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": 0.00033495985553599894, + "objective/train/docs_used": 257001, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.2386419773101807, + "objective/train/original_loss": 2.2386415004730225, + "objective/train/theoretical_loss": 3.9772289663870657, + "objective/train/tokens_used": 452997600, + "objective/train/value_avg": -0.00537109375, + "objective/train/value_loss": 8.69580326252617e-05, + "objective/train/value_max": -0.0004546642303466797, + "objective/train/value_min": -0.07574462890625, + "objective/train/value_reward_corr": 0.5275095751254992, + "objective/train/value_std": 0.004657745361328125, + "objective/train/weight_avg": 1.0071929693222046, + "objective/train/weighted_lm_loss": 2.2366249561309814, + "objective/train/weights_max": 1.5195223093032837, + "objective/train/weights_min": 0.1995791494846344, + "theoretical_loss": 3.9772289663870657, + "tokens_seen": 432537600 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043889669007021064, + "loss": 1.3089, + "theoretical_loss": 3.9772289663870657, + "tokens_seen": 432537600 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004388866599799398, + "loss": 1.4939, + "theoretical_loss": 3.977162437538732, + "tokens_seen": 432603136 + }, + { + "epoch": 0.13, + "learning_rate": 0.000438876629889669, + "loss": 1.5077, + "theoretical_loss": 3.9770959215897586, + "tokens_seen": 432668672 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004388665997993982, + "loss": 1.4696, + "theoretical_loss": 3.97702941853569, + "tokens_seen": 432734208 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004388565697091274, + "loss": 1.5199, + "theoretical_loss": 3.9769629283720738, + "tokens_seen": 432799744 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043884653961885655, + "loss": 1.5117, + "theoretical_loss": 3.9768964510944604, + "tokens_seen": 432865280 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004388365095285858, + "loss": 1.5634, + "theoretical_loss": 3.9768299866984025, + "tokens_seen": 432930816 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043882647943831497, + "loss": 1.5648, + "theoretical_loss": 3.9767635351794532, + "tokens_seen": 432996352 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043881644934804415, + "loss": 1.6541, + "theoretical_loss": 3.976697096533171, + "tokens_seen": 433061888 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043880641925777333, + "loss": 1.5255, + "theoretical_loss": 3.976630670755114, + "tokens_seen": 433127424 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004387963891675025, + "loss": 1.5516, + "theoretical_loss": 3.9765642578408427, + "tokens_seen": 433192960 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004387863590772317, + "loss": 1.5363, + "theoretical_loss": 3.9764978577859216, + "tokens_seen": 433258496 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004387763289869609, + "loss": 1.3922, + "theoretical_loss": 3.976431470585915, + "tokens_seen": 433324032 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043876629889669005, + "loss": 1.4138, + "theoretical_loss": 3.9763650962363917, + "tokens_seen": 433389568 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004387562688064193, + "loss": 1.4166, + "theoretical_loss": 3.9762987347329206, + "tokens_seen": 433455104 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004387462387161484, + "loss": 1.4785, + "theoretical_loss": 3.976232386071074, + "tokens_seen": 433520640 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043873620862587765, + "loss": 1.4676, + "theoretical_loss": 3.976166050246427, + "tokens_seen": 433586176 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043872617853560683, + "loss": 1.4982, + "theoretical_loss": 3.9760997272545557, + "tokens_seen": 433651712 + }, + { + "epoch": 0.13, + "learning_rate": 0.000438716148445336, + "loss": 1.3565, + "theoretical_loss": 3.9760334170910383, + "tokens_seen": 433717248 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004387061183550652, + "loss": 1.5017, + "theoretical_loss": 3.9759671197514557, + "tokens_seen": 433782784 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043869608826479443, + "loss": 1.5902, + "theoretical_loss": 3.975900835231391, + "tokens_seen": 433848320 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043868605817452356, + "loss": 1.5615, + "theoretical_loss": 3.9758345635264307, + "tokens_seen": 433913856 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004386760280842528, + "loss": 1.4645, + "theoretical_loss": 3.9757683046321604, + "tokens_seen": 433979392 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004386659979939819, + "loss": 1.4514, + "theoretical_loss": 3.9757020585441705, + "tokens_seen": 434044928 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043865596790371115, + "loss": 1.4546, + "theoretical_loss": 3.975635825258053, + "tokens_seen": 434110464 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": 0.0013423770433291793, + "objective/train/docs_used": 257982, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.595532178878784, + "objective/train/original_loss": 2.5955324172973633, + "objective/train/theoretical_loss": 3.975569604769402, + "objective/train/tokens_used": 454636000, + "objective/train/value_avg": -0.006107330322265625, + "objective/train/value_loss": 4.8793008318170905e-05, + "objective/train/value_max": -0.0005636215209960938, + "objective/train/value_min": -0.0740966796875, + "objective/train/value_reward_corr": 0.3545783186778816, + "objective/train/value_std": 0.004791259765625, + "objective/train/weight_avg": 1.0158276557922363, + "objective/train/weighted_lm_loss": 2.636737585067749, + "objective/train/weights_max": 2.0979628562927246, + "objective/train/weights_min": 0.40677034854888916, + "theoretical_loss": 3.975569604769402, + "tokens_seen": 434176000 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043864593781344033, + "loss": 1.4688, + "theoretical_loss": 3.975569604769402, + "tokens_seen": 434176000 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004386359077231695, + "loss": 1.4917, + "theoretical_loss": 3.9755033970738127, + "tokens_seen": 434241536 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004386258776328987, + "loss": 1.5891, + "theoretical_loss": 3.9754372021668845, + "tokens_seen": 434307072 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004386158475426279, + "loss": 1.4213, + "theoretical_loss": 3.975371020044218, + "tokens_seen": 434372608 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043860581745235706, + "loss": 1.4275, + "theoretical_loss": 3.9753048507014146, + "tokens_seen": 434438144 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004385957873620863, + "loss": 1.4488, + "theoretical_loss": 3.9752386941340805, + "tokens_seen": 434503680 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004385857572718154, + "loss": 1.4094, + "theoretical_loss": 3.975172550337822, + "tokens_seen": 434569216 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043857572718154466, + "loss": 1.5772, + "theoretical_loss": 3.975106419308249, + "tokens_seen": 434634752 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004385656970912738, + "loss": 1.5516, + "theoretical_loss": 3.9750403010409716, + "tokens_seen": 434700288 + }, + { + "epoch": 0.13, + "learning_rate": 0.000438555667001003, + "loss": 1.5524, + "theoretical_loss": 3.9749741955316056, + "tokens_seen": 434765824 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004385456369107322, + "loss": 1.5509, + "theoretical_loss": 3.974908102775764, + "tokens_seen": 434831360 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004385356068204614, + "loss": 1.5292, + "theoretical_loss": 3.9748420227690673, + "tokens_seen": 434896896 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043852557673019056, + "loss": 1.4777, + "theoretical_loss": 3.9747759555071336, + "tokens_seen": 434962432 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004385155466399198, + "loss": 1.5541, + "theoretical_loss": 3.974709900985586, + "tokens_seen": 435027968 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004385055165496489, + "loss": 1.5404, + "theoretical_loss": 3.9746438592000484, + "tokens_seen": 435093504 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043849548645937816, + "loss": 1.449, + "theoretical_loss": 3.9745778301461483, + "tokens_seen": 435159040 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004384854563691073, + "loss": 1.3852, + "theoretical_loss": 3.9745118138195137, + "tokens_seen": 435224576 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004384754262788365, + "loss": 1.4831, + "theoretical_loss": 3.9744458102157756, + "tokens_seen": 435290112 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004384653961885657, + "loss": 1.4932, + "theoretical_loss": 3.974379819330567, + "tokens_seen": 435355648 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004384553660982949, + "loss": 1.4829, + "theoretical_loss": 3.974313841159523, + "tokens_seen": 435421184 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043844533600802407, + "loss": 1.2756, + "theoretical_loss": 3.974247875698281, + "tokens_seen": 435486720 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043843530591775325, + "loss": 1.4546, + "theoretical_loss": 3.974181922942481, + "tokens_seen": 435552256 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004384252758274825, + "loss": 1.4473, + "theoretical_loss": 3.9741159828877635, + "tokens_seen": 435617792 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043841524573721166, + "loss": 1.541, + "theoretical_loss": 3.9740500555297738, + "tokens_seen": 435683328 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043840521564694084, + "loss": 1.5043, + "theoretical_loss": 3.9739841408641565, + "tokens_seen": 435748864 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -2.813282117131166e-05, + "objective/train/docs_used": 258654, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4682888984680176, + "objective/train/original_loss": 2.4682891368865967, + "objective/train/theoretical_loss": 3.9739182388865606, + "objective/train/tokens_used": 456274400, + "objective/train/value_avg": -0.00496673583984375, + "objective/train/value_loss": 4.8679390602046624e-05, + "objective/train/value_max": -0.0005421638488769531, + "objective/train/value_min": -0.11358642578125, + "objective/train/value_reward_corr": 0.32595411377117167, + "objective/train/value_std": 0.003955841064453125, + "objective/train/weight_avg": 1.0021177530288696, + "objective/train/weighted_lm_loss": 2.4643733501434326, + "objective/train/weights_max": 3.113863706588745, + "objective/train/weights_min": 0.415422648191452, + "theoretical_loss": 3.9739182388865606, + "tokens_seen": 435814400 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043839518555667, + "loss": 1.3776, + "theoretical_loss": 3.9739182388865606, + "tokens_seen": 435814400 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004383851554663992, + "loss": 1.4931, + "theoretical_loss": 3.9738523495926357, + "tokens_seen": 435879936 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004383751253761284, + "loss": 1.5314, + "theoretical_loss": 3.9737864729780346, + "tokens_seen": 435945472 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004383650952858576, + "loss": 1.3892, + "theoretical_loss": 3.973720609038412, + "tokens_seen": 436011008 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043835506519558675, + "loss": 1.4698, + "theoretical_loss": 3.9736547577694235, + "tokens_seen": 436076544 + }, + { + "epoch": 0.13, + "learning_rate": 0.000438345035105316, + "loss": 1.5518, + "theoretical_loss": 3.9735889191667293, + "tokens_seen": 436142080 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043833500501504517, + "loss": 1.4913, + "theoretical_loss": 3.9735230932259893, + "tokens_seen": 436207616 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043832497492477435, + "loss": 1.3632, + "theoretical_loss": 3.973457279942868, + "tokens_seen": 436273152 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043831494483450353, + "loss": 1.5391, + "theoretical_loss": 3.9733914793130287, + "tokens_seen": 436338688 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004383049147442327, + "loss": 1.5225, + "theoretical_loss": 3.9733256913321395, + "tokens_seen": 436404224 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004382948846539619, + "loss": 1.4614, + "theoretical_loss": 3.973259915995871, + "tokens_seen": 436469760 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004382848545636911, + "loss": 1.3207, + "theoretical_loss": 3.973194153299893, + "tokens_seen": 436535296 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043827482447342025, + "loss": 1.4214, + "theoretical_loss": 3.9731284032398806, + "tokens_seen": 436600832 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004382647943831495, + "loss": 1.4271, + "theoretical_loss": 3.9730626658115096, + "tokens_seen": 436666368 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004382547642928786, + "loss": 1.3678, + "theoretical_loss": 3.9729969410104573, + "tokens_seen": 436731904 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043824473420260785, + "loss": 1.5405, + "theoretical_loss": 3.9729312288324037, + "tokens_seen": 436797440 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043823470411233703, + "loss": 1.4417, + "theoretical_loss": 3.972865529273032, + "tokens_seen": 436862976 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004382246740220662, + "loss": 1.5553, + "theoretical_loss": 3.972799842328027, + "tokens_seen": 436928512 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004382146439317954, + "loss": 1.4647, + "theoretical_loss": 3.9727341679930737, + "tokens_seen": 436994048 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043820461384152463, + "loss": 1.4715, + "theoretical_loss": 3.9726685062638616, + "tokens_seen": 437059584 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043819458375125376, + "loss": 1.5879, + "theoretical_loss": 3.9726028571360814, + "tokens_seen": 437125120 + }, + { + "epoch": 0.13, + "learning_rate": 0.000438184553660983, + "loss": 1.4498, + "theoretical_loss": 3.972537220605426, + "tokens_seen": 437190656 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004381745235707121, + "loss": 1.5233, + "theoretical_loss": 3.9724715966675896, + "tokens_seen": 437256192 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043816449348044135, + "loss": 1.5732, + "theoretical_loss": 3.9724059853182707, + "tokens_seen": 437321728 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043815446339017053, + "loss": 1.5364, + "theoretical_loss": 3.9723403865531677, + "tokens_seen": 437387264 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -0.00375038362108171, + "objective/train/docs_used": 259916, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6076810359954834, + "objective/train/original_loss": 2.6076807975769043, + "objective/train/theoretical_loss": 3.9722748003679818, + "objective/train/tokens_used": 457912800, + "objective/train/value_avg": -0.005924224853515625, + "objective/train/value_loss": 0.0001488790294388309, + "objective/train/value_max": -0.0004222393035888672, + "objective/train/value_min": -0.06805419921875, + "objective/train/value_reward_corr": 0.5454137185127929, + "objective/train/value_std": 0.0064544677734375, + "objective/train/weight_avg": 0.9690316915512085, + "objective/train/weighted_lm_loss": 2.5265748500823975, + "objective/train/weights_max": 1.6743781566619873, + "objective/train/weights_min": 0.1534673273563385, + "theoretical_loss": 3.9722748003679818, + "tokens_seen": 437452800 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004381444332998997, + "loss": 1.5466, + "theoretical_loss": 3.9722748003679818, + "tokens_seen": 437452800 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004381344032096289, + "loss": 1.3729, + "theoretical_loss": 3.972209226758417, + "tokens_seen": 437518336 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004381243731193581, + "loss": 1.3296, + "theoretical_loss": 3.972143665720178, + "tokens_seen": 437583872 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043811434302908726, + "loss": 1.4983, + "theoretical_loss": 3.9720781172489734, + "tokens_seen": 437649408 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004381043129388165, + "loss": 1.4729, + "theoretical_loss": 3.972012581340513, + "tokens_seen": 437714944 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004380942828485456, + "loss": 1.6227, + "theoretical_loss": 3.9719470579905076, + "tokens_seen": 437780480 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043808425275827486, + "loss": 1.527, + "theoretical_loss": 3.9718815471946725, + "tokens_seen": 437846016 + }, + { + "epoch": 0.13, + "learning_rate": 0.000438074222668004, + "loss": 1.4483, + "theoretical_loss": 3.971816048948723, + "tokens_seen": 437911552 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004380641925777332, + "loss": 1.5943, + "theoretical_loss": 3.9717505632483774, + "tokens_seen": 437977088 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004380541624874624, + "loss": 1.5218, + "theoretical_loss": 3.9716850900893563, + "tokens_seen": 438042624 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004380441323971916, + "loss": 1.6058, + "theoretical_loss": 3.971619629467382, + "tokens_seen": 438108160 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043803410230692076, + "loss": 1.5516, + "theoretical_loss": 3.971554181378178, + "tokens_seen": 438173696 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043802407221665, + "loss": 1.384, + "theoretical_loss": 3.9714887458174726, + "tokens_seen": 438239232 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004380140421263791, + "loss": 1.552, + "theoretical_loss": 3.9714233227809936, + "tokens_seen": 438304768 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043800401203610836, + "loss": 1.3989, + "theoretical_loss": 3.9713579122644718, + "tokens_seen": 438370304 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004379939819458375, + "loss": 1.3695, + "theoretical_loss": 3.97129251426364, + "tokens_seen": 438435840 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004379839518555667, + "loss": 1.4735, + "theoretical_loss": 3.971227128774234, + "tokens_seen": 438501376 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004379739217652959, + "loss": 1.594, + "theoretical_loss": 3.971161755791989, + "tokens_seen": 438566912 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004379638916750251, + "loss": 1.4373, + "theoretical_loss": 3.971096395312646, + "tokens_seen": 438632448 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043795386158475427, + "loss": 1.4022, + "theoretical_loss": 3.9710310473319455, + "tokens_seen": 438697984 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043794383149448345, + "loss": 1.4396, + "theoretical_loss": 3.970965711845631, + "tokens_seen": 438763520 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043793380140421263, + "loss": 1.5162, + "theoretical_loss": 3.970900388849447, + "tokens_seen": 438829056 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043792377131394186, + "loss": 1.4351, + "theoretical_loss": 3.970835078339143, + "tokens_seen": 438894592 + }, + { + "epoch": 0.13, + "learning_rate": 0.000437913741223671, + "loss": 1.587, + "theoretical_loss": 3.9707697803104667, + "tokens_seen": 438960128 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004379037111334002, + "loss": 1.5086, + "theoretical_loss": 3.970704494759171, + "tokens_seen": 439025664 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": 2.4051150830928236e-05, + "objective/train/docs_used": 260524, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.300790309906006, + "objective/train/original_loss": 2.3007900714874268, + "objective/train/theoretical_loss": 3.9706392216810085, + "objective/train/tokens_used": 459551200, + "objective/train/value_avg": -0.005191802978515625, + "objective/train/value_loss": 5.4619566071778536e-05, + "objective/train/value_max": -0.0006852149963378906, + "objective/train/value_min": -0.04486083984375, + "objective/train/value_reward_corr": 0.39229884491819944, + "objective/train/value_std": 0.00348663330078125, + "objective/train/weight_avg": 1.0028551816940308, + "objective/train/weighted_lm_loss": 2.3135111331939697, + "objective/train/weights_max": 1.4558875560760498, + "objective/train/weights_min": 0.41503456234931946, + "theoretical_loss": 3.9706392216810085, + "tokens_seen": 439091200 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043789368104312935, + "loss": 1.4438, + "theoretical_loss": 3.9706392216810085, + "tokens_seen": 439091200 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004378836509528586, + "loss": 1.4479, + "theoretical_loss": 3.970573961071736, + "tokens_seen": 439156736 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043787362086258777, + "loss": 1.4523, + "theoretical_loss": 3.97050871292711, + "tokens_seen": 439222272 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043786359077231695, + "loss": 1.5417, + "theoretical_loss": 3.9704434772428923, + "tokens_seen": 439287808 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043785356068204613, + "loss": 1.4637, + "theoretical_loss": 3.970378254014844, + "tokens_seen": 439353344 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043784353059177537, + "loss": 1.3721, + "theoretical_loss": 3.9703130432387295, + "tokens_seen": 439418880 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004378335005015045, + "loss": 1.4704, + "theoretical_loss": 3.9702478449103142, + "tokens_seen": 439484416 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043782347041123373, + "loss": 1.526, + "theoretical_loss": 3.9701826590253675, + "tokens_seen": 439549952 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043781344032096286, + "loss": 1.3915, + "theoretical_loss": 3.9701174855796593, + "tokens_seen": 439615488 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004378034102306921, + "loss": 1.4744, + "theoretical_loss": 3.970052324568961, + "tokens_seen": 439681024 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043779338014042127, + "loss": 1.5402, + "theoretical_loss": 3.969987175989049, + "tokens_seen": 439746560 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043778335005015045, + "loss": 1.4662, + "theoretical_loss": 3.9699220398356982, + "tokens_seen": 439812096 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043777331995987963, + "loss": 1.4312, + "theoretical_loss": 3.9698569161046877, + "tokens_seen": 439877632 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004377632898696088, + "loss": 1.5097, + "theoretical_loss": 3.9697918047917984, + "tokens_seen": 439943168 + }, + { + "epoch": 0.13, + "learning_rate": 0.000437753259779338, + "loss": 1.4917, + "theoretical_loss": 3.969726705892813, + "tokens_seen": 440008704 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043774322968906723, + "loss": 1.5274, + "theoretical_loss": 3.9696616194035164, + "tokens_seen": 440074240 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043773319959879636, + "loss": 1.5617, + "theoretical_loss": 3.9695965453196944, + "tokens_seen": 440139776 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004377231695085256, + "loss": 1.3236, + "theoretical_loss": 3.9695314836371365, + "tokens_seen": 440205312 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004377131394182547, + "loss": 1.4698, + "theoretical_loss": 3.969466434351635, + "tokens_seen": 440270848 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043770310932798396, + "loss": 1.5089, + "theoretical_loss": 3.969401397458981, + "tokens_seen": 440336384 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043769307923771314, + "loss": 1.5144, + "theoretical_loss": 3.96933637295497, + "tokens_seen": 440401920 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004376830491474423, + "loss": 1.4166, + "theoretical_loss": 3.9692713608353998, + "tokens_seen": 440467456 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043767301905717155, + "loss": 1.4716, + "theoretical_loss": 3.969206361096069, + "tokens_seen": 440532992 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043766298896690073, + "loss": 1.4926, + "theoretical_loss": 3.9691413737327785, + "tokens_seen": 440598528 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004376529588766299, + "loss": 1.4366, + "theoretical_loss": 3.9690763987413327, + "tokens_seen": 440664064 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": 0.0002946469176094979, + "objective/train/docs_used": 261739, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.782193422317505, + "objective/train/original_loss": 2.782193660736084, + "objective/train/theoretical_loss": 3.969011436117536, + "objective/train/tokens_used": 461189600, + "objective/train/value_avg": -0.0045013427734375, + "objective/train/value_loss": 5.017181683797389e-05, + "objective/train/value_max": -0.0004239082336425781, + "objective/train/value_min": -0.045196533203125, + "objective/train/value_reward_corr": 0.22174705272631276, + "objective/train/value_std": 0.0031986236572265625, + "objective/train/weight_avg": 1.005180835723877, + "objective/train/weighted_lm_loss": 2.788778066635132, + "objective/train/weights_max": 1.5025702714920044, + "objective/train/weights_min": 0.2669898569583893, + "theoretical_loss": 3.969011436117536, + "tokens_seen": 440729600 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004376429287863591, + "loss": 1.5035, + "theoretical_loss": 3.969011436117536, + "tokens_seen": 440729600 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004376328986960883, + "loss": 1.4924, + "theoretical_loss": 3.968946485857196, + "tokens_seen": 440795136 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043762286860581746, + "loss": 1.2869, + "theoretical_loss": 3.968881547956122, + "tokens_seen": 440860672 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004376128385155467, + "loss": 1.5094, + "theoretical_loss": 3.9688166224101256, + "tokens_seen": 440926208 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004376028084252758, + "loss": 1.5305, + "theoretical_loss": 3.96875170921502, + "tokens_seen": 440991744 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043759277833500506, + "loss": 1.5458, + "theoretical_loss": 3.968686808366621, + "tokens_seen": 441057280 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004375827482447342, + "loss": 1.4725, + "theoretical_loss": 3.9686219198607464, + "tokens_seen": 441122816 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004375727181544634, + "loss": 1.4025, + "theoretical_loss": 3.968557043693215, + "tokens_seen": 441188352 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004375626880641926, + "loss": 1.4959, + "theoretical_loss": 3.968492179859849, + "tokens_seen": 441253888 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004375526579739218, + "loss": 1.4997, + "theoretical_loss": 3.968427328356472, + "tokens_seen": 441319424 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043754262788365096, + "loss": 1.5041, + "theoretical_loss": 3.9683624891789098, + "tokens_seen": 441384960 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004375325977933802, + "loss": 1.5839, + "theoretical_loss": 3.96829766232299, + "tokens_seen": 441450496 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004375225677031093, + "loss": 1.4839, + "theoretical_loss": 3.9682328477845425, + "tokens_seen": 441516032 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043751253761283856, + "loss": 1.5501, + "theoretical_loss": 3.9681680455593984, + "tokens_seen": 441581568 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004375025075225677, + "loss": 1.5574, + "theoretical_loss": 3.968103255643393, + "tokens_seen": 441647104 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004374924774322969, + "loss": 1.5661, + "theoretical_loss": 3.9680384780323603, + "tokens_seen": 441712640 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004374824473420261, + "loss": 1.5515, + "theoretical_loss": 3.9679737127221397, + "tokens_seen": 441778176 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004374724172517553, + "loss": 1.5049, + "theoretical_loss": 3.9679089597085704, + "tokens_seen": 441843712 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043746238716148447, + "loss": 1.487, + "theoretical_loss": 3.9678442189874947, + "tokens_seen": 441909248 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043745235707121365, + "loss": 1.4603, + "theoretical_loss": 3.967779490554756, + "tokens_seen": 441974784 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043744232698094283, + "loss": 1.461, + "theoretical_loss": 3.967714774406201, + "tokens_seen": 442040320 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043743229689067206, + "loss": 1.6012, + "theoretical_loss": 3.967650070537677, + "tokens_seen": 442105856 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004374222668004012, + "loss": 1.5558, + "theoretical_loss": 3.9675853789450346, + "tokens_seen": 442171392 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004374122367101304, + "loss": 1.4591, + "theoretical_loss": 3.9675206996241252, + "tokens_seen": 442236928 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043740220661985955, + "loss": 1.4212, + "theoretical_loss": 3.9674560325708037, + "tokens_seen": 442302464 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -0.0007177000516094267, + "objective/train/docs_used": 262441, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4508957862854004, + "objective/train/original_loss": 2.4508960247039795, + "objective/train/theoretical_loss": 3.9673913777809253, + "objective/train/tokens_used": 462828000, + "objective/train/value_avg": -0.006366729736328125, + "objective/train/value_loss": 7.227861351566389e-05, + "objective/train/value_max": -0.0005011558532714844, + "objective/train/value_min": -0.06982421875, + "objective/train/value_reward_corr": 0.49354689334549257, + "objective/train/value_std": 0.007259368896484375, + "objective/train/weight_avg": 0.9963254928588867, + "objective/train/weighted_lm_loss": 2.4518966674804688, + "objective/train/weights_max": 1.7660995721817017, + "objective/train/weights_min": 0.374788373708725, + "theoretical_loss": 3.9673913777809253, + "tokens_seen": 442368000 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004373921765295888, + "loss": 1.4165, + "theoretical_loss": 3.9673913777809253, + "tokens_seen": 442368000 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043738214643931797, + "loss": 1.5022, + "theoretical_loss": 3.9673267352503485, + "tokens_seen": 442433536 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043737211634904715, + "loss": 1.2406, + "theoretical_loss": 3.9672621049749335, + "tokens_seen": 442499072 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043736208625877633, + "loss": 1.5046, + "theoretical_loss": 3.9671974869505418, + "tokens_seen": 442564608 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043735205616850557, + "loss": 1.4333, + "theoretical_loss": 3.967132881173038, + "tokens_seen": 442630144 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004373420260782347, + "loss": 1.4987, + "theoretical_loss": 3.9670682876382886, + "tokens_seen": 442695680 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043733199598796393, + "loss": 1.3909, + "theoretical_loss": 3.967003706342161, + "tokens_seen": 442761216 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043732196589769306, + "loss": 1.5106, + "theoretical_loss": 3.9669391372805256, + "tokens_seen": 442826752 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004373119358074223, + "loss": 1.3646, + "theoretical_loss": 3.9668745804492542, + "tokens_seen": 442892288 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043730190571715147, + "loss": 1.3742, + "theoretical_loss": 3.9668100358442215, + "tokens_seen": 442957824 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043729187562688065, + "loss": 1.4218, + "theoretical_loss": 3.9667455034613033, + "tokens_seen": 443023360 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043728184553660983, + "loss": 1.357, + "theoretical_loss": 3.966680983296378, + "tokens_seen": 443088896 + }, + { + "epoch": 0.13, + "learning_rate": 0.000437271815446339, + "loss": 1.4552, + "theoretical_loss": 3.966616475345326, + "tokens_seen": 443154432 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004372617853560682, + "loss": 1.5156, + "theoretical_loss": 3.966551979604028, + "tokens_seen": 443219968 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043725175526579743, + "loss": 1.5963, + "theoretical_loss": 3.96648749606837, + "tokens_seen": 443285504 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043724172517552656, + "loss": 1.4774, + "theoretical_loss": 3.9664230247342367, + "tokens_seen": 443351040 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004372316950852558, + "loss": 1.4909, + "theoretical_loss": 3.966358565597517, + "tokens_seen": 443416576 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004372216649949849, + "loss": 1.4589, + "theoretical_loss": 3.9662941186541008, + "tokens_seen": 443482112 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043721163490471416, + "loss": 1.4271, + "theoretical_loss": 3.96622968389988, + "tokens_seen": 443547648 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043720160481444334, + "loss": 1.5405, + "theoretical_loss": 3.9661652613307496, + "tokens_seen": 443613184 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004371915747241725, + "loss": 1.3806, + "theoretical_loss": 3.9661008509426043, + "tokens_seen": 443678720 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004371815446339017, + "loss": 1.4882, + "theoretical_loss": 3.9660364527313434, + "tokens_seen": 443744256 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043717151454363093, + "loss": 1.5754, + "theoretical_loss": 3.965972066692866, + "tokens_seen": 443809792 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043716148445336006, + "loss": 1.5225, + "theoretical_loss": 3.965907692823075, + "tokens_seen": 443875328 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004371514543630893, + "loss": 1.4292, + "theoretical_loss": 3.965843331117874, + "tokens_seen": 443940864 + }, + { + "epoch": 0.13, + "objective/train/advantage_avg": -0.002658065175637603, + "objective/train/docs_used": 263839, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1385772228240967, + "objective/train/original_loss": 3.1385769844055176, + "objective/train/theoretical_loss": 3.9657789815731688, + "objective/train/tokens_used": 464466400, + "objective/train/value_avg": -0.005218505859375, + "objective/train/value_loss": 0.00018422740686219186, + "objective/train/value_max": -0.0004858970642089844, + "objective/train/value_min": -0.07305908203125, + "objective/train/value_reward_corr": 0.42256651789272204, + "objective/train/value_std": 0.00472259521484375, + "objective/train/weight_avg": 0.9815102219581604, + "objective/train/weighted_lm_loss": 3.128960609436035, + "objective/train/weights_max": 1.6999365091323853, + "objective/train/weights_min": 0.25453948974609375, + "theoretical_loss": 3.9657789815731688, + "tokens_seen": 444006400 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004371414242728184, + "loss": 1.4642, + "theoretical_loss": 3.9657789815731688, + "tokens_seen": 444006400 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043713139418254766, + "loss": 1.4599, + "theoretical_loss": 3.9657146441848674, + "tokens_seen": 444071936 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043712136409227684, + "loss": 1.5024, + "theoretical_loss": 3.96565031894888, + "tokens_seen": 444137472 + }, + { + "epoch": 0.13, + "learning_rate": 0.000437111334002006, + "loss": 1.5247, + "theoretical_loss": 3.965586005861119, + "tokens_seen": 444203008 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004371013039117352, + "loss": 1.3742, + "theoretical_loss": 3.965521704917498, + "tokens_seen": 444268544 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004370912738214644, + "loss": 1.4845, + "theoretical_loss": 3.9654574161139324, + "tokens_seen": 444334080 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043708124373119356, + "loss": 1.4962, + "theoretical_loss": 3.965393139446341, + "tokens_seen": 444399616 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004370712136409228, + "loss": 1.5118, + "theoretical_loss": 3.965328874910643, + "tokens_seen": 444465152 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004370611835506519, + "loss": 1.3718, + "theoretical_loss": 3.96526462250276, + "tokens_seen": 444530688 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043705115346038116, + "loss": 1.502, + "theoretical_loss": 3.9652003822186166, + "tokens_seen": 444596224 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004370411233701103, + "loss": 1.5704, + "theoretical_loss": 3.9651361540541386, + "tokens_seen": 444661760 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004370310932798395, + "loss": 1.4932, + "theoretical_loss": 3.965071938005252, + "tokens_seen": 444727296 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004370210631895687, + "loss": 1.4453, + "theoretical_loss": 3.965007734067889, + "tokens_seen": 444792832 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004370110330992979, + "loss": 1.4984, + "theoretical_loss": 3.96494354223798, + "tokens_seen": 444858368 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043700100300902707, + "loss": 1.5276, + "theoretical_loss": 3.964879362511459, + "tokens_seen": 444923904 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004369909729187563, + "loss": 1.4867, + "theoretical_loss": 3.9648151948842605, + "tokens_seen": 444989440 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043698094282848543, + "loss": 1.4322, + "theoretical_loss": 3.9647510393523238, + "tokens_seen": 445054976 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043697091273821467, + "loss": 1.5018, + "theoretical_loss": 3.9646868959115875, + "tokens_seen": 445120512 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004369608826479438, + "loss": 1.4825, + "theoretical_loss": 3.9646227645579932, + "tokens_seen": 445186048 + }, + { + "epoch": 0.13, + "learning_rate": 0.00043695085255767303, + "loss": 1.5488, + "theoretical_loss": 3.9645586452874846, + "tokens_seen": 445251584 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004369408224674022, + "loss": 1.3968, + "theoretical_loss": 3.964494538096006, + "tokens_seen": 445317120 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004369307923771314, + "loss": 1.4522, + "theoretical_loss": 3.9644304429795065, + "tokens_seen": 445382656 + }, + { + "epoch": 0.13, + "learning_rate": 0.0004369207622868606, + "loss": 1.5095, + "theoretical_loss": 3.9643663599339343, + "tokens_seen": 445448192 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043691073219658975, + "loss": 1.5488, + "theoretical_loss": 3.9643022889552415, + "tokens_seen": 445513728 + }, + { + "epoch": 0.14, + "learning_rate": 0.000436900702106319, + "loss": 1.4247, + "theoretical_loss": 3.9642382300393804, + "tokens_seen": 445579264 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.0011144819436594844, + "objective/train/docs_used": 264540, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.129345178604126, + "objective/train/original_loss": 3.129345655441284, + "objective/train/theoretical_loss": 3.9641741831823065, + "objective/train/tokens_used": 466104800, + "objective/train/value_avg": -0.0054931640625, + "objective/train/value_loss": 9.558180317981169e-05, + "objective/train/value_max": -0.0004565715789794922, + "objective/train/value_min": -0.0751953125, + "objective/train/value_reward_corr": 0.3229317649072048, + "objective/train/value_std": 0.004425048828125, + "objective/train/weight_avg": 0.992920994758606, + "objective/train/weighted_lm_loss": 3.141315221786499, + "objective/train/weights_max": 2.078843116760254, + "objective/train/weights_min": 0.08501856029033661, + "theoretical_loss": 3.9641741831823065, + "tokens_seen": 445644800 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043689067201604817, + "loss": 1.5297, + "theoretical_loss": 3.9641741831823065, + "tokens_seen": 445644800 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043688064192577735, + "loss": 1.4669, + "theoretical_loss": 3.964110148379977, + "tokens_seen": 445710336 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043687061183550653, + "loss": 1.3985, + "theoretical_loss": 3.964046125628351, + "tokens_seen": 445775872 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043686058174523577, + "loss": 1.3451, + "theoretical_loss": 3.96398211492339, + "tokens_seen": 445841408 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004368505516549649, + "loss": 1.4989, + "theoretical_loss": 3.9639181162610564, + "tokens_seen": 445906944 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043684052156469413, + "loss": 1.4839, + "theoretical_loss": 3.9638541296373155, + "tokens_seen": 445972480 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043683049147442326, + "loss": 1.3801, + "theoretical_loss": 3.963790155048133, + "tokens_seen": 446038016 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004368204613841525, + "loss": 1.6292, + "theoretical_loss": 3.9637261924894798, + "tokens_seen": 446103552 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043681043129388167, + "loss": 1.5472, + "theoretical_loss": 3.9636622419573246, + "tokens_seen": 446169088 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043680040120361085, + "loss": 1.5173, + "theoretical_loss": 3.963598303447642, + "tokens_seen": 446234624 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043679037111334003, + "loss": 1.3829, + "theoretical_loss": 3.9635343769564053, + "tokens_seen": 446300160 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004367803410230692, + "loss": 1.4114, + "theoretical_loss": 3.9634704624795916, + "tokens_seen": 446365696 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004367703109327984, + "loss": 1.5534, + "theoretical_loss": 3.963406560013179, + "tokens_seen": 446431232 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043676028084252763, + "loss": 1.5339, + "theoretical_loss": 3.9633426695531484, + "tokens_seen": 446496768 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043675025075225676, + "loss": 1.3391, + "theoretical_loss": 3.963278791095482, + "tokens_seen": 446562304 + }, + { + "epoch": 0.14, + "learning_rate": 0.000436740220661986, + "loss": 1.4541, + "theoretical_loss": 3.9632149246361643, + "tokens_seen": 446627840 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004367301905717151, + "loss": 1.5302, + "theoretical_loss": 3.9631510701711816, + "tokens_seen": 446693376 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043672016048144436, + "loss": 1.4229, + "theoretical_loss": 3.963087227696521, + "tokens_seen": 446758912 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043671013039117354, + "loss": 1.573, + "theoretical_loss": 3.9630233972081745, + "tokens_seen": 446824448 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004367001003009027, + "loss": 1.476, + "theoretical_loss": 3.962959578702133, + "tokens_seen": 446889984 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004366900702106319, + "loss": 1.4521, + "theoretical_loss": 3.9628957721743907, + "tokens_seen": 446955520 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043668004012036113, + "loss": 1.5783, + "theoretical_loss": 3.962831977620943, + "tokens_seen": 447021056 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043667001003009026, + "loss": 1.4349, + "theoretical_loss": 3.962768195037788, + "tokens_seen": 447086592 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004366599799398195, + "loss": 1.4914, + "theoretical_loss": 3.9627044244209264, + "tokens_seen": 447152128 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004366499498495486, + "loss": 1.4496, + "theoretical_loss": 3.962640665766359, + "tokens_seen": 447217664 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -7.957233901834115e-05, + "objective/train/docs_used": 265849, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.889209032058716, + "objective/train/original_loss": 2.8892087936401367, + "objective/train/theoretical_loss": 3.962576919070089, + "objective/train/tokens_used": 467743200, + "objective/train/value_avg": -0.005268096923828125, + "objective/train/value_loss": 6.445191684179008e-05, + "objective/train/value_max": -0.0005617141723632812, + "objective/train/value_min": -0.0784912109375, + "objective/train/value_reward_corr": 0.33359005637700956, + "objective/train/value_std": 0.004100799560546875, + "objective/train/weight_avg": 1.0021421909332275, + "objective/train/weighted_lm_loss": 2.8997104167938232, + "objective/train/weights_max": 2.192214250564575, + "objective/train/weights_min": 0.2999705374240875, + "theoretical_loss": 3.962576919070089, + "tokens_seen": 447283200 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043663991975927786, + "loss": 1.4599, + "theoretical_loss": 3.962576919070089, + "tokens_seen": 447283200 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043662988966900704, + "loss": 1.512, + "theoretical_loss": 3.962513184328122, + "tokens_seen": 447348736 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004366198595787362, + "loss": 1.3359, + "theoretical_loss": 3.9624494615364667, + "tokens_seen": 447414272 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004366098294884654, + "loss": 1.5308, + "theoretical_loss": 3.9623857506911313, + "tokens_seen": 447479808 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004365997993981946, + "loss": 1.482, + "theoretical_loss": 3.9623220517881275, + "tokens_seen": 447545344 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043658976930792376, + "loss": 1.4586, + "theoretical_loss": 3.962258364823468, + "tokens_seen": 447610880 + }, + { + "epoch": 0.14, + "learning_rate": 0.000436579739217653, + "loss": 1.5576, + "theoretical_loss": 3.9621946897931686, + "tokens_seen": 447676416 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043656970912738213, + "loss": 1.417, + "theoretical_loss": 3.9621310266932457, + "tokens_seen": 447741952 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043655967903711136, + "loss": 1.5849, + "theoretical_loss": 3.9620673755197187, + "tokens_seen": 447807488 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004365496489468405, + "loss": 1.4877, + "theoretical_loss": 3.962003736268608, + "tokens_seen": 447873024 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004365396188565697, + "loss": 1.4404, + "theoretical_loss": 3.961940108935937, + "tokens_seen": 447938560 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004365295887662989, + "loss": 1.3763, + "theoretical_loss": 3.961876493517729, + "tokens_seen": 448004096 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004365195586760281, + "loss": 1.4949, + "theoretical_loss": 3.9618128900100125, + "tokens_seen": 448069632 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043650952858575727, + "loss": 1.4222, + "theoretical_loss": 3.9617492984088143, + "tokens_seen": 448135168 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004364994984954865, + "loss": 1.5624, + "theoretical_loss": 3.9616857187101653, + "tokens_seen": 448200704 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043648946840521563, + "loss": 1.512, + "theoretical_loss": 3.961622150910098, + "tokens_seen": 448266240 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043647943831494487, + "loss": 1.434, + "theoretical_loss": 3.9615585950046466, + "tokens_seen": 448331776 + }, + { + "epoch": 0.14, + "learning_rate": 0.000436469408224674, + "loss": 1.3271, + "theoretical_loss": 3.961495050989847, + "tokens_seen": 448397312 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043645937813440323, + "loss": 1.4332, + "theoretical_loss": 3.9614315188617377, + "tokens_seen": 448462848 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004364493480441324, + "loss": 1.4417, + "theoretical_loss": 3.9613679986163572, + "tokens_seen": 448528384 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004364393179538616, + "loss": 1.4573, + "theoretical_loss": 3.9613044902497485, + "tokens_seen": 448593920 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043642928786359077, + "loss": 1.5222, + "theoretical_loss": 3.961240993757955, + "tokens_seen": 448659456 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043641925777331995, + "loss": 1.4822, + "theoretical_loss": 3.9611775091370225, + "tokens_seen": 448724992 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043640922768304913, + "loss": 1.4561, + "theoretical_loss": 3.9611140363829977, + "tokens_seen": 448790528 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043639919759277837, + "loss": 1.4591, + "theoretical_loss": 3.9610505754919307, + "tokens_seen": 448856064 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.0010582556715235114, + "objective/train/docs_used": 266141, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1079986095428467, + "objective/train/original_loss": 3.107999086380005, + "objective/train/theoretical_loss": 3.960987126459872, + "objective/train/tokens_used": 469381600, + "objective/train/value_avg": -0.005401611328125, + "objective/train/value_loss": 7.677406392758712e-05, + "objective/train/value_max": -0.00044417381286621094, + "objective/train/value_min": -0.073974609375, + "objective/train/value_reward_corr": 0.3534500122878801, + "objective/train/value_std": 0.00518035888671875, + "objective/train/weight_avg": 0.9929940104484558, + "objective/train/weighted_lm_loss": 3.1024329662323, + "objective/train/weights_max": 1.8350971937179565, + "objective/train/weights_min": 0.2953830063343048, + "theoretical_loss": 3.960987126459872, + "tokens_seen": 448921600 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004363891675025075, + "loss": 1.4615, + "theoretical_loss": 3.960987126459872, + "tokens_seen": 448921600 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043637913741223673, + "loss": 1.427, + "theoretical_loss": 3.960923689282876, + "tokens_seen": 448987136 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004363691073219659, + "loss": 1.459, + "theoretical_loss": 3.9608602639569965, + "tokens_seen": 449052672 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004363590772316951, + "loss": 1.465, + "theoretical_loss": 3.9607968504782907, + "tokens_seen": 449118208 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004363490471414243, + "loss": 1.5185, + "theoretical_loss": 3.9607334488428174, + "tokens_seen": 449183744 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043633901705115346, + "loss": 1.5579, + "theoretical_loss": 3.960670059046638, + "tokens_seen": 449249280 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043632898696088264, + "loss": 1.5512, + "theoretical_loss": 3.9606066810858143, + "tokens_seen": 449314816 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043631895687061187, + "loss": 1.6202, + "theoretical_loss": 3.960543314956411, + "tokens_seen": 449380352 + }, + { + "epoch": 0.14, + "learning_rate": 0.000436308926780341, + "loss": 1.5485, + "theoretical_loss": 3.960479960654494, + "tokens_seen": 449445888 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043629889669007023, + "loss": 1.4328, + "theoretical_loss": 3.960416618176132, + "tokens_seen": 449511424 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043628886659979936, + "loss": 1.4489, + "theoretical_loss": 3.9603532875173957, + "tokens_seen": 449576960 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004362788365095286, + "loss": 1.5131, + "theoretical_loss": 3.9602899686743553, + "tokens_seen": 449642496 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004362688064192578, + "loss": 1.535, + "theoretical_loss": 3.9602266616430866, + "tokens_seen": 449708032 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043625877632898696, + "loss": 1.4918, + "theoretical_loss": 3.9601633664196645, + "tokens_seen": 449773568 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043624874623871614, + "loss": 1.527, + "theoretical_loss": 3.9601000830001665, + "tokens_seen": 449839104 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004362387161484453, + "loss": 1.4085, + "theoretical_loss": 3.9600368113806717, + "tokens_seen": 449904640 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004362286860581745, + "loss": 1.6124, + "theoretical_loss": 3.9599735515572627, + "tokens_seen": 449970176 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043621865596790374, + "loss": 1.4278, + "theoretical_loss": 3.959910303526021, + "tokens_seen": 450035712 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043620862587763286, + "loss": 1.375, + "theoretical_loss": 3.9598470672830333, + "tokens_seen": 450101248 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004361985957873621, + "loss": 1.4595, + "theoretical_loss": 3.9597838428243857, + "tokens_seen": 450166784 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004361885656970913, + "loss": 1.4934, + "theoretical_loss": 3.959720630146167, + "tokens_seen": 450232320 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043617853560682046, + "loss": 1.5826, + "theoretical_loss": 3.9596574292444684, + "tokens_seen": 450297856 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004361685055165497, + "loss": 1.4314, + "theoretical_loss": 3.9595942401153823, + "tokens_seen": 450363392 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004361584754262788, + "loss": 1.4859, + "theoretical_loss": 3.959531062755003, + "tokens_seen": 450428928 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043614844533600806, + "loss": 1.43, + "theoretical_loss": 3.959467897159427, + "tokens_seen": 450494464 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.00011363167141098529, + "objective/train/docs_used": 266764, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.59193754196167, + "objective/train/original_loss": 3.591937303543091, + "objective/train/theoretical_loss": 3.959404743324752, + "objective/train/tokens_used": 471020000, + "objective/train/value_avg": -0.005496978759765625, + "objective/train/value_loss": 7.57811576477252e-05, + "objective/train/value_max": -0.0007266998291015625, + "objective/train/value_min": -0.06451416015625, + "objective/train/value_reward_corr": 0.387948988879543, + "objective/train/value_std": 0.004489898681640625, + "objective/train/weight_avg": 1.0022625923156738, + "objective/train/weighted_lm_loss": 3.604342222213745, + "objective/train/weights_max": 1.5185950994491577, + "objective/train/weights_min": 0.19489194452762604, + "theoretical_loss": 3.959404743324752, + "tokens_seen": 450560000 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043613841524573724, + "loss": 1.5347, + "theoretical_loss": 3.959404743324752, + "tokens_seen": 450560000 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004361283851554664, + "loss": 1.4953, + "theoretical_loss": 3.9593416012470786, + "tokens_seen": 450625536 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004361183550651956, + "loss": 1.3933, + "theoretical_loss": 3.9592784709225084, + "tokens_seen": 450691072 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004361083249749248, + "loss": 1.4079, + "theoretical_loss": 3.959215352347145, + "tokens_seen": 450756608 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043609829488465397, + "loss": 1.4336, + "theoretical_loss": 3.959152245517094, + "tokens_seen": 450822144 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004360882647943832, + "loss": 1.4264, + "theoretical_loss": 3.9590891504284635, + "tokens_seen": 450887680 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043607823470411233, + "loss": 1.5414, + "theoretical_loss": 3.959026067077362, + "tokens_seen": 450953216 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043606820461384156, + "loss": 1.4261, + "theoretical_loss": 3.958962995459901, + "tokens_seen": 451018752 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004360581745235707, + "loss": 1.4521, + "theoretical_loss": 3.9588999355721937, + "tokens_seen": 451084288 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004360481444332999, + "loss": 1.4176, + "theoretical_loss": 3.958836887410354, + "tokens_seen": 451149824 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004360381143430291, + "loss": 1.4663, + "theoretical_loss": 3.9587738509705, + "tokens_seen": 451215360 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004360280842527583, + "loss": 1.4549, + "theoretical_loss": 3.95871082624875, + "tokens_seen": 451280896 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043601805416248747, + "loss": 1.4951, + "theoretical_loss": 3.9586478132412233, + "tokens_seen": 451346432 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004360080240722167, + "loss": 1.4784, + "theoretical_loss": 3.9585848119440437, + "tokens_seen": 451411968 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043599799398194583, + "loss": 1.3712, + "theoretical_loss": 3.958521822353334, + "tokens_seen": 451477504 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043598796389167507, + "loss": 1.5882, + "theoretical_loss": 3.9584588444652207, + "tokens_seen": 451543040 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004359779338014042, + "loss": 1.5454, + "theoretical_loss": 3.9583958782758315, + "tokens_seen": 451608576 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043596790371113343, + "loss": 1.5362, + "theoretical_loss": 3.9583329237812968, + "tokens_seen": 451674112 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004359578736208626, + "loss": 1.3686, + "theoretical_loss": 3.9582699809777466, + "tokens_seen": 451739648 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004359478435305918, + "loss": 1.6359, + "theoretical_loss": 3.9582070498613158, + "tokens_seen": 451805184 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043593781344032097, + "loss": 1.3522, + "theoretical_loss": 3.958144130428139, + "tokens_seen": 451870720 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043592778335005015, + "loss": 1.5501, + "theoretical_loss": 3.9580812226743523, + "tokens_seen": 451936256 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043591775325977933, + "loss": 1.4552, + "theoretical_loss": 3.9580183265960955, + "tokens_seen": 452001792 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043590772316950857, + "loss": 1.5052, + "theoretical_loss": 3.9579554421895096, + "tokens_seen": 452067328 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004358976930792377, + "loss": 1.4265, + "theoretical_loss": 3.9578925694507356, + "tokens_seen": 452132864 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.000887122587300837, + "objective/train/docs_used": 268192, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7850632667541504, + "objective/train/original_loss": 2.7850630283355713, + "objective/train/theoretical_loss": 3.9578297083759195, + "objective/train/tokens_used": 472658400, + "objective/train/value_avg": -0.005397796630859375, + "objective/train/value_loss": 3.954356725444086e-05, + "objective/train/value_max": -0.0004172325134277344, + "objective/train/value_min": -0.057586669921875, + "objective/train/value_reward_corr": 0.38587203304583156, + "objective/train/value_std": 0.004154205322265625, + "objective/train/weight_avg": 0.9930213689804077, + "objective/train/weighted_lm_loss": 2.7708094120025635, + "objective/train/weights_max": 1.5255626440048218, + "objective/train/weights_min": 0.41558393836021423, + "theoretical_loss": 3.9578297083759195, + "tokens_seen": 452198400 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043588766298896693, + "loss": 1.4115, + "theoretical_loss": 3.9578297083759195, + "tokens_seen": 452198400 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004358776328986961, + "loss": 1.4373, + "theoretical_loss": 3.957766858961207, + "tokens_seen": 452263936 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004358676028084253, + "loss": 1.5291, + "theoretical_loss": 3.957704021202746, + "tokens_seen": 452329472 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004358575727181545, + "loss": 1.4571, + "theoretical_loss": 3.957641195096685, + "tokens_seen": 452395008 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043584754262788366, + "loss": 1.4879, + "theoretical_loss": 3.9575783806391787, + "tokens_seen": 452460544 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043583751253761284, + "loss": 1.5053, + "theoretical_loss": 3.9575155778263778, + "tokens_seen": 452526080 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043582748244734207, + "loss": 1.5706, + "theoretical_loss": 3.9574527866544384, + "tokens_seen": 452591616 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004358174523570712, + "loss": 1.6039, + "theoretical_loss": 3.9573900071195185, + "tokens_seen": 452657152 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043580742226680043, + "loss": 1.5603, + "theoretical_loss": 3.9573272392177765, + "tokens_seen": 452722688 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043579739217652956, + "loss": 1.503, + "theoretical_loss": 3.957264482945373, + "tokens_seen": 452788224 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004357873620862588, + "loss": 1.4678, + "theoretical_loss": 3.9572017382984708, + "tokens_seen": 452853760 + }, + { + "epoch": 0.14, + "learning_rate": 0.000435777331995988, + "loss": 1.4402, + "theoretical_loss": 3.9571390052732345, + "tokens_seen": 452919296 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043576730190571716, + "loss": 1.4132, + "theoretical_loss": 3.95707628386583, + "tokens_seen": 452984832 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043575727181544634, + "loss": 1.545, + "theoretical_loss": 3.957013574072426, + "tokens_seen": 453050368 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004357472417251755, + "loss": 1.4424, + "theoretical_loss": 3.9569508758891914, + "tokens_seen": 453115904 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004357372116349047, + "loss": 1.4964, + "theoretical_loss": 3.9568881893122994, + "tokens_seen": 453181440 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043572718154463394, + "loss": 1.4326, + "theoretical_loss": 3.956825514337922, + "tokens_seen": 453246976 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043571715145436306, + "loss": 1.428, + "theoretical_loss": 3.9567628509622352, + "tokens_seen": 453312512 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004357071213640923, + "loss": 1.4565, + "theoretical_loss": 3.956700199181416, + "tokens_seen": 453378048 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004356970912738215, + "loss": 1.5881, + "theoretical_loss": 3.9566375589916443, + "tokens_seen": 453443584 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043568706118355066, + "loss": 1.5729, + "theoretical_loss": 3.9565749303891, + "tokens_seen": 453509120 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043567703109327984, + "loss": 1.5762, + "theoretical_loss": 3.956512313369965, + "tokens_seen": 453574656 + }, + { + "epoch": 0.14, + "learning_rate": 0.000435667001003009, + "loss": 1.4572, + "theoretical_loss": 3.956449707930425, + "tokens_seen": 453640192 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004356569709127382, + "loss": 1.4353, + "theoretical_loss": 3.9563871140666658, + "tokens_seen": 453705728 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043564694082246744, + "loss": 1.4345, + "theoretical_loss": 3.9563245317748756, + "tokens_seen": 453771264 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.0008846957352943718, + "objective/train/docs_used": 268717, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8326029777526855, + "objective/train/original_loss": 2.8326029777526855, + "objective/train/theoretical_loss": 3.9562619610512435, + "objective/train/tokens_used": 474296800, + "objective/train/value_avg": -0.00443267822265625, + "objective/train/value_loss": 5.084396980237216e-05, + "objective/train/value_max": -0.0005054473876953125, + "objective/train/value_min": -0.0496826171875, + "objective/train/value_reward_corr": 0.2054265233388738, + "objective/train/value_std": 0.0031871795654296875, + "objective/train/weight_avg": 0.9934982657432556, + "objective/train/weighted_lm_loss": 2.8249032497406006, + "objective/train/weights_max": 1.4230226278305054, + "objective/train/weights_min": 0.3509940207004547, + "theoretical_loss": 3.9562619610512435, + "tokens_seen": 453836800 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043563691073219657, + "loss": 1.4154, + "theoretical_loss": 3.9562619610512435, + "tokens_seen": 453836800 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004356268806419258, + "loss": 1.5026, + "theoretical_loss": 3.956199401891962, + "tokens_seen": 453902336 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043561685055165493, + "loss": 1.4139, + "theoretical_loss": 3.9561368542932245, + "tokens_seen": 453967872 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043560682046138417, + "loss": 1.5907, + "theoretical_loss": 3.9560743182512255, + "tokens_seen": 454033408 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043559679037111335, + "loss": 1.4888, + "theoretical_loss": 3.956011793762162, + "tokens_seen": 454098944 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043558676028084253, + "loss": 1.52, + "theoretical_loss": 3.9559492808222334, + "tokens_seen": 454164480 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004355767301905717, + "loss": 1.4839, + "theoretical_loss": 3.9558867794276407, + "tokens_seen": 454230016 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004355667001003009, + "loss": 1.4408, + "theoretical_loss": 3.955824289574586, + "tokens_seen": 454295552 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043555667001003007, + "loss": 1.4512, + "theoretical_loss": 3.955761811259273, + "tokens_seen": 454361088 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004355466399197593, + "loss": 1.5114, + "theoretical_loss": 3.9556993444779076, + "tokens_seen": 454426624 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043553660982948843, + "loss": 1.5645, + "theoretical_loss": 3.9556368892266986, + "tokens_seen": 454492160 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043552657973921767, + "loss": 1.5314, + "theoretical_loss": 3.9555744455018553, + "tokens_seen": 454557696 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043551654964894685, + "loss": 1.5166, + "theoretical_loss": 3.9555120132995887, + "tokens_seen": 454623232 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043550651955867603, + "loss": 1.4207, + "theoretical_loss": 3.955449592616112, + "tokens_seen": 454688768 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004354964894684052, + "loss": 1.4062, + "theoretical_loss": 3.9553871834476406, + "tokens_seen": 454754304 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004354864593781344, + "loss": 1.4698, + "theoretical_loss": 3.955324785790391, + "tokens_seen": 454819840 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004354764292878636, + "loss": 1.516, + "theoretical_loss": 3.9552623996405822, + "tokens_seen": 454885376 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004354663991975928, + "loss": 1.402, + "theoretical_loss": 3.9552000249944337, + "tokens_seen": 454950912 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043545636910732194, + "loss": 1.4072, + "theoretical_loss": 3.9551376618481684, + "tokens_seen": 455016448 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043544633901705117, + "loss": 1.3793, + "theoretical_loss": 3.9550753101980103, + "tokens_seen": 455081984 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004354363089267803, + "loss": 1.5165, + "theoretical_loss": 3.955012970040184, + "tokens_seen": 455147520 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043542627883650953, + "loss": 1.493, + "theoretical_loss": 3.9549506413709183, + "tokens_seen": 455213056 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043541624874623877, + "loss": 1.6111, + "theoretical_loss": 3.954888324186442, + "tokens_seen": 455278592 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004354062186559679, + "loss": 1.4282, + "theoretical_loss": 3.954826018482986, + "tokens_seen": 455344128 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043539618856569713, + "loss": 1.4509, + "theoretical_loss": 3.9547637242567832, + "tokens_seen": 455409664 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.00016934545419644564, + "objective/train/docs_used": 269967, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.835667371749878, + "objective/train/original_loss": 2.8356668949127197, + "objective/train/theoretical_loss": 3.954701441504068, + "objective/train/tokens_used": 475935200, + "objective/train/value_avg": -0.005481719970703125, + "objective/train/value_loss": 4.968090797774494e-05, + "objective/train/value_max": -0.0005593299865722656, + "objective/train/value_min": -0.0823974609375, + "objective/train/value_reward_corr": 0.2854563299017284, + "objective/train/value_std": 0.004123687744140625, + "objective/train/weight_avg": 1.0006624460220337, + "objective/train/weighted_lm_loss": 2.8471314907073975, + "objective/train/weights_max": 1.8316236734390259, + "objective/train/weights_min": 0.33797284960746765, + "theoretical_loss": 3.954701441504068, + "tokens_seen": 455475200 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004353861584754263, + "loss": 1.448, + "theoretical_loss": 3.954701441504068, + "tokens_seen": 455475200 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004353761283851555, + "loss": 1.4018, + "theoretical_loss": 3.9546391702210766, + "tokens_seen": 455540736 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004353660982948847, + "loss": 1.4431, + "theoretical_loss": 3.9545769104040485, + "tokens_seen": 455606272 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043535606820461386, + "loss": 1.4131, + "theoretical_loss": 3.954514662049222, + "tokens_seen": 455671808 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043534603811434304, + "loss": 1.4618, + "theoretical_loss": 3.95445242515284, + "tokens_seen": 455737344 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043533600802407227, + "loss": 1.423, + "theoretical_loss": 3.9543901997111446, + "tokens_seen": 455802880 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004353259779338014, + "loss": 1.5272, + "theoretical_loss": 3.954327985720383, + "tokens_seen": 455868416 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043531594784353063, + "loss": 1.4364, + "theoretical_loss": 3.9542657831768, + "tokens_seen": 455933952 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043530591775325976, + "loss": 1.5396, + "theoretical_loss": 3.9542035920766465, + "tokens_seen": 455999488 + }, + { + "epoch": 0.14, + "learning_rate": 0.000435295887662989, + "loss": 1.5043, + "theoretical_loss": 3.9541414124161713, + "tokens_seen": 456065024 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004352858575727182, + "loss": 1.4702, + "theoretical_loss": 3.954079244191628, + "tokens_seen": 456130560 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043527582748244736, + "loss": 1.4505, + "theoretical_loss": 3.9540170873992704, + "tokens_seen": 456196096 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043526579739217654, + "loss": 1.4269, + "theoretical_loss": 3.9539549420353532, + "tokens_seen": 456261632 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004352557673019057, + "loss": 1.4264, + "theoretical_loss": 3.9538928080961355, + "tokens_seen": 456327168 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004352457372116349, + "loss": 1.4521, + "theoretical_loss": 3.9538306855778766, + "tokens_seen": 456392704 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043523570712136414, + "loss": 1.4671, + "theoretical_loss": 3.953768574476837, + "tokens_seen": 456458240 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043522567703109326, + "loss": 1.4904, + "theoretical_loss": 3.9537064747892794, + "tokens_seen": 456523776 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004352156469408225, + "loss": 1.3518, + "theoretical_loss": 3.953644386511469, + "tokens_seen": 456589312 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004352056168505517, + "loss": 1.4607, + "theoretical_loss": 3.9535823096396725, + "tokens_seen": 456654848 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043519558676028086, + "loss": 1.492, + "theoretical_loss": 3.953520244170157, + "tokens_seen": 456720384 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043518555667001004, + "loss": 1.4369, + "theoretical_loss": 3.9534581900991936, + "tokens_seen": 456785920 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004351755265797392, + "loss": 1.5075, + "theoretical_loss": 3.9533961474230535, + "tokens_seen": 456851456 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004351654964894684, + "loss": 1.4996, + "theoretical_loss": 3.9533341161380102, + "tokens_seen": 456916992 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043515546639919764, + "loss": 1.4818, + "theoretical_loss": 3.9532720962403385, + "tokens_seen": 456982528 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043514543630892677, + "loss": 1.5295, + "theoretical_loss": 3.953210087726316, + "tokens_seen": 457048064 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.0007574320770800114, + "objective/train/docs_used": 270622, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.511439085006714, + "objective/train/original_loss": 3.511439561843872, + "objective/train/theoretical_loss": 3.9531480905922205, + "objective/train/tokens_used": 477573600, + "objective/train/value_avg": -0.00522613525390625, + "objective/train/value_loss": 5.496545054484159e-05, + "objective/train/value_max": -0.0005235671997070312, + "objective/train/value_min": -0.05499267578125, + "objective/train/value_reward_corr": 0.34607098158721333, + "objective/train/value_std": 0.003765106201171875, + "objective/train/weight_avg": 0.9950311779975891, + "objective/train/weighted_lm_loss": 3.4947266578674316, + "objective/train/weights_max": 1.6567894220352173, + "objective/train/weights_min": 0.4757732152938843, + "theoretical_loss": 3.9531480905922205, + "tokens_seen": 457113600 + }, + { + "epoch": 0.14, + "learning_rate": 0.000435135406218656, + "loss": 1.5295, + "theoretical_loss": 3.9531480905922205, + "tokens_seen": 457113600 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043512537612838513, + "loss": 1.3379, + "theoretical_loss": 3.953086104834334, + "tokens_seen": 457179136 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043511534603811437, + "loss": 1.4363, + "theoretical_loss": 3.953024130448937, + "tokens_seen": 457244672 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043510531594784355, + "loss": 1.372, + "theoretical_loss": 3.952962167432314, + "tokens_seen": 457310208 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043509528585757273, + "loss": 1.5006, + "theoretical_loss": 3.9529002157807516, + "tokens_seen": 457375744 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004350852557673019, + "loss": 1.4055, + "theoretical_loss": 3.952838275490536, + "tokens_seen": 457441280 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004350752256770311, + "loss": 1.4713, + "theoretical_loss": 3.9527763465579575, + "tokens_seen": 457506816 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043506519558676027, + "loss": 1.5097, + "theoretical_loss": 3.9527144289793057, + "tokens_seen": 457572352 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004350551654964895, + "loss": 1.514, + "theoretical_loss": 3.952652522750874, + "tokens_seen": 457637888 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043504513540621863, + "loss": 1.456, + "theoretical_loss": 3.9525906278689575, + "tokens_seen": 457703424 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043503510531594787, + "loss": 1.4617, + "theoretical_loss": 3.952528744329851, + "tokens_seen": 457768960 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043502507522567705, + "loss": 1.5715, + "theoretical_loss": 3.952466872129854, + "tokens_seen": 457834496 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043501504513540623, + "loss": 1.4269, + "theoretical_loss": 3.952405011265264, + "tokens_seen": 457900032 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004350050150451354, + "loss": 1.5419, + "theoretical_loss": 3.9523431617323848, + "tokens_seen": 457965568 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004349949849548646, + "loss": 1.5044, + "theoretical_loss": 3.952281323527517, + "tokens_seen": 458031104 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004349849548645938, + "loss": 1.4321, + "theoretical_loss": 3.9522194966469675, + "tokens_seen": 458096640 + }, + { + "epoch": 0.14, + "learning_rate": 0.000434974924774323, + "loss": 1.5252, + "theoretical_loss": 3.9521576810870416, + "tokens_seen": 458162176 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043496489468405214, + "loss": 1.4823, + "theoretical_loss": 3.9520958768440484, + "tokens_seen": 458227712 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043495486459378137, + "loss": 1.4302, + "theoretical_loss": 3.952034083914297, + "tokens_seen": 458293248 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004349448345035105, + "loss": 1.4376, + "theoretical_loss": 3.9519723022941005, + "tokens_seen": 458358784 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043493480441323973, + "loss": 1.514, + "theoretical_loss": 3.951910531979771, + "tokens_seen": 458424320 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004349247743229689, + "loss": 1.4168, + "theoretical_loss": 3.951848772967625, + "tokens_seen": 458489856 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004349147442326981, + "loss": 1.4794, + "theoretical_loss": 3.9517870252539784, + "tokens_seen": 458555392 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004349047141424273, + "loss": 1.5536, + "theoretical_loss": 3.95172528883515, + "tokens_seen": 458620928 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004348946840521565, + "loss": 1.575, + "theoretical_loss": 3.9516635637074615, + "tokens_seen": 458686464 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.0014067540178075433, + "objective/train/docs_used": 271952, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7703187465667725, + "objective/train/original_loss": 2.7703185081481934, + "objective/train/theoretical_loss": 3.951601849867233, + "objective/train/tokens_used": 479212000, + "objective/train/value_avg": -0.004703521728515625, + "objective/train/value_loss": 7.367751823039725e-05, + "objective/train/value_max": -0.0005974769592285156, + "objective/train/value_min": -0.058135986328125, + "objective/train/value_reward_corr": 0.2568346190190547, + "objective/train/value_std": 0.0036563873291015625, + "objective/train/weight_avg": 0.9892362356185913, + "objective/train/weighted_lm_loss": 2.739786386489868, + "objective/train/weights_max": 1.6488186120986938, + "objective/train/weights_min": 0.29798606038093567, + "theoretical_loss": 3.951601849867233, + "tokens_seen": 458752000 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043488465396188564, + "loss": 1.4336, + "theoretical_loss": 3.951601849867233, + "tokens_seen": 458752000 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004348746238716149, + "loss": 1.4826, + "theoretical_loss": 3.95154014731079, + "tokens_seen": 458817536 + }, + { + "epoch": 0.14, + "learning_rate": 0.000434864593781344, + "loss": 1.4527, + "theoretical_loss": 3.951478456034457, + "tokens_seen": 458883072 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043485456369107324, + "loss": 1.4931, + "theoretical_loss": 3.951416776034562, + "tokens_seen": 458948608 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004348445336008024, + "loss": 1.5584, + "theoretical_loss": 3.951355107307434, + "tokens_seen": 459014144 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004348345035105316, + "loss": 1.5049, + "theoretical_loss": 3.951293449849403, + "tokens_seen": 459079680 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004348244734202608, + "loss": 1.5931, + "theoretical_loss": 3.9512318036568024, + "tokens_seen": 459145216 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043481444332998996, + "loss": 1.3695, + "theoretical_loss": 3.9511701687259655, + "tokens_seen": 459210752 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043480441323971914, + "loss": 1.5103, + "theoretical_loss": 3.951108545053229, + "tokens_seen": 459276288 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004347943831494484, + "loss": 1.3832, + "theoretical_loss": 3.9510469326349296, + "tokens_seen": 459341824 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004347843530591775, + "loss": 1.4849, + "theoretical_loss": 3.950985331467407, + "tokens_seen": 459407360 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043477432296890674, + "loss": 1.4705, + "theoretical_loss": 3.9509237415470024, + "tokens_seen": 459472896 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043476429287863587, + "loss": 1.4825, + "theoretical_loss": 3.9508621628700586, + "tokens_seen": 459538432 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004347542627883651, + "loss": 1.5669, + "theoretical_loss": 3.95080059543292, + "tokens_seen": 459603968 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004347442326980943, + "loss": 1.4063, + "theoretical_loss": 3.950739039231933, + "tokens_seen": 459669504 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043473420260782346, + "loss": 1.4358, + "theoretical_loss": 3.950677494263444, + "tokens_seen": 459735040 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043472417251755265, + "loss": 1.5182, + "theoretical_loss": 3.9506159605238045, + "tokens_seen": 459800576 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004347141424272819, + "loss": 1.5227, + "theoretical_loss": 3.9505544380093642, + "tokens_seen": 459866112 + }, + { + "epoch": 0.14, + "learning_rate": 0.000434704112337011, + "loss": 1.5379, + "theoretical_loss": 3.9504929267164774, + "tokens_seen": 459931648 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043469408224674024, + "loss": 1.4056, + "theoretical_loss": 3.950431426641498, + "tokens_seen": 459997184 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004346840521564694, + "loss": 1.5485, + "theoretical_loss": 3.950369937780783, + "tokens_seen": 460062720 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004346740220661986, + "loss": 1.4462, + "theoretical_loss": 3.9503084601306897, + "tokens_seen": 460128256 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043466399197592784, + "loss": 1.5448, + "theoretical_loss": 3.9502469936875784, + "tokens_seen": 460193792 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043465396188565697, + "loss": 1.4884, + "theoretical_loss": 3.950185538447811, + "tokens_seen": 460259328 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004346439317953862, + "loss": 1.4019, + "theoretical_loss": 3.9501240944077494, + "tokens_seen": 460324864 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": 0.0003696820931509137, + "objective/train/docs_used": 272724, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.5325658321380615, + "objective/train/original_loss": 3.5325653553009033, + "objective/train/theoretical_loss": 3.9500626615637597, + "objective/train/tokens_used": 480850400, + "objective/train/value_avg": -0.004638671875, + "objective/train/value_loss": 3.963897324865684e-05, + "objective/train/value_max": -0.00047850608825683594, + "objective/train/value_min": -0.057373046875, + "objective/train/value_reward_corr": 0.46946887082295424, + "objective/train/value_std": 0.004177093505859375, + "objective/train/weight_avg": 1.005618691444397, + "objective/train/weighted_lm_loss": 3.5627989768981934, + "objective/train/weights_max": 1.5699594020843506, + "objective/train/weights_min": 0.4610699415206909, + "theoretical_loss": 3.9500626615637597, + "tokens_seen": 460390400 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043463390170511533, + "loss": 1.571, + "theoretical_loss": 3.9500626615637597, + "tokens_seen": 460390400 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043462387161484457, + "loss": 1.503, + "theoretical_loss": 3.9500012399122078, + "tokens_seen": 460455936 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043461384152457375, + "loss": 1.481, + "theoretical_loss": 3.9499398294494625, + "tokens_seen": 460521472 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043460381143430293, + "loss": 1.422, + "theoretical_loss": 3.949878430171893, + "tokens_seen": 460587008 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004345937813440321, + "loss": 1.4886, + "theoretical_loss": 3.949817042075872, + "tokens_seen": 460652544 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004345837512537613, + "loss": 1.5707, + "theoretical_loss": 3.9497556651577725, + "tokens_seen": 460718080 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043457372116349047, + "loss": 1.5378, + "theoretical_loss": 3.949694299413969, + "tokens_seen": 460783616 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004345636910732197, + "loss": 1.401, + "theoretical_loss": 3.9496329448408387, + "tokens_seen": 460849152 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043455366098294883, + "loss": 1.4622, + "theoretical_loss": 3.9495716014347604, + "tokens_seen": 460914688 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043454363089267807, + "loss": 1.4483, + "theoretical_loss": 3.9495102691921136, + "tokens_seen": 460980224 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043453360080240725, + "loss": 1.4772, + "theoretical_loss": 3.9494489481092803, + "tokens_seen": 461045760 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043452357071213643, + "loss": 1.4157, + "theoretical_loss": 3.9493876381826443, + "tokens_seen": 461111296 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004345135406218656, + "loss": 1.544, + "theoretical_loss": 3.9493263394085902, + "tokens_seen": 461176832 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004345035105315948, + "loss": 1.5325, + "theoretical_loss": 3.9492650517835055, + "tokens_seen": 461242368 + }, + { + "epoch": 0.14, + "learning_rate": 0.000434493480441324, + "loss": 1.3535, + "theoretical_loss": 3.9492037753037783, + "tokens_seen": 461307904 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004344834503510532, + "loss": 1.4681, + "theoretical_loss": 3.949142509965799, + "tokens_seen": 461373440 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043447342026078234, + "loss": 1.4571, + "theoretical_loss": 3.9490812557659596, + "tokens_seen": 461438976 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043446339017051157, + "loss": 1.5255, + "theoretical_loss": 3.949020012700654, + "tokens_seen": 461504512 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004344533600802407, + "loss": 1.5057, + "theoretical_loss": 3.9489587807662763, + "tokens_seen": 461570048 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043444332998996993, + "loss": 1.4642, + "theoretical_loss": 3.9488975599592244, + "tokens_seen": 461635584 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004344332998996991, + "loss": 1.3738, + "theoretical_loss": 3.948836350275897, + "tokens_seen": 461701120 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004344232698094283, + "loss": 1.5921, + "theoretical_loss": 3.9487751517126948, + "tokens_seen": 461766656 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004344132397191575, + "loss": 1.6299, + "theoretical_loss": 3.9487139642660187, + "tokens_seen": 461832192 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004344032096288867, + "loss": 1.4893, + "theoretical_loss": 3.9486527879322733, + "tokens_seen": 461897728 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043439317953861584, + "loss": 1.4483, + "theoretical_loss": 3.9485916227078626, + "tokens_seen": 461963264 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": 0.0002537076943553984, + "objective/train/docs_used": 273842, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3808553218841553, + "objective/train/original_loss": 3.3808553218841553, + "objective/train/theoretical_loss": 3.948530468589195, + "objective/train/tokens_used": 482488800, + "objective/train/value_avg": -0.005474090576171875, + "objective/train/value_loss": 4.375896241981536e-05, + "objective/train/value_max": -0.000614166259765625, + "objective/train/value_min": -0.059417724609375, + "objective/train/value_reward_corr": 0.4120093612456871, + "objective/train/value_std": 0.004337310791015625, + "objective/train/weight_avg": 1.0046648979187012, + "objective/train/weighted_lm_loss": 3.3942315578460693, + "objective/train/weights_max": 1.628719687461853, + "objective/train/weights_min": 0.6424447894096375, + "theoretical_loss": 3.948530468589195, + "tokens_seen": 462028800 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004343831494483451, + "loss": 1.5746, + "theoretical_loss": 3.948530468589195, + "tokens_seen": 462028800 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004343731193580742, + "loss": 1.4903, + "theoretical_loss": 3.9484693255726793, + "tokens_seen": 462094336 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043436308926780344, + "loss": 1.4762, + "theoretical_loss": 3.9484081936547244, + "tokens_seen": 462159872 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004343530591775326, + "loss": 1.4865, + "theoretical_loss": 3.948347072831744, + "tokens_seen": 462225408 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004343430290872618, + "loss": 1.5376, + "theoretical_loss": 3.94828596310015, + "tokens_seen": 462290944 + }, + { + "epoch": 0.14, + "learning_rate": 0.000434332998996991, + "loss": 1.5543, + "theoretical_loss": 3.9482248644563596, + "tokens_seen": 462356480 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043432296890672016, + "loss": 1.3508, + "theoretical_loss": 3.9481637768967883, + "tokens_seen": 462422016 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043431293881644934, + "loss": 1.3927, + "theoretical_loss": 3.9481027004178557, + "tokens_seen": 462487552 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004343029087261786, + "loss": 1.46, + "theoretical_loss": 3.948041635015982, + "tokens_seen": 462553088 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004342928786359077, + "loss": 1.3532, + "theoretical_loss": 3.947980580687589, + "tokens_seen": 462618624 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043428284854563694, + "loss": 1.5051, + "theoretical_loss": 3.947919537429101, + "tokens_seen": 462684160 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043427281845536607, + "loss": 1.5029, + "theoretical_loss": 3.947858505236942, + "tokens_seen": 462749696 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004342627883650953, + "loss": 1.5049, + "theoretical_loss": 3.9477974841075403, + "tokens_seen": 462815232 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004342527582748245, + "loss": 1.4918, + "theoretical_loss": 3.947736474037324, + "tokens_seen": 462880768 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043424272818455367, + "loss": 1.4142, + "theoretical_loss": 3.9476754750227236, + "tokens_seen": 462946304 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043423269809428285, + "loss": 1.4165, + "theoretical_loss": 3.9476144870601706, + "tokens_seen": 463011840 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004342226680040121, + "loss": 1.4146, + "theoretical_loss": 3.9475535101460997, + "tokens_seen": 463077376 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004342126379137412, + "loss": 1.4384, + "theoretical_loss": 3.947492544276945, + "tokens_seen": 463142912 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043420260782347044, + "loss": 1.4828, + "theoretical_loss": 3.947431589449144, + "tokens_seen": 463208448 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043419257773319957, + "loss": 1.4591, + "theoretical_loss": 3.9473706456591353, + "tokens_seen": 463273984 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004341825476429288, + "loss": 1.4855, + "theoretical_loss": 3.947309712903359, + "tokens_seen": 463339520 + }, + { + "epoch": 0.14, + "learning_rate": 0.000434172517552658, + "loss": 1.4441, + "theoretical_loss": 3.947248791178257, + "tokens_seen": 463405056 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043416248746238717, + "loss": 1.6121, + "theoretical_loss": 3.9471878804802736, + "tokens_seen": 463470592 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043415245737211635, + "loss": 1.5554, + "theoretical_loss": 3.9471269808058524, + "tokens_seen": 463536128 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043414242728184553, + "loss": 1.3957, + "theoretical_loss": 3.9470660921514424, + "tokens_seen": 463601664 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.0019313010852783918, + "objective/train/docs_used": 274430, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.700477361679077, + "objective/train/original_loss": 2.7004776000976562, + "objective/train/theoretical_loss": 3.94700521451349, + "objective/train/tokens_used": 484127200, + "objective/train/value_avg": -0.004924774169921875, + "objective/train/value_loss": 8.6499989265576e-05, + "objective/train/value_max": -0.00042724609375, + "objective/train/value_min": -0.06463623046875, + "objective/train/value_reward_corr": 0.26066309171864355, + "objective/train/value_std": 0.00366973876953125, + "objective/train/weight_avg": 0.9845636487007141, + "objective/train/weighted_lm_loss": 2.696133613586426, + "objective/train/weights_max": 1.721003770828247, + "objective/train/weights_min": 0.037059009075164795, + "theoretical_loss": 3.94700521451349, + "tokens_seen": 463667200 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004341323971915747, + "loss": 1.4576, + "theoretical_loss": 3.94700521451349, + "tokens_seen": 463667200 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043412236710130395, + "loss": 1.4179, + "theoretical_loss": 3.9469443478884463, + "tokens_seen": 463732736 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004341123370110331, + "loss": 1.4756, + "theoretical_loss": 3.9468834922727636, + "tokens_seen": 463798272 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004341023069207623, + "loss": 1.4645, + "theoretical_loss": 3.946822647662894, + "tokens_seen": 463863808 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043409227683049144, + "loss": 1.3559, + "theoretical_loss": 3.946761814055294, + "tokens_seen": 463929344 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043408224674022067, + "loss": 1.4333, + "theoretical_loss": 3.9467009914464186, + "tokens_seen": 463994880 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043407221664994985, + "loss": 1.4826, + "theoretical_loss": 3.946640179832728, + "tokens_seen": 464060416 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043406218655967903, + "loss": 1.4628, + "theoretical_loss": 3.9465793792106814, + "tokens_seen": 464125952 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004340521564694082, + "loss": 1.4563, + "theoretical_loss": 3.94651858957674, + "tokens_seen": 464191488 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043404212637913745, + "loss": 1.5485, + "theoretical_loss": 3.9464578109273676, + "tokens_seen": 464257024 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004340320962888666, + "loss": 1.412, + "theoretical_loss": 3.9463970432590294, + "tokens_seen": 464322560 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004340220661985958, + "loss": 1.4248, + "theoretical_loss": 3.9463362865681915, + "tokens_seen": 464388096 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043401203610832494, + "loss": 1.5613, + "theoretical_loss": 3.946275540851322, + "tokens_seen": 464453632 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004340020060180542, + "loss": 1.404, + "theoretical_loss": 3.9462148061048907, + "tokens_seen": 464519168 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043399197592778336, + "loss": 1.476, + "theoretical_loss": 3.9461540823253696, + "tokens_seen": 464584704 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043398194583751254, + "loss": 1.5019, + "theoretical_loss": 3.9460933695092306, + "tokens_seen": 464650240 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004339719157472417, + "loss": 1.4181, + "theoretical_loss": 3.9460326676529496, + "tokens_seen": 464715776 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004339618856569709, + "loss": 1.4981, + "theoretical_loss": 3.945971976753003, + "tokens_seen": 464781312 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004339518555667001, + "loss": 1.3705, + "theoretical_loss": 3.9459112968058685, + "tokens_seen": 464846848 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004339418254764293, + "loss": 1.4557, + "theoretical_loss": 3.945850627808025, + "tokens_seen": 464912384 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004339317953861585, + "loss": 1.5937, + "theoretical_loss": 3.9457899697559546, + "tokens_seen": 464977920 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004339217652958877, + "loss": 1.6135, + "theoretical_loss": 3.9457293226461396, + "tokens_seen": 465043456 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004339117352056169, + "loss": 1.4665, + "theoretical_loss": 3.9456686864750647, + "tokens_seen": 465108992 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043390170511534604, + "loss": 1.5636, + "theoretical_loss": 3.9456080612392164, + "tokens_seen": 465174528 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004338916750250753, + "loss": 1.4881, + "theoretical_loss": 3.945547446935082, + "tokens_seen": 465240064 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.0009500162559561431, + "objective/train/docs_used": 275809, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.726356029510498, + "objective/train/original_loss": 2.726356029510498, + "objective/train/theoretical_loss": 3.9454868435591504, + "objective/train/tokens_used": 485765600, + "objective/train/value_avg": -0.005336761474609375, + "objective/train/value_loss": 7.541900413343683e-05, + "objective/train/value_max": -0.00042891502380371094, + "objective/train/value_min": -0.08526611328125, + "objective/train/value_reward_corr": 0.3207575617406759, + "objective/train/value_std": 0.005100250244140625, + "objective/train/weight_avg": 0.9940387010574341, + "objective/train/weighted_lm_loss": 2.7278831005096436, + "objective/train/weights_max": 1.9668649435043335, + "objective/train/weights_min": 0.24781832098960876, + "theoretical_loss": 3.9454868435591504, + "tokens_seen": 465305600 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004338816449348044, + "loss": 1.3295, + "theoretical_loss": 3.9454868435591504, + "tokens_seen": 465305600 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043387161484453364, + "loss": 1.4108, + "theoretical_loss": 3.9454262511079135, + "tokens_seen": 465371136 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004338615847542628, + "loss": 1.4063, + "theoretical_loss": 3.9453656695778636, + "tokens_seen": 465436672 + }, + { + "epoch": 0.14, + "learning_rate": 0.000433851554663992, + "loss": 1.4711, + "theoretical_loss": 3.9453050989654948, + "tokens_seen": 465502208 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004338415245737212, + "loss": 1.444, + "theoretical_loss": 3.945244539267303, + "tokens_seen": 465567744 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043383149448345036, + "loss": 1.3618, + "theoretical_loss": 3.9451839904797854, + "tokens_seen": 465633280 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043382146439317954, + "loss": 1.5222, + "theoretical_loss": 3.945123452599441, + "tokens_seen": 465698816 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004338114343029088, + "loss": 1.3861, + "theoretical_loss": 3.9450629256227714, + "tokens_seen": 465764352 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004338014042126379, + "loss": 1.4047, + "theoretical_loss": 3.9450024095462783, + "tokens_seen": 465829888 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043379137412236714, + "loss": 1.5069, + "theoretical_loss": 3.9449419043664653, + "tokens_seen": 465895424 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043378134403209627, + "loss": 1.4988, + "theoretical_loss": 3.944881410079838, + "tokens_seen": 465960960 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004337713139418255, + "loss": 1.5562, + "theoretical_loss": 3.9448209266829046, + "tokens_seen": 466026496 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004337612838515547, + "loss": 1.5266, + "theoretical_loss": 3.9447604541721724, + "tokens_seen": 466092032 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043375125376128387, + "loss": 1.3995, + "theoretical_loss": 3.9446999925441517, + "tokens_seen": 466157568 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043374122367101305, + "loss": 1.448, + "theoretical_loss": 3.944639541795356, + "tokens_seen": 466223104 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004337311935807423, + "loss": 1.522, + "theoretical_loss": 3.944579101922298, + "tokens_seen": 466288640 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004337211634904714, + "loss": 1.4858, + "theoretical_loss": 3.944518672921493, + "tokens_seen": 466354176 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043371113340020064, + "loss": 1.4495, + "theoretical_loss": 3.944458254789458, + "tokens_seen": 466419712 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043370110330992977, + "loss": 1.4475, + "theoretical_loss": 3.94439784752271, + "tokens_seen": 466485248 + }, + { + "epoch": 0.14, + "learning_rate": 0.000433691073219659, + "loss": 1.4486, + "theoretical_loss": 3.9443374511177707, + "tokens_seen": 466550784 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004336810431293882, + "loss": 1.448, + "theoretical_loss": 3.9442770655711614, + "tokens_seen": 466616320 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043367101303911737, + "loss": 1.367, + "theoretical_loss": 3.9442166908794047, + "tokens_seen": 466681856 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043366098294884655, + "loss": 1.3978, + "theoretical_loss": 3.9441563270390256, + "tokens_seen": 466747392 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043365095285857573, + "loss": 1.5099, + "theoretical_loss": 3.944095974046551, + "tokens_seen": 466812928 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004336409227683049, + "loss": 1.4309, + "theoretical_loss": 3.9440356318985086, + "tokens_seen": 466878464 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.0024078802671283484, + "objective/train/docs_used": 276320, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.824169635772705, + "objective/train/original_loss": 2.824169158935547, + "objective/train/theoretical_loss": 3.9439753005914273, + "objective/train/tokens_used": 487404000, + "objective/train/value_avg": -0.00583648681640625, + "objective/train/value_loss": 9.203665103996173e-05, + "objective/train/value_max": -0.0005550384521484375, + "objective/train/value_min": -0.0831298828125, + "objective/train/value_reward_corr": 0.42068211461600996, + "objective/train/value_std": 0.004756927490234375, + "objective/train/weight_avg": 0.9800838232040405, + "objective/train/weighted_lm_loss": 2.768901824951172, + "objective/train/weights_max": 2.296299457550049, + "objective/train/weights_min": 0.08571707457304001, + "theoretical_loss": 3.9439753005914273, + "tokens_seen": 466944000 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043363089267803415, + "loss": 1.4087, + "theoretical_loss": 3.9439753005914273, + "tokens_seen": 466944000 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004336208625877633, + "loss": 1.5798, + "theoretical_loss": 3.9439149801218396, + "tokens_seen": 467009536 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004336108324974925, + "loss": 1.4881, + "theoretical_loss": 3.9438546704862776, + "tokens_seen": 467075072 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043360080240722164, + "loss": 1.4838, + "theoretical_loss": 3.9437943716812756, + "tokens_seen": 467140608 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043359077231695087, + "loss": 1.4629, + "theoretical_loss": 3.9437340837033696, + "tokens_seen": 467206144 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043358074222668005, + "loss": 1.4168, + "theoretical_loss": 3.9436738065490973, + "tokens_seen": 467271680 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043357071213640923, + "loss": 1.5962, + "theoretical_loss": 3.943613540214998, + "tokens_seen": 467337216 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004335606820461384, + "loss": 1.3687, + "theoretical_loss": 3.9435532846976127, + "tokens_seen": 467402752 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043355065195586765, + "loss": 1.4459, + "theoretical_loss": 3.943493039993483, + "tokens_seen": 467468288 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004335406218655968, + "loss": 1.4842, + "theoretical_loss": 3.9434328060991533, + "tokens_seen": 467533824 + }, + { + "epoch": 0.14, + "learning_rate": 0.000433530591775326, + "loss": 1.6496, + "theoretical_loss": 3.943372583011169, + "tokens_seen": 467599360 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043352056168505514, + "loss": 1.5916, + "theoretical_loss": 3.9433123707260775, + "tokens_seen": 467664896 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004335105315947844, + "loss": 1.4899, + "theoretical_loss": 3.9432521692404277, + "tokens_seen": 467730432 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043350050150451356, + "loss": 1.6054, + "theoretical_loss": 3.943191978550769, + "tokens_seen": 467795968 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043349047141424274, + "loss": 1.4514, + "theoretical_loss": 3.943131798653654, + "tokens_seen": 467861504 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004334804413239719, + "loss": 1.3459, + "theoretical_loss": 3.9430716295456363, + "tokens_seen": 467927040 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004334704112337011, + "loss": 1.4746, + "theoretical_loss": 3.9430114712232704, + "tokens_seen": 467992576 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004334603811434303, + "loss": 1.5219, + "theoretical_loss": 3.942951323683113, + "tokens_seen": 468058112 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004334503510531595, + "loss": 1.4605, + "theoretical_loss": 3.942891186921723, + "tokens_seen": 468123648 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043344032096288864, + "loss": 1.4823, + "theoretical_loss": 3.942831060935659, + "tokens_seen": 468189184 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004334302908726179, + "loss": 1.4705, + "theoretical_loss": 3.9427709457214837, + "tokens_seen": 468254720 + }, + { + "epoch": 0.14, + "learning_rate": 0.000433420260782347, + "loss": 1.4007, + "theoretical_loss": 3.942710841275759, + "tokens_seen": 468320256 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043341023069207624, + "loss": 1.443, + "theoretical_loss": 3.9426507475950503, + "tokens_seen": 468385792 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004334002006018054, + "loss": 1.4042, + "theoretical_loss": 3.9425906646759232, + "tokens_seen": 468451328 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004333901705115346, + "loss": 1.4031, + "theoretical_loss": 3.942530592514945, + "tokens_seen": 468516864 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": 0.00025512371212244034, + "objective/train/docs_used": 277290, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7629520893096924, + "objective/train/original_loss": 2.7629525661468506, + "objective/train/theoretical_loss": 3.9424705311086856, + "objective/train/tokens_used": 489042400, + "objective/train/value_avg": -0.005306243896484375, + "objective/train/value_loss": 5.357876216294244e-05, + "objective/train/value_max": -0.0004425048828125, + "objective/train/value_min": -0.058563232421875, + "objective/train/value_reward_corr": 0.2973386703629306, + "objective/train/value_std": 0.00395965576171875, + "objective/train/weight_avg": 1.0050854682922363, + "objective/train/weighted_lm_loss": 2.763037919998169, + "objective/train/weights_max": 1.513967752456665, + "objective/train/weights_min": 0.3819945752620697, + "theoretical_loss": 3.9424705311086856, + "tokens_seen": 468582400 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004333801404212638, + "loss": 1.4795, + "theoretical_loss": 3.9424705311086856, + "tokens_seen": 468582400 + }, + { + "epoch": 0.14, + "learning_rate": 0.000433370110330993, + "loss": 1.4178, + "theoretical_loss": 3.9424104804537157, + "tokens_seen": 468647936 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043336008024072215, + "loss": 1.5641, + "theoretical_loss": 3.9423504405466074, + "tokens_seen": 468713472 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004333500501504514, + "loss": 1.4921, + "theoretical_loss": 3.942290411383935, + "tokens_seen": 468779008 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004333400200601805, + "loss": 1.5044, + "theoretical_loss": 3.942230392962274, + "tokens_seen": 468844544 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043332998996990974, + "loss": 1.5362, + "theoretical_loss": 3.942170385278202, + "tokens_seen": 468910080 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004333199598796389, + "loss": 1.5494, + "theoretical_loss": 3.942110388328296, + "tokens_seen": 468975616 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004333099297893681, + "loss": 1.4358, + "theoretical_loss": 3.9420504021091385, + "tokens_seen": 469041152 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004332998996990973, + "loss": 1.397, + "theoretical_loss": 3.94199042661731, + "tokens_seen": 469106688 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043328986960882647, + "loss": 1.5886, + "theoretical_loss": 3.9419304618493936, + "tokens_seen": 469172224 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043327983951855565, + "loss": 1.2837, + "theoretical_loss": 3.9418705078019753, + "tokens_seen": 469237760 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004332698094282849, + "loss": 1.4373, + "theoretical_loss": 3.941810564471641, + "tokens_seen": 469303296 + }, + { + "epoch": 0.14, + "learning_rate": 0.000433259779338014, + "loss": 1.5059, + "theoretical_loss": 3.9417506318549784, + "tokens_seen": 469368832 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043324974924774325, + "loss": 1.5259, + "theoretical_loss": 3.9416907099485776, + "tokens_seen": 469434368 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043323971915747243, + "loss": 1.381, + "theoretical_loss": 3.9416307987490304, + "tokens_seen": 469499904 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004332296890672016, + "loss": 1.4602, + "theoretical_loss": 3.9415708982529285, + "tokens_seen": 469565440 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004332196589769308, + "loss": 1.4997, + "theoretical_loss": 3.9415110084568665, + "tokens_seen": 469630976 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043320962888665997, + "loss": 1.4812, + "theoretical_loss": 3.941451129357441, + "tokens_seen": 469696512 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043319959879638915, + "loss": 1.412, + "theoretical_loss": 3.9413912609512485, + "tokens_seen": 469762048 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004331895687061184, + "loss": 1.4542, + "theoretical_loss": 3.9413314032348885, + "tokens_seen": 469827584 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043317953861584757, + "loss": 1.5232, + "theoretical_loss": 3.9412715562049616, + "tokens_seen": 469893120 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043316950852557675, + "loss": 1.4501, + "theoretical_loss": 3.94121171985807, + "tokens_seen": 469958656 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043315947843530593, + "loss": 1.4309, + "theoretical_loss": 3.9411518941908166, + "tokens_seen": 470024192 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004331494483450351, + "loss": 1.4675, + "theoretical_loss": 3.9410920791998074, + "tokens_seen": 470089728 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043313941825476435, + "loss": 1.4895, + "theoretical_loss": 3.941032274881649, + "tokens_seen": 470155264 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.00039914037915878, + "objective/train/docs_used": 277851, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.033872365951538, + "objective/train/original_loss": 3.033872127532959, + "objective/train/theoretical_loss": 3.940972481232949, + "objective/train/tokens_used": 490680800, + "objective/train/value_avg": -0.0060882568359375, + "objective/train/value_loss": 9.892592788673937e-05, + "objective/train/value_max": -0.000469207763671875, + "objective/train/value_min": -0.07421875, + "objective/train/value_reward_corr": 0.47091742015516447, + "objective/train/value_std": 0.006103515625, + "objective/train/weight_avg": 1.0004425048828125, + "objective/train/weighted_lm_loss": 3.03958797454834, + "objective/train/weights_max": 1.836028814315796, + "objective/train/weights_min": 0.2501786947250366, + "theoretical_loss": 3.940972481232949, + "tokens_seen": 470220800 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004331293881644935, + "loss": 1.4921, + "theoretical_loss": 3.940972481232949, + "tokens_seen": 470220800 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004331193580742227, + "loss": 1.3558, + "theoretical_loss": 3.9409126982503184, + "tokens_seen": 470286336 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043310932798395184, + "loss": 1.4361, + "theoretical_loss": 3.9408529259303684, + "tokens_seen": 470351872 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043309929789368107, + "loss": 1.4231, + "theoretical_loss": 3.940793164269711, + "tokens_seen": 470417408 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043308926780341025, + "loss": 1.3939, + "theoretical_loss": 3.9407334132649616, + "tokens_seen": 470482944 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043307923771313943, + "loss": 1.4277, + "theoretical_loss": 3.9406736729127365, + "tokens_seen": 470548480 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004330692076228686, + "loss": 1.4614, + "theoretical_loss": 3.9406139432096525, + "tokens_seen": 470614016 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043305917753259785, + "loss": 1.5312, + "theoretical_loss": 3.940554224152329, + "tokens_seen": 470679552 + }, + { + "epoch": 0.14, + "learning_rate": 0.000433049147442327, + "loss": 1.5749, + "theoretical_loss": 3.9404945157373867, + "tokens_seen": 470745088 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004330391173520562, + "loss": 1.4743, + "theoretical_loss": 3.940434817961448, + "tokens_seen": 470810624 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043302908726178534, + "loss": 1.4075, + "theoretical_loss": 3.9403751308211366, + "tokens_seen": 470876160 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004330190571715146, + "loss": 1.4689, + "theoretical_loss": 3.9403154543130783, + "tokens_seen": 470941696 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043300902708124376, + "loss": 1.5978, + "theoretical_loss": 3.940255788433899, + "tokens_seen": 471007232 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043299899699097294, + "loss": 1.5111, + "theoretical_loss": 3.940196133180228, + "tokens_seen": 471072768 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004329889669007021, + "loss": 1.4546, + "theoretical_loss": 3.9401364885486942, + "tokens_seen": 471138304 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004329789368104313, + "loss": 1.4754, + "theoretical_loss": 3.94007685453593, + "tokens_seen": 471203840 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004329689067201605, + "loss": 1.4838, + "theoretical_loss": 3.9400172311385684, + "tokens_seen": 471269376 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004329588766298897, + "loss": 1.4437, + "theoretical_loss": 3.939957618353243, + "tokens_seen": 471334912 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043294884653961884, + "loss": 1.4862, + "theoretical_loss": 3.9398980161765915, + "tokens_seen": 471400448 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004329388164493481, + "loss": 1.415, + "theoretical_loss": 3.9398384246052496, + "tokens_seen": 471465984 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004329287863590772, + "loss": 1.3562, + "theoretical_loss": 3.9397788436358585, + "tokens_seen": 471531520 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043291875626880644, + "loss": 1.3563, + "theoretical_loss": 3.939719273265057, + "tokens_seen": 471597056 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004329087261785356, + "loss": 1.3876, + "theoretical_loss": 3.9396597134894886, + "tokens_seen": 471662592 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004328986960882648, + "loss": 1.4341, + "theoretical_loss": 3.939600164305796, + "tokens_seen": 471728128 + }, + { + "epoch": 0.14, + "learning_rate": 0.000432888665997994, + "loss": 1.5471, + "theoretical_loss": 3.9395406257106256, + "tokens_seen": 471793664 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.001278066891245544, + "objective/train/docs_used": 279270, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.317426919937134, + "objective/train/original_loss": 2.317426919937134, + "objective/train/theoretical_loss": 3.939481097700623, + "objective/train/tokens_used": 492319200, + "objective/train/value_avg": -0.004863739013671875, + "objective/train/value_loss": 0.00011158717097714543, + "objective/train/value_max": -0.0005154609680175781, + "objective/train/value_min": -0.1756591796875, + "objective/train/value_reward_corr": 0.26247189939854804, + "objective/train/value_std": 0.0045318603515625, + "objective/train/weight_avg": 0.9922674298286438, + "objective/train/weighted_lm_loss": 2.3098490238189697, + "objective/train/weights_max": 4.774813652038574, + "objective/train/weights_min": 0.2993699908256531, + "theoretical_loss": 3.939481097700623, + "tokens_seen": 471859200 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004328786359077232, + "loss": 1.3264, + "theoretical_loss": 3.939481097700623, + "tokens_seen": 471859200 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043286860581745235, + "loss": 1.3604, + "theoretical_loss": 3.9394215802724375, + "tokens_seen": 471924736 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004328585757271816, + "loss": 1.4553, + "theoretical_loss": 3.9393620734227186, + "tokens_seen": 471990272 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004328485456369107, + "loss": 1.4186, + "theoretical_loss": 3.9393025771481174, + "tokens_seen": 472055808 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043283851554663994, + "loss": 1.5415, + "theoretical_loss": 3.939243091445287, + "tokens_seen": 472121344 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004328284854563691, + "loss": 1.3685, + "theoretical_loss": 3.9391836163108827, + "tokens_seen": 472186880 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004328184553660983, + "loss": 1.5165, + "theoretical_loss": 3.9391241517415594, + "tokens_seen": 472252416 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004328084252758275, + "loss": 1.5133, + "theoretical_loss": 3.9390646977339747, + "tokens_seen": 472317952 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043279839518555667, + "loss": 1.4274, + "theoretical_loss": 3.9390052542847878, + "tokens_seen": 472383488 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043278836509528585, + "loss": 1.4466, + "theoretical_loss": 3.938945821390659, + "tokens_seen": 472449024 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004327783350050151, + "loss": 1.4664, + "theoretical_loss": 3.9388863990482506, + "tokens_seen": 472514560 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004327683049147442, + "loss": 1.3486, + "theoretical_loss": 3.938826987254226, + "tokens_seen": 472580096 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043275827482447345, + "loss": 1.4196, + "theoretical_loss": 3.9387675860052505, + "tokens_seen": 472645632 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043274824473420263, + "loss": 1.3661, + "theoretical_loss": 3.9387081952979903, + "tokens_seen": 472711168 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004327382146439318, + "loss": 1.4547, + "theoretical_loss": 3.938648815129114, + "tokens_seen": 472776704 + }, + { + "epoch": 0.14, + "learning_rate": 0.000432728184553661, + "loss": 1.4503, + "theoretical_loss": 3.9385894454952908, + "tokens_seen": 472842240 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043271815446339017, + "loss": 1.4087, + "theoretical_loss": 3.9385300863931914, + "tokens_seen": 472907776 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043270812437311935, + "loss": 1.4565, + "theoretical_loss": 3.9384707378194896, + "tokens_seen": 472973312 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004326980942828486, + "loss": 1.5056, + "theoretical_loss": 3.9384113997708585, + "tokens_seen": 473038848 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004326880641925777, + "loss": 1.4038, + "theoretical_loss": 3.9383520722439744, + "tokens_seen": 473104384 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043267803410230695, + "loss": 1.5458, + "theoretical_loss": 3.9382927552355147, + "tokens_seen": 473169920 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004326680040120361, + "loss": 1.2952, + "theoretical_loss": 3.9382334487421575, + "tokens_seen": 473235456 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004326579739217653, + "loss": 1.5319, + "theoretical_loss": 3.938174152760583, + "tokens_seen": 473300992 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004326479438314945, + "loss": 1.3751, + "theoretical_loss": 3.9381148672874735, + "tokens_seen": 473366528 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004326379137412237, + "loss": 1.4445, + "theoretical_loss": 3.938055592319511, + "tokens_seen": 473432064 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.0014715319266542792, + "objective/train/docs_used": 279967, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.932323932647705, + "objective/train/original_loss": 2.932323455810547, + "objective/train/theoretical_loss": 3.937996327853382, + "objective/train/tokens_used": 493957600, + "objective/train/value_avg": -0.006015777587890625, + "objective/train/value_loss": 6.476220733020455e-05, + "objective/train/value_max": -0.00047278404235839844, + "objective/train/value_min": -0.0706787109375, + "objective/train/value_reward_corr": 0.4411889094365081, + "objective/train/value_std": 0.004772186279296875, + "objective/train/weight_avg": 0.9883711934089661, + "objective/train/weighted_lm_loss": 2.9146130084991455, + "objective/train/weights_max": 2.0274667739868164, + "objective/train/weights_min": 0.5767887830734253, + "theoretical_loss": 3.937996327853382, + "tokens_seen": 473497600 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043262788365095286, + "loss": 1.4511, + "theoretical_loss": 3.937996327853382, + "tokens_seen": 473497600 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043261785356068204, + "loss": 1.5242, + "theoretical_loss": 3.9379370738857706, + "tokens_seen": 473563136 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004326078234704112, + "loss": 1.4447, + "theoretical_loss": 3.937877830413366, + "tokens_seen": 473628672 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043259779338014045, + "loss": 1.5264, + "theoretical_loss": 3.9378185974328574, + "tokens_seen": 473694208 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004325877632898696, + "loss": 1.4915, + "theoretical_loss": 3.937759374940935, + "tokens_seen": 473759744 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004325777331995988, + "loss": 1.4922, + "theoretical_loss": 3.9377001629342914, + "tokens_seen": 473825280 + }, + { + "epoch": 0.14, + "learning_rate": 0.000432567703109328, + "loss": 1.412, + "theoretical_loss": 3.9376409614096204, + "tokens_seen": 473890816 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004325576730190572, + "loss": 1.5569, + "theoretical_loss": 3.9375817703636167, + "tokens_seen": 473956352 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043254764292878636, + "loss": 1.5551, + "theoretical_loss": 3.937522589792977, + "tokens_seen": 474021888 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043253761283851554, + "loss": 1.4433, + "theoretical_loss": 3.9374634196944003, + "tokens_seen": 474087424 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004325275827482447, + "loss": 1.5193, + "theoretical_loss": 3.9374042600645858, + "tokens_seen": 474152960 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043251755265797396, + "loss": 1.466, + "theoretical_loss": 3.9373451109002344, + "tokens_seen": 474218496 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004325075225677031, + "loss": 1.4163, + "theoretical_loss": 3.93728597219805, + "tokens_seen": 474284032 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004324974924774323, + "loss": 1.5213, + "theoretical_loss": 3.9372268439547353, + "tokens_seen": 474349568 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043248746238716145, + "loss": 1.413, + "theoretical_loss": 3.9371677261669973, + "tokens_seen": 474415104 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004324774322968907, + "loss": 1.5222, + "theoretical_loss": 3.9371086188315427, + "tokens_seen": 474480640 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043246740220661986, + "loss": 1.4955, + "theoretical_loss": 3.93704952194508, + "tokens_seen": 474546176 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043245737211634904, + "loss": 1.5175, + "theoretical_loss": 3.936990435504319, + "tokens_seen": 474611712 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004324473420260782, + "loss": 1.4733, + "theoretical_loss": 3.936931359505972, + "tokens_seen": 474677248 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004324373119358074, + "loss": 1.5085, + "theoretical_loss": 3.9368722939467524, + "tokens_seen": 474742784 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043242728184553664, + "loss": 1.5141, + "theoretical_loss": 3.9368132388233743, + "tokens_seen": 474808320 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004324172517552658, + "loss": 1.4446, + "theoretical_loss": 3.936754194132554, + "tokens_seen": 474873856 + }, + { + "epoch": 0.14, + "learning_rate": 0.000432407221664995, + "loss": 1.4357, + "theoretical_loss": 3.936695159871009, + "tokens_seen": 474939392 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004323971915747242, + "loss": 1.4977, + "theoretical_loss": 3.9366361360354585, + "tokens_seen": 475004928 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004323871614844534, + "loss": 1.3994, + "theoretical_loss": 3.936577122622624, + "tokens_seen": 475070464 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.0010854334104806185, + "objective/train/docs_used": 280356, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0357110500335693, + "objective/train/original_loss": 3.0357112884521484, + "objective/train/theoretical_loss": 3.936518119629225, + "objective/train/tokens_used": 495596000, + "objective/train/value_avg": -0.00446319580078125, + "objective/train/value_loss": 4.4190881453687325e-05, + "objective/train/value_max": -0.0004565715789794922, + "objective/train/value_min": -0.0615234375, + "objective/train/value_reward_corr": 0.3887160831338593, + "objective/train/value_std": 0.0033321380615234375, + "objective/train/weight_avg": 0.9912521839141846, + "objective/train/weighted_lm_loss": 3.013291835784912, + "objective/train/weights_max": 1.8500901460647583, + "objective/train/weights_min": 0.5425696969032288, + "theoretical_loss": 3.936518119629225, + "tokens_seen": 475136000 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043237713139418255, + "loss": 1.491, + "theoretical_loss": 3.936518119629225, + "tokens_seen": 475136000 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004323671013039118, + "loss": 1.5509, + "theoretical_loss": 3.9364591270519877, + "tokens_seen": 475201536 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004323570712136409, + "loss": 1.5504, + "theoretical_loss": 3.936400144887637, + "tokens_seen": 475267072 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043234704112337014, + "loss": 1.4278, + "theoretical_loss": 3.9363411731328974, + "tokens_seen": 475332608 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004323370110330993, + "loss": 1.4282, + "theoretical_loss": 3.9362822117844987, + "tokens_seen": 475398144 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004323269809428285, + "loss": 1.5004, + "theoretical_loss": 3.936223260839169, + "tokens_seen": 475463680 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004323169508525577, + "loss": 1.3367, + "theoretical_loss": 3.936164320293641, + "tokens_seen": 475529216 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043230692076228687, + "loss": 1.4452, + "theoretical_loss": 3.9361053901446454, + "tokens_seen": 475594752 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043229689067201605, + "loss": 1.468, + "theoretical_loss": 3.9360464703889173, + "tokens_seen": 475660288 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004322868605817453, + "loss": 1.4012, + "theoretical_loss": 3.935987561023192, + "tokens_seen": 475725824 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004322768304914744, + "loss": 1.4127, + "theoretical_loss": 3.9359286620442058, + "tokens_seen": 475791360 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043226680040120365, + "loss": 1.4607, + "theoretical_loss": 3.9358697734486965, + "tokens_seen": 475856896 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043225677031093283, + "loss": 1.5495, + "theoretical_loss": 3.9358108952334057, + "tokens_seen": 475922432 + }, + { + "epoch": 0.14, + "learning_rate": 0.000432246740220662, + "loss": 1.4932, + "theoretical_loss": 3.9357520273950732, + "tokens_seen": 475987968 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004322367101303912, + "loss": 1.4187, + "theoretical_loss": 3.9356931699304427, + "tokens_seen": 476053504 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043222668004012037, + "loss": 1.4906, + "theoretical_loss": 3.9356343228362567, + "tokens_seen": 476119040 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043221664994984955, + "loss": 1.4461, + "theoretical_loss": 3.935575486109263, + "tokens_seen": 476184576 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004322066198595788, + "loss": 1.4171, + "theoretical_loss": 3.9355166597462086, + "tokens_seen": 476250112 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004321965897693079, + "loss": 1.3398, + "theoretical_loss": 3.9354578437438406, + "tokens_seen": 476315648 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043218655967903715, + "loss": 1.4026, + "theoretical_loss": 3.9353990380989097, + "tokens_seen": 476381184 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004321765295887663, + "loss": 1.322, + "theoretical_loss": 3.935340242808168, + "tokens_seen": 476446720 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004321664994984955, + "loss": 1.4638, + "theoretical_loss": 3.935281457868368, + "tokens_seen": 476512256 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004321564694082247, + "loss": 1.3723, + "theoretical_loss": 3.9352226832762645, + "tokens_seen": 476577792 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004321464393179539, + "loss": 1.593, + "theoretical_loss": 3.9351639190286134, + "tokens_seen": 476643328 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043213640922768306, + "loss": 1.5622, + "theoretical_loss": 3.935105165122172, + "tokens_seen": 476708864 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -1.9764773242059164e-05, + "objective/train/docs_used": 281398, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.254239797592163, + "objective/train/original_loss": 3.254239797592163, + "objective/train/theoretical_loss": 3.9350464215537, + "objective/train/tokens_used": 497234400, + "objective/train/value_avg": -0.00551605224609375, + "objective/train/value_loss": 5.45281327504199e-05, + "objective/train/value_max": -0.0006046295166015625, + "objective/train/value_min": -0.08343505859375, + "objective/train/value_reward_corr": 0.47025637600960596, + "objective/train/value_std": 0.00499725341796875, + "objective/train/weight_avg": 1.0024135112762451, + "objective/train/weighted_lm_loss": 3.2566514015197754, + "objective/train/weights_max": 2.3033177852630615, + "objective/train/weights_min": 0.5260658264160156, + "theoretical_loss": 3.9350464215537, + "tokens_seen": 476774400 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043212637913741224, + "loss": 1.4165, + "theoretical_loss": 3.9350464215537, + "tokens_seen": 476774400 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004321163490471414, + "loss": 1.4827, + "theoretical_loss": 3.934987688319956, + "tokens_seen": 476839936 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043210631895687065, + "loss": 1.4363, + "theoretical_loss": 3.9349289654177033, + "tokens_seen": 476905472 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004320962888665998, + "loss": 1.4671, + "theoretical_loss": 3.934870252843705, + "tokens_seen": 476971008 + }, + { + "epoch": 0.14, + "learning_rate": 0.000432086258776329, + "loss": 1.4454, + "theoretical_loss": 3.934811550594725, + "tokens_seen": 477036544 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004320762286860582, + "loss": 1.451, + "theoretical_loss": 3.9347528586675304, + "tokens_seen": 477102080 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004320661985957874, + "loss": 1.5977, + "theoretical_loss": 3.934694177058888, + "tokens_seen": 477167616 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043205616850551656, + "loss": 1.5157, + "theoretical_loss": 3.9346355057655673, + "tokens_seen": 477233152 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043204613841524574, + "loss": 1.3134, + "theoretical_loss": 3.9345768447843392, + "tokens_seen": 477298688 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004320361083249749, + "loss": 1.569, + "theoretical_loss": 3.934518194111975, + "tokens_seen": 477364224 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043202607823470416, + "loss": 1.5144, + "theoretical_loss": 3.934459553745249, + "tokens_seen": 477429760 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004320160481444333, + "loss": 1.5301, + "theoretical_loss": 3.9344009236809354, + "tokens_seen": 477495296 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004320060180541625, + "loss": 1.5334, + "theoretical_loss": 3.9343423039158107, + "tokens_seen": 477560832 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043199598796389165, + "loss": 1.6141, + "theoretical_loss": 3.9342836944466533, + "tokens_seen": 477626368 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004319859578736209, + "loss": 1.3996, + "theoretical_loss": 3.9342250952702416, + "tokens_seen": 477691904 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043197592778335006, + "loss": 1.4746, + "theoretical_loss": 3.934166506383357, + "tokens_seen": 477757440 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043196589769307924, + "loss": 1.4725, + "theoretical_loss": 3.934107927782781, + "tokens_seen": 477822976 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004319558676028084, + "loss": 1.492, + "theoretical_loss": 3.934049359465297, + "tokens_seen": 477888512 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004319458375125376, + "loss": 1.4782, + "theoretical_loss": 3.933990801427691, + "tokens_seen": 477954048 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004319358074222668, + "loss": 1.5511, + "theoretical_loss": 3.9339322536667494, + "tokens_seen": 478019584 + }, + { + "epoch": 0.14, + "learning_rate": 0.000431925777331996, + "loss": 1.5119, + "theoretical_loss": 3.93387371617926, + "tokens_seen": 478085120 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043191574724172515, + "loss": 1.4463, + "theoretical_loss": 3.9338151889620114, + "tokens_seen": 478150656 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004319057171514544, + "loss": 1.5075, + "theoretical_loss": 3.933756672011796, + "tokens_seen": 478216192 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043189568706118357, + "loss": 1.4923, + "theoretical_loss": 3.933698165325404, + "tokens_seen": 478281728 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043188565697091275, + "loss": 1.4851, + "theoretical_loss": 3.933639668899631, + "tokens_seen": 478347264 + }, + { + "epoch": 0.14, + "objective/train/advantage_avg": -0.0015765945427119732, + "objective/train/docs_used": 281924, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.820423126220703, + "objective/train/original_loss": 2.820422887802124, + "objective/train/theoretical_loss": 3.933581182731271, + "objective/train/tokens_used": 498872800, + "objective/train/value_avg": -0.004337310791015625, + "objective/train/value_loss": 7.114662003004923e-05, + "objective/train/value_max": -0.0003712177276611328, + "objective/train/value_min": -0.047943115234375, + "objective/train/value_reward_corr": 0.27483779467416014, + "objective/train/value_std": 0.0031642913818359375, + "objective/train/weight_avg": 0.9874938130378723, + "objective/train/weighted_lm_loss": 2.7843008041381836, + "objective/train/weights_max": 1.6151553392410278, + "objective/train/weights_min": 0.1935657113790512, + "theoretical_loss": 3.933581182731271, + "tokens_seen": 478412800 + }, + { + "epoch": 0.14, + "learning_rate": 0.00043187562688064193, + "loss": 1.4223, + "theoretical_loss": 3.933581182731271, + "tokens_seen": 478412800 + }, + { + "epoch": 0.14, + "learning_rate": 0.0004318655967903711, + "loss": 1.5794, + "theoretical_loss": 3.9335227068171212, + "tokens_seen": 478478336 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004318555667001003, + "loss": 1.3819, + "theoretical_loss": 3.9334642411539793, + "tokens_seen": 478543872 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004318455366098295, + "loss": 1.445, + "theoretical_loss": 3.9334057857386453, + "tokens_seen": 478609408 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043183550651955865, + "loss": 1.3565, + "theoretical_loss": 3.933347340567919, + "tokens_seen": 478674944 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004318254764292879, + "loss": 1.4351, + "theoretical_loss": 3.9332889056386042, + "tokens_seen": 478740480 + }, + { + "epoch": 0.15, + "learning_rate": 0.000431815446339017, + "loss": 1.4069, + "theoretical_loss": 3.933230480947504, + "tokens_seen": 478806016 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043180541624874625, + "loss": 1.4452, + "theoretical_loss": 3.933172066491423, + "tokens_seen": 478871552 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043179538615847543, + "loss": 1.4947, + "theoretical_loss": 3.9331136622671687, + "tokens_seen": 478937088 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004317853560682046, + "loss": 1.4055, + "theoretical_loss": 3.9330552682715485, + "tokens_seen": 479002624 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004317753259779338, + "loss": 1.3739, + "theoretical_loss": 3.9329968845013727, + "tokens_seen": 479068160 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043176529588766303, + "loss": 1.479, + "theoretical_loss": 3.932938510953452, + "tokens_seen": 479133696 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043175526579739215, + "loss": 1.5259, + "theoretical_loss": 3.9328801476245987, + "tokens_seen": 479199232 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004317452357071214, + "loss": 1.3784, + "theoretical_loss": 3.932821794511626, + "tokens_seen": 479264768 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004317352056168505, + "loss": 1.4564, + "theoretical_loss": 3.9327634516113497, + "tokens_seen": 479330304 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043172517552657975, + "loss": 1.4885, + "theoretical_loss": 3.932705118920587, + "tokens_seen": 479395840 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043171514543630893, + "loss": 1.5449, + "theoretical_loss": 3.9326467964361553, + "tokens_seen": 479461376 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004317051153460381, + "loss": 1.556, + "theoretical_loss": 3.9325884841548735, + "tokens_seen": 479526912 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004316950852557673, + "loss": 1.3207, + "theoretical_loss": 3.932530182073564, + "tokens_seen": 479592448 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004316850551654965, + "loss": 1.4756, + "theoretical_loss": 3.9324718901890487, + "tokens_seen": 479657984 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004316750250752257, + "loss": 1.4608, + "theoretical_loss": 3.932413608498151, + "tokens_seen": 479723520 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004316649949849549, + "loss": 1.3946, + "theoretical_loss": 3.932355336997696, + "tokens_seen": 479789056 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004316549648946841, + "loss": 1.4279, + "theoretical_loss": 3.9322970756845113, + "tokens_seen": 479854592 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043164493480441326, + "loss": 1.5213, + "theoretical_loss": 3.9322388245554234, + "tokens_seen": 479920128 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043163490471414244, + "loss": 1.4491, + "theoretical_loss": 3.9321805836072627, + "tokens_seen": 479985664 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": -0.0002444189158268273, + "objective/train/docs_used": 283309, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4823861122131348, + "objective/train/original_loss": 3.482387065887451, + "objective/train/theoretical_loss": 3.9321223528368607, + "objective/train/tokens_used": 500511200, + "objective/train/value_avg": -0.00565338134765625, + "objective/train/value_loss": 6.986682274146006e-05, + "objective/train/value_max": -0.0003077983856201172, + "objective/train/value_min": -0.07977294921875, + "objective/train/value_reward_corr": 0.3710647658552607, + "objective/train/value_std": 0.0042724609375, + "objective/train/weight_avg": 1.0008474588394165, + "objective/train/weighted_lm_loss": 3.4826905727386475, + "objective/train/weights_max": 1.8935014009475708, + "objective/train/weights_min": 0.4172251522541046, + "theoretical_loss": 3.9321223528368607, + "tokens_seen": 480051200 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004316248746238716, + "loss": 1.5146, + "theoretical_loss": 3.9321223528368607, + "tokens_seen": 480051200 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043161484453360085, + "loss": 1.4841, + "theoretical_loss": 3.932064132241049, + "tokens_seen": 480116736 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043160481444333, + "loss": 1.5041, + "theoretical_loss": 3.932005921816661, + "tokens_seen": 480182272 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004315947843530592, + "loss": 1.473, + "theoretical_loss": 3.9319477215605323, + "tokens_seen": 480247808 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004315847542627884, + "loss": 1.4789, + "theoretical_loss": 3.9318895314694995, + "tokens_seen": 480313344 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004315747241725176, + "loss": 1.4776, + "theoretical_loss": 3.931831351540401, + "tokens_seen": 480378880 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043156469408224676, + "loss": 1.3509, + "theoretical_loss": 3.931773181770075, + "tokens_seen": 480444416 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043155466399197594, + "loss": 1.4109, + "theoretical_loss": 3.931715022155364, + "tokens_seen": 480509952 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004315446339017051, + "loss": 1.3833, + "theoretical_loss": 3.9316568726931083, + "tokens_seen": 480575488 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043153460381143436, + "loss": 1.4133, + "theoretical_loss": 3.931598733380153, + "tokens_seen": 480641024 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004315245737211635, + "loss": 1.5679, + "theoretical_loss": 3.9315406042133425, + "tokens_seen": 480706560 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004315145436308927, + "loss": 1.4321, + "theoretical_loss": 3.931482485189524, + "tokens_seen": 480772096 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043150451354062185, + "loss": 1.3575, + "theoretical_loss": 3.9314243763055443, + "tokens_seen": 480837632 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004314944834503511, + "loss": 1.4823, + "theoretical_loss": 3.9313662775582534, + "tokens_seen": 480903168 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043148445336008026, + "loss": 1.4383, + "theoretical_loss": 3.9313081889445014, + "tokens_seen": 480968704 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043147442326980944, + "loss": 1.492, + "theoretical_loss": 3.931250110461141, + "tokens_seen": 481034240 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004314643931795386, + "loss": 1.4915, + "theoretical_loss": 3.9311920421050255, + "tokens_seen": 481099776 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004314543630892678, + "loss": 1.3841, + "theoretical_loss": 3.9311339838730097, + "tokens_seen": 481165312 + }, + { + "epoch": 0.15, + "learning_rate": 0.000431444332998997, + "loss": 1.3734, + "theoretical_loss": 3.9310759357619505, + "tokens_seen": 481230848 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004314343029087262, + "loss": 1.4913, + "theoretical_loss": 3.9310178977687045, + "tokens_seen": 481296384 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043142427281845535, + "loss": 1.4259, + "theoretical_loss": 3.930959869890132, + "tokens_seen": 481361920 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004314142427281846, + "loss": 1.3893, + "theoretical_loss": 3.9309018521230925, + "tokens_seen": 481427456 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043140421263791377, + "loss": 1.4361, + "theoretical_loss": 3.930843844464449, + "tokens_seen": 481492992 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043139418254764295, + "loss": 1.4708, + "theoretical_loss": 3.9307858469110633, + "tokens_seen": 481558528 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043138415245737213, + "loss": 1.4038, + "theoretical_loss": 3.9307278594598016, + "tokens_seen": 481624064 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 0.0012029382633045316, + "objective/train/docs_used": 283520, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.156329393386841, + "objective/train/original_loss": 3.15632963180542, + "objective/train/theoretical_loss": 3.930669882107529, + "objective/train/tokens_used": 502149600, + "objective/train/value_avg": -0.005512237548828125, + "objective/train/value_loss": 3.993565042037517e-05, + "objective/train/value_max": -0.0005359649658203125, + "objective/train/value_min": -0.0809326171875, + "objective/train/value_reward_corr": 0.522637415503463, + "objective/train/value_std": 0.005580902099609375, + "objective/train/weight_avg": 1.0139966011047363, + "objective/train/weighted_lm_loss": 3.2002573013305664, + "objective/train/weights_max": 1.6926867961883545, + "objective/train/weights_min": 0.4094051122665405, + "theoretical_loss": 3.930669882107529, + "tokens_seen": 481689600 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004313741223671013, + "loss": 1.5249, + "theoretical_loss": 3.930669882107529, + "tokens_seen": 481689600 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004313640922768305, + "loss": 1.395, + "theoretical_loss": 3.9306119148511147, + "tokens_seen": 481755136 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004313540621865597, + "loss": 1.4676, + "theoretical_loss": 3.930553957687426, + "tokens_seen": 481820672 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043134403209628885, + "loss": 1.5022, + "theoretical_loss": 3.930496010613333, + "tokens_seen": 481886208 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004313340020060181, + "loss": 1.5151, + "theoretical_loss": 3.9304380736257087, + "tokens_seen": 481951744 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004313239719157472, + "loss": 1.5067, + "theoretical_loss": 3.9303801467214248, + "tokens_seen": 482017280 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043131394182547645, + "loss": 1.4816, + "theoretical_loss": 3.9303222298973575, + "tokens_seen": 482082816 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043130391173520563, + "loss": 1.5535, + "theoretical_loss": 3.930264323150381, + "tokens_seen": 482148352 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004312938816449348, + "loss": 1.563, + "theoretical_loss": 3.9302064264773735, + "tokens_seen": 482213888 + }, + { + "epoch": 0.15, + "learning_rate": 0.000431283851554664, + "loss": 1.6021, + "theoretical_loss": 3.930148539875214, + "tokens_seen": 482279424 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043127382146439323, + "loss": 1.5605, + "theoretical_loss": 3.930090663340782, + "tokens_seen": 482344960 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043126379137412236, + "loss": 1.41, + "theoretical_loss": 3.9300327968709587, + "tokens_seen": 482410496 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004312537612838516, + "loss": 1.4427, + "theoretical_loss": 3.9299749404626274, + "tokens_seen": 482476032 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004312437311935807, + "loss": 1.532, + "theoretical_loss": 3.929917094112673, + "tokens_seen": 482541568 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043123370110330995, + "loss": 1.3307, + "theoretical_loss": 3.92985925781798, + "tokens_seen": 482607104 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043122367101303913, + "loss": 1.2734, + "theoretical_loss": 3.9298014315754357, + "tokens_seen": 482672640 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004312136409227683, + "loss": 1.4509, + "theoretical_loss": 3.9297436153819287, + "tokens_seen": 482738176 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004312036108324975, + "loss": 1.5404, + "theoretical_loss": 3.929685809234349, + "tokens_seen": 482803712 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004311935807422267, + "loss": 1.4415, + "theoretical_loss": 3.929628013129587, + "tokens_seen": 482869248 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043118355065195586, + "loss": 1.532, + "theoretical_loss": 3.929570227064536, + "tokens_seen": 482934784 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004311735205616851, + "loss": 1.5153, + "theoretical_loss": 3.9295124510360893, + "tokens_seen": 483000320 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004311634904714142, + "loss": 1.3432, + "theoretical_loss": 3.929454685041143, + "tokens_seen": 483065856 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043115346038114346, + "loss": 1.4387, + "theoretical_loss": 3.929396929076594, + "tokens_seen": 483131392 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004311434302908726, + "loss": 1.4367, + "theoretical_loss": 3.929339183139339, + "tokens_seen": 483196928 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004311334002006018, + "loss": 1.5399, + "theoretical_loss": 3.9292814472262787, + "tokens_seen": 483262464 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 0.0002975336101371795, + "objective/train/docs_used": 284734, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.89035964012146, + "objective/train/original_loss": 2.890359401702881, + "objective/train/theoretical_loss": 3.9292237213343135, + "objective/train/tokens_used": 503788000, + "objective/train/value_avg": -0.005706787109375, + "objective/train/value_loss": 4.512359373620711e-05, + "objective/train/value_max": -0.000415802001953125, + "objective/train/value_min": -0.06304931640625, + "objective/train/value_reward_corr": 0.43862905920434736, + "objective/train/value_std": 0.0043792724609375, + "objective/train/weight_avg": 1.0051417350769043, + "objective/train/weighted_lm_loss": 2.910647392272949, + "objective/train/weights_max": 1.7999674081802368, + "objective/train/weights_min": 0.5102092623710632, + "theoretical_loss": 3.9292237213343135, + "tokens_seen": 483328000 + }, + { + "epoch": 0.15, + "learning_rate": 0.000431123370110331, + "loss": 1.5112, + "theoretical_loss": 3.9292237213343135, + "tokens_seen": 483328000 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004311133400200602, + "loss": 1.6183, + "theoretical_loss": 3.9291660054603454, + "tokens_seen": 483393536 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043110330992978936, + "loss": 1.4108, + "theoretical_loss": 3.9291082996012783, + "tokens_seen": 483459072 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004310932798395186, + "loss": 1.4853, + "theoretical_loss": 3.9290506037540167, + "tokens_seen": 483524608 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004310832497492477, + "loss": 1.4647, + "theoretical_loss": 3.928992917915468, + "tokens_seen": 483590144 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043107321965897696, + "loss": 1.5225, + "theoretical_loss": 3.928935242082539, + "tokens_seen": 483655680 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004310631895687061, + "loss": 1.4366, + "theoretical_loss": 3.9288775762521393, + "tokens_seen": 483721216 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004310531594784353, + "loss": 1.4229, + "theoretical_loss": 3.928819920421179, + "tokens_seen": 483786752 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004310431293881645, + "loss": 1.437, + "theoretical_loss": 3.9287622745865702, + "tokens_seen": 483852288 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004310330992978937, + "loss": 1.5426, + "theoretical_loss": 3.9287046387452262, + "tokens_seen": 483917824 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043102306920762286, + "loss": 1.5298, + "theoretical_loss": 3.928647012894061, + "tokens_seen": 483983360 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043101303911735205, + "loss": 1.3822, + "theoretical_loss": 3.9285893970299917, + "tokens_seen": 484048896 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004310030090270812, + "loss": 1.5425, + "theoretical_loss": 3.928531791149934, + "tokens_seen": 484114432 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043099297893681046, + "loss": 1.4729, + "theoretical_loss": 3.928474195250808, + "tokens_seen": 484179968 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004309829488465396, + "loss": 1.5183, + "theoretical_loss": 3.928416609329533, + "tokens_seen": 484245504 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004309729187562688, + "loss": 1.5618, + "theoretical_loss": 3.928359033383031, + "tokens_seen": 484311040 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043096288866599795, + "loss": 1.569, + "theoretical_loss": 3.9283014674082244, + "tokens_seen": 484376576 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004309528585757272, + "loss": 1.459, + "theoretical_loss": 3.9282439114020375, + "tokens_seen": 484442112 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043094282848545637, + "loss": 1.4936, + "theoretical_loss": 3.9281863653613964, + "tokens_seen": 484507648 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043093279839518555, + "loss": 1.5846, + "theoretical_loss": 3.928128829283226, + "tokens_seen": 484573184 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004309227683049148, + "loss": 1.6011, + "theoretical_loss": 3.928071303164457, + "tokens_seen": 484638720 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043091273821464397, + "loss": 1.3724, + "theoretical_loss": 3.928013787002018, + "tokens_seen": 484704256 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043090270812437315, + "loss": 1.4509, + "theoretical_loss": 3.9279562807928396, + "tokens_seen": 484769792 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043089267803410233, + "loss": 1.4651, + "theoretical_loss": 3.9278987845338547, + "tokens_seen": 484835328 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004308826479438315, + "loss": 1.4211, + "theoretical_loss": 3.9278412982219963, + "tokens_seen": 484900864 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 0.00023421429796144366, + "objective/train/docs_used": 285372, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.181069850921631, + "objective/train/original_loss": 3.181070327758789, + "objective/train/theoretical_loss": 3.927783821854201, + "objective/train/tokens_used": 505426400, + "objective/train/value_avg": -0.004810333251953125, + "objective/train/value_loss": 6.281316018430516e-05, + "objective/train/value_max": -0.0005035400390625, + "objective/train/value_min": -0.07794189453125, + "objective/train/value_reward_corr": 0.362226251095991, + "objective/train/value_std": 0.0038623809814453125, + "objective/train/weight_avg": 1.0050911903381348, + "objective/train/weighted_lm_loss": 3.202026605606079, + "objective/train/weights_max": 1.5254297256469727, + "objective/train/weights_min": 0.19822101294994354, + "theoretical_loss": 3.927783821854201, + "tokens_seen": 484966400 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004308726178535607, + "loss": 1.4974, + "theoretical_loss": 3.927783821854201, + "tokens_seen": 484966400 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004308625877632899, + "loss": 1.5172, + "theoretical_loss": 3.9277263554274033, + "tokens_seen": 485031936 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043085255767301905, + "loss": 1.4179, + "theoretical_loss": 3.927668898938542, + "tokens_seen": 485097472 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004308425275827483, + "loss": 1.5291, + "theoretical_loss": 3.9276114523845562, + "tokens_seen": 485163008 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004308324974924774, + "loss": 1.38, + "theoretical_loss": 3.9275540157623867, + "tokens_seen": 485228544 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043082246740220665, + "loss": 1.5232, + "theoretical_loss": 3.927496589068974, + "tokens_seen": 485294080 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043081243731193583, + "loss": 1.4731, + "theoretical_loss": 3.9274391723012627, + "tokens_seen": 485359616 + }, + { + "epoch": 0.15, + "learning_rate": 0.000430802407221665, + "loss": 1.4652, + "theoretical_loss": 3.9273817654561967, + "tokens_seen": 485425152 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004307923771313942, + "loss": 1.5228, + "theoretical_loss": 3.927324368530723, + "tokens_seen": 485490688 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043078234704112343, + "loss": 1.3745, + "theoretical_loss": 3.927266981521787, + "tokens_seen": 485556224 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043077231695085256, + "loss": 1.4767, + "theoretical_loss": 3.927209604426338, + "tokens_seen": 485621760 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004307622868605818, + "loss": 1.4598, + "theoretical_loss": 3.9271522372413266, + "tokens_seen": 485687296 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004307522567703109, + "loss": 1.5647, + "theoretical_loss": 3.927094879963704, + "tokens_seen": 485752832 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043074222668004015, + "loss": 1.5719, + "theoretical_loss": 3.927037532590422, + "tokens_seen": 485818368 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043073219658976933, + "loss": 1.5147, + "theoretical_loss": 3.9269801951184355, + "tokens_seen": 485883904 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004307221664994985, + "loss": 1.5642, + "theoretical_loss": 3.9269228675446994, + "tokens_seen": 485949440 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004307121364092277, + "loss": 1.4486, + "theoretical_loss": 3.9268655498661706, + "tokens_seen": 486014976 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004307021063189569, + "loss": 1.4304, + "theoretical_loss": 3.926808242079807, + "tokens_seen": 486080512 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043069207622868606, + "loss": 1.5344, + "theoretical_loss": 3.9267509441825688, + "tokens_seen": 486146048 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004306820461384153, + "loss": 1.4482, + "theoretical_loss": 3.926693656171415, + "tokens_seen": 486211584 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004306720160481444, + "loss": 1.5585, + "theoretical_loss": 3.9266363780433093, + "tokens_seen": 486277120 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043066198595787366, + "loss": 1.4471, + "theoretical_loss": 3.926579109795214, + "tokens_seen": 486342656 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004306519558676028, + "loss": 1.4395, + "theoretical_loss": 3.926521851424095, + "tokens_seen": 486408192 + }, + { + "epoch": 0.15, + "learning_rate": 0.000430641925777332, + "loss": 1.46, + "theoretical_loss": 3.9264646029269175, + "tokens_seen": 486473728 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004306318956870612, + "loss": 1.384, + "theoretical_loss": 3.926407364300649, + "tokens_seen": 486539264 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": -0.0018214911688119173, + "objective/train/docs_used": 285999, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.140460729598999, + "objective/train/original_loss": 3.140460729598999, + "objective/train/theoretical_loss": 3.9263501355422585, + "objective/train/tokens_used": 507064800, + "objective/train/value_avg": -0.005443572998046875, + "objective/train/value_loss": 8.312644058605656e-05, + "objective/train/value_max": -0.0004494190216064453, + "objective/train/value_min": -0.04742431640625, + "objective/train/value_reward_corr": 0.37293149804824466, + "objective/train/value_std": 0.003665924072265625, + "objective/train/weight_avg": 0.9857029914855957, + "objective/train/weighted_lm_loss": 3.111490249633789, + "objective/train/weights_max": 1.606797695159912, + "objective/train/weights_min": 0.4140225052833557, + "theoretical_loss": 3.9263501355422585, + "tokens_seen": 486604800 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004306218655967904, + "loss": 1.5554, + "theoretical_loss": 3.9263501355422585, + "tokens_seen": 486604800 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043061183550651956, + "loss": 1.4407, + "theoretical_loss": 3.926292916648716, + "tokens_seen": 486670336 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004306018054162488, + "loss": 1.4284, + "theoretical_loss": 3.926235707616994, + "tokens_seen": 486735872 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004305917753259779, + "loss": 1.4357, + "theoretical_loss": 3.926178508444064, + "tokens_seen": 486801408 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043058174523570716, + "loss": 1.4683, + "theoretical_loss": 3.9261213191269, + "tokens_seen": 486866944 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004305717151454363, + "loss": 1.456, + "theoretical_loss": 3.9260641396624787, + "tokens_seen": 486932480 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004305616850551655, + "loss": 1.5326, + "theoretical_loss": 3.9260069700477755, + "tokens_seen": 486998016 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004305516549648947, + "loss": 1.4677, + "theoretical_loss": 3.9259498102797696, + "tokens_seen": 487063552 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004305416248746239, + "loss": 1.4715, + "theoretical_loss": 3.9258926603554407, + "tokens_seen": 487129088 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043053159478435306, + "loss": 1.3723, + "theoretical_loss": 3.925835520271768, + "tokens_seen": 487194624 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043052156469408225, + "loss": 1.4046, + "theoretical_loss": 3.9257783900257355, + "tokens_seen": 487260160 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004305115346038114, + "loss": 1.442, + "theoretical_loss": 3.9257212696143258, + "tokens_seen": 487325696 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043050150451354066, + "loss": 1.4494, + "theoretical_loss": 3.9256641590345236, + "tokens_seen": 487391232 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004304914744232698, + "loss": 1.5256, + "theoretical_loss": 3.9256070582833145, + "tokens_seen": 487456768 + }, + { + "epoch": 0.15, + "learning_rate": 0.000430481444332999, + "loss": 1.5082, + "theoretical_loss": 3.9255499673576875, + "tokens_seen": 487522304 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043047141424272815, + "loss": 1.4586, + "theoretical_loss": 3.9254928862546303, + "tokens_seen": 487587840 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004304613841524574, + "loss": 1.4611, + "theoretical_loss": 3.9254358149711335, + "tokens_seen": 487653376 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043045135406218657, + "loss": 1.5979, + "theoretical_loss": 3.925378753504188, + "tokens_seen": 487718912 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043044132397191575, + "loss": 1.382, + "theoretical_loss": 3.925321701850787, + "tokens_seen": 487784448 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043043129388164493, + "loss": 1.5091, + "theoretical_loss": 3.925264660007924, + "tokens_seen": 487849984 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043042126379137417, + "loss": 1.5477, + "theoretical_loss": 3.925207627972596, + "tokens_seen": 487915520 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004304112337011033, + "loss": 1.4623, + "theoretical_loss": 3.9251506057417975, + "tokens_seen": 487981056 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043040120361083253, + "loss": 1.5287, + "theoretical_loss": 3.9250935933125284, + "tokens_seen": 488046592 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043039117352056165, + "loss": 1.5519, + "theoretical_loss": 3.925036590681787, + "tokens_seen": 488112128 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004303811434302909, + "loss": 1.3761, + "theoretical_loss": 3.9249795978465745, + "tokens_seen": 488177664 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 0.0001581201795488596, + "objective/train/docs_used": 287428, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.99515700340271, + "objective/train/original_loss": 2.995157241821289, + "objective/train/theoretical_loss": 3.9249226148038927, + "objective/train/tokens_used": 508703200, + "objective/train/value_avg": -0.00720977783203125, + "objective/train/value_loss": 0.00010957424092339352, + "objective/train/value_max": -0.0004582405090332031, + "objective/train/value_min": -0.126953125, + "objective/train/value_reward_corr": 0.48426141294180153, + "objective/train/value_std": 0.006702423095703125, + "objective/train/weight_avg": 1.0065504312515259, + "objective/train/weighted_lm_loss": 3.017070770263672, + "objective/train/weights_max": 2.995518922805786, + "objective/train/weights_min": 0.3163055181503296, + "theoretical_loss": 3.9249226148038927, + "tokens_seen": 488243200 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043037111334002007, + "loss": 1.501, + "theoretical_loss": 3.9249226148038927, + "tokens_seen": 488243200 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043036108324974925, + "loss": 1.467, + "theoretical_loss": 3.924865641550745, + "tokens_seen": 488308736 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043035105315947843, + "loss": 1.5546, + "theoretical_loss": 3.9248086780841365, + "tokens_seen": 488374272 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004303410230692076, + "loss": 1.4838, + "theoretical_loss": 3.924751724401073, + "tokens_seen": 488439808 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004303309929789368, + "loss": 1.4864, + "theoretical_loss": 3.9246947804985606, + "tokens_seen": 488505344 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043032096288866603, + "loss": 1.4294, + "theoretical_loss": 3.9246378463736096, + "tokens_seen": 488570880 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043031093279839516, + "loss": 1.3692, + "theoretical_loss": 3.9245809220232295, + "tokens_seen": 488636416 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004303009027081244, + "loss": 1.4307, + "theoretical_loss": 3.9245240074444308, + "tokens_seen": 488701952 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004302908726178535, + "loss": 1.5364, + "theoretical_loss": 3.9244671026342273, + "tokens_seen": 488767488 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043028084252758276, + "loss": 1.5326, + "theoretical_loss": 3.9244102075896317, + "tokens_seen": 488833024 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043027081243731194, + "loss": 1.5326, + "theoretical_loss": 3.924353322307659, + "tokens_seen": 488898560 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004302607823470411, + "loss": 1.4552, + "theoretical_loss": 3.924296446785328, + "tokens_seen": 488964096 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004302507522567703, + "loss": 1.5476, + "theoretical_loss": 3.9242395810196538, + "tokens_seen": 489029632 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043024072216649953, + "loss": 1.3518, + "theoretical_loss": 3.9241827250076566, + "tokens_seen": 489095168 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043023069207622866, + "loss": 1.407, + "theoretical_loss": 3.9241258787463567, + "tokens_seen": 489160704 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004302206619859579, + "loss": 1.3685, + "theoretical_loss": 3.9240690422327766, + "tokens_seen": 489226240 + }, + { + "epoch": 0.15, + "learning_rate": 0.000430210631895687, + "loss": 1.4016, + "theoretical_loss": 3.9240122154639385, + "tokens_seen": 489291776 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043020060180541626, + "loss": 1.4764, + "theoretical_loss": 3.923955398436867, + "tokens_seen": 489357312 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043019057171514544, + "loss": 1.5005, + "theoretical_loss": 3.9238985911485873, + "tokens_seen": 489422848 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004301805416248746, + "loss": 1.3251, + "theoretical_loss": 3.923841793596128, + "tokens_seen": 489488384 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043017051153460386, + "loss": 1.4362, + "theoretical_loss": 3.923785005776515, + "tokens_seen": 489553920 + }, + { + "epoch": 0.15, + "learning_rate": 0.000430160481444333, + "loss": 1.442, + "theoretical_loss": 3.92372822768678, + "tokens_seen": 489619456 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004301504513540622, + "loss": 1.4602, + "theoretical_loss": 3.923671459323953, + "tokens_seen": 489684992 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004301404212637914, + "loss": 1.4985, + "theoretical_loss": 3.923614700685066, + "tokens_seen": 489750528 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004301303911735206, + "loss": 1.528, + "theoretical_loss": 3.9235579517671524, + "tokens_seen": 489816064 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 0.001296263886615634, + "objective/train/docs_used": 288116, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.95450496673584, + "objective/train/original_loss": 2.954505205154419, + "objective/train/theoretical_loss": 3.9235012125672473, + "objective/train/tokens_used": 510341600, + "objective/train/value_avg": -0.004627227783203125, + "objective/train/value_loss": 4.548433207673952e-05, + "objective/train/value_max": -0.00036406517028808594, + "objective/train/value_min": -0.060089111328125, + "objective/train/value_reward_corr": 0.35150661730668714, + "objective/train/value_std": 0.003459930419921875, + "objective/train/weight_avg": 1.0151052474975586, + "objective/train/weighted_lm_loss": 3.00226092338562, + "objective/train/weights_max": 1.7213134765625, + "objective/train/weights_min": 0.3875870108604431, + "theoretical_loss": 3.9235012125672473, + "tokens_seen": 489881600 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043012036108324976, + "loss": 1.5211, + "theoretical_loss": 3.9235012125672473, + "tokens_seen": 489881600 + }, + { + "epoch": 0.15, + "learning_rate": 0.000430110330992979, + "loss": 1.4331, + "theoretical_loss": 3.9234444830823874, + "tokens_seen": 489947136 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004301003009027081, + "loss": 1.4512, + "theoretical_loss": 3.9233877633096093, + "tokens_seen": 490012672 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043009027081243736, + "loss": 1.5119, + "theoretical_loss": 3.9233310532459518, + "tokens_seen": 490078208 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004300802407221665, + "loss": 1.4044, + "theoretical_loss": 3.9232743528884546, + "tokens_seen": 490143744 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004300702106318957, + "loss": 1.5545, + "theoretical_loss": 3.92321766223416, + "tokens_seen": 490209280 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004300601805416249, + "loss": 1.456, + "theoretical_loss": 3.92316098128011, + "tokens_seen": 490274816 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004300501504513541, + "loss": 1.4769, + "theoretical_loss": 3.923104310023348, + "tokens_seen": 490340352 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043004012036108326, + "loss": 1.5822, + "theoretical_loss": 3.92304764846092, + "tokens_seen": 490405888 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043003009027081245, + "loss": 1.4882, + "theoretical_loss": 3.9229909965898715, + "tokens_seen": 490471424 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043002006018054163, + "loss": 1.4573, + "theoretical_loss": 3.9229343544072517, + "tokens_seen": 490536960 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043001003009027086, + "loss": 1.3944, + "theoretical_loss": 3.9228777219101088, + "tokens_seen": 490602496 + }, + { + "epoch": 0.15, + "learning_rate": 0.00043, + "loss": 1.4775, + "theoretical_loss": 3.9228210990954935, + "tokens_seen": 490668032 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004299899699097292, + "loss": 1.5051, + "theoretical_loss": 3.9227644859604562, + "tokens_seen": 490733568 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042997993981945835, + "loss": 1.4585, + "theoretical_loss": 3.922707882502052, + "tokens_seen": 490799104 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004299699097291876, + "loss": 1.5223, + "theoretical_loss": 3.922651288717333, + "tokens_seen": 490864640 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042995987963891677, + "loss": 1.4265, + "theoretical_loss": 3.9225947046033562, + "tokens_seen": 490930176 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042994984954864595, + "loss": 1.4759, + "theoretical_loss": 3.922538130157178, + "tokens_seen": 490995712 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042993981945837513, + "loss": 1.4909, + "theoretical_loss": 3.922481565375856, + "tokens_seen": 491061248 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042992978936810437, + "loss": 1.4256, + "theoretical_loss": 3.9224250102564504, + "tokens_seen": 491126784 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004299197592778335, + "loss": 1.5093, + "theoretical_loss": 3.922368464796021, + "tokens_seen": 491192320 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042990972918756273, + "loss": 1.5209, + "theoretical_loss": 3.922311928991631, + "tokens_seen": 491257856 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042989969909729185, + "loss": 1.4646, + "theoretical_loss": 3.9222554028403422, + "tokens_seen": 491323392 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004298896690070211, + "loss": 1.4685, + "theoretical_loss": 3.9221988863392205, + "tokens_seen": 491388928 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042987963891675027, + "loss": 1.5464, + "theoretical_loss": 3.9221423794853303, + "tokens_seen": 491454464 + }, + { + "debugging/Self-BLEU-5": 0.31008426371372083, + "debugging/distinct-1-grams": 0.7945037853782116, + "debugging/distinct-2-grams": 0.9759782108419659, + "debugging/entropy-1-grams": 5.529295001026804, + "debugging/entropy-2-grams": 6.1656973428761805, + "debugging/length": 452.7142857142857, + "debugging/num_segments": 7, + "debugging/raw_token_scores_avg": 0.007705323863774538, + "debugging/raw_token_scores_std": 0.011359859257936478, + "debugging/score": 0.007122610434625429, + "debugging/score_std": 0.004618225724562532, + "epoch": 0.15, + "objective/train/advantage_avg": -0.0012057783314958215, + "objective/train/docs_used": 289197, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.5755741596221924, + "objective/train/original_loss": 3.5755741596221924, + "objective/train/theoretical_loss": 3.9220858822757396, + "objective/train/tokens_used": 511980000, + "objective/train/value_avg": -0.006500244140625, + "objective/train/value_loss": 9.409981430508196e-05, + "objective/train/value_max": -0.0006461143493652344, + "objective/train/value_min": -0.05780029296875, + "objective/train/value_reward_corr": 0.5336071420791404, + "objective/train/value_std": 0.005496978759765625, + "objective/train/weight_avg": 0.9922725558280945, + "objective/train/weighted_lm_loss": 3.5509839057922363, + "objective/train/weights_max": 1.513967752456665, + "objective/train/weights_min": 0.34292078018188477, + "theoretical_loss": 3.9220858822757396, + "tokens_seen": 491520000 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042986960882647945, + "loss": 1.4664, + "theoretical_loss": 3.9220858822757396, + "tokens_seen": 491520000 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042985957873620863, + "loss": 1.5608, + "theoretical_loss": 3.9220293947075167, + "tokens_seen": 491585536 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004298495486459378, + "loss": 1.435, + "theoretical_loss": 3.9219729167777313, + "tokens_seen": 491651072 + }, + { + "epoch": 0.15, + "learning_rate": 0.000429839518555667, + "loss": 1.4766, + "theoretical_loss": 3.9219164484834543, + "tokens_seen": 491716608 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042982948846539623, + "loss": 1.4568, + "theoretical_loss": 3.9218599898217583, + "tokens_seen": 491782144 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042981945837512536, + "loss": 1.5022, + "theoretical_loss": 3.921803540789716, + "tokens_seen": 491847680 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004298094282848546, + "loss": 1.573, + "theoretical_loss": 3.921747101384402, + "tokens_seen": 491913216 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004297993981945837, + "loss": 1.4994, + "theoretical_loss": 3.9216906716028936, + "tokens_seen": 491978752 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042978936810431296, + "loss": 1.4797, + "theoretical_loss": 3.921634251442267, + "tokens_seen": 492044288 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042977933801404214, + "loss": 1.4714, + "theoretical_loss": 3.9215778408996016, + "tokens_seen": 492109824 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004297693079237713, + "loss": 1.4867, + "theoretical_loss": 3.9215214399719773, + "tokens_seen": 492175360 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004297592778335005, + "loss": 1.451, + "theoretical_loss": 3.9214650486564744, + "tokens_seen": 492240896 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042974924774322973, + "loss": 1.5349, + "theoretical_loss": 3.9214086669501755, + "tokens_seen": 492306432 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042973921765295886, + "loss": 1.5053, + "theoretical_loss": 3.921352294850166, + "tokens_seen": 492371968 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004297291875626881, + "loss": 1.4326, + "theoretical_loss": 3.9212959323535284, + "tokens_seen": 492437504 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004297191574724172, + "loss": 1.5347, + "theoretical_loss": 3.9212395794573505, + "tokens_seen": 492503040 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042970912738214646, + "loss": 1.553, + "theoretical_loss": 3.9211832361587193, + "tokens_seen": 492568576 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042969909729187564, + "loss": 1.5462, + "theoretical_loss": 3.9211269024547244, + "tokens_seen": 492634112 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004296890672016048, + "loss": 1.5314, + "theoretical_loss": 3.921070578342455, + "tokens_seen": 492699648 + }, + { + "epoch": 0.15, + "learning_rate": 0.000429679037111334, + "loss": 1.5694, + "theoretical_loss": 3.9210142638190026, + "tokens_seen": 492765184 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004296690070210632, + "loss": 1.5584, + "theoretical_loss": 3.92095795888146, + "tokens_seen": 492830720 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042965897693079236, + "loss": 1.4477, + "theoretical_loss": 3.9209016635269207, + "tokens_seen": 492896256 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004296489468405216, + "loss": 1.5948, + "theoretical_loss": 3.9208453777524808, + "tokens_seen": 492961792 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004296389167502507, + "loss": 1.5985, + "theoretical_loss": 3.9207891015552354, + "tokens_seen": 493027328 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042962888665997996, + "loss": 1.4836, + "theoretical_loss": 3.920732834932283, + "tokens_seen": 493092864 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 0.0003880932927131653, + "objective/train/docs_used": 289918, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2261996269226074, + "objective/train/original_loss": 3.2261993885040283, + "objective/train/theoretical_loss": 3.9206765778807227, + "objective/train/tokens_used": 513618400, + "objective/train/value_avg": -0.004497528076171875, + "objective/train/value_loss": 4.428214015206322e-05, + "objective/train/value_max": -0.0003859996795654297, + "objective/train/value_min": -0.0323486328125, + "objective/train/value_reward_corr": 0.26451952735590095, + "objective/train/value_std": 0.0029621124267578125, + "objective/train/weight_avg": 1.0060094594955444, + "objective/train/weighted_lm_loss": 3.251871109008789, + "objective/train/weights_max": 1.3819372653961182, + "objective/train/weights_min": 0.5402818918228149, + "theoretical_loss": 3.9206765778807227, + "tokens_seen": 493158400 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004296188565697091, + "loss": 1.4291, + "theoretical_loss": 3.9206765778807227, + "tokens_seen": 493158400 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004296088264794383, + "loss": 1.3944, + "theoretical_loss": 3.920620330397654, + "tokens_seen": 493223936 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004295987963891675, + "loss": 1.4698, + "theoretical_loss": 3.920564092480179, + "tokens_seen": 493289472 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004295887662988967, + "loss": 1.5211, + "theoretical_loss": 3.9205078641254003, + "tokens_seen": 493355008 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042957873620862587, + "loss": 1.4913, + "theoretical_loss": 3.920451645330422, + "tokens_seen": 493420544 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004295687061183551, + "loss": 1.4791, + "theoretical_loss": 3.920395436092349, + "tokens_seen": 493486080 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042955867602808423, + "loss": 1.6375, + "theoretical_loss": 3.920339236408288, + "tokens_seen": 493551616 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042954864593781347, + "loss": 1.5906, + "theoretical_loss": 3.920283046275346, + "tokens_seen": 493617152 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004295386158475426, + "loss": 1.643, + "theoretical_loss": 3.920226865690634, + "tokens_seen": 493682688 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042952858575727183, + "loss": 1.4805, + "theoretical_loss": 3.9201706946512607, + "tokens_seen": 493748224 + }, + { + "epoch": 0.15, + "learning_rate": 0.000429518555667001, + "loss": 1.7492, + "theoretical_loss": 3.920114533154338, + "tokens_seen": 493813760 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004295085255767302, + "loss": 1.6469, + "theoretical_loss": 3.9200583811969785, + "tokens_seen": 493879296 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042949849548645937, + "loss": 1.7293, + "theoretical_loss": 3.9200022387762967, + "tokens_seen": 493944832 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042948846539618855, + "loss": 1.5267, + "theoretical_loss": 3.919946105889408, + "tokens_seen": 494010368 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042947843530591773, + "loss": 1.6127, + "theoretical_loss": 3.919889982533429, + "tokens_seen": 494075904 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042946840521564697, + "loss": 1.6545, + "theoretical_loss": 3.919833868705477, + "tokens_seen": 494141440 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004294583751253761, + "loss": 1.4679, + "theoretical_loss": 3.9197777644026717, + "tokens_seen": 494206976 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042944834503510533, + "loss": 1.6872, + "theoretical_loss": 3.9197216696221324, + "tokens_seen": 494272512 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004294383149448345, + "loss": 1.6043, + "theoretical_loss": 3.919665584360982, + "tokens_seen": 494338048 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004294282848545637, + "loss": 1.6418, + "theoretical_loss": 3.9196095086163427, + "tokens_seen": 494403584 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042941825476429293, + "loss": 1.5004, + "theoretical_loss": 3.9195534423853386, + "tokens_seen": 494469120 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042940822467402206, + "loss": 1.546, + "theoretical_loss": 3.919497385665095, + "tokens_seen": 494534656 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004293981945837513, + "loss": 1.5009, + "theoretical_loss": 3.9194413384527387, + "tokens_seen": 494600192 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042938816449348047, + "loss": 1.4791, + "theoretical_loss": 3.9193853007453976, + "tokens_seen": 494665728 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042937813440320965, + "loss": 1.6021, + "theoretical_loss": 3.9193292725402014, + "tokens_seen": 494731264 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 5.0049264245899394e-05, + "objective/train/docs_used": 291263, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3126220703125, + "objective/train/original_loss": 3.312622547149658, + "objective/train/theoretical_loss": 3.9192732538342785, + "objective/train/tokens_used": 515256800, + "objective/train/value_avg": -0.004638671875, + "objective/train/value_loss": 3.8277074054349214e-05, + "objective/train/value_max": -0.000396728515625, + "objective/train/value_min": -0.059967041015625, + "objective/train/value_reward_corr": 0.5074417094809964, + "objective/train/value_std": 0.003787994384765625, + "objective/train/weight_avg": 1.0023363828659058, + "objective/train/weighted_lm_loss": 3.3171286582946777, + "objective/train/weights_max": 1.8215184211730957, + "objective/train/weights_min": 0.5095478892326355, + "theoretical_loss": 3.9192732538342785, + "tokens_seen": 494796800 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042936810431293883, + "loss": 1.538, + "theoretical_loss": 3.9192732538342785, + "tokens_seen": 494796800 + }, + { + "epoch": 0.15, + "learning_rate": 0.000429358074222668, + "loss": 1.6198, + "theoretical_loss": 3.919217244624762, + "tokens_seen": 494862336 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004293480441323972, + "loss": 1.4987, + "theoretical_loss": 3.919161244908785, + "tokens_seen": 494927872 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042933801404212643, + "loss": 1.5269, + "theoretical_loss": 3.9191052546834806, + "tokens_seen": 494993408 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042932798395185556, + "loss": 1.4086, + "theoretical_loss": 3.9190492739459843, + "tokens_seen": 495058944 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004293179538615848, + "loss": 1.5777, + "theoretical_loss": 3.9189933026934334, + "tokens_seen": 495124480 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004293079237713139, + "loss": 1.5827, + "theoretical_loss": 3.918937340922965, + "tokens_seen": 495190016 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042929789368104316, + "loss": 1.5757, + "theoretical_loss": 3.9188813886317186, + "tokens_seen": 495255552 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042928786359077234, + "loss": 1.6098, + "theoretical_loss": 3.918825445816834, + "tokens_seen": 495321088 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004292778335005015, + "loss": 1.4468, + "theoretical_loss": 3.918769512475453, + "tokens_seen": 495386624 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004292678034102307, + "loss": 1.5742, + "theoretical_loss": 3.918713588604718, + "tokens_seen": 495452160 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042925777331995993, + "loss": 1.6228, + "theoretical_loss": 3.918657674201774, + "tokens_seen": 495517696 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042924774322968906, + "loss": 1.5428, + "theoretical_loss": 3.9186017692637654, + "tokens_seen": 495583232 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004292377131394183, + "loss": 1.5739, + "theoretical_loss": 3.918545873787839, + "tokens_seen": 495648768 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004292276830491474, + "loss": 1.4344, + "theoretical_loss": 3.9184899877711423, + "tokens_seen": 495714304 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042921765295887666, + "loss": 1.5071, + "theoretical_loss": 3.918434111210825, + "tokens_seen": 495779840 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042920762286860584, + "loss": 1.5213, + "theoretical_loss": 3.918378244104036, + "tokens_seen": 495845376 + }, + { + "epoch": 0.15, + "learning_rate": 0.000429197592778335, + "loss": 1.4688, + "theoretical_loss": 3.9183223864479277, + "tokens_seen": 495910912 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004291875626880642, + "loss": 1.5805, + "theoretical_loss": 3.918266538239653, + "tokens_seen": 495976448 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004291775325977934, + "loss": 1.5979, + "theoretical_loss": 3.9182106994763646, + "tokens_seen": 496041984 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042916750250752256, + "loss": 1.4737, + "theoretical_loss": 3.918154870155219, + "tokens_seen": 496107520 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004291574724172518, + "loss": 1.5592, + "theoretical_loss": 3.9180990502733715, + "tokens_seen": 496173056 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004291474423269809, + "loss": 1.4917, + "theoretical_loss": 3.91804323982798, + "tokens_seen": 496238592 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042913741223671016, + "loss": 1.6005, + "theoretical_loss": 3.917987438816204, + "tokens_seen": 496304128 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004291273821464393, + "loss": 1.6376, + "theoretical_loss": 3.9179316472352026, + "tokens_seen": 496369664 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": -0.0003037569113075733, + "objective/train/docs_used": 291574, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.323807716369629, + "objective/train/original_loss": 3.32380747795105, + "objective/train/theoretical_loss": 3.917875865082138, + "objective/train/tokens_used": 516895200, + "objective/train/value_avg": -0.00518798828125, + "objective/train/value_loss": 5.42937486898154e-05, + "objective/train/value_max": -0.00041413307189941406, + "objective/train/value_min": -0.0704345703125, + "objective/train/value_reward_corr": 0.3490016330079748, + "objective/train/value_std": 0.0038204193115234375, + "objective/train/weight_avg": 0.9995723366737366, + "objective/train/weighted_lm_loss": 3.324880838394165, + "objective/train/weights_max": 2.0225229263305664, + "objective/train/weights_min": 0.581620454788208, + "theoretical_loss": 3.917875865082138, + "tokens_seen": 496435200 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004291173520561685, + "loss": 1.6258, + "theoretical_loss": 3.917875865082138, + "tokens_seen": 496435200 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004291073219658977, + "loss": 1.4931, + "theoretical_loss": 3.917820092354172, + "tokens_seen": 496500736 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004290972918756269, + "loss": 1.5202, + "theoretical_loss": 3.9177643290484694, + "tokens_seen": 496566272 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042908726178535607, + "loss": 1.5611, + "theoretical_loss": 3.9177085751621936, + "tokens_seen": 496631808 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004290772316950853, + "loss": 1.5478, + "theoretical_loss": 3.9176528306925116, + "tokens_seen": 496697344 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042906720160481443, + "loss": 1.6223, + "theoretical_loss": 3.917597095636591, + "tokens_seen": 496762880 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042905717151454367, + "loss": 1.4345, + "theoretical_loss": 3.9175413699916004, + "tokens_seen": 496828416 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004290471414242728, + "loss": 1.4916, + "theoretical_loss": 3.91748565375471, + "tokens_seen": 496893952 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042903711133400203, + "loss": 1.5188, + "theoretical_loss": 3.91742994692309, + "tokens_seen": 496959488 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004290270812437312, + "loss": 1.4088, + "theoretical_loss": 3.917374249493913, + "tokens_seen": 497025024 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004290170511534604, + "loss": 1.6595, + "theoretical_loss": 3.9173185614643535, + "tokens_seen": 497090560 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042900702106318957, + "loss": 1.5598, + "theoretical_loss": 3.9172628828315847, + "tokens_seen": 497156096 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042899699097291875, + "loss": 1.4466, + "theoretical_loss": 3.9172072135927842, + "tokens_seen": 497221632 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042898696088264793, + "loss": 1.5807, + "theoretical_loss": 3.9171515537451285, + "tokens_seen": 497287168 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042897693079237717, + "loss": 1.4767, + "theoretical_loss": 3.9170959032857953, + "tokens_seen": 497352704 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004289669007021063, + "loss": 1.6483, + "theoretical_loss": 3.9170402622119656, + "tokens_seen": 497418240 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042895687061183553, + "loss": 1.6149, + "theoretical_loss": 3.91698463052082, + "tokens_seen": 497483776 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004289468405215647, + "loss": 1.3543, + "theoretical_loss": 3.9169290082095394, + "tokens_seen": 497549312 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004289368104312939, + "loss": 1.4963, + "theoretical_loss": 3.9168733952753083, + "tokens_seen": 497614848 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004289267803410231, + "loss": 1.5037, + "theoretical_loss": 3.916817791715311, + "tokens_seen": 497680384 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042891675025075226, + "loss": 1.4094, + "theoretical_loss": 3.916762197526733, + "tokens_seen": 497745920 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042890672016048144, + "loss": 1.5549, + "theoretical_loss": 3.916706612706762, + "tokens_seen": 497811456 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042889669007021067, + "loss": 1.6533, + "theoretical_loss": 3.916651037252585, + "tokens_seen": 497876992 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004288866599799398, + "loss": 1.3947, + "theoretical_loss": 3.9165954711613917, + "tokens_seen": 497942528 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042887662988966903, + "loss": 1.6169, + "theoretical_loss": 3.9165399144303734, + "tokens_seen": 498008064 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": -0.0003094740095548332, + "objective/train/docs_used": 293024, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6165244579315186, + "objective/train/original_loss": 2.6165244579315186, + "objective/train/theoretical_loss": 3.9164843670567215, + "objective/train/tokens_used": 518533600, + "objective/train/value_avg": -0.0054168701171875, + "objective/train/value_loss": 8.119555423036218e-05, + "objective/train/value_max": -0.0003249645233154297, + "objective/train/value_min": -0.04638671875, + "objective/train/value_reward_corr": 0.2072452933775116, + "objective/train/value_std": 0.00431060791015625, + "objective/train/weight_avg": 1.0005005598068237, + "objective/train/weighted_lm_loss": 2.6415767669677734, + "objective/train/weights_max": 1.4462262392044067, + "objective/train/weights_min": 0.19856156408786774, + "theoretical_loss": 3.9164843670567215, + "tokens_seen": 498073600 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042886659979939816, + "loss": 1.3917, + "theoretical_loss": 3.9164843670567215, + "tokens_seen": 498073600 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004288565697091274, + "loss": 1.5432, + "theoretical_loss": 3.9164288290376286, + "tokens_seen": 498139136 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004288465396188566, + "loss": 1.4547, + "theoretical_loss": 3.9163733003702896, + "tokens_seen": 498204672 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042883650952858576, + "loss": 1.3813, + "theoretical_loss": 3.9163177810519, + "tokens_seen": 498270208 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042882647943831494, + "loss": 1.4329, + "theoretical_loss": 3.916262271079656, + "tokens_seen": 498335744 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004288164493480441, + "loss": 1.4701, + "theoretical_loss": 3.9162067704507555, + "tokens_seen": 498401280 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004288064192577733, + "loss": 1.4467, + "theoretical_loss": 3.9161512791623974, + "tokens_seen": 498466816 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042879638916750254, + "loss": 1.5424, + "theoretical_loss": 3.9160957972117827, + "tokens_seen": 498532352 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042878635907723166, + "loss": 1.4987, + "theoretical_loss": 3.9160403245961124, + "tokens_seen": 498597888 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004287763289869609, + "loss": 1.4762, + "theoretical_loss": 3.915984861312589, + "tokens_seen": 498663424 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004287662988966901, + "loss": 1.4281, + "theoretical_loss": 3.9159294073584174, + "tokens_seen": 498728960 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042875626880641926, + "loss": 1.4839, + "theoretical_loss": 3.9158739627308012, + "tokens_seen": 498794496 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042874623871614844, + "loss": 1.4406, + "theoretical_loss": 3.9158185274269477, + "tokens_seen": 498860032 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004287362086258776, + "loss": 1.6458, + "theoretical_loss": 3.915763101444065, + "tokens_seen": 498925568 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004287261785356068, + "loss": 1.4569, + "theoretical_loss": 3.9157076847793597, + "tokens_seen": 498991104 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042871614844533604, + "loss": 1.5051, + "theoretical_loss": 3.915652277430044, + "tokens_seen": 499056640 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042870611835506517, + "loss": 1.5005, + "theoretical_loss": 3.9155968793933273, + "tokens_seen": 499122176 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004286960882647944, + "loss": 1.5554, + "theoretical_loss": 3.915541490666423, + "tokens_seen": 499187712 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042868605817452353, + "loss": 1.437, + "theoretical_loss": 3.915486111246545, + "tokens_seen": 499253248 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042867602808425276, + "loss": 1.4132, + "theoretical_loss": 3.9154307411309066, + "tokens_seen": 499318784 + }, + { + "epoch": 0.15, + "learning_rate": 0.000428665997993982, + "loss": 1.4584, + "theoretical_loss": 3.915375380316725, + "tokens_seen": 499384320 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004286559679037111, + "loss": 1.5703, + "theoretical_loss": 3.915320028801216, + "tokens_seen": 499449856 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042864593781344036, + "loss": 1.4651, + "theoretical_loss": 3.915264686581599, + "tokens_seen": 499515392 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004286359077231695, + "loss": 1.4597, + "theoretical_loss": 3.9152093536550936, + "tokens_seen": 499580928 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004286258776328987, + "loss": 1.4673, + "theoretical_loss": 3.9151540300189196, + "tokens_seen": 499646464 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": -0.001099871820770204, + "objective/train/docs_used": 293514, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0356030464172363, + "objective/train/original_loss": 3.0356032848358154, + "objective/train/theoretical_loss": 3.9150987156702994, + "objective/train/tokens_used": 520172000, + "objective/train/value_avg": -0.006656646728515625, + "objective/train/value_loss": 7.028997788438573e-05, + "objective/train/value_max": -0.00036263465881347656, + "objective/train/value_min": -0.06866455078125, + "objective/train/value_reward_corr": 0.3753322126077656, + "objective/train/value_std": 0.00528717041015625, + "objective/train/weight_avg": 0.9922950863838196, + "objective/train/weighted_lm_loss": 3.0110652446746826, + "objective/train/weights_max": 1.6621595621109009, + "objective/train/weights_min": 0.35039201378822327, + "theoretical_loss": 3.9150987156702994, + "tokens_seen": 499712000 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004286158475426279, + "loss": 1.4753, + "theoretical_loss": 3.9150987156702994, + "tokens_seen": 499712000 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004286058174523571, + "loss": 1.5346, + "theoretical_loss": 3.915043410606457, + "tokens_seen": 499777536 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042859578736208627, + "loss": 1.4744, + "theoretical_loss": 3.914988114824615, + "tokens_seen": 499843072 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004285857572718155, + "loss": 1.4212, + "theoretical_loss": 3.914932828322, + "tokens_seen": 499908608 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042857572718154463, + "loss": 1.3466, + "theoretical_loss": 3.914877551095838, + "tokens_seen": 499974144 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042856569709127387, + "loss": 1.4774, + "theoretical_loss": 3.9148222831433577, + "tokens_seen": 500039680 + }, + { + "epoch": 0.15, + "learning_rate": 0.000428555667001003, + "loss": 1.529, + "theoretical_loss": 3.914767024461788, + "tokens_seen": 500105216 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042854563691073223, + "loss": 1.602, + "theoretical_loss": 3.9147117750483584, + "tokens_seen": 500170752 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004285356068204614, + "loss": 1.5604, + "theoretical_loss": 3.9146565349003017, + "tokens_seen": 500236288 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004285255767301906, + "loss": 1.5534, + "theoretical_loss": 3.9146013040148486, + "tokens_seen": 500301824 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042851554663991977, + "loss": 1.573, + "theoretical_loss": 3.9145460823892346, + "tokens_seen": 500367360 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042850551654964895, + "loss": 1.5646, + "theoretical_loss": 3.9144908700206944, + "tokens_seen": 500432896 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042849548645937813, + "loss": 1.5335, + "theoretical_loss": 3.9144356669064635, + "tokens_seen": 500498432 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042848545636910737, + "loss": 1.4737, + "theoretical_loss": 3.9143804730437797, + "tokens_seen": 500563968 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004284754262788365, + "loss": 1.4752, + "theoretical_loss": 3.9143252884298816, + "tokens_seen": 500629504 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042846539618856573, + "loss": 1.332, + "theoretical_loss": 3.914270113062009, + "tokens_seen": 500695040 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004284553660982949, + "loss": 1.5137, + "theoretical_loss": 3.9142149469374026, + "tokens_seen": 500760576 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004284453360080241, + "loss": 1.5019, + "theoretical_loss": 3.914159790053305, + "tokens_seen": 500826112 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004284353059177533, + "loss": 1.4506, + "theoretical_loss": 3.9141046424069597, + "tokens_seen": 500891648 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042842527582748246, + "loss": 1.4821, + "theoretical_loss": 3.91404950399561, + "tokens_seen": 500957184 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042841524573721164, + "loss": 1.422, + "theoretical_loss": 3.913994374816503, + "tokens_seen": 501022720 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042840521564694087, + "loss": 1.5882, + "theoretical_loss": 3.913939254866884, + "tokens_seen": 501088256 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042839518555667, + "loss": 1.542, + "theoretical_loss": 3.9138841441440024, + "tokens_seen": 501153792 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042838515546639923, + "loss": 1.6219, + "theoretical_loss": 3.913829042645107, + "tokens_seen": 501219328 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042837512537612836, + "loss": 1.455, + "theoretical_loss": 3.9137739503674482, + "tokens_seen": 501284864 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 0.00012907716154586524, + "objective/train/docs_used": 294722, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3063626289367676, + "objective/train/original_loss": 3.3063628673553467, + "objective/train/theoretical_loss": 3.913718867308278, + "objective/train/tokens_used": 521810400, + "objective/train/value_avg": -0.0045928955078125, + "objective/train/value_loss": 5.9336489357519895e-05, + "objective/train/value_max": -0.00038743019104003906, + "objective/train/value_min": -0.045257568359375, + "objective/train/value_reward_corr": 0.2815498589815858, + "objective/train/value_std": 0.0030765533447265625, + "objective/train/weight_avg": 1.003886103630066, + "objective/train/weighted_lm_loss": 3.3169116973876953, + "objective/train/weights_max": 1.398805022239685, + "objective/train/weights_min": 0.17430275678634644, + "theoretical_loss": 3.913718867308278, + "tokens_seen": 501350400 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004283650952858576, + "loss": 1.5585, + "theoretical_loss": 3.913718867308278, + "tokens_seen": 501350400 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004283550651955868, + "loss": 1.4036, + "theoretical_loss": 3.9136637934648477, + "tokens_seen": 501415936 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042834503510531596, + "loss": 1.5644, + "theoretical_loss": 3.9136087288344124, + "tokens_seen": 501481472 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042833500501504514, + "loss": 1.4515, + "theoretical_loss": 3.9135536734142278, + "tokens_seen": 501547008 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004283249749247743, + "loss": 1.4106, + "theoretical_loss": 3.913498627201549, + "tokens_seen": 501612544 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004283149448345035, + "loss": 1.4476, + "theoretical_loss": 3.9134435901936335, + "tokens_seen": 501678080 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042830491474423274, + "loss": 1.4843, + "theoretical_loss": 3.9133885623877407, + "tokens_seen": 501743616 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042829488465396186, + "loss": 1.4024, + "theoretical_loss": 3.91333354378113, + "tokens_seen": 501809152 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004282848545636911, + "loss": 1.6111, + "theoretical_loss": 3.9132785343710625, + "tokens_seen": 501874688 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004282748244734203, + "loss": 1.5852, + "theoretical_loss": 3.9132235341548007, + "tokens_seen": 501940224 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042826479438314946, + "loss": 1.5115, + "theoretical_loss": 3.913168543129607, + "tokens_seen": 502005760 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042825476429287864, + "loss": 1.5763, + "theoretical_loss": 3.9131135612927466, + "tokens_seen": 502071296 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004282447342026078, + "loss": 1.6105, + "theoretical_loss": 3.913058588641485, + "tokens_seen": 502136832 + }, + { + "epoch": 0.15, + "learning_rate": 0.000428234704112337, + "loss": 1.5919, + "theoretical_loss": 3.913003625173089, + "tokens_seen": 502202368 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042822467402206624, + "loss": 1.5008, + "theoretical_loss": 3.912948670884827, + "tokens_seen": 502267904 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042821464393179537, + "loss": 1.6204, + "theoretical_loss": 3.9128937257739675, + "tokens_seen": 502333440 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004282046138415246, + "loss": 1.5041, + "theoretical_loss": 3.912838789837781, + "tokens_seen": 502398976 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042819458375125373, + "loss": 1.4902, + "theoretical_loss": 3.91278386307354, + "tokens_seen": 502464512 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042818455366098296, + "loss": 1.501, + "theoretical_loss": 3.912728945478516, + "tokens_seen": 502530048 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042817452357071215, + "loss": 1.5533, + "theoretical_loss": 3.9126740370499835, + "tokens_seen": 502595584 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004281644934804413, + "loss": 1.532, + "theoretical_loss": 3.912619137785217, + "tokens_seen": 502661120 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004281544633901705, + "loss": 1.5162, + "theoretical_loss": 3.912564247681493, + "tokens_seen": 502726656 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004281444332998997, + "loss": 1.5079, + "theoretical_loss": 3.912509366736089, + "tokens_seen": 502792192 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042813440320962887, + "loss": 1.4609, + "theoretical_loss": 3.9124544949462834, + "tokens_seen": 502857728 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004281243731193581, + "loss": 1.4954, + "theoretical_loss": 3.9123996323093557, + "tokens_seen": 502923264 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 0.0006668232381343842, + "objective/train/docs_used": 295112, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2483341693878174, + "objective/train/original_loss": 3.2483344078063965, + "objective/train/theoretical_loss": 3.912344778822587, + "objective/train/tokens_used": 523448800, + "objective/train/value_avg": -0.004558563232421875, + "objective/train/value_loss": 5.9485046222107485e-05, + "objective/train/value_max": -0.0003936290740966797, + "objective/train/value_min": -0.056640625, + "objective/train/value_reward_corr": 0.31711217712375006, + "objective/train/value_std": 0.0034580230712890625, + "objective/train/weight_avg": 1.009423851966858, + "objective/train/weighted_lm_loss": 3.2740743160247803, + "objective/train/weights_max": 1.7619236707687378, + "objective/train/weights_min": 0.3493243157863617, + "theoretical_loss": 3.912344778822587, + "tokens_seen": 502988800 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042811434302908723, + "loss": 1.4821, + "theoretical_loss": 3.912344778822587, + "tokens_seen": 502988800 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042810431293881647, + "loss": 1.4967, + "theoretical_loss": 3.912289934483259, + "tokens_seen": 503054336 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042809428284854565, + "loss": 1.4217, + "theoretical_loss": 3.9122350992886554, + "tokens_seen": 503119872 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042808425275827483, + "loss": 1.5179, + "theoretical_loss": 3.91218027323606, + "tokens_seen": 503185408 + }, + { + "epoch": 0.15, + "learning_rate": 0.000428074222668004, + "loss": 1.5933, + "theoretical_loss": 3.9121254563227588, + "tokens_seen": 503250944 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004280641925777332, + "loss": 1.4756, + "theoretical_loss": 3.912070648546038, + "tokens_seen": 503316480 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004280541624874624, + "loss": 1.4089, + "theoretical_loss": 3.9120158499031854, + "tokens_seen": 503382016 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004280441323971916, + "loss": 1.5473, + "theoretical_loss": 3.91196106039149, + "tokens_seen": 503447552 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042803410230692074, + "loss": 1.5783, + "theoretical_loss": 3.9119062800082416, + "tokens_seen": 503513088 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042802407221664997, + "loss": 1.547, + "theoretical_loss": 3.9118515087507326, + "tokens_seen": 503578624 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004280140421263791, + "loss": 1.5256, + "theoretical_loss": 3.911796746616254, + "tokens_seen": 503644160 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042800401203610833, + "loss": 1.458, + "theoretical_loss": 3.911741993602101, + "tokens_seen": 503709696 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004279939819458375, + "loss": 1.4733, + "theoretical_loss": 3.911687249705567, + "tokens_seen": 503775232 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004279839518555667, + "loss": 1.4814, + "theoretical_loss": 3.911632514923948, + "tokens_seen": 503840768 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004279739217652959, + "loss": 1.4986, + "theoretical_loss": 3.911577789254542, + "tokens_seen": 503906304 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004279638916750251, + "loss": 1.5193, + "theoretical_loss": 3.9115230726946457, + "tokens_seen": 503971840 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042795386158475424, + "loss": 1.4022, + "theoretical_loss": 3.9114683652415603, + "tokens_seen": 504037376 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004279438314944835, + "loss": 1.4439, + "theoretical_loss": 3.911413666892585, + "tokens_seen": 504102912 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004279338014042126, + "loss": 1.4996, + "theoretical_loss": 3.9113589776450217, + "tokens_seen": 504168448 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042792377131394184, + "loss": 1.4373, + "theoretical_loss": 3.9113042974961734, + "tokens_seen": 504233984 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042791374122367107, + "loss": 1.5078, + "theoretical_loss": 3.911249626443344, + "tokens_seen": 504299520 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004279037111334002, + "loss": 1.5312, + "theoretical_loss": 3.9111949644838386, + "tokens_seen": 504365056 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042789368104312943, + "loss": 1.3891, + "theoretical_loss": 3.9111403116149637, + "tokens_seen": 504430592 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042788365095285856, + "loss": 1.6347, + "theoretical_loss": 3.9110856678340253, + "tokens_seen": 504496128 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004278736208625878, + "loss": 1.4748, + "theoretical_loss": 3.9110310331383342, + "tokens_seen": 504561664 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": -0.00033795152558013797, + "objective/train/docs_used": 296352, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.330782651901245, + "objective/train/original_loss": 3.330782890319824, + "objective/train/theoretical_loss": 3.910976407525199, + "objective/train/tokens_used": 525087200, + "objective/train/value_avg": -0.005130767822265625, + "objective/train/value_loss": 5.675953070749529e-05, + "objective/train/value_max": -0.00036835670471191406, + "objective/train/value_min": -0.07794189453125, + "objective/train/value_reward_corr": 0.5432023053776815, + "objective/train/value_std": 0.005153656005859375, + "objective/train/weight_avg": 0.9992616176605225, + "objective/train/weighted_lm_loss": 3.317734479904175, + "objective/train/weights_max": 1.6865912675857544, + "objective/train/weights_min": 0.3442854583263397, + "theoretical_loss": 3.910976407525199, + "tokens_seen": 504627200 + }, + { + "epoch": 0.15, + "learning_rate": 0.000427863590772317, + "loss": 1.481, + "theoretical_loss": 3.910976407525199, + "tokens_seen": 504627200 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042785356068204616, + "loss": 1.6073, + "theoretical_loss": 3.9109217909919303, + "tokens_seen": 504692736 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042784353059177534, + "loss": 1.5129, + "theoretical_loss": 3.9108671835358404, + "tokens_seen": 504758272 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004278335005015045, + "loss": 1.4511, + "theoretical_loss": 3.910812585154242, + "tokens_seen": 504823808 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004278234704112337, + "loss": 1.542, + "theoretical_loss": 3.91075799584445, + "tokens_seen": 504889344 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042781344032096294, + "loss": 1.5328, + "theoretical_loss": 3.9107034156037797, + "tokens_seen": 504954880 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042780341023069206, + "loss": 1.5232, + "theoretical_loss": 3.9106488444295473, + "tokens_seen": 505020416 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004277933801404213, + "loss": 1.5291, + "theoretical_loss": 3.9105942823190705, + "tokens_seen": 505085952 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004277833500501505, + "loss": 1.4773, + "theoretical_loss": 3.9105397292696686, + "tokens_seen": 505151488 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042777331995987966, + "loss": 1.4332, + "theoretical_loss": 3.9104851852786617, + "tokens_seen": 505217024 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042776328986960884, + "loss": 1.4, + "theoretical_loss": 3.91043065034337, + "tokens_seen": 505282560 + }, + { + "epoch": 0.15, + "learning_rate": 0.000427753259779338, + "loss": 1.5033, + "theoretical_loss": 3.9103761244611164, + "tokens_seen": 505348096 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004277432296890672, + "loss": 1.4895, + "theoretical_loss": 3.910321607629225, + "tokens_seen": 505413632 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042773319959879644, + "loss": 1.4349, + "theoretical_loss": 3.9102670998450186, + "tokens_seen": 505479168 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042772316950852557, + "loss": 1.4274, + "theoretical_loss": 3.9102126011058247, + "tokens_seen": 505544704 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004277131394182548, + "loss": 1.5055, + "theoretical_loss": 3.910158111408969, + "tokens_seen": 505610240 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042770310932798393, + "loss": 1.5056, + "theoretical_loss": 3.91010363075178, + "tokens_seen": 505675776 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042769307923771316, + "loss": 1.4133, + "theoretical_loss": 3.910049159131586, + "tokens_seen": 505741312 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042768304914744235, + "loss": 1.4902, + "theoretical_loss": 3.909994696545718, + "tokens_seen": 505806848 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042767301905717153, + "loss": 1.3613, + "theoretical_loss": 3.9099402429915076, + "tokens_seen": 505872384 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004276629889669007, + "loss": 1.5459, + "theoretical_loss": 3.909885798466286, + "tokens_seen": 505937920 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004276529588766299, + "loss": 1.5555, + "theoretical_loss": 3.909831362967388, + "tokens_seen": 506003456 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042764292878635907, + "loss": 1.6134, + "theoretical_loss": 3.9097769364921486, + "tokens_seen": 506068992 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004276328986960883, + "loss": 1.511, + "theoretical_loss": 3.909722519037903, + "tokens_seen": 506134528 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042762286860581743, + "loss": 1.5198, + "theoretical_loss": 3.909668110601988, + "tokens_seen": 506200064 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 0.00013089663116261363, + "objective/train/docs_used": 297118, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1883790493011475, + "objective/train/original_loss": 3.1883795261383057, + "objective/train/theoretical_loss": 3.9096137111817413, + "objective/train/tokens_used": 526725600, + "objective/train/value_avg": -0.005306243896484375, + "objective/train/value_loss": 6.752050103386864e-05, + "objective/train/value_max": -0.0005860328674316406, + "objective/train/value_min": -0.063720703125, + "objective/train/value_reward_corr": 0.37129315922492, + "objective/train/value_std": 0.00396728515625, + "objective/train/weight_avg": 1.004215955734253, + "objective/train/weighted_lm_loss": 3.1992785930633545, + "objective/train/weights_max": 1.388625144958496, + "objective/train/weights_min": 0.11551831662654877, + "theoretical_loss": 3.9096137111817413, + "tokens_seen": 506265600 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042761283851554667, + "loss": 1.5394, + "theoretical_loss": 3.9096137111817413, + "tokens_seen": 506265600 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042760280842527585, + "loss": 1.5416, + "theoretical_loss": 3.9095593207745045, + "tokens_seen": 506331136 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042759277833500503, + "loss": 1.4888, + "theoretical_loss": 3.9095049393776153, + "tokens_seen": 506396672 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004275827482447342, + "loss": 1.5152, + "theoretical_loss": 3.9094505669884168, + "tokens_seen": 506462208 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004275727181544634, + "loss": 1.4736, + "theoretical_loss": 3.9093962036042518, + "tokens_seen": 506527744 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004275626880641926, + "loss": 1.4481, + "theoretical_loss": 3.909341849222463, + "tokens_seen": 506593280 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004275526579739218, + "loss": 1.5651, + "theoretical_loss": 3.909287503840396, + "tokens_seen": 506658816 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042754262788365094, + "loss": 1.5386, + "theoretical_loss": 3.9092331674553975, + "tokens_seen": 506724352 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042753259779338017, + "loss": 1.592, + "theoretical_loss": 3.909178840064813, + "tokens_seen": 506789888 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004275225677031093, + "loss": 1.4321, + "theoretical_loss": 3.9091245216659924, + "tokens_seen": 506855424 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042751253761283853, + "loss": 1.5521, + "theoretical_loss": 3.9090702122562844, + "tokens_seen": 506920960 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004275025075225677, + "loss": 1.3453, + "theoretical_loss": 3.9090159118330394, + "tokens_seen": 506986496 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004274924774322969, + "loss": 1.5039, + "theoretical_loss": 3.90896162039361, + "tokens_seen": 507052032 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004274824473420261, + "loss": 1.4691, + "theoretical_loss": 3.9089073379353474, + "tokens_seen": 507117568 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004274724172517553, + "loss": 1.5837, + "theoretical_loss": 3.9088530644556068, + "tokens_seen": 507183104 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042746238716148444, + "loss": 1.4255, + "theoretical_loss": 3.908798799951743, + "tokens_seen": 507248640 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004274523570712137, + "loss": 1.5571, + "theoretical_loss": 3.908744544421112, + "tokens_seen": 507314176 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004274423269809428, + "loss": 1.4572, + "theoretical_loss": 3.908690297861071, + "tokens_seen": 507379712 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042743229689067204, + "loss": 1.4695, + "theoretical_loss": 3.9086360602689787, + "tokens_seen": 507445248 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004274222668004012, + "loss": 1.4562, + "theoretical_loss": 3.9085818316421945, + "tokens_seen": 507510784 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004274122367101304, + "loss": 1.4799, + "theoretical_loss": 3.9085276119780787, + "tokens_seen": 507576320 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004274022066198596, + "loss": 1.3921, + "theoretical_loss": 3.9084734012739935, + "tokens_seen": 507641856 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042739217652958876, + "loss": 1.4651, + "theoretical_loss": 3.9084191995273017, + "tokens_seen": 507707392 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042738214643931794, + "loss": 1.5048, + "theoretical_loss": 3.9083650067353677, + "tokens_seen": 507772928 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004273721163490472, + "loss": 1.4874, + "theoretical_loss": 3.9083108228955554, + "tokens_seen": 507838464 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 0.0008280467009171844, + "objective/train/docs_used": 298304, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.433581829071045, + "objective/train/original_loss": 3.433581829071045, + "objective/train/theoretical_loss": 3.9082566480052314, + "objective/train/tokens_used": 528364000, + "objective/train/value_avg": -0.005306243896484375, + "objective/train/value_loss": 4.823599738301709e-05, + "objective/train/value_max": -0.00046181678771972656, + "objective/train/value_min": -0.06561279296875, + "objective/train/value_reward_corr": 0.45068084847750345, + "objective/train/value_std": 0.004474639892578125, + "objective/train/weight_avg": 1.0106085538864136, + "objective/train/weighted_lm_loss": 3.461263656616211, + "objective/train/weights_max": 1.5234229564666748, + "objective/train/weights_min": 0.4902639091014862, + "theoretical_loss": 3.9082566480052314, + "tokens_seen": 507904000 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004273620862587763, + "loss": 1.6341, + "theoretical_loss": 3.9082566480052314, + "tokens_seen": 507904000 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042735205616850554, + "loss": 1.5849, + "theoretical_loss": 3.908202482061764, + "tokens_seen": 507969536 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042734202607823467, + "loss": 1.4649, + "theoretical_loss": 3.9081483250625206, + "tokens_seen": 508035072 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004273319959879639, + "loss": 1.2901, + "theoretical_loss": 3.908094177004872, + "tokens_seen": 508100608 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004273219658976931, + "loss": 1.4223, + "theoretical_loss": 3.9080400378861873, + "tokens_seen": 508166144 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042731193580742226, + "loss": 1.5196, + "theoretical_loss": 3.9079859077038392, + "tokens_seen": 508231680 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042730190571715145, + "loss": 1.4844, + "theoretical_loss": 3.907931786455201, + "tokens_seen": 508297216 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004272918756268807, + "loss": 1.663, + "theoretical_loss": 3.9078776741376453, + "tokens_seen": 508362752 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004272818455366098, + "loss": 1.3604, + "theoretical_loss": 3.9078235707485485, + "tokens_seen": 508428288 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042727181544633904, + "loss": 1.4608, + "theoretical_loss": 3.907769476285287, + "tokens_seen": 508493824 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042726178535606817, + "loss": 1.3891, + "theoretical_loss": 3.9077153907452367, + "tokens_seen": 508559360 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004272517552657974, + "loss": 1.5402, + "theoretical_loss": 3.907661314125777, + "tokens_seen": 508624896 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004272417251755266, + "loss": 1.5032, + "theoretical_loss": 3.9076072464242877, + "tokens_seen": 508690432 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042723169508525577, + "loss": 1.4959, + "theoretical_loss": 3.9075531876381486, + "tokens_seen": 508755968 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042722166499498495, + "loss": 1.5247, + "theoretical_loss": 3.9074991377647423, + "tokens_seen": 508821504 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042721163490471413, + "loss": 1.4446, + "theoretical_loss": 3.9074450968014514, + "tokens_seen": 508887040 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004272016048144433, + "loss": 1.4401, + "theoretical_loss": 3.90739106474566, + "tokens_seen": 508952576 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042719157472417255, + "loss": 1.6473, + "theoretical_loss": 3.9073370415947526, + "tokens_seen": 509018112 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004271815446339017, + "loss": 1.5353, + "theoretical_loss": 3.907283027346116, + "tokens_seen": 509083648 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004271715145436309, + "loss": 1.5313, + "theoretical_loss": 3.9072290219971375, + "tokens_seen": 509149184 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004271614844533601, + "loss": 1.5175, + "theoretical_loss": 3.907175025545205, + "tokens_seen": 509214720 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042715145436308927, + "loss": 1.4976, + "theoretical_loss": 3.907121037987708, + "tokens_seen": 509280256 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004271414242728185, + "loss": 1.6366, + "theoretical_loss": 3.907067059322038, + "tokens_seen": 509345792 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042713139418254763, + "loss": 1.439, + "theoretical_loss": 3.907013089545586, + "tokens_seen": 509411328 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042712136409227687, + "loss": 1.4492, + "theoretical_loss": 3.906959128655745, + "tokens_seen": 509476864 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": 0.0011374519672244787, + "objective/train/docs_used": 298986, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2314672470092773, + "objective/train/original_loss": 3.2314672470092773, + "objective/train/theoretical_loss": 3.906905176649909, + "objective/train/tokens_used": 530002400, + "objective/train/value_avg": -0.005523681640625, + "objective/train/value_loss": 5.263795537757687e-05, + "objective/train/value_max": -0.0004916191101074219, + "objective/train/value_min": -0.053009033203125, + "objective/train/value_reward_corr": 0.31358945864091453, + "objective/train/value_std": 0.00405120849609375, + "objective/train/weight_avg": 1.013885259628296, + "objective/train/weighted_lm_loss": 3.2692224979400635, + "objective/train/weights_max": 1.430525541305542, + "objective/train/weights_min": 0.4343506097793579, + "theoretical_loss": 3.906905176649909, + "tokens_seen": 509542400 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042711133400200605, + "loss": 1.6244, + "theoretical_loss": 3.906905176649909, + "tokens_seen": 509542400 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042710130391173523, + "loss": 1.5663, + "theoretical_loss": 3.9068512335254724, + "tokens_seen": 509607936 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004270912738214644, + "loss": 1.585, + "theoretical_loss": 3.906797299279831, + "tokens_seen": 509673472 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004270812437311936, + "loss": 1.5039, + "theoretical_loss": 3.906743373910384, + "tokens_seen": 509739008 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004270712136409228, + "loss": 1.6057, + "theoretical_loss": 3.9066894574145277, + "tokens_seen": 509804544 + }, + { + "epoch": 0.15, + "learning_rate": 0.000427061183550652, + "loss": 1.6074, + "theoretical_loss": 3.9066355497896623, + "tokens_seen": 509870080 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042705115346038114, + "loss": 1.5432, + "theoretical_loss": 3.9065816510331883, + "tokens_seen": 509935616 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042704112337011037, + "loss": 1.4257, + "theoretical_loss": 3.9065277611425073, + "tokens_seen": 510001152 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004270310932798395, + "loss": 1.5111, + "theoretical_loss": 3.906473880115022, + "tokens_seen": 510066688 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042702106318956873, + "loss": 1.4656, + "theoretical_loss": 3.9064200079481357, + "tokens_seen": 510132224 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004270110330992979, + "loss": 1.4421, + "theoretical_loss": 3.9063661446392537, + "tokens_seen": 510197760 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004270010030090271, + "loss": 1.4467, + "theoretical_loss": 3.9063122901857814, + "tokens_seen": 510263296 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004269909729187563, + "loss": 1.4696, + "theoretical_loss": 3.9062584445851267, + "tokens_seen": 510328832 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004269809428284855, + "loss": 1.5821, + "theoretical_loss": 3.9062046078346975, + "tokens_seen": 510394368 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042697091273821464, + "loss": 1.4342, + "theoretical_loss": 3.906150779931903, + "tokens_seen": 510459904 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004269608826479439, + "loss": 1.4112, + "theoretical_loss": 3.906096960874153, + "tokens_seen": 510525440 + }, + { + "epoch": 0.15, + "learning_rate": 0.000426950852557673, + "loss": 1.4431, + "theoretical_loss": 3.9060431506588595, + "tokens_seen": 510590976 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042694082246740224, + "loss": 1.6105, + "theoretical_loss": 3.905989349283435, + "tokens_seen": 510656512 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004269307923771314, + "loss": 1.4524, + "theoretical_loss": 3.905935556745293, + "tokens_seen": 510722048 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004269207622868606, + "loss": 1.4407, + "theoretical_loss": 3.9058817730418474, + "tokens_seen": 510787584 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004269107321965898, + "loss": 1.4264, + "theoretical_loss": 3.9058279981705155, + "tokens_seen": 510853120 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042690070210631896, + "loss": 1.4942, + "theoretical_loss": 3.9057742321287128, + "tokens_seen": 510918656 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042689067201604814, + "loss": 1.4189, + "theoretical_loss": 3.905720474913858, + "tokens_seen": 510984192 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004268806419257774, + "loss": 1.3722, + "theoretical_loss": 3.9056667265233704, + "tokens_seen": 511049728 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004268706118355065, + "loss": 1.4215, + "theoretical_loss": 3.9056129869546696, + "tokens_seen": 511115264 + }, + { + "epoch": 0.15, + "objective/train/advantage_avg": -0.0005821535014547408, + "objective/train/docs_used": 299996, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6964073181152344, + "objective/train/original_loss": 2.6964073181152344, + "objective/train/theoretical_loss": 3.9055592562051764, + "objective/train/tokens_used": 531640800, + "objective/train/value_avg": -0.00511932373046875, + "objective/train/value_loss": 6.881616718601435e-05, + "objective/train/value_max": -0.00033283233642578125, + "objective/train/value_min": -0.052703857421875, + "objective/train/value_reward_corr": 0.4307956233117146, + "objective/train/value_std": 0.0050811767578125, + "objective/train/weight_avg": 0.9973248839378357, + "objective/train/weighted_lm_loss": 2.674621343612671, + "objective/train/weights_max": 1.471144676208496, + "objective/train/weights_min": 0.250302791595459, + "theoretical_loss": 3.9055592562051764, + "tokens_seen": 511180800 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042686058174523574, + "loss": 1.4753, + "theoretical_loss": 3.9055592562051764, + "tokens_seen": 511180800 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042685055165496487, + "loss": 1.4782, + "theoretical_loss": 3.9055055342723137, + "tokens_seen": 511246336 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004268405215646941, + "loss": 1.4353, + "theoretical_loss": 3.905451821153505, + "tokens_seen": 511311872 + }, + { + "epoch": 0.15, + "learning_rate": 0.0004268304914744233, + "loss": 1.5497, + "theoretical_loss": 3.9053981168461744, + "tokens_seen": 511377408 + }, + { + "epoch": 0.15, + "learning_rate": 0.00042682046138415246, + "loss": 1.422, + "theoretical_loss": 3.9053444213477473, + "tokens_seen": 511442944 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042681043129388165, + "loss": 1.4181, + "theoretical_loss": 3.905290734655651, + "tokens_seen": 511508480 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004268004012036109, + "loss": 1.5432, + "theoretical_loss": 3.9052370567673123, + "tokens_seen": 511574016 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042679037111334, + "loss": 1.4505, + "theoretical_loss": 3.905183387680161, + "tokens_seen": 511639552 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042678034102306924, + "loss": 1.5322, + "theoretical_loss": 3.9051297273916257, + "tokens_seen": 511705088 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042677031093279837, + "loss": 1.4477, + "theoretical_loss": 3.9050760758991387, + "tokens_seen": 511770624 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004267602808425276, + "loss": 1.4227, + "theoretical_loss": 3.9050224332001306, + "tokens_seen": 511836160 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004267502507522568, + "loss": 1.5478, + "theoretical_loss": 3.904968799292036, + "tokens_seen": 511901696 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042674022066198597, + "loss": 1.5123, + "theoretical_loss": 3.904915174172288, + "tokens_seen": 511967232 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042673019057171515, + "loss": 1.4629, + "theoretical_loss": 3.9048615578383226, + "tokens_seen": 512032768 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042672016048144433, + "loss": 1.4484, + "theoretical_loss": 3.9048079502875748, + "tokens_seen": 512098304 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004267101303911735, + "loss": 1.4562, + "theoretical_loss": 3.9047543515174836, + "tokens_seen": 512163840 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042670010030090275, + "loss": 1.4725, + "theoretical_loss": 3.904700761525487, + "tokens_seen": 512229376 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004266900702106319, + "loss": 1.5376, + "theoretical_loss": 3.904647180309024, + "tokens_seen": 512294912 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004266800401203611, + "loss": 1.5465, + "theoretical_loss": 3.904593607865536, + "tokens_seen": 512360448 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042667001003009024, + "loss": 1.4216, + "theoretical_loss": 3.904540044192464, + "tokens_seen": 512425984 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042665997993981947, + "loss": 1.4764, + "theoretical_loss": 3.904486489287251, + "tokens_seen": 512491520 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042664994984954865, + "loss": 1.5139, + "theoretical_loss": 3.9044329431473406, + "tokens_seen": 512557056 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042663991975927783, + "loss": 1.3431, + "theoretical_loss": 3.904379405770179, + "tokens_seen": 512622592 + }, + { + "epoch": 0.16, + "learning_rate": 0.000426629889669007, + "loss": 1.4545, + "theoretical_loss": 3.9043258771532106, + "tokens_seen": 512688128 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042661985957873625, + "loss": 1.394, + "theoretical_loss": 3.9042723572938836, + "tokens_seen": 512753664 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.000995925161987543, + "objective/train/docs_used": 300643, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7111122608184814, + "objective/train/original_loss": 2.7111122608184814, + "objective/train/theoretical_loss": 3.904218846189645, + "objective/train/tokens_used": 533279200, + "objective/train/value_avg": -0.004970550537109375, + "objective/train/value_loss": 5.3629315516445786e-05, + "objective/train/value_max": -0.0004673004150390625, + "objective/train/value_min": -0.0439453125, + "objective/train/value_reward_corr": 0.37371548730980375, + "objective/train/value_std": 0.0036754608154296875, + "objective/train/weight_avg": 0.9925470352172852, + "objective/train/weighted_lm_loss": 2.6825408935546875, + "objective/train/weights_max": 1.3785483837127686, + "objective/train/weights_min": 0.33631378412246704, + "theoretical_loss": 3.904218846189645, + "tokens_seen": 512819200 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004266098294884654, + "loss": 1.4103, + "theoretical_loss": 3.904218846189645, + "tokens_seen": 512819200 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004265997993981946, + "loss": 1.4875, + "theoretical_loss": 3.9041653438379456, + "tokens_seen": 512884736 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042658976930792374, + "loss": 1.4576, + "theoretical_loss": 3.9041118502362337, + "tokens_seen": 512950272 + }, + { + "epoch": 0.16, + "learning_rate": 0.000426579739217653, + "loss": 1.5474, + "theoretical_loss": 3.904058365381962, + "tokens_seen": 513015808 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042656970912738216, + "loss": 1.4626, + "theoretical_loss": 3.904004889272583, + "tokens_seen": 513081344 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042655967903711134, + "loss": 1.4664, + "theoretical_loss": 3.90395142190555, + "tokens_seen": 513146880 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004265496489468405, + "loss": 1.6694, + "theoretical_loss": 3.9038979632783164, + "tokens_seen": 513212416 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004265396188565697, + "loss": 1.5247, + "theoretical_loss": 3.9038445133883393, + "tokens_seen": 513277952 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004265295887662989, + "loss": 1.5096, + "theoretical_loss": 3.9037910722330746, + "tokens_seen": 513343488 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004265195586760281, + "loss": 1.5119, + "theoretical_loss": 3.9037376398099806, + "tokens_seen": 513409024 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042650952858575724, + "loss": 1.5305, + "theoretical_loss": 3.903684216116516, + "tokens_seen": 513474560 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004264994984954865, + "loss": 1.3819, + "theoretical_loss": 3.9036308011501397, + "tokens_seen": 513540096 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004264894684052156, + "loss": 1.3932, + "theoretical_loss": 3.9035773949083135, + "tokens_seen": 513605632 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042647943831494484, + "loss": 1.5427, + "theoretical_loss": 3.9035239973884988, + "tokens_seen": 513671168 + }, + { + "epoch": 0.16, + "learning_rate": 0.000426469408224674, + "loss": 1.4432, + "theoretical_loss": 3.90347060858816, + "tokens_seen": 513736704 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004264593781344032, + "loss": 1.5126, + "theoretical_loss": 3.9034172285047597, + "tokens_seen": 513802240 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004264493480441324, + "loss": 1.5131, + "theoretical_loss": 3.903363857135764, + "tokens_seen": 513867776 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004264393179538616, + "loss": 1.3617, + "theoretical_loss": 3.9033104944786388, + "tokens_seen": 513933312 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042642928786359075, + "loss": 1.4975, + "theoretical_loss": 3.903257140530852, + "tokens_seen": 513998848 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042641925777332, + "loss": 1.4549, + "theoretical_loss": 3.90320379528987, + "tokens_seen": 514064384 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042640922768304916, + "loss": 1.5981, + "theoretical_loss": 3.9031504587531645, + "tokens_seen": 514129920 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042639919759277834, + "loss": 1.4795, + "theoretical_loss": 3.903097130918205, + "tokens_seen": 514195456 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004263891675025076, + "loss": 1.3173, + "theoretical_loss": 3.9030438117824633, + "tokens_seen": 514260992 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004263791374122367, + "loss": 1.3499, + "theoretical_loss": 3.9029905013434116, + "tokens_seen": 514326528 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042636910732196594, + "loss": 1.5883, + "theoretical_loss": 3.9029371995985236, + "tokens_seen": 514392064 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": 5.073475404060446e-05, + "objective/train/docs_used": 301818, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.5425896644592285, + "objective/train/original_loss": 3.5425891876220703, + "objective/train/theoretical_loss": 3.9028839065452745, + "objective/train/tokens_used": 534917600, + "objective/train/value_avg": -0.006366729736328125, + "objective/train/value_loss": 9.619225602364168e-05, + "objective/train/value_max": -0.0004935264587402344, + "objective/train/value_min": -0.055511474609375, + "objective/train/value_reward_corr": 0.37820335132157396, + "objective/train/value_std": 0.004909515380859375, + "objective/train/weight_avg": 1.004836916923523, + "objective/train/weighted_lm_loss": 3.564011573791504, + "objective/train/weights_max": 1.534434199333191, + "objective/train/weights_min": 0.20070146024227142, + "theoretical_loss": 3.9028839065452745, + "tokens_seen": 514457600 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042635907723169507, + "loss": 1.5695, + "theoretical_loss": 3.9028839065452745, + "tokens_seen": 514457600 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004263490471414243, + "loss": 1.4947, + "theoretical_loss": 3.9028306221811393, + "tokens_seen": 514523136 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004263390170511535, + "loss": 1.5258, + "theoretical_loss": 3.9027773465035955, + "tokens_seen": 514588672 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042632898696088266, + "loss": 1.5329, + "theoretical_loss": 3.9027240795101203, + "tokens_seen": 514654208 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042631895687061185, + "loss": 1.4838, + "theoretical_loss": 3.9026708211981935, + "tokens_seen": 514719744 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004263089267803411, + "loss": 1.5355, + "theoretical_loss": 3.9026175715652944, + "tokens_seen": 514785280 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004262988966900702, + "loss": 1.5766, + "theoretical_loss": 3.902564330608904, + "tokens_seen": 514850816 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042628886659979944, + "loss": 1.4915, + "theoretical_loss": 3.902511098326505, + "tokens_seen": 514916352 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042627883650952857, + "loss": 1.5972, + "theoretical_loss": 3.90245787471558, + "tokens_seen": 514981888 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004262688064192578, + "loss": 1.3746, + "theoretical_loss": 3.9024046597736124, + "tokens_seen": 515047424 + }, + { + "epoch": 0.16, + "learning_rate": 0.000426258776328987, + "loss": 1.5353, + "theoretical_loss": 3.902351453498089, + "tokens_seen": 515112960 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042624874623871617, + "loss": 1.4107, + "theoretical_loss": 3.902298255886495, + "tokens_seen": 515178496 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042623871614844535, + "loss": 1.4688, + "theoretical_loss": 3.902245066936319, + "tokens_seen": 515244032 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042622868605817453, + "loss": 1.4308, + "theoretical_loss": 3.9021918866450473, + "tokens_seen": 515309568 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004262186559679037, + "loss": 1.4092, + "theoretical_loss": 3.902138715010171, + "tokens_seen": 515375104 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042620862587763295, + "loss": 1.4788, + "theoretical_loss": 3.90208555202918, + "tokens_seen": 515440640 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004261985957873621, + "loss": 1.3966, + "theoretical_loss": 3.9020323976995663, + "tokens_seen": 515506176 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004261885656970913, + "loss": 1.4853, + "theoretical_loss": 3.9019792520188217, + "tokens_seen": 515571712 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042617853560682044, + "loss": 1.4658, + "theoretical_loss": 3.9019261149844393, + "tokens_seen": 515637248 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042616850551654967, + "loss": 1.4831, + "theoretical_loss": 3.901872986593916, + "tokens_seen": 515702784 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042615847542627885, + "loss": 1.5284, + "theoretical_loss": 3.901819866844745, + "tokens_seen": 515768320 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042614844533600803, + "loss": 1.4553, + "theoretical_loss": 3.9017667557344247, + "tokens_seen": 515833856 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004261384152457372, + "loss": 1.5041, + "theoretical_loss": 3.901713653260452, + "tokens_seen": 515899392 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042612838515546645, + "loss": 1.4338, + "theoretical_loss": 3.9016605594203257, + "tokens_seen": 515964928 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004261183550651956, + "loss": 1.5553, + "theoretical_loss": 3.9016074742115463, + "tokens_seen": 516030464 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -7.630381878698245e-05, + "objective/train/docs_used": 302480, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.353384494781494, + "objective/train/original_loss": 3.353384494781494, + "objective/train/theoretical_loss": 3.901554397631614, + "objective/train/tokens_used": 536556000, + "objective/train/value_avg": -0.00579833984375, + "objective/train/value_loss": 6.970482354518026e-05, + "objective/train/value_max": -0.0003528594970703125, + "objective/train/value_min": -0.07196044921875, + "objective/train/value_reward_corr": 0.32405412044581877, + "objective/train/value_std": 0.004558563232421875, + "objective/train/weight_avg": 1.0024510622024536, + "objective/train/weighted_lm_loss": 3.3737504482269287, + "objective/train/weights_max": 1.7551186084747314, + "objective/train/weights_min": 0.29337289929389954, + "theoretical_loss": 3.901554397631614, + "tokens_seen": 516096000 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004261083249749248, + "loss": 1.6306, + "theoretical_loss": 3.901554397631614, + "tokens_seen": 516096000 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042609829488465394, + "loss": 1.3849, + "theoretical_loss": 3.9015013296780316, + "tokens_seen": 516161536 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004260882647943832, + "loss": 1.499, + "theoretical_loss": 3.9014482703483013, + "tokens_seen": 516227072 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042607823470411236, + "loss": 1.4678, + "theoretical_loss": 3.9013952196399275, + "tokens_seen": 516292608 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042606820461384154, + "loss": 1.4301, + "theoretical_loss": 3.9013421775504153, + "tokens_seen": 516358144 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004260581745235707, + "loss": 1.4538, + "theoretical_loss": 3.901289144077271, + "tokens_seen": 516423680 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004260481444332999, + "loss": 1.3022, + "theoretical_loss": 3.9012361192180007, + "tokens_seen": 516489216 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004260381143430291, + "loss": 1.4559, + "theoretical_loss": 3.901183102970114, + "tokens_seen": 516554752 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004260280842527583, + "loss": 1.3907, + "theoretical_loss": 3.9011300953311188, + "tokens_seen": 516620288 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042601805416248744, + "loss": 1.3988, + "theoretical_loss": 3.9010770962985264, + "tokens_seen": 516685824 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004260080240722167, + "loss": 1.5075, + "theoretical_loss": 3.9010241058698476, + "tokens_seen": 516751360 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004259979939819458, + "loss": 1.4933, + "theoretical_loss": 3.9009711240425946, + "tokens_seen": 516816896 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042598796389167504, + "loss": 1.4388, + "theoretical_loss": 3.900918150814281, + "tokens_seen": 516882432 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004259779338014042, + "loss": 1.3835, + "theoretical_loss": 3.900865186182421, + "tokens_seen": 516947968 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004259679037111334, + "loss": 1.5181, + "theoretical_loss": 3.9008122301445307, + "tokens_seen": 517013504 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004259578736208626, + "loss": 1.3769, + "theoretical_loss": 3.9007592826981257, + "tokens_seen": 517079040 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004259478435305918, + "loss": 1.5151, + "theoretical_loss": 3.9007063438407235, + "tokens_seen": 517144576 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042593781344032095, + "loss": 1.3798, + "theoretical_loss": 3.9006534135698434, + "tokens_seen": 517210112 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004259277833500502, + "loss": 1.4438, + "theoretical_loss": 3.9006004918830044, + "tokens_seen": 517275648 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004259177532597793, + "loss": 1.4194, + "theoretical_loss": 3.900547578777727, + "tokens_seen": 517341184 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042590772316950854, + "loss": 1.4814, + "theoretical_loss": 3.900494674251533, + "tokens_seen": 517406720 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004258976930792377, + "loss": 1.4721, + "theoretical_loss": 3.9004417783019454, + "tokens_seen": 517472256 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004258876629889669, + "loss": 1.5358, + "theoretical_loss": 3.900388890926487, + "tokens_seen": 517537792 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004258776328986961, + "loss": 1.475, + "theoretical_loss": 3.9003360121226835, + "tokens_seen": 517603328 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042586760280842527, + "loss": 1.4653, + "theoretical_loss": 3.90028314188806, + "tokens_seen": 517668864 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.00043348217150196433, + "objective/train/docs_used": 303675, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0490334033966064, + "objective/train/original_loss": 3.0490341186523438, + "objective/train/theoretical_loss": 3.9002302802201427, + "objective/train/tokens_used": 538194400, + "objective/train/value_avg": -0.00638580322265625, + "objective/train/value_loss": 0.00010200186807196587, + "objective/train/value_max": -0.00046563148498535156, + "objective/train/value_min": -0.0880126953125, + "objective/train/value_reward_corr": 0.4535957520144748, + "objective/train/value_std": 0.00562286376953125, + "objective/train/weight_avg": 1.0002171993255615, + "objective/train/weighted_lm_loss": 3.046177625656128, + "objective/train/weights_max": 2.045781135559082, + "objective/train/weights_min": 0.29611068964004517, + "theoretical_loss": 3.9002302802201427, + "tokens_seen": 517734400 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042585757271815445, + "loss": 1.4709, + "theoretical_loss": 3.9002302802201427, + "tokens_seen": 517734400 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004258475426278837, + "loss": 1.5156, + "theoretical_loss": 3.900177427116461, + "tokens_seen": 517799936 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004258375125376128, + "loss": 1.5579, + "theoretical_loss": 3.9001245825745423, + "tokens_seen": 517865472 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042582748244734205, + "loss": 1.3411, + "theoretical_loss": 3.9000717465919177, + "tokens_seen": 517931008 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004258174523570712, + "loss": 1.4168, + "theoretical_loss": 3.9000189191661163, + "tokens_seen": 517996544 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004258074222668004, + "loss": 1.5178, + "theoretical_loss": 3.8999661002946713, + "tokens_seen": 518062080 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004257973921765296, + "loss": 1.4456, + "theoretical_loss": 3.8999132899751157, + "tokens_seen": 518127616 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042578736208625877, + "loss": 1.4667, + "theoretical_loss": 3.8998604882049825, + "tokens_seen": 518193152 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042577733199598795, + "loss": 1.5702, + "theoretical_loss": 3.8998076949818072, + "tokens_seen": 518258688 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004257673019057172, + "loss": 1.428, + "theoretical_loss": 3.899754910303126, + "tokens_seen": 518324224 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004257572718154463, + "loss": 1.4895, + "theoretical_loss": 3.8997021341664757, + "tokens_seen": 518389760 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042574724172517555, + "loss": 1.5445, + "theoretical_loss": 3.8996493665693936, + "tokens_seen": 518455296 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004257372116349047, + "loss": 1.378, + "theoretical_loss": 3.8995966075094204, + "tokens_seen": 518520832 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004257271815446339, + "loss": 1.4643, + "theoretical_loss": 3.899543856984095, + "tokens_seen": 518586368 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004257171514543631, + "loss": 1.3845, + "theoretical_loss": 3.8994911149909575, + "tokens_seen": 518651904 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004257071213640923, + "loss": 1.3667, + "theoretical_loss": 3.8994383815275517, + "tokens_seen": 518717440 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042569709127382145, + "loss": 1.3933, + "theoretical_loss": 3.899385656591421, + "tokens_seen": 518782976 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042568706118355064, + "loss": 1.4637, + "theoretical_loss": 3.8993329401801082, + "tokens_seen": 518848512 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004256770310932798, + "loss": 1.458, + "theoretical_loss": 3.899280232291159, + "tokens_seen": 518914048 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042566700100300905, + "loss": 1.4177, + "theoretical_loss": 3.899227532922119, + "tokens_seen": 518979584 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042565697091273823, + "loss": 1.3291, + "theoretical_loss": 3.8991748420705363, + "tokens_seen": 519045120 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004256469408224674, + "loss": 1.4889, + "theoretical_loss": 3.8991221597339587, + "tokens_seen": 519110656 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042563691073219665, + "loss": 1.4092, + "theoretical_loss": 3.8990694859099357, + "tokens_seen": 519176192 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004256268806419258, + "loss": 1.4823, + "theoretical_loss": 3.8990168205960174, + "tokens_seen": 519241728 + }, + { + "epoch": 0.16, + "learning_rate": 0.000425616850551655, + "loss": 1.6213, + "theoretical_loss": 3.8989641637897545, + "tokens_seen": 519307264 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.0006906199851073325, + "objective/train/docs_used": 304298, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0201995372772217, + "objective/train/original_loss": 3.0201990604400635, + "objective/train/theoretical_loss": 3.8989115154886997, + "objective/train/tokens_used": 539832800, + "objective/train/value_avg": -0.004955291748046875, + "objective/train/value_loss": 5.876619616174139e-05, + "objective/train/value_max": -0.00036978721618652344, + "objective/train/value_min": -0.047607421875, + "objective/train/value_reward_corr": 0.39599682638288947, + "objective/train/value_std": 0.004108428955078125, + "objective/train/weight_avg": 0.9958828091621399, + "objective/train/weighted_lm_loss": 3.0168018341064453, + "objective/train/weights_max": 1.534434199333191, + "objective/train/weights_min": 0.5119540691375732, + "theoretical_loss": 3.8989115154886997, + "tokens_seen": 519372800 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042560682046138414, + "loss": 1.5872, + "theoretical_loss": 3.8989115154886997, + "tokens_seen": 519372800 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004255967903711134, + "loss": 1.5045, + "theoretical_loss": 3.898858875690406, + "tokens_seen": 519438336 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042558676028084256, + "loss": 1.461, + "theoretical_loss": 3.8988062443924285, + "tokens_seen": 519503872 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042557673019057174, + "loss": 1.5863, + "theoretical_loss": 3.8987536215923217, + "tokens_seen": 519569408 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004255667001003009, + "loss": 1.5437, + "theoretical_loss": 3.8987010072876425, + "tokens_seen": 519634944 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004255566700100301, + "loss": 1.5302, + "theoretical_loss": 3.8986484014759473, + "tokens_seen": 519700480 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004255466399197593, + "loss": 1.4853, + "theoretical_loss": 3.898595804154795, + "tokens_seen": 519766016 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004255366098294885, + "loss": 1.4245, + "theoretical_loss": 3.8985432153217454, + "tokens_seen": 519831552 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042552657973921764, + "loss": 1.4544, + "theoretical_loss": 3.898490634974358, + "tokens_seen": 519897088 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004255165496489469, + "loss": 1.6151, + "theoretical_loss": 3.898438063110195, + "tokens_seen": 519962624 + }, + { + "epoch": 0.16, + "learning_rate": 0.000425506519558676, + "loss": 1.4594, + "theoretical_loss": 3.8983854997268175, + "tokens_seen": 520028160 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042549648946840524, + "loss": 1.3063, + "theoretical_loss": 3.8983329448217905, + "tokens_seen": 520093696 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004254864593781344, + "loss": 1.4192, + "theoretical_loss": 3.898280398392677, + "tokens_seen": 520159232 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004254764292878636, + "loss": 1.358, + "theoretical_loss": 3.8982278604370437, + "tokens_seen": 520224768 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004254663991975928, + "loss": 1.5729, + "theoretical_loss": 3.898175330952456, + "tokens_seen": 520290304 + }, + { + "epoch": 0.16, + "learning_rate": 0.000425456369107322, + "loss": 1.4603, + "theoretical_loss": 3.8981228099364804, + "tokens_seen": 520355840 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042544633901705115, + "loss": 1.4878, + "theoretical_loss": 3.8980702973866874, + "tokens_seen": 520421376 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004254363089267804, + "loss": 1.3821, + "theoretical_loss": 3.8980177933006455, + "tokens_seen": 520486912 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004254262788365095, + "loss": 1.5592, + "theoretical_loss": 3.897965297675925, + "tokens_seen": 520552448 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042541624874623874, + "loss": 1.4751, + "theoretical_loss": 3.897912810510097, + "tokens_seen": 520617984 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004254062186559679, + "loss": 1.4697, + "theoretical_loss": 3.897860331800735, + "tokens_seen": 520683520 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004253961885656971, + "loss": 1.3613, + "theoretical_loss": 3.8978078615454113, + "tokens_seen": 520749056 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004253861584754263, + "loss": 1.4805, + "theoretical_loss": 3.8977553997417003, + "tokens_seen": 520814592 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042537612838515547, + "loss": 1.4607, + "theoretical_loss": 3.897702946387178, + "tokens_seen": 520880128 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042536609829488465, + "loss": 1.3317, + "theoretical_loss": 3.897650501479421, + "tokens_seen": 520945664 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": 0.00011712282139342278, + "objective/train/docs_used": 305033, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9775986671447754, + "objective/train/original_loss": 2.9775984287261963, + "objective/train/theoretical_loss": 3.8975980650160067, + "objective/train/tokens_used": 541471200, + "objective/train/value_avg": -0.005527496337890625, + "objective/train/value_loss": 6.948047666810453e-05, + "objective/train/value_max": -0.0004425048828125, + "objective/train/value_min": -0.138671875, + "objective/train/value_reward_corr": 0.23663074296124056, + "objective/train/value_std": 0.004547119140625, + "objective/train/weight_avg": 1.004296898841858, + "objective/train/weighted_lm_loss": 2.9862453937530518, + "objective/train/weights_max": 3.759247303009033, + "objective/train/weights_min": 0.14313025772571564, + "theoretical_loss": 3.8975980650160067, + "tokens_seen": 521011200 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004253560682046139, + "loss": 1.3644, + "theoretical_loss": 3.8975980650160067, + "tokens_seen": 521011200 + }, + { + "epoch": 0.16, + "learning_rate": 0.000425346038114343, + "loss": 1.41, + "theoretical_loss": 3.897545636994513, + "tokens_seen": 521076736 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042533600802407225, + "loss": 1.5458, + "theoretical_loss": 3.8974932174125194, + "tokens_seen": 521142272 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004253259779338014, + "loss": 1.3946, + "theoretical_loss": 3.8974408062676065, + "tokens_seen": 521207808 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004253159478435306, + "loss": 1.4531, + "theoretical_loss": 3.897388403557356, + "tokens_seen": 521273344 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004253059177532598, + "loss": 1.4934, + "theoretical_loss": 3.897336009279349, + "tokens_seen": 521338880 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042529588766298897, + "loss": 1.5111, + "theoretical_loss": 3.8972836234311705, + "tokens_seen": 521404416 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042528585757271815, + "loss": 1.4745, + "theoretical_loss": 3.8972312460104046, + "tokens_seen": 521469952 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004252758274824474, + "loss": 1.4703, + "theoretical_loss": 3.8971788770146363, + "tokens_seen": 521535488 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004252657973921765, + "loss": 1.556, + "theoretical_loss": 3.8971265164414515, + "tokens_seen": 521601024 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042525576730190575, + "loss": 1.5611, + "theoretical_loss": 3.897074164288439, + "tokens_seen": 521666560 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004252457372116349, + "loss": 1.4462, + "theoretical_loss": 3.8970218205531855, + "tokens_seen": 521732096 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004252357071213641, + "loss": 1.4403, + "theoretical_loss": 3.8969694852332815, + "tokens_seen": 521797632 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004252256770310933, + "loss": 1.5777, + "theoretical_loss": 3.8969171583263176, + "tokens_seen": 521863168 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004252156469408225, + "loss": 1.3389, + "theoretical_loss": 3.8968648398298846, + "tokens_seen": 521928704 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042520561685055165, + "loss": 1.4261, + "theoretical_loss": 3.8968125297415748, + "tokens_seen": 521994240 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042519558676028084, + "loss": 1.4746, + "theoretical_loss": 3.896760228058982, + "tokens_seen": 522059776 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042518555667001, + "loss": 1.417, + "theoretical_loss": 3.8967079347797, + "tokens_seen": 522125312 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042517552657973925, + "loss": 1.4242, + "theoretical_loss": 3.896655649901324, + "tokens_seen": 522190848 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004251654964894684, + "loss": 1.5261, + "theoretical_loss": 3.896603373421451, + "tokens_seen": 522256384 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004251554663991976, + "loss": 1.5049, + "theoretical_loss": 3.896551105337678, + "tokens_seen": 522321920 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004251454363089268, + "loss": 1.5797, + "theoretical_loss": 3.8964988456476033, + "tokens_seen": 522387456 + }, + { + "epoch": 0.16, + "learning_rate": 0.000425135406218656, + "loss": 1.3391, + "theoretical_loss": 3.8964465943488262, + "tokens_seen": 522452992 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042512537612838516, + "loss": 1.4375, + "theoretical_loss": 3.896394351438947, + "tokens_seen": 522518528 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042511534603811434, + "loss": 1.3932, + "theoretical_loss": 3.8963421169155668, + "tokens_seen": 522584064 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.001180545543320477, + "objective/train/docs_used": 306243, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3002638816833496, + "objective/train/original_loss": 3.3002641201019287, + "objective/train/theoretical_loss": 3.8962898907762877, + "objective/train/tokens_used": 543109600, + "objective/train/value_avg": -0.005214691162109375, + "objective/train/value_loss": 0.00010788897634483874, + "objective/train/value_max": -0.0004954338073730469, + "objective/train/value_min": -0.06536865234375, + "objective/train/value_reward_corr": 0.32034625495776975, + "objective/train/value_std": 0.0040130615234375, + "objective/train/weight_avg": 0.992983341217041, + "objective/train/weighted_lm_loss": 3.29087233543396, + "objective/train/weights_max": 1.709223985671997, + "objective/train/weights_min": 0.3038864731788635, + "theoretical_loss": 3.8962898907762877, + "tokens_seen": 522649600 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004251053159478435, + "loss": 1.4251, + "theoretical_loss": 3.8962898907762877, + "tokens_seen": 522649600 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042509528585757276, + "loss": 1.6082, + "theoretical_loss": 3.896237673018714, + "tokens_seen": 522715136 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004250852557673019, + "loss": 1.5069, + "theoretical_loss": 3.8961854636404487, + "tokens_seen": 522780672 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004250752256770311, + "loss": 1.483, + "theoretical_loss": 3.896133262639097, + "tokens_seen": 522846208 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042506519558676024, + "loss": 1.4153, + "theoretical_loss": 3.896081070012266, + "tokens_seen": 522911744 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004250551654964895, + "loss": 1.4896, + "theoretical_loss": 3.8960288857575622, + "tokens_seen": 522977280 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042504513540621866, + "loss": 1.4095, + "theoretical_loss": 3.8959767098725937, + "tokens_seen": 523042816 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042503510531594784, + "loss": 1.471, + "theoretical_loss": 3.89592454235497, + "tokens_seen": 523108352 + }, + { + "epoch": 0.16, + "learning_rate": 0.000425025075225677, + "loss": 1.5439, + "theoretical_loss": 3.8958723832023017, + "tokens_seen": 523173888 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004250150451354062, + "loss": 1.4235, + "theoretical_loss": 3.8958202324121984, + "tokens_seen": 523239424 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004250050150451354, + "loss": 1.482, + "theoretical_loss": 3.8957680899822735, + "tokens_seen": 523304960 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004249949849548646, + "loss": 1.5433, + "theoretical_loss": 3.8957159559101395, + "tokens_seen": 523370496 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042498495486459375, + "loss": 1.4213, + "theoretical_loss": 3.895663830193411, + "tokens_seen": 523436032 + }, + { + "epoch": 0.16, + "learning_rate": 0.000424974924774323, + "loss": 1.5212, + "theoretical_loss": 3.8956117128297025, + "tokens_seen": 523501568 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042496489468405216, + "loss": 1.3993, + "theoretical_loss": 3.89555960381663, + "tokens_seen": 523567104 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042495486459378135, + "loss": 1.4409, + "theoretical_loss": 3.8955075031518103, + "tokens_seen": 523632640 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004249448345035105, + "loss": 1.5027, + "theoretical_loss": 3.8954554108328616, + "tokens_seen": 523698176 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004249348044132397, + "loss": 1.4056, + "theoretical_loss": 3.8954033268574033, + "tokens_seen": 523763712 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042492477432296894, + "loss": 1.58, + "theoretical_loss": 3.8953512512230546, + "tokens_seen": 523829248 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004249147442326981, + "loss": 1.4839, + "theoretical_loss": 3.8952991839274373, + "tokens_seen": 523894784 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004249047141424273, + "loss": 1.3958, + "theoretical_loss": 3.895247124968172, + "tokens_seen": 523960320 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004248946840521565, + "loss": 1.4039, + "theoretical_loss": 3.8951950743428827, + "tokens_seen": 524025856 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042488465396188567, + "loss": 1.4897, + "theoretical_loss": 3.895143032049192, + "tokens_seen": 524091392 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042487462387161485, + "loss": 1.4178, + "theoretical_loss": 3.8950909980847266, + "tokens_seen": 524156928 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004248645937813441, + "loss": 1.5784, + "theoretical_loss": 3.89503897244711, + "tokens_seen": 524222464 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": 0.0004966599517501891, + "objective/train/docs_used": 306972, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7057361602783203, + "objective/train/original_loss": 2.705735921859741, + "objective/train/theoretical_loss": 3.8949869551339704, + "objective/train/tokens_used": 544748000, + "objective/train/value_avg": -0.004886627197265625, + "objective/train/value_loss": 5.151869117980823e-05, + "objective/train/value_max": -0.0005998611450195312, + "objective/train/value_min": -0.06982421875, + "objective/train/value_reward_corr": 0.24553296181240006, + "objective/train/value_std": 0.003681182861328125, + "objective/train/weight_avg": 1.0072760581970215, + "objective/train/weighted_lm_loss": 2.723297357559204, + "objective/train/weights_max": 1.9516133069992065, + "objective/train/weights_min": 0.2971573770046234, + "theoretical_loss": 3.8949869551339704, + "tokens_seen": 524288000 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004248545636910732, + "loss": 1.4283, + "theoretical_loss": 3.8949869551339704, + "tokens_seen": 524288000 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042484453360080245, + "loss": 1.5679, + "theoretical_loss": 3.8949349461429357, + "tokens_seen": 524353536 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004248345035105316, + "loss": 1.4888, + "theoretical_loss": 3.8948829454716334, + "tokens_seen": 524419072 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004248244734202608, + "loss": 1.3674, + "theoretical_loss": 3.894830953117694, + "tokens_seen": 524484608 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042481444332999, + "loss": 1.4968, + "theoretical_loss": 3.8947789690787475, + "tokens_seen": 524550144 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042480441323971917, + "loss": 1.5373, + "theoretical_loss": 3.8947269933524264, + "tokens_seen": 524615680 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042479438314944835, + "loss": 1.5318, + "theoretical_loss": 3.8946750259363627, + "tokens_seen": 524681216 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004247843530591776, + "loss": 1.3598, + "theoretical_loss": 3.8946230668281903, + "tokens_seen": 524746752 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004247743229689067, + "loss": 1.4371, + "theoretical_loss": 3.894571116025543, + "tokens_seen": 524812288 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042476429287863595, + "loss": 1.5447, + "theoretical_loss": 3.894519173526057, + "tokens_seen": 524877824 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004247542627883651, + "loss": 1.4868, + "theoretical_loss": 3.8944672393273683, + "tokens_seen": 524943360 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004247442326980943, + "loss": 1.3411, + "theoretical_loss": 3.894415313427115, + "tokens_seen": 525008896 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004247342026078235, + "loss": 1.5857, + "theoretical_loss": 3.8943633958229347, + "tokens_seen": 525074432 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004247241725175527, + "loss": 1.4954, + "theoretical_loss": 3.894311486512467, + "tokens_seen": 525139968 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042471414242728186, + "loss": 1.357, + "theoretical_loss": 3.8942595854933515, + "tokens_seen": 525205504 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042470411233701104, + "loss": 1.3453, + "theoretical_loss": 3.894207692763231, + "tokens_seen": 525271040 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004246940822467402, + "loss": 1.4514, + "theoretical_loss": 3.8941558083197467, + "tokens_seen": 525336576 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042468405215646945, + "loss": 1.531, + "theoretical_loss": 3.8941039321605424, + "tokens_seen": 525402112 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004246740220661986, + "loss": 1.524, + "theoretical_loss": 3.8940520642832617, + "tokens_seen": 525467648 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004246639919759278, + "loss": 1.4977, + "theoretical_loss": 3.8940002046855495, + "tokens_seen": 525533184 + }, + { + "epoch": 0.16, + "learning_rate": 0.000424653961885657, + "loss": 1.4492, + "theoretical_loss": 3.8939483533650536, + "tokens_seen": 525598720 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004246439317953862, + "loss": 1.434, + "theoretical_loss": 3.8938965103194185, + "tokens_seen": 525664256 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042463390170511536, + "loss": 1.3869, + "theoretical_loss": 3.8938446755462945, + "tokens_seen": 525729792 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042462387161484454, + "loss": 1.3904, + "theoretical_loss": 3.8937928490433293, + "tokens_seen": 525795328 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004246138415245737, + "loss": 1.4649, + "theoretical_loss": 3.8937410308081737, + "tokens_seen": 525860864 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.0003395440580789, + "objective/train/docs_used": 308263, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2173409461975098, + "objective/train/original_loss": 3.2173409461975098, + "objective/train/theoretical_loss": 3.8936892208384775, + "objective/train/tokens_used": 546386400, + "objective/train/value_avg": -0.0048828125, + "objective/train/value_loss": 5.36350162292365e-05, + "objective/train/value_max": -0.00034332275390625, + "objective/train/value_min": -0.0616455078125, + "objective/train/value_reward_corr": 0.32400220150155223, + "objective/train/value_std": 0.003978729248046875, + "objective/train/weight_avg": 0.9991657137870789, + "objective/train/weighted_lm_loss": 3.223982095718384, + "objective/train/weights_max": 1.6014128923416138, + "objective/train/weights_min": 0.456885427236557, + "theoretical_loss": 3.8936892208384775, + "tokens_seen": 525926400 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042460381143430296, + "loss": 1.4642, + "theoretical_loss": 3.8936892208384775, + "tokens_seen": 525926400 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004245937813440321, + "loss": 1.4943, + "theoretical_loss": 3.8936374191318937, + "tokens_seen": 525991936 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004245837512537613, + "loss": 1.4249, + "theoretical_loss": 3.893585625686074, + "tokens_seen": 526057472 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042457372116349044, + "loss": 1.467, + "theoretical_loss": 3.8935338404986735, + "tokens_seen": 526123008 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004245636910732197, + "loss": 1.5515, + "theoretical_loss": 3.8934820635673457, + "tokens_seen": 526188544 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042455366098294886, + "loss": 1.525, + "theoretical_loss": 3.8934302948897472, + "tokens_seen": 526254080 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042454363089267804, + "loss": 1.4656, + "theoretical_loss": 3.893378534463534, + "tokens_seen": 526319616 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004245336008024072, + "loss": 1.4927, + "theoretical_loss": 3.8933267822863646, + "tokens_seen": 526385152 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004245235707121364, + "loss": 1.4632, + "theoretical_loss": 3.8932750383558963, + "tokens_seen": 526450688 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004245135406218656, + "loss": 1.4837, + "theoretical_loss": 3.8932233026697896, + "tokens_seen": 526516224 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004245035105315948, + "loss": 1.4652, + "theoretical_loss": 3.8931715752257054, + "tokens_seen": 526581760 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042449348044132395, + "loss": 1.3641, + "theoretical_loss": 3.893119856021304, + "tokens_seen": 526647296 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004244834503510532, + "loss": 1.4897, + "theoretical_loss": 3.893068145054248, + "tokens_seen": 526712832 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042447342026078236, + "loss": 1.3805, + "theoretical_loss": 3.8930164423222013, + "tokens_seen": 526778368 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042446339017051155, + "loss": 1.486, + "theoretical_loss": 3.8929647478228278, + "tokens_seen": 526843904 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004244533600802407, + "loss": 1.4769, + "theoretical_loss": 3.8929130615537932, + "tokens_seen": 526909440 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004244433299899699, + "loss": 1.385, + "theoretical_loss": 3.8928613835127632, + "tokens_seen": 526974976 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004244332998996991, + "loss": 1.3842, + "theoretical_loss": 3.892809713697405, + "tokens_seen": 527040512 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004244232698094283, + "loss": 1.4354, + "theoretical_loss": 3.8927580521053873, + "tokens_seen": 527106048 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042441323971915745, + "loss": 1.4219, + "theoretical_loss": 3.8927063987343784, + "tokens_seen": 527171584 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004244032096288867, + "loss": 1.4414, + "theoretical_loss": 3.8926547535820486, + "tokens_seen": 527237120 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004243931795386158, + "loss": 1.4801, + "theoretical_loss": 3.892603116646069, + "tokens_seen": 527302656 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042438314944834505, + "loss": 1.4785, + "theoretical_loss": 3.8925514879241114, + "tokens_seen": 527368192 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042437311935807423, + "loss": 1.4202, + "theoretical_loss": 3.8924998674138487, + "tokens_seen": 527433728 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004243630892678034, + "loss": 1.5064, + "theoretical_loss": 3.8924482551129547, + "tokens_seen": 527499264 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": 0.0004965656553395092, + "objective/train/docs_used": 309026, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.927673816680908, + "objective/train/original_loss": 2.927673816680908, + "objective/train/theoretical_loss": 3.892396651019104, + "objective/train/tokens_used": 548024800, + "objective/train/value_avg": -0.005062103271484375, + "objective/train/value_loss": 4.2534913518466055e-05, + "objective/train/value_max": -0.0004513263702392578, + "objective/train/value_min": -0.059326171875, + "objective/train/value_reward_corr": 0.31015984832595, + "objective/train/value_std": 0.003692626953125, + "objective/train/weight_avg": 1.007013201713562, + "objective/train/weighted_lm_loss": 2.941283941268921, + "objective/train/weights_max": 1.5938950777053833, + "objective/train/weights_min": 0.4454784691333771, + "theoretical_loss": 3.892396651019104, + "tokens_seen": 527564800 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004243530591775326, + "loss": 1.4663, + "theoretical_loss": 3.892396651019104, + "tokens_seen": 527564800 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004243430290872618, + "loss": 1.4001, + "theoretical_loss": 3.8923450551299728, + "tokens_seen": 527630336 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042433299899699095, + "loss": 1.4459, + "theoretical_loss": 3.8922934674432375, + "tokens_seen": 527695872 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004243229689067202, + "loss": 1.6213, + "theoretical_loss": 3.892241887956575, + "tokens_seen": 527761408 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004243129388164493, + "loss": 1.5355, + "theoretical_loss": 3.892190316667665, + "tokens_seen": 527826944 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042430290872617855, + "loss": 1.5439, + "theoretical_loss": 3.8921387535741863, + "tokens_seen": 527892480 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042429287863590773, + "loss": 1.5188, + "theoretical_loss": 3.8920871986738197, + "tokens_seen": 527958016 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004242828485456369, + "loss": 1.376, + "theoretical_loss": 3.892035651964246, + "tokens_seen": 528023552 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004242728184553661, + "loss": 1.3552, + "theoretical_loss": 3.891984113443148, + "tokens_seen": 528089088 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004242627883650953, + "loss": 1.487, + "theoretical_loss": 3.8919325831082086, + "tokens_seen": 528154624 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042425275827482446, + "loss": 1.4685, + "theoretical_loss": 3.891881060957113, + "tokens_seen": 528220160 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004242427281845537, + "loss": 1.5243, + "theoretical_loss": 3.8918295469875446, + "tokens_seen": 528285696 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004242326980942828, + "loss": 1.6006, + "theoretical_loss": 3.891778041197191, + "tokens_seen": 528351232 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042422266800401206, + "loss": 1.3285, + "theoretical_loss": 3.891726543583739, + "tokens_seen": 528416768 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004242126379137412, + "loss": 1.4077, + "theoretical_loss": 3.8916750541448764, + "tokens_seen": 528482304 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004242026078234704, + "loss": 1.4507, + "theoretical_loss": 3.8916235728782915, + "tokens_seen": 528547840 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004241925777331996, + "loss": 1.4545, + "theoretical_loss": 3.8915720997816745, + "tokens_seen": 528613376 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004241825476429288, + "loss": 1.4429, + "theoretical_loss": 3.891520634852717, + "tokens_seen": 528678912 + }, + { + "epoch": 0.16, + "learning_rate": 0.000424172517552658, + "loss": 1.4783, + "theoretical_loss": 3.89146917808911, + "tokens_seen": 528744448 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004241624874623872, + "loss": 1.5021, + "theoretical_loss": 3.8914177294885457, + "tokens_seen": 528809984 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004241524573721164, + "loss": 1.5972, + "theoretical_loss": 3.891366289048719, + "tokens_seen": 528875520 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042414242728184556, + "loss": 1.5251, + "theoretical_loss": 3.8913148567673232, + "tokens_seen": 528941056 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042413239719157474, + "loss": 1.4795, + "theoretical_loss": 3.891263432642055, + "tokens_seen": 529006592 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004241223671013039, + "loss": 1.4731, + "theoretical_loss": 3.8912120166706092, + "tokens_seen": 529072128 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042411233701103316, + "loss": 1.3924, + "theoretical_loss": 3.8911606088506847, + "tokens_seen": 529137664 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.0010372486431151628, + "objective/train/docs_used": 310459, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8341526985168457, + "objective/train/original_loss": 2.8341526985168457, + "objective/train/theoretical_loss": 3.8911092091799793, + "objective/train/tokens_used": 549663200, + "objective/train/value_avg": -0.005580902099609375, + "objective/train/value_loss": 7.557449134765193e-05, + "objective/train/value_max": -0.0003814697265625, + "objective/train/value_min": -0.057373046875, + "objective/train/value_reward_corr": 0.24528624023858148, + "objective/train/value_std": 0.00391387939453125, + "objective/train/weight_avg": 0.9930875897407532, + "objective/train/weighted_lm_loss": 2.813767194747925, + "objective/train/weights_max": 1.7748757600784302, + "objective/train/weights_min": 0.44356733560562134, + "theoretical_loss": 3.8911092091799793, + "tokens_seen": 529203200 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004241023069207623, + "loss": 1.4054, + "theoretical_loss": 3.8911092091799793, + "tokens_seen": 529203200 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004240922768304915, + "loss": 1.4428, + "theoretical_loss": 3.891057817656192, + "tokens_seen": 529268736 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042408224674022065, + "loss": 1.4814, + "theoretical_loss": 3.891006434277023, + "tokens_seen": 529334272 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004240722166499499, + "loss": 1.4433, + "theoretical_loss": 3.8909550590401736, + "tokens_seen": 529399808 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042406218655967906, + "loss": 1.4717, + "theoretical_loss": 3.890903691943346, + "tokens_seen": 529465344 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042405215646940824, + "loss": 1.4588, + "theoretical_loss": 3.890852332984242, + "tokens_seen": 529530880 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004240421263791374, + "loss": 1.3779, + "theoretical_loss": 3.890800982160567, + "tokens_seen": 529596416 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004240320962888666, + "loss": 1.5021, + "theoretical_loss": 3.890749639470025, + "tokens_seen": 529661952 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004240220661985958, + "loss": 1.427, + "theoretical_loss": 3.8906983049103223, + "tokens_seen": 529727488 + }, + { + "epoch": 0.16, + "learning_rate": 0.000424012036108325, + "loss": 1.4344, + "theoretical_loss": 3.8906469784791646, + "tokens_seen": 529793024 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042400200601805415, + "loss": 1.6088, + "theoretical_loss": 3.8905956601742604, + "tokens_seen": 529858560 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004239919759277834, + "loss": 1.4073, + "theoretical_loss": 3.890544349993318, + "tokens_seen": 529924096 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042398194583751256, + "loss": 1.5037, + "theoretical_loss": 3.890493047934047, + "tokens_seen": 529989632 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042397191574724175, + "loss": 1.444, + "theoretical_loss": 3.890441753994158, + "tokens_seen": 530055168 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004239618856569709, + "loss": 1.4905, + "theoretical_loss": 3.8903904681713612, + "tokens_seen": 530120704 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004239518555667001, + "loss": 1.6773, + "theoretical_loss": 3.89033919046337, + "tokens_seen": 530186240 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004239418254764293, + "loss": 1.4863, + "theoretical_loss": 3.8902879208678973, + "tokens_seen": 530251776 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004239317953861585, + "loss": 1.6211, + "theoretical_loss": 3.8902366593826567, + "tokens_seen": 530317312 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042392176529588765, + "loss": 1.4357, + "theoretical_loss": 3.890185406005364, + "tokens_seen": 530382848 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004239117352056169, + "loss": 1.5428, + "theoretical_loss": 3.8901341607337354, + "tokens_seen": 530448384 + }, + { + "epoch": 0.16, + "learning_rate": 0.000423901705115346, + "loss": 1.6132, + "theoretical_loss": 3.8900829235654872, + "tokens_seen": 530513920 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042389167502507525, + "loss": 1.4233, + "theoretical_loss": 3.890031694498337, + "tokens_seen": 530579456 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042388164493480443, + "loss": 1.3628, + "theoretical_loss": 3.889980473530004, + "tokens_seen": 530644992 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004238716148445336, + "loss": 1.5239, + "theoretical_loss": 3.8899292606582074, + "tokens_seen": 530710528 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004238615847542628, + "loss": 1.4623, + "theoretical_loss": 3.8898780558806685, + "tokens_seen": 530776064 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.00044801601325161755, + "objective/train/docs_used": 311126, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0486738681793213, + "objective/train/original_loss": 3.0486738681793213, + "objective/train/theoretical_loss": 3.889826859195108, + "objective/train/tokens_used": 551301600, + "objective/train/value_avg": -0.004863739013671875, + "objective/train/value_loss": 7.436619489453733e-05, + "objective/train/value_max": -0.00038909912109375, + "objective/train/value_min": -0.1153564453125, + "objective/train/value_reward_corr": 0.3750347464561856, + "objective/train/value_std": 0.00449371337890625, + "objective/train/weight_avg": 0.9986662864685059, + "objective/train/weighted_lm_loss": 3.05319881439209, + "objective/train/weights_max": 2.2633907794952393, + "objective/train/weights_min": 0.14089293777942657, + "theoretical_loss": 3.889826859195108, + "tokens_seen": 530841600 + }, + { + "epoch": 0.16, + "learning_rate": 0.000423851554663992, + "loss": 1.5185, + "theoretical_loss": 3.889826859195108, + "tokens_seen": 530841600 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042384152457372115, + "loss": 1.5411, + "theoretical_loss": 3.8897756705992492, + "tokens_seen": 530907136 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004238314944834504, + "loss": 1.4838, + "theoretical_loss": 3.8897244900908152, + "tokens_seen": 530972672 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004238214643931795, + "loss": 1.5772, + "theoretical_loss": 3.8896733176675298, + "tokens_seen": 531038208 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042381143430290875, + "loss": 1.5168, + "theoretical_loss": 3.8896221533271182, + "tokens_seen": 531103744 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042380140421263793, + "loss": 1.3181, + "theoretical_loss": 3.8895709970673074, + "tokens_seen": 531169280 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004237913741223671, + "loss": 1.4534, + "theoretical_loss": 3.889519848885824, + "tokens_seen": 531234816 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004237813440320963, + "loss": 1.4986, + "theoretical_loss": 3.889468708780395, + "tokens_seen": 531300352 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004237713139418255, + "loss": 1.4326, + "theoretical_loss": 3.8894175767487504, + "tokens_seen": 531365888 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042376128385155466, + "loss": 1.5127, + "theoretical_loss": 3.88936645278862, + "tokens_seen": 531431424 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004237512537612839, + "loss": 1.5224, + "theoretical_loss": 3.889315336897734, + "tokens_seen": 531496960 + }, + { + "epoch": 0.16, + "learning_rate": 0.000423741223671013, + "loss": 1.4485, + "theoretical_loss": 3.8892642290738246, + "tokens_seen": 531562496 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042373119358074226, + "loss": 1.5206, + "theoretical_loss": 3.8892131293146237, + "tokens_seen": 531628032 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004237211634904714, + "loss": 1.4825, + "theoretical_loss": 3.889162037617865, + "tokens_seen": 531693568 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004237111334002006, + "loss": 1.3945, + "theoretical_loss": 3.8891109539812834, + "tokens_seen": 531759104 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004237011033099298, + "loss": 1.4766, + "theoretical_loss": 3.889059878402614, + "tokens_seen": 531824640 + }, + { + "epoch": 0.16, + "learning_rate": 0.000423691073219659, + "loss": 1.5367, + "theoretical_loss": 3.8890088108795924, + "tokens_seen": 531890176 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042368104312938816, + "loss": 1.428, + "theoretical_loss": 3.888957751409956, + "tokens_seen": 531955712 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004236710130391174, + "loss": 1.5015, + "theoretical_loss": 3.888906699991443, + "tokens_seen": 532021248 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004236609829488465, + "loss": 1.6368, + "theoretical_loss": 3.8888556566217924, + "tokens_seen": 532086784 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042365095285857576, + "loss": 1.4428, + "theoretical_loss": 3.8888046212987444, + "tokens_seen": 532152320 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004236409227683049, + "loss": 1.5453, + "theoretical_loss": 3.888753594020039, + "tokens_seen": 532217856 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004236308926780341, + "loss": 1.4047, + "theoretical_loss": 3.8887025747834185, + "tokens_seen": 532283392 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004236208625877633, + "loss": 1.4502, + "theoretical_loss": 3.8886515635866252, + "tokens_seen": 532348928 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004236108324974925, + "loss": 1.5762, + "theoretical_loss": 3.8886005604274025, + "tokens_seen": 532414464 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.0006169916014187038, + "objective/train/docs_used": 312276, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3148512840270996, + "objective/train/original_loss": 3.3148512840270996, + "objective/train/theoretical_loss": 3.888549565303496, + "objective/train/tokens_used": 552940000, + "objective/train/value_avg": -0.005603790283203125, + "objective/train/value_loss": 6.020202272338793e-05, + "objective/train/value_max": -0.00047659873962402344, + "objective/train/value_min": -0.10089111328125, + "objective/train/value_reward_corr": 0.34087108409766964, + "objective/train/value_std": 0.004718780517578125, + "objective/train/weight_avg": 0.996694028377533, + "objective/train/weighted_lm_loss": 3.3053524494171143, + "objective/train/weights_max": 1.7404758930206299, + "objective/train/weights_min": 0.45903417468070984, + "theoretical_loss": 3.888549565303496, + "tokens_seen": 532480000 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042360080240722166, + "loss": 1.562, + "theoretical_loss": 3.888549565303496, + "tokens_seen": 532480000 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042359077231695085, + "loss": 1.4258, + "theoretical_loss": 3.8884985782126495, + "tokens_seen": 532545536 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042358074222668, + "loss": 1.3476, + "theoretical_loss": 3.8884475991526095, + "tokens_seen": 532611072 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042357071213640926, + "loss": 1.5009, + "theoretical_loss": 3.888396628121124, + "tokens_seen": 532676608 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004235606820461384, + "loss": 1.3679, + "theoretical_loss": 3.8883456651159407, + "tokens_seen": 532742144 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004235506519558676, + "loss": 1.3793, + "theoretical_loss": 3.888294710134808, + "tokens_seen": 532807680 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042354062186559675, + "loss": 1.4004, + "theoretical_loss": 3.8882437631754767, + "tokens_seen": 532873216 + }, + { + "epoch": 0.16, + "learning_rate": 0.000423530591775326, + "loss": 1.5597, + "theoretical_loss": 3.8881928242356976, + "tokens_seen": 532938752 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042352056168505517, + "loss": 1.5309, + "theoretical_loss": 3.8881418933132212, + "tokens_seen": 533004288 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042351053159478435, + "loss": 1.5781, + "theoretical_loss": 3.888090970405801, + "tokens_seen": 533069824 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042350050150451353, + "loss": 1.4232, + "theoretical_loss": 3.8880400555111914, + "tokens_seen": 533135360 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042349047141424276, + "loss": 1.4737, + "theoretical_loss": 3.8879891486271445, + "tokens_seen": 533200896 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004234804413239719, + "loss": 1.5108, + "theoretical_loss": 3.887938249751418, + "tokens_seen": 533266432 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042347041123370113, + "loss": 1.4571, + "theoretical_loss": 3.887887358881766, + "tokens_seen": 533331968 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042346038114343025, + "loss": 1.4804, + "theoretical_loss": 3.8878364760159476, + "tokens_seen": 533397504 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004234503510531595, + "loss": 1.4205, + "theoretical_loss": 3.88778560115172, + "tokens_seen": 533463040 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042344032096288867, + "loss": 1.4499, + "theoretical_loss": 3.887734734286842, + "tokens_seen": 533528576 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042343029087261785, + "loss": 1.4576, + "theoretical_loss": 3.8876838754190732, + "tokens_seen": 533594112 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004234202607823471, + "loss": 1.5201, + "theoretical_loss": 3.887633024546175, + "tokens_seen": 533659648 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004234102306920762, + "loss": 1.3853, + "theoretical_loss": 3.887582181665909, + "tokens_seen": 533725184 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042340020060180545, + "loss": 1.4281, + "theoretical_loss": 3.8875313467760373, + "tokens_seen": 533790720 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042339017051153463, + "loss": 1.5312, + "theoretical_loss": 3.887480519874323, + "tokens_seen": 533856256 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004233801404212638, + "loss": 1.4842, + "theoretical_loss": 3.887429700958532, + "tokens_seen": 533921792 + }, + { + "epoch": 0.16, + "learning_rate": 0.000423370110330993, + "loss": 1.4823, + "theoretical_loss": 3.8873788900264286, + "tokens_seen": 533987328 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004233600802407222, + "loss": 1.4048, + "theoretical_loss": 3.887328087075778, + "tokens_seen": 534052864 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.0005910081672482193, + "objective/train/docs_used": 312792, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2758595943450928, + "objective/train/original_loss": 3.2758595943450928, + "objective/train/theoretical_loss": 3.887277292104349, + "objective/train/tokens_used": 554578400, + "objective/train/value_avg": -0.005718231201171875, + "objective/train/value_loss": 5.6720367865636945e-05, + "objective/train/value_max": -0.0004305839538574219, + "objective/train/value_min": -0.07305908203125, + "objective/train/value_reward_corr": 0.45237446889768895, + "objective/train/value_std": 0.004978179931640625, + "objective/train/weight_avg": 0.9968087673187256, + "objective/train/weighted_lm_loss": 3.2597076892852783, + "objective/train/weights_max": 1.7367568016052246, + "objective/train/weights_min": 0.5062252283096313, + "theoretical_loss": 3.887277292104349, + "tokens_seen": 534118400 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042335005015045135, + "loss": 1.5922, + "theoretical_loss": 3.887277292104349, + "tokens_seen": 534118400 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004233400200601806, + "loss": 1.4751, + "theoretical_loss": 3.8872265051099086, + "tokens_seen": 534183936 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004233299899699097, + "loss": 1.4123, + "theoretical_loss": 3.8871757260902253, + "tokens_seen": 534249472 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042331995987963895, + "loss": 1.4989, + "theoretical_loss": 3.88712495504307, + "tokens_seen": 534315008 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042330992978936813, + "loss": 1.5537, + "theoretical_loss": 3.887074191966213, + "tokens_seen": 534380544 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004232998996990973, + "loss": 1.4716, + "theoretical_loss": 3.887023436857424, + "tokens_seen": 534446080 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004232898696088265, + "loss": 1.4481, + "theoretical_loss": 3.8869726897144785, + "tokens_seen": 534511616 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004232798395185557, + "loss": 1.5901, + "theoretical_loss": 3.8869219505351476, + "tokens_seen": 534577152 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042326980942828486, + "loss": 1.5087, + "theoretical_loss": 3.8868712193172064, + "tokens_seen": 534642688 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004232597793380141, + "loss": 1.4295, + "theoretical_loss": 3.8868204960584296, + "tokens_seen": 534708224 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004232497492477432, + "loss": 1.4337, + "theoretical_loss": 3.8867697807565937, + "tokens_seen": 534773760 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042323971915747246, + "loss": 1.5016, + "theoretical_loss": 3.886719073409475, + "tokens_seen": 534839296 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004232296890672016, + "loss": 1.4271, + "theoretical_loss": 3.886668374014852, + "tokens_seen": 534904832 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004232196589769308, + "loss": 1.3708, + "theoretical_loss": 3.8866176825705034, + "tokens_seen": 534970368 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042320962888666, + "loss": 1.3279, + "theoretical_loss": 3.8865669990742084, + "tokens_seen": 535035904 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004231995987963892, + "loss": 1.458, + "theoretical_loss": 3.8865163235237468, + "tokens_seen": 535101440 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042318956870611836, + "loss": 1.3693, + "theoretical_loss": 3.8864656559169015, + "tokens_seen": 535166976 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004231795386158476, + "loss": 1.3109, + "theoretical_loss": 3.886414996251454, + "tokens_seen": 535232512 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004231695085255767, + "loss": 1.3996, + "theoretical_loss": 3.8863643445251874, + "tokens_seen": 535298048 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042315947843530596, + "loss": 1.4359, + "theoretical_loss": 3.8863137007358857, + "tokens_seen": 535363584 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004231494483450351, + "loss": 1.3846, + "theoretical_loss": 3.8862630648813337, + "tokens_seen": 535429120 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004231394182547643, + "loss": 1.3918, + "theoretical_loss": 3.886212436959317, + "tokens_seen": 535494656 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004231293881644935, + "loss": 1.5077, + "theoretical_loss": 3.8861618169676237, + "tokens_seen": 535560192 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004231193580742227, + "loss": 1.4635, + "theoretical_loss": 3.88611120490404, + "tokens_seen": 535625728 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042310932798395186, + "loss": 1.4942, + "theoretical_loss": 3.8860606007663554, + "tokens_seen": 535691264 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.00108234747312963, + "objective/train/docs_used": 314102, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.086606502532959, + "objective/train/original_loss": 3.08660626411438, + "objective/train/theoretical_loss": 3.8860100045523582, + "objective/train/tokens_used": 556216800, + "objective/train/value_avg": -0.00588226318359375, + "objective/train/value_loss": 8.009065641090274e-05, + "objective/train/value_max": -0.0006022453308105469, + "objective/train/value_min": -0.05877685546875, + "objective/train/value_reward_corr": 0.3782611135424178, + "objective/train/value_std": 0.004642486572265625, + "objective/train/weight_avg": 0.9927505254745483, + "objective/train/weighted_lm_loss": 3.0595030784606934, + "objective/train/weights_max": 1.5272431373596191, + "objective/train/weights_min": 0.2938489317893982, + "theoretical_loss": 3.8860100045523582, + "tokens_seen": 535756800 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042309929789368105, + "loss": 1.3563, + "theoretical_loss": 3.8860100045523582, + "tokens_seen": 535756800 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004230892678034102, + "loss": 1.4602, + "theoretical_loss": 3.8859594162598396, + "tokens_seen": 535822336 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042307923771313946, + "loss": 1.322, + "theoretical_loss": 3.88590883588659, + "tokens_seen": 535887872 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004230692076228686, + "loss": 1.5086, + "theoretical_loss": 3.8858582634304017, + "tokens_seen": 535953408 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004230591775325978, + "loss": 1.4916, + "theoretical_loss": 3.885807698889068, + "tokens_seen": 536018944 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042304914744232695, + "loss": 1.5533, + "theoretical_loss": 3.885757142260382, + "tokens_seen": 536084480 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004230391173520562, + "loss": 1.4692, + "theoretical_loss": 3.8857065935421398, + "tokens_seen": 536150016 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042302908726178537, + "loss": 1.4425, + "theoretical_loss": 3.885656052732135, + "tokens_seen": 536215552 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042301905717151455, + "loss": 1.5015, + "theoretical_loss": 3.8856055198281654, + "tokens_seen": 536281088 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042300902708124373, + "loss": 1.44, + "theoretical_loss": 3.885554994828028, + "tokens_seen": 536346624 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042299899699097297, + "loss": 1.5219, + "theoretical_loss": 3.885504477729521, + "tokens_seen": 536412160 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004229889669007021, + "loss": 1.4953, + "theoretical_loss": 3.8854539685304434, + "tokens_seen": 536477696 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042297893681043133, + "loss": 1.5903, + "theoretical_loss": 3.885403467228595, + "tokens_seen": 536543232 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042296890672016045, + "loss": 1.3965, + "theoretical_loss": 3.885352973821778, + "tokens_seen": 536608768 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004229588766298897, + "loss": 1.444, + "theoretical_loss": 3.885302488307792, + "tokens_seen": 536674304 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042294884653961887, + "loss": 1.4466, + "theoretical_loss": 3.8852520106844413, + "tokens_seen": 536739840 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042293881644934805, + "loss": 1.4118, + "theoretical_loss": 3.885201540949529, + "tokens_seen": 536805376 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042292878635907723, + "loss": 1.512, + "theoretical_loss": 3.8851510791008588, + "tokens_seen": 536870912 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004229187562688064, + "loss": 1.4859, + "theoretical_loss": 3.885100625136237, + "tokens_seen": 536936448 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004229087261785356, + "loss": 1.437, + "theoretical_loss": 3.8850501790534695, + "tokens_seen": 537001984 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042289869608826483, + "loss": 1.365, + "theoretical_loss": 3.884999740850363, + "tokens_seen": 537067520 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042288866599799396, + "loss": 1.4937, + "theoretical_loss": 3.884949310524725, + "tokens_seen": 537133056 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004228786359077232, + "loss": 1.5021, + "theoretical_loss": 3.884898888074365, + "tokens_seen": 537198592 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004228686058174523, + "loss": 1.373, + "theoretical_loss": 3.8848484734970925, + "tokens_seen": 537264128 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042285857572718155, + "loss": 1.584, + "theoretical_loss": 3.884798066790718, + "tokens_seen": 537329664 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": 0.0002927820896729827, + "objective/train/docs_used": 314596, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.328258752822876, + "objective/train/original_loss": 2.328258991241455, + "objective/train/theoretical_loss": 3.884747667953053, + "objective/train/tokens_used": 557855200, + "objective/train/value_avg": -0.004749298095703125, + "objective/train/value_loss": 5.69921285205055e-05, + "objective/train/value_max": -0.00046372413635253906, + "objective/train/value_min": -0.047943115234375, + "objective/train/value_reward_corr": 0.2522648476345838, + "objective/train/value_std": 0.0035686492919921875, + "objective/train/weight_avg": 1.0056113004684448, + "objective/train/weighted_lm_loss": 2.3502960205078125, + "objective/train/weights_max": 1.5452420711517334, + "objective/train/weights_min": 0.5170876979827881, + "theoretical_loss": 3.884747667953053, + "tokens_seen": 537395200 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042284854563691074, + "loss": 1.4538, + "theoretical_loss": 3.884747667953053, + "tokens_seen": 537395200 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004228385155466399, + "loss": 1.4513, + "theoretical_loss": 3.8846972769819095, + "tokens_seen": 537460736 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004228284854563691, + "loss": 1.5847, + "theoretical_loss": 3.884646893875101, + "tokens_seen": 537526272 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042281845536609833, + "loss": 1.5015, + "theoretical_loss": 3.884596518630441, + "tokens_seen": 537591808 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042280842527582746, + "loss": 1.5024, + "theoretical_loss": 3.884546151245745, + "tokens_seen": 537657344 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004227983951855567, + "loss": 1.3751, + "theoretical_loss": 3.8844957917188285, + "tokens_seen": 537722880 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004227883650952858, + "loss": 1.5057, + "theoretical_loss": 3.8844454400475086, + "tokens_seen": 537788416 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042277833500501506, + "loss": 1.3993, + "theoretical_loss": 3.8843950962296026, + "tokens_seen": 537853952 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042276830491474424, + "loss": 1.5454, + "theoretical_loss": 3.8843447602629277, + "tokens_seen": 537919488 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004227582748244734, + "loss": 1.4718, + "theoretical_loss": 3.884294432145305, + "tokens_seen": 537985024 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004227482447342026, + "loss": 1.4714, + "theoretical_loss": 3.884244111874554, + "tokens_seen": 538050560 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004227382146439318, + "loss": 1.4933, + "theoretical_loss": 3.8841937994484956, + "tokens_seen": 538116096 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042272818455366096, + "loss": 1.4337, + "theoretical_loss": 3.8841434948649507, + "tokens_seen": 538181632 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004227181544633902, + "loss": 1.3551, + "theoretical_loss": 3.884093198121744, + "tokens_seen": 538247168 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004227081243731193, + "loss": 1.4228, + "theoretical_loss": 3.884042909216698, + "tokens_seen": 538312704 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042269809428284856, + "loss": 1.3709, + "theoretical_loss": 3.883992628147637, + "tokens_seen": 538378240 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004226880641925777, + "loss": 1.5063, + "theoretical_loss": 3.883942354912387, + "tokens_seen": 538443776 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004226780341023069, + "loss": 1.4789, + "theoretical_loss": 3.883892089508774, + "tokens_seen": 538509312 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042266800401203616, + "loss": 1.3818, + "theoretical_loss": 3.8838418319346246, + "tokens_seen": 538574848 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004226579739217653, + "loss": 1.4527, + "theoretical_loss": 3.8837915821877678, + "tokens_seen": 538640384 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004226479438314945, + "loss": 1.5355, + "theoretical_loss": 3.883741340266031, + "tokens_seen": 538705920 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004226379137412237, + "loss": 1.3715, + "theoretical_loss": 3.8836911061672454, + "tokens_seen": 538771456 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004226278836509529, + "loss": 1.5309, + "theoretical_loss": 3.883640879889241, + "tokens_seen": 538836992 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042261785356068206, + "loss": 1.5024, + "theoretical_loss": 3.8835906614298485, + "tokens_seen": 538902528 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042260782347041125, + "loss": 1.4658, + "theoretical_loss": 3.883540450786901, + "tokens_seen": 538968064 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -4.542338501778431e-05, + "objective/train/docs_used": 315427, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9225897789001465, + "objective/train/original_loss": 2.9225897789001465, + "objective/train/theoretical_loss": 3.8834902479582314, + "objective/train/tokens_used": 559493600, + "objective/train/value_avg": -0.005252838134765625, + "objective/train/value_loss": 6.01616884523537e-05, + "objective/train/value_max": -0.0004172325134277344, + "objective/train/value_min": -0.089111328125, + "objective/train/value_reward_corr": 0.27252133054743305, + "objective/train/value_std": 0.003940582275390625, + "objective/train/weight_avg": 1.0023977756500244, + "objective/train/weighted_lm_loss": 2.923440456390381, + "objective/train/weights_max": 2.262220621109009, + "objective/train/weights_min": 0.3588053584098816, + "theoretical_loss": 3.8834902479582314, + "tokens_seen": 539033600 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004225977933801404, + "loss": 1.4406, + "theoretical_loss": 3.8834902479582314, + "tokens_seen": 539033600 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042258776328986966, + "loss": 1.4306, + "theoretical_loss": 3.883440052941674, + "tokens_seen": 539099136 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004225777331995988, + "loss": 1.3287, + "theoretical_loss": 3.8833898657350634, + "tokens_seen": 539164672 + }, + { + "epoch": 0.16, + "learning_rate": 0.000422567703109328, + "loss": 1.4187, + "theoretical_loss": 3.8833396863362353, + "tokens_seen": 539230208 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042255767301905715, + "loss": 1.4525, + "theoretical_loss": 3.8832895147430264, + "tokens_seen": 539295744 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004225476429287864, + "loss": 1.4354, + "theoretical_loss": 3.8832393509532745, + "tokens_seen": 539361280 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042253761283851557, + "loss": 1.4476, + "theoretical_loss": 3.8831891949648174, + "tokens_seen": 539426816 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042252758274824475, + "loss": 1.4355, + "theoretical_loss": 3.883139046775495, + "tokens_seen": 539492352 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042251755265797393, + "loss": 1.2931, + "theoretical_loss": 3.883088906383146, + "tokens_seen": 539557888 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042250752256770317, + "loss": 1.4519, + "theoretical_loss": 3.8830387737856134, + "tokens_seen": 539623424 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004224974924774323, + "loss": 1.3726, + "theoretical_loss": 3.882988648980737, + "tokens_seen": 539688960 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042248746238716153, + "loss": 1.4176, + "theoretical_loss": 3.88293853196636, + "tokens_seen": 539754496 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042247743229689065, + "loss": 1.3698, + "theoretical_loss": 3.8828884227403266, + "tokens_seen": 539820032 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004224674022066199, + "loss": 1.4903, + "theoretical_loss": 3.882838321300481, + "tokens_seen": 539885568 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042245737211634907, + "loss": 1.5154, + "theoretical_loss": 3.8827882276446672, + "tokens_seen": 539951104 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042244734202607825, + "loss": 1.4751, + "theoretical_loss": 3.8827381417707327, + "tokens_seen": 540016640 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042243731193580743, + "loss": 1.4518, + "theoretical_loss": 3.8826880636765235, + "tokens_seen": 540082176 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004224272818455366, + "loss": 1.5049, + "theoretical_loss": 3.8826379933598885, + "tokens_seen": 540147712 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004224172517552658, + "loss": 1.4508, + "theoretical_loss": 3.8825879308186755, + "tokens_seen": 540213248 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042240722166499503, + "loss": 1.3766, + "theoretical_loss": 3.8825378760507334, + "tokens_seen": 540278784 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042239719157472416, + "loss": 1.483, + "theoretical_loss": 3.8824878290539138, + "tokens_seen": 540344320 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004223871614844534, + "loss": 1.374, + "theoretical_loss": 3.8824377898260676, + "tokens_seen": 540409856 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004223771313941825, + "loss": 1.4442, + "theoretical_loss": 3.882387758365046, + "tokens_seen": 540475392 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042236710130391176, + "loss": 1.3766, + "theoretical_loss": 3.8823377346687034, + "tokens_seen": 540540928 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042235707121364094, + "loss": 1.4505, + "theoretical_loss": 3.882287718734892, + "tokens_seen": 540606464 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.0007516786572523415, + "objective/train/docs_used": 316530, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.384727954864502, + "objective/train/original_loss": 3.384727716445923, + "objective/train/theoretical_loss": 3.8822377105614674, + "objective/train/tokens_used": 561132000, + "objective/train/value_avg": -0.006496429443359375, + "objective/train/value_loss": 0.00010967024718411267, + "objective/train/value_max": -0.0005793571472167969, + "objective/train/value_min": -0.09552001953125, + "objective/train/value_reward_corr": 0.5731518154413667, + "objective/train/value_std": 0.00687408447265625, + "objective/train/weight_avg": 0.9973310232162476, + "objective/train/weighted_lm_loss": 3.37292218208313, + "objective/train/weights_max": 2.5991909503936768, + "objective/train/weights_min": 0.0862945094704628, + "theoretical_loss": 3.8822377105614674, + "tokens_seen": 540672000 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004223470411233701, + "loss": 1.4423, + "theoretical_loss": 3.8822377105614674, + "tokens_seen": 540672000 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004223370110330993, + "loss": 1.4007, + "theoretical_loss": 3.882187710146285, + "tokens_seen": 540737536 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042232698094282853, + "loss": 1.4461, + "theoretical_loss": 3.882137717487201, + "tokens_seen": 540803072 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042231695085255766, + "loss": 1.51, + "theoretical_loss": 3.8820877325820726, + "tokens_seen": 540868608 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004223069207622869, + "loss": 1.4054, + "theoretical_loss": 3.8820377554287573, + "tokens_seen": 540934144 + }, + { + "epoch": 0.16, + "learning_rate": 0.000422296890672016, + "loss": 1.4696, + "theoretical_loss": 3.881987786025115, + "tokens_seen": 540999680 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042228686058174526, + "loss": 1.4745, + "theoretical_loss": 3.8819378243690044, + "tokens_seen": 541065216 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042227683049147444, + "loss": 1.3847, + "theoretical_loss": 3.8818878704582866, + "tokens_seen": 541130752 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004222668004012036, + "loss": 1.4434, + "theoretical_loss": 3.8818379242908234, + "tokens_seen": 541196288 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004222567703109328, + "loss": 1.3386, + "theoretical_loss": 3.8817879858644764, + "tokens_seen": 541261824 + }, + { + "epoch": 0.16, + "learning_rate": 0.000422246740220662, + "loss": 1.4273, + "theoretical_loss": 3.8817380551771095, + "tokens_seen": 541327360 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042223671013039116, + "loss": 1.5126, + "theoretical_loss": 3.881688132226586, + "tokens_seen": 541392896 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004222266800401204, + "loss": 1.4342, + "theoretical_loss": 3.881638217010771, + "tokens_seen": 541458432 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004222166499498495, + "loss": 1.4523, + "theoretical_loss": 3.88158830952753, + "tokens_seen": 541523968 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042220661985957876, + "loss": 1.4776, + "theoretical_loss": 3.88153840977473, + "tokens_seen": 541589504 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004221965897693079, + "loss": 1.4109, + "theoretical_loss": 3.8814885177502374, + "tokens_seen": 541655040 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004221865596790371, + "loss": 1.4802, + "theoretical_loss": 3.8814386334519213, + "tokens_seen": 541720576 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004221765295887663, + "loss": 1.4483, + "theoretical_loss": 3.881388756877651, + "tokens_seen": 541786112 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004221664994984955, + "loss": 1.4595, + "theoretical_loss": 3.8813388880252955, + "tokens_seen": 541851648 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042215646940822467, + "loss": 1.4092, + "theoretical_loss": 3.881289026892726, + "tokens_seen": 541917184 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004221464393179539, + "loss": 1.4279, + "theoretical_loss": 3.881239173477814, + "tokens_seen": 541982720 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042213640922768303, + "loss": 1.4611, + "theoretical_loss": 3.8811893277784324, + "tokens_seen": 542048256 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042212637913741226, + "loss": 1.4572, + "theoretical_loss": 3.881139489792454, + "tokens_seen": 542113792 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004221163490471414, + "loss": 1.3132, + "theoretical_loss": 3.8810896595177526, + "tokens_seen": 542179328 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004221063189568706, + "loss": 1.3937, + "theoretical_loss": 3.881039836952204, + "tokens_seen": 542244864 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": -0.001498803379945457, + "objective/train/docs_used": 317270, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.229557514190674, + "objective/train/original_loss": 3.2295570373535156, + "objective/train/theoretical_loss": 3.880990022093684, + "objective/train/tokens_used": 562770400, + "objective/train/value_avg": -0.00662994384765625, + "objective/train/value_loss": 9.023771417560056e-05, + "objective/train/value_max": -0.0004322528839111328, + "objective/train/value_min": -0.092529296875, + "objective/train/value_reward_corr": 0.42108966837786643, + "objective/train/value_std": 0.00572967529296875, + "objective/train/weight_avg": 0.9892557859420776, + "objective/train/weighted_lm_loss": 3.1949257850646973, + "objective/train/weights_max": 2.052600145339966, + "objective/train/weights_min": 0.5585831999778748, + "theoretical_loss": 3.880990022093684, + "tokens_seen": 542310400 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004220962888665998, + "loss": 1.479, + "theoretical_loss": 3.880990022093684, + "tokens_seen": 542310400 + }, + { + "epoch": 0.16, + "learning_rate": 0.000422086258776329, + "loss": 1.3737, + "theoretical_loss": 3.8809402149400682, + "tokens_seen": 542375936 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042207622868605817, + "loss": 1.5073, + "theoretical_loss": 3.880890415489235, + "tokens_seen": 542441472 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042206619859578735, + "loss": 1.496, + "theoretical_loss": 3.8808406237390622, + "tokens_seen": 542507008 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042205616850551653, + "loss": 1.3518, + "theoretical_loss": 3.8807908396874295, + "tokens_seen": 542572544 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042204613841524577, + "loss": 1.4741, + "theoretical_loss": 3.880741063332217, + "tokens_seen": 542638080 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004220361083249749, + "loss": 1.5, + "theoretical_loss": 3.880691294671305, + "tokens_seen": 542703616 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042202607823470413, + "loss": 1.3724, + "theoretical_loss": 3.8806415337025753, + "tokens_seen": 542769152 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004220160481444333, + "loss": 1.5337, + "theoretical_loss": 3.880591780423911, + "tokens_seen": 542834688 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004220060180541625, + "loss": 1.5304, + "theoretical_loss": 3.880542034833195, + "tokens_seen": 542900224 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004219959879638917, + "loss": 1.5294, + "theoretical_loss": 3.880492296928312, + "tokens_seen": 542965760 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042198595787362085, + "loss": 1.4764, + "theoretical_loss": 3.880442566707146, + "tokens_seen": 543031296 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042197592778335004, + "loss": 1.4513, + "theoretical_loss": 3.880392844167584, + "tokens_seen": 543096832 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042196589769307927, + "loss": 1.4967, + "theoretical_loss": 3.880343129307512, + "tokens_seen": 543162368 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004219558676028084, + "loss": 1.5368, + "theoretical_loss": 3.8802934221248178, + "tokens_seen": 543227904 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042194583751253763, + "loss": 1.395, + "theoretical_loss": 3.88024372261739, + "tokens_seen": 543293440 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042193580742226676, + "loss": 1.4483, + "theoretical_loss": 3.8801940307831178, + "tokens_seen": 543358976 + }, + { + "epoch": 0.16, + "learning_rate": 0.000421925777331996, + "loss": 1.5502, + "theoretical_loss": 3.8801443466198906, + "tokens_seen": 543424512 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042191574724172523, + "loss": 1.4742, + "theoretical_loss": 3.8800946701256005, + "tokens_seen": 543490048 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042190571715145436, + "loss": 1.3724, + "theoretical_loss": 3.880045001298138, + "tokens_seen": 543555584 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004218956870611836, + "loss": 1.434, + "theoretical_loss": 3.8799953401353964, + "tokens_seen": 543621120 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004218856569709127, + "loss": 1.3147, + "theoretical_loss": 3.8799456866352693, + "tokens_seen": 543686656 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042187562688064196, + "loss": 1.4653, + "theoretical_loss": 3.8798960407956504, + "tokens_seen": 543752192 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042186559679037114, + "loss": 1.4127, + "theoretical_loss": 3.879846402614435, + "tokens_seen": 543817728 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004218555667001003, + "loss": 1.4958, + "theoretical_loss": 3.8797967720895183, + "tokens_seen": 543883264 + }, + { + "epoch": 0.16, + "objective/train/advantage_avg": 0.00016097772459033877, + "objective/train/docs_used": 318285, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.101067543029785, + "objective/train/original_loss": 3.1010682582855225, + "objective/train/theoretical_loss": 3.8797471492187987, + "objective/train/tokens_used": 564408800, + "objective/train/value_avg": -0.005077362060546875, + "objective/train/value_loss": 5.1315360906301066e-05, + "objective/train/value_max": -0.00042724609375, + "objective/train/value_min": -0.0755615234375, + "objective/train/value_reward_corr": 0.4226775529317669, + "objective/train/value_std": 0.00395965576171875, + "objective/train/weight_avg": 1.0040329694747925, + "objective/train/weighted_lm_loss": 3.117164134979248, + "objective/train/weights_max": 1.4245123863220215, + "objective/train/weights_min": 0.3883121609687805, + "theoretical_loss": 3.8797471492187987, + "tokens_seen": 543948800 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004218455366098295, + "loss": 1.5351, + "theoretical_loss": 3.8797471492187987, + "tokens_seen": 543948800 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042183550651955873, + "loss": 1.4399, + "theoretical_loss": 3.8796975340001723, + "tokens_seen": 544014336 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042182547642928786, + "loss": 1.4938, + "theoretical_loss": 3.879647926431538, + "tokens_seen": 544079872 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004218154463390171, + "loss": 1.6045, + "theoretical_loss": 3.879598326510795, + "tokens_seen": 544145408 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004218054162487462, + "loss": 1.5685, + "theoretical_loss": 3.879548734235843, + "tokens_seen": 544210944 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042179538615847546, + "loss": 1.5257, + "theoretical_loss": 3.879499149604583, + "tokens_seen": 544276480 + }, + { + "epoch": 0.16, + "learning_rate": 0.00042178535606820464, + "loss": 1.4375, + "theoretical_loss": 3.879449572614918, + "tokens_seen": 544342016 + }, + { + "epoch": 0.16, + "learning_rate": 0.0004217753259779338, + "loss": 1.5252, + "theoretical_loss": 3.8794000032647484, + "tokens_seen": 544407552 + }, + { + "epoch": 0.16, + "learning_rate": 0.000421765295887663, + "loss": 1.3757, + "theoretical_loss": 3.8793504415519795, + "tokens_seen": 544473088 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004217552657973922, + "loss": 1.491, + "theoretical_loss": 3.879300887474514, + "tokens_seen": 544538624 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042174523570712136, + "loss": 1.562, + "theoretical_loss": 3.8792513410302574, + "tokens_seen": 544604160 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004217352056168506, + "loss": 1.3727, + "theoretical_loss": 3.8792018022171164, + "tokens_seen": 544669696 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004217251755265797, + "loss": 1.4933, + "theoretical_loss": 3.8791522710329964, + "tokens_seen": 544735232 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042171514543630896, + "loss": 1.4499, + "theoretical_loss": 3.8791027474758057, + "tokens_seen": 544800768 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004217051153460381, + "loss": 1.544, + "theoretical_loss": 3.8790532315434523, + "tokens_seen": 544866304 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004216950852557673, + "loss": 1.3466, + "theoretical_loss": 3.879003723233846, + "tokens_seen": 544931840 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004216850551654965, + "loss": 1.527, + "theoretical_loss": 3.8789542225448956, + "tokens_seen": 544997376 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004216750250752257, + "loss": 1.4065, + "theoretical_loss": 3.878904729474513, + "tokens_seen": 545062912 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042166499498495487, + "loss": 1.5059, + "theoretical_loss": 3.8788552440206088, + "tokens_seen": 545128448 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004216549648946841, + "loss": 1.447, + "theoretical_loss": 3.8788057661810966, + "tokens_seen": 545193984 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042164493480441323, + "loss": 1.3553, + "theoretical_loss": 3.878756295953889, + "tokens_seen": 545259520 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042163490471414246, + "loss": 1.5949, + "theoretical_loss": 3.8787068333369, + "tokens_seen": 545325056 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004216248746238716, + "loss": 1.5526, + "theoretical_loss": 3.878657378328045, + "tokens_seen": 545390592 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004216148445336008, + "loss": 1.4327, + "theoretical_loss": 3.878607930925239, + "tokens_seen": 545456128 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042160481444333, + "loss": 1.4306, + "theoretical_loss": 3.878558491126399, + "tokens_seen": 545521664 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.0012106276117265224, + "objective/train/docs_used": 318961, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6248929500579834, + "objective/train/original_loss": 2.624892473220825, + "objective/train/theoretical_loss": 3.878509058929443, + "objective/train/tokens_used": 566047200, + "objective/train/value_avg": -0.006053924560546875, + "objective/train/value_loss": 0.00011356089089531451, + "objective/train/value_max": -0.00032258033752441406, + "objective/train/value_min": -0.1201171875, + "objective/train/value_reward_corr": 0.54100585689991, + "objective/train/value_std": 0.006488800048828125, + "objective/train/weight_avg": 0.9929178953170776, + "objective/train/weighted_lm_loss": 2.5845723152160645, + "objective/train/weights_max": 2.382310390472412, + "objective/train/weights_min": 0.2555903494358063, + "theoretical_loss": 3.878509058929443, + "tokens_seen": 545587200 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004215947843530592, + "loss": 1.5052, + "theoretical_loss": 3.878509058929443, + "tokens_seen": 545587200 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042158475426278837, + "loss": 1.5185, + "theoretical_loss": 3.8784596343322884, + "tokens_seen": 545652736 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042157472417251755, + "loss": 1.4122, + "theoretical_loss": 3.8784102173328545, + "tokens_seen": 545718272 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042156469408224673, + "loss": 1.4225, + "theoretical_loss": 3.8783608079290603, + "tokens_seen": 545783808 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042155466399197597, + "loss": 1.354, + "theoretical_loss": 3.8783114061188284, + "tokens_seen": 545849344 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004215446339017051, + "loss": 1.4921, + "theoretical_loss": 3.8782620119000786, + "tokens_seen": 545914880 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042153460381143433, + "loss": 1.3674, + "theoretical_loss": 3.8782126252707334, + "tokens_seen": 545980416 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004215245737211635, + "loss": 1.4207, + "theoretical_loss": 3.8781632462287163, + "tokens_seen": 546045952 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004215145436308927, + "loss": 1.3707, + "theoretical_loss": 3.8781138747719517, + "tokens_seen": 546111488 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004215045135406219, + "loss": 1.4939, + "theoretical_loss": 3.8780645108983633, + "tokens_seen": 546177024 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042149448345035105, + "loss": 1.378, + "theoretical_loss": 3.8780151546058774, + "tokens_seen": 546242560 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042148445336008024, + "loss": 1.4986, + "theoretical_loss": 3.87796580589242, + "tokens_seen": 546308096 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042147442326980947, + "loss": 1.4016, + "theoretical_loss": 3.8779164647559186, + "tokens_seen": 546373632 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004214643931795386, + "loss": 1.3902, + "theoretical_loss": 3.877867131194301, + "tokens_seen": 546439168 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042145436308926783, + "loss": 1.4998, + "theoretical_loss": 3.8778178052054963, + "tokens_seen": 546504704 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042144433299899696, + "loss": 1.4254, + "theoretical_loss": 3.877768486787434, + "tokens_seen": 546570240 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004214343029087262, + "loss": 1.3265, + "theoretical_loss": 3.877719175938044, + "tokens_seen": 546635776 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004214242728184554, + "loss": 1.4561, + "theoretical_loss": 3.8776698726552583, + "tokens_seen": 546701312 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042141424272818456, + "loss": 1.3997, + "theoretical_loss": 3.877620576937009, + "tokens_seen": 546766848 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042140421263791374, + "loss": 1.5411, + "theoretical_loss": 3.877571288781229, + "tokens_seen": 546832384 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004213941825476429, + "loss": 1.4921, + "theoretical_loss": 3.877522008185851, + "tokens_seen": 546897920 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004213841524573721, + "loss": 1.4289, + "theoretical_loss": 3.877472735148811, + "tokens_seen": 546963456 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042137412236710134, + "loss": 1.423, + "theoretical_loss": 3.8774234696680434, + "tokens_seen": 547028992 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042136409227683046, + "loss": 1.4102, + "theoretical_loss": 3.8773742117414844, + "tokens_seen": 547094528 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004213540621865597, + "loss": 1.4794, + "theoretical_loss": 3.877324961367072, + "tokens_seen": 547160064 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": 0.0011614927789196372, + "objective/train/docs_used": 320353, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.294041156768799, + "objective/train/original_loss": 3.294041395187378, + "objective/train/theoretical_loss": 3.877275718542742, + "objective/train/tokens_used": 567685600, + "objective/train/value_avg": -0.0060272216796875, + "objective/train/value_loss": 6.0587288317037746e-05, + "objective/train/value_max": -0.0005507469177246094, + "objective/train/value_min": -0.08465576171875, + "objective/train/value_reward_corr": 0.3952569817658803, + "objective/train/value_std": 0.005512237548828125, + "objective/train/weight_avg": 1.0146359205245972, + "objective/train/weighted_lm_loss": 3.3354640007019043, + "objective/train/weights_max": 2.331606864929199, + "objective/train/weights_min": 0.553996741771698, + "theoretical_loss": 3.877275718542742, + "tokens_seen": 547225600 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004213440320962889, + "loss": 1.4804, + "theoretical_loss": 3.877275718542742, + "tokens_seen": 547225600 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042133400200601806, + "loss": 1.405, + "theoretical_loss": 3.8772264832664347, + "tokens_seen": 547291136 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042132397191574724, + "loss": 1.3493, + "theoretical_loss": 3.877177255536089, + "tokens_seen": 547356672 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004213139418254764, + "loss": 1.4431, + "theoretical_loss": 3.8771280353496445, + "tokens_seen": 547422208 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004213039117352056, + "loss": 1.4638, + "theoretical_loss": 3.877078822705043, + "tokens_seen": 547487744 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042129388164493484, + "loss": 1.4659, + "theoretical_loss": 3.877029617600226, + "tokens_seen": 547553280 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042128385155466397, + "loss": 1.4953, + "theoretical_loss": 3.876980420033136, + "tokens_seen": 547618816 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004212738214643932, + "loss": 1.4492, + "theoretical_loss": 3.8769312300017167, + "tokens_seen": 547684352 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042126379137412233, + "loss": 1.486, + "theoretical_loss": 3.876882047503912, + "tokens_seen": 547749888 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042125376128385156, + "loss": 1.4609, + "theoretical_loss": 3.8768328725376673, + "tokens_seen": 547815424 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042124373119358075, + "loss": 1.3481, + "theoretical_loss": 3.8767837051009284, + "tokens_seen": 547880960 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004212337011033099, + "loss": 1.3516, + "theoretical_loss": 3.876734545191642, + "tokens_seen": 547946496 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004212236710130391, + "loss": 1.4596, + "theoretical_loss": 3.876685392807755, + "tokens_seen": 548012032 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004212136409227683, + "loss": 1.3525, + "theoretical_loss": 3.876636247947216, + "tokens_seen": 548077568 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042120361083249747, + "loss": 1.5671, + "theoretical_loss": 3.8765871106079746, + "tokens_seen": 548143104 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004211935807422267, + "loss": 1.4487, + "theoretical_loss": 3.8765379807879796, + "tokens_seen": 548208640 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042118355065195583, + "loss": 1.3894, + "theoretical_loss": 3.8764888584851827, + "tokens_seen": 548274176 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042117352056168507, + "loss": 1.4206, + "theoretical_loss": 3.876439743697535, + "tokens_seen": 548339712 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004211634904714143, + "loss": 1.6228, + "theoretical_loss": 3.8763906364229888, + "tokens_seen": 548405248 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042115346038114343, + "loss": 1.4821, + "theoretical_loss": 3.8763415366594964, + "tokens_seen": 548470784 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042114343029087266, + "loss": 1.4456, + "theoretical_loss": 3.8762924444050135, + "tokens_seen": 548536320 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004211334002006018, + "loss": 1.4463, + "theoretical_loss": 3.8762433596574932, + "tokens_seen": 548601856 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042112337011033103, + "loss": 1.3741, + "theoretical_loss": 3.8761942824148914, + "tokens_seen": 548667392 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004211133400200602, + "loss": 1.3791, + "theoretical_loss": 3.8761452126751648, + "tokens_seen": 548732928 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004211033099297894, + "loss": 1.3893, + "theoretical_loss": 3.87609615043627, + "tokens_seen": 548798464 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.0010866671800613403, + "objective/train/docs_used": 320989, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.84834623336792, + "objective/train/original_loss": 2.8483455181121826, + "objective/train/theoretical_loss": 3.8760470956961655, + "objective/train/tokens_used": 569324000, + "objective/train/value_avg": -0.005306243896484375, + "objective/train/value_loss": 0.00013385532656684518, + "objective/train/value_max": -0.00038909912109375, + "objective/train/value_min": -0.05792236328125, + "objective/train/value_reward_corr": 0.29129117276942723, + "objective/train/value_std": 0.004261016845703125, + "objective/train/weight_avg": 0.99492347240448, + "objective/train/weighted_lm_loss": 2.8269236087799072, + "objective/train/weights_max": 1.7846524715423584, + "objective/train/weights_min": 0.2655084431171417, + "theoretical_loss": 3.8760470956961655, + "tokens_seen": 548864000 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042109327983951857, + "loss": 1.4658, + "theoretical_loss": 3.8760470956961655, + "tokens_seen": 548864000 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042108324974924775, + "loss": 1.4006, + "theoretical_loss": 3.8759980484528094, + "tokens_seen": 548929536 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042107321965897693, + "loss": 1.3735, + "theoretical_loss": 3.8759490087041613, + "tokens_seen": 548995072 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042106318956870617, + "loss": 1.4441, + "theoretical_loss": 3.875899976448182, + "tokens_seen": 549060608 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004210531594784353, + "loss": 1.4457, + "theoretical_loss": 3.8758509516828314, + "tokens_seen": 549126144 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042104312938816453, + "loss": 1.4746, + "theoretical_loss": 3.8758019344060726, + "tokens_seen": 549191680 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004210330992978937, + "loss": 1.4031, + "theoretical_loss": 3.8757529246158677, + "tokens_seen": 549257216 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004210230692076229, + "loss": 1.5383, + "theoretical_loss": 3.875703922310181, + "tokens_seen": 549322752 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004210130391173521, + "loss": 1.4536, + "theoretical_loss": 3.875654927486975, + "tokens_seen": 549388288 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042100300902708125, + "loss": 1.3744, + "theoretical_loss": 3.875605940144217, + "tokens_seen": 549453824 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042099297893681044, + "loss": 1.4127, + "theoretical_loss": 3.875556960279871, + "tokens_seen": 549519360 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042098294884653967, + "loss": 1.2785, + "theoretical_loss": 3.875507987891905, + "tokens_seen": 549584896 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004209729187562688, + "loss": 1.442, + "theoretical_loss": 3.8754590229782853, + "tokens_seen": 549650432 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042096288866599803, + "loss": 1.3361, + "theoretical_loss": 3.8754100655369808, + "tokens_seen": 549715968 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042095285857572716, + "loss": 1.4406, + "theoretical_loss": 3.875361115565961, + "tokens_seen": 549781504 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004209428284854564, + "loss": 1.2832, + "theoretical_loss": 3.875312173063195, + "tokens_seen": 549847040 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004209327983951856, + "loss": 1.4596, + "theoretical_loss": 3.875263238026654, + "tokens_seen": 549912576 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042092276830491476, + "loss": 1.3975, + "theoretical_loss": 3.8752143104543086, + "tokens_seen": 549978112 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042091273821464394, + "loss": 1.4588, + "theoretical_loss": 3.875165390344132, + "tokens_seen": 550043648 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004209027081243731, + "loss": 1.4961, + "theoretical_loss": 3.8751164776940965, + "tokens_seen": 550109184 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004208926780341023, + "loss": 1.3795, + "theoretical_loss": 3.875067572502177, + "tokens_seen": 550174720 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042088264794383154, + "loss": 1.4394, + "theoretical_loss": 3.875018674766346, + "tokens_seen": 550240256 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042087261785356066, + "loss": 1.4223, + "theoretical_loss": 3.874969784484581, + "tokens_seen": 550305792 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004208625877632899, + "loss": 1.4536, + "theoretical_loss": 3.874920901654857, + "tokens_seen": 550371328 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004208525576730191, + "loss": 1.4243, + "theoretical_loss": 3.8748720262751517, + "tokens_seen": 550436864 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.0016120861982926726, + "objective/train/docs_used": 322374, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6729934215545654, + "objective/train/original_loss": 2.6729936599731445, + "objective/train/theoretical_loss": 3.8748231583434425, + "objective/train/tokens_used": 570962400, + "objective/train/value_avg": -0.005035400390625, + "objective/train/value_loss": 7.337633724091575e-05, + "objective/train/value_max": -0.00044083595275878906, + "objective/train/value_min": -0.08209228515625, + "objective/train/value_reward_corr": 0.5092933986153994, + "objective/train/value_std": 0.004680633544921875, + "objective/train/weight_avg": 0.9872429966926575, + "objective/train/weighted_lm_loss": 2.6535942554473877, + "objective/train/weights_max": 1.730299949645996, + "objective/train/weights_min": 0.3253864347934723, + "theoretical_loss": 3.8748231583434425, + "tokens_seen": 550502400 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042084252758274826, + "loss": 1.4657, + "theoretical_loss": 3.8748231583434425, + "tokens_seen": 550502400 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042083249749247744, + "loss": 1.4984, + "theoretical_loss": 3.8747742978577078, + "tokens_seen": 550567936 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004208224674022066, + "loss": 1.4995, + "theoretical_loss": 3.874725444815927, + "tokens_seen": 550633472 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004208124373119358, + "loss": 1.2981, + "theoretical_loss": 3.8746765992160808, + "tokens_seen": 550699008 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042080240722166504, + "loss": 1.3473, + "theoretical_loss": 3.874627761056149, + "tokens_seen": 550764544 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042079237713139417, + "loss": 1.4481, + "theoretical_loss": 3.874578930334115, + "tokens_seen": 550830080 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004207823470411234, + "loss": 1.5565, + "theoretical_loss": 3.8745301070479594, + "tokens_seen": 550895616 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042077231695085253, + "loss": 1.5748, + "theoretical_loss": 3.874481291195667, + "tokens_seen": 550961152 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042076228686058176, + "loss": 1.5114, + "theoretical_loss": 3.8744324827752203, + "tokens_seen": 551026688 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042075225677031095, + "loss": 1.529, + "theoretical_loss": 3.8743836817846065, + "tokens_seen": 551092224 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004207422266800401, + "loss": 1.3004, + "theoretical_loss": 3.8743348882218083, + "tokens_seen": 551157760 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004207321965897693, + "loss": 1.4477, + "theoretical_loss": 3.874286102084815, + "tokens_seen": 551223296 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004207221664994985, + "loss": 1.4462, + "theoretical_loss": 3.8742373233716116, + "tokens_seen": 551288832 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042071213640922767, + "loss": 1.3555, + "theoretical_loss": 3.874188552080187, + "tokens_seen": 551354368 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004207021063189569, + "loss": 1.5285, + "theoretical_loss": 3.8741397882085304, + "tokens_seen": 551419904 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042069207622868603, + "loss": 1.4474, + "theoretical_loss": 3.874091031754631, + "tokens_seen": 551485440 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042068204613841527, + "loss": 1.4117, + "theoretical_loss": 3.8740422827164784, + "tokens_seen": 551550976 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042067201604814445, + "loss": 1.5051, + "theoretical_loss": 3.873993541092065, + "tokens_seen": 551616512 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042066198595787363, + "loss": 1.3426, + "theoretical_loss": 3.873944806879382, + "tokens_seen": 551682048 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004206519558676028, + "loss": 1.4136, + "theoretical_loss": 3.873896080076422, + "tokens_seen": 551747584 + }, + { + "epoch": 0.17, + "learning_rate": 0.000420641925777332, + "loss": 1.4871, + "theoretical_loss": 3.8738473606811787, + "tokens_seen": 551813120 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004206318956870612, + "loss": 1.3863, + "theoretical_loss": 3.873798648691647, + "tokens_seen": 551878656 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004206218655967904, + "loss": 1.4364, + "theoretical_loss": 3.8737499441058207, + "tokens_seen": 551944192 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042061183550651954, + "loss": 1.4787, + "theoretical_loss": 3.8737012469216965, + "tokens_seen": 552009728 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042060180541624877, + "loss": 1.399, + "theoretical_loss": 3.8736525571372704, + "tokens_seen": 552075264 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.0006489845109172165, + "objective/train/docs_used": 322945, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0486667156219482, + "objective/train/original_loss": 3.0486669540405273, + "objective/train/theoretical_loss": 3.8736038747505406, + "objective/train/tokens_used": 572600800, + "objective/train/value_avg": -0.00473785400390625, + "objective/train/value_loss": 4.944832471664995e-05, + "objective/train/value_max": -0.0005254745483398438, + "objective/train/value_min": -0.06756591796875, + "objective/train/value_reward_corr": 0.4606175053481691, + "objective/train/value_std": 0.00394439697265625, + "objective/train/weight_avg": 0.9958677887916565, + "objective/train/weighted_lm_loss": 3.04479718208313, + "objective/train/weights_max": 1.747678518295288, + "objective/train/weights_min": 0.5367447733879089, + "theoretical_loss": 3.8736038747505406, + "tokens_seen": 552140800 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004205917753259779, + "loss": 1.4381, + "theoretical_loss": 3.8736038747505406, + "tokens_seen": 552140800 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042058174523570713, + "loss": 1.3872, + "theoretical_loss": 3.8735551997595046, + "tokens_seen": 552206336 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004205717151454363, + "loss": 1.4076, + "theoretical_loss": 3.8735065321621613, + "tokens_seen": 552271872 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004205616850551655, + "loss": 1.391, + "theoretical_loss": 3.8734578719565107, + "tokens_seen": 552337408 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004205516549648947, + "loss": 1.3572, + "theoretical_loss": 3.873409219140554, + "tokens_seen": 552402944 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004205416248746239, + "loss": 1.4423, + "theoretical_loss": 3.873360573712291, + "tokens_seen": 552468480 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042053159478435304, + "loss": 1.401, + "theoretical_loss": 3.873311935669725, + "tokens_seen": 552534016 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004205215646940823, + "loss": 1.4644, + "theoretical_loss": 3.873263305010858, + "tokens_seen": 552599552 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004205115346038114, + "loss": 1.3782, + "theoretical_loss": 3.8732146817336934, + "tokens_seen": 552665088 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042050150451354064, + "loss": 1.248, + "theoretical_loss": 3.873166065836237, + "tokens_seen": 552730624 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004204914744232698, + "loss": 1.4145, + "theoretical_loss": 3.873117457316493, + "tokens_seen": 552796160 + }, + { + "epoch": 0.17, + "learning_rate": 0.000420481444332999, + "loss": 1.4391, + "theoretical_loss": 3.8730688561724667, + "tokens_seen": 552861696 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004204714142427282, + "loss": 1.5536, + "theoretical_loss": 3.873020262402166, + "tokens_seen": 552927232 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042046138415245736, + "loss": 1.5718, + "theoretical_loss": 3.872971676003598, + "tokens_seen": 552992768 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042045135406218654, + "loss": 1.3747, + "theoretical_loss": 3.872923096974771, + "tokens_seen": 553058304 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004204413239719158, + "loss": 1.5056, + "theoretical_loss": 3.872874525313694, + "tokens_seen": 553123840 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004204312938816449, + "loss": 1.4369, + "theoretical_loss": 3.872825961018376, + "tokens_seen": 553189376 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042042126379137414, + "loss": 1.3868, + "theoretical_loss": 3.872777404086829, + "tokens_seen": 553254912 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004204112337011033, + "loss": 1.4094, + "theoretical_loss": 3.8727288545170637, + "tokens_seen": 553320448 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004204012036108325, + "loss": 1.4534, + "theoretical_loss": 3.872680312307092, + "tokens_seen": 553385984 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042039117352056174, + "loss": 1.4096, + "theoretical_loss": 3.8726317774549273, + "tokens_seen": 553451520 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042038114343029086, + "loss": 1.52, + "theoretical_loss": 3.8725832499585824, + "tokens_seen": 553517056 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004203711133400201, + "loss": 1.3898, + "theoretical_loss": 3.872534729816073, + "tokens_seen": 553582592 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004203610832497493, + "loss": 1.5966, + "theoretical_loss": 3.872486217025413, + "tokens_seen": 553648128 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042035105315947846, + "loss": 1.4591, + "theoretical_loss": 3.8724377115846194, + "tokens_seen": 553713664 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": 0.00028381278389133513, + "objective/train/docs_used": 324208, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9770047664642334, + "objective/train/original_loss": 2.9770047664642334, + "objective/train/theoretical_loss": 3.872389213491709, + "objective/train/tokens_used": 574239200, + "objective/train/value_avg": -0.004970550537109375, + "objective/train/value_loss": 4.092148446943611e-05, + "objective/train/value_max": -0.0005955696105957031, + "objective/train/value_min": -0.033843994140625, + "objective/train/value_reward_corr": 0.32237790162077457, + "objective/train/value_std": 0.003498077392578125, + "objective/train/weight_avg": 1.0047663450241089, + "objective/train/weighted_lm_loss": 2.995063304901123, + "objective/train/weights_max": 1.323330044746399, + "objective/train/weights_min": 0.37870821356773376, + "theoretical_loss": 3.872389213491709, + "tokens_seen": 553779200 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042034102306920764, + "loss": 1.3508, + "theoretical_loss": 3.872389213491709, + "tokens_seen": 553779200 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004203309929789368, + "loss": 1.4109, + "theoretical_loss": 3.872340722744698, + "tokens_seen": 553844736 + }, + { + "epoch": 0.17, + "learning_rate": 0.000420320962888666, + "loss": 1.5406, + "theoretical_loss": 3.872292239341606, + "tokens_seen": 553910272 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042031093279839524, + "loss": 1.4923, + "theoretical_loss": 3.872243763280452, + "tokens_seen": 553975808 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042030090270812437, + "loss": 1.3755, + "theoretical_loss": 3.872195294559255, + "tokens_seen": 554041344 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004202908726178536, + "loss": 1.5283, + "theoretical_loss": 3.8721468331760365, + "tokens_seen": 554106880 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042028084252758273, + "loss": 1.4299, + "theoretical_loss": 3.872098379128817, + "tokens_seen": 554172416 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042027081243731196, + "loss": 1.3068, + "theoretical_loss": 3.87204993241562, + "tokens_seen": 554237952 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042026078234704115, + "loss": 1.4403, + "theoretical_loss": 3.8720014930344666, + "tokens_seen": 554303488 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004202507522567703, + "loss": 1.3994, + "theoretical_loss": 3.8719530609833814, + "tokens_seen": 554369024 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004202407221664995, + "loss": 1.3271, + "theoretical_loss": 3.8719046362603895, + "tokens_seen": 554434560 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004202306920762287, + "loss": 1.4685, + "theoretical_loss": 3.871856218863515, + "tokens_seen": 554500096 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042022066198595787, + "loss": 1.5043, + "theoretical_loss": 3.8718078087907846, + "tokens_seen": 554565632 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004202106318956871, + "loss": 1.455, + "theoretical_loss": 3.8717594060402245, + "tokens_seen": 554631168 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042020060180541623, + "loss": 1.4737, + "theoretical_loss": 3.8717110106098627, + "tokens_seen": 554696704 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042019057171514547, + "loss": 1.464, + "theoretical_loss": 3.871662622497727, + "tokens_seen": 554762240 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042018054162487465, + "loss": 1.5453, + "theoretical_loss": 3.871614241701846, + "tokens_seen": 554827776 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042017051153460383, + "loss": 1.4236, + "theoretical_loss": 3.8715658682202507, + "tokens_seen": 554893312 + }, + { + "epoch": 0.17, + "learning_rate": 0.000420160481444333, + "loss": 1.3308, + "theoretical_loss": 3.8715175020509713, + "tokens_seen": 554958848 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004201504513540622, + "loss": 1.4776, + "theoretical_loss": 3.8714691431920385, + "tokens_seen": 555024384 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004201404212637914, + "loss": 1.5149, + "theoretical_loss": 3.871420791641485, + "tokens_seen": 555089920 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004201303911735206, + "loss": 1.4608, + "theoretical_loss": 3.871372447397343, + "tokens_seen": 555155456 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042012036108324974, + "loss": 1.4158, + "theoretical_loss": 3.871324110457647, + "tokens_seen": 555220992 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042011033099297897, + "loss": 1.3343, + "theoretical_loss": 3.8712757808204303, + "tokens_seen": 555286528 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004201003009027081, + "loss": 1.2674, + "theoretical_loss": 3.8712274584837294, + "tokens_seen": 555352064 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -9.176450112136081e-05, + "objective/train/docs_used": 324930, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5180530548095703, + "objective/train/original_loss": 2.518052577972412, + "objective/train/theoretical_loss": 3.8711791434455787, + "objective/train/tokens_used": 575877600, + "objective/train/value_avg": -0.005298614501953125, + "objective/train/value_loss": 6.33446907158941e-05, + "objective/train/value_max": -0.00046372413635253906, + "objective/train/value_min": -0.056243896484375, + "objective/train/value_reward_corr": 0.35896752839000956, + "objective/train/value_std": 0.004688262939453125, + "objective/train/weight_avg": 1.0020614862442017, + "objective/train/weighted_lm_loss": 2.5130300521850586, + "objective/train/weights_max": 1.7262638807296753, + "objective/train/weights_min": 0.2512786090373993, + "theoretical_loss": 3.8711791434455787, + "tokens_seen": 555417600 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042009027081243733, + "loss": 1.3606, + "theoretical_loss": 3.8711791434455787, + "tokens_seen": 555417600 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004200802407221665, + "loss": 1.3629, + "theoretical_loss": 3.8711308357040157, + "tokens_seen": 555483136 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004200702106318957, + "loss": 1.3282, + "theoretical_loss": 3.8710825352570772, + "tokens_seen": 555548672 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004200601805416249, + "loss": 1.4148, + "theoretical_loss": 3.8710342421028026, + "tokens_seen": 555614208 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004200501504513541, + "loss": 1.3948, + "theoretical_loss": 3.8709859562392293, + "tokens_seen": 555679744 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042004012036108324, + "loss": 1.4973, + "theoretical_loss": 3.870937677664398, + "tokens_seen": 555745280 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004200300902708125, + "loss": 1.3968, + "theoretical_loss": 3.8708894063763486, + "tokens_seen": 555810816 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004200200601805416, + "loss": 1.5265, + "theoretical_loss": 3.8708411423731226, + "tokens_seen": 555876352 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042001003009027084, + "loss": 1.3469, + "theoretical_loss": 3.870792885652762, + "tokens_seen": 555941888 + }, + { + "epoch": 0.17, + "learning_rate": 0.00042, + "loss": 1.3426, + "theoretical_loss": 3.8707446362133098, + "tokens_seen": 556007424 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004199899699097292, + "loss": 1.4988, + "theoretical_loss": 3.870696394052808, + "tokens_seen": 556072960 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004199799398194584, + "loss": 1.5123, + "theoretical_loss": 3.8706481591693027, + "tokens_seen": 556138496 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041996990972918756, + "loss": 1.3008, + "theoretical_loss": 3.8705999315608377, + "tokens_seen": 556204032 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041995987963891674, + "loss": 1.2454, + "theoretical_loss": 3.8705517112254593, + "tokens_seen": 556269568 + }, + { + "epoch": 0.17, + "learning_rate": 0.000419949849548646, + "loss": 1.473, + "theoretical_loss": 3.870503498161214, + "tokens_seen": 556335104 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004199398194583751, + "loss": 1.486, + "theoretical_loss": 3.870455292366148, + "tokens_seen": 556400640 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041992978936810434, + "loss": 1.563, + "theoretical_loss": 3.8704070938383106, + "tokens_seen": 556466176 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041991975927783347, + "loss": 1.5599, + "theoretical_loss": 3.87035890257575, + "tokens_seen": 556531712 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004199097291875627, + "loss": 1.3372, + "theoretical_loss": 3.870310718576516, + "tokens_seen": 556597248 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004198996990972919, + "loss": 1.5843, + "theoretical_loss": 3.8702625418386587, + "tokens_seen": 556662784 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041988966900702106, + "loss": 1.465, + "theoretical_loss": 3.8702143723602287, + "tokens_seen": 556728320 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041987963891675025, + "loss": 1.3143, + "theoretical_loss": 3.870166210139278, + "tokens_seen": 556793856 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004198696088264795, + "loss": 1.3574, + "theoretical_loss": 3.87011805517386, + "tokens_seen": 556859392 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004198595787362086, + "loss": 1.3448, + "theoretical_loss": 3.8700699074620264, + "tokens_seen": 556924928 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041984954864593784, + "loss": 1.4818, + "theoretical_loss": 3.870021767001832, + "tokens_seen": 556990464 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": 3.792497227550484e-05, + "objective/train/docs_used": 326252, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.905352830886841, + "objective/train/original_loss": 2.9053525924682617, + "objective/train/theoretical_loss": 3.869973633791332, + "objective/train/tokens_used": 577516000, + "objective/train/value_avg": -0.00499725341796875, + "objective/train/value_loss": 4.222976713208482e-05, + "objective/train/value_max": -0.00038743019104003906, + "objective/train/value_min": -0.06500244140625, + "objective/train/value_reward_corr": 0.3195320414675105, + "objective/train/value_std": 0.00389862060546875, + "objective/train/weight_avg": 1.0023061037063599, + "objective/train/weighted_lm_loss": 2.9244327545166016, + "objective/train/weights_max": 1.8683499097824097, + "objective/train/weights_min": 0.19603650271892548, + "theoretical_loss": 3.869973633791332, + "tokens_seen": 557056000 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041983951855566697, + "loss": 1.4077, + "theoretical_loss": 3.869973633791332, + "tokens_seen": 557056000 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004198294884653962, + "loss": 1.4179, + "theoretical_loss": 3.869925507828582, + "tokens_seen": 557121536 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004198194583751254, + "loss": 1.5231, + "theoretical_loss": 3.8698773891116374, + "tokens_seen": 557187072 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041980942828485457, + "loss": 1.4999, + "theoretical_loss": 3.8698292776385554, + "tokens_seen": 557252608 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041979939819458375, + "loss": 1.4137, + "theoretical_loss": 3.869781173407394, + "tokens_seen": 557318144 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041978936810431293, + "loss": 1.4645, + "theoretical_loss": 3.869733076416212, + "tokens_seen": 557383680 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004197793380140421, + "loss": 1.4442, + "theoretical_loss": 3.869684986663068, + "tokens_seen": 557449216 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041976930792377135, + "loss": 1.3469, + "theoretical_loss": 3.869636904146023, + "tokens_seen": 557514752 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041975927783350047, + "loss": 1.2557, + "theoretical_loss": 3.8695888288631366, + "tokens_seen": 557580288 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004197492477432297, + "loss": 1.395, + "theoretical_loss": 3.8695407608124714, + "tokens_seen": 557645824 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041973921765295883, + "loss": 1.4363, + "theoretical_loss": 3.869492699992089, + "tokens_seen": 557711360 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041972918756268807, + "loss": 1.4536, + "theoretical_loss": 3.8694446464000523, + "tokens_seen": 557776896 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041971915747241725, + "loss": 1.4496, + "theoretical_loss": 3.8693966000344253, + "tokens_seen": 557842432 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041970912738214643, + "loss": 1.4601, + "theoretical_loss": 3.8693485608932727, + "tokens_seen": 557907968 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004196990972918756, + "loss": 1.2763, + "theoretical_loss": 3.8693005289746596, + "tokens_seen": 557973504 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041968906720160485, + "loss": 1.2852, + "theoretical_loss": 3.869252504276652, + "tokens_seen": 558039040 + }, + { + "epoch": 0.17, + "learning_rate": 0.000419679037111334, + "loss": 1.4898, + "theoretical_loss": 3.869204486797316, + "tokens_seen": 558104576 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004196690070210632, + "loss": 1.3628, + "theoretical_loss": 3.8691564765347204, + "tokens_seen": 558170112 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004196589769307924, + "loss": 1.4738, + "theoretical_loss": 3.869108473486932, + "tokens_seen": 558235648 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004196489468405216, + "loss": 1.3015, + "theoretical_loss": 3.869060477652021, + "tokens_seen": 558301184 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004196389167502508, + "loss": 1.4888, + "theoretical_loss": 3.8690124890280564, + "tokens_seen": 558366720 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041962888665997994, + "loss": 1.3898, + "theoretical_loss": 3.8689645076131085, + "tokens_seen": 558432256 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041961885656970917, + "loss": 1.4261, + "theoretical_loss": 3.8689165334052493, + "tokens_seen": 558497792 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004196088264794383, + "loss": 1.3672, + "theoretical_loss": 3.8688685664025497, + "tokens_seen": 558563328 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041959879638916753, + "loss": 1.3102, + "theoretical_loss": 3.8688206066030837, + "tokens_seen": 558628864 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.0009410632774233818, + "objective/train/docs_used": 326886, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5288729667663574, + "objective/train/original_loss": 2.5288729667663574, + "objective/train/theoretical_loss": 3.8687726540049234, + "objective/train/tokens_used": 579154400, + "objective/train/value_avg": -0.005527496337890625, + "objective/train/value_loss": 0.0001022286742227152, + "objective/train/value_max": -0.0007042884826660156, + "objective/train/value_min": -0.042083740234375, + "objective/train/value_reward_corr": 0.3772882134053812, + "objective/train/value_std": 0.00439453125, + "objective/train/weight_avg": 0.9949741959571838, + "objective/train/weighted_lm_loss": 2.5262866020202637, + "objective/train/weights_max": 1.4036139249801636, + "objective/train/weights_min": 0.14057083427906036, + "theoretical_loss": 3.8687726540049234, + "tokens_seen": 558694400 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004195887662988967, + "loss": 1.4041, + "theoretical_loss": 3.8687726540049234, + "tokens_seen": 558694400 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004195787362086259, + "loss": 1.344, + "theoretical_loss": 3.868724708606144, + "tokens_seen": 558759936 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004195687061183551, + "loss": 1.4445, + "theoretical_loss": 3.8686767704048197, + "tokens_seen": 558825472 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004195586760280843, + "loss": 1.4713, + "theoretical_loss": 3.868628839399026, + "tokens_seen": 558891008 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041954864593781344, + "loss": 1.3845, + "theoretical_loss": 3.86858091558684, + "tokens_seen": 558956544 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004195386158475427, + "loss": 1.4239, + "theoretical_loss": 3.8685329989663386, + "tokens_seen": 559022080 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004195285857572718, + "loss": 1.4789, + "theoretical_loss": 3.868485089535599, + "tokens_seen": 559087616 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041951855566700104, + "loss": 1.5102, + "theoretical_loss": 3.8684371872927015, + "tokens_seen": 559153152 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004195085255767302, + "loss": 1.4894, + "theoretical_loss": 3.8683892922357233, + "tokens_seen": 559218688 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004194984954864594, + "loss": 1.3305, + "theoretical_loss": 3.868341404362746, + "tokens_seen": 559284224 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004194884653961886, + "loss": 1.4622, + "theoretical_loss": 3.8682935236718494, + "tokens_seen": 559349760 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041947843530591776, + "loss": 1.4232, + "theoretical_loss": 3.868245650161116, + "tokens_seen": 559415296 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041946840521564694, + "loss": 1.4213, + "theoretical_loss": 3.8681977838286272, + "tokens_seen": 559480832 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004194583751253762, + "loss": 1.3927, + "theoretical_loss": 3.868149924672467, + "tokens_seen": 559546368 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004194483450351053, + "loss": 1.4711, + "theoretical_loss": 3.8681020726907187, + "tokens_seen": 559611904 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041943831494483454, + "loss": 1.3915, + "theoretical_loss": 3.868054227881466, + "tokens_seen": 559677440 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041942828485456367, + "loss": 1.4506, + "theoretical_loss": 3.8680063902427957, + "tokens_seen": 559742976 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004194182547642929, + "loss": 1.4624, + "theoretical_loss": 3.8679585597727923, + "tokens_seen": 559808512 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004194082246740221, + "loss": 1.3146, + "theoretical_loss": 3.8679107364695433, + "tokens_seen": 559874048 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041939819458375126, + "loss": 1.4474, + "theoretical_loss": 3.8678629203311368, + "tokens_seen": 559939584 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041938816449348045, + "loss": 1.4544, + "theoretical_loss": 3.867815111355659, + "tokens_seen": 560005120 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004193781344032097, + "loss": 1.4265, + "theoretical_loss": 3.8677673095412013, + "tokens_seen": 560070656 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004193681043129388, + "loss": 1.4135, + "theoretical_loss": 3.867719514885851, + "tokens_seen": 560136192 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041935807422266804, + "loss": 1.374, + "theoretical_loss": 3.8676717273877, + "tokens_seen": 560201728 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041934804413239717, + "loss": 1.4857, + "theoretical_loss": 3.867623947044839, + "tokens_seen": 560267264 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.0008310521952807903, + "objective/train/docs_used": 327930, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.557345151901245, + "objective/train/original_loss": 2.557345151901245, + "objective/train/theoretical_loss": 3.8675761738553596, + "objective/train/tokens_used": 580792800, + "objective/train/value_avg": -0.0044403076171875, + "objective/train/value_loss": 4.478639311855659e-05, + "objective/train/value_max": -0.0003554821014404297, + "objective/train/value_min": -0.04217529296875, + "objective/train/value_reward_corr": 0.3910883345443772, + "objective/train/value_std": 0.0036029815673828125, + "objective/train/weight_avg": 0.9938300251960754, + "objective/train/weighted_lm_loss": 2.5387563705444336, + "objective/train/weights_max": 1.4462262392044067, + "objective/train/weights_min": 0.6084666848182678, + "theoretical_loss": 3.8675761738553596, + "tokens_seen": 560332800 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004193380140421264, + "loss": 1.3762, + "theoretical_loss": 3.8675761738553596, + "tokens_seen": 560332800 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004193279839518556, + "loss": 1.4212, + "theoretical_loss": 3.8675284078173546, + "tokens_seen": 560398336 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041931795386158477, + "loss": 1.4218, + "theoretical_loss": 3.8674806489289173, + "tokens_seen": 560463872 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041930792377131395, + "loss": 1.3932, + "theoretical_loss": 3.8674328971881415, + "tokens_seen": 560529408 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041929789368104313, + "loss": 1.4687, + "theoretical_loss": 3.867385152593122, + "tokens_seen": 560594944 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004192878635907723, + "loss": 1.4417, + "theoretical_loss": 3.8673374151419546, + "tokens_seen": 560660480 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041927783350050155, + "loss": 1.4744, + "theoretical_loss": 3.867289684832736, + "tokens_seen": 560726016 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041926780341023067, + "loss": 1.3187, + "theoretical_loss": 3.867241961663562, + "tokens_seen": 560791552 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004192577733199599, + "loss": 1.4151, + "theoretical_loss": 3.867194245632531, + "tokens_seen": 560857088 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041924774322968904, + "loss": 1.4826, + "theoretical_loss": 3.867146536737741, + "tokens_seen": 560922624 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041923771313941827, + "loss": 1.4916, + "theoretical_loss": 3.8670988349772912, + "tokens_seen": 560988160 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041922768304914745, + "loss": 1.5858, + "theoretical_loss": 3.8670511403492824, + "tokens_seen": 561053696 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041921765295887663, + "loss": 1.3867, + "theoretical_loss": 3.8670034528518142, + "tokens_seen": 561119232 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004192076228686058, + "loss": 1.2996, + "theoretical_loss": 3.866955772482988, + "tokens_seen": 561184768 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041919759277833505, + "loss": 1.401, + "theoretical_loss": 3.866908099240906, + "tokens_seen": 561250304 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004191875626880642, + "loss": 1.4459, + "theoretical_loss": 3.866860433123671, + "tokens_seen": 561315840 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004191775325977934, + "loss": 1.3574, + "theoretical_loss": 3.8668127741293867, + "tokens_seen": 561381376 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041916750250752254, + "loss": 1.3291, + "theoretical_loss": 3.866765122256158, + "tokens_seen": 561446912 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004191574724172518, + "loss": 1.4939, + "theoretical_loss": 3.866717477502088, + "tokens_seen": 561512448 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041914744232698095, + "loss": 1.4992, + "theoretical_loss": 3.866669839865284, + "tokens_seen": 561577984 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041913741223671014, + "loss": 1.5028, + "theoretical_loss": 3.866622209343851, + "tokens_seen": 561643520 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004191273821464393, + "loss": 1.43, + "theoretical_loss": 3.866574585935898, + "tokens_seen": 561709056 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004191173520561685, + "loss": 1.4464, + "theoretical_loss": 3.8665269696395312, + "tokens_seen": 561774592 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004191073219658977, + "loss": 1.5075, + "theoretical_loss": 3.8664793604528604, + "tokens_seen": 561840128 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004190972918756269, + "loss": 1.5104, + "theoretical_loss": 3.8664317583739938, + "tokens_seen": 561905664 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.0004535694606602192, + "objective/train/docs_used": 328746, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7742114067077637, + "objective/train/original_loss": 2.7742114067077637, + "objective/train/theoretical_loss": 3.866384163401042, + "objective/train/tokens_used": 582431200, + "objective/train/value_avg": -0.004947662353515625, + "objective/train/value_loss": 4.937005360261537e-05, + "objective/train/value_max": -0.00048041343688964844, + "objective/train/value_min": -0.06256103515625, + "objective/train/value_reward_corr": 0.33958593003957954, + "objective/train/value_std": 0.003620147705078125, + "objective/train/weight_avg": 0.9977539777755737, + "objective/train/weighted_lm_loss": 2.764643430709839, + "objective/train/weights_max": 1.450614333152771, + "objective/train/weights_min": 0.29432570934295654, + "theoretical_loss": 3.866384163401042, + "tokens_seen": 561971200 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041908726178535604, + "loss": 1.4401, + "theoretical_loss": 3.866384163401042, + "tokens_seen": 561971200 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004190772316950853, + "loss": 1.4014, + "theoretical_loss": 3.8663365755321157, + "tokens_seen": 562036736 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004190672016048144, + "loss": 1.4804, + "theoretical_loss": 3.866288994765327, + "tokens_seen": 562102272 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041905717151454364, + "loss": 1.297, + "theoretical_loss": 3.866241421098787, + "tokens_seen": 562167808 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004190471414242728, + "loss": 1.3957, + "theoretical_loss": 3.8661938545306094, + "tokens_seen": 562233344 + }, + { + "epoch": 0.17, + "learning_rate": 0.000419037111334002, + "loss": 1.4601, + "theoretical_loss": 3.8661462950589076, + "tokens_seen": 562298880 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004190270812437312, + "loss": 1.4622, + "theoretical_loss": 3.8660987426817957, + "tokens_seen": 562364416 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004190170511534604, + "loss": 1.5536, + "theoretical_loss": 3.866051197397389, + "tokens_seen": 562429952 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041900702106318954, + "loss": 1.3815, + "theoretical_loss": 3.8660036592038036, + "tokens_seen": 562495488 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004189969909729188, + "loss": 1.4027, + "theoretical_loss": 3.865956128099155, + "tokens_seen": 562561024 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004189869608826479, + "loss": 1.3964, + "theoretical_loss": 3.865908604081562, + "tokens_seen": 562626560 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041897693079237714, + "loss": 1.3311, + "theoretical_loss": 3.8658610871491415, + "tokens_seen": 562692096 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004189669007021063, + "loss": 1.6676, + "theoretical_loss": 3.8658135773000124, + "tokens_seen": 562757632 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004189568706118355, + "loss": 1.4184, + "theoretical_loss": 3.8657660745322944, + "tokens_seen": 562823168 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004189468405215647, + "loss": 1.449, + "theoretical_loss": 3.865718578844107, + "tokens_seen": 562888704 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041893681043129387, + "loss": 1.5675, + "theoretical_loss": 3.8656710902335716, + "tokens_seen": 562954240 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041892678034102305, + "loss": 1.4239, + "theoretical_loss": 3.8656236086988094, + "tokens_seen": 563019776 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004189167502507523, + "loss": 1.4909, + "theoretical_loss": 3.865576134237943, + "tokens_seen": 563085312 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041890672016048146, + "loss": 1.3937, + "theoretical_loss": 3.865528666849095, + "tokens_seen": 563150848 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041889669007021065, + "loss": 1.3403, + "theoretical_loss": 3.8654812065303896, + "tokens_seen": 563216384 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004188866599799399, + "loss": 1.3919, + "theoretical_loss": 3.8654337532799508, + "tokens_seen": 563281920 + }, + { + "epoch": 0.17, + "learning_rate": 0.000418876629889669, + "loss": 1.3896, + "theoretical_loss": 3.865386307095904, + "tokens_seen": 563347456 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041886659979939824, + "loss": 1.4828, + "theoretical_loss": 3.8653388679763747, + "tokens_seen": 563412992 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041885656970912737, + "loss": 1.3252, + "theoretical_loss": 3.86529143591949, + "tokens_seen": 563478528 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004188465396188566, + "loss": 1.4742, + "theoretical_loss": 3.8652440109233765, + "tokens_seen": 563544064 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": 0.00042970350477844477, + "objective/train/docs_used": 329425, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.759270429611206, + "objective/train/original_loss": 2.759270668029785, + "objective/train/theoretical_loss": 3.8651965929861625, + "objective/train/tokens_used": 584069600, + "objective/train/value_avg": -0.004791259765625, + "objective/train/value_loss": 4.6083532652119175e-05, + "objective/train/value_max": -0.00033283233642578125, + "objective/train/value_min": -0.06756591796875, + "objective/train/value_reward_corr": 0.3173810472958858, + "objective/train/value_std": 0.0036334991455078125, + "objective/train/weight_avg": 1.0064619779586792, + "objective/train/weighted_lm_loss": 2.775268316268921, + "objective/train/weights_max": 1.6221644878387451, + "objective/train/weights_min": 0.44679397344589233, + "theoretical_loss": 3.8651965929861625, + "tokens_seen": 563609600 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004188365095285858, + "loss": 1.4532, + "theoretical_loss": 3.8651965929861625, + "tokens_seen": 563609600 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041882647943831497, + "loss": 1.3598, + "theoretical_loss": 3.865149182105977, + "tokens_seen": 563675136 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041881644934804415, + "loss": 1.3577, + "theoretical_loss": 3.865101778280949, + "tokens_seen": 563740672 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041880641925777333, + "loss": 1.4459, + "theoretical_loss": 3.8650543815092098, + "tokens_seen": 563806208 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004187963891675025, + "loss": 1.4661, + "theoretical_loss": 3.865006991788888, + "tokens_seen": 563871744 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041878635907723175, + "loss": 1.4922, + "theoretical_loss": 3.864959609118117, + "tokens_seen": 563937280 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004187763289869609, + "loss": 1.5486, + "theoretical_loss": 3.864912233495029, + "tokens_seen": 564002816 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004187662988966901, + "loss": 1.3539, + "theoretical_loss": 3.864864864917756, + "tokens_seen": 564068352 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041875626880641924, + "loss": 1.5322, + "theoretical_loss": 3.8648175033844323, + "tokens_seen": 564133888 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041874623871614847, + "loss": 1.4537, + "theoretical_loss": 3.864770148893192, + "tokens_seen": 564199424 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041873620862587765, + "loss": 1.4465, + "theoretical_loss": 3.864722801442171, + "tokens_seen": 564264960 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041872617853560683, + "loss": 1.564, + "theoretical_loss": 3.864675461029504, + "tokens_seen": 564330496 + }, + { + "epoch": 0.17, + "learning_rate": 0.000418716148445336, + "loss": 1.42, + "theoretical_loss": 3.8646281276533285, + "tokens_seen": 564396032 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041870611835506525, + "loss": 1.5235, + "theoretical_loss": 3.864580801311781, + "tokens_seen": 564461568 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004186960882647944, + "loss": 1.4264, + "theoretical_loss": 3.864533482003, + "tokens_seen": 564527104 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004186860581745236, + "loss": 1.4669, + "theoretical_loss": 3.864486169725124, + "tokens_seen": 564592640 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041867602808425274, + "loss": 1.4608, + "theoretical_loss": 3.8644388644762926, + "tokens_seen": 564658176 + }, + { + "epoch": 0.17, + "learning_rate": 0.000418665997993982, + "loss": 1.3322, + "theoretical_loss": 3.8643915662546453, + "tokens_seen": 564723712 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041865596790371115, + "loss": 1.4042, + "theoretical_loss": 3.8643442750583237, + "tokens_seen": 564789248 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041864593781344034, + "loss": 1.5326, + "theoretical_loss": 3.8642969908854683, + "tokens_seen": 564854784 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004186359077231695, + "loss": 1.564, + "theoretical_loss": 3.864249713734222, + "tokens_seen": 564920320 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004186258776328987, + "loss": 1.5209, + "theoretical_loss": 3.8642024436027276, + "tokens_seen": 564985856 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004186158475426279, + "loss": 1.4136, + "theoretical_loss": 3.8641551804891288, + "tokens_seen": 565051392 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004186058174523571, + "loss": 1.429, + "theoretical_loss": 3.86410792439157, + "tokens_seen": 565116928 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041859578736208624, + "loss": 1.4392, + "theoretical_loss": 3.8640606753081954, + "tokens_seen": 565182464 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.0007120417430996895, + "objective/train/docs_used": 329612, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.216221570968628, + "objective/train/original_loss": 3.216221809387207, + "objective/train/theoretical_loss": 3.864013433237152, + "objective/train/tokens_used": 585708000, + "objective/train/value_avg": -0.006916046142578125, + "objective/train/value_loss": 5.4738091421313584e-05, + "objective/train/value_max": -0.0005860328674316406, + "objective/train/value_min": -0.052703857421875, + "objective/train/value_reward_corr": 0.41493190221847015, + "objective/train/value_std": 0.005115509033203125, + "objective/train/weight_avg": 0.9955516457557678, + "objective/train/weighted_lm_loss": 3.211469888687134, + "objective/train/weights_max": 1.621080994606018, + "objective/train/weights_min": 0.5702959895133972, + "theoretical_loss": 3.864013433237152, + "tokens_seen": 565248000 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004185857572718155, + "loss": 1.5231, + "theoretical_loss": 3.864013433237152, + "tokens_seen": 565248000 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004185757271815446, + "loss": 1.484, + "theoretical_loss": 3.8639661981765854, + "tokens_seen": 565313536 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041856569709127384, + "loss": 1.5585, + "theoretical_loss": 3.863918970124643, + "tokens_seen": 565379072 + }, + { + "epoch": 0.17, + "learning_rate": 0.000418555667001003, + "loss": 1.5306, + "theoretical_loss": 3.8638717490794723, + "tokens_seen": 565444608 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004185456369107322, + "loss": 1.5157, + "theoretical_loss": 3.863824535039223, + "tokens_seen": 565510144 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004185356068204614, + "loss": 1.3978, + "theoretical_loss": 3.863777328002042, + "tokens_seen": 565575680 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004185255767301906, + "loss": 1.6058, + "theoretical_loss": 3.8637301279660825, + "tokens_seen": 565641216 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041851554663991974, + "loss": 1.5348, + "theoretical_loss": 3.8636829349294928, + "tokens_seen": 565706752 + }, + { + "epoch": 0.17, + "learning_rate": 0.000418505516549649, + "loss": 1.4942, + "theoretical_loss": 3.8636357488904247, + "tokens_seen": 565772288 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004184954864593781, + "loss": 1.4286, + "theoretical_loss": 3.863588569847031, + "tokens_seen": 565837824 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041848545636910734, + "loss": 1.5224, + "theoretical_loss": 3.8635413977974635, + "tokens_seen": 565903360 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004184754262788365, + "loss": 1.4936, + "theoretical_loss": 3.8634942327398765, + "tokens_seen": 565968896 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004184653961885657, + "loss": 1.3822, + "theoretical_loss": 3.8634470746724237, + "tokens_seen": 566034432 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004184553660982949, + "loss": 1.5464, + "theoretical_loss": 3.86339992359326, + "tokens_seen": 566099968 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041844533600802407, + "loss": 1.4947, + "theoretical_loss": 3.8633527795005413, + "tokens_seen": 566165504 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041843530591775325, + "loss": 1.3006, + "theoretical_loss": 3.8633056423924232, + "tokens_seen": 566231040 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004184252758274825, + "loss": 1.3795, + "theoretical_loss": 3.8632585122670635, + "tokens_seen": 566296576 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004184152457372116, + "loss": 1.684, + "theoretical_loss": 3.86321138912262, + "tokens_seen": 566362112 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041840521564694085, + "loss": 1.4931, + "theoretical_loss": 3.86316427295725, + "tokens_seen": 566427648 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041839518555666997, + "loss": 1.5441, + "theoretical_loss": 3.8631171637691137, + "tokens_seen": 566493184 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004183851554663992, + "loss": 1.4514, + "theoretical_loss": 3.86307006155637, + "tokens_seen": 566558720 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004183751253761284, + "loss": 1.4293, + "theoretical_loss": 3.86302296631718, + "tokens_seen": 566624256 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041836509528585757, + "loss": 1.4244, + "theoretical_loss": 3.8629758780497037, + "tokens_seen": 566689792 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041835506519558675, + "loss": 1.3968, + "theoretical_loss": 3.8629287967521044, + "tokens_seen": 566755328 + }, + { + "epoch": 0.17, + "learning_rate": 0.000418345035105316, + "loss": 1.5553, + "theoretical_loss": 3.8628817224225447, + "tokens_seen": 566820864 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.0008303835056722164, + "objective/train/docs_used": 331071, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1410727500915527, + "objective/train/original_loss": 3.141073226928711, + "objective/train/theoretical_loss": 3.8628346550591868, + "objective/train/tokens_used": 587346400, + "objective/train/value_avg": -0.00494384765625, + "objective/train/value_loss": 8.217508002417162e-05, + "objective/train/value_max": -0.00036978721618652344, + "objective/train/value_min": -0.042572021484375, + "objective/train/value_reward_corr": 0.3854804737937326, + "objective/train/value_std": 0.0037250518798828125, + "objective/train/weight_avg": 0.9954090118408203, + "objective/train/weighted_lm_loss": 3.1276376247406006, + "objective/train/weights_max": 1.3304688930511475, + "objective/train/weights_min": 0.3107560873031616, + "theoretical_loss": 3.8628346550591868, + "tokens_seen": 566886400 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004183350050150451, + "loss": 1.4268, + "theoretical_loss": 3.8628346550591868, + "tokens_seen": 566886400 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041832497492477435, + "loss": 1.5227, + "theoretical_loss": 3.862787594660195, + "tokens_seen": 566951936 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004183149448345035, + "loss": 1.5614, + "theoretical_loss": 3.862740541223734, + "tokens_seen": 567017472 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004183049147442327, + "loss": 1.4557, + "theoretical_loss": 3.86269349474797, + "tokens_seen": 567083008 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004182948846539619, + "loss": 1.4398, + "theoretical_loss": 3.8626464552310678, + "tokens_seen": 567148544 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004182848545636911, + "loss": 1.4813, + "theoretical_loss": 3.8625994226711944, + "tokens_seen": 567214080 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041827482447342025, + "loss": 1.4022, + "theoretical_loss": 3.8625523970665174, + "tokens_seen": 567279616 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041826479438314944, + "loss": 1.47, + "theoretical_loss": 3.8625053784152055, + "tokens_seen": 567345152 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004182547642928786, + "loss": 1.5669, + "theoretical_loss": 3.8624583667154266, + "tokens_seen": 567410688 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041824473420260785, + "loss": 1.4744, + "theoretical_loss": 3.862411361965351, + "tokens_seen": 567476224 + }, + { + "epoch": 0.17, + "learning_rate": 0.000418234704112337, + "loss": 1.4352, + "theoretical_loss": 3.862364364163148, + "tokens_seen": 567541760 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004182246740220662, + "loss": 1.5347, + "theoretical_loss": 3.8623173733069898, + "tokens_seen": 567607296 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004182146439317954, + "loss": 1.4536, + "theoretical_loss": 3.8622703893950465, + "tokens_seen": 567672832 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004182046138415246, + "loss": 1.4904, + "theoretical_loss": 3.8622234124254917, + "tokens_seen": 567738368 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041819458375125376, + "loss": 1.5434, + "theoretical_loss": 3.862176442396498, + "tokens_seen": 567803904 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041818455366098294, + "loss": 1.4374, + "theoretical_loss": 3.862129479306238, + "tokens_seen": 567869440 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004181745235707121, + "loss": 1.517, + "theoretical_loss": 3.8620825231528877, + "tokens_seen": 567934976 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041816449348044136, + "loss": 1.5273, + "theoretical_loss": 3.8620355739346213, + "tokens_seen": 568000512 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041815446339017054, + "loss": 1.5057, + "theoretical_loss": 3.861988631649614, + "tokens_seen": 568066048 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004181444332998997, + "loss": 1.5355, + "theoretical_loss": 3.861941696296044, + "tokens_seen": 568131584 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004181344032096289, + "loss": 1.5104, + "theoretical_loss": 3.8618947678720863, + "tokens_seen": 568197120 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004181243731193581, + "loss": 1.4229, + "theoretical_loss": 3.86184784637592, + "tokens_seen": 568262656 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004181143430290873, + "loss": 1.5303, + "theoretical_loss": 3.8618009318057234, + "tokens_seen": 568328192 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041810431293881644, + "loss": 1.6433, + "theoretical_loss": 3.8617540241596755, + "tokens_seen": 568393728 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004180942828485457, + "loss": 1.493, + "theoretical_loss": 3.8617071234359566, + "tokens_seen": 568459264 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": 0.00019693221838679165, + "objective/train/docs_used": 331416, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.016237258911133, + "objective/train/original_loss": 3.016237735748291, + "objective/train/theoretical_loss": 3.8616602296327467, + "objective/train/tokens_used": 588984800, + "objective/train/value_avg": -0.0042877197265625, + "objective/train/value_loss": 3.3429707400500774e-05, + "objective/train/value_max": -0.00023233890533447266, + "objective/train/value_min": -0.04443359375, + "objective/train/value_reward_corr": 0.31550933062206143, + "objective/train/value_std": 0.00319671630859375, + "objective/train/weight_avg": 1.0035258531570435, + "objective/train/weighted_lm_loss": 3.0285801887512207, + "objective/train/weights_max": 1.3665543794631958, + "objective/train/weights_min": 0.3331280052661896, + "theoretical_loss": 3.8616602296327467, + "tokens_seen": 568524800 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004180842527582748, + "loss": 1.3196, + "theoretical_loss": 3.8616602296327467, + "tokens_seen": 568524800 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041807422266800404, + "loss": 1.3948, + "theoretical_loss": 3.8616133427482273, + "tokens_seen": 568590336 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004180641925777332, + "loss": 1.5907, + "theoretical_loss": 3.861566462780581, + "tokens_seen": 568655872 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004180541624874624, + "loss": 1.5284, + "theoretical_loss": 3.8615195897279886, + "tokens_seen": 568721408 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004180441323971916, + "loss": 1.5984, + "theoretical_loss": 3.8614727235886352, + "tokens_seen": 568786944 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004180341023069208, + "loss": 1.3418, + "theoretical_loss": 3.8614258643607045, + "tokens_seen": 568852480 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041802407221664994, + "loss": 1.5408, + "theoretical_loss": 3.8613790120423808, + "tokens_seen": 568918016 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004180140421263792, + "loss": 1.4579, + "theoretical_loss": 3.8613321666318496, + "tokens_seen": 568983552 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004180040120361083, + "loss": 1.4938, + "theoretical_loss": 3.8612853281272965, + "tokens_seen": 569049088 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041799398194583754, + "loss": 1.4849, + "theoretical_loss": 3.8612384965269095, + "tokens_seen": 569114624 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004179839518555667, + "loss": 1.4431, + "theoretical_loss": 3.8611916718288746, + "tokens_seen": 569180160 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004179739217652959, + "loss": 1.4468, + "theoretical_loss": 3.86114485403138, + "tokens_seen": 569245696 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004179638916750251, + "loss": 1.4239, + "theoretical_loss": 3.8610980431326163, + "tokens_seen": 569311232 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041795386158475427, + "loss": 1.5425, + "theoretical_loss": 3.861051239130771, + "tokens_seen": 569376768 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041794383149448345, + "loss": 1.4443, + "theoretical_loss": 3.8610044420240355, + "tokens_seen": 569442304 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004179338014042127, + "loss": 1.4485, + "theoretical_loss": 3.8609576518105992, + "tokens_seen": 569507840 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004179237713139418, + "loss": 1.5154, + "theoretical_loss": 3.8609108684886557, + "tokens_seen": 569573376 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041791374122367105, + "loss": 1.3991, + "theoretical_loss": 3.8608640920563957, + "tokens_seen": 569638912 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041790371113340017, + "loss": 1.3591, + "theoretical_loss": 3.8608173225120126, + "tokens_seen": 569704448 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004178936810431294, + "loss": 1.5695, + "theoretical_loss": 3.8607705598536994, + "tokens_seen": 569769984 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004178836509528586, + "loss": 1.2912, + "theoretical_loss": 3.860723804079651, + "tokens_seen": 569835520 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041787362086258777, + "loss": 1.3986, + "theoretical_loss": 3.8606770551880625, + "tokens_seen": 569901056 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041786359077231695, + "loss": 1.4416, + "theoretical_loss": 3.860630313177129, + "tokens_seen": 569966592 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004178535606820462, + "loss": 1.5966, + "theoretical_loss": 3.860583578045047, + "tokens_seen": 570032128 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004178435305917753, + "loss": 1.4193, + "theoretical_loss": 3.860536849790013, + "tokens_seen": 570097664 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.0012760485988110304, + "objective/train/docs_used": 332819, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4292571544647217, + "objective/train/original_loss": 3.4292569160461426, + "objective/train/theoretical_loss": 3.8604901284102264, + "objective/train/tokens_used": 590623200, + "objective/train/value_avg": -0.00725555419921875, + "objective/train/value_loss": 8.924637222662568e-05, + "objective/train/value_max": -0.0003542900085449219, + "objective/train/value_min": -0.06842041015625, + "objective/train/value_reward_corr": 0.5522105204661499, + "objective/train/value_std": 0.00670623779296875, + "objective/train/weight_avg": 0.9913996458053589, + "objective/train/weighted_lm_loss": 3.3955256938934326, + "objective/train/weights_max": 1.7254784107208252, + "objective/train/weights_min": 0.3445335924625397, + "theoretical_loss": 3.8604901284102264, + "tokens_seen": 570163200 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041783350050150455, + "loss": 1.4734, + "theoretical_loss": 3.8604901284102264, + "tokens_seen": 570163200 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004178234704112337, + "loss": 1.2962, + "theoretical_loss": 3.8604434139038837, + "tokens_seen": 570228736 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004178134403209629, + "loss": 1.4206, + "theoretical_loss": 3.860396706269184, + "tokens_seen": 570294272 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004178034102306921, + "loss": 1.4903, + "theoretical_loss": 3.860350005504328, + "tokens_seen": 570359808 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004177933801404213, + "loss": 1.3933, + "theoretical_loss": 3.860303311607516, + "tokens_seen": 570425344 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041778335005015045, + "loss": 1.5933, + "theoretical_loss": 3.860256624576948, + "tokens_seen": 570490880 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041777331995987964, + "loss": 1.557, + "theoretical_loss": 3.860209944410827, + "tokens_seen": 570556416 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004177632898696088, + "loss": 1.3911, + "theoretical_loss": 3.8601632711073544, + "tokens_seen": 570621952 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041775325977933805, + "loss": 1.421, + "theoretical_loss": 3.860116604664734, + "tokens_seen": 570687488 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004177432296890672, + "loss": 1.4515, + "theoretical_loss": 3.860069945081169, + "tokens_seen": 570753024 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004177331995987964, + "loss": 1.4769, + "theoretical_loss": 3.8600232923548643, + "tokens_seen": 570818560 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004177231695085256, + "loss": 1.2995, + "theoretical_loss": 3.8599766464840246, + "tokens_seen": 570884096 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004177131394182548, + "loss": 1.4199, + "theoretical_loss": 3.8599300074668563, + "tokens_seen": 570949632 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041770310932798396, + "loss": 1.4195, + "theoretical_loss": 3.8598833753015653, + "tokens_seen": 571015168 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041769307923771314, + "loss": 1.4699, + "theoretical_loss": 3.8598367499863593, + "tokens_seen": 571080704 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004176830491474423, + "loss": 1.4147, + "theoretical_loss": 3.8597901315194454, + "tokens_seen": 571146240 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041767301905717156, + "loss": 1.334, + "theoretical_loss": 3.859743519899032, + "tokens_seen": 571211776 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004176629889669007, + "loss": 1.4937, + "theoretical_loss": 3.8596969151233296, + "tokens_seen": 571277312 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004176529588766299, + "loss": 1.4498, + "theoretical_loss": 3.859650317190547, + "tokens_seen": 571342848 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041764292878635904, + "loss": 1.5747, + "theoretical_loss": 3.8596037260988947, + "tokens_seen": 571408384 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004176328986960883, + "loss": 1.3433, + "theoretical_loss": 3.859557141846584, + "tokens_seen": 571473920 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041762286860581746, + "loss": 1.3597, + "theoretical_loss": 3.8595105644318273, + "tokens_seen": 571539456 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041761283851554664, + "loss": 1.5978, + "theoretical_loss": 3.8594639938528363, + "tokens_seen": 571604992 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004176028084252758, + "loss": 1.5293, + "theoretical_loss": 3.8594174301078246, + "tokens_seen": 571670528 + }, + { + "epoch": 0.17, + "learning_rate": 0.000417592778335005, + "loss": 1.4964, + "theoretical_loss": 3.8593708731950063, + "tokens_seen": 571736064 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -5.678942034137435e-05, + "objective/train/docs_used": 333556, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1807687282562256, + "objective/train/original_loss": 3.1807687282562256, + "objective/train/theoretical_loss": 3.8593243231125958, + "objective/train/tokens_used": 592261600, + "objective/train/value_avg": -0.00470733642578125, + "objective/train/value_loss": 6.684012623736635e-05, + "objective/train/value_max": -0.0003249645233154297, + "objective/train/value_min": -0.06500244140625, + "objective/train/value_reward_corr": 0.3981941094322816, + "objective/train/value_std": 0.004009246826171875, + "objective/train/weight_avg": 1.0024025440216064, + "objective/train/weighted_lm_loss": 3.1824562549591064, + "objective/train/weights_max": 1.563743233680725, + "objective/train/weights_min": 0.20692078769207, + "theoretical_loss": 3.8593243231125958, + "tokens_seen": 571801600 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004175827482447342, + "loss": 1.4382, + "theoretical_loss": 3.8593243231125958, + "tokens_seen": 571801600 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004175727181544634, + "loss": 1.4168, + "theoretical_loss": 3.8592777798588083, + "tokens_seen": 571867136 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041756268806419255, + "loss": 1.4595, + "theoretical_loss": 3.8592312434318594, + "tokens_seen": 571932672 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004175526579739218, + "loss": 1.4086, + "theoretical_loss": 3.8591847138299658, + "tokens_seen": 571998208 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041754262788365096, + "loss": 1.4408, + "theoretical_loss": 3.859138191051345, + "tokens_seen": 572063744 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041753259779338015, + "loss": 1.5005, + "theoretical_loss": 3.8590916750942146, + "tokens_seen": 572129280 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004175225677031093, + "loss": 1.4992, + "theoretical_loss": 3.8590451659567937, + "tokens_seen": 572194816 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004175125376128385, + "loss": 1.476, + "theoretical_loss": 3.8589986636373013, + "tokens_seen": 572260352 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004175025075225677, + "loss": 1.4601, + "theoretical_loss": 3.8589521681339565, + "tokens_seen": 572325888 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004174924774322969, + "loss": 1.4039, + "theoretical_loss": 3.8589056794449808, + "tokens_seen": 572391424 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041748244734202605, + "loss": 1.4379, + "theoretical_loss": 3.858859197568595, + "tokens_seen": 572456960 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004174724172517553, + "loss": 1.5036, + "theoretical_loss": 3.858812722503022, + "tokens_seen": 572522496 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004174623871614844, + "loss": 1.3495, + "theoretical_loss": 3.8587662542464827, + "tokens_seen": 572588032 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041745235707121365, + "loss": 1.3736, + "theoretical_loss": 3.858719792797202, + "tokens_seen": 572653568 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041744232698094283, + "loss": 1.4316, + "theoretical_loss": 3.8586733381534026, + "tokens_seen": 572719104 + }, + { + "epoch": 0.17, + "learning_rate": 0.000417432296890672, + "loss": 1.4259, + "theoretical_loss": 3.8586268903133094, + "tokens_seen": 572784640 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004174222668004012, + "loss": 1.3376, + "theoretical_loss": 3.8585804492751485, + "tokens_seen": 572850176 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041741223671013037, + "loss": 1.49, + "theoretical_loss": 3.8585340150371445, + "tokens_seen": 572915712 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004174022066198596, + "loss": 1.5019, + "theoretical_loss": 3.8584875875975246, + "tokens_seen": 572981248 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004173921765295888, + "loss": 1.4579, + "theoretical_loss": 3.858441166954516, + "tokens_seen": 573046784 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041738214643931797, + "loss": 1.4967, + "theoretical_loss": 3.858394753106347, + "tokens_seen": 573112320 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041737211634904715, + "loss": 1.4285, + "theoretical_loss": 3.8583483460512458, + "tokens_seen": 573177856 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004173620862587764, + "loss": 1.5695, + "theoretical_loss": 3.8583019457874412, + "tokens_seen": 573243392 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004173520561685055, + "loss": 1.4472, + "theoretical_loss": 3.8582555523131643, + "tokens_seen": 573308928 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041734202607823475, + "loss": 1.6052, + "theoretical_loss": 3.8582091656266444, + "tokens_seen": 573374464 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": 0.0011146464385092258, + "objective/train/docs_used": 334721, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0009419918060303, + "objective/train/original_loss": 3.0009424686431885, + "objective/train/theoretical_loss": 3.8581627857261136, + "objective/train/tokens_used": 593900000, + "objective/train/value_avg": -0.005340576171875, + "objective/train/value_loss": 6.311726610874757e-05, + "objective/train/value_max": -0.0004458427429199219, + "objective/train/value_min": -0.06463623046875, + "objective/train/value_reward_corr": 0.39587060771291194, + "objective/train/value_std": 0.004878997802734375, + "objective/train/weight_avg": 1.0141488313674927, + "objective/train/weighted_lm_loss": 3.0371830463409424, + "objective/train/weights_max": 1.4617563486099243, + "objective/train/weights_min": 0.2500069737434387, + "theoretical_loss": 3.8581627857261136, + "tokens_seen": 573440000 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004173319959879639, + "loss": 1.5167, + "theoretical_loss": 3.8581627857261136, + "tokens_seen": 573440000 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004173219658976931, + "loss": 1.339, + "theoretical_loss": 3.858116412609803, + "tokens_seen": 573505536 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004173119358074223, + "loss": 1.4627, + "theoretical_loss": 3.8580700462759463, + "tokens_seen": 573571072 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004173019057171515, + "loss": 1.4482, + "theoretical_loss": 3.8580236867227757, + "tokens_seen": 573636608 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041729187562688065, + "loss": 1.4791, + "theoretical_loss": 3.8579773339485257, + "tokens_seen": 573702144 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041728184553660984, + "loss": 1.511, + "theoretical_loss": 3.85793098795143, + "tokens_seen": 573767680 + }, + { + "epoch": 0.17, + "learning_rate": 0.000417271815446339, + "loss": 1.5094, + "theoretical_loss": 3.8578846487297245, + "tokens_seen": 573833216 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041726178535606825, + "loss": 1.3742, + "theoretical_loss": 3.8578383162816454, + "tokens_seen": 573898752 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004172517552657974, + "loss": 1.4235, + "theoretical_loss": 3.8577919906054285, + "tokens_seen": 573964288 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004172417251755266, + "loss": 1.5088, + "theoretical_loss": 3.8577456716993117, + "tokens_seen": 574029824 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004172316950852558, + "loss": 1.3335, + "theoretical_loss": 3.857699359561532, + "tokens_seen": 574095360 + }, + { + "epoch": 0.17, + "learning_rate": 0.000417221664994985, + "loss": 1.3571, + "theoretical_loss": 3.857653054190328, + "tokens_seen": 574160896 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041721163490471416, + "loss": 1.5293, + "theoretical_loss": 3.8576067555839395, + "tokens_seen": 574226432 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041720160481444334, + "loss": 1.3148, + "theoretical_loss": 3.8575604637406062, + "tokens_seen": 574291968 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004171915747241725, + "loss": 1.3109, + "theoretical_loss": 3.857514178658568, + "tokens_seen": 574357504 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041718154463390176, + "loss": 1.48, + "theoretical_loss": 3.857467900336067, + "tokens_seen": 574423040 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004171715145436309, + "loss": 1.4146, + "theoretical_loss": 3.8574216287713434, + "tokens_seen": 574488576 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004171614844533601, + "loss": 1.4299, + "theoretical_loss": 3.8573753639626416, + "tokens_seen": 574554112 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041715145436308924, + "loss": 1.4054, + "theoretical_loss": 3.857329105908203, + "tokens_seen": 574619648 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004171414242728185, + "loss": 1.3235, + "theoretical_loss": 3.857282854606273, + "tokens_seen": 574685184 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041713139418254766, + "loss": 1.5374, + "theoretical_loss": 3.8572366100550948, + "tokens_seen": 574750720 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041712136409227684, + "loss": 1.5653, + "theoretical_loss": 3.857190372252914, + "tokens_seen": 574816256 + }, + { + "epoch": 0.17, + "learning_rate": 0.000417111334002006, + "loss": 1.4996, + "theoretical_loss": 3.8571441411979763, + "tokens_seen": 574881792 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004171013039117352, + "loss": 1.4241, + "theoretical_loss": 3.857097916888528, + "tokens_seen": 574947328 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004170912738214644, + "loss": 1.4439, + "theoretical_loss": 3.857051699322816, + "tokens_seen": 575012864 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.0015174535801634192, + "objective/train/docs_used": 335233, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.962026357650757, + "objective/train/original_loss": 2.9620258808135986, + "objective/train/theoretical_loss": 3.8570054884990883, + "objective/train/tokens_used": 595538400, + "objective/train/value_avg": -0.005954742431640625, + "objective/train/value_loss": 5.404729017755017e-05, + "objective/train/value_max": -0.0003859996795654297, + "objective/train/value_min": -0.0504150390625, + "objective/train/value_reward_corr": 0.5702445229719927, + "objective/train/value_std": 0.004886627197265625, + "objective/train/weight_avg": 0.987414538860321, + "objective/train/weighted_lm_loss": 2.92812180519104, + "objective/train/weights_max": 1.4777401685714722, + "objective/train/weights_min": 0.46350353956222534, + "theoretical_loss": 3.8570054884990883, + "tokens_seen": 575078400 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004170812437311936, + "loss": 1.5053, + "theoretical_loss": 3.8570054884990883, + "tokens_seen": 575078400 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041707121364092275, + "loss": 1.3845, + "theoretical_loss": 3.856959284415593, + "tokens_seen": 575143936 + }, + { + "epoch": 0.17, + "learning_rate": 0.000417061183550652, + "loss": 1.4164, + "theoretical_loss": 3.8569130870705797, + "tokens_seen": 575209472 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041705115346038116, + "loss": 1.5487, + "theoretical_loss": 3.8568668964622974, + "tokens_seen": 575275008 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041704112337011035, + "loss": 1.4074, + "theoretical_loss": 3.8568207125889966, + "tokens_seen": 575340544 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004170310932798395, + "loss": 1.5303, + "theoretical_loss": 3.8567745354489285, + "tokens_seen": 575406080 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004170210631895687, + "loss": 1.5085, + "theoretical_loss": 3.856728365040345, + "tokens_seen": 575471616 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004170110330992979, + "loss": 1.5041, + "theoretical_loss": 3.8566822013614974, + "tokens_seen": 575537152 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004170010030090271, + "loss": 1.4197, + "theoretical_loss": 3.8566360444106396, + "tokens_seen": 575602688 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041699097291875625, + "loss": 1.351, + "theoretical_loss": 3.8565898941860244, + "tokens_seen": 575668224 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004169809428284855, + "loss": 1.4728, + "theoretical_loss": 3.8565437506859066, + "tokens_seen": 575733760 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004169709127382146, + "loss": 1.5529, + "theoretical_loss": 3.856497613908541, + "tokens_seen": 575799296 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041696088264794385, + "loss": 1.5085, + "theoretical_loss": 3.856451483852183, + "tokens_seen": 575864832 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041695085255767303, + "loss": 1.383, + "theoretical_loss": 3.8564053605150894, + "tokens_seen": 575930368 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004169408224674022, + "loss": 1.4172, + "theoretical_loss": 3.8563592438955157, + "tokens_seen": 575995904 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004169307923771314, + "loss": 1.3874, + "theoretical_loss": 3.856313133991721, + "tokens_seen": 576061440 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004169207622868606, + "loss": 1.4498, + "theoretical_loss": 3.856267030801962, + "tokens_seen": 576126976 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041691073219658975, + "loss": 1.4973, + "theoretical_loss": 3.856220934324499, + "tokens_seen": 576192512 + }, + { + "epoch": 0.17, + "learning_rate": 0.000416900702106319, + "loss": 1.3865, + "theoretical_loss": 3.8561748445575903, + "tokens_seen": 576258048 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004168906720160481, + "loss": 1.3188, + "theoretical_loss": 3.8561287614994963, + "tokens_seen": 576323584 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041688064192577735, + "loss": 1.3545, + "theoretical_loss": 3.8560826851484777, + "tokens_seen": 576389120 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041687061183550653, + "loss": 1.3487, + "theoretical_loss": 3.8560366155027963, + "tokens_seen": 576454656 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004168605817452357, + "loss": 1.4505, + "theoretical_loss": 3.855990552560714, + "tokens_seen": 576520192 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004168505516549649, + "loss": 1.3536, + "theoretical_loss": 3.8559444963204923, + "tokens_seen": 576585728 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004168405215646941, + "loss": 1.299, + "theoretical_loss": 3.8558984467803965, + "tokens_seen": 576651264 + }, + { + "epoch": 0.17, + "objective/train/advantage_avg": -0.00022400562011171132, + "objective/train/docs_used": 335838, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.563371419906616, + "objective/train/original_loss": 2.563371419906616, + "objective/train/theoretical_loss": 3.855852403938689, + "objective/train/tokens_used": 597176800, + "objective/train/value_avg": -0.0047607421875, + "objective/train/value_loss": 4.464864105102606e-05, + "objective/train/value_max": -0.00029587745666503906, + "objective/train/value_min": -0.072265625, + "objective/train/value_reward_corr": 0.45563854936696824, + "objective/train/value_std": 0.0042724609375, + "objective/train/weight_avg": 0.9999141097068787, + "objective/train/weighted_lm_loss": 2.5609426498413086, + "objective/train/weights_max": 1.7110536098480225, + "objective/train/weights_min": 0.5323176383972168, + "theoretical_loss": 3.855852403938689, + "tokens_seen": 576716800 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041683049147442326, + "loss": 1.4496, + "theoretical_loss": 3.855852403938689, + "tokens_seen": 576716800 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004168204613841525, + "loss": 1.3465, + "theoretical_loss": 3.855806367793636, + "tokens_seen": 576782336 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004168104312938816, + "loss": 1.3757, + "theoretical_loss": 3.8557603383435013, + "tokens_seen": 576847872 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041680040120361085, + "loss": 1.4507, + "theoretical_loss": 3.8557143155865514, + "tokens_seen": 576913408 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041679037111334, + "loss": 1.3556, + "theoretical_loss": 3.855668299521053, + "tokens_seen": 576978944 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004167803410230692, + "loss": 1.6045, + "theoretical_loss": 3.8556222901452735, + "tokens_seen": 577044480 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004167703109327984, + "loss": 1.5305, + "theoretical_loss": 3.85557628745748, + "tokens_seen": 577110016 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004167602808425276, + "loss": 1.5337, + "theoretical_loss": 3.8555302914559415, + "tokens_seen": 577175552 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041675025075225676, + "loss": 1.3515, + "theoretical_loss": 3.855484302138927, + "tokens_seen": 577241088 + }, + { + "epoch": 0.17, + "learning_rate": 0.000416740220661986, + "loss": 1.4601, + "theoretical_loss": 3.8554383195047066, + "tokens_seen": 577306624 + }, + { + "epoch": 0.17, + "learning_rate": 0.0004167301905717151, + "loss": 1.5698, + "theoretical_loss": 3.855392343551551, + "tokens_seen": 577372160 + }, + { + "epoch": 0.17, + "learning_rate": 0.00041672016048144436, + "loss": 1.448, + "theoretical_loss": 3.8553463742777305, + "tokens_seen": 577437696 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004167101303911735, + "loss": 1.3331, + "theoretical_loss": 3.855300411681517, + "tokens_seen": 577503232 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004167001003009027, + "loss": 1.4327, + "theoretical_loss": 3.8552544557611834, + "tokens_seen": 577568768 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004166900702106319, + "loss": 1.4955, + "theoretical_loss": 3.855208506515002, + "tokens_seen": 577634304 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004166800401203611, + "loss": 1.3847, + "theoretical_loss": 3.8551625639412475, + "tokens_seen": 577699840 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041667001003009026, + "loss": 1.5221, + "theoretical_loss": 3.8551166280381928, + "tokens_seen": 577765376 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041665997993981944, + "loss": 1.3211, + "theoretical_loss": 3.8550706988041137, + "tokens_seen": 577830912 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004166499498495487, + "loss": 1.4244, + "theoretical_loss": 3.8550247762372862, + "tokens_seen": 577896448 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041663991975927786, + "loss": 1.4952, + "theoretical_loss": 3.8549788603359856, + "tokens_seen": 577961984 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041662988966900704, + "loss": 1.3884, + "theoretical_loss": 3.8549329510984887, + "tokens_seen": 578027520 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004166198595787362, + "loss": 1.3185, + "theoretical_loss": 3.854887048523074, + "tokens_seen": 578093056 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004166098294884654, + "loss": 1.542, + "theoretical_loss": 3.8548411526080195, + "tokens_seen": 578158592 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004165997993981946, + "loss": 1.4435, + "theoretical_loss": 3.854795263351603, + "tokens_seen": 578224128 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004165897693079238, + "loss": 1.5921, + "theoretical_loss": 3.854749380752105, + "tokens_seen": 578289664 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": 2.0141118511673994e-05, + "objective/train/docs_used": 337192, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.156684160232544, + "objective/train/original_loss": 3.1566836833953857, + "objective/train/theoretical_loss": 3.854703504807805, + "objective/train/tokens_used": 598815200, + "objective/train/value_avg": -0.0051422119140625, + "objective/train/value_loss": 4.323995381128043e-05, + "objective/train/value_max": -0.0004076957702636719, + "objective/train/value_min": -0.048126220703125, + "objective/train/value_reward_corr": 0.39946240616661066, + "objective/train/value_std": 0.00374603271484375, + "objective/train/weight_avg": 1.0022892951965332, + "objective/train/weighted_lm_loss": 3.164708137512207, + "objective/train/weights_max": 1.5699594020843506, + "objective/train/weights_min": 0.5566262602806091, + "theoretical_loss": 3.854703504807805, + "tokens_seen": 578355200 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041657973921765295, + "loss": 1.4528, + "theoretical_loss": 3.854703504807805, + "tokens_seen": 578355200 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004165697091273822, + "loss": 1.3836, + "theoretical_loss": 3.8546576355169835, + "tokens_seen": 578420736 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041655967903711136, + "loss": 1.5349, + "theoretical_loss": 3.8546117728779223, + "tokens_seen": 578486272 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041654964894684055, + "loss": 1.4213, + "theoretical_loss": 3.8545659168889035, + "tokens_seen": 578551808 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004165396188565697, + "loss": 1.5018, + "theoretical_loss": 3.854520067548209, + "tokens_seen": 578617344 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004165295887662989, + "loss": 1.5902, + "theoretical_loss": 3.854474224854123, + "tokens_seen": 578682880 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004165195586760281, + "loss": 1.4258, + "theoretical_loss": 3.8544283888049287, + "tokens_seen": 578748416 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004165095285857573, + "loss": 1.456, + "theoretical_loss": 3.854382559398911, + "tokens_seen": 578813952 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041649949849548645, + "loss": 1.334, + "theoretical_loss": 3.854336736634355, + "tokens_seen": 578879488 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004164894684052157, + "loss": 1.4736, + "theoretical_loss": 3.854290920509546, + "tokens_seen": 578945024 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004164794383149448, + "loss": 1.5614, + "theoretical_loss": 3.854245111022771, + "tokens_seen": 579010560 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041646940822467405, + "loss": 1.4439, + "theoretical_loss": 3.8541993081723174, + "tokens_seen": 579076096 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041645937813440323, + "loss": 1.4576, + "theoretical_loss": 3.8541535119564725, + "tokens_seen": 579141632 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004164493480441324, + "loss": 1.4813, + "theoretical_loss": 3.8541077223735245, + "tokens_seen": 579207168 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004164393179538616, + "loss": 1.366, + "theoretical_loss": 3.8540619394217623, + "tokens_seen": 579272704 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004164292878635908, + "loss": 1.418, + "theoretical_loss": 3.8540161630994763, + "tokens_seen": 579338240 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041641925777331995, + "loss": 1.4719, + "theoretical_loss": 3.853970393404956, + "tokens_seen": 579403776 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004164092276830492, + "loss": 1.5609, + "theoretical_loss": 3.8539246303364925, + "tokens_seen": 579469312 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004163991975927783, + "loss": 1.504, + "theoretical_loss": 3.853878873892377, + "tokens_seen": 579534848 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041638916750250755, + "loss": 1.4952, + "theoretical_loss": 3.8538331240709027, + "tokens_seen": 579600384 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041637913741223673, + "loss": 1.4584, + "theoretical_loss": 3.8537873808703615, + "tokens_seen": 579665920 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004163691073219659, + "loss": 1.2469, + "theoretical_loss": 3.8537416442890464, + "tokens_seen": 579731456 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004163590772316951, + "loss": 1.5515, + "theoretical_loss": 3.853695914325253, + "tokens_seen": 579796992 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004163490471414243, + "loss": 1.3754, + "theoretical_loss": 3.8536501909772745, + "tokens_seen": 579862528 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041633901705115346, + "loss": 1.3554, + "theoretical_loss": 3.853604474243407, + "tokens_seen": 579928064 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.00037389210774563253, + "objective/train/docs_used": 337878, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.1813926696777344, + "objective/train/original_loss": 2.1813926696777344, + "objective/train/theoretical_loss": 3.8535587641219466, + "objective/train/tokens_used": 600453600, + "objective/train/value_avg": -0.004009246826171875, + "objective/train/value_loss": 3.174396260874346e-05, + "objective/train/value_max": -0.0003612041473388672, + "objective/train/value_min": -0.038818359375, + "objective/train/value_reward_corr": 0.34062886043628054, + "objective/train/value_std": 0.0030574798583984375, + "objective/train/weight_avg": 0.9978086948394775, + "objective/train/weighted_lm_loss": 2.178893804550171, + "objective/train/weights_max": 1.3701545000076294, + "objective/train/weights_min": 0.7234852910041809, + "theoretical_loss": 3.8535587641219466, + "tokens_seen": 579993600 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004163289869608827, + "loss": 1.2411, + "theoretical_loss": 3.8535587641219466, + "tokens_seen": 579993600 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004163189568706118, + "loss": 1.4029, + "theoretical_loss": 3.853513060611189, + "tokens_seen": 580059136 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041630892678034105, + "loss": 1.468, + "theoretical_loss": 3.853467363709432, + "tokens_seen": 580124672 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004162988966900702, + "loss": 1.4673, + "theoretical_loss": 3.8534216734149735, + "tokens_seen": 580190208 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004162888665997994, + "loss": 1.2655, + "theoretical_loss": 3.8533759897261124, + "tokens_seen": 580255744 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004162788365095286, + "loss": 1.4492, + "theoretical_loss": 3.853330312641147, + "tokens_seen": 580321280 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004162688064192578, + "loss": 1.4499, + "theoretical_loss": 3.8532846421583775, + "tokens_seen": 580386816 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041625877632898696, + "loss": 1.4697, + "theoretical_loss": 3.853238978276104, + "tokens_seen": 580452352 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004162487462387162, + "loss": 1.5496, + "theoretical_loss": 3.8531933209926272, + "tokens_seen": 580517888 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004162387161484453, + "loss": 1.3882, + "theoretical_loss": 3.8531476703062495, + "tokens_seen": 580583424 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041622868605817456, + "loss": 1.3882, + "theoretical_loss": 3.8531020262152733, + "tokens_seen": 580648960 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004162186559679037, + "loss": 1.5659, + "theoretical_loss": 3.8530563887180005, + "tokens_seen": 580714496 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004162086258776329, + "loss": 1.484, + "theoretical_loss": 3.853010757812735, + "tokens_seen": 580780032 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004161985957873621, + "loss": 1.4862, + "theoretical_loss": 3.852965133497781, + "tokens_seen": 580845568 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004161885656970913, + "loss": 1.438, + "theoretical_loss": 3.852919515771444, + "tokens_seen": 580911104 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041617853560682046, + "loss": 1.5092, + "theoretical_loss": 3.852873904632028, + "tokens_seen": 580976640 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041616850551654964, + "loss": 1.3958, + "theoretical_loss": 3.8528283000778405, + "tokens_seen": 581042176 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004161584754262788, + "loss": 1.3763, + "theoretical_loss": 3.852782702107186, + "tokens_seen": 581107712 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041614844533600806, + "loss": 1.3472, + "theoretical_loss": 3.8527371107183748, + "tokens_seen": 581173248 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004161384152457372, + "loss": 1.4913, + "theoretical_loss": 3.8526915259097123, + "tokens_seen": 581238784 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004161283851554664, + "loss": 1.4204, + "theoretical_loss": 3.852645947679508, + "tokens_seen": 581304320 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041611835506519555, + "loss": 1.3466, + "theoretical_loss": 3.8526003760260714, + "tokens_seen": 581369856 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004161083249749248, + "loss": 1.4593, + "theoretical_loss": 3.8525548109477112, + "tokens_seen": 581435392 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041609829488465397, + "loss": 1.4838, + "theoretical_loss": 3.852509252442739, + "tokens_seen": 581500928 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041608826479438315, + "loss": 1.4523, + "theoretical_loss": 3.852463700509465, + "tokens_seen": 581566464 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": 0.0006574012222699821, + "objective/train/docs_used": 339259, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.559018135070801, + "objective/train/original_loss": 2.559018135070801, + "objective/train/theoretical_loss": 3.8524181551462013, + "objective/train/tokens_used": 602092000, + "objective/train/value_avg": -0.00492095947265625, + "objective/train/value_loss": 4.027660907013342e-05, + "objective/train/value_max": -0.0003829002380371094, + "objective/train/value_min": -0.134033203125, + "objective/train/value_reward_corr": 0.39133521154980405, + "objective/train/value_std": 0.004878997802734375, + "objective/train/weight_avg": 1.0086015462875366, + "objective/train/weighted_lm_loss": 2.5830817222595215, + "objective/train/weights_max": 3.41213321685791, + "objective/train/weights_min": 0.6523916125297546, + "theoretical_loss": 3.8524181551462013, + "tokens_seen": 581632000 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041607823470411233, + "loss": 1.2642, + "theoretical_loss": 3.8524181551462013, + "tokens_seen": 581632000 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041606820461384156, + "loss": 1.3584, + "theoretical_loss": 3.8523726163512597, + "tokens_seen": 581697536 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004160581745235707, + "loss": 1.4391, + "theoretical_loss": 3.8523270841229538, + "tokens_seen": 581763072 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004160481444332999, + "loss": 1.4948, + "theoretical_loss": 3.8522815584595964, + "tokens_seen": 581828608 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041603811434302905, + "loss": 1.4333, + "theoretical_loss": 3.852236039359502, + "tokens_seen": 581894144 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004160280842527583, + "loss": 1.4847, + "theoretical_loss": 3.8521905268209857, + "tokens_seen": 581959680 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041601805416248747, + "loss": 1.479, + "theoretical_loss": 3.8521450208423618, + "tokens_seen": 582025216 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041600802407221665, + "loss": 1.39, + "theoretical_loss": 3.852099521421948, + "tokens_seen": 582090752 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041599799398194583, + "loss": 1.4472, + "theoretical_loss": 3.8520540285580593, + "tokens_seen": 582156288 + }, + { + "epoch": 0.18, + "learning_rate": 0.000415987963891675, + "loss": 1.4575, + "theoretical_loss": 3.8520085422490142, + "tokens_seen": 582221824 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004159779338014042, + "loss": 1.4813, + "theoretical_loss": 3.8519630624931294, + "tokens_seen": 582287360 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041596790371113343, + "loss": 1.4798, + "theoretical_loss": 3.8519175892887243, + "tokens_seen": 582352896 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041595787362086256, + "loss": 1.2892, + "theoretical_loss": 3.851872122634118, + "tokens_seen": 582418432 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004159478435305918, + "loss": 1.3364, + "theoretical_loss": 3.851826662527629, + "tokens_seen": 582483968 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004159378134403209, + "loss": 1.3533, + "theoretical_loss": 3.85178120896758, + "tokens_seen": 582549504 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041592778335005015, + "loss": 1.4368, + "theoretical_loss": 3.85173576195229, + "tokens_seen": 582615040 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041591775325977934, + "loss": 1.4426, + "theoretical_loss": 3.8516903214800813, + "tokens_seen": 582680576 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004159077231695085, + "loss": 1.3164, + "theoretical_loss": 3.851644887549276, + "tokens_seen": 582746112 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041589769307923775, + "loss": 1.4411, + "theoretical_loss": 3.851599460158197, + "tokens_seen": 582811648 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041588766298896693, + "loss": 1.3678, + "theoretical_loss": 3.8515540393051673, + "tokens_seen": 582877184 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004158776328986961, + "loss": 1.4426, + "theoretical_loss": 3.851508624988512, + "tokens_seen": 582942720 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004158676028084253, + "loss": 1.4367, + "theoretical_loss": 3.851463217206555, + "tokens_seen": 583008256 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004158575727181545, + "loss": 1.3877, + "theoretical_loss": 3.851417815957622, + "tokens_seen": 583073792 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041584754262788366, + "loss": 1.498, + "theoretical_loss": 3.8513724212400384, + "tokens_seen": 583139328 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004158375125376129, + "loss": 1.3999, + "theoretical_loss": 3.851327033052131, + "tokens_seen": 583204864 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -9.340403630631045e-05, + "objective/train/docs_used": 340083, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.184685707092285, + "objective/train/original_loss": 3.184685468673706, + "objective/train/theoretical_loss": 3.8512816513922274, + "objective/train/tokens_used": 603730400, + "objective/train/value_avg": -0.004932403564453125, + "objective/train/value_loss": 5.0688962801359594e-05, + "objective/train/value_max": -0.000507354736328125, + "objective/train/value_min": -0.0469970703125, + "objective/train/value_reward_corr": 0.40696887192817155, + "objective/train/value_std": 0.003749847412109375, + "objective/train/weight_avg": 1.001389980316162, + "objective/train/weighted_lm_loss": 3.1980154514312744, + "objective/train/weights_max": 1.537089228630066, + "objective/train/weights_min": 0.41419631242752075, + "theoretical_loss": 3.8512816513922274, + "tokens_seen": 583270400 + }, + { + "epoch": 0.18, + "learning_rate": 0.000415827482447342, + "loss": 1.5278, + "theoretical_loss": 3.8512816513922274, + "tokens_seen": 583270400 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041581745235707126, + "loss": 1.4935, + "theoretical_loss": 3.851236276258655, + "tokens_seen": 583335936 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004158074222668004, + "loss": 1.5199, + "theoretical_loss": 3.8511909076497424, + "tokens_seen": 583401472 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004157973921765296, + "loss": 1.3763, + "theoretical_loss": 3.851145545563818, + "tokens_seen": 583467008 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004157873620862588, + "loss": 1.4693, + "theoretical_loss": 3.8511001899992126, + "tokens_seen": 583532544 + }, + { + "epoch": 0.18, + "learning_rate": 0.000415777331995988, + "loss": 1.3824, + "theoretical_loss": 3.851054840954255, + "tokens_seen": 583598080 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041576730190571716, + "loss": 1.4405, + "theoretical_loss": 3.8510094984272767, + "tokens_seen": 583663616 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004157572718154464, + "loss": 1.4254, + "theoretical_loss": 3.8509641624166098, + "tokens_seen": 583729152 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004157472417251755, + "loss": 1.298, + "theoretical_loss": 3.850918832920585, + "tokens_seen": 583794688 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041573721163490476, + "loss": 1.4994, + "theoretical_loss": 3.8508735099375366, + "tokens_seen": 583860224 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004157271815446339, + "loss": 1.3714, + "theoretical_loss": 3.850828193465797, + "tokens_seen": 583925760 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004157171514543631, + "loss": 1.5279, + "theoretical_loss": 3.8507828835036997, + "tokens_seen": 583991296 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004157071213640923, + "loss": 1.4378, + "theoretical_loss": 3.85073758004958, + "tokens_seen": 584056832 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004156970912738215, + "loss": 1.5338, + "theoretical_loss": 3.850692283101773, + "tokens_seen": 584122368 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041568706118355066, + "loss": 1.4297, + "theoretical_loss": 3.850646992658614, + "tokens_seen": 584187904 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041567703109327984, + "loss": 1.3485, + "theoretical_loss": 3.8506017087184405, + "tokens_seen": 584253440 + }, + { + "epoch": 0.18, + "learning_rate": 0.000415667001003009, + "loss": 1.4295, + "theoretical_loss": 3.850556431279588, + "tokens_seen": 584318976 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041565697091273826, + "loss": 1.3938, + "theoretical_loss": 3.8505111603403948, + "tokens_seen": 584384512 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004156469408224674, + "loss": 1.5231, + "theoretical_loss": 3.850465895899199, + "tokens_seen": 584450048 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004156369107321966, + "loss": 1.4147, + "theoretical_loss": 3.8504206379543398, + "tokens_seen": 584515584 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041562688064192575, + "loss": 1.4851, + "theoretical_loss": 3.8503753865041563, + "tokens_seen": 584581120 + }, + { + "epoch": 0.18, + "learning_rate": 0.000415616850551655, + "loss": 1.3626, + "theoretical_loss": 3.8503301415469884, + "tokens_seen": 584646656 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041560682046138417, + "loss": 1.3908, + "theoretical_loss": 3.8502849030811768, + "tokens_seen": 584712192 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041559679037111335, + "loss": 1.4926, + "theoretical_loss": 3.8502396711050633, + "tokens_seen": 584777728 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041558676028084253, + "loss": 1.3342, + "theoretical_loss": 3.850194445616989, + "tokens_seen": 584843264 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -5.9544922805798706e-06, + "objective/train/docs_used": 341398, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9288594722747803, + "objective/train/original_loss": 2.9288597106933594, + "objective/train/theoretical_loss": 3.8501492266152972, + "objective/train/tokens_used": 605368800, + "objective/train/value_avg": -0.004913330078125, + "objective/train/value_loss": 3.58720681106206e-05, + "objective/train/value_max": -0.0004992485046386719, + "objective/train/value_min": -0.075439453125, + "objective/train/value_reward_corr": 0.45142834764118434, + "objective/train/value_std": 0.004009246826171875, + "objective/train/weight_avg": 1.0016957521438599, + "objective/train/weighted_lm_loss": 2.9301867485046387, + "objective/train/weights_max": 2.126323699951172, + "objective/train/weights_min": 0.5543906688690186, + "theoretical_loss": 3.8501492266152972, + "tokens_seen": 584908800 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041557673019057176, + "loss": 1.5108, + "theoretical_loss": 3.8501492266152972, + "tokens_seen": 584908800 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004155667001003009, + "loss": 1.3251, + "theoretical_loss": 3.8501040140983305, + "tokens_seen": 584974336 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004155566700100301, + "loss": 1.4043, + "theoretical_loss": 3.850058808064432, + "tokens_seen": 585039872 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041554663991975925, + "loss": 1.4848, + "theoretical_loss": 3.850013608511947, + "tokens_seen": 585105408 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004155366098294885, + "loss": 1.4003, + "theoretical_loss": 3.8499684154392204, + "tokens_seen": 585170944 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041552657973921767, + "loss": 1.4428, + "theoretical_loss": 3.849923228844598, + "tokens_seen": 585236480 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041551654964894685, + "loss": 1.4207, + "theoretical_loss": 3.849878048726424, + "tokens_seen": 585302016 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041550651955867603, + "loss": 1.3495, + "theoretical_loss": 3.849832875083047, + "tokens_seen": 585367552 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004154964894684052, + "loss": 1.4563, + "theoretical_loss": 3.849787707912814, + "tokens_seen": 585433088 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004154864593781344, + "loss": 1.3505, + "theoretical_loss": 3.849742547214073, + "tokens_seen": 585498624 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041547642928786363, + "loss": 1.4201, + "theoretical_loss": 3.8496973929851723, + "tokens_seen": 585564160 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041546639919759276, + "loss": 1.3728, + "theoretical_loss": 3.8496522452244606, + "tokens_seen": 585629696 + }, + { + "epoch": 0.18, + "learning_rate": 0.000415456369107322, + "loss": 1.3398, + "theoretical_loss": 3.8496071039302886, + "tokens_seen": 585695232 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004154463390170511, + "loss": 1.4361, + "theoretical_loss": 3.849561969101006, + "tokens_seen": 585760768 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041543630892678035, + "loss": 1.4226, + "theoretical_loss": 3.8495168407349647, + "tokens_seen": 585826304 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041542627883650954, + "loss": 1.437, + "theoretical_loss": 3.8494717188305154, + "tokens_seen": 585891840 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004154162487462387, + "loss": 1.5313, + "theoretical_loss": 3.8494266033860103, + "tokens_seen": 585957376 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004154062186559679, + "loss": 1.4834, + "theoretical_loss": 3.849381494399802, + "tokens_seen": 586022912 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041539618856569713, + "loss": 1.4287, + "theoretical_loss": 3.8493363918702457, + "tokens_seen": 586088448 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041538615847542626, + "loss": 1.5025, + "theoretical_loss": 3.8492912957956933, + "tokens_seen": 586153984 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004153761283851555, + "loss": 1.4398, + "theoretical_loss": 3.8492462061745, + "tokens_seen": 586219520 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004153660982948846, + "loss": 1.3674, + "theoretical_loss": 3.849201123005021, + "tokens_seen": 586285056 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041535606820461386, + "loss": 1.4391, + "theoretical_loss": 3.849156046285613, + "tokens_seen": 586350592 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041534603811434304, + "loss": 1.4821, + "theoretical_loss": 3.8491109760146314, + "tokens_seen": 586416128 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004153360080240722, + "loss": 1.3437, + "theoretical_loss": 3.849065912190434, + "tokens_seen": 586481664 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.00048402781249023974, + "objective/train/docs_used": 342071, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8706774711608887, + "objective/train/original_loss": 2.8706774711608887, + "objective/train/theoretical_loss": 3.849020854811377, + "objective/train/tokens_used": 607007200, + "objective/train/value_avg": -0.00543212890625, + "objective/train/value_loss": 6.362679414451122e-05, + "objective/train/value_max": -0.0004305839538574219, + "objective/train/value_min": -0.06231689453125, + "objective/train/value_reward_corr": 0.35792577677322807, + "objective/train/value_std": 0.004547119140625, + "objective/train/weight_avg": 0.9980591535568237, + "objective/train/weighted_lm_loss": 2.8647050857543945, + "objective/train/weights_max": 1.8648282289505005, + "objective/train/weights_min": 0.15006330609321594, + "theoretical_loss": 3.849020854811377, + "tokens_seen": 586547200 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004153259779338014, + "loss": 1.4703, + "theoretical_loss": 3.849020854811377, + "tokens_seen": 586547200 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004153159478435306, + "loss": 1.4526, + "theoretical_loss": 3.8489758038758204, + "tokens_seen": 586612736 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041530591775325976, + "loss": 1.4676, + "theoretical_loss": 3.848930759382122, + "tokens_seen": 586678272 + }, + { + "epoch": 0.18, + "learning_rate": 0.000415295887662989, + "loss": 1.5065, + "theoretical_loss": 3.8488857213286414, + "tokens_seen": 586743808 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004152858575727181, + "loss": 1.3767, + "theoretical_loss": 3.8488406897137386, + "tokens_seen": 586809344 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041527582748244736, + "loss": 1.4422, + "theoretical_loss": 3.8487956645357744, + "tokens_seen": 586874880 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004152657973921765, + "loss": 1.4107, + "theoretical_loss": 3.84875064579311, + "tokens_seen": 586940416 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004152557673019057, + "loss": 1.4835, + "theoretical_loss": 3.848705633484107, + "tokens_seen": 587005952 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004152457372116349, + "loss": 1.4848, + "theoretical_loss": 3.8486606276071287, + "tokens_seen": 587071488 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004152357071213641, + "loss": 1.4485, + "theoretical_loss": 3.8486156281605366, + "tokens_seen": 587137024 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041522567703109327, + "loss": 1.5496, + "theoretical_loss": 3.848570635142696, + "tokens_seen": 587202560 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004152156469408225, + "loss": 1.3474, + "theoretical_loss": 3.84852564855197, + "tokens_seen": 587268096 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041520561685055163, + "loss": 1.484, + "theoretical_loss": 3.848480668386723, + "tokens_seen": 587333632 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041519558676028086, + "loss": 1.3881, + "theoretical_loss": 3.848435694645322, + "tokens_seen": 587399168 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041518555667001, + "loss": 1.4171, + "theoretical_loss": 3.848390727326132, + "tokens_seen": 587464704 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004151755265797392, + "loss": 1.3782, + "theoretical_loss": 3.84834576642752, + "tokens_seen": 587530240 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004151654964894684, + "loss": 1.4817, + "theoretical_loss": 3.848300811947852, + "tokens_seen": 587595776 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004151554663991976, + "loss": 1.4087, + "theoretical_loss": 3.8482558638854982, + "tokens_seen": 587661312 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004151454363089268, + "loss": 1.5041, + "theoretical_loss": 3.8482109222388248, + "tokens_seen": 587726848 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041513540621865595, + "loss": 1.4636, + "theoretical_loss": 3.848165987006202, + "tokens_seen": 587792384 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004151253761283852, + "loss": 1.5251, + "theoretical_loss": 3.848121058185999, + "tokens_seen": 587857920 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041511534603811437, + "loss": 1.3266, + "theoretical_loss": 3.8480761357765862, + "tokens_seen": 587923456 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041510531594784355, + "loss": 1.4351, + "theoretical_loss": 3.848031219776334, + "tokens_seen": 587988992 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041509528585757273, + "loss": 1.4506, + "theoretical_loss": 3.8479863101836145, + "tokens_seen": 588054528 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041508525576730196, + "loss": 1.3168, + "theoretical_loss": 3.847941406996799, + "tokens_seen": 588120064 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": 0.0004657788958866149, + "objective/train/docs_used": 342650, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9296154975891113, + "objective/train/original_loss": 2.9296154975891113, + "objective/train/theoretical_loss": 3.8478965102142606, + "objective/train/tokens_used": 608645600, + "objective/train/value_avg": -0.00496673583984375, + "objective/train/value_loss": 4.150447784923017e-05, + "objective/train/value_max": -0.0005235671997070312, + "objective/train/value_min": -0.054901123046875, + "objective/train/value_reward_corr": 0.22232206283608624, + "objective/train/value_std": 0.0033435821533203125, + "objective/train/weight_avg": 1.0066713094711304, + "objective/train/weighted_lm_loss": 2.9509437084198, + "objective/train/weights_max": 1.5349984169006348, + "objective/train/weights_min": 0.5600444078445435, + "theoretical_loss": 3.8478965102142606, + "tokens_seen": 588185600 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004150752256770311, + "loss": 1.4392, + "theoretical_loss": 3.8478965102142606, + "tokens_seen": 588185600 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004150651955867603, + "loss": 1.4054, + "theoretical_loss": 3.8478516198343717, + "tokens_seen": 588251136 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041505516549648945, + "loss": 1.5095, + "theoretical_loss": 3.8478067358555075, + "tokens_seen": 588316672 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004150451354062187, + "loss": 1.3355, + "theoretical_loss": 3.8477618582760407, + "tokens_seen": 588382208 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041503510531594787, + "loss": 1.513, + "theoretical_loss": 3.847716987094347, + "tokens_seen": 588447744 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041502507522567705, + "loss": 1.4631, + "theoretical_loss": 3.8476721223088024, + "tokens_seen": 588513280 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041501504513540623, + "loss": 1.3483, + "theoretical_loss": 3.8476272639177824, + "tokens_seen": 588578816 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004150050150451354, + "loss": 1.4681, + "theoretical_loss": 3.8475824119196638, + "tokens_seen": 588644352 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004149949849548646, + "loss": 1.4723, + "theoretical_loss": 3.8475375663128246, + "tokens_seen": 588709888 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041498495486459383, + "loss": 1.3955, + "theoretical_loss": 3.8474927270956423, + "tokens_seen": 588775424 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041497492477432296, + "loss": 1.3157, + "theoretical_loss": 3.8474478942664954, + "tokens_seen": 588840960 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004149648946840522, + "loss": 1.4147, + "theoretical_loss": 3.8474030678237625, + "tokens_seen": 588906496 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004149548645937813, + "loss": 1.5282, + "theoretical_loss": 3.8473582477658237, + "tokens_seen": 588972032 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041494483450351055, + "loss": 1.3886, + "theoretical_loss": 3.8473134340910597, + "tokens_seen": 589037568 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041493480441323974, + "loss": 1.488, + "theoretical_loss": 3.847268626797851, + "tokens_seen": 589103104 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004149247743229689, + "loss": 1.5236, + "theoretical_loss": 3.847223825884579, + "tokens_seen": 589168640 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004149147442326981, + "loss": 1.5187, + "theoretical_loss": 3.847179031349626, + "tokens_seen": 589234176 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041490471414242733, + "loss": 1.5146, + "theoretical_loss": 3.847134243191375, + "tokens_seen": 589299712 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041489468405215646, + "loss": 1.5082, + "theoretical_loss": 3.847089461408208, + "tokens_seen": 589365248 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004148846539618857, + "loss": 1.3773, + "theoretical_loss": 3.8470446859985103, + "tokens_seen": 589430784 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004148746238716148, + "loss": 1.348, + "theoretical_loss": 3.846999916960666, + "tokens_seen": 589496320 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041486459378134406, + "loss": 1.4395, + "theoretical_loss": 3.8469551542930587, + "tokens_seen": 589561856 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041485456369107324, + "loss": 1.3606, + "theoretical_loss": 3.8469103979940757, + "tokens_seen": 589627392 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004148445336008024, + "loss": 1.5437, + "theoretical_loss": 3.846865648062103, + "tokens_seen": 589692928 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004148345035105316, + "loss": 1.3613, + "theoretical_loss": 3.846820904495526, + "tokens_seen": 589758464 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": 0.0004962650709785521, + "objective/train/docs_used": 343935, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.642508029937744, + "objective/train/original_loss": 2.6425082683563232, + "objective/train/theoretical_loss": 3.8467761672927336, + "objective/train/tokens_used": 610284000, + "objective/train/value_avg": -0.005344390869140625, + "objective/train/value_loss": 5.461870023282245e-05, + "objective/train/value_max": -0.0004839897155761719, + "objective/train/value_min": -0.06292724609375, + "objective/train/value_reward_corr": 0.4047107110544551, + "objective/train/value_std": 0.004207611083984375, + "objective/train/weight_avg": 1.0075479745864868, + "objective/train/weighted_lm_loss": 2.6544737815856934, + "objective/train/weights_max": 1.6704580783843994, + "objective/train/weights_min": 0.46043717861175537, + "theoretical_loss": 3.8467761672927336, + "tokens_seen": 589824000 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004148244734202608, + "loss": 1.509, + "theoretical_loss": 3.8467761672927336, + "tokens_seen": 589824000 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041481444332998996, + "loss": 1.5502, + "theoretical_loss": 3.8467314364521132, + "tokens_seen": 589889536 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004148044132397192, + "loss": 1.3735, + "theoretical_loss": 3.8466867119720534, + "tokens_seen": 589955072 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004147943831494483, + "loss": 1.3729, + "theoretical_loss": 3.846641993850943, + "tokens_seen": 590020608 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041478435305917756, + "loss": 1.4221, + "theoretical_loss": 3.846597282087172, + "tokens_seen": 590086144 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004147743229689067, + "loss": 1.3877, + "theoretical_loss": 3.846552576679131, + "tokens_seen": 590151680 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004147642928786359, + "loss": 1.5103, + "theoretical_loss": 3.8465078776252097, + "tokens_seen": 590217216 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004147542627883651, + "loss": 1.4153, + "theoretical_loss": 3.8464631849238016, + "tokens_seen": 590282752 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004147442326980943, + "loss": 1.387, + "theoretical_loss": 3.8464184985732968, + "tokens_seen": 590348288 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041473420260782347, + "loss": 1.5239, + "theoretical_loss": 3.846373818572089, + "tokens_seen": 590413824 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004147241725175527, + "loss": 1.4378, + "theoretical_loss": 3.8463291449185713, + "tokens_seen": 590479360 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041471414242728183, + "loss": 1.414, + "theoretical_loss": 3.846284477611137, + "tokens_seen": 590544896 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041470411233701106, + "loss": 1.5222, + "theoretical_loss": 3.8462398166481817, + "tokens_seen": 590610432 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004146940822467402, + "loss": 1.494, + "theoretical_loss": 3.846195162028099, + "tokens_seen": 590675968 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004146840521564694, + "loss": 1.5144, + "theoretical_loss": 3.8461505137492855, + "tokens_seen": 590741504 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004146740220661986, + "loss": 1.3858, + "theoretical_loss": 3.8461058718101366, + "tokens_seen": 590807040 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004146639919759278, + "loss": 1.5279, + "theoretical_loss": 3.846061236209049, + "tokens_seen": 590872576 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041465396188565697, + "loss": 1.3774, + "theoretical_loss": 3.8460166069444206, + "tokens_seen": 590938112 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041464393179538615, + "loss": 1.5018, + "theoretical_loss": 3.84597198401465, + "tokens_seen": 591003648 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041463390170511533, + "loss": 1.3481, + "theoretical_loss": 3.845927367418134, + "tokens_seen": 591069184 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041462387161484457, + "loss": 1.3914, + "theoretical_loss": 3.845882757153272, + "tokens_seen": 591134720 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004146138415245737, + "loss": 1.3102, + "theoretical_loss": 3.8458381532184647, + "tokens_seen": 591200256 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041460381143430293, + "loss": 1.5543, + "theoretical_loss": 3.8457935556121114, + "tokens_seen": 591265792 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004145937813440321, + "loss": 1.4896, + "theoretical_loss": 3.8457489643326133, + "tokens_seen": 591331328 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004145837512537613, + "loss": 1.307, + "theoretical_loss": 3.845704379378372, + "tokens_seen": 591396864 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.001070996979251504, + "objective/train/docs_used": 344744, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9186270236968994, + "objective/train/original_loss": 2.9186270236968994, + "objective/train/theoretical_loss": 3.8456598007477893, + "objective/train/tokens_used": 611922400, + "objective/train/value_avg": -0.0046844482421875, + "objective/train/value_loss": 5.819456055178307e-05, + "objective/train/value_max": -0.0004565715789794922, + "objective/train/value_min": -0.049774169921875, + "objective/train/value_reward_corr": 0.4066705554115347, + "objective/train/value_std": 0.0034961700439453125, + "objective/train/weight_avg": 0.9919410943984985, + "objective/train/weighted_lm_loss": 2.895192861557007, + "objective/train/weights_max": 1.4874060153961182, + "objective/train/weights_min": 0.2912098169326782, + "theoretical_loss": 3.8456598007477893, + "tokens_seen": 591462400 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004145737211634905, + "loss": 1.4104, + "theoretical_loss": 3.8456598007477893, + "tokens_seen": 591462400 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041456369107321965, + "loss": 1.4247, + "theoretical_loss": 3.845615228439267, + "tokens_seen": 591527936 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041455366098294884, + "loss": 1.3193, + "theoretical_loss": 3.8455706624512094, + "tokens_seen": 591593472 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041454363089267807, + "loss": 1.3579, + "theoretical_loss": 3.84552610278202, + "tokens_seen": 591659008 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004145336008024072, + "loss": 1.5372, + "theoretical_loss": 3.845481549430102, + "tokens_seen": 591724544 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041452357071213643, + "loss": 1.3687, + "theoretical_loss": 3.8454370023938615, + "tokens_seen": 591790080 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041451354062186556, + "loss": 1.4755, + "theoretical_loss": 3.845392461671704, + "tokens_seen": 591855616 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004145035105315948, + "loss": 1.3949, + "theoretical_loss": 3.8453479272620346, + "tokens_seen": 591921152 + }, + { + "epoch": 0.18, + "learning_rate": 0.000414493480441324, + "loss": 1.3034, + "theoretical_loss": 3.8453033991632606, + "tokens_seen": 591986688 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041448345035105316, + "loss": 1.3798, + "theoretical_loss": 3.8452588773737886, + "tokens_seen": 592052224 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041447342026078234, + "loss": 1.3955, + "theoretical_loss": 3.845214361892027, + "tokens_seen": 592117760 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004144633901705115, + "loss": 1.4536, + "theoretical_loss": 3.845169852716384, + "tokens_seen": 592183296 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004144533600802407, + "loss": 1.3842, + "theoretical_loss": 3.8451253498452687, + "tokens_seen": 592248832 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041444332998996994, + "loss": 1.3113, + "theoretical_loss": 3.84508085327709, + "tokens_seen": 592314368 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041443329989969906, + "loss": 1.4162, + "theoretical_loss": 3.8450363630102586, + "tokens_seen": 592379904 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004144232698094283, + "loss": 1.3908, + "theoretical_loss": 3.8449918790431843, + "tokens_seen": 592445440 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041441323971915753, + "loss": 1.4539, + "theoretical_loss": 3.84494740137428, + "tokens_seen": 592510976 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041440320962888666, + "loss": 1.2594, + "theoretical_loss": 3.8449029300019557, + "tokens_seen": 592576512 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004143931795386159, + "loss": 1.4634, + "theoretical_loss": 3.8448584649246245, + "tokens_seen": 592642048 + }, + { + "epoch": 0.18, + "learning_rate": 0.000414383149448345, + "loss": 1.4909, + "theoretical_loss": 3.844814006140699, + "tokens_seen": 592707584 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041437311935807426, + "loss": 1.3517, + "theoretical_loss": 3.844769553648594, + "tokens_seen": 592773120 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041436308926780344, + "loss": 1.5908, + "theoretical_loss": 3.8447251074467226, + "tokens_seen": 592838656 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004143530591775326, + "loss": 1.5092, + "theoretical_loss": 3.844680667533499, + "tokens_seen": 592904192 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004143430290872618, + "loss": 1.3165, + "theoretical_loss": 3.8446362339073397, + "tokens_seen": 592969728 + }, + { + "epoch": 0.18, + "learning_rate": 0.000414332998996991, + "loss": 1.4547, + "theoretical_loss": 3.84459180656666, + "tokens_seen": 593035264 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.0008750300039537251, + "objective/train/docs_used": 345976, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9947049617767334, + "objective/train/original_loss": 2.9947052001953125, + "objective/train/theoretical_loss": 3.844547385509876, + "objective/train/tokens_used": 613560800, + "objective/train/value_avg": -0.004901885986328125, + "objective/train/value_loss": 6.508078513434157e-05, + "objective/train/value_max": -0.0003905296325683594, + "objective/train/value_min": -0.0775146484375, + "objective/train/value_reward_corr": 0.3141445733933946, + "objective/train/value_std": 0.0037860870361328125, + "objective/train/weight_avg": 0.9940714240074158, + "objective/train/weighted_lm_loss": 2.981045722961426, + "objective/train/weights_max": 1.8335015773773193, + "objective/train/weights_min": 0.08474171906709671, + "theoretical_loss": 3.844547385509876, + "tokens_seen": 593100800 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041432296890672016, + "loss": 1.5391, + "theoretical_loss": 3.844547385509876, + "tokens_seen": 593100800 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004143129388164494, + "loss": 1.4454, + "theoretical_loss": 3.8445029707354053, + "tokens_seen": 593166336 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004143029087261785, + "loss": 1.4797, + "theoretical_loss": 3.8444585622416643, + "tokens_seen": 593231872 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041429287863590776, + "loss": 1.4406, + "theoretical_loss": 3.844414160027072, + "tokens_seen": 593297408 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004142828485456369, + "loss": 1.3971, + "theoretical_loss": 3.844369764090048, + "tokens_seen": 593362944 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004142728184553661, + "loss": 1.3826, + "theoretical_loss": 3.84432537442901, + "tokens_seen": 593428480 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004142627883650953, + "loss": 1.3789, + "theoretical_loss": 3.8442809910423783, + "tokens_seen": 593494016 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004142527582748245, + "loss": 1.4966, + "theoretical_loss": 3.844236613928574, + "tokens_seen": 593559552 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041424272818455367, + "loss": 1.3208, + "theoretical_loss": 3.844192243086017, + "tokens_seen": 593625088 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004142326980942829, + "loss": 1.4714, + "theoretical_loss": 3.8441478785131302, + "tokens_seen": 593690624 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041422266800401203, + "loss": 1.4839, + "theoretical_loss": 3.8441035202083347, + "tokens_seen": 593756160 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041421263791374126, + "loss": 1.4429, + "theoretical_loss": 3.844059168170053, + "tokens_seen": 593821696 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004142026078234704, + "loss": 1.4193, + "theoretical_loss": 3.8440148223967094, + "tokens_seen": 593887232 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004141925777331996, + "loss": 1.3051, + "theoretical_loss": 3.843970482886727, + "tokens_seen": 593952768 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004141825476429288, + "loss": 1.5823, + "theoretical_loss": 3.8439261496385306, + "tokens_seen": 594018304 + }, + { + "epoch": 0.18, + "learning_rate": 0.000414172517552658, + "loss": 1.4607, + "theoretical_loss": 3.843881822650545, + "tokens_seen": 594083840 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041416248746238717, + "loss": 1.3626, + "theoretical_loss": 3.843837501921196, + "tokens_seen": 594149376 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041415245737211635, + "loss": 1.4483, + "theoretical_loss": 3.8437931874489095, + "tokens_seen": 594214912 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041414242728184553, + "loss": 1.3897, + "theoretical_loss": 3.8437488792321126, + "tokens_seen": 594280448 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041413239719157477, + "loss": 1.5026, + "theoretical_loss": 3.8437045772692313, + "tokens_seen": 594345984 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004141223671013039, + "loss": 1.4321, + "theoretical_loss": 3.8436602815586953, + "tokens_seen": 594411520 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041411233701103313, + "loss": 1.4073, + "theoretical_loss": 3.8436159920989317, + "tokens_seen": 594477056 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004141023069207623, + "loss": 1.5568, + "theoretical_loss": 3.8435717088883696, + "tokens_seen": 594542592 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004140922768304915, + "loss": 1.3814, + "theoretical_loss": 3.8435274319254393, + "tokens_seen": 594608128 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004140822467402207, + "loss": 1.2783, + "theoretical_loss": 3.84348316120857, + "tokens_seen": 594673664 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.0010994907934218645, + "objective/train/docs_used": 346603, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.020670175552368, + "objective/train/original_loss": 3.020670175552368, + "objective/train/theoretical_loss": 3.8434388967361928, + "objective/train/tokens_used": 615199200, + "objective/train/value_avg": -0.00510406494140625, + "objective/train/value_loss": 6.0760852647945285e-05, + "objective/train/value_max": -0.0004897117614746094, + "objective/train/value_min": -0.061981201171875, + "objective/train/value_reward_corr": 0.29017754182132705, + "objective/train/value_std": 0.0040130615234375, + "objective/train/weight_avg": 0.9919084906578064, + "objective/train/weighted_lm_loss": 3.006418228149414, + "objective/train/weights_max": 1.528584361076355, + "objective/train/weights_min": 0.48324528336524963, + "theoretical_loss": 3.8434388967361928, + "tokens_seen": 594739200 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041407221664994985, + "loss": 1.4693, + "theoretical_loss": 3.8434388967361928, + "tokens_seen": 594739200 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041406218655967904, + "loss": 1.4511, + "theoretical_loss": 3.843394638506739, + "tokens_seen": 594804736 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041405215646940827, + "loss": 1.4953, + "theoretical_loss": 3.8433503865186402, + "tokens_seen": 594870272 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004140421263791374, + "loss": 1.3999, + "theoretical_loss": 3.8433061407703293, + "tokens_seen": 594935808 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041403209628886663, + "loss": 1.3666, + "theoretical_loss": 3.843261901260239, + "tokens_seen": 595001344 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041402206619859576, + "loss": 1.4072, + "theoretical_loss": 3.843217667986802, + "tokens_seen": 595066880 + }, + { + "epoch": 0.18, + "learning_rate": 0.000414012036108325, + "loss": 1.3787, + "theoretical_loss": 3.843173440948454, + "tokens_seen": 595132416 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004140020060180542, + "loss": 1.4496, + "theoretical_loss": 3.843129220143628, + "tokens_seen": 595197952 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041399197592778336, + "loss": 1.4702, + "theoretical_loss": 3.8430850055707597, + "tokens_seen": 595263488 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041398194583751254, + "loss": 1.4218, + "theoretical_loss": 3.8430407972282854, + "tokens_seen": 595329024 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004139719157472417, + "loss": 1.3913, + "theoretical_loss": 3.8429965951146414, + "tokens_seen": 595394560 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004139618856569709, + "loss": 1.423, + "theoretical_loss": 3.842952399228264, + "tokens_seen": 595460096 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041395185556670014, + "loss": 1.3715, + "theoretical_loss": 3.8429082095675913, + "tokens_seen": 595525632 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041394182547642926, + "loss": 1.3547, + "theoretical_loss": 3.842864026131061, + "tokens_seen": 595591168 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004139317953861585, + "loss": 1.4682, + "theoretical_loss": 3.8428198489171117, + "tokens_seen": 595656704 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004139217652958877, + "loss": 1.4037, + "theoretical_loss": 3.8427756779241826, + "tokens_seen": 595722240 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041391173520561686, + "loss": 1.618, + "theoretical_loss": 3.842731513150713, + "tokens_seen": 595787776 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041390170511534604, + "loss": 1.4151, + "theoretical_loss": 3.842687354595144, + "tokens_seen": 595853312 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004138916750250752, + "loss": 1.5306, + "theoretical_loss": 3.842643202255916, + "tokens_seen": 595918848 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004138816449348044, + "loss": 1.4266, + "theoretical_loss": 3.8425990561314705, + "tokens_seen": 595984384 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041387161484453364, + "loss": 1.3338, + "theoretical_loss": 3.8425549162202493, + "tokens_seen": 596049920 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041386158475426277, + "loss": 1.3252, + "theoretical_loss": 3.8425107825206952, + "tokens_seen": 596115456 + }, + { + "epoch": 0.18, + "learning_rate": 0.000413851554663992, + "loss": 1.2664, + "theoretical_loss": 3.842466655031251, + "tokens_seen": 596180992 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041384152457372113, + "loss": 1.3976, + "theoretical_loss": 3.8424225337503604, + "tokens_seen": 596246528 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041383149448345036, + "loss": 1.448, + "theoretical_loss": 3.842378418676468, + "tokens_seen": 596312064 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.0013460576301440597, + "objective/train/docs_used": 348035, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0235414505004883, + "objective/train/original_loss": 3.0235414505004883, + "objective/train/theoretical_loss": 3.8423343098080185, + "objective/train/tokens_used": 616837600, + "objective/train/value_avg": -0.005954742431640625, + "objective/train/value_loss": 9.998626046581194e-05, + "objective/train/value_max": -0.00042724609375, + "objective/train/value_min": -0.0775146484375, + "objective/train/value_reward_corr": 0.540108919207995, + "objective/train/value_std": 0.005329132080078125, + "objective/train/weight_avg": 0.9910869598388672, + "objective/train/weighted_lm_loss": 3.012766122817993, + "objective/train/weights_max": 1.4756507873535156, + "objective/train/weights_min": 0.35738468170166016, + "theoretical_loss": 3.8423343098080185, + "tokens_seen": 596377600 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041382146439317954, + "loss": 1.412, + "theoretical_loss": 3.8423343098080185, + "tokens_seen": 596377600 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004138114343029087, + "loss": 1.2718, + "theoretical_loss": 3.8422902071434564, + "tokens_seen": 596443136 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004138014042126379, + "loss": 1.38, + "theoretical_loss": 3.842246110681229, + "tokens_seen": 596508672 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004137913741223671, + "loss": 1.513, + "theoretical_loss": 3.8422020204197813, + "tokens_seen": 596574208 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041378134403209627, + "loss": 1.4653, + "theoretical_loss": 3.8421579363575615, + "tokens_seen": 596639744 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004137713139418255, + "loss": 1.426, + "theoretical_loss": 3.8421138584930166, + "tokens_seen": 596705280 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041376128385155463, + "loss": 1.4526, + "theoretical_loss": 3.8420697868245943, + "tokens_seen": 596770816 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041375125376128387, + "loss": 1.4021, + "theoretical_loss": 3.8420257213507445, + "tokens_seen": 596836352 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041374122367101305, + "loss": 1.4948, + "theoretical_loss": 3.841981662069916, + "tokens_seen": 596901888 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041373119358074223, + "loss": 1.3418, + "theoretical_loss": 3.8419376089805573, + "tokens_seen": 596967424 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004137211634904714, + "loss": 1.5165, + "theoretical_loss": 3.841893562081121, + "tokens_seen": 597032960 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004137111334002006, + "loss": 1.3437, + "theoretical_loss": 3.8418495213700563, + "tokens_seen": 597098496 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041370110330992977, + "loss": 1.3893, + "theoretical_loss": 3.841805486845815, + "tokens_seen": 597164032 + }, + { + "epoch": 0.18, + "learning_rate": 0.000413691073219659, + "loss": 1.4401, + "theoretical_loss": 3.8417614585068502, + "tokens_seen": 597229568 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041368104312938813, + "loss": 1.5017, + "theoretical_loss": 3.8417174363516136, + "tokens_seen": 597295104 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041367101303911737, + "loss": 1.3951, + "theoretical_loss": 3.8416734203785587, + "tokens_seen": 597360640 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041366098294884655, + "loss": 1.3839, + "theoretical_loss": 3.8416294105861386, + "tokens_seen": 597426176 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041365095285857573, + "loss": 1.3964, + "theoretical_loss": 3.8415854069728077, + "tokens_seen": 597491712 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041364092276830497, + "loss": 1.4056, + "theoretical_loss": 3.8415414095370215, + "tokens_seen": 597557248 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004136308926780341, + "loss": 1.5006, + "theoretical_loss": 3.8414974182772355, + "tokens_seen": 597622784 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041362086258776333, + "loss": 1.4694, + "theoretical_loss": 3.841453433191904, + "tokens_seen": 597688320 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004136108324974925, + "loss": 1.364, + "theoretical_loss": 3.8414094542794857, + "tokens_seen": 597753856 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004136008024072217, + "loss": 1.5281, + "theoretical_loss": 3.841365481538436, + "tokens_seen": 597819392 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004135907723169509, + "loss": 1.3721, + "theoretical_loss": 3.841321514967213, + "tokens_seen": 597884928 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041358074222668005, + "loss": 1.4177, + "theoretical_loss": 3.841277554564275, + "tokens_seen": 597950464 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.002356248442083597, + "objective/train/docs_used": 348629, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5139925479888916, + "objective/train/original_loss": 2.5139923095703125, + "objective/train/theoretical_loss": 3.8412336003280814, + "objective/train/tokens_used": 618476000, + "objective/train/value_avg": -0.005138397216796875, + "objective/train/value_loss": 8.077573147602379e-05, + "objective/train/value_max": -0.0003199577331542969, + "objective/train/value_min": -0.07452392578125, + "objective/train/value_reward_corr": 0.6007512903418734, + "objective/train/value_std": 0.00522613525390625, + "objective/train/weight_avg": 0.980080246925354, + "objective/train/weighted_lm_loss": 2.485438823699951, + "objective/train/weights_max": 1.7074023485183716, + "objective/train/weights_min": 0.18112695217132568, + "theoretical_loss": 3.8412336003280814, + "tokens_seen": 598016000 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041357071213640924, + "loss": 1.2613, + "theoretical_loss": 3.8412336003280814, + "tokens_seen": 598016000 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041356068204613847, + "loss": 1.4514, + "theoretical_loss": 3.8411896522570896, + "tokens_seen": 598081536 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004135506519558676, + "loss": 1.3672, + "theoretical_loss": 3.8411457103497613, + "tokens_seen": 598147072 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041354062186559683, + "loss": 1.3952, + "theoretical_loss": 3.8411017746045557, + "tokens_seen": 598212608 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041353059177532596, + "loss": 1.3431, + "theoretical_loss": 3.8410578450199346, + "tokens_seen": 598278144 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004135205616850552, + "loss": 1.4923, + "theoretical_loss": 3.841013921594359, + "tokens_seen": 598343680 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004135105315947844, + "loss": 1.4444, + "theoretical_loss": 3.840970004326291, + "tokens_seen": 598409216 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041350050150451356, + "loss": 1.3769, + "theoretical_loss": 3.8409260932141924, + "tokens_seen": 598474752 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041349047141424274, + "loss": 1.502, + "theoretical_loss": 3.8408821882565274, + "tokens_seen": 598540288 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004134804413239719, + "loss": 1.4913, + "theoretical_loss": 3.84083828945176, + "tokens_seen": 598605824 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004134704112337011, + "loss": 1.354, + "theoretical_loss": 3.840794396798353, + "tokens_seen": 598671360 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041346038114343034, + "loss": 1.4357, + "theoretical_loss": 3.8407505102947725, + "tokens_seen": 598736896 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041345035105315946, + "loss": 1.4685, + "theoretical_loss": 3.8407066299394836, + "tokens_seen": 598802432 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004134403209628887, + "loss": 1.5141, + "theoretical_loss": 3.8406627557309516, + "tokens_seen": 598867968 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004134302908726179, + "loss": 1.3351, + "theoretical_loss": 3.840618887667643, + "tokens_seen": 598933504 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041342026078234706, + "loss": 1.465, + "theoretical_loss": 3.8405750257480253, + "tokens_seen": 598999040 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041341023069207624, + "loss": 1.436, + "theoretical_loss": 3.840531169970566, + "tokens_seen": 599064576 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004134002006018054, + "loss": 1.376, + "theoretical_loss": 3.840487320333733, + "tokens_seen": 599130112 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004133901705115346, + "loss": 1.3969, + "theoretical_loss": 3.8404434768359943, + "tokens_seen": 599195648 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041338014042126384, + "loss": 1.3929, + "theoretical_loss": 3.8403996394758204, + "tokens_seen": 599261184 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041337011033099297, + "loss": 1.3953, + "theoretical_loss": 3.8403558082516804, + "tokens_seen": 599326720 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004133600802407222, + "loss": 1.5206, + "theoretical_loss": 3.8403119831620445, + "tokens_seen": 599392256 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041335005015045133, + "loss": 1.5138, + "theoretical_loss": 3.8402681642053835, + "tokens_seen": 599457792 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041334002006018056, + "loss": 1.2985, + "theoretical_loss": 3.8402243513801695, + "tokens_seen": 599523328 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041332998996990975, + "loss": 1.4164, + "theoretical_loss": 3.8401805446848734, + "tokens_seen": 599588864 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.002151534426957369, + "objective/train/docs_used": 349996, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2134175300598145, + "objective/train/original_loss": 3.2134170532226562, + "objective/train/theoretical_loss": 3.8401367441179683, + "objective/train/tokens_used": 620114400, + "objective/train/value_avg": -0.005855560302734375, + "objective/train/value_loss": 7.351388194365427e-05, + "objective/train/value_max": -0.0004107952117919922, + "objective/train/value_min": -0.04986572265625, + "objective/train/value_reward_corr": 0.5253496017182366, + "objective/train/value_std": 0.00484466552734375, + "objective/train/weight_avg": 0.9819043874740601, + "objective/train/weighted_lm_loss": 3.193875312805176, + "objective/train/weights_max": 1.6150277853012085, + "objective/train/weights_min": 0.3417259454727173, + "theoretical_loss": 3.8401367441179683, + "tokens_seen": 599654400 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004133199598796389, + "loss": 1.507, + "theoretical_loss": 3.8401367441179683, + "tokens_seen": 599654400 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004133099297893681, + "loss": 1.4823, + "theoretical_loss": 3.840092949677927, + "tokens_seen": 599719936 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004132998996990973, + "loss": 1.3219, + "theoretical_loss": 3.840049161363223, + "tokens_seen": 599785472 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041328986960882647, + "loss": 1.4592, + "theoretical_loss": 3.8400053791723305, + "tokens_seen": 599851008 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004132798395185557, + "loss": 1.5553, + "theoretical_loss": 3.839961603103725, + "tokens_seen": 599916544 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041326980942828483, + "loss": 1.3544, + "theoretical_loss": 3.83991783315588, + "tokens_seen": 599982080 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041325977933801407, + "loss": 1.4926, + "theoretical_loss": 3.8398740693272733, + "tokens_seen": 600047616 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041324974924774325, + "loss": 1.3292, + "theoretical_loss": 3.8398303116163794, + "tokens_seen": 600113152 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041323971915747243, + "loss": 1.3816, + "theoretical_loss": 3.8397865600216763, + "tokens_seen": 600178688 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004132296890672016, + "loss": 1.5957, + "theoretical_loss": 3.8397428145416406, + "tokens_seen": 600244224 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004132196589769308, + "loss": 1.3518, + "theoretical_loss": 3.8396990751747513, + "tokens_seen": 600309760 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041320962888665997, + "loss": 1.3327, + "theoretical_loss": 3.839655341919486, + "tokens_seen": 600375296 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004131995987963892, + "loss": 1.5223, + "theoretical_loss": 3.8396116147743236, + "tokens_seen": 600440832 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041318956870611833, + "loss": 1.347, + "theoretical_loss": 3.8395678937377444, + "tokens_seen": 600506368 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041317953861584757, + "loss": 1.4996, + "theoretical_loss": 3.839524178808228, + "tokens_seen": 600571904 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004131695085255767, + "loss": 1.3465, + "theoretical_loss": 3.839480469984256, + "tokens_seen": 600637440 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041315947843530593, + "loss": 1.3817, + "theoretical_loss": 3.839436767264308, + "tokens_seen": 600702976 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004131494483450351, + "loss": 1.2754, + "theoretical_loss": 3.8393930706468673, + "tokens_seen": 600768512 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004131394182547643, + "loss": 1.4676, + "theoretical_loss": 3.839349380130415, + "tokens_seen": 600834048 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004131293881644935, + "loss": 1.383, + "theoretical_loss": 3.839305695713435, + "tokens_seen": 600899584 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004131193580742227, + "loss": 1.3419, + "theoretical_loss": 3.83926201739441, + "tokens_seen": 600965120 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041310932798395184, + "loss": 1.4443, + "theoretical_loss": 3.8392183451718243, + "tokens_seen": 601030656 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004130992978936811, + "loss": 1.2729, + "theoretical_loss": 3.839174679044162, + "tokens_seen": 601096192 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004130892678034102, + "loss": 1.3299, + "theoretical_loss": 3.839131019009908, + "tokens_seen": 601161728 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041307923771313944, + "loss": 1.4002, + "theoretical_loss": 3.8390873650675488, + "tokens_seen": 601227264 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.0006775893270969391, + "objective/train/docs_used": 350686, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6989638805389404, + "objective/train/original_loss": 2.6989634037017822, + "objective/train/theoretical_loss": 3.839043717215569, + "objective/train/tokens_used": 621752800, + "objective/train/value_avg": -0.004558563232421875, + "objective/train/value_loss": 7.221358828246593e-05, + "objective/train/value_max": -0.0003859996795654297, + "objective/train/value_min": -0.0699462890625, + "objective/train/value_reward_corr": 0.3935164382062976, + "objective/train/value_std": 0.0040283203125, + "objective/train/weight_avg": 0.9965484738349915, + "objective/train/weighted_lm_loss": 2.6946287155151367, + "objective/train/weights_max": 1.9414830207824707, + "objective/train/weights_min": 0.45705971121788025, + "theoretical_loss": 3.839043717215569, + "tokens_seen": 601292800 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004130692076228686, + "loss": 1.3989, + "theoretical_loss": 3.839043717215569, + "tokens_seen": 601292800 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004130591775325978, + "loss": 1.3418, + "theoretical_loss": 3.839000075452457, + "tokens_seen": 601358336 + }, + { + "epoch": 0.18, + "learning_rate": 0.000413049147442327, + "loss": 1.4402, + "theoretical_loss": 3.8389564397766986, + "tokens_seen": 601423872 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041303911735205616, + "loss": 1.4171, + "theoretical_loss": 3.838912810186782, + "tokens_seen": 601489408 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041302908726178534, + "loss": 1.3943, + "theoretical_loss": 3.8388691866811957, + "tokens_seen": 601554944 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004130190571715146, + "loss": 1.3492, + "theoretical_loss": 3.838825569258428, + "tokens_seen": 601620480 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004130090270812437, + "loss": 1.3844, + "theoretical_loss": 3.8387819579169684, + "tokens_seen": 601686016 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041299899699097294, + "loss": 1.4261, + "theoretical_loss": 3.838738352655307, + "tokens_seen": 601751552 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041298896690070207, + "loss": 1.4107, + "theoretical_loss": 3.838694753471934, + "tokens_seen": 601817088 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004129789368104313, + "loss": 1.376, + "theoretical_loss": 3.838651160365341, + "tokens_seen": 601882624 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004129689067201605, + "loss": 1.4662, + "theoretical_loss": 3.838607573334018, + "tokens_seen": 601948160 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041295887662988966, + "loss": 1.3628, + "theoretical_loss": 3.838563992376459, + "tokens_seen": 602013696 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041294884653961884, + "loss": 1.3614, + "theoretical_loss": 3.8385204174911554, + "tokens_seen": 602079232 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004129388164493481, + "loss": 1.3629, + "theoretical_loss": 3.8384768486765997, + "tokens_seen": 602144768 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004129287863590772, + "loss": 1.3551, + "theoretical_loss": 3.838433285931287, + "tokens_seen": 602210304 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041291875626880644, + "loss": 1.4026, + "theoretical_loss": 3.83838972925371, + "tokens_seen": 602275840 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004129087261785356, + "loss": 1.4475, + "theoretical_loss": 3.8383461786423654, + "tokens_seen": 602341376 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004128986960882648, + "loss": 1.3048, + "theoretical_loss": 3.8383026340957462, + "tokens_seen": 602406912 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041288866599799404, + "loss": 1.3824, + "theoretical_loss": 3.83825909561235, + "tokens_seen": 602472448 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041287863590772317, + "loss": 1.3569, + "theoretical_loss": 3.838215563190672, + "tokens_seen": 602537984 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004128686058174524, + "loss": 1.5042, + "theoretical_loss": 3.83817203682921, + "tokens_seen": 602603520 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041285857572718153, + "loss": 1.3851, + "theoretical_loss": 3.83812851652646, + "tokens_seen": 602669056 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041284854563691076, + "loss": 1.3944, + "theoretical_loss": 3.8380850022809216, + "tokens_seen": 602734592 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041283851554663995, + "loss": 1.3612, + "theoretical_loss": 3.838041494091093, + "tokens_seen": 602800128 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004128284854563691, + "loss": 1.3936, + "theoretical_loss": 3.8379979919554716, + "tokens_seen": 602865664 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": 0.00036190127138979733, + "objective/train/docs_used": 352047, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.903890371322632, + "objective/train/original_loss": 2.90389084815979, + "objective/train/theoretical_loss": 3.837954495872559, + "objective/train/tokens_used": 623391200, + "objective/train/value_avg": -0.00579071044921875, + "objective/train/value_loss": 5.543830047827214e-05, + "objective/train/value_max": -0.0004222393035888672, + "objective/train/value_min": -0.0880126953125, + "objective/train/value_reward_corr": 0.38771536001142, + "objective/train/value_std": 0.0053253173828125, + "objective/train/weight_avg": 1.006245732307434, + "objective/train/weighted_lm_loss": 2.9230806827545166, + "objective/train/weights_max": 2.231675148010254, + "objective/train/weights_min": 0.2208838313817978, + "theoretical_loss": 3.837954495872559, + "tokens_seen": 602931200 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004128184553660983, + "loss": 1.396, + "theoretical_loss": 3.837954495872559, + "tokens_seen": 602931200 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004128084252758275, + "loss": 1.3391, + "theoretical_loss": 3.837911005840854, + "tokens_seen": 602996736 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041279839518555667, + "loss": 1.3655, + "theoretical_loss": 3.8378675218588576, + "tokens_seen": 603062272 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004127883650952859, + "loss": 1.4547, + "theoretical_loss": 3.837824043925071, + "tokens_seen": 603127808 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041277833500501503, + "loss": 1.4575, + "theoretical_loss": 3.837780572037995, + "tokens_seen": 603193344 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041276830491474427, + "loss": 1.4179, + "theoretical_loss": 3.8377371061961343, + "tokens_seen": 603258880 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041275827482447345, + "loss": 1.476, + "theoretical_loss": 3.837693646397989, + "tokens_seen": 603324416 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041274824473420263, + "loss": 1.4962, + "theoretical_loss": 3.8376501926420636, + "tokens_seen": 603389952 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004127382146439318, + "loss": 1.425, + "theoretical_loss": 3.8376067449268616, + "tokens_seen": 603455488 + }, + { + "epoch": 0.18, + "learning_rate": 0.000412728184553661, + "loss": 1.4187, + "theoretical_loss": 3.837563303250888, + "tokens_seen": 603521024 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041271815446339017, + "loss": 1.3577, + "theoretical_loss": 3.837519867612647, + "tokens_seen": 603586560 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004127081243731194, + "loss": 1.4168, + "theoretical_loss": 3.8374764380106434, + "tokens_seen": 603652096 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041269809428284854, + "loss": 1.3761, + "theoretical_loss": 3.837433014443385, + "tokens_seen": 603717632 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041268806419257777, + "loss": 1.3123, + "theoretical_loss": 3.837389596909377, + "tokens_seen": 603783168 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004126780341023069, + "loss": 1.3592, + "theoretical_loss": 3.8373461854071262, + "tokens_seen": 603848704 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041266800401203613, + "loss": 1.4213, + "theoretical_loss": 3.837302779935141, + "tokens_seen": 603914240 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004126579739217653, + "loss": 1.4893, + "theoretical_loss": 3.837259380491929, + "tokens_seen": 603979776 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004126479438314945, + "loss": 1.368, + "theoretical_loss": 3.8372159870759988, + "tokens_seen": 604045312 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004126379137412237, + "loss": 1.4017, + "theoretical_loss": 3.83717259968586, + "tokens_seen": 604110848 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004126278836509529, + "loss": 1.4498, + "theoretical_loss": 3.8371292183200216, + "tokens_seen": 604176384 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041261785356068204, + "loss": 1.4946, + "theoretical_loss": 3.837085842976994, + "tokens_seen": 604241920 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004126078234704113, + "loss": 1.4834, + "theoretical_loss": 3.837042473655288, + "tokens_seen": 604307456 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004125977933801404, + "loss": 1.4043, + "theoretical_loss": 3.836999110353415, + "tokens_seen": 604372992 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041258776328986964, + "loss": 1.46, + "theoretical_loss": 3.836955753069887, + "tokens_seen": 604438528 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004125777331995988, + "loss": 1.4601, + "theoretical_loss": 3.8369124018032155, + "tokens_seen": 604504064 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.0007791193784214556, + "objective/train/docs_used": 352465, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.015281915664673, + "objective/train/original_loss": 3.0152814388275146, + "objective/train/theoretical_loss": 3.8368690565519143, + "objective/train/tokens_used": 625029600, + "objective/train/value_avg": -0.005126953125, + "objective/train/value_loss": 6.628908886341378e-05, + "objective/train/value_max": -0.00040459632873535156, + "objective/train/value_min": -0.053192138671875, + "objective/train/value_reward_corr": 0.3356659163482802, + "objective/train/value_std": 0.00384521484375, + "objective/train/weight_avg": 0.9953336715698242, + "objective/train/weighted_lm_loss": 3.00372052192688, + "objective/train/weights_max": 1.4833263158798218, + "objective/train/weights_min": 0.46762049198150635, + "theoretical_loss": 3.8368690565519143, + "tokens_seen": 604569600 + }, + { + "epoch": 0.18, + "learning_rate": 0.000412567703109328, + "loss": 1.3878, + "theoretical_loss": 3.8368690565519143, + "tokens_seen": 604569600 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004125576730190572, + "loss": 1.4024, + "theoretical_loss": 3.836825717314496, + "tokens_seen": 604635136 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041254764292878636, + "loss": 1.5651, + "theoretical_loss": 3.8367823840894753, + "tokens_seen": 604700672 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041253761283851554, + "loss": 1.4159, + "theoretical_loss": 3.8367390568753654, + "tokens_seen": 604766208 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004125275827482448, + "loss": 1.4891, + "theoretical_loss": 3.8366957356706832, + "tokens_seen": 604831744 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004125175526579739, + "loss": 1.4268, + "theoretical_loss": 3.836652420473942, + "tokens_seen": 604897280 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041250752256770314, + "loss": 1.3535, + "theoretical_loss": 3.8366091112836593, + "tokens_seen": 604962816 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041249749247743227, + "loss": 1.4815, + "theoretical_loss": 3.836565808098351, + "tokens_seen": 605028352 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004124874623871615, + "loss": 1.5236, + "theoretical_loss": 3.8365225109165344, + "tokens_seen": 605093888 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004124774322968907, + "loss": 1.3194, + "theoretical_loss": 3.8364792197367272, + "tokens_seen": 605159424 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041246740220661986, + "loss": 1.5565, + "theoretical_loss": 3.8364359345574472, + "tokens_seen": 605224960 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041245737211634904, + "loss": 1.4753, + "theoretical_loss": 3.8363926553772134, + "tokens_seen": 605290496 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004124473420260783, + "loss": 1.4185, + "theoretical_loss": 3.8363493821945447, + "tokens_seen": 605356032 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004124373119358074, + "loss": 1.4351, + "theoretical_loss": 3.8363061150079614, + "tokens_seen": 605421568 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041242728184553664, + "loss": 1.3688, + "theoretical_loss": 3.836262853815983, + "tokens_seen": 605487104 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041241725175526577, + "loss": 1.4709, + "theoretical_loss": 3.83621959861713, + "tokens_seen": 605552640 + }, + { + "epoch": 0.18, + "learning_rate": 0.000412407221664995, + "loss": 1.522, + "theoretical_loss": 3.8361763494099246, + "tokens_seen": 605618176 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004123971915747242, + "loss": 1.3608, + "theoretical_loss": 3.836133106192888, + "tokens_seen": 605683712 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041238716148445337, + "loss": 1.4366, + "theoretical_loss": 3.8360898689645433, + "tokens_seen": 605749248 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041237713139418255, + "loss": 1.4593, + "theoretical_loss": 3.836046637723412, + "tokens_seen": 605814784 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041236710130391173, + "loss": 1.3373, + "theoretical_loss": 3.8360034124680187, + "tokens_seen": 605880320 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004123570712136409, + "loss": 1.3402, + "theoretical_loss": 3.8359601931968865, + "tokens_seen": 605945856 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041234704112337015, + "loss": 1.4378, + "theoretical_loss": 3.83591697990854, + "tokens_seen": 606011392 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041233701103309927, + "loss": 1.278, + "theoretical_loss": 3.835873772601505, + "tokens_seen": 606076928 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004123269809428285, + "loss": 1.356, + "theoretical_loss": 3.8358305712743057, + "tokens_seen": 606142464 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.0008668186492286623, + "objective/train/docs_used": 353634, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9781720638275146, + "objective/train/original_loss": 2.9781720638275146, + "objective/train/theoretical_loss": 3.8357873759254693, + "objective/train/tokens_used": 626668000, + "objective/train/value_avg": -0.0065155029296875, + "objective/train/value_loss": 0.00010502720397198573, + "objective/train/value_max": -0.00028252601623535156, + "objective/train/value_min": -0.1209716796875, + "objective/train/value_reward_corr": 0.5798390780509668, + "objective/train/value_std": 0.006946563720703125, + "objective/train/weight_avg": 0.9960699677467346, + "objective/train/weighted_lm_loss": 2.954066753387451, + "objective/train/weights_max": 2.619027853012085, + "objective/train/weights_min": 0.254073828458786, + "theoretical_loss": 3.8357873759254693, + "tokens_seen": 606208000 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041231695085255763, + "loss": 1.3734, + "theoretical_loss": 3.8357873759254693, + "tokens_seen": 606208000 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041230692076228687, + "loss": 1.396, + "theoretical_loss": 3.835744186553521, + "tokens_seen": 606273536 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041229689067201605, + "loss": 1.4028, + "theoretical_loss": 3.835701003156989, + "tokens_seen": 606339072 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041228686058174523, + "loss": 1.3294, + "theoretical_loss": 3.8356578257344003, + "tokens_seen": 606404608 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004122768304914744, + "loss": 1.4977, + "theoretical_loss": 3.8356146542842833, + "tokens_seen": 606470144 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041226680040120365, + "loss": 1.3422, + "theoretical_loss": 3.835571488805166, + "tokens_seen": 606535680 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004122567703109328, + "loss": 1.4746, + "theoretical_loss": 3.835528329295578, + "tokens_seen": 606601216 + }, + { + "epoch": 0.18, + "learning_rate": 0.000412246740220662, + "loss": 1.4335, + "theoretical_loss": 3.835485175754049, + "tokens_seen": 606666752 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041223671013039114, + "loss": 1.2712, + "theoretical_loss": 3.8354420281791093, + "tokens_seen": 606732288 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004122266800401204, + "loss": 1.5645, + "theoretical_loss": 3.8353988865692887, + "tokens_seen": 606797824 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041221664994984955, + "loss": 1.4456, + "theoretical_loss": 3.835355750923119, + "tokens_seen": 606863360 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041220661985957874, + "loss": 1.5141, + "theoretical_loss": 3.835312621239132, + "tokens_seen": 606928896 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004121965897693079, + "loss": 1.4134, + "theoretical_loss": 3.8352694975158608, + "tokens_seen": 606994432 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004121865596790371, + "loss": 1.464, + "theoretical_loss": 3.8352263797518367, + "tokens_seen": 607059968 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004121765295887663, + "loss": 1.4904, + "theoretical_loss": 3.8351832679455935, + "tokens_seen": 607125504 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004121664994984955, + "loss": 1.4713, + "theoretical_loss": 3.835140162095665, + "tokens_seen": 607191040 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004121564694082247, + "loss": 1.3631, + "theoretical_loss": 3.8350970622005858, + "tokens_seen": 607256576 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004121464393179539, + "loss": 1.4442, + "theoretical_loss": 3.8350539682588902, + "tokens_seen": 607322112 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004121364092276831, + "loss": 1.4347, + "theoretical_loss": 3.835010880269114, + "tokens_seen": 607387648 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041212637913741224, + "loss": 1.449, + "theoretical_loss": 3.834967798229793, + "tokens_seen": 607453184 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004121163490471415, + "loss": 1.3977, + "theoretical_loss": 3.834924722139464, + "tokens_seen": 607518720 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004121063189568706, + "loss": 1.4464, + "theoretical_loss": 3.834881651996663, + "tokens_seen": 607584256 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041209628886659984, + "loss": 1.2961, + "theoretical_loss": 3.8348385877999283, + "tokens_seen": 607649792 + }, + { + "epoch": 0.18, + "learning_rate": 0.000412086258776329, + "loss": 1.5806, + "theoretical_loss": 3.8347955295477973, + "tokens_seen": 607715328 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004120762286860582, + "loss": 1.5163, + "theoretical_loss": 3.834752477238809, + "tokens_seen": 607780864 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": -0.0004724613390862942, + "objective/train/docs_used": 354263, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5432941913604736, + "objective/train/original_loss": 2.5432939529418945, + "objective/train/theoretical_loss": 3.8347094308715017, + "objective/train/tokens_used": 628306400, + "objective/train/value_avg": -0.0047149658203125, + "objective/train/value_loss": 3.8290505472104996e-05, + "objective/train/value_max": -0.00031495094299316406, + "objective/train/value_min": -0.0592041015625, + "objective/train/value_reward_corr": 0.6059013258047012, + "objective/train/value_std": 0.00484466552734375, + "objective/train/weight_avg": 0.9971039891242981, + "objective/train/weighted_lm_loss": 2.543642282485962, + "objective/train/weights_max": 1.603647232055664, + "objective/train/weights_min": 0.6036920547485352, + "theoretical_loss": 3.8347094308715017, + "tokens_seen": 607846400 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004120661985957874, + "loss": 1.368, + "theoretical_loss": 3.8347094308715017, + "tokens_seen": 607846400 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041205616850551656, + "loss": 1.4786, + "theoretical_loss": 3.834666390444415, + "tokens_seen": 607911936 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041204613841524574, + "loss": 1.45, + "theoretical_loss": 3.8346233559560896, + "tokens_seen": 607977472 + }, + { + "epoch": 0.18, + "learning_rate": 0.000412036108324975, + "loss": 1.4088, + "theoretical_loss": 3.8345803274050656, + "tokens_seen": 608043008 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004120260782347041, + "loss": 1.4472, + "theoretical_loss": 3.834537304789884, + "tokens_seen": 608108544 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041201604814443334, + "loss": 1.342, + "theoretical_loss": 3.834494288109086, + "tokens_seen": 608174080 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041200601805416247, + "loss": 1.3387, + "theoretical_loss": 3.8344512773612154, + "tokens_seen": 608239616 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004119959879638917, + "loss": 1.5113, + "theoretical_loss": 3.8344082725448123, + "tokens_seen": 608305152 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004119859578736209, + "loss": 1.3422, + "theoretical_loss": 3.8343652736584213, + "tokens_seen": 608370688 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041197592778335006, + "loss": 1.4595, + "theoretical_loss": 3.8343222807005866, + "tokens_seen": 608436224 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041196589769307924, + "loss": 1.4738, + "theoretical_loss": 3.8342792936698507, + "tokens_seen": 608501760 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004119558676028085, + "loss": 1.5493, + "theoretical_loss": 3.8342363125647587, + "tokens_seen": 608567296 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004119458375125376, + "loss": 1.2937, + "theoretical_loss": 3.834193337383857, + "tokens_seen": 608632832 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041193580742226684, + "loss": 1.3058, + "theoretical_loss": 3.83415036812569, + "tokens_seen": 608698368 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041192577733199597, + "loss": 1.5839, + "theoretical_loss": 3.8341074047888046, + "tokens_seen": 608763904 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004119157472417252, + "loss": 1.4919, + "theoretical_loss": 3.8340644473717465, + "tokens_seen": 608829440 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004119057171514544, + "loss": 1.4749, + "theoretical_loss": 3.8340214958730643, + "tokens_seen": 608894976 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041189568706118357, + "loss": 1.4822, + "theoretical_loss": 3.833978550291305, + "tokens_seen": 608960512 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041188565697091275, + "loss": 1.472, + "theoretical_loss": 3.833935610625017, + "tokens_seen": 609026048 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041187562688064193, + "loss": 1.3882, + "theoretical_loss": 3.8338926768727486, + "tokens_seen": 609091584 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004118655967903711, + "loss": 1.4054, + "theoretical_loss": 3.8338497490330505, + "tokens_seen": 609157120 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041185556670010035, + "loss": 1.3039, + "theoretical_loss": 3.8338068271044703, + "tokens_seen": 609222656 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041184553660982947, + "loss": 1.3893, + "theoretical_loss": 3.83376391108556, + "tokens_seen": 609288192 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004118355065195587, + "loss": 1.525, + "theoretical_loss": 3.8337210009748697, + "tokens_seen": 609353728 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041182547642928783, + "loss": 1.4093, + "theoretical_loss": 3.833678096770951, + "tokens_seen": 609419264 + }, + { + "epoch": 0.18, + "objective/train/advantage_avg": 0.00012119229359086603, + "objective/train/docs_used": 355721, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4365923404693604, + "objective/train/original_loss": 2.436591863632202, + "objective/train/theoretical_loss": 3.833635198472356, + "objective/train/tokens_used": 629944800, + "objective/train/value_avg": -0.005245208740234375, + "objective/train/value_loss": 5.155423059477471e-05, + "objective/train/value_max": -0.0002715587615966797, + "objective/train/value_min": -0.08062744140625, + "objective/train/value_reward_corr": 0.28839641878431543, + "objective/train/value_std": 0.003986358642578125, + "objective/train/weight_avg": 1.0036152601242065, + "objective/train/weighted_lm_loss": 2.4475293159484863, + "objective/train/weights_max": 1.591707468032837, + "objective/train/weights_min": 0.2779873311519623, + "theoretical_loss": 3.833635198472356, + "tokens_seen": 609484800 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041181544633901707, + "loss": 1.432, + "theoretical_loss": 3.833635198472356, + "tokens_seen": 609484800 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041180541624874625, + "loss": 1.5357, + "theoretical_loss": 3.8335923060776365, + "tokens_seen": 609550336 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041179538615847543, + "loss": 1.4168, + "theoretical_loss": 3.8335494195853452, + "tokens_seen": 609615872 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004117853560682046, + "loss": 1.4199, + "theoretical_loss": 3.833506538994036, + "tokens_seen": 609681408 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041177532597793385, + "loss": 1.5568, + "theoretical_loss": 3.8334636643022626, + "tokens_seen": 609746944 + }, + { + "epoch": 0.18, + "learning_rate": 0.000411765295887663, + "loss": 1.3606, + "theoretical_loss": 3.8334207955085793, + "tokens_seen": 609812480 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004117552657973922, + "loss": 1.5544, + "theoretical_loss": 3.833377932611541, + "tokens_seen": 609878016 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041174523570712134, + "loss": 1.4444, + "theoretical_loss": 3.8333350756097033, + "tokens_seen": 609943552 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004117352056168506, + "loss": 1.5175, + "theoretical_loss": 3.833292224501622, + "tokens_seen": 610009088 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041172517552657975, + "loss": 1.4089, + "theoretical_loss": 3.833249379285853, + "tokens_seen": 610074624 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041171514543630894, + "loss": 1.3587, + "theoretical_loss": 3.833206539960954, + "tokens_seen": 610140160 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004117051153460381, + "loss": 1.3832, + "theoretical_loss": 3.8331637065254824, + "tokens_seen": 610205696 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004116950852557673, + "loss": 1.3754, + "theoretical_loss": 3.8331208789779954, + "tokens_seen": 610271232 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004116850551654965, + "loss": 1.4523, + "theoretical_loss": 3.833078057317052, + "tokens_seen": 610336768 + }, + { + "epoch": 0.18, + "learning_rate": 0.0004116750250752257, + "loss": 1.4115, + "theoretical_loss": 3.8330352415412117, + "tokens_seen": 610402304 + }, + { + "epoch": 0.18, + "learning_rate": 0.00041166499498495484, + "loss": 1.4106, + "theoretical_loss": 3.832992431649033, + "tokens_seen": 610467840 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004116549648946841, + "loss": 1.4026, + "theoretical_loss": 3.8329496276390764, + "tokens_seen": 610533376 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004116449348044132, + "loss": 1.3668, + "theoretical_loss": 3.832906829509902, + "tokens_seen": 610598912 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041163490471414244, + "loss": 1.3818, + "theoretical_loss": 3.832864037260071, + "tokens_seen": 610664448 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004116248746238716, + "loss": 1.4271, + "theoretical_loss": 3.832821250888145, + "tokens_seen": 610729984 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004116148445336008, + "loss": 1.3449, + "theoretical_loss": 3.832778470392686, + "tokens_seen": 610795520 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041160481444333, + "loss": 1.433, + "theoretical_loss": 3.8327356957722563, + "tokens_seen": 610861056 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004115947843530592, + "loss": 1.4328, + "theoretical_loss": 3.8326929270254193, + "tokens_seen": 610926592 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041158475426278834, + "loss": 1.3445, + "theoretical_loss": 3.8326501641507376, + "tokens_seen": 610992128 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004115747241725176, + "loss": 1.4461, + "theoretical_loss": 3.832607407146777, + "tokens_seen": 611057664 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.0011835522018373013, + "objective/train/docs_used": 356348, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.895150899887085, + "objective/train/original_loss": 2.895151376724243, + "objective/train/theoretical_loss": 3.8325646560121003, + "objective/train/tokens_used": 631583200, + "objective/train/value_avg": -0.005405426025390625, + "objective/train/value_loss": 5.338992195902392e-05, + "objective/train/value_max": -0.0005130767822265625, + "objective/train/value_min": -0.08404541015625, + "objective/train/value_reward_corr": 0.4496619571507148, + "objective/train/value_std": 0.004497528076171875, + "objective/train/weight_avg": 0.9907354116439819, + "objective/train/weighted_lm_loss": 2.8775503635406494, + "objective/train/weights_max": 2.3174190521240234, + "objective/train/weights_min": 0.474260151386261, + "theoretical_loss": 3.8325646560121003, + "tokens_seen": 611123200 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004115646940822467, + "loss": 1.5363, + "theoretical_loss": 3.8325646560121003, + "tokens_seen": 611123200 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041155466399197594, + "loss": 1.3141, + "theoretical_loss": 3.832521910745273, + "tokens_seen": 611188736 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004115446339017051, + "loss": 1.4842, + "theoretical_loss": 3.8324791713448603, + "tokens_seen": 611254272 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004115346038114343, + "loss": 1.3796, + "theoretical_loss": 3.83243643780943, + "tokens_seen": 611319808 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004115245737211635, + "loss": 1.4834, + "theoretical_loss": 3.8323937101375467, + "tokens_seen": 611385344 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041151454363089267, + "loss": 1.4451, + "theoretical_loss": 3.8323509883277778, + "tokens_seen": 611450880 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041150451354062185, + "loss": 1.389, + "theoretical_loss": 3.8323082723786914, + "tokens_seen": 611516416 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004114944834503511, + "loss": 1.3192, + "theoretical_loss": 3.832265562288856, + "tokens_seen": 611581952 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004114844533600802, + "loss": 1.3617, + "theoretical_loss": 3.8322228580568387, + "tokens_seen": 611647488 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041147442326980944, + "loss": 1.3965, + "theoretical_loss": 3.8321801596812097, + "tokens_seen": 611713024 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041146439317953857, + "loss": 1.4503, + "theoretical_loss": 3.832137467160538, + "tokens_seen": 611778560 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004114543630892678, + "loss": 1.5145, + "theoretical_loss": 3.8320947804933945, + "tokens_seen": 611844096 + }, + { + "epoch": 0.19, + "learning_rate": 0.000411444332998997, + "loss": 1.4991, + "theoretical_loss": 3.8320520996783483, + "tokens_seen": 611909632 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041143430290872617, + "loss": 1.3803, + "theoretical_loss": 3.832009424713972, + "tokens_seen": 611975168 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041142427281845535, + "loss": 1.4483, + "theoretical_loss": 3.831966755598837, + "tokens_seen": 612040704 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004114142427281846, + "loss": 1.3323, + "theoretical_loss": 3.831924092331514, + "tokens_seen": 612106240 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041140421263791377, + "loss": 1.4738, + "theoretical_loss": 3.831881434910577, + "tokens_seen": 612171776 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041139418254764295, + "loss": 1.4117, + "theoretical_loss": 3.8318387833345984, + "tokens_seen": 612237312 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041138415245737213, + "loss": 1.353, + "theoretical_loss": 3.8317961376021517, + "tokens_seen": 612302848 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004113741223671013, + "loss": 1.3562, + "theoretical_loss": 3.8317534977118117, + "tokens_seen": 612368384 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041136409227683055, + "loss": 1.4006, + "theoretical_loss": 3.8317108636621526, + "tokens_seen": 612433920 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041135406218655967, + "loss": 1.3987, + "theoretical_loss": 3.831668235451749, + "tokens_seen": 612499456 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004113440320962889, + "loss": 1.4072, + "theoretical_loss": 3.8316256130791775, + "tokens_seen": 612564992 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041133400200601803, + "loss": 1.4286, + "theoretical_loss": 3.8315829965430126, + "tokens_seen": 612630528 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041132397191574727, + "loss": 1.5979, + "theoretical_loss": 3.8315403858418327, + "tokens_seen": 612696064 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": 0.0007694467785768211, + "objective/train/docs_used": 356825, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.709606647491455, + "objective/train/original_loss": 2.709606885910034, + "objective/train/theoretical_loss": 3.831497780974214, + "objective/train/tokens_used": 633221600, + "objective/train/value_avg": -0.00493621826171875, + "objective/train/value_loss": 4.9751022743294016e-05, + "objective/train/value_max": -0.0002892017364501953, + "objective/train/value_min": -0.095703125, + "objective/train/value_reward_corr": 0.2920276987305115, + "objective/train/value_std": 0.003849029541015625, + "objective/train/weight_avg": 1.0100648403167725, + "objective/train/weighted_lm_loss": 2.741302490234375, + "objective/train/weights_max": 2.603954315185547, + "objective/train/weights_min": 0.34019118547439575, + "theoretical_loss": 3.831497780974214, + "tokens_seen": 612761600 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041131394182547645, + "loss": 1.4145, + "theoretical_loss": 3.831497780974214, + "tokens_seen": 612761600 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041130391173520563, + "loss": 1.4502, + "theoretical_loss": 3.8314551819387335, + "tokens_seen": 612827136 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004112938816449348, + "loss": 1.4062, + "theoretical_loss": 3.8314125887339703, + "tokens_seen": 612892672 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041128385155466405, + "loss": 1.4986, + "theoretical_loss": 3.8313700013585024, + "tokens_seen": 612958208 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004112738214643932, + "loss": 1.6044, + "theoretical_loss": 3.8313274198109095, + "tokens_seen": 613023744 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004112637913741224, + "loss": 1.4265, + "theoretical_loss": 3.8312848440897707, + "tokens_seen": 613089280 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041125376128385154, + "loss": 1.4272, + "theoretical_loss": 3.8312422741936656, + "tokens_seen": 613154816 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004112437311935808, + "loss": 1.4755, + "theoretical_loss": 3.831199710121176, + "tokens_seen": 613220352 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041123370110330995, + "loss": 1.3622, + "theoretical_loss": 3.8311571518708813, + "tokens_seen": 613285888 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041122367101303914, + "loss": 1.356, + "theoretical_loss": 3.8311145994413653, + "tokens_seen": 613351424 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004112136409227683, + "loss": 1.4478, + "theoretical_loss": 3.8310720528312077, + "tokens_seen": 613416960 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004112036108324975, + "loss": 1.4091, + "theoretical_loss": 3.831029512038993, + "tokens_seen": 613482496 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004111935807422267, + "loss": 1.3905, + "theoretical_loss": 3.8309869770633025, + "tokens_seen": 613548032 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004111835506519559, + "loss": 1.4988, + "theoretical_loss": 3.8309444479027217, + "tokens_seen": 613613568 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041117352056168504, + "loss": 1.4234, + "theoretical_loss": 3.8309019245558327, + "tokens_seen": 613679104 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004111634904714143, + "loss": 1.534, + "theoretical_loss": 3.8308594070212214, + "tokens_seen": 613744640 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004111534603811434, + "loss": 1.5083, + "theoretical_loss": 3.830816895297472, + "tokens_seen": 613810176 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041114343029087264, + "loss": 1.4315, + "theoretical_loss": 3.830774389383171, + "tokens_seen": 613875712 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004111334002006018, + "loss": 1.4195, + "theoretical_loss": 3.830731889276904, + "tokens_seen": 613941248 + }, + { + "epoch": 0.19, + "learning_rate": 0.000411123370110331, + "loss": 1.3798, + "theoretical_loss": 3.8306893949772567, + "tokens_seen": 614006784 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004111133400200602, + "loss": 1.492, + "theoretical_loss": 3.830646906482817, + "tokens_seen": 614072320 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004111033099297894, + "loss": 1.4596, + "theoretical_loss": 3.8306044237921726, + "tokens_seen": 614137856 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041109327983951854, + "loss": 1.4586, + "theoretical_loss": 3.830561946903911, + "tokens_seen": 614203392 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004110832497492478, + "loss": 1.3982, + "theoretical_loss": 3.8305194758166206, + "tokens_seen": 614268928 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004110732196589769, + "loss": 1.4799, + "theoretical_loss": 3.830477010528891, + "tokens_seen": 614334464 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": 0.00040838521090336144, + "objective/train/docs_used": 357491, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5830750465393066, + "objective/train/original_loss": 2.5830750465393066, + "objective/train/theoretical_loss": 3.8304345510393105, + "objective/train/tokens_used": 634860000, + "objective/train/value_avg": -0.005023956298828125, + "objective/train/value_loss": 6.467890489147976e-05, + "objective/train/value_max": -0.0004954338073730469, + "objective/train/value_min": -0.063720703125, + "objective/train/value_reward_corr": 0.40609285184092786, + "objective/train/value_std": 0.003993988037109375, + "objective/train/weight_avg": 1.00700843334198, + "objective/train/weighted_lm_loss": 2.5919137001037598, + "objective/train/weights_max": 1.7862870693206787, + "objective/train/weights_min": 0.28271418809890747, + "theoretical_loss": 3.8304345510393105, + "tokens_seen": 614400000 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041106318956870614, + "loss": 1.3466, + "theoretical_loss": 3.8304345510393105, + "tokens_seen": 614400000 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004110531594784353, + "loss": 1.5388, + "theoretical_loss": 3.830392097346471, + "tokens_seen": 614465536 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004110431293881645, + "loss": 1.4483, + "theoretical_loss": 3.8303496494489613, + "tokens_seen": 614531072 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004110330992978937, + "loss": 1.4032, + "theoretical_loss": 3.830307207345373, + "tokens_seen": 614596608 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041102306920762287, + "loss": 1.4992, + "theoretical_loss": 3.8302647710342983, + "tokens_seen": 614662144 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041101303911735205, + "loss": 1.4494, + "theoretical_loss": 3.830222340514328, + "tokens_seen": 614727680 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004110030090270813, + "loss": 1.4227, + "theoretical_loss": 3.830179915784054, + "tokens_seen": 614793216 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004109929789368104, + "loss": 1.4232, + "theoretical_loss": 3.830137496842071, + "tokens_seen": 614858752 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041098294884653965, + "loss": 1.3462, + "theoretical_loss": 3.830095083686972, + "tokens_seen": 614924288 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041097291875626877, + "loss": 1.4652, + "theoretical_loss": 3.83005267631735, + "tokens_seen": 614989824 + }, + { + "epoch": 0.19, + "learning_rate": 0.000410962888665998, + "loss": 1.5455, + "theoretical_loss": 3.8300102747318006, + "tokens_seen": 615055360 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004109528585757272, + "loss": 1.5253, + "theoretical_loss": 3.829967878928917, + "tokens_seen": 615120896 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041094282848545637, + "loss": 1.4272, + "theoretical_loss": 3.8299254889072962, + "tokens_seen": 615186432 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041093279839518555, + "loss": 1.3641, + "theoretical_loss": 3.8298831046655337, + "tokens_seen": 615251968 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004109227683049148, + "loss": 1.3389, + "theoretical_loss": 3.8298407262022263, + "tokens_seen": 615317504 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004109127382146439, + "loss": 1.4244, + "theoretical_loss": 3.8297983535159696, + "tokens_seen": 615383040 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041090270812437315, + "loss": 1.3281, + "theoretical_loss": 3.8297559866053614, + "tokens_seen": 615448576 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004108926780341023, + "loss": 1.4943, + "theoretical_loss": 3.8297136254690005, + "tokens_seen": 615514112 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004108826479438315, + "loss": 1.3725, + "theoretical_loss": 3.8296712701054845, + "tokens_seen": 615579648 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004108726178535607, + "loss": 1.4874, + "theoretical_loss": 3.8296289205134117, + "tokens_seen": 615645184 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041086258776328987, + "loss": 1.4046, + "theoretical_loss": 3.8295865766913826, + "tokens_seen": 615710720 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041085255767301905, + "loss": 1.4379, + "theoretical_loss": 3.8295442386379968, + "tokens_seen": 615776256 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041084252758274823, + "loss": 1.3833, + "theoretical_loss": 3.8295019063518536, + "tokens_seen": 615841792 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004108324974924774, + "loss": 1.3202, + "theoretical_loss": 3.8294595798315543, + "tokens_seen": 615907328 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041082246740220665, + "loss": 1.3316, + "theoretical_loss": 3.8294172590757007, + "tokens_seen": 615972864 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.0022922707721590996, + "objective/train/docs_used": 358741, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8150625228881836, + "objective/train/original_loss": 2.8150622844696045, + "objective/train/theoretical_loss": 3.829374944082894, + "objective/train/tokens_used": 636498400, + "objective/train/value_avg": -0.006221771240234375, + "objective/train/value_loss": 9.553904237691313e-05, + "objective/train/value_max": -0.00036835670471191406, + "objective/train/value_min": -0.06292724609375, + "objective/train/value_reward_corr": 0.46005279698345486, + "objective/train/value_std": 0.00518798828125, + "objective/train/weight_avg": 0.9815552234649658, + "objective/train/weighted_lm_loss": 2.7694568634033203, + "objective/train/weights_max": 1.4723329544067383, + "objective/train/weights_min": 0.37550029158592224, + "theoretical_loss": 3.829374944082894, + "tokens_seen": 616038400 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004108124373119358, + "loss": 1.431, + "theoretical_loss": 3.829374944082894, + "tokens_seen": 616038400 + }, + { + "epoch": 0.19, + "learning_rate": 0.000410802407221665, + "loss": 1.3261, + "theoretical_loss": 3.8293326348517365, + "tokens_seen": 616103936 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004107923771313942, + "loss": 1.4618, + "theoretical_loss": 3.8292903313808306, + "tokens_seen": 616169472 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004107823470411234, + "loss": 1.4024, + "theoretical_loss": 3.829248033668781, + "tokens_seen": 616235008 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041077231695085256, + "loss": 1.3909, + "theoretical_loss": 3.8292057417141896, + "tokens_seen": 616300544 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041076228686058174, + "loss": 1.3711, + "theoretical_loss": 3.8291634555156615, + "tokens_seen": 616366080 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004107522567703109, + "loss": 1.3336, + "theoretical_loss": 3.8291211750718013, + "tokens_seen": 616431616 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041074222668004015, + "loss": 1.345, + "theoretical_loss": 3.8290789003812145, + "tokens_seen": 616497152 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004107321965897693, + "loss": 1.5221, + "theoretical_loss": 3.829036631442506, + "tokens_seen": 616562688 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004107221664994985, + "loss": 1.3389, + "theoretical_loss": 3.828994368254283, + "tokens_seen": 616628224 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041071213640922764, + "loss": 1.5368, + "theoretical_loss": 3.8289521108151505, + "tokens_seen": 616693760 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004107021063189569, + "loss": 1.4944, + "theoretical_loss": 3.8289098591237174, + "tokens_seen": 616759296 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041069207622868606, + "loss": 1.4334, + "theoretical_loss": 3.8288676131785895, + "tokens_seen": 616824832 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041068204613841524, + "loss": 1.496, + "theoretical_loss": 3.8288253729783763, + "tokens_seen": 616890368 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004106720160481444, + "loss": 1.4087, + "theoretical_loss": 3.828783138521686, + "tokens_seen": 616955904 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004106619859578736, + "loss": 1.4885, + "theoretical_loss": 3.828740909807128, + "tokens_seen": 617021440 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041065195586760284, + "loss": 1.4962, + "theoretical_loss": 3.8286986868333113, + "tokens_seen": 617086976 + }, + { + "epoch": 0.19, + "learning_rate": 0.000410641925777332, + "loss": 1.3238, + "theoretical_loss": 3.8286564695988456, + "tokens_seen": 617152512 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004106318956870612, + "loss": 1.4398, + "theoretical_loss": 3.828614258102342, + "tokens_seen": 617218048 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004106218655967904, + "loss": 1.5079, + "theoretical_loss": 3.8285720523424116, + "tokens_seen": 617283584 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004106118355065196, + "loss": 1.4036, + "theoretical_loss": 3.828529852317666, + "tokens_seen": 617349120 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041060180541624874, + "loss": 1.5272, + "theoretical_loss": 3.8284876580267158, + "tokens_seen": 617414656 + }, + { + "epoch": 0.19, + "learning_rate": 0.000410591775325978, + "loss": 1.563, + "theoretical_loss": 3.8284454694681753, + "tokens_seen": 617480192 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004105817452357071, + "loss": 1.4252, + "theoretical_loss": 3.8284032866406563, + "tokens_seen": 617545728 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041057171514543634, + "loss": 1.4511, + "theoretical_loss": 3.8283611095427723, + "tokens_seen": 617611264 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.0005288867396302521, + "objective/train/docs_used": 359404, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7554526329040527, + "objective/train/original_loss": 2.7554526329040527, + "objective/train/theoretical_loss": 3.8283189381731377, + "objective/train/tokens_used": 638136800, + "objective/train/value_avg": -0.0048065185546875, + "objective/train/value_loss": 4.319595609558746e-05, + "objective/train/value_max": -0.00044083595275878906, + "objective/train/value_min": -0.06512451171875, + "objective/train/value_reward_corr": 0.3861817312284917, + "objective/train/value_std": 0.0037841796875, + "objective/train/weight_avg": 0.996771514415741, + "objective/train/weighted_lm_loss": 2.7509071826934814, + "objective/train/weights_max": 1.7299555540084839, + "objective/train/weights_min": 0.4805246889591217, + "theoretical_loss": 3.8283189381731377, + "tokens_seen": 617676800 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004105616850551655, + "loss": 1.4144, + "theoretical_loss": 3.8283189381731377, + "tokens_seen": 617676800 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004105516549648947, + "loss": 1.4782, + "theoretical_loss": 3.8282767725303666, + "tokens_seen": 617742336 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004105416248746239, + "loss": 1.4732, + "theoretical_loss": 3.8282346126130733, + "tokens_seen": 617807872 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041053159478435307, + "loss": 1.4202, + "theoretical_loss": 3.828192458419874, + "tokens_seen": 617873408 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041052156469408225, + "loss": 1.4696, + "theoretical_loss": 3.8281503099493843, + "tokens_seen": 617938944 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004105115346038115, + "loss": 1.5251, + "theoretical_loss": 3.82810816720022, + "tokens_seen": 618004480 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004105015045135406, + "loss": 1.4731, + "theoretical_loss": 3.828066030170999, + "tokens_seen": 618070016 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041049147442326985, + "loss": 1.4375, + "theoretical_loss": 3.8280238988603363, + "tokens_seen": 618135552 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041048144433299897, + "loss": 1.4583, + "theoretical_loss": 3.8279817732668526, + "tokens_seen": 618201088 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004104714142427282, + "loss": 1.4385, + "theoretical_loss": 3.827939653389164, + "tokens_seen": 618266624 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004104613841524574, + "loss": 1.4199, + "theoretical_loss": 3.8278975392258903, + "tokens_seen": 618332160 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041045135406218657, + "loss": 1.5003, + "theoretical_loss": 3.82785543077565, + "tokens_seen": 618397696 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041044132397191575, + "loss": 1.3591, + "theoretical_loss": 3.827813328037063, + "tokens_seen": 618463232 + }, + { + "epoch": 0.19, + "learning_rate": 0.000410431293881645, + "loss": 1.5254, + "theoretical_loss": 3.8277712310087493, + "tokens_seen": 618528768 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004104212637913741, + "loss": 1.3475, + "theoretical_loss": 3.82772913968933, + "tokens_seen": 618594304 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041041123370110335, + "loss": 1.3919, + "theoretical_loss": 3.827687054077426, + "tokens_seen": 618659840 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004104012036108325, + "loss": 1.3476, + "theoretical_loss": 3.827644974171659, + "tokens_seen": 618725376 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004103911735205617, + "loss": 1.4007, + "theoretical_loss": 3.8276028999706506, + "tokens_seen": 618790912 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004103811434302909, + "loss": 1.4739, + "theoretical_loss": 3.8275608314730234, + "tokens_seen": 618856448 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041037111334002007, + "loss": 1.3645, + "theoretical_loss": 3.8275187686774013, + "tokens_seen": 618921984 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041036108324974925, + "loss": 1.3052, + "theoretical_loss": 3.827476711582406, + "tokens_seen": 618987520 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041035105315947844, + "loss": 1.4937, + "theoretical_loss": 3.8274346601866633, + "tokens_seen": 619053056 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004103410230692076, + "loss": 1.4721, + "theoretical_loss": 3.8273926144887973, + "tokens_seen": 619118592 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041033099297893685, + "loss": 1.5285, + "theoretical_loss": 3.827350574487432, + "tokens_seen": 619184128 + }, + { + "epoch": 0.19, + "learning_rate": 0.000410320962888666, + "loss": 1.4328, + "theoretical_loss": 3.8273085401811935, + "tokens_seen": 619249664 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.0003412722435314208, + "objective/train/docs_used": 360794, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0334064960479736, + "objective/train/original_loss": 3.0334064960479736, + "objective/train/theoretical_loss": 3.8272665115687077, + "objective/train/tokens_used": 639775200, + "objective/train/value_avg": -0.004543304443359375, + "objective/train/value_loss": 7.250312046380714e-05, + "objective/train/value_max": -0.00035834312438964844, + "objective/train/value_min": -0.04351806640625, + "objective/train/value_reward_corr": 0.32233759983756055, + "objective/train/value_std": 0.003574371337890625, + "objective/train/weight_avg": 0.9998151659965515, + "objective/train/weighted_lm_loss": 3.0383431911468506, + "objective/train/weights_max": 1.5442993640899658, + "objective/train/weights_min": 0.14976593852043152, + "theoretical_loss": 3.8272665115687077, + "tokens_seen": 619315200 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004103109327983952, + "loss": 1.4843, + "theoretical_loss": 3.8272665115687077, + "tokens_seen": 619315200 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004103009027081244, + "loss": 1.4, + "theoretical_loss": 3.827224488648601, + "tokens_seen": 619380736 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004102908726178536, + "loss": 1.4326, + "theoretical_loss": 3.8271824714194995, + "tokens_seen": 619446272 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041028084252758276, + "loss": 1.3177, + "theoretical_loss": 3.8271404598800314, + "tokens_seen": 619511808 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041027081243731194, + "loss": 1.4399, + "theoretical_loss": 3.8270984540288238, + "tokens_seen": 619577344 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004102607823470411, + "loss": 1.4622, + "theoretical_loss": 3.8270564538645058, + "tokens_seen": 619642880 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041025075225677035, + "loss": 1.4432, + "theoretical_loss": 3.8270144593857056, + "tokens_seen": 619708416 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004102407221664995, + "loss": 1.5016, + "theoretical_loss": 3.826972470591053, + "tokens_seen": 619773952 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004102306920762287, + "loss": 1.3807, + "theoretical_loss": 3.8269304874791765, + "tokens_seen": 619839488 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041022066198595784, + "loss": 1.4844, + "theoretical_loss": 3.826888510048707, + "tokens_seen": 619905024 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004102106318956871, + "loss": 1.4293, + "theoretical_loss": 3.8268465382982755, + "tokens_seen": 619970560 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041020060180541626, + "loss": 1.4251, + "theoretical_loss": 3.8268045722265125, + "tokens_seen": 620036096 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041019057171514544, + "loss": 1.5657, + "theoretical_loss": 3.8267626118320504, + "tokens_seen": 620101632 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004101805416248746, + "loss": 1.4167, + "theoretical_loss": 3.82672065711352, + "tokens_seen": 620167168 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004101705115346038, + "loss": 1.4241, + "theoretical_loss": 3.826678708069555, + "tokens_seen": 620232704 + }, + { + "epoch": 0.19, + "learning_rate": 0.000410160481444333, + "loss": 1.2871, + "theoretical_loss": 3.8266367646987876, + "tokens_seen": 620298240 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004101504513540622, + "loss": 1.4621, + "theoretical_loss": 3.8265948269998518, + "tokens_seen": 620363776 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041014042126379135, + "loss": 1.4197, + "theoretical_loss": 3.826552894971381, + "tokens_seen": 620429312 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004101303911735206, + "loss": 1.4298, + "theoretical_loss": 3.826510968612011, + "tokens_seen": 620494848 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041012036108324976, + "loss": 1.4776, + "theoretical_loss": 3.8264690479203747, + "tokens_seen": 620560384 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041011033099297894, + "loss": 1.5065, + "theoretical_loss": 3.826427132895109, + "tokens_seen": 620625920 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004101003009027081, + "loss": 1.4369, + "theoretical_loss": 3.8263852235348494, + "tokens_seen": 620691456 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004100902708124373, + "loss": 1.4084, + "theoretical_loss": 3.8263433198382324, + "tokens_seen": 620756992 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004100802407221665, + "loss": 1.3102, + "theoretical_loss": 3.826301421803894, + "tokens_seen": 620822528 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004100702106318957, + "loss": 1.4963, + "theoretical_loss": 3.826259529430472, + "tokens_seen": 620888064 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.00018338988593313843, + "objective/train/docs_used": 361341, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1340506076812744, + "objective/train/original_loss": 3.134049892425537, + "objective/train/theoretical_loss": 3.8262176427166046, + "objective/train/tokens_used": 641413600, + "objective/train/value_avg": -0.005413055419921875, + "objective/train/value_loss": 5.764554953202605e-05, + "objective/train/value_max": -0.00036263465881347656, + "objective/train/value_min": -0.08331298828125, + "objective/train/value_reward_corr": 0.38218622426656834, + "objective/train/value_std": 0.0054473876953125, + "objective/train/weight_avg": 1.0009756088256836, + "objective/train/weighted_lm_loss": 3.137744426727295, + "objective/train/weights_max": 2.0203278064727783, + "objective/train/weights_min": 0.44182372093200684, + "theoretical_loss": 3.8262176427166046, + "tokens_seen": 620953600 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041006018054162485, + "loss": 1.4261, + "theoretical_loss": 3.8262176427166046, + "tokens_seen": 620953600 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004100501504513541, + "loss": 1.2824, + "theoretical_loss": 3.826175761660929, + "tokens_seen": 621019136 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004100401203610832, + "loss": 1.476, + "theoretical_loss": 3.826133886262085, + "tokens_seen": 621084672 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041003009027081245, + "loss": 1.4041, + "theoretical_loss": 3.826092016518711, + "tokens_seen": 621150208 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041002006018054163, + "loss": 1.3925, + "theoretical_loss": 3.826050152429447, + "tokens_seen": 621215744 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004100100300902708, + "loss": 1.467, + "theoretical_loss": 3.8260082939929325, + "tokens_seen": 621281280 + }, + { + "epoch": 0.19, + "learning_rate": 0.00041, + "loss": 1.4341, + "theoretical_loss": 3.825966441207809, + "tokens_seen": 621346816 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040998996990972917, + "loss": 1.4051, + "theoretical_loss": 3.825924594072717, + "tokens_seen": 621412352 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040997993981945835, + "loss": 1.3476, + "theoretical_loss": 3.8258827525862973, + "tokens_seen": 621477888 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004099699097291876, + "loss": 1.5332, + "theoretical_loss": 3.825840916747193, + "tokens_seen": 621543424 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004099598796389167, + "loss": 1.3656, + "theoretical_loss": 3.8257990865540465, + "tokens_seen": 621608960 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040994984954864595, + "loss": 1.4367, + "theoretical_loss": 3.8257572620055003, + "tokens_seen": 621674496 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040993981945837513, + "loss": 1.4805, + "theoretical_loss": 3.8257154431001976, + "tokens_seen": 621740032 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004099297893681043, + "loss": 1.535, + "theoretical_loss": 3.825673629836783, + "tokens_seen": 621805568 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004099197592778335, + "loss": 1.4494, + "theoretical_loss": 3.8256318222139, + "tokens_seen": 621871104 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004099097291875627, + "loss": 1.4471, + "theoretical_loss": 3.8255900202301936, + "tokens_seen": 621936640 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004098996990972919, + "loss": 1.3815, + "theoretical_loss": 3.8255482238843097, + "tokens_seen": 622002176 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004098896690070211, + "loss": 1.5153, + "theoretical_loss": 3.825506433174893, + "tokens_seen": 622067712 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004098796389167503, + "loss": 1.4812, + "theoretical_loss": 3.825464648100591, + "tokens_seen": 622133248 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040986960882647945, + "loss": 1.4341, + "theoretical_loss": 3.8254228686600493, + "tokens_seen": 622198784 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040985957873620864, + "loss": 1.4527, + "theoretical_loss": 3.825381094851915, + "tokens_seen": 622264320 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004098495486459378, + "loss": 1.3743, + "theoretical_loss": 3.8253393266748366, + "tokens_seen": 622329856 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040983951855566705, + "loss": 1.3794, + "theoretical_loss": 3.825297564127461, + "tokens_seen": 622395392 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004098294884653962, + "loss": 1.3492, + "theoretical_loss": 3.825255807208438, + "tokens_seen": 622460928 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004098194583751254, + "loss": 1.4786, + "theoretical_loss": 3.825214055916416, + "tokens_seen": 622526464 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": 0.001000298303551972, + "objective/train/docs_used": 362042, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9639761447906494, + "objective/train/original_loss": 2.9639759063720703, + "objective/train/theoretical_loss": 3.8251723102500437, + "objective/train/tokens_used": 643052000, + "objective/train/value_avg": -0.004291534423828125, + "objective/train/value_loss": 4.87987526867073e-05, + "objective/train/value_max": -0.0003407001495361328, + "objective/train/value_min": -0.0303802490234375, + "objective/train/value_reward_corr": 0.24863282187323088, + "objective/train/value_std": 0.0029544830322265625, + "objective/train/weight_avg": 1.0121382474899292, + "objective/train/weighted_lm_loss": 2.9953410625457764, + "objective/train/weights_max": 1.3481956720352173, + "objective/train/weights_min": 0.13881763815879822, + "theoretical_loss": 3.8251723102500437, + "tokens_seen": 622592000 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004098094282848546, + "loss": 1.3251, + "theoretical_loss": 3.8251723102500437, + "tokens_seen": 622592000 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004097993981945838, + "loss": 1.49, + "theoretical_loss": 3.8251305702079725, + "tokens_seen": 622657536 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040978936810431296, + "loss": 1.4578, + "theoretical_loss": 3.8250888357888515, + "tokens_seen": 622723072 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040977933801404214, + "loss": 1.4698, + "theoretical_loss": 3.8250471069913328, + "tokens_seen": 622788608 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004097693079237713, + "loss": 1.3218, + "theoretical_loss": 3.8250053838140663, + "tokens_seen": 622854144 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040975927783350055, + "loss": 1.4042, + "theoretical_loss": 3.824963666255705, + "tokens_seen": 622919680 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004097492477432297, + "loss": 1.3498, + "theoretical_loss": 3.8249219543149007, + "tokens_seen": 622985216 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004097392176529589, + "loss": 1.462, + "theoretical_loss": 3.824880247990306, + "tokens_seen": 623050752 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040972918756268804, + "loss": 1.4301, + "theoretical_loss": 3.8248385472805744, + "tokens_seen": 623116288 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004097191574724173, + "loss": 1.4808, + "theoretical_loss": 3.8247968521843596, + "tokens_seen": 623181824 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040970912738214646, + "loss": 1.3726, + "theoretical_loss": 3.824755162700315, + "tokens_seen": 623247360 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040969909729187564, + "loss": 1.4023, + "theoretical_loss": 3.8247134788270962, + "tokens_seen": 623312896 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004096890672016048, + "loss": 1.3651, + "theoretical_loss": 3.8246718005633573, + "tokens_seen": 623378432 + }, + { + "epoch": 0.19, + "learning_rate": 0.000409679037111334, + "loss": 1.3987, + "theoretical_loss": 3.8246301279077546, + "tokens_seen": 623443968 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004096690070210632, + "loss": 1.4828, + "theoretical_loss": 3.8245884608589438, + "tokens_seen": 623509504 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004096589769307924, + "loss": 1.3872, + "theoretical_loss": 3.824546799415581, + "tokens_seen": 623575040 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040964894684052155, + "loss": 1.41, + "theoretical_loss": 3.8245051435763235, + "tokens_seen": 623640576 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004096389167502508, + "loss": 1.5479, + "theoretical_loss": 3.824463493339829, + "tokens_seen": 623706112 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040962888665997996, + "loss": 1.433, + "theoretical_loss": 3.8244218487047545, + "tokens_seen": 623771648 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040961885656970914, + "loss": 1.518, + "theoretical_loss": 3.8243802096697586, + "tokens_seen": 623837184 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004096088264794383, + "loss": 1.5158, + "theoretical_loss": 3.8243385762335, + "tokens_seen": 623902720 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004095987963891675, + "loss": 1.3734, + "theoretical_loss": 3.8242969483946387, + "tokens_seen": 623968256 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004095887662988967, + "loss": 1.4322, + "theoretical_loss": 3.8242553261518335, + "tokens_seen": 624033792 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004095787362086259, + "loss": 1.4305, + "theoretical_loss": 3.8242137095037445, + "tokens_seen": 624099328 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040956870611835505, + "loss": 1.445, + "theoretical_loss": 3.824172098449033, + "tokens_seen": 624164864 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.0012944451300427318, + "objective/train/docs_used": 363016, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0887224674224854, + "objective/train/original_loss": 3.0887224674224854, + "objective/train/theoretical_loss": 3.824130492986359, + "objective/train/tokens_used": 644690400, + "objective/train/value_avg": -0.005950927734375, + "objective/train/value_loss": 7.32181069906801e-05, + "objective/train/value_max": -0.0003275871276855469, + "objective/train/value_min": -0.073974609375, + "objective/train/value_reward_corr": 0.44503589629658885, + "objective/train/value_std": 0.0057220458984375, + "objective/train/weight_avg": 0.9903890490531921, + "objective/train/weighted_lm_loss": 3.0638296604156494, + "objective/train/weights_max": 1.647890567779541, + "objective/train/weights_min": 0.13807298243045807, + "theoretical_loss": 3.824130492986359, + "tokens_seen": 624230400 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004095586760280843, + "loss": 1.5146, + "theoretical_loss": 3.824130492986359, + "tokens_seen": 624230400 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004095486459378134, + "loss": 1.4644, + "theoretical_loss": 3.824088893114385, + "tokens_seen": 624295936 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040953861584754265, + "loss": 1.4633, + "theoretical_loss": 3.824047298831773, + "tokens_seen": 624361472 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040952858575727183, + "loss": 1.3625, + "theoretical_loss": 3.8240057101371843, + "tokens_seen": 624427008 + }, + { + "epoch": 0.19, + "learning_rate": 0.000409518555667001, + "loss": 1.4843, + "theoretical_loss": 3.8239641270292832, + "tokens_seen": 624492544 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004095085255767302, + "loss": 1.4796, + "theoretical_loss": 3.823922549506732, + "tokens_seen": 624558080 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040949849548645937, + "loss": 1.443, + "theoretical_loss": 3.8238809775681952, + "tokens_seen": 624623616 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040948846539618855, + "loss": 1.3936, + "theoretical_loss": 3.823839411212337, + "tokens_seen": 624689152 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004094784353059178, + "loss": 1.5173, + "theoretical_loss": 3.8237978504378223, + "tokens_seen": 624754688 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004094684052156469, + "loss": 1.3423, + "theoretical_loss": 3.8237562952433155, + "tokens_seen": 624820224 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040945837512537615, + "loss": 1.3395, + "theoretical_loss": 3.823714745627483, + "tokens_seen": 624885760 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040944834503510533, + "loss": 1.4833, + "theoretical_loss": 3.8236732015889903, + "tokens_seen": 624951296 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004094383149448345, + "loss": 1.4414, + "theoretical_loss": 3.823631663126505, + "tokens_seen": 625016832 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004094282848545637, + "loss": 1.4119, + "theoretical_loss": 3.823590130238693, + "tokens_seen": 625082368 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004094182547642929, + "loss": 1.3513, + "theoretical_loss": 3.823548602924223, + "tokens_seen": 625147904 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040940822467402206, + "loss": 1.5837, + "theoretical_loss": 3.8235070811817615, + "tokens_seen": 625213440 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004093981945837513, + "loss": 1.457, + "theoretical_loss": 3.823465565009978, + "tokens_seen": 625278976 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004093881644934804, + "loss": 1.3633, + "theoretical_loss": 3.8234240544075417, + "tokens_seen": 625344512 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040937813440320965, + "loss": 1.3969, + "theoretical_loss": 3.8233825493731204, + "tokens_seen": 625410048 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004093681043129388, + "loss": 1.4378, + "theoretical_loss": 3.823341049905385, + "tokens_seen": 625475584 + }, + { + "epoch": 0.19, + "learning_rate": 0.000409358074222668, + "loss": 1.4244, + "theoretical_loss": 3.823299556003006, + "tokens_seen": 625541120 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004093480441323972, + "loss": 1.4999, + "theoretical_loss": 3.823258067664653, + "tokens_seen": 625606656 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004093380140421264, + "loss": 1.3712, + "theoretical_loss": 3.823216584888998, + "tokens_seen": 625672192 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040932798395185556, + "loss": 1.4986, + "theoretical_loss": 3.8231751076747127, + "tokens_seen": 625737728 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004093179538615848, + "loss": 1.4423, + "theoretical_loss": 3.8231336360204686, + "tokens_seen": 625803264 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.00013564123946707696, + "objective/train/docs_used": 363672, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.734769821166992, + "objective/train/original_loss": 2.734769344329834, + "objective/train/theoretical_loss": 3.823092169924938, + "objective/train/tokens_used": 646328800, + "objective/train/value_avg": -0.0045623779296875, + "objective/train/value_loss": 4.5378488721325994e-05, + "objective/train/value_max": -0.0003936290740966797, + "objective/train/value_min": -0.06951904296875, + "objective/train/value_reward_corr": 0.3738723347477858, + "objective/train/value_std": 0.003353118896484375, + "objective/train/weight_avg": 1.00083589553833, + "objective/train/weighted_lm_loss": 2.727391004562378, + "objective/train/weights_max": 2.0040905475616455, + "objective/train/weights_min": 0.6634183526039124, + "theoretical_loss": 3.823092169924938, + "tokens_seen": 625868800 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004093079237713139, + "loss": 1.3905, + "theoretical_loss": 3.823092169924938, + "tokens_seen": 625868800 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040929789368104316, + "loss": 1.4291, + "theoretical_loss": 3.8230507093867954, + "tokens_seen": 625934336 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004092878635907723, + "loss": 1.4096, + "theoretical_loss": 3.8230092544047123, + "tokens_seen": 625999872 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004092778335005015, + "loss": 1.4696, + "theoretical_loss": 3.822967804977364, + "tokens_seen": 626065408 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004092678034102307, + "loss": 1.4605, + "theoretical_loss": 3.8229263611034243, + "tokens_seen": 626130944 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004092577733199599, + "loss": 1.4082, + "theoretical_loss": 3.822884922781568, + "tokens_seen": 626196480 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040924774322968906, + "loss": 1.396, + "theoretical_loss": 3.8228434900104706, + "tokens_seen": 626262016 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040923771313941824, + "loss": 1.4253, + "theoretical_loss": 3.8228020627888073, + "tokens_seen": 626327552 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004092276830491474, + "loss": 1.3279, + "theoretical_loss": 3.822760641115255, + "tokens_seen": 626393088 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040921765295887666, + "loss": 1.4274, + "theoretical_loss": 3.8227192249884894, + "tokens_seen": 626458624 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004092076228686058, + "loss": 1.5849, + "theoretical_loss": 3.8226778144071885, + "tokens_seen": 626524160 + }, + { + "epoch": 0.19, + "learning_rate": 0.000409197592778335, + "loss": 1.4942, + "theoretical_loss": 3.8226364093700296, + "tokens_seen": 626589696 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040918756268806415, + "loss": 1.5358, + "theoretical_loss": 3.82259500987569, + "tokens_seen": 626655232 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004091775325977934, + "loss": 1.2832, + "theoretical_loss": 3.822553615922849, + "tokens_seen": 626720768 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040916750250752257, + "loss": 1.3749, + "theoretical_loss": 3.8225122275101846, + "tokens_seen": 626786304 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040915747241725175, + "loss": 1.4716, + "theoretical_loss": 3.8224708446363778, + "tokens_seen": 626851840 + }, + { + "epoch": 0.19, + "learning_rate": 0.000409147442326981, + "loss": 1.4019, + "theoretical_loss": 3.8224294673001062, + "tokens_seen": 626917376 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040913741223671016, + "loss": 1.5066, + "theoretical_loss": 3.8223880955000515, + "tokens_seen": 626982912 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040912738214643934, + "loss": 1.3707, + "theoretical_loss": 3.8223467292348943, + "tokens_seen": 627048448 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004091173520561685, + "loss": 1.4488, + "theoretical_loss": 3.822305368503315, + "tokens_seen": 627113984 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004091073219658977, + "loss": 1.4898, + "theoretical_loss": 3.822264013303996, + "tokens_seen": 627179520 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004090972918756269, + "loss": 1.5978, + "theoretical_loss": 3.822222663635619, + "tokens_seen": 627245056 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004090872617853561, + "loss": 1.4788, + "theoretical_loss": 3.822181319496867, + "tokens_seen": 627310592 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040907723169508525, + "loss": 1.4685, + "theoretical_loss": 3.822139980886422, + "tokens_seen": 627376128 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004090672016048145, + "loss": 1.411, + "theoretical_loss": 3.8220986478029677, + "tokens_seen": 627441664 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.001985265640541911, + "objective/train/docs_used": 364826, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5260703563690186, + "objective/train/original_loss": 2.5260705947875977, + "objective/train/theoretical_loss": 3.8220573202451886, + "objective/train/tokens_used": 647967200, + "objective/train/value_avg": -0.0057830810546875, + "objective/train/value_loss": 9.660975047154352e-05, + "objective/train/value_max": -0.0004222393035888672, + "objective/train/value_min": -0.1026611328125, + "objective/train/value_reward_corr": 0.6263685443641615, + "objective/train/value_std": 0.005985260009765625, + "objective/train/weight_avg": 0.9845865964889526, + "objective/train/weighted_lm_loss": 2.4923324584960938, + "objective/train/weights_max": 2.5736124515533447, + "objective/train/weights_min": 0.4245847463607788, + "theoretical_loss": 3.8220573202451886, + "tokens_seen": 627507200 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004090571715145436, + "loss": 1.3546, + "theoretical_loss": 3.8220573202451886, + "tokens_seen": 627507200 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040904714142427285, + "loss": 1.5017, + "theoretical_loss": 3.822015998211768, + "tokens_seen": 627572736 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040903711133400203, + "loss": 1.4171, + "theoretical_loss": 3.821974681701392, + "tokens_seen": 627638272 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004090270812437312, + "loss": 1.4972, + "theoretical_loss": 3.8219333707127445, + "tokens_seen": 627703808 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004090170511534604, + "loss": 1.4566, + "theoretical_loss": 3.8218920652445125, + "tokens_seen": 627769344 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040900702106318957, + "loss": 1.4578, + "theoretical_loss": 3.8218507652953813, + "tokens_seen": 627834880 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040899699097291875, + "loss": 1.5472, + "theoretical_loss": 3.8218094708640367, + "tokens_seen": 627900416 + }, + { + "epoch": 0.19, + "learning_rate": 0.000408986960882648, + "loss": 1.4972, + "theoretical_loss": 3.8217681819491673, + "tokens_seen": 627965952 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004089769307923771, + "loss": 1.494, + "theoretical_loss": 3.8217268985494597, + "tokens_seen": 628031488 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040896690070210635, + "loss": 1.4942, + "theoretical_loss": 3.8216856206636014, + "tokens_seen": 628097024 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040895687061183553, + "loss": 1.4574, + "theoretical_loss": 3.821644348290282, + "tokens_seen": 628162560 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004089468405215647, + "loss": 1.4798, + "theoretical_loss": 3.821603081428189, + "tokens_seen": 628228096 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004089368104312939, + "loss": 1.4643, + "theoretical_loss": 3.8215618200760124, + "tokens_seen": 628293632 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004089267803410231, + "loss": 1.4777, + "theoretical_loss": 3.8215205642324417, + "tokens_seen": 628359168 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040891675025075226, + "loss": 1.3972, + "theoretical_loss": 3.8214793138961674, + "tokens_seen": 628424704 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004089067201604815, + "loss": 1.4845, + "theoretical_loss": 3.82143806906588, + "tokens_seen": 628490240 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004088966900702106, + "loss": 1.3856, + "theoretical_loss": 3.82139682974027, + "tokens_seen": 628555776 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040888665997993985, + "loss": 1.5314, + "theoretical_loss": 3.821355595918029, + "tokens_seen": 628621312 + }, + { + "epoch": 0.19, + "learning_rate": 0.000408876629889669, + "loss": 1.3979, + "theoretical_loss": 3.8213143675978496, + "tokens_seen": 628686848 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004088665997993982, + "loss": 1.4982, + "theoretical_loss": 3.821273144778424, + "tokens_seen": 628752384 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004088565697091274, + "loss": 1.435, + "theoretical_loss": 3.8212319274584448, + "tokens_seen": 628817920 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004088465396188566, + "loss": 1.5055, + "theoretical_loss": 3.8211907156366047, + "tokens_seen": 628883456 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040883650952858576, + "loss": 1.445, + "theoretical_loss": 3.821149509311599, + "tokens_seen": 628948992 + }, + { + "epoch": 0.19, + "learning_rate": 0.000408826479438315, + "loss": 1.4697, + "theoretical_loss": 3.8211083084821205, + "tokens_seen": 629014528 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004088164493480441, + "loss": 1.5606, + "theoretical_loss": 3.821067113146864, + "tokens_seen": 629080064 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.00023212366795632988, + "objective/train/docs_used": 365375, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0437633991241455, + "objective/train/original_loss": 3.0437636375427246, + "objective/train/theoretical_loss": 3.8210259233045254, + "objective/train/tokens_used": 649605600, + "objective/train/value_avg": -0.00505828857421875, + "objective/train/value_loss": 4.853124846704304e-05, + "objective/train/value_max": -0.0003654956817626953, + "objective/train/value_min": -0.060760498046875, + "objective/train/value_reward_corr": 0.3708112922211409, + "objective/train/value_std": 0.0038242340087890625, + "objective/train/weight_avg": 0.9999001026153564, + "objective/train/weighted_lm_loss": 3.0467894077301025, + "objective/train/weights_max": 1.6350330114364624, + "objective/train/weights_min": 0.25109657645225525, + "theoretical_loss": 3.8210259233045254, + "tokens_seen": 629145600 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040880641925777336, + "loss": 1.3891, + "theoretical_loss": 3.8210259233045254, + "tokens_seen": 629145600 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004087963891675025, + "loss": 1.4625, + "theoretical_loss": 3.8209847389537996, + "tokens_seen": 629211136 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004087863590772317, + "loss": 1.4259, + "theoretical_loss": 3.820943560093382, + "tokens_seen": 629276672 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004087763289869609, + "loss": 1.5545, + "theoretical_loss": 3.8209023867219702, + "tokens_seen": 629342208 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004087662988966901, + "loss": 1.5665, + "theoretical_loss": 3.8208612188382607, + "tokens_seen": 629407744 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040875626880641926, + "loss": 1.5187, + "theoretical_loss": 3.8208200564409505, + "tokens_seen": 629473280 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040874623871614844, + "loss": 1.4038, + "theoretical_loss": 3.8207788995287375, + "tokens_seen": 629538816 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004087362086258776, + "loss": 1.4002, + "theoretical_loss": 3.8207377481003197, + "tokens_seen": 629604352 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040872617853560686, + "loss": 1.5252, + "theoretical_loss": 3.8206966021543955, + "tokens_seen": 629669888 + }, + { + "epoch": 0.19, + "learning_rate": 0.000408716148445336, + "loss": 1.3847, + "theoretical_loss": 3.8206554616896646, + "tokens_seen": 629735424 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004087061183550652, + "loss": 1.4353, + "theoretical_loss": 3.8206143267048267, + "tokens_seen": 629800960 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040869608826479435, + "loss": 1.4648, + "theoretical_loss": 3.820573197198581, + "tokens_seen": 629866496 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004086860581745236, + "loss": 1.4873, + "theoretical_loss": 3.8205320731696277, + "tokens_seen": 629932032 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040867602808425277, + "loss": 1.4187, + "theoretical_loss": 3.820490954616669, + "tokens_seen": 629997568 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040866599799398195, + "loss": 1.373, + "theoretical_loss": 3.820449841538405, + "tokens_seen": 630063104 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040865596790371113, + "loss": 1.5901, + "theoretical_loss": 3.8204087339335384, + "tokens_seen": 630128640 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040864593781344036, + "loss": 1.3996, + "theoretical_loss": 3.8203676318007704, + "tokens_seen": 630194176 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004086359077231695, + "loss": 1.5377, + "theoretical_loss": 3.8203265351388045, + "tokens_seen": 630259712 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004086258776328987, + "loss": 1.3439, + "theoretical_loss": 3.820285443946343, + "tokens_seen": 630325248 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040861584754262785, + "loss": 1.394, + "theoretical_loss": 3.820244358222089, + "tokens_seen": 630390784 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004086058174523571, + "loss": 1.5146, + "theoretical_loss": 3.8202032779647483, + "tokens_seen": 630456320 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040859578736208627, + "loss": 1.3806, + "theoretical_loss": 3.820162203173024, + "tokens_seen": 630521856 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040858575727181545, + "loss": 1.4316, + "theoretical_loss": 3.8201211338456202, + "tokens_seen": 630587392 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040857572718154463, + "loss": 1.4148, + "theoretical_loss": 3.820080069981244, + "tokens_seen": 630652928 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004085656970912738, + "loss": 1.5049, + "theoretical_loss": 3.8200390115786003, + "tokens_seen": 630718464 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": 0.0007909638807177544, + "objective/train/docs_used": 366612, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.370457410812378, + "objective/train/original_loss": 2.370457172393799, + "objective/train/theoretical_loss": 3.8199979586363946, + "objective/train/tokens_used": 651244000, + "objective/train/value_avg": -0.005023956298828125, + "objective/train/value_loss": 3.8355454307748005e-05, + "objective/train/value_max": -0.0003669261932373047, + "objective/train/value_min": -0.10089111328125, + "objective/train/value_reward_corr": 0.36696957249733486, + "objective/train/value_std": 0.003963470458984375, + "objective/train/weight_avg": 1.0097923278808594, + "objective/train/weighted_lm_loss": 2.387725830078125, + "objective/train/weights_max": 2.212933301925659, + "objective/train/weights_min": 0.4468451142311096, + "theoretical_loss": 3.8199979586363946, + "tokens_seen": 630784000 + }, + { + "epoch": 0.19, + "learning_rate": 0.000408555667001003, + "loss": 1.3716, + "theoretical_loss": 3.8199979586363946, + "tokens_seen": 630784000 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040854563691073223, + "loss": 1.5334, + "theoretical_loss": 3.8199569111533345, + "tokens_seen": 630849536 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040853560682046136, + "loss": 1.4605, + "theoretical_loss": 3.8199158691281263, + "tokens_seen": 630915072 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004085255767301906, + "loss": 1.3688, + "theoretical_loss": 3.819874832559478, + "tokens_seen": 630980608 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004085155466399197, + "loss": 1.5115, + "theoretical_loss": 3.819833801446097, + "tokens_seen": 631046144 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040850551654964895, + "loss": 1.5422, + "theoretical_loss": 3.8197927757866923, + "tokens_seen": 631111680 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040849548645937814, + "loss": 1.3619, + "theoretical_loss": 3.819751755579972, + "tokens_seen": 631177216 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004084854563691073, + "loss": 1.4737, + "theoretical_loss": 3.819710740824646, + "tokens_seen": 631242752 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004084754262788365, + "loss": 1.4498, + "theoretical_loss": 3.819669731519424, + "tokens_seen": 631308288 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040846539618856573, + "loss": 1.4051, + "theoretical_loss": 3.819628727663015, + "tokens_seen": 631373824 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040845536609829486, + "loss": 1.4507, + "theoretical_loss": 3.819587729254131, + "tokens_seen": 631439360 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004084453360080241, + "loss": 1.4397, + "theoretical_loss": 3.819546736291483, + "tokens_seen": 631504896 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004084353059177532, + "loss": 1.4421, + "theoretical_loss": 3.8195057487737802, + "tokens_seen": 631570432 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040842527582748246, + "loss": 1.4566, + "theoretical_loss": 3.8194647666997374, + "tokens_seen": 631635968 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040841524573721164, + "loss": 1.5414, + "theoretical_loss": 3.8194237900680648, + "tokens_seen": 631701504 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004084052156469408, + "loss": 1.4257, + "theoretical_loss": 3.819382818877477, + "tokens_seen": 631767040 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040839518555667005, + "loss": 1.4319, + "theoretical_loss": 3.819341853126685, + "tokens_seen": 631832576 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004083851554663992, + "loss": 1.4315, + "theoretical_loss": 3.819300892814404, + "tokens_seen": 631898112 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004083751253761284, + "loss": 1.4434, + "theoretical_loss": 3.8192599379393477, + "tokens_seen": 631963648 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004083650952858576, + "loss": 1.4716, + "theoretical_loss": 3.819218988500231, + "tokens_seen": 632029184 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004083550651955868, + "loss": 1.4205, + "theoretical_loss": 3.8191780444957675, + "tokens_seen": 632094720 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040834503510531596, + "loss": 1.4608, + "theoretical_loss": 3.819137105924674, + "tokens_seen": 632160256 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004083350050150452, + "loss": 1.5071, + "theoretical_loss": 3.819096172785666, + "tokens_seen": 632225792 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004083249749247743, + "loss": 1.4391, + "theoretical_loss": 3.8190552450774584, + "tokens_seen": 632291328 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040831494483450356, + "loss": 1.4361, + "theoretical_loss": 3.81901432279877, + "tokens_seen": 632356864 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": 0.0007279632263816893, + "objective/train/docs_used": 367323, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1712894439697266, + "objective/train/original_loss": 3.1712894439697266, + "objective/train/theoretical_loss": 3.8189734059483165, + "objective/train/tokens_used": 652882400, + "objective/train/value_avg": -0.0045166015625, + "objective/train/value_loss": 5.571155634243041e-05, + "objective/train/value_max": -0.0004546642303466797, + "objective/train/value_min": -0.0341796875, + "objective/train/value_reward_corr": 0.3290753124104708, + "objective/train/value_std": 0.0030002593994140625, + "objective/train/weight_avg": 1.0098387002944946, + "objective/train/weighted_lm_loss": 3.2078089714050293, + "objective/train/weights_max": 1.373528242111206, + "objective/train/weights_min": 0.3322650194168091, + "theoretical_loss": 3.8189734059483165, + "tokens_seen": 632422400 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004083049147442327, + "loss": 1.5409, + "theoretical_loss": 3.8189734059483165, + "tokens_seen": 632422400 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004082948846539619, + "loss": 1.3212, + "theoretical_loss": 3.8189324945248164, + "tokens_seen": 632487936 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004082848545636911, + "loss": 1.3468, + "theoretical_loss": 3.818891588526987, + "tokens_seen": 632553472 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004082748244734203, + "loss": 1.4693, + "theoretical_loss": 3.818850687953547, + "tokens_seen": 632619008 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040826479438314946, + "loss": 1.3905, + "theoretical_loss": 3.8188097928032145, + "tokens_seen": 632684544 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040825476429287864, + "loss": 1.4909, + "theoretical_loss": 3.81876890307471, + "tokens_seen": 632750080 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004082447342026078, + "loss": 1.4838, + "theoretical_loss": 3.8187280187667527, + "tokens_seen": 632815616 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040823470411233706, + "loss": 1.4876, + "theoretical_loss": 3.8186871398780626, + "tokens_seen": 632881152 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004082246740220662, + "loss": 1.3489, + "theoretical_loss": 3.8186462664073613, + "tokens_seen": 632946688 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004082146439317954, + "loss": 1.4224, + "theoretical_loss": 3.818605398353368, + "tokens_seen": 633012224 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040820461384152455, + "loss": 1.2972, + "theoretical_loss": 3.8185645357148057, + "tokens_seen": 633077760 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004081945837512538, + "loss": 1.4291, + "theoretical_loss": 3.8185236784903958, + "tokens_seen": 633143296 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040818455366098297, + "loss": 1.495, + "theoretical_loss": 3.8184828266788604, + "tokens_seen": 633208832 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040817452357071215, + "loss": 1.3201, + "theoretical_loss": 3.818441980278923, + "tokens_seen": 633274368 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040816449348044133, + "loss": 1.2873, + "theoretical_loss": 3.818401139289306, + "tokens_seen": 633339904 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040815446339017056, + "loss": 1.387, + "theoretical_loss": 3.818360303708733, + "tokens_seen": 633405440 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004081444332998997, + "loss": 1.3898, + "theoretical_loss": 3.8183194735359294, + "tokens_seen": 633470976 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004081344032096289, + "loss": 1.5504, + "theoretical_loss": 3.8182786487696183, + "tokens_seen": 633536512 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040812437311935805, + "loss": 1.5383, + "theoretical_loss": 3.818237829408525, + "tokens_seen": 633602048 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004081143430290873, + "loss": 1.3671, + "theoretical_loss": 3.818197015451375, + "tokens_seen": 633667584 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040810431293881647, + "loss": 1.4047, + "theoretical_loss": 3.8181562068968944, + "tokens_seen": 633733120 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040809428284854565, + "loss": 1.3904, + "theoretical_loss": 3.8181154037438088, + "tokens_seen": 633798656 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040808425275827483, + "loss": 1.4488, + "theoretical_loss": 3.8180746059908457, + "tokens_seen": 633864192 + }, + { + "epoch": 0.19, + "learning_rate": 0.000408074222668004, + "loss": 1.3946, + "theoretical_loss": 3.818033813636731, + "tokens_seen": 633929728 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004080641925777332, + "loss": 1.4034, + "theoretical_loss": 3.8179930266801936, + "tokens_seen": 633995264 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": 0.0008688066154718399, + "objective/train/docs_used": 367883, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0341057777404785, + "objective/train/original_loss": 3.0341057777404785, + "objective/train/theoretical_loss": 3.817952245119961, + "objective/train/tokens_used": 654520800, + "objective/train/value_avg": -0.006900787353515625, + "objective/train/value_loss": 6.59573488519527e-05, + "objective/train/value_max": -0.00047659873962402344, + "objective/train/value_min": -0.0638427734375, + "objective/train/value_reward_corr": 0.4046551917730105, + "objective/train/value_std": 0.00551605224609375, + "objective/train/weight_avg": 1.011883020401001, + "objective/train/weighted_lm_loss": 3.0675508975982666, + "objective/train/weights_max": 1.8374848365783691, + "objective/train/weights_min": 0.3128972351551056, + "theoretical_loss": 3.817952245119961, + "tokens_seen": 634060800 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040805416248746243, + "loss": 1.4881, + "theoretical_loss": 3.817952245119961, + "tokens_seen": 634060800 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040804413239719156, + "loss": 1.3976, + "theoretical_loss": 3.8179114689547604, + "tokens_seen": 634126336 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004080341023069208, + "loss": 1.4939, + "theoretical_loss": 3.8178706981833224, + "tokens_seen": 634191872 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004080240722166499, + "loss": 1.4173, + "theoretical_loss": 3.8178299328043757, + "tokens_seen": 634257408 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040801404212637915, + "loss": 1.4195, + "theoretical_loss": 3.817789172816649, + "tokens_seen": 634322944 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040800401203610834, + "loss": 1.4289, + "theoretical_loss": 3.8177484182188737, + "tokens_seen": 634388480 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004079939819458375, + "loss": 1.2998, + "theoretical_loss": 3.8177076690097795, + "tokens_seen": 634454016 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004079839518555667, + "loss": 1.4566, + "theoretical_loss": 3.817666925188098, + "tokens_seen": 634519552 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040797392176529593, + "loss": 1.3848, + "theoretical_loss": 3.8176261867525607, + "tokens_seen": 634585088 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040796389167502506, + "loss": 1.3445, + "theoretical_loss": 3.8175854537018985, + "tokens_seen": 634650624 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004079538615847543, + "loss": 1.4849, + "theoretical_loss": 3.817544726034844, + "tokens_seen": 634716160 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004079438314944834, + "loss": 1.4371, + "theoretical_loss": 3.8175040037501304, + "tokens_seen": 634781696 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040793380140421266, + "loss": 1.4285, + "theoretical_loss": 3.8174632868464915, + "tokens_seen": 634847232 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040792377131394184, + "loss": 1.4353, + "theoretical_loss": 3.817422575322659, + "tokens_seen": 634912768 + }, + { + "epoch": 0.19, + "learning_rate": 0.000407913741223671, + "loss": 1.4363, + "theoretical_loss": 3.817381869177368, + "tokens_seen": 634978304 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004079037111334002, + "loss": 1.4757, + "theoretical_loss": 3.817341168409353, + "tokens_seen": 635043840 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004078936810431294, + "loss": 1.5685, + "theoretical_loss": 3.8173004730173483, + "tokens_seen": 635109376 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040788365095285856, + "loss": 1.5025, + "theoretical_loss": 3.8172597830000896, + "tokens_seen": 635174912 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004078736208625878, + "loss": 1.496, + "theoretical_loss": 3.817219098356312, + "tokens_seen": 635240448 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004078635907723169, + "loss": 1.4412, + "theoretical_loss": 3.8171784190847524, + "tokens_seen": 635305984 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040785356068204616, + "loss": 1.4486, + "theoretical_loss": 3.817137745184147, + "tokens_seen": 635371520 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004078435305917753, + "loss": 1.4396, + "theoretical_loss": 3.8170970766532326, + "tokens_seen": 635437056 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004078335005015045, + "loss": 1.407, + "theoretical_loss": 3.8170564134907474, + "tokens_seen": 635502592 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004078234704112337, + "loss": 1.308, + "theoretical_loss": 3.8170157556954285, + "tokens_seen": 635568128 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004078134403209629, + "loss": 1.5092, + "theoretical_loss": 3.8169751032660137, + "tokens_seen": 635633664 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.0012931732926517725, + "objective/train/docs_used": 369288, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.056506395339966, + "objective/train/original_loss": 3.056506872177124, + "objective/train/theoretical_loss": 3.816934456201243, + "objective/train/tokens_used": 656159200, + "objective/train/value_avg": -0.00664520263671875, + "objective/train/value_loss": 8.782654913375154e-05, + "objective/train/value_max": -0.0002892017364501953, + "objective/train/value_min": -0.07666015625, + "objective/train/value_reward_corr": 0.458306632682683, + "objective/train/value_std": 0.006256103515625, + "objective/train/weight_avg": 0.9912649989128113, + "objective/train/weighted_lm_loss": 3.0469367504119873, + "objective/train/weights_max": 1.8395730257034302, + "objective/train/weights_min": 0.3075718283653259, + "theoretical_loss": 3.816934456201243, + "tokens_seen": 635699200 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040780341023069207, + "loss": 1.4665, + "theoretical_loss": 3.816934456201243, + "tokens_seen": 635699200 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004077933801404213, + "loss": 1.3351, + "theoretical_loss": 3.8168938144998545, + "tokens_seen": 635764736 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040778335005015043, + "loss": 1.4036, + "theoretical_loss": 3.8168531781605877, + "tokens_seen": 635830272 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040777331995987966, + "loss": 1.4294, + "theoretical_loss": 3.8168125471821837, + "tokens_seen": 635895808 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004077632898696088, + "loss": 1.4982, + "theoretical_loss": 3.8167719215633817, + "tokens_seen": 635961344 + }, + { + "epoch": 0.19, + "learning_rate": 0.000407753259779338, + "loss": 1.2885, + "theoretical_loss": 3.816731301302923, + "tokens_seen": 636026880 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004077432296890672, + "loss": 1.3944, + "theoretical_loss": 3.816690686399549, + "tokens_seen": 636092416 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004077331995987964, + "loss": 1.4026, + "theoretical_loss": 3.8166500768520004, + "tokens_seen": 636157952 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040772316950852557, + "loss": 1.4214, + "theoretical_loss": 3.816609472659021, + "tokens_seen": 636223488 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040771313941825475, + "loss": 1.4258, + "theoretical_loss": 3.8165688738193517, + "tokens_seen": 636289024 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040770310932798393, + "loss": 1.4884, + "theoretical_loss": 3.8165282803317364, + "tokens_seen": 636354560 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040769307923771317, + "loss": 1.586, + "theoretical_loss": 3.8164876921949182, + "tokens_seen": 636420096 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004076830491474423, + "loss": 1.4465, + "theoretical_loss": 3.816447109407641, + "tokens_seen": 636485632 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040767301905717153, + "loss": 1.3432, + "theoretical_loss": 3.8164065319686484, + "tokens_seen": 636551168 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004076629889669007, + "loss": 1.4766, + "theoretical_loss": 3.816365959876686, + "tokens_seen": 636616704 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004076529588766299, + "loss": 1.4413, + "theoretical_loss": 3.8163253931304983, + "tokens_seen": 636682240 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004076429287863591, + "loss": 1.4668, + "theoretical_loss": 3.816284831728831, + "tokens_seen": 636747776 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040763289869608825, + "loss": 1.4385, + "theoretical_loss": 3.81624427567043, + "tokens_seen": 636813312 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004076228686058175, + "loss": 1.4394, + "theoretical_loss": 3.816203724954041, + "tokens_seen": 636878848 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040761283851554667, + "loss": 1.3475, + "theoretical_loss": 3.8161631795784117, + "tokens_seen": 636944384 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040760280842527585, + "loss": 1.3186, + "theoretical_loss": 3.816122639542289, + "tokens_seen": 637009920 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040759277833500503, + "loss": 1.4209, + "theoretical_loss": 3.81608210484442, + "tokens_seen": 637075456 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004075827482447342, + "loss": 1.4763, + "theoretical_loss": 3.8160415754835535, + "tokens_seen": 637140992 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004075727181544634, + "loss": 1.376, + "theoretical_loss": 3.8160010514584375, + "tokens_seen": 637206528 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040756268806419263, + "loss": 1.3745, + "theoretical_loss": 3.815960532767821, + "tokens_seen": 637272064 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -7.100084621924907e-05, + "objective/train/docs_used": 369983, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3376622200012207, + "objective/train/original_loss": 3.3376619815826416, + "objective/train/theoretical_loss": 3.815920019410453, + "objective/train/tokens_used": 657797600, + "objective/train/value_avg": -0.006072998046875, + "objective/train/value_loss": 5.7765006204135716e-05, + "objective/train/value_max": -0.0003800392150878906, + "objective/train/value_min": -0.076416015625, + "objective/train/value_reward_corr": 0.3881194444287867, + "objective/train/value_std": 0.00452423095703125, + "objective/train/weight_avg": 1.0020982027053833, + "objective/train/weighted_lm_loss": 3.3506548404693604, + "objective/train/weights_max": 1.8824633359909058, + "objective/train/weights_min": 0.4820117652416229, + "theoretical_loss": 3.815920019410453, + "tokens_seen": 637337600 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040755265797392176, + "loss": 1.6045, + "theoretical_loss": 3.815920019410453, + "tokens_seen": 637337600 + }, + { + "epoch": 0.19, + "learning_rate": 0.000407542627883651, + "loss": 1.3591, + "theoretical_loss": 3.8158795113850834, + "tokens_seen": 637403136 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004075325977933801, + "loss": 1.4685, + "theoretical_loss": 3.815839008690463, + "tokens_seen": 637468672 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040752256770310935, + "loss": 1.3444, + "theoretical_loss": 3.815798511325341, + "tokens_seen": 637534208 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040751253761283854, + "loss": 1.487, + "theoretical_loss": 3.8157580192884692, + "tokens_seen": 637599744 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004075025075225677, + "loss": 1.385, + "theoretical_loss": 3.8157175325785997, + "tokens_seen": 637665280 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004074924774322969, + "loss": 1.4503, + "theoretical_loss": 3.8156770511944833, + "tokens_seen": 637730816 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040748244734202613, + "loss": 1.3892, + "theoretical_loss": 3.8156365751348718, + "tokens_seen": 637796352 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040747241725175526, + "loss": 1.4145, + "theoretical_loss": 3.8155961043985194, + "tokens_seen": 637861888 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004074623871614845, + "loss": 1.5702, + "theoretical_loss": 3.8155556389841783, + "tokens_seen": 637927424 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004074523570712136, + "loss": 1.3779, + "theoretical_loss": 3.8155151788906014, + "tokens_seen": 637992960 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040744232698094286, + "loss": 1.3629, + "theoretical_loss": 3.8154747241165436, + "tokens_seen": 638058496 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040743229689067204, + "loss": 1.5618, + "theoretical_loss": 3.815434274660759, + "tokens_seen": 638124032 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004074222668004012, + "loss": 1.5581, + "theoretical_loss": 3.8153938305220025, + "tokens_seen": 638189568 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004074122367101304, + "loss": 1.5416, + "theoretical_loss": 3.8153533916990288, + "tokens_seen": 638255104 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004074022066198596, + "loss": 1.4133, + "theoretical_loss": 3.815312958190594, + "tokens_seen": 638320640 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040739217652958876, + "loss": 1.3311, + "theoretical_loss": 3.815272529995454, + "tokens_seen": 638386176 + }, + { + "epoch": 0.19, + "learning_rate": 0.000407382146439318, + "loss": 1.3362, + "theoretical_loss": 3.8152321071123643, + "tokens_seen": 638451712 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004073721163490471, + "loss": 1.3468, + "theoretical_loss": 3.8151916895400833, + "tokens_seen": 638517248 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040736208625877636, + "loss": 1.3644, + "theoretical_loss": 3.8151512772773675, + "tokens_seen": 638582784 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004073520561685055, + "loss": 1.4636, + "theoretical_loss": 3.8151108703229744, + "tokens_seen": 638648320 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004073420260782347, + "loss": 1.4908, + "theoretical_loss": 3.8150704686756627, + "tokens_seen": 638713856 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004073319959879639, + "loss": 1.3386, + "theoretical_loss": 3.8150300723341903, + "tokens_seen": 638779392 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004073219658976931, + "loss": 1.4567, + "theoretical_loss": 3.814989681297317, + "tokens_seen": 638844928 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040731193580742227, + "loss": 1.4641, + "theoretical_loss": 3.8149492955638014, + "tokens_seen": 638910464 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.002221066039055586, + "objective/train/docs_used": 370882, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.06880521774292, + "objective/train/original_loss": 3.068805456161499, + "objective/train/theoretical_loss": 3.8149089151324036, + "objective/train/tokens_used": 659436000, + "objective/train/value_avg": -0.00457000732421875, + "objective/train/value_loss": 9.291571768699214e-05, + "objective/train/value_max": -0.0003612041473388672, + "objective/train/value_min": -0.07501220703125, + "objective/train/value_reward_corr": 0.27690937477099326, + "objective/train/value_std": 0.002918243408203125, + "objective/train/weight_avg": 0.982032060623169, + "objective/train/weighted_lm_loss": 3.0313332080841064, + "objective/train/weights_max": 1.603749394416809, + "objective/train/weights_min": 0.4122888743877411, + "theoretical_loss": 3.8149089151324036, + "tokens_seen": 638976000 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004073019057171515, + "loss": 1.4139, + "theoretical_loss": 3.8149089151324036, + "tokens_seen": 638976000 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040729187562688063, + "loss": 1.4039, + "theoretical_loss": 3.814868540001883, + "tokens_seen": 639041536 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040728184553660986, + "loss": 1.4993, + "theoretical_loss": 3.814828170171002, + "tokens_seen": 639107072 + }, + { + "epoch": 0.19, + "learning_rate": 0.000407271815446339, + "loss": 1.4793, + "theoretical_loss": 3.8147878056385203, + "tokens_seen": 639172608 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004072617853560682, + "loss": 1.569, + "theoretical_loss": 3.814747446403199, + "tokens_seen": 639238144 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004072517552657974, + "loss": 1.5236, + "theoretical_loss": 3.814707092463801, + "tokens_seen": 639303680 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004072417251755266, + "loss": 1.5095, + "theoretical_loss": 3.814666743819088, + "tokens_seen": 639369216 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040723169508525577, + "loss": 1.4963, + "theoretical_loss": 3.814626400467823, + "tokens_seen": 639434752 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040722166499498495, + "loss": 1.3612, + "theoretical_loss": 3.814586062408769, + "tokens_seen": 639500288 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040721163490471413, + "loss": 1.4289, + "theoretical_loss": 3.81454572964069, + "tokens_seen": 639565824 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040720160481444337, + "loss": 1.357, + "theoretical_loss": 3.814505402162349, + "tokens_seen": 639631360 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004071915747241725, + "loss": 1.5463, + "theoretical_loss": 3.8144650799725106, + "tokens_seen": 639696896 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040718154463390173, + "loss": 1.453, + "theoretical_loss": 3.8144247630699404, + "tokens_seen": 639762432 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004071715145436309, + "loss": 1.5921, + "theoretical_loss": 3.8143844514534027, + "tokens_seen": 639827968 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004071614844533601, + "loss": 1.5095, + "theoretical_loss": 3.8143441451216633, + "tokens_seen": 639893504 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040715145436308927, + "loss": 1.5038, + "theoretical_loss": 3.814303844073488, + "tokens_seen": 639959040 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040714142427281845, + "loss": 1.3953, + "theoretical_loss": 3.8142635483076446, + "tokens_seen": 640024576 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040713139418254763, + "loss": 1.5538, + "theoretical_loss": 3.8142232578228983, + "tokens_seen": 640090112 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040712136409227687, + "loss": 1.4256, + "theoretical_loss": 3.814182972618017, + "tokens_seen": 640155648 + }, + { + "epoch": 0.19, + "learning_rate": 0.000407111334002006, + "loss": 1.4003, + "theoretical_loss": 3.8141426926917683, + "tokens_seen": 640221184 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040710130391173523, + "loss": 1.4925, + "theoretical_loss": 3.8141024180429204, + "tokens_seen": 640286720 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040709127382146436, + "loss": 1.4487, + "theoretical_loss": 3.814062148670242, + "tokens_seen": 640352256 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004070812437311936, + "loss": 1.3586, + "theoretical_loss": 3.814021884572502, + "tokens_seen": 640417792 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004070712136409228, + "loss": 1.5184, + "theoretical_loss": 3.813981625748469, + "tokens_seen": 640483328 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040706118355065196, + "loss": 1.4499, + "theoretical_loss": 3.813941372196914, + "tokens_seen": 640548864 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.00033274994348175824, + "objective/train/docs_used": 371543, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8811452388763428, + "objective/train/original_loss": 2.881145477294922, + "objective/train/theoretical_loss": 3.8139011239166054, + "objective/train/tokens_used": 661074400, + "objective/train/value_avg": -0.005161285400390625, + "objective/train/value_loss": 4.976023410563357e-05, + "objective/train/value_max": -0.0003829002380371094, + "objective/train/value_min": -0.0509033203125, + "objective/train/value_reward_corr": 0.4002772024165608, + "objective/train/value_std": 0.00418853759765625, + "objective/train/weight_avg": 0.9990168213844299, + "objective/train/weighted_lm_loss": 2.8986542224884033, + "objective/train/weights_max": 1.408333659172058, + "objective/train/weights_min": 0.41447290778160095, + "theoretical_loss": 3.8139011239166054, + "tokens_seen": 640614400 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040705115346038114, + "loss": 1.485, + "theoretical_loss": 3.8139011239166054, + "tokens_seen": 640614400 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004070411233701103, + "loss": 1.4169, + "theoretical_loss": 3.813860880906316, + "tokens_seen": 640679936 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004070310932798395, + "loss": 1.474, + "theoretical_loss": 3.8138206431648145, + "tokens_seen": 640745472 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040702106318956874, + "loss": 1.4852, + "theoretical_loss": 3.8137804106908737, + "tokens_seen": 640811008 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040701103309929786, + "loss": 1.3943, + "theoretical_loss": 3.8137401834832656, + "tokens_seen": 640876544 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004070010030090271, + "loss": 1.4688, + "theoretical_loss": 3.8136999615407614, + "tokens_seen": 640942080 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004069909729187563, + "loss": 1.4151, + "theoretical_loss": 3.8136597448621345, + "tokens_seen": 641007616 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040698094282848546, + "loss": 1.5946, + "theoretical_loss": 3.8136195334461576, + "tokens_seen": 641073152 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040697091273821464, + "loss": 1.4279, + "theoretical_loss": 3.8135793272916043, + "tokens_seen": 641138688 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004069608826479438, + "loss": 1.4749, + "theoretical_loss": 3.8135391263972487, + "tokens_seen": 641204224 + }, + { + "epoch": 0.19, + "learning_rate": 0.000406950852557673, + "loss": 1.367, + "theoretical_loss": 3.813498930761864, + "tokens_seen": 641269760 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040694082246740224, + "loss": 1.4181, + "theoretical_loss": 3.8134587403842266, + "tokens_seen": 641335296 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040693079237713137, + "loss": 1.5346, + "theoretical_loss": 3.8134185552631097, + "tokens_seen": 641400832 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004069207622868606, + "loss": 1.4524, + "theoretical_loss": 3.8133783753972903, + "tokens_seen": 641466368 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040691073219658973, + "loss": 1.5114, + "theoretical_loss": 3.8133382007855445, + "tokens_seen": 641531904 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040690070210631896, + "loss": 1.5138, + "theoretical_loss": 3.8132980314266476, + "tokens_seen": 641597440 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004068906720160482, + "loss": 1.4319, + "theoretical_loss": 3.8132578673193755, + "tokens_seen": 641662976 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004068806419257773, + "loss": 1.4571, + "theoretical_loss": 3.813217708462508, + "tokens_seen": 641728512 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040687061183550656, + "loss": 1.3996, + "theoretical_loss": 3.8131775548548204, + "tokens_seen": 641794048 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004068605817452357, + "loss": 1.4778, + "theoretical_loss": 3.813137406495092, + "tokens_seen": 641859584 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004068505516549649, + "loss": 1.3555, + "theoretical_loss": 3.8130972633821, + "tokens_seen": 641925120 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004068405215646941, + "loss": 1.4292, + "theoretical_loss": 3.8130571255146246, + "tokens_seen": 641990656 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004068304914744233, + "loss": 1.4375, + "theoretical_loss": 3.813016992891444, + "tokens_seen": 642056192 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040682046138415247, + "loss": 1.468, + "theoretical_loss": 3.812976865511337, + "tokens_seen": 642121728 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004068104312938817, + "loss": 1.3418, + "theoretical_loss": 3.8129367433730863, + "tokens_seen": 642187264 + }, + { + "epoch": 0.19, + "objective/train/advantage_avg": -0.0002571732038632035, + "objective/train/docs_used": 373013, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0892927646636963, + "objective/train/original_loss": 3.0892930030822754, + "objective/train/theoretical_loss": 3.81289662647547, + "objective/train/tokens_used": 662712800, + "objective/train/value_avg": -0.005565643310546875, + "objective/train/value_loss": 5.419699300546199e-05, + "objective/train/value_max": -0.00041413307189941406, + "objective/train/value_min": -0.08740234375, + "objective/train/value_reward_corr": 0.44098173148511677, + "objective/train/value_std": 0.004619598388671875, + "objective/train/weight_avg": 0.9999920129776001, + "objective/train/weighted_lm_loss": 3.083749532699585, + "objective/train/weights_max": 2.1564366817474365, + "objective/train/weights_min": 0.34271153807640076, + "theoretical_loss": 3.81289662647547, + "tokens_seen": 642252800 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040680040120361083, + "loss": 1.4811, + "theoretical_loss": 3.81289662647547, + "tokens_seen": 642252800 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040679037111334006, + "loss": 1.5098, + "theoretical_loss": 3.8128565148172693, + "tokens_seen": 642318336 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004067803410230692, + "loss": 1.4947, + "theoretical_loss": 3.812816408397266, + "tokens_seen": 642383872 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004067703109327984, + "loss": 1.4558, + "theoretical_loss": 3.812776307214241, + "tokens_seen": 642449408 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004067602808425276, + "loss": 1.4283, + "theoretical_loss": 3.8127362112669774, + "tokens_seen": 642514944 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004067502507522568, + "loss": 1.3186, + "theoretical_loss": 3.812696120554257, + "tokens_seen": 642580480 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040674022066198597, + "loss": 1.3726, + "theoretical_loss": 3.8126560350748626, + "tokens_seen": 642646016 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040673019057171515, + "loss": 1.4325, + "theoretical_loss": 3.812615954827578, + "tokens_seen": 642711552 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040672016048144433, + "loss": 1.5953, + "theoretical_loss": 3.8125758798111864, + "tokens_seen": 642777088 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040671013039117357, + "loss": 1.2708, + "theoretical_loss": 3.8125358100244715, + "tokens_seen": 642842624 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004067001003009027, + "loss": 1.4221, + "theoretical_loss": 3.8124957454662187, + "tokens_seen": 642908160 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040669007021063193, + "loss": 1.4363, + "theoretical_loss": 3.8124556861352126, + "tokens_seen": 642973696 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004066800401203611, + "loss": 1.4697, + "theoretical_loss": 3.8124156320302385, + "tokens_seen": 643039232 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004066700100300903, + "loss": 1.3989, + "theoretical_loss": 3.8123755831500814, + "tokens_seen": 643104768 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040665997993981947, + "loss": 1.4088, + "theoretical_loss": 3.8123355394935285, + "tokens_seen": 643170304 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040664994984954865, + "loss": 1.4569, + "theoretical_loss": 3.812295501059366, + "tokens_seen": 643235840 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040663991975927783, + "loss": 1.5972, + "theoretical_loss": 3.8122554678463807, + "tokens_seen": 643301376 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040662988966900707, + "loss": 1.3771, + "theoretical_loss": 3.8122154398533596, + "tokens_seen": 643366912 + }, + { + "epoch": 0.19, + "learning_rate": 0.0004066198595787362, + "loss": 1.4705, + "theoretical_loss": 3.812175417079091, + "tokens_seen": 643432448 + }, + { + "epoch": 0.19, + "learning_rate": 0.00040660982948846543, + "loss": 1.4239, + "theoretical_loss": 3.8121353995223624, + "tokens_seen": 643497984 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040659979939819456, + "loss": 1.4709, + "theoretical_loss": 3.812095387181963, + "tokens_seen": 643563520 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004065897693079238, + "loss": 1.3984, + "theoretical_loss": 3.812055380056682, + "tokens_seen": 643629056 + }, + { + "epoch": 0.2, + "learning_rate": 0.000406579739217653, + "loss": 1.3883, + "theoretical_loss": 3.812015378145308, + "tokens_seen": 643694592 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040656970912738216, + "loss": 1.4565, + "theoretical_loss": 3.811975381446631, + "tokens_seen": 643760128 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040655967903711134, + "loss": 1.5671, + "theoretical_loss": 3.8119353899594413, + "tokens_seen": 643825664 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": -0.001024759141728282, + "objective/train/docs_used": 373533, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.713315486907959, + "objective/train/original_loss": 2.71331524848938, + "objective/train/theoretical_loss": 3.811895403682529, + "objective/train/tokens_used": 664351200, + "objective/train/value_avg": -0.005062103271484375, + "objective/train/value_loss": 5.717682506656274e-05, + "objective/train/value_max": -0.0003528594970703125, + "objective/train/value_min": -0.06561279296875, + "objective/train/value_reward_corr": 0.34230135332911527, + "objective/train/value_std": 0.0035800933837890625, + "objective/train/weight_avg": 0.9923568964004517, + "objective/train/weighted_lm_loss": 2.7033419609069824, + "objective/train/weights_max": 1.791646957397461, + "objective/train/weights_min": 0.19736839830875397, + "theoretical_loss": 3.811895403682529, + "tokens_seen": 643891200 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004065496489468405, + "loss": 1.466, + "theoretical_loss": 3.811895403682529, + "tokens_seen": 643891200 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004065396188565697, + "loss": 1.329, + "theoretical_loss": 3.811855422614686, + "tokens_seen": 643956736 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040652958876629894, + "loss": 1.4104, + "theoretical_loss": 3.811815446754703, + "tokens_seen": 644022272 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040651955867602806, + "loss": 1.4338, + "theoretical_loss": 3.8117754761013725, + "tokens_seen": 644087808 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004065095285857573, + "loss": 1.391, + "theoretical_loss": 3.8117355106534854, + "tokens_seen": 644153344 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004064994984954865, + "loss": 1.4658, + "theoretical_loss": 3.811695550409835, + "tokens_seen": 644218880 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040648946840521566, + "loss": 1.4343, + "theoretical_loss": 3.8116555953692144, + "tokens_seen": 644284416 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040647943831494484, + "loss": 1.4776, + "theoretical_loss": 3.811615645530417, + "tokens_seen": 644349952 + }, + { + "epoch": 0.2, + "learning_rate": 0.000406469408224674, + "loss": 1.5275, + "theoretical_loss": 3.811575700892236, + "tokens_seen": 644415488 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004064593781344032, + "loss": 1.4425, + "theoretical_loss": 3.8115357614534666, + "tokens_seen": 644481024 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040644934804413244, + "loss": 1.4569, + "theoretical_loss": 3.8114958272129034, + "tokens_seen": 644546560 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040643931795386157, + "loss": 1.3862, + "theoretical_loss": 3.8114558981693403, + "tokens_seen": 644612096 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004064292878635908, + "loss": 1.4262, + "theoretical_loss": 3.8114159743215734, + "tokens_seen": 644677632 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040641925777331993, + "loss": 1.4265, + "theoretical_loss": 3.811376055668398, + "tokens_seen": 644743168 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040640922768304916, + "loss": 1.4354, + "theoretical_loss": 3.8113361422086114, + "tokens_seen": 644808704 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040639919759277834, + "loss": 1.4413, + "theoretical_loss": 3.8112962339410092, + "tokens_seen": 644874240 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004063891675025075, + "loss": 1.3785, + "theoretical_loss": 3.8112563308643885, + "tokens_seen": 644939776 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004063791374122367, + "loss": 1.4553, + "theoretical_loss": 3.8112164329775475, + "tokens_seen": 645005312 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004063691073219659, + "loss": 1.4573, + "theoretical_loss": 3.811176540279283, + "tokens_seen": 645070848 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040635907723169507, + "loss": 1.5636, + "theoretical_loss": 3.8111366527683943, + "tokens_seen": 645136384 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004063490471414243, + "loss": 1.4626, + "theoretical_loss": 3.8110967704436787, + "tokens_seen": 645201920 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040633901705115343, + "loss": 1.3756, + "theoretical_loss": 3.8110568933039364, + "tokens_seen": 645267456 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040632898696088267, + "loss": 1.4868, + "theoretical_loss": 3.811017021347966, + "tokens_seen": 645332992 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040631895687061185, + "loss": 1.5283, + "theoretical_loss": 3.810977154574568, + "tokens_seen": 645398528 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040630892678034103, + "loss": 1.4239, + "theoretical_loss": 3.8109372929825422, + "tokens_seen": 645464064 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": 0.0006595560116693377, + "objective/train/docs_used": 374553, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.480903148651123, + "objective/train/original_loss": 3.480903148651123, + "objective/train/theoretical_loss": 3.8108974365706887, + "objective/train/tokens_used": 665989600, + "objective/train/value_avg": -0.004787445068359375, + "objective/train/value_loss": 3.74656701751519e-05, + "objective/train/value_max": -0.00040602684020996094, + "objective/train/value_min": -0.04302978515625, + "objective/train/value_reward_corr": 0.4495311285533097, + "objective/train/value_std": 0.00345611572265625, + "objective/train/weight_avg": 1.008427381515503, + "objective/train/weighted_lm_loss": 3.509122371673584, + "objective/train/weights_max": 1.4383035898208618, + "objective/train/weights_min": 0.6702840924263, + "theoretical_loss": 3.8108974365706887, + "tokens_seen": 645529600 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004062988966900702, + "loss": 1.5038, + "theoretical_loss": 3.8108974365706887, + "tokens_seen": 645529600 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004062888665997994, + "loss": 1.3868, + "theoretical_loss": 3.8108575853378097, + "tokens_seen": 645595136 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040627883650952857, + "loss": 1.5625, + "theoretical_loss": 3.8108177392827063, + "tokens_seen": 645660672 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004062688064192578, + "loss": 1.4397, + "theoretical_loss": 3.810777898404179, + "tokens_seen": 645726208 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040625877632898693, + "loss": 1.4201, + "theoretical_loss": 3.8107380627010317, + "tokens_seen": 645791744 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040624874623871617, + "loss": 1.5192, + "theoretical_loss": 3.810698232172066, + "tokens_seen": 645857280 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004062387161484453, + "loss": 1.434, + "theoretical_loss": 3.810658406816085, + "tokens_seen": 645922816 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040622868605817453, + "loss": 1.3718, + "theoretical_loss": 3.810618586631893, + "tokens_seen": 645988352 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004062186559679037, + "loss": 1.5303, + "theoretical_loss": 3.810578771618293, + "tokens_seen": 646053888 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004062086258776329, + "loss": 1.5252, + "theoretical_loss": 3.810538961774089, + "tokens_seen": 646119424 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004061985957873621, + "loss": 1.4987, + "theoretical_loss": 3.8104991570980857, + "tokens_seen": 646184960 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040618856569709126, + "loss": 1.5106, + "theoretical_loss": 3.810459357589089, + "tokens_seen": 646250496 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040617853560682044, + "loss": 1.5084, + "theoretical_loss": 3.8104195632459033, + "tokens_seen": 646316032 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040616850551654967, + "loss": 1.4082, + "theoretical_loss": 3.8103797740673344, + "tokens_seen": 646381568 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004061584754262788, + "loss": 1.4376, + "theoretical_loss": 3.8103399900521895, + "tokens_seen": 646447104 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040614844533600804, + "loss": 1.6411, + "theoretical_loss": 3.810300211199274, + "tokens_seen": 646512640 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040613841524573727, + "loss": 1.4473, + "theoretical_loss": 3.810260437507395, + "tokens_seen": 646578176 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004061283851554664, + "loss": 1.6345, + "theoretical_loss": 3.810220668975361, + "tokens_seen": 646643712 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040611835506519563, + "loss": 1.5423, + "theoretical_loss": 3.810180905601979, + "tokens_seen": 646709248 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040610832497492476, + "loss": 1.447, + "theoretical_loss": 3.810141147386057, + "tokens_seen": 646774784 + }, + { + "epoch": 0.2, + "learning_rate": 0.000406098294884654, + "loss": 1.5737, + "theoretical_loss": 3.8101013943264035, + "tokens_seen": 646840320 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004060882647943832, + "loss": 1.456, + "theoretical_loss": 3.810061646421828, + "tokens_seen": 646905856 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040607823470411236, + "loss": 1.5758, + "theoretical_loss": 3.8100219036711396, + "tokens_seen": 646971392 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040606820461384154, + "loss": 1.4504, + "theoretical_loss": 3.809982166073148, + "tokens_seen": 647036928 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004060581745235707, + "loss": 1.4926, + "theoretical_loss": 3.8099424336266634, + "tokens_seen": 647102464 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": -0.0006219503120519221, + "objective/train/docs_used": 375146, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9216995239257812, + "objective/train/original_loss": 2.921699285507202, + "objective/train/theoretical_loss": 3.8099027063304964, + "objective/train/tokens_used": 667628000, + "objective/train/value_avg": -0.006786346435546875, + "objective/train/value_loss": 0.0001116600979003124, + "objective/train/value_max": -0.00041413307189941406, + "objective/train/value_min": -0.08209228515625, + "objective/train/value_reward_corr": 0.44806246266548827, + "objective/train/value_std": 0.006389617919921875, + "objective/train/weight_avg": 0.9988403916358948, + "objective/train/weighted_lm_loss": 2.929866075515747, + "objective/train/weights_max": 1.8530654907226562, + "objective/train/weights_min": 0.16970185935497284, + "theoretical_loss": 3.8099027063304964, + "tokens_seen": 647168000 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004060481444332999, + "loss": 1.5044, + "theoretical_loss": 3.8099027063304964, + "tokens_seen": 647168000 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040603811434302914, + "loss": 1.5338, + "theoretical_loss": 3.8098629841834577, + "tokens_seen": 647233536 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040602808425275826, + "loss": 1.4628, + "theoretical_loss": 3.809823267184359, + "tokens_seen": 647299072 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004060180541624875, + "loss": 1.3729, + "theoretical_loss": 3.809783555332012, + "tokens_seen": 647364608 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004060080240722167, + "loss": 1.5267, + "theoretical_loss": 3.809743848625228, + "tokens_seen": 647430144 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040599799398194586, + "loss": 1.5126, + "theoretical_loss": 3.8097041470628206, + "tokens_seen": 647495680 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040598796389167504, + "loss": 1.5135, + "theoretical_loss": 3.809664450643602, + "tokens_seen": 647561216 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004059779338014042, + "loss": 1.3773, + "theoretical_loss": 3.8096247593663866, + "tokens_seen": 647626752 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004059679037111334, + "loss": 1.5147, + "theoretical_loss": 3.8095850732299867, + "tokens_seen": 647692288 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040595787362086264, + "loss": 1.457, + "theoretical_loss": 3.8095453922332174, + "tokens_seen": 647757824 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040594784353059177, + "loss": 1.4191, + "theoretical_loss": 3.8095057163748924, + "tokens_seen": 647823360 + }, + { + "epoch": 0.2, + "learning_rate": 0.000405937813440321, + "loss": 1.4503, + "theoretical_loss": 3.809466045653827, + "tokens_seen": 647888896 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040592778335005013, + "loss": 1.4644, + "theoretical_loss": 3.8094263800688366, + "tokens_seen": 647954432 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040591775325977936, + "loss": 1.5182, + "theoretical_loss": 3.809386719618737, + "tokens_seen": 648019968 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040590772316950854, + "loss": 1.5385, + "theoretical_loss": 3.8093470643023433, + "tokens_seen": 648085504 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004058976930792377, + "loss": 1.5172, + "theoretical_loss": 3.809307414118473, + "tokens_seen": 648151040 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004058876629889669, + "loss": 1.4157, + "theoretical_loss": 3.8092677690659427, + "tokens_seen": 648216576 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004058776328986961, + "loss": 1.6208, + "theoretical_loss": 3.809228129143569, + "tokens_seen": 648282112 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040586760280842527, + "loss": 1.4727, + "theoretical_loss": 3.8091884943501704, + "tokens_seen": 648347648 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004058575727181545, + "loss": 1.3137, + "theoretical_loss": 3.8091488646845644, + "tokens_seen": 648413184 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040584754262788363, + "loss": 1.5193, + "theoretical_loss": 3.80910924014557, + "tokens_seen": 648478720 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040583751253761287, + "loss": 1.3693, + "theoretical_loss": 3.8090696207320054, + "tokens_seen": 648544256 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040582748244734205, + "loss": 1.4389, + "theoretical_loss": 3.80903000644269, + "tokens_seen": 648609792 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040581745235707123, + "loss": 1.5762, + "theoretical_loss": 3.808990397276443, + "tokens_seen": 648675328 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004058074222668004, + "loss": 1.5118, + "theoretical_loss": 3.808950793232085, + "tokens_seen": 648740864 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": 0.00014857824135106057, + "objective/train/docs_used": 375860, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1593120098114014, + "objective/train/original_loss": 3.1593117713928223, + "objective/train/theoretical_loss": 3.808911194308436, + "objective/train/tokens_used": 669266400, + "objective/train/value_avg": -0.004436492919921875, + "objective/train/value_loss": 5.5715670896461233e-05, + "objective/train/value_max": -0.0003459453582763672, + "objective/train/value_min": -0.04559326171875, + "objective/train/value_reward_corr": 0.4191939838821378, + "objective/train/value_std": 0.003345489501953125, + "objective/train/weight_avg": 1.00397789478302, + "objective/train/weighted_lm_loss": 3.1699836254119873, + "objective/train/weights_max": 1.426936388015747, + "objective/train/weights_min": 0.14573626220226288, + "theoretical_loss": 3.808911194308436, + "tokens_seen": 648806400 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004057973921765296, + "loss": 1.4763, + "theoretical_loss": 3.808911194308436, + "tokens_seen": 648806400 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040578736208625877, + "loss": 1.4953, + "theoretical_loss": 3.8088716005043173, + "tokens_seen": 648871936 + }, + { + "epoch": 0.2, + "learning_rate": 0.000405777331995988, + "loss": 1.4736, + "theoretical_loss": 3.8088320118185495, + "tokens_seen": 648937472 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040576730190571713, + "loss": 1.5422, + "theoretical_loss": 3.808792428249954, + "tokens_seen": 649003008 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040575727181544637, + "loss": 1.584, + "theoretical_loss": 3.808752849797353, + "tokens_seen": 649068544 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004057472417251755, + "loss": 1.5014, + "theoretical_loss": 3.808713276459569, + "tokens_seen": 649134080 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040573721163490473, + "loss": 1.5466, + "theoretical_loss": 3.8086737082354247, + "tokens_seen": 649199616 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004057271815446339, + "loss": 1.5264, + "theoretical_loss": 3.808634145123743, + "tokens_seen": 649265152 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004057171514543631, + "loss": 1.3594, + "theoretical_loss": 3.8085945871233475, + "tokens_seen": 649330688 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004057071213640923, + "loss": 1.5367, + "theoretical_loss": 3.8085550342330614, + "tokens_seen": 649396224 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040569709127382146, + "loss": 1.4875, + "theoretical_loss": 3.8085154864517103, + "tokens_seen": 649461760 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040568706118355064, + "loss": 1.535, + "theoretical_loss": 3.808475943778118, + "tokens_seen": 649527296 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004056770310932799, + "loss": 1.4897, + "theoretical_loss": 3.8084364062111096, + "tokens_seen": 649592832 + }, + { + "epoch": 0.2, + "learning_rate": 0.000405667001003009, + "loss": 1.4683, + "theoretical_loss": 3.8083968737495106, + "tokens_seen": 649658368 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040565697091273824, + "loss": 1.4776, + "theoretical_loss": 3.808357346392147, + "tokens_seen": 649723904 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004056469408224674, + "loss": 1.5593, + "theoretical_loss": 3.808317824137845, + "tokens_seen": 649789440 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004056369107321966, + "loss": 1.3559, + "theoretical_loss": 3.8082783069854305, + "tokens_seen": 649854976 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004056268806419258, + "loss": 1.4061, + "theoretical_loss": 3.8082387949337315, + "tokens_seen": 649920512 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040561685055165496, + "loss": 1.4472, + "theoretical_loss": 3.8081992879815747, + "tokens_seen": 649986048 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040560682046138414, + "loss": 1.5265, + "theoretical_loss": 3.8081597861277885, + "tokens_seen": 650051584 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004055967903711134, + "loss": 1.4736, + "theoretical_loss": 3.8081202893712005, + "tokens_seen": 650117120 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004055867602808425, + "loss": 1.4689, + "theoretical_loss": 3.8080807977106397, + "tokens_seen": 650182656 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040557673019057174, + "loss": 1.5489, + "theoretical_loss": 3.808041311144934, + "tokens_seen": 650248192 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040556670010030087, + "loss": 1.4167, + "theoretical_loss": 3.8080018296729143, + "tokens_seen": 650313728 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004055566700100301, + "loss": 1.4051, + "theoretical_loss": 3.8079623532934086, + "tokens_seen": 650379264 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": 0.000985439051873982, + "objective/train/docs_used": 377374, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.674614191055298, + "objective/train/original_loss": 2.6746139526367188, + "objective/train/theoretical_loss": 3.807922882005249, + "objective/train/tokens_used": 670904800, + "objective/train/value_avg": -0.00516510009765625, + "objective/train/value_loss": 4.335774065111764e-05, + "objective/train/value_max": -0.00032258033752441406, + "objective/train/value_min": -0.06561279296875, + "objective/train/value_reward_corr": 0.3310315878086955, + "objective/train/value_std": 0.0039520263671875, + "objective/train/weight_avg": 1.0119447708129883, + "objective/train/weighted_lm_loss": 2.7049167156219482, + "objective/train/weights_max": 1.7246373891830444, + "objective/train/weights_min": 0.45697256922721863, + "theoretical_loss": 3.807922882005249, + "tokens_seen": 650444800 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004055466399197593, + "loss": 1.3811, + "theoretical_loss": 3.807922882005249, + "tokens_seen": 650444800 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040553660982948846, + "loss": 1.4707, + "theoretical_loss": 3.807883415807264, + "tokens_seen": 650510336 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040552657973921764, + "loss": 1.4979, + "theoretical_loss": 3.8078439546982854, + "tokens_seen": 650575872 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004055165496489469, + "loss": 1.3402, + "theoretical_loss": 3.807804498677145, + "tokens_seen": 650641408 + }, + { + "epoch": 0.2, + "learning_rate": 0.000405506519558676, + "loss": 1.4388, + "theoretical_loss": 3.8077650477426728, + "tokens_seen": 650706944 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040549648946840524, + "loss": 1.4069, + "theoretical_loss": 3.8077256018937025, + "tokens_seen": 650772480 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040548645937813437, + "loss": 1.5833, + "theoretical_loss": 3.807686161129065, + "tokens_seen": 650838016 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004054764292878636, + "loss": 1.461, + "theoretical_loss": 3.807646725447595, + "tokens_seen": 650903552 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004054663991975928, + "loss": 1.459, + "theoretical_loss": 3.8076072948481237, + "tokens_seen": 650969088 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040545636910732197, + "loss": 1.4134, + "theoretical_loss": 3.8075678693294863, + "tokens_seen": 651034624 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040544633901705115, + "loss": 1.4437, + "theoretical_loss": 3.8075284488905154, + "tokens_seen": 651100160 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040543630892678033, + "loss": 1.5133, + "theoretical_loss": 3.807489033530046, + "tokens_seen": 651165696 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004054262788365095, + "loss": 1.3758, + "theoretical_loss": 3.807449623246913, + "tokens_seen": 651231232 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040541624874623874, + "loss": 1.5032, + "theoretical_loss": 3.8074102180399514, + "tokens_seen": 651296768 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040540621865596787, + "loss": 1.4023, + "theoretical_loss": 3.807370817907996, + "tokens_seen": 651362304 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004053961885656971, + "loss": 1.4679, + "theoretical_loss": 3.8073314228498836, + "tokens_seen": 651427840 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004053861584754263, + "loss": 1.5099, + "theoretical_loss": 3.80729203286445, + "tokens_seen": 651493376 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040537612838515547, + "loss": 1.4531, + "theoretical_loss": 3.8072526479505315, + "tokens_seen": 651558912 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004053660982948847, + "loss": 1.4785, + "theoretical_loss": 3.807213268106966, + "tokens_seen": 651624448 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040535606820461383, + "loss": 1.4526, + "theoretical_loss": 3.8071738933325903, + "tokens_seen": 651689984 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040534603811434307, + "loss": 1.4244, + "theoretical_loss": 3.8071345236262424, + "tokens_seen": 651755520 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040533600802407225, + "loss": 1.3497, + "theoretical_loss": 3.8070951589867605, + "tokens_seen": 651821056 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040532597793380143, + "loss": 1.5408, + "theoretical_loss": 3.807055799412983, + "tokens_seen": 651886592 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004053159478435306, + "loss": 1.3815, + "theoretical_loss": 3.8070164449037485, + "tokens_seen": 651952128 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004053059177532598, + "loss": 1.5109, + "theoretical_loss": 3.8069770954578974, + "tokens_seen": 652017664 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": -0.0008754754671826959, + "objective/train/docs_used": 377873, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.99634051322937, + "objective/train/original_loss": 2.996340751647949, + "objective/train/theoretical_loss": 3.806937751074268, + "objective/train/tokens_used": 672543200, + "objective/train/value_avg": -0.005138397216796875, + "objective/train/value_loss": 5.539569247048348e-05, + "objective/train/value_max": -0.00022518634796142578, + "objective/train/value_min": -0.07904052734375, + "objective/train/value_reward_corr": 0.4417257341774938, + "objective/train/value_std": 0.004741668701171875, + "objective/train/weight_avg": 0.9939041137695312, + "objective/train/weighted_lm_loss": 2.973649024963379, + "objective/train/weights_max": 1.7272021770477295, + "objective/train/weights_min": 0.5332638025283813, + "theoretical_loss": 3.806937751074268, + "tokens_seen": 652083200 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040529588766298897, + "loss": 1.5106, + "theoretical_loss": 3.806937751074268, + "tokens_seen": 652083200 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004052858575727182, + "loss": 1.5151, + "theoretical_loss": 3.8068984117517015, + "tokens_seen": 652148736 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040527582748244733, + "loss": 1.5543, + "theoretical_loss": 3.806859077489038, + "tokens_seen": 652214272 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040526579739217657, + "loss": 1.4092, + "theoretical_loss": 3.8068197482851183, + "tokens_seen": 652279808 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004052557673019057, + "loss": 1.3442, + "theoretical_loss": 3.806780424138784, + "tokens_seen": 652345344 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040524573721163493, + "loss": 1.4672, + "theoretical_loss": 3.806741105048876, + "tokens_seen": 652410880 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004052357071213641, + "loss": 1.4859, + "theoretical_loss": 3.8067017910142367, + "tokens_seen": 652476416 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004052256770310933, + "loss": 1.5351, + "theoretical_loss": 3.8066624820337083, + "tokens_seen": 652541952 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004052156469408225, + "loss": 1.5718, + "theoretical_loss": 3.8066231781061344, + "tokens_seen": 652607488 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040520561685055166, + "loss": 1.3194, + "theoretical_loss": 3.806583879230357, + "tokens_seen": 652673024 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040519558676028084, + "loss": 1.3596, + "theoretical_loss": 3.80654458540522, + "tokens_seen": 652738560 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004051855566700101, + "loss": 1.3007, + "theoretical_loss": 3.806505296629567, + "tokens_seen": 652804096 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004051755265797392, + "loss": 1.4825, + "theoretical_loss": 3.8064660129022436, + "tokens_seen": 652869632 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040516549648946844, + "loss": 1.3764, + "theoretical_loss": 3.806426734222093, + "tokens_seen": 652935168 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004051554663991976, + "loss": 1.505, + "theoretical_loss": 3.806387460587961, + "tokens_seen": 653000704 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004051454363089268, + "loss": 1.5417, + "theoretical_loss": 3.806348191998693, + "tokens_seen": 653066240 + }, + { + "epoch": 0.2, + "learning_rate": 0.000405135406218656, + "loss": 1.3769, + "theoretical_loss": 3.806308928453134, + "tokens_seen": 653131776 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040512537612838516, + "loss": 1.469, + "theoretical_loss": 3.8062696699501313, + "tokens_seen": 653197312 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040511534603811434, + "loss": 1.5034, + "theoretical_loss": 3.806230416488531, + "tokens_seen": 653262848 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004051053159478436, + "loss": 1.4649, + "theoretical_loss": 3.80619116806718, + "tokens_seen": 653328384 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004050952858575727, + "loss": 1.4497, + "theoretical_loss": 3.806151924684925, + "tokens_seen": 653393920 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040508525576730194, + "loss": 1.46, + "theoretical_loss": 3.8061126863406147, + "tokens_seen": 653459456 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040507522567703107, + "loss": 1.5262, + "theoretical_loss": 3.806073453033097, + "tokens_seen": 653524992 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004050651955867603, + "loss": 1.3239, + "theoretical_loss": 3.8060342247612207, + "tokens_seen": 653590528 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004050551654964895, + "loss": 1.4829, + "theoretical_loss": 3.805995001523833, + "tokens_seen": 653656064 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": 0.0009562300401739776, + "objective/train/docs_used": 379015, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9973948001861572, + "objective/train/original_loss": 2.9973948001861572, + "objective/train/theoretical_loss": 3.805955783319785, + "objective/train/tokens_used": 674181600, + "objective/train/value_avg": -0.005420684814453125, + "objective/train/value_loss": 5.002917896490544e-05, + "objective/train/value_max": -0.00030303001403808594, + "objective/train/value_min": -0.1019287109375, + "objective/train/value_reward_corr": 0.4527878574990713, + "objective/train/value_std": 0.004940032958984375, + "objective/train/weight_avg": 1.0119174718856812, + "objective/train/weighted_lm_loss": 3.0309956073760986, + "objective/train/weights_max": 2.0950276851654053, + "objective/train/weights_min": 0.34370657801628113, + "theoretical_loss": 3.805955783319785, + "tokens_seen": 653721600 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040504513540621866, + "loss": 1.4265, + "theoretical_loss": 3.805955783319785, + "tokens_seen": 653721600 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040503510531594784, + "loss": 1.4445, + "theoretical_loss": 3.805916570147925, + "tokens_seen": 653787136 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004050250752256771, + "loss": 1.4706, + "theoretical_loss": 3.805877362007104, + "tokens_seen": 653852672 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004050150451354062, + "loss": 1.4099, + "theoretical_loss": 3.8058381588961714, + "tokens_seen": 653918208 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040500501504513544, + "loss": 1.494, + "theoretical_loss": 3.805798960813979, + "tokens_seen": 653983744 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040499498495486457, + "loss": 1.3144, + "theoretical_loss": 3.805759767759377, + "tokens_seen": 654049280 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004049849548645938, + "loss": 1.4083, + "theoretical_loss": 3.805720579731217, + "tokens_seen": 654114816 + }, + { + "epoch": 0.2, + "learning_rate": 0.000404974924774323, + "loss": 1.457, + "theoretical_loss": 3.8056813967283514, + "tokens_seen": 654180352 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040496489468405217, + "loss": 1.323, + "theoretical_loss": 3.8056422187496315, + "tokens_seen": 654245888 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040495486459378135, + "loss": 1.467, + "theoretical_loss": 3.8056030457939114, + "tokens_seen": 654311424 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040494483450351053, + "loss": 1.4095, + "theoretical_loss": 3.8055638778600427, + "tokens_seen": 654376960 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004049348044132397, + "loss": 1.4668, + "theoretical_loss": 3.805524714946879, + "tokens_seen": 654442496 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040492477432296894, + "loss": 1.4335, + "theoretical_loss": 3.8054855570532746, + "tokens_seen": 654508032 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040491474423269807, + "loss": 1.5069, + "theoretical_loss": 3.8054464041780838, + "tokens_seen": 654573568 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004049047141424273, + "loss": 1.4972, + "theoretical_loss": 3.80540725632016, + "tokens_seen": 654639104 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040489468405215643, + "loss": 1.3757, + "theoretical_loss": 3.8053681134783592, + "tokens_seen": 654704640 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040488465396188567, + "loss": 1.5187, + "theoretical_loss": 3.805328975651536, + "tokens_seen": 654770176 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040487462387161485, + "loss": 1.384, + "theoretical_loss": 3.8052898428385458, + "tokens_seen": 654835712 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040486459378134403, + "loss": 1.4186, + "theoretical_loss": 3.8052507150382455, + "tokens_seen": 654901248 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004048545636910732, + "loss": 1.4962, + "theoretical_loss": 3.8052115922494907, + "tokens_seen": 654966784 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040484453360080245, + "loss": 1.4305, + "theoretical_loss": 3.8051724744711386, + "tokens_seen": 655032320 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004048345035105316, + "loss": 1.4315, + "theoretical_loss": 3.805133361702046, + "tokens_seen": 655097856 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004048244734202608, + "loss": 1.4603, + "theoretical_loss": 3.805094253941071, + "tokens_seen": 655163392 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040481444332998994, + "loss": 1.4321, + "theoretical_loss": 3.8050551511870707, + "tokens_seen": 655228928 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040480441323971917, + "loss": 1.3948, + "theoretical_loss": 3.805016053438904, + "tokens_seen": 655294464 + }, + { + "debugging/Self-BLEU-5": 0.28723315400076466, + "debugging/distinct-1-grams": 0.831755762171831, + "debugging/distinct-2-grams": 0.9583018392542203, + "debugging/entropy-1-grams": 5.453202391069032, + "debugging/entropy-2-grams": 5.995093667215148, + "debugging/length": 442.0, + "debugging/num_segments": 7, + "debugging/raw_token_scores_avg": 0.006729413755238056, + "debugging/raw_token_scores_std": 0.008488241583108902, + "debugging/score": 0.009230784774488613, + "debugging/score_std": 0.005285791546055549, + "epoch": 0.2, + "objective/train/advantage_avg": -0.0020214514806866646, + "objective/train/docs_used": 379813, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.894972324371338, + "objective/train/original_loss": 2.894972324371338, + "objective/train/theoretical_loss": 3.804976960695429, + "objective/train/tokens_used": 675820000, + "objective/train/value_avg": -0.004711151123046875, + "objective/train/value_loss": 7.106566044967622e-05, + "objective/train/value_max": -0.0003669261932373047, + "objective/train/value_min": -0.05108642578125, + "objective/train/value_reward_corr": 0.28904580023997, + "objective/train/value_std": 0.0034122467041015625, + "objective/train/weight_avg": 0.9830307960510254, + "objective/train/weighted_lm_loss": 2.8512866497039795, + "objective/train/weights_max": 1.6667311191558838, + "objective/train/weights_min": 0.2096860110759735, + "theoretical_loss": 3.804976960695429, + "tokens_seen": 655360000 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040479438314944835, + "loss": 1.4539, + "theoretical_loss": 3.804976960695429, + "tokens_seen": 655360000 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040478435305917753, + "loss": 1.383, + "theoretical_loss": 3.804937872955505, + "tokens_seen": 655425536 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004047743229689067, + "loss": 1.6268, + "theoretical_loss": 3.804898790217991, + "tokens_seen": 655491072 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004047642928786359, + "loss": 1.5234, + "theoretical_loss": 3.804859712481748, + "tokens_seen": 655556608 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004047542627883651, + "loss": 1.4406, + "theoretical_loss": 3.804820639745634, + "tokens_seen": 655622144 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004047442326980943, + "loss": 1.5384, + "theoretical_loss": 3.804781572008511, + "tokens_seen": 655687680 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040473420260782344, + "loss": 1.4252, + "theoretical_loss": 3.80474250926924, + "tokens_seen": 655753216 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004047241725175527, + "loss": 1.3883, + "theoretical_loss": 3.8047034515266818, + "tokens_seen": 655818752 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004047141424272818, + "loss": 1.4972, + "theoretical_loss": 3.804664398779698, + "tokens_seen": 655884288 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040470411233701104, + "loss": 1.5793, + "theoretical_loss": 3.80462535102715, + "tokens_seen": 655949824 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004046940822467402, + "loss": 1.5726, + "theoretical_loss": 3.8045863082679015, + "tokens_seen": 656015360 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004046840521564694, + "loss": 1.4044, + "theoretical_loss": 3.8045472705008145, + "tokens_seen": 656080896 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004046740220661986, + "loss": 1.5171, + "theoretical_loss": 3.804508237724752, + "tokens_seen": 656146432 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004046639919759278, + "loss": 1.4502, + "theoretical_loss": 3.8044692099385777, + "tokens_seen": 656211968 + }, + { + "epoch": 0.2, + "learning_rate": 0.000404653961885657, + "loss": 1.4012, + "theoretical_loss": 3.8044301871411554, + "tokens_seen": 656277504 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004046439317953862, + "loss": 1.4879, + "theoretical_loss": 3.804391169331349, + "tokens_seen": 656343040 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040463390170511536, + "loss": 1.5091, + "theoretical_loss": 3.8043521565080236, + "tokens_seen": 656408576 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040462387161484454, + "loss": 1.4364, + "theoretical_loss": 3.804313148670044, + "tokens_seen": 656474112 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004046138415245738, + "loss": 1.4712, + "theoretical_loss": 3.804274145816276, + "tokens_seen": 656539648 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004046038114343029, + "loss": 1.5096, + "theoretical_loss": 3.804235147945584, + "tokens_seen": 656605184 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040459378134403214, + "loss": 1.5253, + "theoretical_loss": 3.804196155056835, + "tokens_seen": 656670720 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040458375125376127, + "loss": 1.4782, + "theoretical_loss": 3.8041571671488956, + "tokens_seen": 656736256 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004045737211634905, + "loss": 1.3764, + "theoretical_loss": 3.8041181842206324, + "tokens_seen": 656801792 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004045636910732197, + "loss": 1.4508, + "theoretical_loss": 3.804079206270913, + "tokens_seen": 656867328 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040455366098294886, + "loss": 1.5204, + "theoretical_loss": 3.804040233298604, + "tokens_seen": 656932864 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": -0.0017896050121635199, + "objective/train/docs_used": 381022, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8255677223205566, + "objective/train/original_loss": 2.8255677223205566, + "objective/train/theoretical_loss": 3.804001265302574, + "objective/train/tokens_used": 677458400, + "objective/train/value_avg": -0.00616455078125, + "objective/train/value_loss": 8.186986087821424e-05, + "objective/train/value_max": -0.00044417381286621094, + "objective/train/value_min": -0.06781005859375, + "objective/train/value_reward_corr": 0.374044511890477, + "objective/train/value_std": 0.00469970703125, + "objective/train/weight_avg": 0.9859269261360168, + "objective/train/weighted_lm_loss": 2.7991292476654053, + "objective/train/weights_max": 1.8231782913208008, + "objective/train/weights_min": 0.42926082015037537, + "theoretical_loss": 3.804001265302574, + "tokens_seen": 656998400 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040454363089267804, + "loss": 1.4009, + "theoretical_loss": 3.804001265302574, + "tokens_seen": 656998400 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004045336008024073, + "loss": 1.5292, + "theoretical_loss": 3.8039623022816915, + "tokens_seen": 657063936 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004045235707121364, + "loss": 1.4645, + "theoretical_loss": 3.803923344234825, + "tokens_seen": 657129472 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040451354062186564, + "loss": 1.469, + "theoretical_loss": 3.803884391160843, + "tokens_seen": 657195008 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040450351053159477, + "loss": 1.4873, + "theoretical_loss": 3.8038454430586155, + "tokens_seen": 657260544 + }, + { + "epoch": 0.2, + "learning_rate": 0.000404493480441324, + "loss": 1.5228, + "theoretical_loss": 3.803806499927012, + "tokens_seen": 657326080 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004044834503510532, + "loss": 1.489, + "theoretical_loss": 3.8037675617649036, + "tokens_seen": 657391616 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040447342026078237, + "loss": 1.6321, + "theoretical_loss": 3.803728628571159, + "tokens_seen": 657457152 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040446339017051155, + "loss": 1.531, + "theoretical_loss": 3.8036897003446506, + "tokens_seen": 657522688 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040445336008024073, + "loss": 1.4415, + "theoretical_loss": 3.8036507770842496, + "tokens_seen": 657588224 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004044433299899699, + "loss": 1.4353, + "theoretical_loss": 3.803611858788827, + "tokens_seen": 657653760 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040443329989969915, + "loss": 1.5869, + "theoretical_loss": 3.8035729454572547, + "tokens_seen": 657719296 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040442326980942827, + "loss": 1.3183, + "theoretical_loss": 3.8035340370884057, + "tokens_seen": 657784832 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004044132397191575, + "loss": 1.3959, + "theoretical_loss": 3.8034951336811527, + "tokens_seen": 657850368 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040440320962888663, + "loss": 1.5953, + "theoretical_loss": 3.803456235234368, + "tokens_seen": 657915904 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040439317953861587, + "loss": 1.2797, + "theoretical_loss": 3.8034173417469264, + "tokens_seen": 657981440 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040438314944834505, + "loss": 1.4942, + "theoretical_loss": 3.8033784532177, + "tokens_seen": 658046976 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040437311935807423, + "loss": 1.4039, + "theoretical_loss": 3.803339569645565, + "tokens_seen": 658112512 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004043630892678034, + "loss": 1.5388, + "theoretical_loss": 3.8033006910293947, + "tokens_seen": 658178048 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040435305917753265, + "loss": 1.3077, + "theoretical_loss": 3.8032618173680643, + "tokens_seen": 658243584 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004043430290872618, + "loss": 1.5249, + "theoretical_loss": 3.803222948660449, + "tokens_seen": 658309120 + }, + { + "epoch": 0.2, + "learning_rate": 0.000404332998996991, + "loss": 1.4197, + "theoretical_loss": 3.803184084905425, + "tokens_seen": 658374656 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040432296890672014, + "loss": 1.4484, + "theoretical_loss": 3.8031452261018677, + "tokens_seen": 658440192 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040431293881644937, + "loss": 1.4437, + "theoretical_loss": 3.803106372248654, + "tokens_seen": 658505728 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040430290872617855, + "loss": 1.4588, + "theoretical_loss": 3.8030675233446605, + "tokens_seen": 658571264 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": -0.0005694591673091054, + "objective/train/docs_used": 381585, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9409446716308594, + "objective/train/original_loss": 2.9409446716308594, + "objective/train/theoretical_loss": 3.8030286793887647, + "objective/train/tokens_used": 679096800, + "objective/train/value_avg": -0.005634307861328125, + "objective/train/value_loss": 6.29876449238509e-05, + "objective/train/value_max": -0.0003101825714111328, + "objective/train/value_min": -0.0699462890625, + "objective/train/value_reward_corr": 0.4639499716591038, + "objective/train/value_std": 0.0052642822265625, + "objective/train/weight_avg": 0.9973323941230774, + "objective/train/weighted_lm_loss": 2.9282989501953125, + "objective/train/weights_max": 1.6035022735595703, + "objective/train/weights_min": 0.294904500246048, + "theoretical_loss": 3.8030286793887647, + "tokens_seen": 658636800 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040429287863590773, + "loss": 1.5053, + "theoretical_loss": 3.8030286793887647, + "tokens_seen": 658636800 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004042828485456369, + "loss": 1.4362, + "theoretical_loss": 3.8029898403798432, + "tokens_seen": 658702336 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004042728184553661, + "loss": 1.3726, + "theoretical_loss": 3.802951006316775, + "tokens_seen": 658767872 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004042627883650953, + "loss": 1.4922, + "theoretical_loss": 3.8029121771984373, + "tokens_seen": 658833408 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004042527582748245, + "loss": 1.4568, + "theoretical_loss": 3.8028733530237093, + "tokens_seen": 658898944 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040424272818455364, + "loss": 1.4402, + "theoretical_loss": 3.8028345337914704, + "tokens_seen": 658964480 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004042326980942829, + "loss": 1.4609, + "theoretical_loss": 3.8027957195005992, + "tokens_seen": 659030016 + }, + { + "epoch": 0.2, + "learning_rate": 0.000404222668004012, + "loss": 1.4824, + "theoretical_loss": 3.8027569101499754, + "tokens_seen": 659095552 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040421263791374124, + "loss": 1.48, + "theoretical_loss": 3.8027181057384793, + "tokens_seen": 659161088 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004042026078234704, + "loss": 1.4097, + "theoretical_loss": 3.8026793062649915, + "tokens_seen": 659226624 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004041925777331996, + "loss": 1.406, + "theoretical_loss": 3.8026405117283932, + "tokens_seen": 659292160 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004041825476429288, + "loss": 1.5106, + "theoretical_loss": 3.802601722127565, + "tokens_seen": 659357696 + }, + { + "epoch": 0.2, + "learning_rate": 0.000404172517552658, + "loss": 1.3482, + "theoretical_loss": 3.8025629374613876, + "tokens_seen": 659423232 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040416248746238714, + "loss": 1.4128, + "theoretical_loss": 3.8025241577287447, + "tokens_seen": 659488768 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004041524573721164, + "loss": 1.5184, + "theoretical_loss": 3.8024853829285172, + "tokens_seen": 659554304 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004041424272818455, + "loss": 1.3766, + "theoretical_loss": 3.8024466130595886, + "tokens_seen": 659619840 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040413239719157474, + "loss": 1.3645, + "theoretical_loss": 3.8024078481208416, + "tokens_seen": 659685376 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004041223671013039, + "loss": 1.4603, + "theoretical_loss": 3.802369088111159, + "tokens_seen": 659750912 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004041123370110331, + "loss": 1.449, + "theoretical_loss": 3.802330333029426, + "tokens_seen": 659816448 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004041023069207623, + "loss": 1.5078, + "theoretical_loss": 3.802291582874525, + "tokens_seen": 659881984 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040409227683049147, + "loss": 1.396, + "theoretical_loss": 3.802252837645341, + "tokens_seen": 659947520 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040408224674022065, + "loss": 1.4852, + "theoretical_loss": 3.8022140973407588, + "tokens_seen": 660013056 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004040722166499499, + "loss": 1.4461, + "theoretical_loss": 3.802175361959664, + "tokens_seen": 660078592 + }, + { + "epoch": 0.2, + "learning_rate": 0.000404062186559679, + "loss": 1.5345, + "theoretical_loss": 3.8021366315009417, + "tokens_seen": 660144128 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040405215646940824, + "loss": 1.4226, + "theoretical_loss": 3.8020979059634783, + "tokens_seen": 660209664 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": 0.00032565149012953043, + "objective/train/docs_used": 382875, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2607667446136475, + "objective/train/original_loss": 3.2607665061950684, + "objective/train/theoretical_loss": 3.8020591853461596, + "objective/train/tokens_used": 680735200, + "objective/train/value_avg": -0.0057220458984375, + "objective/train/value_loss": 6.170400592964143e-05, + "objective/train/value_max": -0.0005397796630859375, + "objective/train/value_min": -0.060516357421875, + "objective/train/value_reward_corr": 0.3315311947170764, + "objective/train/value_std": 0.004283905029296875, + "objective/train/weight_avg": 1.006174087524414, + "objective/train/weighted_lm_loss": 3.293900966644287, + "objective/train/weights_max": 1.5628312826156616, + "objective/train/weights_min": 0.4095066487789154, + "theoretical_loss": 3.8020591853461596, + "tokens_seen": 660275200 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040404212637913737, + "loss": 1.4958, + "theoretical_loss": 3.8020591853461596, + "tokens_seen": 660275200 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004040320962888666, + "loss": 1.4104, + "theoretical_loss": 3.8020204696478723, + "tokens_seen": 660340736 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004040220661985958, + "loss": 1.3732, + "theoretical_loss": 3.8019817588675036, + "tokens_seen": 660406272 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040401203610832497, + "loss": 1.4205, + "theoretical_loss": 3.8019430530039404, + "tokens_seen": 660471808 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040400200601805415, + "loss": 1.4375, + "theoretical_loss": 3.801904352056071, + "tokens_seen": 660537344 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004039919759277834, + "loss": 1.3942, + "theoretical_loss": 3.801865656022783, + "tokens_seen": 660602880 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004039819458375125, + "loss": 1.4974, + "theoretical_loss": 3.8018269649029652, + "tokens_seen": 660668416 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040397191574724175, + "loss": 1.4205, + "theoretical_loss": 3.801788278695507, + "tokens_seen": 660733952 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004039618856569709, + "loss": 1.4557, + "theoretical_loss": 3.8017495973992954, + "tokens_seen": 660799488 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004039518555667001, + "loss": 1.4566, + "theoretical_loss": 3.801710921013222, + "tokens_seen": 660865024 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004039418254764293, + "loss": 1.3559, + "theoretical_loss": 3.801672249536176, + "tokens_seen": 660930560 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040393179538615847, + "loss": 1.4502, + "theoretical_loss": 3.801633582967048, + "tokens_seen": 660996096 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040392176529588765, + "loss": 1.4717, + "theoretical_loss": 3.8015949213047278, + "tokens_seen": 661061632 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040391173520561683, + "loss": 1.3786, + "theoretical_loss": 3.8015562645481067, + "tokens_seen": 661127168 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040390170511534607, + "loss": 1.4069, + "theoretical_loss": 3.8015176126960766, + "tokens_seen": 661192704 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040389167502507525, + "loss": 1.4127, + "theoretical_loss": 3.8014789657475285, + "tokens_seen": 661258240 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040388164493480443, + "loss": 1.3798, + "theoretical_loss": 3.801440323701355, + "tokens_seen": 661323776 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004038716148445336, + "loss": 1.3953, + "theoretical_loss": 3.801401686556448, + "tokens_seen": 661389312 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040386158475426285, + "loss": 1.4133, + "theoretical_loss": 3.8013630543117003, + "tokens_seen": 661454848 + }, + { + "epoch": 0.2, + "learning_rate": 0.000403851554663992, + "loss": 1.5192, + "theoretical_loss": 3.801324426966006, + "tokens_seen": 661520384 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004038415245737212, + "loss": 1.4761, + "theoretical_loss": 3.8012858045182565, + "tokens_seen": 661585920 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040383149448345034, + "loss": 1.4558, + "theoretical_loss": 3.801247186967348, + "tokens_seen": 661651456 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040382146439317957, + "loss": 1.4932, + "theoretical_loss": 3.8012085743121733, + "tokens_seen": 661716992 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040381143430290875, + "loss": 1.5187, + "theoretical_loss": 3.801169966551627, + "tokens_seen": 661782528 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040380140421263794, + "loss": 1.3698, + "theoretical_loss": 3.801131363684605, + "tokens_seen": 661848064 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": 0.00030529306968674064, + "objective/train/docs_used": 383560, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8767240047454834, + "objective/train/original_loss": 2.8767242431640625, + "objective/train/theoretical_loss": 3.8010927657100013, + "objective/train/tokens_used": 682373600, + "objective/train/value_avg": -0.00548553466796875, + "objective/train/value_loss": 7.152105536079034e-05, + "objective/train/value_max": -0.00035834312438964844, + "objective/train/value_min": -0.06634521484375, + "objective/train/value_reward_corr": 0.32383399420096554, + "objective/train/value_std": 0.00482177734375, + "objective/train/weight_avg": 1.0063387155532837, + "objective/train/weighted_lm_loss": 2.8878862857818604, + "objective/train/weights_max": 1.8048174381256104, + "objective/train/weights_min": 0.14162038266658783, + "theoretical_loss": 3.8010927657100013, + "tokens_seen": 661913600 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004037913741223671, + "loss": 1.37, + "theoretical_loss": 3.8010927657100013, + "tokens_seen": 661913600 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004037813440320963, + "loss": 1.4241, + "theoretical_loss": 3.801054172626712, + "tokens_seen": 661979136 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004037713139418255, + "loss": 1.5975, + "theoretical_loss": 3.8010155844336344, + "tokens_seen": 662044672 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004037612838515547, + "loss": 1.4355, + "theoretical_loss": 3.8009770011296626, + "tokens_seen": 662110208 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040375125376128384, + "loss": 1.3476, + "theoretical_loss": 3.8009384227136946, + "tokens_seen": 662175744 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004037412236710131, + "loss": 1.3724, + "theoretical_loss": 3.8008998491846278, + "tokens_seen": 662241280 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004037311935807422, + "loss": 1.4538, + "theoretical_loss": 3.8008612805413593, + "tokens_seen": 662306816 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040372116349047144, + "loss": 1.3694, + "theoretical_loss": 3.800822716782786, + "tokens_seen": 662372352 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004037111334002006, + "loss": 1.3911, + "theoretical_loss": 3.800784157907807, + "tokens_seen": 662437888 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004037011033099298, + "loss": 1.3419, + "theoretical_loss": 3.800745603915321, + "tokens_seen": 662503424 + }, + { + "epoch": 0.2, + "learning_rate": 0.000403691073219659, + "loss": 1.4048, + "theoretical_loss": 3.800707054804226, + "tokens_seen": 662568960 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004036810431293882, + "loss": 1.6183, + "theoretical_loss": 3.800668510573422, + "tokens_seen": 662634496 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040367101303911734, + "loss": 1.4859, + "theoretical_loss": 3.8006299712218086, + "tokens_seen": 662700032 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004036609829488466, + "loss": 1.349, + "theoretical_loss": 3.800591436748285, + "tokens_seen": 662765568 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004036509528585757, + "loss": 1.4884, + "theoretical_loss": 3.8005529071517516, + "tokens_seen": 662831104 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040364092276830494, + "loss": 1.3204, + "theoretical_loss": 3.8005143824311096, + "tokens_seen": 662896640 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004036308926780341, + "loss": 1.4055, + "theoretical_loss": 3.8004758625852597, + "tokens_seen": 662962176 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004036208625877633, + "loss": 1.3878, + "theoretical_loss": 3.8004373476131037, + "tokens_seen": 663027712 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004036108324974925, + "loss": 1.3915, + "theoretical_loss": 3.8003988375135425, + "tokens_seen": 663093248 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040360080240722167, + "loss": 1.5062, + "theoretical_loss": 3.8003603322854786, + "tokens_seen": 663158784 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040359077231695085, + "loss": 1.3281, + "theoretical_loss": 3.800321831927815, + "tokens_seen": 663224320 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004035807422266801, + "loss": 1.5103, + "theoretical_loss": 3.8002833364394535, + "tokens_seen": 663289856 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004035707121364092, + "loss": 1.2867, + "theoretical_loss": 3.8002448458192974, + "tokens_seen": 663355392 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040356068204613844, + "loss": 1.4555, + "theoretical_loss": 3.800206360066251, + "tokens_seen": 663420928 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040355065195586757, + "loss": 1.2659, + "theoretical_loss": 3.8001678791792175, + "tokens_seen": 663486464 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": 0.0006718810182064772, + "objective/train/docs_used": 384103, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.494300603866577, + "objective/train/original_loss": 2.4943008422851562, + "objective/train/theoretical_loss": 3.8001294031571007, + "objective/train/tokens_used": 684012000, + "objective/train/value_avg": -0.0042724609375, + "objective/train/value_loss": 3.468226350378245e-05, + "objective/train/value_max": -0.00051116943359375, + "objective/train/value_min": -0.05108642578125, + "objective/train/value_reward_corr": 0.3042554563204558, + "objective/train/value_std": 0.0030307769775390625, + "objective/train/weight_avg": 1.0083433389663696, + "objective/train/weighted_lm_loss": 2.508721351623535, + "objective/train/weights_max": 1.5858501195907593, + "objective/train/weights_min": 0.29260730743408203, + "theoretical_loss": 3.8001294031571007, + "tokens_seen": 663552000 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004035406218655968, + "loss": 1.318, + "theoretical_loss": 3.8001294031571007, + "tokens_seen": 663552000 + }, + { + "epoch": 0.2, + "learning_rate": 0.000403530591775326, + "loss": 1.4652, + "theoretical_loss": 3.800090931998806, + "tokens_seen": 663617536 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040352056168505517, + "loss": 1.4697, + "theoretical_loss": 3.800052465703238, + "tokens_seen": 663683072 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040351053159478435, + "loss": 1.4613, + "theoretical_loss": 3.8000140042693022, + "tokens_seen": 663748608 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004035005015045136, + "loss": 1.3568, + "theoretical_loss": 3.7999755476959036, + "tokens_seen": 663814144 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004034904714142427, + "loss": 1.38, + "theoretical_loss": 3.7999370959819494, + "tokens_seen": 663879680 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040348044132397195, + "loss": 1.3822, + "theoretical_loss": 3.799898649126344, + "tokens_seen": 663945216 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004034704112337011, + "loss": 1.4736, + "theoretical_loss": 3.799860207127996, + "tokens_seen": 664010752 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004034603811434303, + "loss": 1.3457, + "theoretical_loss": 3.7998217699858112, + "tokens_seen": 664076288 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004034503510531595, + "loss": 1.4212, + "theoretical_loss": 3.799783337698697, + "tokens_seen": 664141824 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040344032096288867, + "loss": 1.5063, + "theoretical_loss": 3.799744910265563, + "tokens_seen": 664207360 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040343029087261785, + "loss": 1.2918, + "theoretical_loss": 3.799706487685315, + "tokens_seen": 664272896 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040342026078234703, + "loss": 1.4689, + "theoretical_loss": 3.799668069956862, + "tokens_seen": 664338432 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004034102306920762, + "loss": 1.3474, + "theoretical_loss": 3.7996296570791133, + "tokens_seen": 664403968 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040340020060180545, + "loss": 1.4804, + "theoretical_loss": 3.7995912490509784, + "tokens_seen": 664469504 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004033901705115346, + "loss": 1.5381, + "theoretical_loss": 3.799552845871366, + "tokens_seen": 664535040 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004033801404212638, + "loss": 1.3918, + "theoretical_loss": 3.7995144475391864, + "tokens_seen": 664600576 + }, + { + "epoch": 0.2, + "learning_rate": 0.000403370110330993, + "loss": 1.3272, + "theoretical_loss": 3.79947605405335, + "tokens_seen": 664666112 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004033600802407222, + "loss": 1.4586, + "theoretical_loss": 3.799437665412767, + "tokens_seen": 664731648 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040335005015045136, + "loss": 1.3686, + "theoretical_loss": 3.799399281616348, + "tokens_seen": 664797184 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040334002006018054, + "loss": 1.3383, + "theoretical_loss": 3.799360902663005, + "tokens_seen": 664862720 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004033299899699097, + "loss": 1.4548, + "theoretical_loss": 3.7993225285516496, + "tokens_seen": 664928256 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040331995987963895, + "loss": 1.4375, + "theoretical_loss": 3.799284159281193, + "tokens_seen": 664993792 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004033099297893681, + "loss": 1.5022, + "theoretical_loss": 3.799245794850548, + "tokens_seen": 665059328 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004032998996990973, + "loss": 1.3734, + "theoretical_loss": 3.799207435258628, + "tokens_seen": 665124864 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": -0.003535562427714467, + "objective/train/docs_used": 385247, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0831186771392822, + "objective/train/original_loss": 3.0831189155578613, + "objective/train/theoretical_loss": 3.7991690805043445, + "objective/train/tokens_used": 685650400, + "objective/train/value_avg": -0.006744384765625, + "objective/train/value_loss": 0.00010911256686085835, + "objective/train/value_max": -0.0002779960632324219, + "objective/train/value_min": -0.0816650390625, + "objective/train/value_reward_corr": 0.5083459119675269, + "objective/train/value_std": 0.0069427490234375, + "objective/train/weight_avg": 0.9697949290275574, + "objective/train/weighted_lm_loss": 3.020491361618042, + "objective/train/weights_max": 2.262907028198242, + "objective/train/weights_min": 0.616550624370575, + "theoretical_loss": 3.7991690805043445, + "tokens_seen": 665190400 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040328986960882644, + "loss": 1.5063, + "theoretical_loss": 3.7991690805043445, + "tokens_seen": 665190400 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004032798395185557, + "loss": 1.4629, + "theoretical_loss": 3.7991307305866124, + "tokens_seen": 665255936 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040326980942828486, + "loss": 1.4177, + "theoretical_loss": 3.799092385504345, + "tokens_seen": 665321472 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040325977933801404, + "loss": 1.439, + "theoretical_loss": 3.7990540452564554, + "tokens_seen": 665387008 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004032497492477432, + "loss": 1.4609, + "theoretical_loss": 3.799015709841859, + "tokens_seen": 665452544 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004032397191574724, + "loss": 1.4992, + "theoretical_loss": 3.7989773792594708, + "tokens_seen": 665518080 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004032296890672016, + "loss": 1.3733, + "theoretical_loss": 3.7989390535082057, + "tokens_seen": 665583616 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004032196589769308, + "loss": 1.3345, + "theoretical_loss": 3.7989007325869784, + "tokens_seen": 665649152 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040320962888665995, + "loss": 1.5054, + "theoretical_loss": 3.7988624164947056, + "tokens_seen": 665714688 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004031995987963892, + "loss": 1.347, + "theoretical_loss": 3.7988241052303033, + "tokens_seen": 665780224 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040318956870611836, + "loss": 1.5363, + "theoretical_loss": 3.798785798792688, + "tokens_seen": 665845760 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040317953861584754, + "loss": 1.4182, + "theoretical_loss": 3.798747497180777, + "tokens_seen": 665911296 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004031695085255767, + "loss": 1.4402, + "theoretical_loss": 3.7987092003934864, + "tokens_seen": 665976832 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004031594784353059, + "loss": 1.3593, + "theoretical_loss": 3.7986709084297354, + "tokens_seen": 666042368 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040314944834503514, + "loss": 1.4424, + "theoretical_loss": 3.7986326212884407, + "tokens_seen": 666107904 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004031394182547643, + "loss": 1.4033, + "theoretical_loss": 3.7985943389685213, + "tokens_seen": 666173440 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004031293881644935, + "loss": 1.367, + "theoretical_loss": 3.7985560614688945, + "tokens_seen": 666238976 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004031193580742227, + "loss": 1.317, + "theoretical_loss": 3.798517788788481, + "tokens_seen": 666304512 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040310932798395187, + "loss": 1.5601, + "theoretical_loss": 3.7984795209261994, + "tokens_seen": 666370048 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040309929789368105, + "loss": 1.4547, + "theoretical_loss": 3.79844125788097, + "tokens_seen": 666435584 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004030892678034103, + "loss": 1.4879, + "theoretical_loss": 3.7984029996517115, + "tokens_seen": 666501120 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004030792377131394, + "loss": 1.3324, + "theoretical_loss": 3.798364746237345, + "tokens_seen": 666566656 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040306920762286864, + "loss": 1.4284, + "theoretical_loss": 3.7983264976367916, + "tokens_seen": 666632192 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040305917753259777, + "loss": 1.4018, + "theoretical_loss": 3.798288253848972, + "tokens_seen": 666697728 + }, + { + "epoch": 0.2, + "learning_rate": 0.000403049147442327, + "loss": 1.4337, + "theoretical_loss": 3.7982500148728073, + "tokens_seen": 666763264 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": -0.0008156599360518157, + "objective/train/docs_used": 385855, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.86568021774292, + "objective/train/original_loss": 2.8656792640686035, + "objective/train/theoretical_loss": 3.79821178070722, + "objective/train/tokens_used": 687288800, + "objective/train/value_avg": -0.0047760009765625, + "objective/train/value_loss": 4.647484456654638e-05, + "objective/train/value_max": -0.0003845691680908203, + "objective/train/value_min": -0.06524658203125, + "objective/train/value_reward_corr": 0.38135231122683855, + "objective/train/value_std": 0.00424957275390625, + "objective/train/weight_avg": 0.99409419298172, + "objective/train/weighted_lm_loss": 2.8590381145477295, + "objective/train/weights_max": 1.609742522239685, + "objective/train/weights_min": 0.47739124298095703, + "theoretical_loss": 3.79821178070722, + "tokens_seen": 666828800 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004030391173520562, + "loss": 1.5063, + "theoretical_loss": 3.79821178070722, + "tokens_seen": 666828800 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040302908726178537, + "loss": 1.4133, + "theoretical_loss": 3.798173551351132, + "tokens_seen": 666894336 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040301905717151455, + "loss": 1.5195, + "theoretical_loss": 3.7981353268034654, + "tokens_seen": 666959872 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004030090270812438, + "loss": 1.4106, + "theoretical_loss": 3.7980971070631435, + "tokens_seen": 667025408 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004029989969909729, + "loss": 1.4171, + "theoretical_loss": 3.7980588921290885, + "tokens_seen": 667090944 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040298896690070215, + "loss": 1.5635, + "theoretical_loss": 3.7980206820002254, + "tokens_seen": 667156480 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004029789368104313, + "loss": 1.4005, + "theoretical_loss": 3.797982476675477, + "tokens_seen": 667222016 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004029689067201605, + "loss": 1.527, + "theoretical_loss": 3.7979442761537676, + "tokens_seen": 667287552 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004029588766298897, + "loss": 1.5303, + "theoretical_loss": 3.7979060804340223, + "tokens_seen": 667353088 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040294884653961887, + "loss": 1.4934, + "theoretical_loss": 3.797867889515165, + "tokens_seen": 667418624 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040293881644934805, + "loss": 1.3877, + "theoretical_loss": 3.797829703396122, + "tokens_seen": 667484160 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040292878635907723, + "loss": 1.5713, + "theoretical_loss": 3.7977915220758183, + "tokens_seen": 667549696 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004029187562688064, + "loss": 1.4542, + "theoretical_loss": 3.79775334555318, + "tokens_seen": 667615232 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040290872617853565, + "loss": 1.3589, + "theoretical_loss": 3.7977151738271333, + "tokens_seen": 667680768 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004028986960882648, + "loss": 1.5331, + "theoretical_loss": 3.7976770068966053, + "tokens_seen": 667746304 + }, + { + "epoch": 0.2, + "learning_rate": 0.000402888665997994, + "loss": 1.4039, + "theoretical_loss": 3.797638844760522, + "tokens_seen": 667811840 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004028786359077232, + "loss": 1.3296, + "theoretical_loss": 3.7976006874178116, + "tokens_seen": 667877376 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004028686058174524, + "loss": 1.3494, + "theoretical_loss": 3.797562534867401, + "tokens_seen": 667942912 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040285857572718156, + "loss": 1.3886, + "theoretical_loss": 3.7975243871082194, + "tokens_seen": 668008448 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040284854563691074, + "loss": 1.465, + "theoretical_loss": 3.797486244139194, + "tokens_seen": 668073984 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004028385155466399, + "loss": 1.3478, + "theoretical_loss": 3.7974481059592535, + "tokens_seen": 668139520 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040282848545636915, + "loss": 1.4126, + "theoretical_loss": 3.7974099725673276, + "tokens_seen": 668205056 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004028184553660983, + "loss": 1.3185, + "theoretical_loss": 3.7973718439623454, + "tokens_seen": 668270592 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004028084252758275, + "loss": 1.4499, + "theoretical_loss": 3.7973337201432367, + "tokens_seen": 668336128 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040279839518555664, + "loss": 1.4413, + "theoretical_loss": 3.7972956011089316, + "tokens_seen": 668401664 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": -0.0009316555806435645, + "objective/train/docs_used": 387212, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0824198722839355, + "objective/train/original_loss": 3.0824198722839355, + "objective/train/theoretical_loss": 3.797257486858361, + "objective/train/tokens_used": 688927200, + "objective/train/value_avg": -0.004840850830078125, + "objective/train/value_loss": 5.8762027038028464e-05, + "objective/train/value_max": -0.0004935264587402344, + "objective/train/value_min": -0.07763671875, + "objective/train/value_reward_corr": 0.3053916445899925, + "objective/train/value_std": 0.0034351348876953125, + "objective/train/weight_avg": 0.9933707118034363, + "objective/train/weighted_lm_loss": 3.0610787868499756, + "objective/train/weights_max": 1.6509770154953003, + "objective/train/weights_min": 0.136525958776474, + "theoretical_loss": 3.797257486858361, + "tokens_seen": 668467200 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004027883650952859, + "loss": 1.4969, + "theoretical_loss": 3.797257486858361, + "tokens_seen": 668467200 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040277833500501506, + "loss": 1.3654, + "theoretical_loss": 3.797219377390454, + "tokens_seen": 668532736 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040276830491474424, + "loss": 1.3521, + "theoretical_loss": 3.797181272704143, + "tokens_seen": 668598272 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004027582748244734, + "loss": 1.3748, + "theoretical_loss": 3.79714317279836, + "tokens_seen": 668663808 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004027482447342026, + "loss": 1.3647, + "theoretical_loss": 3.797105077672035, + "tokens_seen": 668729344 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004027382146439318, + "loss": 1.4048, + "theoretical_loss": 3.797066987324102, + "tokens_seen": 668794880 + }, + { + "epoch": 0.2, + "learning_rate": 0.000402728184553661, + "loss": 1.4029, + "theoretical_loss": 3.7970289017534924, + "tokens_seen": 668860416 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040271815446339015, + "loss": 1.4252, + "theoretical_loss": 3.796990820959139, + "tokens_seen": 668925952 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004027081243731194, + "loss": 1.3682, + "theoretical_loss": 3.796952744939976, + "tokens_seen": 668991488 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040269809428284856, + "loss": 1.4033, + "theoretical_loss": 3.7969146736949355, + "tokens_seen": 669057024 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040268806419257774, + "loss": 1.4277, + "theoretical_loss": 3.796876607222952, + "tokens_seen": 669122560 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004026780341023069, + "loss": 1.4405, + "theoretical_loss": 3.7968385455229603, + "tokens_seen": 669188096 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004026680040120361, + "loss": 1.3973, + "theoretical_loss": 3.7968004885938935, + "tokens_seen": 669253632 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004026579739217653, + "loss": 1.4332, + "theoretical_loss": 3.7967624364346877, + "tokens_seen": 669319168 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004026479438314945, + "loss": 1.4643, + "theoretical_loss": 3.796724389044278, + "tokens_seen": 669384704 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040263791374122365, + "loss": 1.4523, + "theoretical_loss": 3.796686346421599, + "tokens_seen": 669450240 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004026278836509529, + "loss": 1.2812, + "theoretical_loss": 3.7966483085655884, + "tokens_seen": 669515776 + }, + { + "epoch": 0.2, + "learning_rate": 0.000402617853560682, + "loss": 1.4258, + "theoretical_loss": 3.7966102754751803, + "tokens_seen": 669581312 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040260782347041125, + "loss": 1.3217, + "theoretical_loss": 3.796572247149313, + "tokens_seen": 669646848 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040259779338014043, + "loss": 1.4328, + "theoretical_loss": 3.7965342235869226, + "tokens_seen": 669712384 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004025877632898696, + "loss": 1.4808, + "theoretical_loss": 3.796496204786946, + "tokens_seen": 669777920 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004025777331995988, + "loss": 1.466, + "theoretical_loss": 3.796458190748322, + "tokens_seen": 669843456 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040256770310932797, + "loss": 1.3807, + "theoretical_loss": 3.7964201814699874, + "tokens_seen": 669908992 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040255767301905715, + "loss": 1.4469, + "theoretical_loss": 3.796382176950881, + "tokens_seen": 669974528 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004025476429287864, + "loss": 1.454, + "theoretical_loss": 3.7963441771899418, + "tokens_seen": 670040064 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": 0.00015963672194629908, + "objective/train/docs_used": 387914, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.554715156555176, + "objective/train/original_loss": 2.5547146797180176, + "objective/train/theoretical_loss": 3.7963061821861084, + "objective/train/tokens_used": 690565600, + "objective/train/value_avg": -0.0047760009765625, + "objective/train/value_loss": 3.433601159485988e-05, + "objective/train/value_max": -0.0003612041473388672, + "objective/train/value_min": -0.048492431640625, + "objective/train/value_reward_corr": 0.2857588210349699, + "objective/train/value_std": 0.00351715087890625, + "objective/train/weight_avg": 1.0032563209533691, + "objective/train/weighted_lm_loss": 2.562591075897217, + "objective/train/weights_max": 1.5618356466293335, + "objective/train/weights_min": 0.5127261281013489, + "theoretical_loss": 3.7963061821861084, + "tokens_seen": 670105600 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004025376128385155, + "loss": 1.4037, + "theoretical_loss": 3.7963061821861084, + "tokens_seen": 670105600 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040252758274824475, + "loss": 1.4253, + "theoretical_loss": 3.79626819193832, + "tokens_seen": 670171136 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040251755265797393, + "loss": 1.3655, + "theoretical_loss": 3.7962302064455162, + "tokens_seen": 670236672 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004025075225677031, + "loss": 1.4614, + "theoretical_loss": 3.796192225706637, + "tokens_seen": 670302208 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004024974924774323, + "loss": 1.4304, + "theoretical_loss": 3.796154249720623, + "tokens_seen": 670367744 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004024874623871615, + "loss": 1.4551, + "theoretical_loss": 3.796116278486415, + "tokens_seen": 670433280 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040247743229689066, + "loss": 1.3223, + "theoretical_loss": 3.7960783120029533, + "tokens_seen": 670498816 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004024674022066199, + "loss": 1.4594, + "theoretical_loss": 3.79604035026918, + "tokens_seen": 670564352 + }, + { + "epoch": 0.2, + "learning_rate": 0.000402457372116349, + "loss": 1.3553, + "theoretical_loss": 3.7960023932840365, + "tokens_seen": 670629888 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040244734202607825, + "loss": 1.4279, + "theoretical_loss": 3.795964441046465, + "tokens_seen": 670695424 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004024373119358074, + "loss": 1.5132, + "theoretical_loss": 3.795926493555407, + "tokens_seen": 670760960 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004024272818455366, + "loss": 1.3434, + "theoretical_loss": 3.7958885508098064, + "tokens_seen": 670826496 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004024172517552658, + "loss": 1.3444, + "theoretical_loss": 3.795850612808606, + "tokens_seen": 670892032 + }, + { + "epoch": 0.2, + "learning_rate": 0.000402407221664995, + "loss": 1.43, + "theoretical_loss": 3.7958126795507487, + "tokens_seen": 670957568 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004023971915747242, + "loss": 1.4195, + "theoretical_loss": 3.7957747510351787, + "tokens_seen": 671023104 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004023871614844534, + "loss": 1.3821, + "theoretical_loss": 3.795736827260839, + "tokens_seen": 671088640 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004023771313941826, + "loss": 1.447, + "theoretical_loss": 3.795698908226676, + "tokens_seen": 671154176 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040236710130391176, + "loss": 1.4426, + "theoretical_loss": 3.795660993931633, + "tokens_seen": 671219712 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040235707121364094, + "loss": 1.5957, + "theoretical_loss": 3.795623084374655, + "tokens_seen": 671285248 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004023470411233701, + "loss": 1.3821, + "theoretical_loss": 3.7955851795546875, + "tokens_seen": 671350784 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040233701103309935, + "loss": 1.4114, + "theoretical_loss": 3.795547279470677, + "tokens_seen": 671416320 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004023269809428285, + "loss": 1.5198, + "theoretical_loss": 3.7955093841215692, + "tokens_seen": 671481856 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004023169508525577, + "loss": 1.4237, + "theoretical_loss": 3.79547149350631, + "tokens_seen": 671547392 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040230692076228684, + "loss": 1.5098, + "theoretical_loss": 3.7954336076238473, + "tokens_seen": 671612928 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004022968906720161, + "loss": 1.475, + "theoretical_loss": 3.795395726473127, + "tokens_seen": 671678464 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": -0.0018293379107490182, + "objective/train/docs_used": 389368, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8948922157287598, + "objective/train/original_loss": 2.894892930984497, + "objective/train/theoretical_loss": 3.795357850053097, + "objective/train/tokens_used": 692204000, + "objective/train/value_avg": -0.005279541015625, + "objective/train/value_loss": 7.62626004870981e-05, + "objective/train/value_max": -0.0002758502960205078, + "objective/train/value_min": -0.059326171875, + "objective/train/value_reward_corr": 0.40995138127631986, + "objective/train/value_std": 0.00473785400390625, + "objective/train/weight_avg": 0.9853152632713318, + "objective/train/weighted_lm_loss": 2.857041358947754, + "objective/train/weights_max": 1.4756507873535156, + "objective/train/weights_min": 0.4919312298297882, + "theoretical_loss": 3.795357850053097, + "tokens_seen": 671744000 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040228686058174526, + "loss": 1.4697, + "theoretical_loss": 3.795357850053097, + "tokens_seen": 671744000 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040227683049147444, + "loss": 1.4836, + "theoretical_loss": 3.795319978362705, + "tokens_seen": 671809536 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004022668004012036, + "loss": 1.4211, + "theoretical_loss": 3.7952821114008994, + "tokens_seen": 671875072 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004022567703109328, + "loss": 1.4512, + "theoretical_loss": 3.795244249166629, + "tokens_seen": 671940608 + }, + { + "epoch": 0.2, + "learning_rate": 0.000402246740220662, + "loss": 1.3599, + "theoretical_loss": 3.7952063916588417, + "tokens_seen": 672006144 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004022367101303912, + "loss": 1.4162, + "theoretical_loss": 3.7951685388764864, + "tokens_seen": 672071680 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040222668004012035, + "loss": 1.3622, + "theoretical_loss": 3.795130690818514, + "tokens_seen": 672137216 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004022166499498496, + "loss": 1.3906, + "theoretical_loss": 3.7950928474838728, + "tokens_seen": 672202752 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040220661985957876, + "loss": 1.6345, + "theoretical_loss": 3.795055008871514, + "tokens_seen": 672268288 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040219658976930794, + "loss": 1.5371, + "theoretical_loss": 3.7950171749803876, + "tokens_seen": 672333824 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004021865596790371, + "loss": 1.5103, + "theoretical_loss": 3.794979345809444, + "tokens_seen": 672399360 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004021765295887663, + "loss": 1.4102, + "theoretical_loss": 3.794941521357635, + "tokens_seen": 672464896 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004021664994984955, + "loss": 1.4421, + "theoretical_loss": 3.794903701623912, + "tokens_seen": 672530432 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004021564694082247, + "loss": 1.4598, + "theoretical_loss": 3.7948658866072265, + "tokens_seen": 672595968 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040214643931795385, + "loss": 1.419, + "theoretical_loss": 3.7948280763065307, + "tokens_seen": 672661504 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004021364092276831, + "loss": 1.5044, + "theoretical_loss": 3.7947902707207777, + "tokens_seen": 672727040 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004021263791374122, + "loss": 1.4456, + "theoretical_loss": 3.7947524698489197, + "tokens_seen": 672792576 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040211634904714145, + "loss": 1.4491, + "theoretical_loss": 3.794714673689909, + "tokens_seen": 672858112 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040210631895687063, + "loss": 1.4583, + "theoretical_loss": 3.794676882242701, + "tokens_seen": 672923648 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004020962888665998, + "loss": 1.4557, + "theoretical_loss": 3.7946390955062483, + "tokens_seen": 672989184 + }, + { + "epoch": 0.2, + "learning_rate": 0.000402086258776329, + "loss": 1.5086, + "theoretical_loss": 3.7946013134795047, + "tokens_seen": 673054720 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040207622868605817, + "loss": 1.3672, + "theoretical_loss": 3.7945635361614256, + "tokens_seen": 673120256 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040206619859578735, + "loss": 1.4442, + "theoretical_loss": 3.7945257635509657, + "tokens_seen": 673185792 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004020561685055166, + "loss": 1.511, + "theoretical_loss": 3.7944879956470796, + "tokens_seen": 673251328 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004020461384152457, + "loss": 1.4998, + "theoretical_loss": 3.7944502324487233, + "tokens_seen": 673316864 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": -0.00207929196767509, + "objective/train/docs_used": 390091, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.508676528930664, + "objective/train/original_loss": 2.508676528930664, + "objective/train/theoretical_loss": 3.7944124739548526, + "objective/train/tokens_used": 693842400, + "objective/train/value_avg": -0.005290985107421875, + "objective/train/value_loss": 7.492768781958148e-05, + "objective/train/value_max": -0.0002397298812866211, + "objective/train/value_min": -0.06915283203125, + "objective/train/value_reward_corr": 0.5440714643260203, + "objective/train/value_std": 0.0055389404296875, + "objective/train/weight_avg": 0.9827744364738464, + "objective/train/weighted_lm_loss": 2.46648907661438, + "objective/train/weights_max": 1.606627106666565, + "objective/train/weights_min": 0.44229164719581604, + "theoretical_loss": 3.7944124739548526, + "tokens_seen": 673382400 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040203610832497495, + "loss": 1.3553, + "theoretical_loss": 3.7944124739548526, + "tokens_seen": 673382400 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040202607823470413, + "loss": 1.5885, + "theoretical_loss": 3.7943747201644227, + "tokens_seen": 673447936 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004020160481444333, + "loss": 1.3685, + "theoretical_loss": 3.7943369710763912, + "tokens_seen": 673513472 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004020060180541625, + "loss": 1.4157, + "theoretical_loss": 3.7942992266897146, + "tokens_seen": 673579008 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004019959879638917, + "loss": 1.3336, + "theoretical_loss": 3.79426148700335, + "tokens_seen": 673644544 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040198595787362086, + "loss": 1.3713, + "theoretical_loss": 3.794223752016255, + "tokens_seen": 673710080 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004019759277833501, + "loss": 1.4312, + "theoretical_loss": 3.7941860217273877, + "tokens_seen": 673775616 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004019658976930792, + "loss": 1.4569, + "theoretical_loss": 3.794148296135705, + "tokens_seen": 673841152 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040195586760280845, + "loss": 1.4455, + "theoretical_loss": 3.794110575240167, + "tokens_seen": 673906688 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004019458375125376, + "loss": 1.4393, + "theoretical_loss": 3.7940728590397312, + "tokens_seen": 673972224 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004019358074222668, + "loss": 1.3644, + "theoretical_loss": 3.7940351475333576, + "tokens_seen": 674037760 + }, + { + "epoch": 0.2, + "learning_rate": 0.000401925777331996, + "loss": 1.3943, + "theoretical_loss": 3.7939974407200054, + "tokens_seen": 674103296 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004019157472417252, + "loss": 1.4235, + "theoretical_loss": 3.793959738598634, + "tokens_seen": 674168832 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040190571715145436, + "loss": 1.3983, + "theoretical_loss": 3.793922041168204, + "tokens_seen": 674234368 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004018956870611836, + "loss": 1.5166, + "theoretical_loss": 3.793884348427676, + "tokens_seen": 674299904 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004018856569709127, + "loss": 1.3889, + "theoretical_loss": 3.79384666037601, + "tokens_seen": 674365440 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040187562688064196, + "loss": 1.3711, + "theoretical_loss": 3.7938089770121683, + "tokens_seen": 674430976 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004018655967903711, + "loss": 1.4679, + "theoretical_loss": 3.7937712983351117, + "tokens_seen": 674496512 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004018555667001003, + "loss": 1.4759, + "theoretical_loss": 3.793733624343802, + "tokens_seen": 674562048 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004018455366098295, + "loss": 1.2889, + "theoretical_loss": 3.793695955037202, + "tokens_seen": 674627584 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004018355065195587, + "loss": 1.4176, + "theoretical_loss": 3.793658290414273, + "tokens_seen": 674693120 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040182547642928786, + "loss": 1.4787, + "theoretical_loss": 3.7936206304739777, + "tokens_seen": 674758656 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040181544633901704, + "loss": 1.4123, + "theoretical_loss": 3.7935829752152808, + "tokens_seen": 674824192 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004018054162487462, + "loss": 1.5298, + "theoretical_loss": 3.7935453246371442, + "tokens_seen": 674889728 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040179538615847546, + "loss": 1.3857, + "theoretical_loss": 3.7935076787385325, + "tokens_seen": 674955264 + }, + { + "epoch": 0.2, + "objective/train/advantage_avg": -0.0013673411449417472, + "objective/train/docs_used": 391518, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8769445419311523, + "objective/train/original_loss": 2.8769447803497314, + "objective/train/theoretical_loss": 3.79347003751841, + "objective/train/tokens_used": 695480800, + "objective/train/value_avg": -0.006633758544921875, + "objective/train/value_loss": 0.00010743663005996495, + "objective/train/value_max": -0.00036263465881347656, + "objective/train/value_min": -0.09588623046875, + "objective/train/value_reward_corr": 0.4394265127061108, + "objective/train/value_std": 0.005889892578125, + "objective/train/weight_avg": 0.9909010529518127, + "objective/train/weighted_lm_loss": 2.845916509628296, + "objective/train/weights_max": 2.514920473098755, + "objective/train/weights_min": 0.08854537457227707, + "theoretical_loss": 3.79347003751841, + "tokens_seen": 675020800 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004017853560682046, + "loss": 1.3772, + "theoretical_loss": 3.79347003751841, + "tokens_seen": 675020800 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004017753259779338, + "loss": 1.3709, + "theoretical_loss": 3.79343240097574, + "tokens_seen": 675086336 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040176529588766295, + "loss": 1.3353, + "theoretical_loss": 3.7933947691094883, + "tokens_seen": 675151872 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004017552657973922, + "loss": 1.4716, + "theoretical_loss": 3.79335714191862, + "tokens_seen": 675217408 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040174523570712137, + "loss": 1.4318, + "theoretical_loss": 3.7933195194020994, + "tokens_seen": 675282944 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040173520561685055, + "loss": 1.3271, + "theoretical_loss": 3.793281901558894, + "tokens_seen": 675348480 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040172517552657973, + "loss": 1.3451, + "theoretical_loss": 3.7932442883879687, + "tokens_seen": 675414016 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040171514543630896, + "loss": 1.418, + "theoretical_loss": 3.79320667988829, + "tokens_seen": 675479552 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004017051153460381, + "loss": 1.5083, + "theoretical_loss": 3.793169076058825, + "tokens_seen": 675545088 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004016950852557673, + "loss": 1.42, + "theoretical_loss": 3.79313147689854, + "tokens_seen": 675610624 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040168505516549645, + "loss": 1.5278, + "theoretical_loss": 3.7930938824064038, + "tokens_seen": 675676160 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004016750250752257, + "loss": 1.3532, + "theoretical_loss": 3.793056292581383, + "tokens_seen": 675741696 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040166499498495487, + "loss": 1.4155, + "theoretical_loss": 3.793018707422446, + "tokens_seen": 675807232 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040165496489468405, + "loss": 1.4822, + "theoretical_loss": 3.792981126928561, + "tokens_seen": 675872768 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004016449348044133, + "loss": 1.3706, + "theoretical_loss": 3.792943551098697, + "tokens_seen": 675938304 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004016349047141424, + "loss": 1.391, + "theoretical_loss": 3.792905979931823, + "tokens_seen": 676003840 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040162487462387165, + "loss": 1.46, + "theoretical_loss": 3.792868413426908, + "tokens_seen": 676069376 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040161484453360083, + "loss": 1.3037, + "theoretical_loss": 3.7928308515829223, + "tokens_seen": 676134912 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040160481444333, + "loss": 1.4825, + "theoretical_loss": 3.7927932943988356, + "tokens_seen": 676200448 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004015947843530592, + "loss": 1.5223, + "theoretical_loss": 3.792755741873618, + "tokens_seen": 676265984 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040158475426278837, + "loss": 1.3888, + "theoretical_loss": 3.7927181940062407, + "tokens_seen": 676331520 + }, + { + "epoch": 0.2, + "learning_rate": 0.00040157472417251755, + "loss": 1.4069, + "theoretical_loss": 3.792680650795674, + "tokens_seen": 676397056 + }, + { + "epoch": 0.2, + "learning_rate": 0.0004015646940822468, + "loss": 1.4381, + "theoretical_loss": 3.7926431122408903, + "tokens_seen": 676462592 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004015546639919759, + "loss": 1.3516, + "theoretical_loss": 3.79260557834086, + "tokens_seen": 676528128 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040154463390170515, + "loss": 1.3252, + "theoretical_loss": 3.792568049094556, + "tokens_seen": 676593664 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0009807548485696316, + "objective/train/docs_used": 392138, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.082443952560425, + "objective/train/original_loss": 3.082444190979004, + "objective/train/theoretical_loss": 3.7925305245009504, + "objective/train/tokens_used": 697119200, + "objective/train/value_avg": -0.0060577392578125, + "objective/train/value_loss": 7.427798846038058e-05, + "objective/train/value_max": -0.00036835670471191406, + "objective/train/value_min": -0.10284423828125, + "objective/train/value_reward_corr": 0.22153630589432183, + "objective/train/value_std": 0.004512786865234375, + "objective/train/weight_avg": 0.993705689907074, + "objective/train/weighted_lm_loss": 3.0736422538757324, + "objective/train/weights_max": 2.421736001968384, + "objective/train/weights_min": 0.2689950168132782, + "theoretical_loss": 3.7925305245009504, + "tokens_seen": 676659200 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040153460381143433, + "loss": 1.5127, + "theoretical_loss": 3.7925305245009504, + "tokens_seen": 676659200 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004015245737211635, + "loss": 1.4373, + "theoretical_loss": 3.7924930045590157, + "tokens_seen": 676724736 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004015145436308927, + "loss": 1.4625, + "theoretical_loss": 3.7924554892677254, + "tokens_seen": 676790272 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004015045135406219, + "loss": 1.4164, + "theoretical_loss": 3.7924179786260517, + "tokens_seen": 676855808 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040149448345035106, + "loss": 1.428, + "theoretical_loss": 3.792380472632969, + "tokens_seen": 676921344 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004014844533600803, + "loss": 1.4125, + "theoretical_loss": 3.792342971287451, + "tokens_seen": 676986880 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004014744232698094, + "loss": 1.2464, + "theoretical_loss": 3.7923054745884723, + "tokens_seen": 677052416 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040146439317953865, + "loss": 1.3952, + "theoretical_loss": 3.7922679825350074, + "tokens_seen": 677117952 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004014543630892678, + "loss": 1.5525, + "theoretical_loss": 3.792230495126031, + "tokens_seen": 677183488 + }, + { + "epoch": 0.21, + "learning_rate": 0.000401444332998997, + "loss": 1.4692, + "theoretical_loss": 3.792193012360518, + "tokens_seen": 677249024 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004014343029087262, + "loss": 1.4071, + "theoretical_loss": 3.7921555342374447, + "tokens_seen": 677314560 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004014242728184554, + "loss": 1.4434, + "theoretical_loss": 3.792118060755787, + "tokens_seen": 677380096 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040141424272818456, + "loss": 1.4957, + "theoretical_loss": 3.7920805919145204, + "tokens_seen": 677445632 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004014042126379138, + "loss": 1.4823, + "theoretical_loss": 3.792043127712622, + "tokens_seen": 677511168 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004013941825476429, + "loss": 1.4134, + "theoretical_loss": 3.792005668149068, + "tokens_seen": 677576704 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040138415245737216, + "loss": 1.4987, + "theoretical_loss": 3.791968213222837, + "tokens_seen": 677642240 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004013741223671013, + "loss": 1.5482, + "theoretical_loss": 3.791930762932905, + "tokens_seen": 677707776 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004013640922768305, + "loss": 1.4795, + "theoretical_loss": 3.791893317278251, + "tokens_seen": 677773312 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004013540621865597, + "loss": 1.4018, + "theoretical_loss": 3.7918558762578525, + "tokens_seen": 677838848 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004013440320962889, + "loss": 1.5612, + "theoretical_loss": 3.791818439870688, + "tokens_seen": 677904384 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040133400200601806, + "loss": 1.4198, + "theoretical_loss": 3.791781008115737, + "tokens_seen": 677969920 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040132397191574724, + "loss": 1.3894, + "theoretical_loss": 3.7917435809919775, + "tokens_seen": 678035456 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004013139418254764, + "loss": 1.4078, + "theoretical_loss": 3.79170615849839, + "tokens_seen": 678100992 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040130391173520566, + "loss": 1.4163, + "theoretical_loss": 3.7916687406339538, + "tokens_seen": 678166528 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004012938816449348, + "loss": 1.3554, + "theoretical_loss": 3.7916313273976483, + "tokens_seen": 678232064 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.001137252664193511, + "objective/train/docs_used": 393619, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.31144642829895, + "objective/train/original_loss": 3.3114466667175293, + "objective/train/theoretical_loss": 3.7915939187884558, + "objective/train/tokens_used": 698757600, + "objective/train/value_avg": -0.0057525634765625, + "objective/train/value_loss": 0.00013182320981286466, + "objective/train/value_max": -0.000396728515625, + "objective/train/value_min": -0.091064453125, + "objective/train/value_reward_corr": 0.6010935199649243, + "objective/train/value_std": 0.0066070556640625, + "objective/train/weight_avg": 0.9943047165870667, + "objective/train/weighted_lm_loss": 3.3013014793395996, + "objective/train/weights_max": 1.793517827987671, + "objective/train/weights_min": 0.24960675835609436, + "theoretical_loss": 3.7915939187884558, + "tokens_seen": 678297600 + }, + { + "epoch": 0.21, + "learning_rate": 0.000401283851554664, + "loss": 1.426, + "theoretical_loss": 3.7915939187884558, + "tokens_seen": 678297600 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040127382146439315, + "loss": 1.4424, + "theoretical_loss": 3.791556514805355, + "tokens_seen": 678363136 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004012637913741224, + "loss": 1.4755, + "theoretical_loss": 3.7915191154473287, + "tokens_seen": 678428672 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040125376128385157, + "loss": 1.4377, + "theoretical_loss": 3.7914817207133575, + "tokens_seen": 678494208 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040124373119358075, + "loss": 1.3459, + "theoretical_loss": 3.791444330602423, + "tokens_seen": 678559744 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040123370110330993, + "loss": 1.3345, + "theoretical_loss": 3.7914069451135073, + "tokens_seen": 678625280 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040122367101303916, + "loss": 1.4476, + "theoretical_loss": 3.791369564245593, + "tokens_seen": 678690816 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004012136409227683, + "loss": 1.5266, + "theoretical_loss": 3.7913321879976625, + "tokens_seen": 678756352 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004012036108324975, + "loss": 1.3599, + "theoretical_loss": 3.7912948163686995, + "tokens_seen": 678821888 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040119358074222665, + "loss": 1.3503, + "theoretical_loss": 3.791257449357686, + "tokens_seen": 678887424 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004011835506519559, + "loss": 1.4196, + "theoretical_loss": 3.791220086963608, + "tokens_seen": 678952960 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040117352056168507, + "loss": 1.4329, + "theoretical_loss": 3.7911827291854467, + "tokens_seen": 679018496 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040116349047141425, + "loss": 1.3905, + "theoretical_loss": 3.791145376022188, + "tokens_seen": 679084032 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040115346038114343, + "loss": 1.4156, + "theoretical_loss": 3.7911080274728164, + "tokens_seen": 679149568 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004011434302908726, + "loss": 1.2899, + "theoretical_loss": 3.791070683536316, + "tokens_seen": 679215104 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004011334002006018, + "loss": 1.3943, + "theoretical_loss": 3.791033344211673, + "tokens_seen": 679280640 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040112337011033103, + "loss": 1.4427, + "theoretical_loss": 3.7909960094978734, + "tokens_seen": 679346176 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040111334002006016, + "loss": 1.2896, + "theoretical_loss": 3.790958679393902, + "tokens_seen": 679411712 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004011033099297894, + "loss": 1.4495, + "theoretical_loss": 3.790921353898745, + "tokens_seen": 679477248 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004010932798395185, + "loss": 1.3654, + "theoretical_loss": 3.79088403301139, + "tokens_seen": 679542784 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040108324974924775, + "loss": 1.5155, + "theoretical_loss": 3.790846716730823, + "tokens_seen": 679608320 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040107321965897693, + "loss": 1.4103, + "theoretical_loss": 3.7908094050560317, + "tokens_seen": 679673856 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004010631895687061, + "loss": 1.4833, + "theoretical_loss": 3.790772097986003, + "tokens_seen": 679739392 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004010531594784353, + "loss": 1.3879, + "theoretical_loss": 3.7907347955197253, + "tokens_seen": 679804928 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040104312938816453, + "loss": 1.386, + "theoretical_loss": 3.7906974976561862, + "tokens_seen": 679870464 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": 2.200900053139776e-05, + "objective/train/docs_used": 393966, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0114197731018066, + "objective/train/original_loss": 3.0114197731018066, + "objective/train/theoretical_loss": 3.790660204394375, + "objective/train/tokens_used": 700396000, + "objective/train/value_avg": -0.006107330322265625, + "objective/train/value_loss": 8.331832941621542e-05, + "objective/train/value_max": -0.0002982616424560547, + "objective/train/value_min": -0.053497314453125, + "objective/train/value_reward_corr": 0.3020037152482923, + "objective/train/value_std": 0.0049896240234375, + "objective/train/weight_avg": 1.0037702322006226, + "objective/train/weighted_lm_loss": 3.0152485370635986, + "objective/train/weights_max": 1.5926400423049927, + "objective/train/weights_min": 0.08595611155033112, + "theoretical_loss": 3.790660204394375, + "tokens_seen": 679936000 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040103309929789366, + "loss": 1.4532, + "theoretical_loss": 3.790660204394375, + "tokens_seen": 679936000 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004010230692076229, + "loss": 1.423, + "theoretical_loss": 3.79062291573328, + "tokens_seen": 680001536 + }, + { + "epoch": 0.21, + "learning_rate": 0.000401013039117352, + "loss": 1.4417, + "theoretical_loss": 3.79058563167189, + "tokens_seen": 680067072 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040100300902708126, + "loss": 1.4213, + "theoretical_loss": 3.7905483522091954, + "tokens_seen": 680132608 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040099297893681044, + "loss": 1.4494, + "theoretical_loss": 3.7905110773441844, + "tokens_seen": 680198144 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004009829488465396, + "loss": 1.4039, + "theoretical_loss": 3.7904738070758484, + "tokens_seen": 680263680 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004009729187562688, + "loss": 1.4184, + "theoretical_loss": 3.790436541403177, + "tokens_seen": 680329216 + }, + { + "epoch": 0.21, + "learning_rate": 0.000400962888665998, + "loss": 1.4379, + "theoretical_loss": 3.790399280325162, + "tokens_seen": 680394752 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040095285857572716, + "loss": 1.3992, + "theoretical_loss": 3.7903620238407933, + "tokens_seen": 680460288 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004009428284854564, + "loss": 1.3744, + "theoretical_loss": 3.790324771949063, + "tokens_seen": 680525824 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004009327983951855, + "loss": 1.3869, + "theoretical_loss": 3.790287524648962, + "tokens_seen": 680591360 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040092276830491476, + "loss": 1.4198, + "theoretical_loss": 3.7902502819394828, + "tokens_seen": 680656896 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004009127382146439, + "loss": 1.3914, + "theoretical_loss": 3.7902130438196173, + "tokens_seen": 680722432 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004009027081243731, + "loss": 1.4118, + "theoretical_loss": 3.7901758102883587, + "tokens_seen": 680787968 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040089267803410236, + "loss": 1.4291, + "theoretical_loss": 3.7901385813446997, + "tokens_seen": 680853504 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004008826479438315, + "loss": 1.2854, + "theoretical_loss": 3.790101356987633, + "tokens_seen": 680919040 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004008726178535607, + "loss": 1.3517, + "theoretical_loss": 3.790064137216153, + "tokens_seen": 680984576 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004008625877632899, + "loss": 1.4075, + "theoretical_loss": 3.7900269220292535, + "tokens_seen": 681050112 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004008525576730191, + "loss": 1.4145, + "theoretical_loss": 3.789989711425928, + "tokens_seen": 681115648 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040084252758274826, + "loss": 1.4788, + "theoretical_loss": 3.7899525054051724, + "tokens_seen": 681181184 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040083249749247744, + "loss": 1.3765, + "theoretical_loss": 3.7899153039659796, + "tokens_seen": 681246720 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004008224674022066, + "loss": 1.4039, + "theoretical_loss": 3.789878107107346, + "tokens_seen": 681312256 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040081243731193586, + "loss": 1.5512, + "theoretical_loss": 3.7898409148282672, + "tokens_seen": 681377792 + }, + { + "epoch": 0.21, + "learning_rate": 0.000400802407221665, + "loss": 1.5046, + "theoretical_loss": 3.7898037271277385, + "tokens_seen": 681443328 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004007923771313942, + "loss": 1.3767, + "theoretical_loss": 3.789766544004756, + "tokens_seen": 681508864 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0005630032392218709, + "objective/train/docs_used": 395160, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.831752061843872, + "objective/train/original_loss": 2.831751585006714, + "objective/train/theoretical_loss": 3.7897293654583164, + "objective/train/tokens_used": 702034400, + "objective/train/value_avg": -0.005855560302734375, + "objective/train/value_loss": 7.794718840159476e-05, + "objective/train/value_max": -0.0004076957702636719, + "objective/train/value_min": -0.058563232421875, + "objective/train/value_reward_corr": 0.3942439707430735, + "objective/train/value_std": 0.00443267822265625, + "objective/train/weight_avg": 0.9979328513145447, + "objective/train/weighted_lm_loss": 2.8136613368988037, + "objective/train/weights_max": 1.5303258895874023, + "objective/train/weights_min": 0.2568511962890625, + "theoretical_loss": 3.7897293654583164, + "tokens_seen": 681574400 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040078234704112335, + "loss": 1.3622, + "theoretical_loss": 3.7897293654583164, + "tokens_seen": 681574400 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004007723169508526, + "loss": 1.3679, + "theoretical_loss": 3.7896921914874167, + "tokens_seen": 681639936 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040076228686058177, + "loss": 1.4294, + "theoretical_loss": 3.789655022091053, + "tokens_seen": 681705472 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040075225677031095, + "loss": 1.4584, + "theoretical_loss": 3.7896178572682238, + "tokens_seen": 681771008 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040074222668004013, + "loss": 1.4327, + "theoretical_loss": 3.789580697017926, + "tokens_seen": 681836544 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040073219658976936, + "loss": 1.3609, + "theoretical_loss": 3.7895435413391576, + "tokens_seen": 681902080 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004007221664994985, + "loss": 1.4901, + "theoretical_loss": 3.7895063902309176, + "tokens_seen": 681967616 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004007121364092277, + "loss": 1.4302, + "theoretical_loss": 3.789469243692204, + "tokens_seen": 682033152 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040070210631895685, + "loss": 1.3235, + "theoretical_loss": 3.7894321017220163, + "tokens_seen": 682098688 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004006920762286861, + "loss": 1.3608, + "theoretical_loss": 3.789394964319353, + "tokens_seen": 682164224 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040068204613841527, + "loss": 1.476, + "theoretical_loss": 3.7893578314832146, + "tokens_seen": 682229760 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040067201604814445, + "loss": 1.4632, + "theoretical_loss": 3.7893207032126, + "tokens_seen": 682295296 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040066198595787363, + "loss": 1.4054, + "theoretical_loss": 3.7892835795065105, + "tokens_seen": 682360832 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004006519558676028, + "loss": 1.4652, + "theoretical_loss": 3.7892464603639455, + "tokens_seen": 682426368 + }, + { + "epoch": 0.21, + "learning_rate": 0.000400641925777332, + "loss": 1.4481, + "theoretical_loss": 3.789209345783907, + "tokens_seen": 682491904 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040063189568706123, + "loss": 1.4301, + "theoretical_loss": 3.7891722357653954, + "tokens_seen": 682557440 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040062186559679036, + "loss": 1.4003, + "theoretical_loss": 3.7891351303074123, + "tokens_seen": 682622976 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004006118355065196, + "loss": 1.3686, + "theoretical_loss": 3.78909802940896, + "tokens_seen": 682688512 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004006018054162487, + "loss": 1.4913, + "theoretical_loss": 3.789060933069039, + "tokens_seen": 682754048 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040059177532597795, + "loss": 1.5835, + "theoretical_loss": 3.789023841286654, + "tokens_seen": 682819584 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040058174523570713, + "loss": 1.4181, + "theoretical_loss": 3.788986754060806, + "tokens_seen": 682885120 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004005717151454363, + "loss": 1.3422, + "theoretical_loss": 3.7889496713904993, + "tokens_seen": 682950656 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004005616850551655, + "loss": 1.3929, + "theoretical_loss": 3.788912593274736, + "tokens_seen": 683016192 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040055165496489473, + "loss": 1.4039, + "theoretical_loss": 3.7888755197125206, + "tokens_seen": 683081728 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040054162487462386, + "loss": 1.4777, + "theoretical_loss": 3.7888384507028565, + "tokens_seen": 683147264 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0001535853516543284, + "objective/train/docs_used": 395993, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.732649326324463, + "objective/train/original_loss": 2.732649087905884, + "objective/train/theoretical_loss": 3.788801386244749, + "objective/train/tokens_used": 703672800, + "objective/train/value_avg": -0.004093170166015625, + "objective/train/value_loss": 3.648161145974882e-05, + "objective/train/value_max": -0.0004322528839111328, + "objective/train/value_min": -0.06634521484375, + "objective/train/value_reward_corr": 0.2239522294334949, + "objective/train/value_std": 0.002902984619140625, + "objective/train/weight_avg": 1.0001040697097778, + "objective/train/weighted_lm_loss": 2.736762046813965, + "objective/train/weights_max": 1.8157305717468262, + "objective/train/weights_min": 0.13696478307247162, + "theoretical_loss": 3.788801386244749, + "tokens_seen": 683212800 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004005315947843531, + "loss": 1.3075, + "theoretical_loss": 3.788801386244749, + "tokens_seen": 683212800 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004005215646940822, + "loss": 1.3871, + "theoretical_loss": 3.7887643263372017, + "tokens_seen": 683278336 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040051153460381146, + "loss": 1.6004, + "theoretical_loss": 3.78872727097922, + "tokens_seen": 683343872 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040050150451354064, + "loss": 1.3811, + "theoretical_loss": 3.788690220169809, + "tokens_seen": 683409408 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004004914744232698, + "loss": 1.4433, + "theoretical_loss": 3.7886531739079743, + "tokens_seen": 683474944 + }, + { + "epoch": 0.21, + "learning_rate": 0.000400481444332999, + "loss": 1.4427, + "theoretical_loss": 3.788616132192722, + "tokens_seen": 683540480 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004004714142427282, + "loss": 1.5393, + "theoretical_loss": 3.7885790950230582, + "tokens_seen": 683606016 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040046138415245736, + "loss": 1.506, + "theoretical_loss": 3.7885420623979886, + "tokens_seen": 683671552 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004004513540621866, + "loss": 1.3863, + "theoretical_loss": 3.788505034316521, + "tokens_seen": 683737088 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004004413239719157, + "loss": 1.341, + "theoretical_loss": 3.7884680107776623, + "tokens_seen": 683802624 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040043129388164496, + "loss": 1.5273, + "theoretical_loss": 3.788430991780419, + "tokens_seen": 683868160 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004004212637913741, + "loss": 1.3901, + "theoretical_loss": 3.7883939773238002, + "tokens_seen": 683933696 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004004112337011033, + "loss": 1.431, + "theoretical_loss": 3.7883569674068136, + "tokens_seen": 683999232 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004004012036108325, + "loss": 1.3907, + "theoretical_loss": 3.788319962028467, + "tokens_seen": 684064768 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004003911735205617, + "loss": 1.4015, + "theoretical_loss": 3.788282961187769, + "tokens_seen": 684130304 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040038114343029087, + "loss": 1.4309, + "theoretical_loss": 3.78824596488373, + "tokens_seen": 684195840 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004003711133400201, + "loss": 1.4443, + "theoretical_loss": 3.7882089731153576, + "tokens_seen": 684261376 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040036108324974923, + "loss": 1.3559, + "theoretical_loss": 3.788171985881662, + "tokens_seen": 684326912 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040035105315947846, + "loss": 1.4792, + "theoretical_loss": 3.7881350031816536, + "tokens_seen": 684392448 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004003410230692076, + "loss": 1.3663, + "theoretical_loss": 3.788098025014342, + "tokens_seen": 684457984 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004003309929789368, + "loss": 1.3738, + "theoretical_loss": 3.788061051378738, + "tokens_seen": 684523520 + }, + { + "epoch": 0.21, + "learning_rate": 0.000400320962888666, + "loss": 1.3486, + "theoretical_loss": 3.7880240822738527, + "tokens_seen": 684589056 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004003109327983952, + "loss": 1.3065, + "theoretical_loss": 3.7879871176986963, + "tokens_seen": 684654592 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040030090270812437, + "loss": 1.4929, + "theoretical_loss": 3.787950157652282, + "tokens_seen": 684720128 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040029087261785355, + "loss": 1.4267, + "theoretical_loss": 3.7879132021336197, + "tokens_seen": 684785664 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0004173258494120091, + "objective/train/docs_used": 397274, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7942774295806885, + "objective/train/original_loss": 2.7942776679992676, + "objective/train/theoretical_loss": 3.7878762511417223, + "objective/train/tokens_used": 705311200, + "objective/train/value_avg": -0.005207061767578125, + "objective/train/value_loss": 7.875426672399044e-05, + "objective/train/value_max": -0.0001838207244873047, + "objective/train/value_min": -0.05145263671875, + "objective/train/value_reward_corr": 0.3541456388705349, + "objective/train/value_std": 0.00437164306640625, + "objective/train/weight_avg": 0.999458909034729, + "objective/train/weighted_lm_loss": 2.7882285118103027, + "objective/train/weights_max": 1.5943233966827393, + "objective/train/weights_min": 0.339717835187912, + "theoretical_loss": 3.7878762511417223, + "tokens_seen": 684851200 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040028084252758273, + "loss": 1.3843, + "theoretical_loss": 3.7878762511417223, + "tokens_seen": 684851200 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040027081243731197, + "loss": 1.4301, + "theoretical_loss": 3.7878393046756025, + "tokens_seen": 684916736 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004002607823470411, + "loss": 1.4857, + "theoretical_loss": 3.787802362734273, + "tokens_seen": 684982272 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040025075225677033, + "loss": 1.3956, + "theoretical_loss": 3.787765425316746, + "tokens_seen": 685047808 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040024072216649946, + "loss": 1.4118, + "theoretical_loss": 3.787728492422036, + "tokens_seen": 685113344 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004002306920762287, + "loss": 1.3866, + "theoretical_loss": 3.7876915640491555, + "tokens_seen": 685178880 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040022066198595787, + "loss": 1.4736, + "theoretical_loss": 3.7876546401971196, + "tokens_seen": 685244416 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040021063189568705, + "loss": 1.4879, + "theoretical_loss": 3.787617720864942, + "tokens_seen": 685309952 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040020060180541623, + "loss": 1.3804, + "theoretical_loss": 3.7875808060516363, + "tokens_seen": 685375488 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040019057171514547, + "loss": 1.4769, + "theoretical_loss": 3.787543895756219, + "tokens_seen": 685441024 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004001805416248746, + "loss": 1.4697, + "theoretical_loss": 3.7875069899777047, + "tokens_seen": 685506560 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040017051153460383, + "loss": 1.5586, + "theoretical_loss": 3.7874700887151085, + "tokens_seen": 685572096 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040016048144433296, + "loss": 1.4832, + "theoretical_loss": 3.7874331919674464, + "tokens_seen": 685637632 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004001504513540622, + "loss": 1.4768, + "theoretical_loss": 3.787396299733735, + "tokens_seen": 685703168 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040014042126379143, + "loss": 1.3859, + "theoretical_loss": 3.78735941201299, + "tokens_seen": 685768704 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040013039117352056, + "loss": 1.5109, + "theoretical_loss": 3.7873225288042285, + "tokens_seen": 685834240 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004001203610832498, + "loss": 1.4322, + "theoretical_loss": 3.787285650106468, + "tokens_seen": 685899776 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004001103309929789, + "loss": 1.4804, + "theoretical_loss": 3.7872487759187248, + "tokens_seen": 685965312 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040010030090270815, + "loss": 1.5331, + "theoretical_loss": 3.7872119062400174, + "tokens_seen": 686030848 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040009027081243733, + "loss": 1.4015, + "theoretical_loss": 3.787175041069363, + "tokens_seen": 686096384 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004000802407221665, + "loss": 1.3456, + "theoretical_loss": 3.7871381804057807, + "tokens_seen": 686161920 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004000702106318957, + "loss": 1.3674, + "theoretical_loss": 3.7871013242482885, + "tokens_seen": 686227456 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040006018054162493, + "loss": 1.3824, + "theoretical_loss": 3.7870644725959055, + "tokens_seen": 686292992 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040005015045135406, + "loss": 1.3463, + "theoretical_loss": 3.787027625447651, + "tokens_seen": 686358528 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004000401203610833, + "loss": 1.3642, + "theoretical_loss": 3.7869907828025444, + "tokens_seen": 686424064 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0014091921038925648, + "objective/train/docs_used": 397954, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.757758617401123, + "objective/train/original_loss": 2.757758617401123, + "objective/train/theoretical_loss": 3.786953944659605, + "objective/train/tokens_used": 706949600, + "objective/train/value_avg": -0.006214141845703125, + "objective/train/value_loss": 6.799768016207963e-05, + "objective/train/value_max": -0.0004322528839111328, + "objective/train/value_min": -0.06866455078125, + "objective/train/value_reward_corr": 0.30771262689717543, + "objective/train/value_std": 0.00482177734375, + "objective/train/weight_avg": 0.989201009273529, + "objective/train/weighted_lm_loss": 2.7274909019470215, + "objective/train/weights_max": 1.9870387315750122, + "objective/train/weights_min": 0.6444298624992371, + "theoretical_loss": 3.786953944659605, + "tokens_seen": 686489600 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004000300902708124, + "loss": 1.3976, + "theoretical_loss": 3.786953944659605, + "tokens_seen": 686489600 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040002006018054166, + "loss": 1.4277, + "theoretical_loss": 3.7869171110178543, + "tokens_seen": 686555136 + }, + { + "epoch": 0.21, + "learning_rate": 0.00040001003009027084, + "loss": 1.4079, + "theoretical_loss": 3.7868802818763116, + "tokens_seen": 686620672 + }, + { + "epoch": 0.21, + "learning_rate": 0.0004, + "loss": 1.372, + "theoretical_loss": 3.7868434572339975, + "tokens_seen": 686686208 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003999899699097292, + "loss": 1.4618, + "theoretical_loss": 3.786806637089934, + "tokens_seen": 686751744 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003999799398194584, + "loss": 1.3831, + "theoretical_loss": 3.786769821443141, + "tokens_seen": 686817280 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039996990972918756, + "loss": 1.4302, + "theoretical_loss": 3.7867330102926413, + "tokens_seen": 686882816 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003999598796389168, + "loss": 1.4878, + "theoretical_loss": 3.7866962036374567, + "tokens_seen": 686948352 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003999498495486459, + "loss": 1.341, + "theoretical_loss": 3.7866594014766095, + "tokens_seen": 687013888 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039993981945837516, + "loss": 1.4732, + "theoretical_loss": 3.7866226038091213, + "tokens_seen": 687079424 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003999297893681043, + "loss": 1.4171, + "theoretical_loss": 3.786585810634016, + "tokens_seen": 687144960 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003999197592778335, + "loss": 1.5042, + "theoretical_loss": 3.7865490219503166, + "tokens_seen": 687210496 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003999097291875627, + "loss": 1.4872, + "theoretical_loss": 3.786512237757046, + "tokens_seen": 687276032 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003998996990972919, + "loss": 1.3718, + "theoretical_loss": 3.7864754580532294, + "tokens_seen": 687341568 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039988966900702107, + "loss": 1.4446, + "theoretical_loss": 3.7864386828378898, + "tokens_seen": 687407104 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003998796389167503, + "loss": 1.3252, + "theoretical_loss": 3.786401912110051, + "tokens_seen": 687472640 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039986960882647943, + "loss": 1.4941, + "theoretical_loss": 3.786365145868739, + "tokens_seen": 687538176 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039985957873620866, + "loss": 1.4244, + "theoretical_loss": 3.786328384112978, + "tokens_seen": 687603712 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003998495486459378, + "loss": 1.3731, + "theoretical_loss": 3.786291626841794, + "tokens_seen": 687669248 + }, + { + "epoch": 0.21, + "learning_rate": 0.000399839518555667, + "loss": 1.4039, + "theoretical_loss": 3.7862548740542117, + "tokens_seen": 687734784 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003998294884653962, + "loss": 1.4125, + "theoretical_loss": 3.7862181257492575, + "tokens_seen": 687800320 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003998194583751254, + "loss": 1.3662, + "theoretical_loss": 3.7861813819259575, + "tokens_seen": 687865856 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039980942828485457, + "loss": 1.5638, + "theoretical_loss": 3.7861446425833387, + "tokens_seen": 687931392 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039979939819458375, + "loss": 1.4921, + "theoretical_loss": 3.786107907720428, + "tokens_seen": 687996928 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039978936810431293, + "loss": 1.3043, + "theoretical_loss": 3.786071177336251, + "tokens_seen": 688062464 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0018266692059114575, + "objective/train/docs_used": 398552, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0198585987091064, + "objective/train/original_loss": 3.0198583602905273, + "objective/train/theoretical_loss": 3.7860344514298374, + "objective/train/tokens_used": 708588000, + "objective/train/value_avg": -0.00617218017578125, + "objective/train/value_loss": 7.404025382129475e-05, + "objective/train/value_max": -0.0005440711975097656, + "objective/train/value_min": -0.08538818359375, + "objective/train/value_reward_corr": 0.4056631869233186, + "objective/train/value_std": 0.005191802978515625, + "objective/train/weight_avg": 0.98524409532547, + "objective/train/weighted_lm_loss": 2.965506076812744, + "objective/train/weights_max": 2.008152961730957, + "objective/train/weights_min": 0.4606480002403259, + "theoretical_loss": 3.7860344514298374, + "tokens_seen": 688128000 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039977933801404217, + "loss": 1.4073, + "theoretical_loss": 3.7860344514298374, + "tokens_seen": 688128000 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003997693079237713, + "loss": 1.491, + "theoretical_loss": 3.7859977300002132, + "tokens_seen": 688193536 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039975927783350053, + "loss": 1.4312, + "theoretical_loss": 3.785961013046407, + "tokens_seen": 688259072 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039974924774322966, + "loss": 1.439, + "theoretical_loss": 3.785924300567448, + "tokens_seen": 688324608 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003997392176529589, + "loss": 1.4612, + "theoretical_loss": 3.7858875925623634, + "tokens_seen": 688390144 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039972918756268807, + "loss": 1.5216, + "theoretical_loss": 3.7858508890301827, + "tokens_seen": 688455680 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039971915747241725, + "loss": 1.4281, + "theoretical_loss": 3.785814189969935, + "tokens_seen": 688521216 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039970912738214643, + "loss": 1.5499, + "theoretical_loss": 3.7857774953806507, + "tokens_seen": 688586752 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039969909729187567, + "loss": 1.4518, + "theoretical_loss": 3.7857408052613595, + "tokens_seen": 688652288 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003996890672016048, + "loss": 1.4678, + "theoretical_loss": 3.7857041196110908, + "tokens_seen": 688717824 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039967903711133403, + "loss": 1.4806, + "theoretical_loss": 3.7856674384288755, + "tokens_seen": 688783360 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039966900702106316, + "loss": 1.3005, + "theoretical_loss": 3.785630761713744, + "tokens_seen": 688848896 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003996589769307924, + "loss": 1.4234, + "theoretical_loss": 3.7855940894647278, + "tokens_seen": 688914432 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003996489468405216, + "loss": 1.4535, + "theoretical_loss": 3.785557421680859, + "tokens_seen": 688979968 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039963891675025076, + "loss": 1.53, + "theoretical_loss": 3.7855207583611676, + "tokens_seen": 689045504 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039962888665997994, + "loss": 1.3621, + "theoretical_loss": 3.7854840995046866, + "tokens_seen": 689111040 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003996188565697091, + "loss": 1.3164, + "theoretical_loss": 3.7854474451104485, + "tokens_seen": 689176576 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003996088264794383, + "loss": 1.4157, + "theoretical_loss": 3.7854107951774854, + "tokens_seen": 689242112 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039959879638916754, + "loss": 1.3577, + "theoretical_loss": 3.7853741497048303, + "tokens_seen": 689307648 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039958876629889666, + "loss": 1.3424, + "theoretical_loss": 3.7853375086915166, + "tokens_seen": 689373184 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003995787362086259, + "loss": 1.4384, + "theoretical_loss": 3.785300872136577, + "tokens_seen": 689438720 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003995687061183551, + "loss": 1.4011, + "theoretical_loss": 3.785264240039046, + "tokens_seen": 689504256 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039955867602808426, + "loss": 1.2324, + "theoretical_loss": 3.7852276123979585, + "tokens_seen": 689569792 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039954864593781344, + "loss": 1.2975, + "theoretical_loss": 3.7851909892123476, + "tokens_seen": 689635328 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003995386158475426, + "loss": 1.4154, + "theoretical_loss": 3.7851543704812487, + "tokens_seen": 689700864 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0011306093074381351, + "objective/train/docs_used": 399765, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.528136968612671, + "objective/train/original_loss": 2.528136730194092, + "objective/train/theoretical_loss": 3.7851177562036957, + "objective/train/tokens_used": 710226400, + "objective/train/value_avg": -0.004253387451171875, + "objective/train/value_loss": 6.689542351523414e-05, + "objective/train/value_max": -0.00034737586975097656, + "objective/train/value_min": -0.05889892578125, + "objective/train/value_reward_corr": 0.2246658727631872, + "objective/train/value_std": 0.0034122467041015625, + "objective/train/weight_avg": 0.9914727210998535, + "objective/train/weighted_lm_loss": 2.51432466506958, + "objective/train/weights_max": 1.4216703176498413, + "objective/train/weights_min": 0.13721320033073425, + "theoretical_loss": 3.7851177562036957, + "tokens_seen": 689766400 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003995285857572718, + "loss": 1.302, + "theoretical_loss": 3.7851177562036957, + "tokens_seen": 689766400 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039951855566700104, + "loss": 1.3741, + "theoretical_loss": 3.785081146378725, + "tokens_seen": 689831936 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039950852557673017, + "loss": 1.3723, + "theoretical_loss": 3.785044541005372, + "tokens_seen": 689897472 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003994984954864594, + "loss": 1.5307, + "theoretical_loss": 3.785007940082673, + "tokens_seen": 689963008 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039948846539618853, + "loss": 1.5728, + "theoretical_loss": 3.784971343609663, + "tokens_seen": 690028544 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039947843530591776, + "loss": 1.2875, + "theoretical_loss": 3.7849347515853795, + "tokens_seen": 690094080 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039946840521564694, + "loss": 1.4453, + "theoretical_loss": 3.784898164008859, + "tokens_seen": 690159616 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003994583751253761, + "loss": 1.4092, + "theoretical_loss": 3.784861580879139, + "tokens_seen": 690225152 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003994483450351053, + "loss": 1.5741, + "theoretical_loss": 3.784825002195256, + "tokens_seen": 690290688 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003994383149448345, + "loss": 1.4242, + "theoretical_loss": 3.7847884279562485, + "tokens_seen": 690356224 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039942828485456367, + "loss": 1.4386, + "theoretical_loss": 3.7847518581611537, + "tokens_seen": 690421760 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003994182547642929, + "loss": 1.466, + "theoretical_loss": 3.784715292809011, + "tokens_seen": 690487296 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039940822467402203, + "loss": 1.2885, + "theoretical_loss": 3.7846787318988584, + "tokens_seen": 690552832 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039939819458375127, + "loss": 1.4117, + "theoretical_loss": 3.7846421754297346, + "tokens_seen": 690618368 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003993881644934805, + "loss": 1.371, + "theoretical_loss": 3.7846056234006786, + "tokens_seen": 690683904 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039937813440320963, + "loss": 1.415, + "theoretical_loss": 3.7845690758107313, + "tokens_seen": 690749440 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039936810431293886, + "loss": 1.4295, + "theoretical_loss": 3.7845325326589307, + "tokens_seen": 690814976 + }, + { + "epoch": 0.21, + "learning_rate": 0.000399358074222668, + "loss": 1.4, + "theoretical_loss": 3.784495993944318, + "tokens_seen": 690880512 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003993480441323972, + "loss": 1.5017, + "theoretical_loss": 3.784459459665934, + "tokens_seen": 690946048 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003993380140421264, + "loss": 1.5037, + "theoretical_loss": 3.7844229298228176, + "tokens_seen": 691011584 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003993279839518556, + "loss": 1.4138, + "theoretical_loss": 3.784386404414011, + "tokens_seen": 691077120 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039931795386158477, + "loss": 1.3743, + "theoretical_loss": 3.784349883438556, + "tokens_seen": 691142656 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039930792377131395, + "loss": 1.3362, + "theoretical_loss": 3.784313366895493, + "tokens_seen": 691208192 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039929789368104313, + "loss": 1.4172, + "theoretical_loss": 3.7842768547838648, + "tokens_seen": 691273728 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039928786359077237, + "loss": 1.4294, + "theoretical_loss": 3.7842403471027133, + "tokens_seen": 691339264 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0011779865017160773, + "objective/train/docs_used": 400468, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.890582799911499, + "objective/train/original_loss": 2.890582323074341, + "objective/train/theoretical_loss": 3.7842038438510803, + "objective/train/tokens_used": 711864800, + "objective/train/value_avg": -0.005191802978515625, + "objective/train/value_loss": 7.314129470614716e-05, + "objective/train/value_max": -0.0003275871276855469, + "objective/train/value_min": -0.06951904296875, + "objective/train/value_reward_corr": 0.35832292656574233, + "objective/train/value_std": 0.0039825439453125, + "objective/train/weight_avg": 0.9915988445281982, + "objective/train/weighted_lm_loss": 2.858529567718506, + "objective/train/weights_max": 1.8371838331222534, + "objective/train/weights_min": 0.3478482961654663, + "theoretical_loss": 3.7842038438510803, + "tokens_seen": 691404800 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003992778335005015, + "loss": 1.4007, + "theoretical_loss": 3.7842038438510803, + "tokens_seen": 691404800 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039926780341023073, + "loss": 1.2796, + "theoretical_loss": 3.7841673450280098, + "tokens_seen": 691470336 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039925777331995986, + "loss": 1.424, + "theoretical_loss": 3.784130850632544, + "tokens_seen": 691535872 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003992477432296891, + "loss": 1.4384, + "theoretical_loss": 3.7840943606637265, + "tokens_seen": 691601408 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039923771313941827, + "loss": 1.5405, + "theoretical_loss": 3.7840578751206015, + "tokens_seen": 691666944 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039922768304914745, + "loss": 1.3552, + "theoretical_loss": 3.784021394002212, + "tokens_seen": 691732480 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039921765295887663, + "loss": 1.4291, + "theoretical_loss": 3.7839849173076026, + "tokens_seen": 691798016 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039920762286860587, + "loss": 1.4427, + "theoretical_loss": 3.783948445035818, + "tokens_seen": 691863552 + }, + { + "epoch": 0.21, + "learning_rate": 0.000399197592778335, + "loss": 1.3863, + "theoretical_loss": 3.7839119771859036, + "tokens_seen": 691929088 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039918756268806423, + "loss": 1.4255, + "theoretical_loss": 3.783875513756904, + "tokens_seen": 691994624 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039917753259779336, + "loss": 1.4672, + "theoretical_loss": 3.7838390547478635, + "tokens_seen": 692060160 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003991675025075226, + "loss": 1.3962, + "theoretical_loss": 3.7838026001578298, + "tokens_seen": 692125696 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003991574724172518, + "loss": 1.3573, + "theoretical_loss": 3.7837661499858486, + "tokens_seen": 692191232 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039914744232698096, + "loss": 1.3974, + "theoretical_loss": 3.783729704230965, + "tokens_seen": 692256768 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039913741223671014, + "loss": 1.4071, + "theoretical_loss": 3.783693262892227, + "tokens_seen": 692322304 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003991273821464393, + "loss": 1.6062, + "theoretical_loss": 3.7836568259686807, + "tokens_seen": 692387840 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003991173520561685, + "loss": 1.4611, + "theoretical_loss": 3.7836203934593735, + "tokens_seen": 692453376 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039910732196589774, + "loss": 1.441, + "theoretical_loss": 3.7835839653633534, + "tokens_seen": 692518912 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039909729187562686, + "loss": 1.4622, + "theoretical_loss": 3.783547541679668, + "tokens_seen": 692584448 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003990872617853561, + "loss": 1.4482, + "theoretical_loss": 3.783511122407364, + "tokens_seen": 692649984 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003990772316950853, + "loss": 1.3805, + "theoretical_loss": 3.7834747075454924, + "tokens_seen": 692715520 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039906720160481446, + "loss": 1.3816, + "theoretical_loss": 3.7834382970930998, + "tokens_seen": 692781056 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039905717151454364, + "loss": 1.4423, + "theoretical_loss": 3.7834018910492366, + "tokens_seen": 692846592 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003990471414242728, + "loss": 1.3758, + "theoretical_loss": 3.783365489412951, + "tokens_seen": 692912128 + }, + { + "epoch": 0.21, + "learning_rate": 0.000399037111334002, + "loss": 1.453, + "theoretical_loss": 3.783329092183293, + "tokens_seen": 692977664 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0020834931638091803, + "objective/train/docs_used": 401917, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.978667974472046, + "objective/train/original_loss": 2.9786674976348877, + "objective/train/theoretical_loss": 3.783292699359313, + "objective/train/tokens_used": 713503200, + "objective/train/value_avg": -0.00670623779296875, + "objective/train/value_loss": 0.00012582865019794554, + "objective/train/value_max": -0.00036263465881347656, + "objective/train/value_min": -0.055938720703125, + "objective/train/value_reward_corr": 0.46476739156483293, + "objective/train/value_std": 0.0057525634765625, + "objective/train/weight_avg": 0.9847881197929382, + "objective/train/weighted_lm_loss": 2.942373752593994, + "objective/train/weights_max": 1.6141698360443115, + "objective/train/weights_min": 0.294904500246048, + "theoretical_loss": 3.783292699359313, + "tokens_seen": 693043200 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039902708124373124, + "loss": 1.4085, + "theoretical_loss": 3.783292699359313, + "tokens_seen": 693043200 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039901705115346037, + "loss": 1.6002, + "theoretical_loss": 3.78325631094006, + "tokens_seen": 693108736 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003990070210631896, + "loss": 1.4508, + "theoretical_loss": 3.7832199269245863, + "tokens_seen": 693174272 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039899699097291873, + "loss": 1.475, + "theoretical_loss": 3.783183547311941, + "tokens_seen": 693239808 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039898696088264796, + "loss": 1.4269, + "theoretical_loss": 3.7831471721011756, + "tokens_seen": 693305344 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039897693079237714, + "loss": 1.4067, + "theoretical_loss": 3.7831108012913415, + "tokens_seen": 693370880 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003989669007021063, + "loss": 1.4003, + "theoretical_loss": 3.783074434881491, + "tokens_seen": 693436416 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003989568706118355, + "loss": 1.4248, + "theoretical_loss": 3.783038072870675, + "tokens_seen": 693501952 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003989468405215647, + "loss": 1.4721, + "theoretical_loss": 3.783001715257946, + "tokens_seen": 693567488 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039893681043129387, + "loss": 1.4432, + "theoretical_loss": 3.7829653620423573, + "tokens_seen": 693633024 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003989267803410231, + "loss": 1.4539, + "theoretical_loss": 3.782929013222961, + "tokens_seen": 693698560 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039891675025075223, + "loss": 1.3794, + "theoretical_loss": 3.7828926687988105, + "tokens_seen": 693764096 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039890672016048147, + "loss": 1.3627, + "theoretical_loss": 3.7828563287689594, + "tokens_seen": 693829632 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039889669007021065, + "loss": 1.3886, + "theoretical_loss": 3.7828199931324606, + "tokens_seen": 693895168 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039888665997993983, + "loss": 1.4079, + "theoretical_loss": 3.782783661888369, + "tokens_seen": 693960704 + }, + { + "epoch": 0.21, + "learning_rate": 0.000398876629889669, + "loss": 1.4133, + "theoretical_loss": 3.7827473350357383, + "tokens_seen": 694026240 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003988665997993982, + "loss": 1.5054, + "theoretical_loss": 3.782711012573623, + "tokens_seen": 694091776 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039885656970912737, + "loss": 1.4955, + "theoretical_loss": 3.782674694501079, + "tokens_seen": 694157312 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003988465396188566, + "loss": 1.34, + "theoretical_loss": 3.7826383808171604, + "tokens_seen": 694222848 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039883650952858573, + "loss": 1.4562, + "theoretical_loss": 3.782602071520923, + "tokens_seen": 694288384 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039882647943831497, + "loss": 1.4971, + "theoretical_loss": 3.782565766611423, + "tokens_seen": 694353920 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003988164493480441, + "loss": 1.4342, + "theoretical_loss": 3.782529466087716, + "tokens_seen": 694419456 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039880641925777333, + "loss": 1.4, + "theoretical_loss": 3.782493169948858, + "tokens_seen": 694484992 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003987963891675025, + "loss": 1.5088, + "theoretical_loss": 3.782456878193906, + "tokens_seen": 694550528 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003987863590772317, + "loss": 1.5043, + "theoretical_loss": 3.7824205908219173, + "tokens_seen": 694616064 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0023984520230442286, + "objective/train/docs_used": 402587, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6212234497070312, + "objective/train/original_loss": 2.6212234497070312, + "objective/train/theoretical_loss": 3.782384307831949, + "objective/train/tokens_used": 715141600, + "objective/train/value_avg": -0.005939483642578125, + "objective/train/value_loss": 6.871621735626832e-05, + "objective/train/value_max": -0.0004189014434814453, + "objective/train/value_min": -0.055419921875, + "objective/train/value_reward_corr": 0.4350300906988164, + "objective/train/value_std": 0.005146026611328125, + "objective/train/weight_avg": 0.9792167544364929, + "objective/train/weighted_lm_loss": 2.5742788314819336, + "objective/train/weights_max": 1.4915393590927124, + "objective/train/weights_min": 0.2954900562763214, + "theoretical_loss": 3.782384307831949, + "tokens_seen": 694681600 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003987763289869609, + "loss": 1.3739, + "theoretical_loss": 3.782384307831949, + "tokens_seen": 694681600 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039876629889669006, + "loss": 1.5737, + "theoretical_loss": 3.7823480292230576, + "tokens_seen": 694747136 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039875626880641924, + "loss": 1.4432, + "theoretical_loss": 3.7823117549943017, + "tokens_seen": 694812672 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039874623871614847, + "loss": 1.4481, + "theoretical_loss": 3.7822754851447398, + "tokens_seen": 694878208 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003987362086258776, + "loss": 1.5116, + "theoretical_loss": 3.78223921967343, + "tokens_seen": 694943744 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039872617853560683, + "loss": 1.405, + "theoretical_loss": 3.782202958579431, + "tokens_seen": 695009280 + }, + { + "epoch": 0.21, + "learning_rate": 0.000398716148445336, + "loss": 1.4009, + "theoretical_loss": 3.782166701861801, + "tokens_seen": 695074816 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003987061183550652, + "loss": 1.3527, + "theoretical_loss": 3.7821304495195998, + "tokens_seen": 695140352 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003986960882647944, + "loss": 1.4321, + "theoretical_loss": 3.782094201551887, + "tokens_seen": 695205888 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039868605817452356, + "loss": 1.4129, + "theoretical_loss": 3.782057957957723, + "tokens_seen": 695271424 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039867602808425274, + "loss": 1.5016, + "theoretical_loss": 3.782021718736167, + "tokens_seen": 695336960 + }, + { + "epoch": 0.21, + "learning_rate": 0.000398665997993982, + "loss": 1.5312, + "theoretical_loss": 3.7819854838862796, + "tokens_seen": 695402496 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003986559679037111, + "loss": 1.4342, + "theoretical_loss": 3.781949253407122, + "tokens_seen": 695468032 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039864593781344034, + "loss": 1.4769, + "theoretical_loss": 3.7819130272977546, + "tokens_seen": 695533568 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003986359077231695, + "loss": 1.4171, + "theoretical_loss": 3.781876805557239, + "tokens_seen": 695599104 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003986258776328987, + "loss": 1.3821, + "theoretical_loss": 3.781840588184637, + "tokens_seen": 695664640 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039861584754262794, + "loss": 1.3214, + "theoretical_loss": 3.7818043751790102, + "tokens_seen": 695730176 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039860581745235706, + "loss": 1.4037, + "theoretical_loss": 3.781768166539421, + "tokens_seen": 695795712 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003985957873620863, + "loss": 1.3791, + "theoretical_loss": 3.7817319622649315, + "tokens_seen": 695861248 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003985857572718155, + "loss": 1.3501, + "theoretical_loss": 3.7816957623546044, + "tokens_seen": 695926784 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039857572718154466, + "loss": 1.5179, + "theoretical_loss": 3.7816595668075026, + "tokens_seen": 695992320 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039856569709127384, + "loss": 1.365, + "theoretical_loss": 3.78162337562269, + "tokens_seen": 696057856 + }, + { + "epoch": 0.21, + "learning_rate": 0.000398555667001003, + "loss": 1.3958, + "theoretical_loss": 3.7815871887992305, + "tokens_seen": 696123392 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003985456369107322, + "loss": 1.4331, + "theoretical_loss": 3.781551006336187, + "tokens_seen": 696188928 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039853560682046144, + "loss": 1.3404, + "theoretical_loss": 3.7815148282326243, + "tokens_seen": 696254464 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": 3.4161446819780394e-05, + "objective/train/docs_used": 403173, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5027289390563965, + "objective/train/original_loss": 2.5027291774749756, + "objective/train/theoretical_loss": 3.7814786544876062, + "objective/train/tokens_used": 716780000, + "objective/train/value_avg": -0.004634857177734375, + "objective/train/value_loss": 5.077426612842828e-05, + "objective/train/value_max": -0.0003407001495361328, + "objective/train/value_min": -0.06475830078125, + "objective/train/value_reward_corr": 0.2752202185149376, + "objective/train/value_std": 0.003475189208984375, + "objective/train/weight_avg": 1.0026977062225342, + "objective/train/weighted_lm_loss": 2.499101161956787, + "objective/train/weights_max": 1.9109165668487549, + "objective/train/weights_min": 0.38144850730895996, + "theoretical_loss": 3.7814786544876062, + "tokens_seen": 696320000 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039852557673019057, + "loss": 1.2826, + "theoretical_loss": 3.7814786544876062, + "tokens_seen": 696320000 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003985155466399198, + "loss": 1.4323, + "theoretical_loss": 3.7814424851001984, + "tokens_seen": 696385536 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039850551654964893, + "loss": 1.4305, + "theoretical_loss": 3.7814063200694656, + "tokens_seen": 696451072 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039849548645937816, + "loss": 1.3383, + "theoretical_loss": 3.7813701593944726, + "tokens_seen": 696516608 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039848545636910734, + "loss": 1.4199, + "theoretical_loss": 3.781334003074286, + "tokens_seen": 696582144 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003984754262788365, + "loss": 1.3333, + "theoretical_loss": 3.781297851107971, + "tokens_seen": 696647680 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003984653961885657, + "loss": 1.4123, + "theoretical_loss": 3.781261703494594, + "tokens_seen": 696713216 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003984553660982949, + "loss": 1.3889, + "theoretical_loss": 3.7812255602332216, + "tokens_seen": 696778752 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039844533600802407, + "loss": 1.4863, + "theoretical_loss": 3.78118942132292, + "tokens_seen": 696844288 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003984353059177533, + "loss": 1.4124, + "theoretical_loss": 3.7811532867627573, + "tokens_seen": 696909824 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039842527582748243, + "loss": 1.4478, + "theoretical_loss": 3.7811171565518, + "tokens_seen": 696975360 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039841524573721167, + "loss": 1.3115, + "theoretical_loss": 3.7810810306891165, + "tokens_seen": 697040896 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039840521564694085, + "loss": 1.4383, + "theoretical_loss": 3.781044909173774, + "tokens_seen": 697106432 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039839518555667003, + "loss": 1.4781, + "theoretical_loss": 3.7810087920048407, + "tokens_seen": 697171968 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003983851554663992, + "loss": 1.2833, + "theoretical_loss": 3.780972679181386, + "tokens_seen": 697237504 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003983751253761284, + "loss": 1.4284, + "theoretical_loss": 3.780936570702478, + "tokens_seen": 697303040 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039836509528585757, + "loss": 1.4614, + "theoretical_loss": 3.7809004665671857, + "tokens_seen": 697368576 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003983550651955868, + "loss": 1.3807, + "theoretical_loss": 3.7808643667745785, + "tokens_seen": 697434112 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039834503510531593, + "loss": 1.4364, + "theoretical_loss": 3.7808282713237262, + "tokens_seen": 697499648 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039833500501504517, + "loss": 1.4322, + "theoretical_loss": 3.780792180213699, + "tokens_seen": 697565184 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003983249749247743, + "loss": 1.4072, + "theoretical_loss": 3.780756093443567, + "tokens_seen": 697630720 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039831494483450353, + "loss": 1.3993, + "theoretical_loss": 3.7807200110124004, + "tokens_seen": 697696256 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003983049147442327, + "loss": 1.406, + "theoretical_loss": 3.7806839329192705, + "tokens_seen": 697761792 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003982948846539619, + "loss": 1.465, + "theoretical_loss": 3.780647859163248, + "tokens_seen": 697827328 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003982848545636911, + "loss": 1.3053, + "theoretical_loss": 3.7806117897434044, + "tokens_seen": 697892864 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0006442750454880297, + "objective/train/docs_used": 404489, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.015889883041382, + "objective/train/original_loss": 3.015889883041382, + "objective/train/theoretical_loss": 3.780575724658811, + "objective/train/tokens_used": 718418400, + "objective/train/value_avg": -0.005626678466796875, + "objective/train/value_loss": 6.982935883570462e-05, + "objective/train/value_max": -0.0005464553833007812, + "objective/train/value_min": -0.06256103515625, + "objective/train/value_reward_corr": 0.3467223510936132, + "objective/train/value_std": 0.004390716552734375, + "objective/train/weight_avg": 0.9967947602272034, + "objective/train/weighted_lm_loss": 2.995854139328003, + "objective/train/weights_max": 1.8693865537643433, + "objective/train/weights_min": 0.19551745057106018, + "theoretical_loss": 3.780575724658811, + "tokens_seen": 697958400 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039827482447342026, + "loss": 1.3662, + "theoretical_loss": 3.780575724658811, + "tokens_seen": 697958400 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039826479438314944, + "loss": 1.3908, + "theoretical_loss": 3.7805396639085407, + "tokens_seen": 698023936 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039825476429287867, + "loss": 1.4541, + "theoretical_loss": 3.7805036074916645, + "tokens_seen": 698089472 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003982447342026078, + "loss": 1.2691, + "theoretical_loss": 3.780467555407255, + "tokens_seen": 698155008 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039823470411233703, + "loss": 1.3891, + "theoretical_loss": 3.7804315076543866, + "tokens_seen": 698220544 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003982246740220662, + "loss": 1.3482, + "theoretical_loss": 3.7803954642321314, + "tokens_seen": 698286080 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003982146439317954, + "loss": 1.3077, + "theoretical_loss": 3.780359425139562, + "tokens_seen": 698351616 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003982046138415246, + "loss": 1.4327, + "theoretical_loss": 3.7803233903757527, + "tokens_seen": 698417152 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039819458375125376, + "loss": 1.3523, + "theoretical_loss": 3.780287359939778, + "tokens_seen": 698482688 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039818455366098294, + "loss": 1.5071, + "theoretical_loss": 3.7802513338307113, + "tokens_seen": 698548224 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003981745235707122, + "loss": 1.345, + "theoretical_loss": 3.780215312047628, + "tokens_seen": 698613760 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003981644934804413, + "loss": 1.4533, + "theoretical_loss": 3.780179294589601, + "tokens_seen": 698679296 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039815446339017054, + "loss": 1.3612, + "theoretical_loss": 3.7801432814557083, + "tokens_seen": 698744832 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039814443329989966, + "loss": 1.3255, + "theoretical_loss": 3.780107272645023, + "tokens_seen": 698810368 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003981344032096289, + "loss": 1.3268, + "theoretical_loss": 3.780071268156621, + "tokens_seen": 698875904 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003981243731193581, + "loss": 1.377, + "theoretical_loss": 3.7800352679895797, + "tokens_seen": 698941440 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039811434302908726, + "loss": 1.3963, + "theoretical_loss": 3.779999272142973, + "tokens_seen": 699006976 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039810431293881644, + "loss": 1.4844, + "theoretical_loss": 3.7799632806158794, + "tokens_seen": 699072512 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003980942828485457, + "loss": 1.4458, + "theoretical_loss": 3.779927293407375, + "tokens_seen": 699138048 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003980842527582748, + "loss": 1.4437, + "theoretical_loss": 3.779891310516537, + "tokens_seen": 699203584 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039807422266800404, + "loss": 1.2793, + "theoretical_loss": 3.7798553319424424, + "tokens_seen": 699269120 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039806419257773317, + "loss": 1.2488, + "theoretical_loss": 3.779819357684169, + "tokens_seen": 699334656 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003980541624874624, + "loss": 1.3689, + "theoretical_loss": 3.7797833877407947, + "tokens_seen": 699400192 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003980441323971916, + "loss": 1.2899, + "theoretical_loss": 3.779747422111398, + "tokens_seen": 699465728 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039803410230692077, + "loss": 1.4314, + "theoretical_loss": 3.779711460795057, + "tokens_seen": 699531264 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.000683393853250891, + "objective/train/docs_used": 405172, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7430992126464844, + "objective/train/original_loss": 2.7430992126464844, + "objective/train/theoretical_loss": 3.779675503790851, + "objective/train/tokens_used": 720056800, + "objective/train/value_avg": -0.005153656005859375, + "objective/train/value_loss": 4.9213547754334286e-05, + "objective/train/value_max": -0.0004839897155761719, + "objective/train/value_min": -0.05059814453125, + "objective/train/value_reward_corr": 0.3875030433278481, + "objective/train/value_std": 0.004150390625, + "objective/train/weight_avg": 0.9955289959907532, + "objective/train/weighted_lm_loss": 2.7319252490997314, + "objective/train/weights_max": 1.520914077758789, + "objective/train/weights_min": 0.30897122621536255, + "theoretical_loss": 3.779675503790851, + "tokens_seen": 699596800 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039802407221664995, + "loss": 1.4494, + "theoretical_loss": 3.779675503790851, + "tokens_seen": 699596800 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039801404212637913, + "loss": 1.5399, + "theoretical_loss": 3.7796395510978584, + "tokens_seen": 699662336 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003980040120361083, + "loss": 1.4901, + "theoretical_loss": 3.7796036027151585, + "tokens_seen": 699727872 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039799398194583754, + "loss": 1.451, + "theoretical_loss": 3.779567658641832, + "tokens_seen": 699793408 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039798395185556667, + "loss": 1.2681, + "theoretical_loss": 3.7795317188769575, + "tokens_seen": 699858944 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003979739217652959, + "loss": 1.4049, + "theoretical_loss": 3.779495783419616, + "tokens_seen": 699924480 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039796389167502503, + "loss": 1.388, + "theoretical_loss": 3.779459852268888, + "tokens_seen": 699990016 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039795386158475427, + "loss": 1.3778, + "theoretical_loss": 3.7794239254238535, + "tokens_seen": 700055552 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039794383149448345, + "loss": 1.357, + "theoretical_loss": 3.779388002883594, + "tokens_seen": 700121088 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039793380140421263, + "loss": 1.3501, + "theoretical_loss": 3.7793520846471913, + "tokens_seen": 700186624 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003979237713139418, + "loss": 1.3835, + "theoretical_loss": 3.7793161707137264, + "tokens_seen": 700252160 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039791374122367105, + "loss": 1.4427, + "theoretical_loss": 3.779280261082281, + "tokens_seen": 700317696 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003979037111334002, + "loss": 1.4314, + "theoretical_loss": 3.7792443557519375, + "tokens_seen": 700383232 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003978936810431294, + "loss": 1.38, + "theoretical_loss": 3.779208454721779, + "tokens_seen": 700448768 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003978836509528586, + "loss": 1.3629, + "theoretical_loss": 3.7791725579908872, + "tokens_seen": 700514304 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039787362086258777, + "loss": 1.4799, + "theoretical_loss": 3.7791366655583456, + "tokens_seen": 700579840 + }, + { + "epoch": 0.21, + "learning_rate": 0.000397863590772317, + "loss": 1.4159, + "theoretical_loss": 3.7791007774232375, + "tokens_seen": 700645376 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039785356068204613, + "loss": 1.5243, + "theoretical_loss": 3.7790648935846463, + "tokens_seen": 700710912 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039784353059177537, + "loss": 1.4947, + "theoretical_loss": 3.7790290140416554, + "tokens_seen": 700776448 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003978335005015045, + "loss": 1.3832, + "theoretical_loss": 3.7789931387933504, + "tokens_seen": 700841984 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039782347041123373, + "loss": 1.4318, + "theoretical_loss": 3.7789572678388144, + "tokens_seen": 700907520 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003978134403209629, + "loss": 1.4004, + "theoretical_loss": 3.778921401177133, + "tokens_seen": 700973056 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003978034102306921, + "loss": 1.4174, + "theoretical_loss": 3.7788855388073896, + "tokens_seen": 701038592 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003977933801404213, + "loss": 1.3626, + "theoretical_loss": 3.778849680728671, + "tokens_seen": 701104128 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039778335005015046, + "loss": 1.4981, + "theoretical_loss": 3.7788138269400626, + "tokens_seen": 701169664 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.00028744677547365427, + "objective/train/docs_used": 406046, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6484410762786865, + "objective/train/original_loss": 2.6484413146972656, + "objective/train/theoretical_loss": 3.778777977440649, + "objective/train/tokens_used": 721695200, + "objective/train/value_avg": -0.005695343017578125, + "objective/train/value_loss": 5.690029502147809e-05, + "objective/train/value_max": -0.0003514289855957031, + "objective/train/value_min": -0.0943603515625, + "objective/train/value_reward_corr": 0.469985296112285, + "objective/train/value_std": 0.004932403564453125, + "objective/train/weight_avg": 0.9998338222503662, + "objective/train/weighted_lm_loss": 2.6438395977020264, + "objective/train/weights_max": 2.569222927093506, + "objective/train/weights_min": 0.35479021072387695, + "theoretical_loss": 3.778777977440649, + "tokens_seen": 701235200 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039777331995987964, + "loss": 1.343, + "theoretical_loss": 3.778777977440649, + "tokens_seen": 701235200 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039776328986960887, + "loss": 1.4511, + "theoretical_loss": 3.7787421322295183, + "tokens_seen": 701300736 + }, + { + "epoch": 0.21, + "learning_rate": 0.000397753259779338, + "loss": 1.3036, + "theoretical_loss": 3.778706291305755, + "tokens_seen": 701366272 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039774322968906723, + "loss": 1.4209, + "theoretical_loss": 3.778670454668447, + "tokens_seen": 701431808 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003977331995987964, + "loss": 1.2605, + "theoretical_loss": 3.7786346223166802, + "tokens_seen": 701497344 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003977231695085256, + "loss": 1.2755, + "theoretical_loss": 3.7785987942495423, + "tokens_seen": 701562880 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003977131394182548, + "loss": 1.5554, + "theoretical_loss": 3.7785629704661217, + "tokens_seen": 701628416 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039770310932798396, + "loss": 1.402, + "theoretical_loss": 3.7785271509655045, + "tokens_seen": 701693952 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039769307923771314, + "loss": 1.4076, + "theoretical_loss": 3.7784913357467795, + "tokens_seen": 701759488 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003976830491474424, + "loss": 1.39, + "theoretical_loss": 3.7784555248090355, + "tokens_seen": 701825024 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003976730190571715, + "loss": 1.3775, + "theoretical_loss": 3.77841971815136, + "tokens_seen": 701890560 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039766298896690074, + "loss": 1.3655, + "theoretical_loss": 3.778383915772843, + "tokens_seen": 701956096 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039765295887662987, + "loss": 1.3597, + "theoretical_loss": 3.7783481176725737, + "tokens_seen": 702021632 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003976429287863591, + "loss": 1.3795, + "theoretical_loss": 3.778312323849641, + "tokens_seen": 702087168 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003976328986960883, + "loss": 1.4519, + "theoretical_loss": 3.778276534303134, + "tokens_seen": 702152704 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039762286860581746, + "loss": 1.4689, + "theoretical_loss": 3.778240749032144, + "tokens_seen": 702218240 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039761283851554664, + "loss": 1.3274, + "theoretical_loss": 3.778204968035761, + "tokens_seen": 702283776 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003976028084252759, + "loss": 1.2791, + "theoretical_loss": 3.7781691913130753, + "tokens_seen": 702349312 + }, + { + "epoch": 0.21, + "learning_rate": 0.000397592778335005, + "loss": 1.3645, + "theoretical_loss": 3.7781334188631774, + "tokens_seen": 702414848 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039758274824473424, + "loss": 1.4357, + "theoretical_loss": 3.778097650685159, + "tokens_seen": 702480384 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039757271815446337, + "loss": 1.3409, + "theoretical_loss": 3.778061886778111, + "tokens_seen": 702545920 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003975626880641926, + "loss": 1.4408, + "theoretical_loss": 3.7780261271411253, + "tokens_seen": 702611456 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003975526579739218, + "loss": 1.365, + "theoretical_loss": 3.777990371773294, + "tokens_seen": 702676992 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039754262788365097, + "loss": 1.5398, + "theoretical_loss": 3.777954620673709, + "tokens_seen": 702742528 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039753259779338015, + "loss": 1.3854, + "theoretical_loss": 3.7779188738414637, + "tokens_seen": 702808064 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0016024294309318066, + "objective/train/docs_used": 406800, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8573050498962402, + "objective/train/original_loss": 2.8573057651519775, + "objective/train/theoretical_loss": 3.77788313127565, + "objective/train/tokens_used": 723333600, + "objective/train/value_avg": -0.00478363037109375, + "objective/train/value_loss": 7.03486439306289e-05, + "objective/train/value_max": -0.0004458427429199219, + "objective/train/value_min": -0.0517578125, + "objective/train/value_reward_corr": 0.2962702639220285, + "objective/train/value_std": 0.0036983489990234375, + "objective/train/weight_avg": 0.9870296716690063, + "objective/train/weighted_lm_loss": 2.834770679473877, + "objective/train/weights_max": 1.4765517711639404, + "objective/train/weights_min": 0.21637825667858124, + "theoretical_loss": 3.77788313127565, + "tokens_seen": 702873600 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039752256770310933, + "loss": 1.3996, + "theoretical_loss": 3.77788313127565, + "tokens_seen": 702873600 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003975125376128385, + "loss": 1.5403, + "theoretical_loss": 3.7778473929753607, + "tokens_seen": 702939136 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039750250752256774, + "loss": 1.4541, + "theoretical_loss": 3.77781165893969, + "tokens_seen": 703004672 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039749247743229687, + "loss": 1.3418, + "theoretical_loss": 3.7777759291677313, + "tokens_seen": 703070208 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003974824473420261, + "loss": 1.4162, + "theoretical_loss": 3.7777402036585777, + "tokens_seen": 703135744 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039747241725175523, + "loss": 1.4423, + "theoretical_loss": 3.7777044824113246, + "tokens_seen": 703201280 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039746238716148447, + "loss": 1.4788, + "theoretical_loss": 3.777668765425066, + "tokens_seen": 703266816 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039745235707121365, + "loss": 1.4116, + "theoretical_loss": 3.7776330526988957, + "tokens_seen": 703332352 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039744232698094283, + "loss": 1.47, + "theoretical_loss": 3.7775973442319097, + "tokens_seen": 703397888 + }, + { + "epoch": 0.21, + "learning_rate": 0.000397432296890672, + "loss": 1.5335, + "theoretical_loss": 3.7775616400232037, + "tokens_seen": 703463424 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039742226680040125, + "loss": 1.498, + "theoretical_loss": 3.777525940071872, + "tokens_seen": 703528960 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003974122367101304, + "loss": 1.3236, + "theoretical_loss": 3.7774902443770113, + "tokens_seen": 703594496 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003974022066198596, + "loss": 1.4086, + "theoretical_loss": 3.7774545529377175, + "tokens_seen": 703660032 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039739217652958874, + "loss": 1.3625, + "theoretical_loss": 3.777418865753087, + "tokens_seen": 703725568 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039738214643931797, + "loss": 1.389, + "theoretical_loss": 3.7773831828222164, + "tokens_seen": 703791104 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039737211634904715, + "loss": 1.3732, + "theoretical_loss": 3.7773475041442026, + "tokens_seen": 703856640 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039736208625877633, + "loss": 1.5073, + "theoretical_loss": 3.7773118297181427, + "tokens_seen": 703922176 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003973520561685055, + "loss": 1.3871, + "theoretical_loss": 3.777276159543135, + "tokens_seen": 703987712 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003973420260782347, + "loss": 1.3692, + "theoretical_loss": 3.777240493618276, + "tokens_seen": 704053248 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003973319959879639, + "loss": 1.4508, + "theoretical_loss": 3.7772048319426643, + "tokens_seen": 704118784 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003973219658976931, + "loss": 1.3923, + "theoretical_loss": 3.7771691745153984, + "tokens_seen": 704184320 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039731193580742224, + "loss": 1.4513, + "theoretical_loss": 3.7771335213355766, + "tokens_seen": 704249856 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003973019057171515, + "loss": 1.3783, + "theoretical_loss": 3.7770978724022983, + "tokens_seen": 704315392 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003972918756268806, + "loss": 1.4125, + "theoretical_loss": 3.777062227714662, + "tokens_seen": 704380928 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039728184553660984, + "loss": 1.3807, + "theoretical_loss": 3.7770265872717674, + "tokens_seen": 704446464 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": 0.00024578391457907856, + "objective/train/docs_used": 408057, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8373727798461914, + "objective/train/original_loss": 2.8373727798461914, + "objective/train/theoretical_loss": 3.7769909510727144, + "objective/train/tokens_used": 724972000, + "objective/train/value_avg": -0.005779266357421875, + "objective/train/value_loss": 5.6677963584661484e-05, + "objective/train/value_max": -0.000469207763671875, + "objective/train/value_min": -0.06439208984375, + "objective/train/value_reward_corr": 0.4544331485413781, + "objective/train/value_std": 0.005260467529296875, + "objective/train/weight_avg": 1.005242109298706, + "objective/train/weighted_lm_loss": 2.8459408283233643, + "objective/train/weights_max": 1.7087008953094482, + "objective/train/weights_min": 0.5960291028022766, + "theoretical_loss": 3.7769909510727144, + "tokens_seen": 704512000 + }, + { + "epoch": 0.21, + "learning_rate": 0.000397271815446339, + "loss": 1.4187, + "theoretical_loss": 3.7769909510727144, + "tokens_seen": 704512000 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003972617853560682, + "loss": 1.5268, + "theoretical_loss": 3.7769553191166025, + "tokens_seen": 704577536 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003972517552657974, + "loss": 1.5677, + "theoretical_loss": 3.776919691402532, + "tokens_seen": 704643072 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003972417251755266, + "loss": 1.3829, + "theoretical_loss": 3.776884067929604, + "tokens_seen": 704708608 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039723169508525574, + "loss": 1.4867, + "theoretical_loss": 3.776848448696919, + "tokens_seen": 704774144 + }, + { + "epoch": 0.21, + "learning_rate": 0.000397221664994985, + "loss": 1.4161, + "theoretical_loss": 3.776812833703578, + "tokens_seen": 704839680 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003972116349047141, + "loss": 1.4867, + "theoretical_loss": 3.776777222948682, + "tokens_seen": 704905216 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039720160481444334, + "loss": 1.506, + "theoretical_loss": 3.776741616431332, + "tokens_seen": 704970752 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003971915747241725, + "loss": 1.2603, + "theoretical_loss": 3.776706014150632, + "tokens_seen": 705036288 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003971815446339017, + "loss": 1.5985, + "theoretical_loss": 3.7766704161056825, + "tokens_seen": 705101824 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003971715145436309, + "loss": 1.4896, + "theoretical_loss": 3.776634822295587, + "tokens_seen": 705167360 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039716148445336007, + "loss": 1.5281, + "theoretical_loss": 3.7765992327194464, + "tokens_seen": 705232896 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039715145436308925, + "loss": 1.4005, + "theoretical_loss": 3.7765636473763653, + "tokens_seen": 705298432 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003971414242728185, + "loss": 1.4853, + "theoretical_loss": 3.776528066265447, + "tokens_seen": 705363968 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039713139418254766, + "loss": 1.4413, + "theoretical_loss": 3.776492489385794, + "tokens_seen": 705429504 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039712136409227684, + "loss": 1.3918, + "theoretical_loss": 3.7764569167365103, + "tokens_seen": 705495040 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003971113340020061, + "loss": 1.3756, + "theoretical_loss": 3.776421348316701, + "tokens_seen": 705560576 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003971013039117352, + "loss": 1.5084, + "theoretical_loss": 3.7763857841254698, + "tokens_seen": 705626112 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039709127382146444, + "loss": 1.2797, + "theoretical_loss": 3.7763502241619205, + "tokens_seen": 705691648 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039708124373119357, + "loss": 1.4899, + "theoretical_loss": 3.776314668425159, + "tokens_seen": 705757184 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003970712136409228, + "loss": 1.4247, + "theoretical_loss": 3.7762791169142904, + "tokens_seen": 705822720 + }, + { + "epoch": 0.21, + "learning_rate": 0.000397061183550652, + "loss": 1.3541, + "theoretical_loss": 3.77624356962842, + "tokens_seen": 705888256 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039705115346038117, + "loss": 1.4298, + "theoretical_loss": 3.7762080265666533, + "tokens_seen": 705953792 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039704112337011035, + "loss": 1.3839, + "theoretical_loss": 3.7761724877280964, + "tokens_seen": 706019328 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039703109327983953, + "loss": 1.4105, + "theoretical_loss": 3.776136953111856, + "tokens_seen": 706084864 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0004112911701668054, + "objective/train/docs_used": 408602, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.346721887588501, + "objective/train/original_loss": 3.346721887588501, + "objective/train/theoretical_loss": 3.7761014227170375, + "objective/train/tokens_used": 726610400, + "objective/train/value_avg": -0.004772186279296875, + "objective/train/value_loss": 5.541608697967604e-05, + "objective/train/value_max": -0.00026726722717285156, + "objective/train/value_min": -0.05511474609375, + "objective/train/value_reward_corr": 0.4482914173243421, + "objective/train/value_std": 0.00402069091796875, + "objective/train/weight_avg": 0.998458981513977, + "objective/train/weighted_lm_loss": 3.3471128940582275, + "objective/train/weights_max": 1.7352429628372192, + "objective/train/weights_min": 0.3525371849536896, + "theoretical_loss": 3.7761014227170375, + "tokens_seen": 706150400 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003970210631895687, + "loss": 1.5056, + "theoretical_loss": 3.7761014227170375, + "tokens_seen": 706150400 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039701103309929794, + "loss": 1.4246, + "theoretical_loss": 3.776065896542749, + "tokens_seen": 706215936 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039700100300902707, + "loss": 1.5597, + "theoretical_loss": 3.7760303745880965, + "tokens_seen": 706281472 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003969909729187563, + "loss": 1.5001, + "theoretical_loss": 3.775994856852188, + "tokens_seen": 706347008 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039698094282848543, + "loss": 1.5076, + "theoretical_loss": 3.775959343334131, + "tokens_seen": 706412544 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039697091273821467, + "loss": 1.3899, + "theoretical_loss": 3.7759238340330334, + "tokens_seen": 706478080 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039696088264794385, + "loss": 1.4522, + "theoretical_loss": 3.775888328948003, + "tokens_seen": 706543616 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039695085255767303, + "loss": 1.3386, + "theoretical_loss": 3.7758528280781487, + "tokens_seen": 706609152 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003969408224674022, + "loss": 1.4404, + "theoretical_loss": 3.7758173314225787, + "tokens_seen": 706674688 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039693079237713145, + "loss": 1.4543, + "theoretical_loss": 3.7757818389804023, + "tokens_seen": 706740224 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003969207622868606, + "loss": 1.3678, + "theoretical_loss": 3.7757463507507287, + "tokens_seen": 706805760 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003969107321965898, + "loss": 1.2916, + "theoretical_loss": 3.775710866732668, + "tokens_seen": 706871296 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039690070210631894, + "loss": 1.4841, + "theoretical_loss": 3.775675386925329, + "tokens_seen": 706936832 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039689067201604817, + "loss": 1.4775, + "theoretical_loss": 3.775639911327821, + "tokens_seen": 707002368 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039688064192577735, + "loss": 1.4193, + "theoretical_loss": 3.775604439939257, + "tokens_seen": 707067904 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039687061183550653, + "loss": 1.468, + "theoretical_loss": 3.7755689727587454, + "tokens_seen": 707133440 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003968605817452357, + "loss": 1.4783, + "theoretical_loss": 3.775533509785398, + "tokens_seen": 707198976 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003968505516549649, + "loss": 1.3778, + "theoretical_loss": 3.7754980510183254, + "tokens_seen": 707264512 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003968405215646941, + "loss": 1.3753, + "theoretical_loss": 3.775462596456639, + "tokens_seen": 707330048 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003968304914744233, + "loss": 1.3828, + "theoretical_loss": 3.7754271460994513, + "tokens_seen": 707395584 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039682046138415244, + "loss": 1.4981, + "theoretical_loss": 3.775391699945873, + "tokens_seen": 707461120 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003968104312938817, + "loss": 1.4732, + "theoretical_loss": 3.7753562579950177, + "tokens_seen": 707526656 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003968004012036108, + "loss": 1.4096, + "theoretical_loss": 3.7753208202459962, + "tokens_seen": 707592192 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039679037111334004, + "loss": 1.3226, + "theoretical_loss": 3.7752853866979232, + "tokens_seen": 707657728 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003967803410230692, + "loss": 1.4928, + "theoretical_loss": 3.77524995734991, + "tokens_seen": 707723264 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": -0.0005370975704863667, + "objective/train/docs_used": 409727, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6752877235412598, + "objective/train/original_loss": 2.6752877235412598, + "objective/train/theoretical_loss": 3.775214532201071, + "objective/train/tokens_used": 728248800, + "objective/train/value_avg": -0.00438690185546875, + "objective/train/value_loss": 6.0333055444061756e-05, + "objective/train/value_max": -0.0003845691680908203, + "objective/train/value_min": -0.08135986328125, + "objective/train/value_reward_corr": 0.5083803720559075, + "objective/train/value_std": 0.00453948974609375, + "objective/train/weight_avg": 0.9974194169044495, + "objective/train/weighted_lm_loss": 2.66400146484375, + "objective/train/weights_max": 1.8929946422576904, + "objective/train/weights_min": 0.4110603630542755, + "theoretical_loss": 3.775214532201071, + "tokens_seen": 707788800 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003967703109327984, + "loss": 1.448, + "theoretical_loss": 3.775214532201071, + "tokens_seen": 707788800 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003967602808425276, + "loss": 1.4998, + "theoretical_loss": 3.7751791112505195, + "tokens_seen": 707854336 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003967502507522568, + "loss": 1.5344, + "theoretical_loss": 3.775143694497369, + "tokens_seen": 707919872 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039674022066198594, + "loss": 1.4265, + "theoretical_loss": 3.7751082819407333, + "tokens_seen": 707985408 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003967301905717152, + "loss": 1.4467, + "theoretical_loss": 3.775072873579728, + "tokens_seen": 708050944 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003967201604814443, + "loss": 1.3946, + "theoretical_loss": 3.7750374694134665, + "tokens_seen": 708116480 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039671013039117354, + "loss": 1.4601, + "theoretical_loss": 3.7750020694410646, + "tokens_seen": 708182016 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003967001003009027, + "loss": 1.4293, + "theoretical_loss": 3.774966673661637, + "tokens_seen": 708247552 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003966900702106319, + "loss": 1.401, + "theoretical_loss": 3.7749312820742995, + "tokens_seen": 708313088 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003966800401203611, + "loss": 1.4601, + "theoretical_loss": 3.774895894678167, + "tokens_seen": 708378624 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039667001003009027, + "loss": 1.4041, + "theoretical_loss": 3.7748605114723564, + "tokens_seen": 708444160 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039665997993981945, + "loss": 1.4762, + "theoretical_loss": 3.7748251324559834, + "tokens_seen": 708509696 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003966499498495487, + "loss": 1.4181, + "theoretical_loss": 3.774789757628165, + "tokens_seen": 708575232 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003966399197592778, + "loss": 1.4496, + "theoretical_loss": 3.7747543869880174, + "tokens_seen": 708640768 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039662988966900704, + "loss": 1.4847, + "theoretical_loss": 3.7747190205346577, + "tokens_seen": 708706304 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039661985957873617, + "loss": 1.4164, + "theoretical_loss": 3.7746836582672034, + "tokens_seen": 708771840 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003966098294884654, + "loss": 1.3733, + "theoretical_loss": 3.774648300184772, + "tokens_seen": 708837376 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003965997993981946, + "loss": 1.3914, + "theoretical_loss": 3.7746129462864824, + "tokens_seen": 708902912 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039658976930792377, + "loss": 1.4146, + "theoretical_loss": 3.7745775965714508, + "tokens_seen": 708968448 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039657973921765295, + "loss": 1.339, + "theoretical_loss": 3.774542251038797, + "tokens_seen": 709033984 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003965697091273822, + "loss": 1.3527, + "theoretical_loss": 3.7745069096876396, + "tokens_seen": 709099520 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003965596790371113, + "loss": 1.5139, + "theoretical_loss": 3.7744715725170965, + "tokens_seen": 709165056 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039654964894684055, + "loss": 1.3849, + "theoretical_loss": 3.7744362395262883, + "tokens_seen": 709230592 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003965396188565697, + "loss": 1.4361, + "theoretical_loss": 3.774400910714333, + "tokens_seen": 709296128 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003965295887662989, + "loss": 1.4104, + "theoretical_loss": 3.7743655860803513, + "tokens_seen": 709361664 + }, + { + "epoch": 0.21, + "objective/train/advantage_avg": 0.00012102701293770224, + "objective/train/docs_used": 410190, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7830140590667725, + "objective/train/original_loss": 2.7830140590667725, + "objective/train/theoretical_loss": 3.7743302656234627, + "objective/train/tokens_used": 729887200, + "objective/train/value_avg": -0.0054779052734375, + "objective/train/value_loss": 5.2099789172643796e-05, + "objective/train/value_max": -0.0004425048828125, + "objective/train/value_min": -0.05340576171875, + "objective/train/value_reward_corr": 0.4207512884409758, + "objective/train/value_std": 0.00417327880859375, + "objective/train/weight_avg": 1.0037100315093994, + "objective/train/weighted_lm_loss": 2.7912328243255615, + "objective/train/weights_max": 1.6300104856491089, + "objective/train/weights_min": 0.5474342107772827, + "theoretical_loss": 3.7743302656234627, + "tokens_seen": 709427200 + }, + { + "epoch": 0.21, + "learning_rate": 0.0003965195586760281, + "loss": 1.4238, + "theoretical_loss": 3.7743302656234627, + "tokens_seen": 709427200 + }, + { + "epoch": 0.21, + "learning_rate": 0.00039650952858575727, + "loss": 1.5012, + "theoretical_loss": 3.774294949342788, + "tokens_seen": 709492736 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039649949849548645, + "loss": 1.3672, + "theoretical_loss": 3.7742596372374475, + "tokens_seen": 709558272 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039648946840521563, + "loss": 1.427, + "theoretical_loss": 3.774224329306562, + "tokens_seen": 709623808 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003964794383149448, + "loss": 1.4492, + "theoretical_loss": 3.774189025549252, + "tokens_seen": 709689344 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039646940822467405, + "loss": 1.4391, + "theoretical_loss": 3.7741537259646396, + "tokens_seen": 709754880 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003964593781344032, + "loss": 1.4867, + "theoretical_loss": 3.7741184305518454, + "tokens_seen": 709820416 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003964493480441324, + "loss": 1.3778, + "theoretical_loss": 3.774083139309993, + "tokens_seen": 709885952 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003964393179538616, + "loss": 1.3795, + "theoretical_loss": 3.774047852238203, + "tokens_seen": 709951488 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003964292878635908, + "loss": 1.5779, + "theoretical_loss": 3.774012569335598, + "tokens_seen": 710017024 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039641925777331996, + "loss": 1.4009, + "theoretical_loss": 3.7739772906013007, + "tokens_seen": 710082560 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039640922768304914, + "loss": 1.2874, + "theoretical_loss": 3.773942016034435, + "tokens_seen": 710148096 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003963991975927783, + "loss": 1.524, + "theoretical_loss": 3.7739067456341227, + "tokens_seen": 710213632 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039638916750250755, + "loss": 1.5525, + "theoretical_loss": 3.773871479399488, + "tokens_seen": 710279168 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039637913741223673, + "loss": 1.3692, + "theoretical_loss": 3.7738362173296545, + "tokens_seen": 710344704 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003963691073219659, + "loss": 1.5174, + "theoretical_loss": 3.7738009594237463, + "tokens_seen": 710410240 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003963590772316951, + "loss": 1.4304, + "theoretical_loss": 3.773765705680887, + "tokens_seen": 710475776 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003963490471414243, + "loss": 1.5137, + "theoretical_loss": 3.7737304561002016, + "tokens_seen": 710541312 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003963390170511535, + "loss": 1.3623, + "theoretical_loss": 3.7736952106808155, + "tokens_seen": 710606848 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039632898696088264, + "loss": 1.4874, + "theoretical_loss": 3.7736599694218524, + "tokens_seen": 710672384 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003963189568706119, + "loss": 1.2975, + "theoretical_loss": 3.773624732322439, + "tokens_seen": 710737920 + }, + { + "epoch": 0.22, + "learning_rate": 0.000396308926780341, + "loss": 1.551, + "theoretical_loss": 3.7735894993817, + "tokens_seen": 710803456 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039629889669007024, + "loss": 1.4304, + "theoretical_loss": 3.773554270598761, + "tokens_seen": 710868992 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003962888665997994, + "loss": 1.4288, + "theoretical_loss": 3.7735190459727486, + "tokens_seen": 710934528 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003962788365095286, + "loss": 1.3551, + "theoretical_loss": 3.7734838255027894, + "tokens_seen": 711000064 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0015487928176298738, + "objective/train/docs_used": 411414, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.732069492340088, + "objective/train/original_loss": 2.732069492340088, + "objective/train/theoretical_loss": 3.7734486091880095, + "objective/train/tokens_used": 731525600, + "objective/train/value_avg": -0.005321502685546875, + "objective/train/value_loss": 0.0001087427226593718, + "objective/train/value_max": -0.00045299530029296875, + "objective/train/value_min": -0.051849365234375, + "objective/train/value_reward_corr": 0.4461286072337971, + "objective/train/value_std": 0.004627227783203125, + "objective/train/weight_avg": 0.989113450050354, + "objective/train/weighted_lm_loss": 2.7105815410614014, + "objective/train/weights_max": 1.528296947479248, + "objective/train/weights_min": 0.15332689881324768, + "theoretical_loss": 3.7734486091880095, + "tokens_seen": 711065600 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003962688064192578, + "loss": 1.5048, + "theoretical_loss": 3.7734486091880095, + "tokens_seen": 711065600 + }, + { + "epoch": 0.22, + "learning_rate": 0.000396258776328987, + "loss": 1.4236, + "theoretical_loss": 3.773413397027536, + "tokens_seen": 711131136 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039624874623871614, + "loss": 1.573, + "theoretical_loss": 3.773378189020496, + "tokens_seen": 711196672 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003962387161484454, + "loss": 1.4044, + "theoretical_loss": 3.7733429851660167, + "tokens_seen": 711262208 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003962286860581745, + "loss": 1.4171, + "theoretical_loss": 3.773307785463226, + "tokens_seen": 711327744 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039621865596790374, + "loss": 1.3992, + "theoretical_loss": 3.7732725899112527, + "tokens_seen": 711393280 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003962086258776329, + "loss": 1.3705, + "theoretical_loss": 3.773237398509223, + "tokens_seen": 711458816 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003961985957873621, + "loss": 1.4021, + "theoretical_loss": 3.773202211256267, + "tokens_seen": 711524352 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003961885656970913, + "loss": 1.4402, + "theoretical_loss": 3.773167028151514, + "tokens_seen": 711589888 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039617853560682047, + "loss": 1.3597, + "theoretical_loss": 3.773131849194091, + "tokens_seen": 711655424 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039616850551654965, + "loss": 1.3621, + "theoretical_loss": 3.7730966743831282, + "tokens_seen": 711720960 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003961584754262789, + "loss": 1.4675, + "theoretical_loss": 3.7730615037177557, + "tokens_seen": 711786496 + }, + { + "epoch": 0.22, + "learning_rate": 0.000396148445336008, + "loss": 1.4942, + "theoretical_loss": 3.773026337197102, + "tokens_seen": 711852032 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039613841524573724, + "loss": 1.5107, + "theoretical_loss": 3.7729911748202984, + "tokens_seen": 711917568 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039612838515546637, + "loss": 1.3955, + "theoretical_loss": 3.7729560165864746, + "tokens_seen": 711983104 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003961183550651956, + "loss": 1.4795, + "theoretical_loss": 3.7729208624947614, + "tokens_seen": 712048640 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003961083249749248, + "loss": 1.4232, + "theoretical_loss": 3.7728857125442894, + "tokens_seen": 712114176 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039609829488465397, + "loss": 1.4011, + "theoretical_loss": 3.77285056673419, + "tokens_seen": 712179712 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039608826479438315, + "loss": 1.4361, + "theoretical_loss": 3.772815425063594, + "tokens_seen": 712245248 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003960782347041124, + "loss": 1.484, + "theoretical_loss": 3.772780287531634, + "tokens_seen": 712310784 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003960682046138415, + "loss": 1.4615, + "theoretical_loss": 3.772745154137441, + "tokens_seen": 712376320 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039605817452357075, + "loss": 1.5249, + "theoretical_loss": 3.7727100248801473, + "tokens_seen": 712441856 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003960481444332999, + "loss": 1.3733, + "theoretical_loss": 3.7726748997588855, + "tokens_seen": 712507392 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003960381143430291, + "loss": 1.5553, + "theoretical_loss": 3.7726397787727883, + "tokens_seen": 712572928 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003960280842527583, + "loss": 1.4823, + "theoretical_loss": 3.772604661920989, + "tokens_seen": 712638464 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": 8.77750208019279e-05, + "objective/train/docs_used": 412148, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8942267894744873, + "objective/train/original_loss": 2.8942267894744873, + "objective/train/theoretical_loss": 3.77256954920262, + "objective/train/tokens_used": 733164000, + "objective/train/value_avg": -0.005207061767578125, + "objective/train/value_loss": 3.9809965528547764e-05, + "objective/train/value_max": -0.0003249645233154297, + "objective/train/value_min": -0.06658935546875, + "objective/train/value_reward_corr": 0.5092445404282959, + "objective/train/value_std": 0.004665374755859375, + "objective/train/weight_avg": 1.0028184652328491, + "objective/train/weighted_lm_loss": 2.9064576625823975, + "objective/train/weights_max": 1.5400114059448242, + "objective/train/weights_min": 0.5033369064331055, + "theoretical_loss": 3.77256954920262, + "tokens_seen": 712704000 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039601805416248747, + "loss": 1.4334, + "theoretical_loss": 3.77256954920262, + "tokens_seen": 712704000 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039600802407221665, + "loss": 1.4477, + "theoretical_loss": 3.772534440616815, + "tokens_seen": 712769536 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039599799398194583, + "loss": 1.4573, + "theoretical_loss": 3.7724993361627077, + "tokens_seen": 712835072 + }, + { + "epoch": 0.22, + "learning_rate": 0.000395987963891675, + "loss": 1.4749, + "theoretical_loss": 3.7724642358394322, + "tokens_seen": 712900608 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039597793380140425, + "loss": 1.4648, + "theoretical_loss": 3.772429139646123, + "tokens_seen": 712966144 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003959679037111334, + "loss": 1.4956, + "theoretical_loss": 3.7723940475819147, + "tokens_seen": 713031680 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003959578736208626, + "loss": 1.5448, + "theoretical_loss": 3.772358959645941, + "tokens_seen": 713097216 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003959478435305918, + "loss": 1.376, + "theoretical_loss": 3.772323875837338, + "tokens_seen": 713162752 + }, + { + "epoch": 0.22, + "learning_rate": 0.000395937813440321, + "loss": 1.3854, + "theoretical_loss": 3.7722887961552405, + "tokens_seen": 713228288 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039592778335005016, + "loss": 1.3987, + "theoretical_loss": 3.7722537205987843, + "tokens_seen": 713293824 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039591775325977934, + "loss": 1.4637, + "theoretical_loss": 3.772218649167105, + "tokens_seen": 713359360 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003959077231695085, + "loss": 1.4822, + "theoretical_loss": 3.7721835818593386, + "tokens_seen": 713424896 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039589769307923775, + "loss": 1.4229, + "theoretical_loss": 3.772148518674622, + "tokens_seen": 713490432 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003958876629889669, + "loss": 1.421, + "theoretical_loss": 3.7721134596120907, + "tokens_seen": 713555968 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003958776328986961, + "loss": 1.4498, + "theoretical_loss": 3.7720784046708826, + "tokens_seen": 713621504 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039586760280842524, + "loss": 1.4679, + "theoretical_loss": 3.772043353850134, + "tokens_seen": 713687040 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003958575727181545, + "loss": 1.3944, + "theoretical_loss": 3.772008307148983, + "tokens_seen": 713752576 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039584754262788366, + "loss": 1.4472, + "theoretical_loss": 3.771973264566567, + "tokens_seen": 713818112 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039583751253761284, + "loss": 1.3919, + "theoretical_loss": 3.7719382261020233, + "tokens_seen": 713883648 + }, + { + "epoch": 0.22, + "learning_rate": 0.000395827482447342, + "loss": 1.5312, + "theoretical_loss": 3.771903191754491, + "tokens_seen": 713949184 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003958174523570712, + "loss": 1.4442, + "theoretical_loss": 3.7718681615231078, + "tokens_seen": 714014720 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003958074222668004, + "loss": 1.411, + "theoretical_loss": 3.7718331354070127, + "tokens_seen": 714080256 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003957973921765296, + "loss": 1.4478, + "theoretical_loss": 3.771798113405344, + "tokens_seen": 714145792 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039578736208625875, + "loss": 1.3945, + "theoretical_loss": 3.771763095517242, + "tokens_seen": 714211328 + }, + { + "epoch": 0.22, + "learning_rate": 0.000395777331995988, + "loss": 1.4374, + "theoretical_loss": 3.771728081741845, + "tokens_seen": 714276864 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.00034032543771900237, + "objective/train/docs_used": 413348, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3562114238739014, + "objective/train/original_loss": 2.3562114238739014, + "objective/train/theoretical_loss": 3.7716930720782935, + "objective/train/tokens_used": 734802400, + "objective/train/value_avg": -0.004364013671875, + "objective/train/value_loss": 3.755267971428111e-05, + "objective/train/value_max": -0.0003249645233154297, + "objective/train/value_min": -0.08416748046875, + "objective/train/value_reward_corr": 0.4844795269454044, + "objective/train/value_std": 0.00438690185546875, + "objective/train/weight_avg": 0.9983370304107666, + "objective/train/weighted_lm_loss": 2.363457441329956, + "objective/train/weights_max": 1.9408061504364014, + "objective/train/weights_min": 0.2974976599216461, + "theoretical_loss": 3.7716930720782935, + "tokens_seen": 714342400 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039576730190571716, + "loss": 1.3249, + "theoretical_loss": 3.7716930720782935, + "tokens_seen": 714342400 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039575727181544634, + "loss": 1.5128, + "theoretical_loss": 3.7716580665257267, + "tokens_seen": 714407936 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003957472417251755, + "loss": 1.4489, + "theoretical_loss": 3.7716230650832863, + "tokens_seen": 714473472 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003957372116349047, + "loss": 1.3513, + "theoretical_loss": 3.771588067750111, + "tokens_seen": 714539008 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003957271815446339, + "loss": 1.3697, + "theoretical_loss": 3.7715530745253423, + "tokens_seen": 714604544 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003957171514543631, + "loss": 1.4301, + "theoretical_loss": 3.7715180854081214, + "tokens_seen": 714670080 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039570712136409225, + "loss": 1.4774, + "theoretical_loss": 3.771483100397589, + "tokens_seen": 714735616 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003956970912738215, + "loss": 1.3924, + "theoretical_loss": 3.7714481194928875, + "tokens_seen": 714801152 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003956870611835506, + "loss": 1.4692, + "theoretical_loss": 3.7714131426931576, + "tokens_seen": 714866688 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039567703109327985, + "loss": 1.2952, + "theoretical_loss": 3.771378169997542, + "tokens_seen": 714932224 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039566700100300903, + "loss": 1.5191, + "theoretical_loss": 3.771343201405183, + "tokens_seen": 714997760 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003956569709127382, + "loss": 1.3602, + "theoretical_loss": 3.771308236915223, + "tokens_seen": 715063296 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003956469408224674, + "loss": 1.4988, + "theoretical_loss": 3.771273276526805, + "tokens_seen": 715128832 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039563691073219657, + "loss": 1.4824, + "theoretical_loss": 3.771238320239071, + "tokens_seen": 715194368 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003956268806419258, + "loss": 1.4097, + "theoretical_loss": 3.7712033680511654, + "tokens_seen": 715259904 + }, + { + "epoch": 0.22, + "learning_rate": 0.000395616850551655, + "loss": 1.4669, + "theoretical_loss": 3.7711684199622324, + "tokens_seen": 715325440 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039560682046138417, + "loss": 1.4652, + "theoretical_loss": 3.771133475971414, + "tokens_seen": 715390976 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039559679037111335, + "loss": 1.3975, + "theoretical_loss": 3.7710985360778553, + "tokens_seen": 715456512 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003955867602808426, + "loss": 1.4486, + "theoretical_loss": 3.771063600280701, + "tokens_seen": 715522048 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003955767301905717, + "loss": 1.4112, + "theoretical_loss": 3.7710286685790946, + "tokens_seen": 715587584 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039556670010030095, + "loss": 1.5097, + "theoretical_loss": 3.770993740972182, + "tokens_seen": 715653120 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003955566700100301, + "loss": 1.5173, + "theoretical_loss": 3.7709588174591087, + "tokens_seen": 715718656 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003955466399197593, + "loss": 1.3154, + "theoretical_loss": 3.770923898039018, + "tokens_seen": 715784192 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003955366098294885, + "loss": 1.4225, + "theoretical_loss": 3.7708889827110577, + "tokens_seen": 715849728 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039552657973921767, + "loss": 1.3934, + "theoretical_loss": 3.770854071474372, + "tokens_seen": 715915264 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0011027351720258594, + "objective/train/docs_used": 413910, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6835694313049316, + "objective/train/original_loss": 2.6835691928863525, + "objective/train/theoretical_loss": 3.770819164328109, + "objective/train/tokens_used": 736440800, + "objective/train/value_avg": -0.00522613525390625, + "objective/train/value_loss": 7.152266334742308e-05, + "objective/train/value_max": -0.0004546642303466797, + "objective/train/value_min": -0.056549072265625, + "objective/train/value_reward_corr": 0.27525110007278225, + "objective/train/value_std": 0.0038738250732421875, + "objective/train/weight_avg": 0.9922711253166199, + "objective/train/weighted_lm_loss": 2.6811962127685547, + "objective/train/weights_max": 1.4671193361282349, + "objective/train/weights_min": 0.1971050649881363, + "theoretical_loss": 3.770819164328109, + "tokens_seen": 715980800 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039551654964894685, + "loss": 1.4688, + "theoretical_loss": 3.770819164328109, + "tokens_seen": 715980800 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039550651955867603, + "loss": 1.4658, + "theoretical_loss": 3.7707842612714133, + "tokens_seen": 716046336 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003954964894684052, + "loss": 1.3211, + "theoretical_loss": 3.770749362303432, + "tokens_seen": 716111872 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039548645937813445, + "loss": 1.4568, + "theoretical_loss": 3.770714467423313, + "tokens_seen": 716177408 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003954764292878636, + "loss": 1.3797, + "theoretical_loss": 3.7706795766302017, + "tokens_seen": 716242944 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003954663991975928, + "loss": 1.3683, + "theoretical_loss": 3.7706446899232473, + "tokens_seen": 716308480 + }, + { + "epoch": 0.22, + "learning_rate": 0.000395456369107322, + "loss": 1.5948, + "theoretical_loss": 3.7706098073015966, + "tokens_seen": 716374016 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003954463390170512, + "loss": 1.335, + "theoretical_loss": 3.7705749287643973, + "tokens_seen": 716439552 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039543630892678036, + "loss": 1.4586, + "theoretical_loss": 3.770540054310798, + "tokens_seen": 716505088 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039542627883650954, + "loss": 1.6003, + "theoretical_loss": 3.770505183939947, + "tokens_seen": 716570624 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003954162487462387, + "loss": 1.3787, + "theoretical_loss": 3.770470317650993, + "tokens_seen": 716636160 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039540621865596795, + "loss": 1.3409, + "theoretical_loss": 3.770435455443085, + "tokens_seen": 716701696 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003953961885656971, + "loss": 1.4425, + "theoretical_loss": 3.7704005973153722, + "tokens_seen": 716767232 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003953861584754263, + "loss": 1.4916, + "theoretical_loss": 3.7703657432670044, + "tokens_seen": 716832768 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039537612838515544, + "loss": 1.3228, + "theoretical_loss": 3.770330893297131, + "tokens_seen": 716898304 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003953660982948847, + "loss": 1.3952, + "theoretical_loss": 3.7702960474049014, + "tokens_seen": 716963840 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039535606820461386, + "loss": 1.6947, + "theoretical_loss": 3.770261205589467, + "tokens_seen": 717029376 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039534603811434304, + "loss": 1.4548, + "theoretical_loss": 3.7702263678499772, + "tokens_seen": 717094912 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003953360080240722, + "loss": 1.3732, + "theoretical_loss": 3.7701915341855834, + "tokens_seen": 717160448 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003953259779338014, + "loss": 1.4656, + "theoretical_loss": 3.7701567045954367, + "tokens_seen": 717225984 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003953159478435306, + "loss": 1.4485, + "theoretical_loss": 3.7701218790786877, + "tokens_seen": 717291520 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003953059177532598, + "loss": 1.2789, + "theoretical_loss": 3.7700870576344885, + "tokens_seen": 717357056 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039529588766298895, + "loss": 1.4756, + "theoretical_loss": 3.7700522402619905, + "tokens_seen": 717422592 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003952858575727182, + "loss": 1.4874, + "theoretical_loss": 3.7700174269603464, + "tokens_seen": 717488128 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039527582748244736, + "loss": 1.4726, + "theoretical_loss": 3.769982617728707, + "tokens_seen": 717553664 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.00046009148354642093, + "objective/train/docs_used": 414484, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.593858480453491, + "objective/train/original_loss": 2.5938591957092285, + "objective/train/theoretical_loss": 3.769947812566226, + "objective/train/tokens_used": 738079200, + "objective/train/value_avg": -0.005779266357421875, + "objective/train/value_loss": 6.590461998712271e-05, + "objective/train/value_max": -0.0003905296325683594, + "objective/train/value_min": -0.06793212890625, + "objective/train/value_reward_corr": 0.5424954112536489, + "objective/train/value_std": 0.0055999755859375, + "objective/train/weight_avg": 0.9984509348869324, + "objective/train/weighted_lm_loss": 2.5886809825897217, + "objective/train/weights_max": 1.6294434070587158, + "objective/train/weights_min": 0.2673785984516144, + "theoretical_loss": 3.769947812566226, + "tokens_seen": 717619200 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039526579739217654, + "loss": 1.3274, + "theoretical_loss": 3.769947812566226, + "tokens_seen": 717619200 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003952557673019057, + "loss": 1.4098, + "theoretical_loss": 3.769913011472056, + "tokens_seen": 717684736 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003952457372116349, + "loss": 1.283, + "theoretical_loss": 3.76987821444535, + "tokens_seen": 717750272 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003952357071213641, + "loss": 1.3989, + "theoretical_loss": 3.769843421485261, + "tokens_seen": 717815808 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003952256770310933, + "loss": 1.3046, + "theoretical_loss": 3.7698086325909435, + "tokens_seen": 717881344 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039521564694082245, + "loss": 1.4265, + "theoretical_loss": 3.7697738477615497, + "tokens_seen": 717946880 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003952056168505517, + "loss": 1.5235, + "theoretical_loss": 3.7697390669962347, + "tokens_seen": 718012416 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003951955867602808, + "loss": 1.4853, + "theoretical_loss": 3.769704290294152, + "tokens_seen": 718077952 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039518555667001005, + "loss": 1.3101, + "theoretical_loss": 3.7696695176544575, + "tokens_seen": 718143488 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039517552657973923, + "loss": 1.3501, + "theoretical_loss": 3.769634749076305, + "tokens_seen": 718209024 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003951654964894684, + "loss": 1.3841, + "theoretical_loss": 3.76959998455885, + "tokens_seen": 718274560 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003951554663991976, + "loss": 1.3159, + "theoretical_loss": 3.769565224101247, + "tokens_seen": 718340096 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039514543630892677, + "loss": 1.3575, + "theoretical_loss": 3.769530467702652, + "tokens_seen": 718405632 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039513540621865595, + "loss": 1.5424, + "theoretical_loss": 3.769495715362222, + "tokens_seen": 718471168 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003951253761283852, + "loss": 1.3626, + "theoretical_loss": 3.769460967079111, + "tokens_seen": 718536704 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003951153460381143, + "loss": 1.4919, + "theoretical_loss": 3.769426222852477, + "tokens_seen": 718602240 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039510531594784355, + "loss": 1.4006, + "theoretical_loss": 3.769391482681475, + "tokens_seen": 718667776 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039509528585757273, + "loss": 1.4148, + "theoretical_loss": 3.769356746565263, + "tokens_seen": 718733312 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003950852557673019, + "loss": 1.4611, + "theoretical_loss": 3.7693220145029986, + "tokens_seen": 718798848 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003950752256770311, + "loss": 1.3486, + "theoretical_loss": 3.7692872864938374, + "tokens_seen": 718864384 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003950651955867603, + "loss": 1.3747, + "theoretical_loss": 3.769252562536938, + "tokens_seen": 718929920 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039505516549648946, + "loss": 1.4352, + "theoretical_loss": 3.769217842631458, + "tokens_seen": 718995456 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003950451354062187, + "loss": 1.4361, + "theoretical_loss": 3.7691831267765563, + "tokens_seen": 719060992 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003950351053159478, + "loss": 1.527, + "theoretical_loss": 3.76914841497139, + "tokens_seen": 719126528 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039502507522567705, + "loss": 1.3473, + "theoretical_loss": 3.7691137072151184, + "tokens_seen": 719192064 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0008880366804078221, + "objective/train/docs_used": 415542, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0178205966949463, + "objective/train/original_loss": 3.017820119857788, + "objective/train/theoretical_loss": 3.7690790035069, + "objective/train/tokens_used": 739717600, + "objective/train/value_avg": -0.005401611328125, + "objective/train/value_loss": 5.410028461483307e-05, + "objective/train/value_max": -0.00023233890533447266, + "objective/train/value_min": -0.043609619140625, + "objective/train/value_reward_corr": 0.44803282708622627, + "objective/train/value_std": 0.00467681884765625, + "objective/train/weight_avg": 0.9936684370040894, + "objective/train/weighted_lm_loss": 3.005601644515991, + "objective/train/weights_max": 1.5181353092193604, + "objective/train/weights_min": 0.33955585956573486, + "theoretical_loss": 3.7690790035069, + "tokens_seen": 719257600 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003950150451354062, + "loss": 1.4772, + "theoretical_loss": 3.7690790035069, + "tokens_seen": 719257600 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003950050150451354, + "loss": 1.5151, + "theoretical_loss": 3.7690443038458943, + "tokens_seen": 719323136 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003949949849548646, + "loss": 1.4098, + "theoretical_loss": 3.7690096082312596, + "tokens_seen": 719388672 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003949849548645938, + "loss": 1.4521, + "theoretical_loss": 3.768974916662157, + "tokens_seen": 719454208 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039497492477432296, + "loss": 1.4605, + "theoretical_loss": 3.768940229137745, + "tokens_seen": 719519744 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003949648946840522, + "loss": 1.4351, + "theoretical_loss": 3.768905545657185, + "tokens_seen": 719585280 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003949548645937813, + "loss": 1.3684, + "theoretical_loss": 3.768870866219636, + "tokens_seen": 719650816 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039494483450351056, + "loss": 1.3638, + "theoretical_loss": 3.76883619082426, + "tokens_seen": 719716352 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003949348044132397, + "loss": 1.4737, + "theoretical_loss": 3.768801519470216, + "tokens_seen": 719781888 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003949247743229689, + "loss": 1.5049, + "theoretical_loss": 3.7687668521566664, + "tokens_seen": 719847424 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003949147442326981, + "loss": 1.4619, + "theoretical_loss": 3.768732188882773, + "tokens_seen": 719912960 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003949047141424273, + "loss": 1.4175, + "theoretical_loss": 3.768697529647696, + "tokens_seen": 719978496 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003948946840521565, + "loss": 1.3883, + "theoretical_loss": 3.768662874450598, + "tokens_seen": 720044032 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039488465396188564, + "loss": 1.4476, + "theoretical_loss": 3.7686282232906416, + "tokens_seen": 720109568 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003948746238716149, + "loss": 1.3808, + "theoretical_loss": 3.768593576166988, + "tokens_seen": 720175104 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039486459378134406, + "loss": 1.454, + "theoretical_loss": 3.7685589330788005, + "tokens_seen": 720240640 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039485456369107324, + "loss": 1.3666, + "theoretical_loss": 3.768524294025242, + "tokens_seen": 720306176 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003948445336008024, + "loss": 1.4761, + "theoretical_loss": 3.7684896590054757, + "tokens_seen": 720371712 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003948345035105316, + "loss": 1.364, + "theoretical_loss": 3.768455028018664, + "tokens_seen": 720437248 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003948244734202608, + "loss": 1.3404, + "theoretical_loss": 3.7684204010639717, + "tokens_seen": 720502784 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039481444332999, + "loss": 1.438, + "theoretical_loss": 3.768385778140562, + "tokens_seen": 720568320 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039480441323971915, + "loss": 1.5521, + "theoretical_loss": 3.768351159247599, + "tokens_seen": 720633856 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003947943831494484, + "loss": 1.4871, + "theoretical_loss": 3.768316544384247, + "tokens_seen": 720699392 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039478435305917756, + "loss": 1.5212, + "theoretical_loss": 3.768281933549671, + "tokens_seen": 720764928 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039477432296890674, + "loss": 1.4623, + "theoretical_loss": 3.7682473267430354, + "tokens_seen": 720830464 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0002611952368170023, + "objective/train/docs_used": 416165, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.921515703201294, + "objective/train/original_loss": 2.921515941619873, + "objective/train/theoretical_loss": 3.7682127239635053, + "objective/train/tokens_used": 741356000, + "objective/train/value_avg": -0.005306243896484375, + "objective/train/value_loss": 4.886410170001909e-05, + "objective/train/value_max": -0.0004322528839111328, + "objective/train/value_min": -0.0701904296875, + "objective/train/value_reward_corr": 0.5817178585888143, + "objective/train/value_std": 0.004974365234375, + "objective/train/weight_avg": 0.9997338652610779, + "objective/train/weighted_lm_loss": 2.9155681133270264, + "objective/train/weights_max": 2.0175912380218506, + "objective/train/weights_min": 0.5441913604736328, + "theoretical_loss": 3.7682127239635053, + "tokens_seen": 720896000 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003947642928786359, + "loss": 1.4278, + "theoretical_loss": 3.7682127239635053, + "tokens_seen": 720896000 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003947542627883651, + "loss": 1.4884, + "theoretical_loss": 3.7681781252102464, + "tokens_seen": 720961536 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003947442326980943, + "loss": 1.3318, + "theoretical_loss": 3.768143530482424, + "tokens_seen": 721027072 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003947342026078235, + "loss": 1.4336, + "theoretical_loss": 3.7681089397792045, + "tokens_seen": 721092608 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039472417251755265, + "loss": 1.6182, + "theoretical_loss": 3.7680743530997525, + "tokens_seen": 721158144 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003947141424272819, + "loss": 1.354, + "theoretical_loss": 3.7680397704432362, + "tokens_seen": 721223680 + }, + { + "epoch": 0.22, + "learning_rate": 0.000394704112337011, + "loss": 1.3655, + "theoretical_loss": 3.768005191808821, + "tokens_seen": 721289216 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039469408224674025, + "loss": 1.4882, + "theoretical_loss": 3.767970617195674, + "tokens_seen": 721354752 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039468405215646943, + "loss": 1.524, + "theoretical_loss": 3.767936046602963, + "tokens_seen": 721420288 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003946740220661986, + "loss": 1.4109, + "theoretical_loss": 3.7679014800298543, + "tokens_seen": 721485824 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003946639919759278, + "loss": 1.5235, + "theoretical_loss": 3.7678669174755153, + "tokens_seen": 721551360 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039465396188565697, + "loss": 1.4458, + "theoretical_loss": 3.767832358939115, + "tokens_seen": 721616896 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039464393179538615, + "loss": 1.4545, + "theoretical_loss": 3.7677978044198204, + "tokens_seen": 721682432 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003946339017051154, + "loss": 1.5207, + "theoretical_loss": 3.7677632539168004, + "tokens_seen": 721747968 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003946238716148445, + "loss": 1.4488, + "theoretical_loss": 3.7677287074292236, + "tokens_seen": 721813504 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039461384152457375, + "loss": 1.4548, + "theoretical_loss": 3.7676941649562585, + "tokens_seen": 721879040 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039460381143430293, + "loss": 1.426, + "theoretical_loss": 3.7676596264970743, + "tokens_seen": 721944576 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003945937813440321, + "loss": 1.324, + "theoretical_loss": 3.7676250920508405, + "tokens_seen": 722010112 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003945837512537613, + "loss": 1.4485, + "theoretical_loss": 3.7675905616167262, + "tokens_seen": 722075648 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003945737211634905, + "loss": 1.4102, + "theoretical_loss": 3.7675560351939015, + "tokens_seen": 722141184 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039456369107321966, + "loss": 1.4565, + "theoretical_loss": 3.7675215127815367, + "tokens_seen": 722206720 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003945536609829489, + "loss": 1.4244, + "theoretical_loss": 3.7674869943788014, + "tokens_seen": 722272256 + }, + { + "epoch": 0.22, + "learning_rate": 0.000394543630892678, + "loss": 1.4264, + "theoretical_loss": 3.767452479984867, + "tokens_seen": 722337792 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039453360080240725, + "loss": 1.4119, + "theoretical_loss": 3.7674179695989034, + "tokens_seen": 722403328 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003945235707121364, + "loss": 1.3922, + "theoretical_loss": 3.7673834632200824, + "tokens_seen": 722468864 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0009570175316184759, + "objective/train/docs_used": 417606, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2690956592559814, + "objective/train/original_loss": 3.2690958976745605, + "objective/train/theoretical_loss": 3.767348960847575, + "objective/train/tokens_used": 742994400, + "objective/train/value_avg": -0.006336212158203125, + "objective/train/value_loss": 6.878429849166423e-05, + "objective/train/value_max": -0.00038909912109375, + "objective/train/value_min": -0.09771728515625, + "objective/train/value_reward_corr": 0.5613554607427923, + "objective/train/value_std": 0.005756378173828125, + "objective/train/weight_avg": 0.9936969876289368, + "objective/train/weighted_lm_loss": 3.248875856399536, + "objective/train/weights_max": 2.045689582824707, + "objective/train/weights_min": 0.5132936239242554, + "theoretical_loss": 3.767348960847575, + "tokens_seen": 722534400 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003945135406218656, + "loss": 1.4443, + "theoretical_loss": 3.767348960847575, + "tokens_seen": 722534400 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003945035105315948, + "loss": 1.3469, + "theoretical_loss": 3.7673144624805524, + "tokens_seen": 722599936 + }, + { + "epoch": 0.22, + "learning_rate": 0.000394493480441324, + "loss": 1.3472, + "theoretical_loss": 3.767279968118187, + "tokens_seen": 722665472 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039448345035105316, + "loss": 1.3959, + "theoretical_loss": 3.7672454777596505, + "tokens_seen": 722731008 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003944734202607824, + "loss": 1.355, + "theoretical_loss": 3.7672109914041156, + "tokens_seen": 722796544 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003944633901705115, + "loss": 1.4118, + "theoretical_loss": 3.7671765090507536, + "tokens_seen": 722862080 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039445336008024076, + "loss": 1.4101, + "theoretical_loss": 3.7671420306987384, + "tokens_seen": 722927616 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003944433299899699, + "loss": 1.448, + "theoretical_loss": 3.7671075563472427, + "tokens_seen": 722993152 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003944332998996991, + "loss": 1.423, + "theoretical_loss": 3.76707308599544, + "tokens_seen": 723058688 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003944232698094283, + "loss": 1.453, + "theoretical_loss": 3.767038619642503, + "tokens_seen": 723124224 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003944132397191575, + "loss": 1.5978, + "theoretical_loss": 3.767004157287606, + "tokens_seen": 723189760 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039440320962888666, + "loss": 1.4565, + "theoretical_loss": 3.7669696989299233, + "tokens_seen": 723255296 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039439317953861584, + "loss": 1.539, + "theoretical_loss": 3.7669352445686286, + "tokens_seen": 723320832 + }, + { + "epoch": 0.22, + "learning_rate": 0.000394383149448345, + "loss": 1.4586, + "theoretical_loss": 3.7669007942028965, + "tokens_seen": 723386368 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039437311935807426, + "loss": 1.5002, + "theoretical_loss": 3.766866347831902, + "tokens_seen": 723451904 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003943630892678034, + "loss": 1.3339, + "theoretical_loss": 3.76683190545482, + "tokens_seen": 723517440 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003943530591775326, + "loss": 1.3, + "theoretical_loss": 3.766797467070825, + "tokens_seen": 723582976 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039434302908726175, + "loss": 1.4738, + "theoretical_loss": 3.766763032679094, + "tokens_seen": 723648512 + }, + { + "epoch": 0.22, + "learning_rate": 0.000394332998996991, + "loss": 1.4602, + "theoretical_loss": 3.7667286022788007, + "tokens_seen": 723714048 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039432296890672017, + "loss": 1.4042, + "theoretical_loss": 3.766694175869123, + "tokens_seen": 723779584 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039431293881644935, + "loss": 1.3101, + "theoretical_loss": 3.766659753449236, + "tokens_seen": 723845120 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039430290872617853, + "loss": 1.4935, + "theoretical_loss": 3.7666253350183165, + "tokens_seen": 723910656 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039429287863590776, + "loss": 1.5401, + "theoretical_loss": 3.7665909205755406, + "tokens_seen": 723976192 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003942828485456369, + "loss": 1.4063, + "theoretical_loss": 3.766556510120086, + "tokens_seen": 724041728 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003942728184553661, + "loss": 1.4867, + "theoretical_loss": 3.7665221036511296, + "tokens_seen": 724107264 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0011746912496164441, + "objective/train/docs_used": 418215, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9619202613830566, + "objective/train/original_loss": 2.9619202613830566, + "objective/train/theoretical_loss": 3.7664877011678484, + "objective/train/tokens_used": 744632800, + "objective/train/value_avg": -0.006206512451171875, + "objective/train/value_loss": 0.00010273342923028395, + "objective/train/value_max": -0.00046563148498535156, + "objective/train/value_min": -0.0836181640625, + "objective/train/value_reward_corr": 0.26718589912393587, + "objective/train/value_std": 0.004673004150390625, + "objective/train/weight_avg": 0.9929325580596924, + "objective/train/weighted_lm_loss": 2.945633888244629, + "objective/train/weights_max": 2.307539224624634, + "objective/train/weights_min": 0.14065396785736084, + "theoretical_loss": 3.7664877011678484, + "tokens_seen": 724172800 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039426278836509525, + "loss": 1.4814, + "theoretical_loss": 3.7664877011678484, + "tokens_seen": 724172800 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003942527582748245, + "loss": 1.4763, + "theoretical_loss": 3.7664533026694205, + "tokens_seen": 724238336 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039424272818455367, + "loss": 1.4169, + "theoretical_loss": 3.766418908155024, + "tokens_seen": 724303872 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039423269809428285, + "loss": 1.3809, + "theoretical_loss": 3.7663845176238366, + "tokens_seen": 724369408 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039422266800401203, + "loss": 1.3731, + "theoretical_loss": 3.766350131075037, + "tokens_seen": 724434944 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003942126379137412, + "loss": 1.3606, + "theoretical_loss": 3.7663157485078047, + "tokens_seen": 724500480 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003942026078234704, + "loss": 1.3531, + "theoretical_loss": 3.766281369921316, + "tokens_seen": 724566016 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039419257773319963, + "loss": 1.3993, + "theoretical_loss": 3.766246995314753, + "tokens_seen": 724631552 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039418254764292876, + "loss": 1.3768, + "theoretical_loss": 3.7662126246872925, + "tokens_seen": 724697088 + }, + { + "epoch": 0.22, + "learning_rate": 0.000394172517552658, + "loss": 1.4228, + "theoretical_loss": 3.7661782580381162, + "tokens_seen": 724762624 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003941624874623871, + "loss": 1.1836, + "theoretical_loss": 3.766143895366403, + "tokens_seen": 724828160 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039415245737211635, + "loss": 1.4036, + "theoretical_loss": 3.7661095366713324, + "tokens_seen": 724893696 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003941424272818456, + "loss": 1.3538, + "theoretical_loss": 3.766075181952086, + "tokens_seen": 724959232 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003941323971915747, + "loss": 1.4863, + "theoretical_loss": 3.766040831207844, + "tokens_seen": 725024768 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039412236710130395, + "loss": 1.4278, + "theoretical_loss": 3.7660064844377867, + "tokens_seen": 725090304 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039411233701103313, + "loss": 1.3849, + "theoretical_loss": 3.765972141641096, + "tokens_seen": 725155840 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003941023069207623, + "loss": 1.4839, + "theoretical_loss": 3.765937802816952, + "tokens_seen": 725221376 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003940922768304915, + "loss": 1.4206, + "theoretical_loss": 3.7659034679645376, + "tokens_seen": 725286912 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003940822467402207, + "loss": 1.442, + "theoretical_loss": 3.7658691370830337, + "tokens_seen": 725352448 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039407221664994986, + "loss": 1.3148, + "theoretical_loss": 3.7658348101716226, + "tokens_seen": 725417984 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003940621865596791, + "loss": 1.3253, + "theoretical_loss": 3.7658004872294866, + "tokens_seen": 725483520 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003940521564694082, + "loss": 1.2629, + "theoretical_loss": 3.7657661682558086, + "tokens_seen": 725549056 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039404212637913745, + "loss": 1.44, + "theoretical_loss": 3.765731853249771, + "tokens_seen": 725614592 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003940320962888666, + "loss": 1.4327, + "theoretical_loss": 3.7656975422105563, + "tokens_seen": 725680128 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003940220661985958, + "loss": 1.3403, + "theoretical_loss": 3.7656632351373487, + "tokens_seen": 725745664 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0013077526818960905, + "objective/train/docs_used": 419550, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1983745098114014, + "objective/train/original_loss": 3.1983742713928223, + "objective/train/theoretical_loss": 3.7656289320293315, + "objective/train/tokens_used": 746271200, + "objective/train/value_avg": -0.005321502685546875, + "objective/train/value_loss": 7.396401633741334e-05, + "objective/train/value_max": -0.0004513263702392578, + "objective/train/value_min": -0.051666259765625, + "objective/train/value_reward_corr": 0.36807163761106904, + "objective/train/value_std": 0.00405120849609375, + "objective/train/weight_avg": 0.9903941750526428, + "objective/train/weighted_lm_loss": 3.175907850265503, + "objective/train/weights_max": 1.6764233112335205, + "objective/train/weights_min": 0.4493493139743805, + "theoretical_loss": 3.7656289320293315, + "tokens_seen": 725811200 + }, + { + "epoch": 0.22, + "learning_rate": 0.000394012036108325, + "loss": 1.5252, + "theoretical_loss": 3.7656289320293315, + "tokens_seen": 725811200 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003940020060180542, + "loss": 1.4573, + "theoretical_loss": 3.7655946328856875, + "tokens_seen": 725876736 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039399197592778336, + "loss": 1.3547, + "theoretical_loss": 3.765560337705602, + "tokens_seen": 725942272 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003939819458375126, + "loss": 1.4393, + "theoretical_loss": 3.765526046488258, + "tokens_seen": 726007808 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003939719157472417, + "loss": 1.4429, + "theoretical_loss": 3.7654917592328414, + "tokens_seen": 726073344 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039396188565697096, + "loss": 1.5217, + "theoretical_loss": 3.7654574759385353, + "tokens_seen": 726138880 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003939518555667001, + "loss": 1.3556, + "theoretical_loss": 3.7654231966045257, + "tokens_seen": 726204416 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003939418254764293, + "loss": 1.5384, + "theoretical_loss": 3.765388921229998, + "tokens_seen": 726269952 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003939317953861585, + "loss": 1.4303, + "theoretical_loss": 3.7653546498141366, + "tokens_seen": 726335488 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003939217652958877, + "loss": 1.3103, + "theoretical_loss": 3.765320382356128, + "tokens_seen": 726401024 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039391173520561686, + "loss": 1.3705, + "theoretical_loss": 3.7652861188551574, + "tokens_seen": 726466560 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039390170511534604, + "loss": 1.368, + "theoretical_loss": 3.7652518593104114, + "tokens_seen": 726532096 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003938916750250752, + "loss": 1.4181, + "theoretical_loss": 3.765217603721077, + "tokens_seen": 726597632 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039388164493480446, + "loss": 1.4201, + "theoretical_loss": 3.7651833520863396, + "tokens_seen": 726663168 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003938716148445336, + "loss": 1.275, + "theoretical_loss": 3.7651491044053866, + "tokens_seen": 726728704 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003938615847542628, + "loss": 1.4077, + "theoretical_loss": 3.7651148606774054, + "tokens_seen": 726794240 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039385155466399195, + "loss": 1.5057, + "theoretical_loss": 3.7650806209015837, + "tokens_seen": 726859776 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003938415245737212, + "loss": 1.3478, + "theoretical_loss": 3.765046385077108, + "tokens_seen": 726925312 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039383149448345037, + "loss": 1.412, + "theoretical_loss": 3.765012153203166, + "tokens_seen": 726990848 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039382146439317955, + "loss": 1.3384, + "theoretical_loss": 3.764977925278947, + "tokens_seen": 727056384 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039381143430290873, + "loss": 1.3267, + "theoretical_loss": 3.764943701303639, + "tokens_seen": 727121920 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039380140421263796, + "loss": 1.4619, + "theoretical_loss": 3.7649094812764297, + "tokens_seen": 727187456 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003937913741223671, + "loss": 1.3213, + "theoretical_loss": 3.764875265196509, + "tokens_seen": 727252992 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003937813440320963, + "loss": 1.4955, + "theoretical_loss": 3.7648410530630656, + "tokens_seen": 727318528 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039377131394182545, + "loss": 1.2897, + "theoretical_loss": 3.764806844875288, + "tokens_seen": 727384064 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": 0.0008633521501906216, + "objective/train/docs_used": 420047, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.386106252670288, + "objective/train/original_loss": 2.386106014251709, + "objective/train/theoretical_loss": 3.7647726406323665, + "objective/train/tokens_used": 747909600, + "objective/train/value_avg": -0.00525665283203125, + "objective/train/value_loss": 4.9348094762535766e-05, + "objective/train/value_max": -0.0004093647003173828, + "objective/train/value_min": -0.0634765625, + "objective/train/value_reward_corr": 0.5067742894753223, + "objective/train/value_std": 0.004802703857421875, + "objective/train/weight_avg": 1.0110055208206177, + "objective/train/weighted_lm_loss": 2.4105145931243896, + "objective/train/weights_max": 1.4400534629821777, + "objective/train/weights_min": 0.43484586477279663, + "theoretical_loss": 3.7647726406323665, + "tokens_seen": 727449600 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003937612838515547, + "loss": 1.421, + "theoretical_loss": 3.7647726406323665, + "tokens_seen": 727449600 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039375125376128387, + "loss": 1.448, + "theoretical_loss": 3.764738440333491, + "tokens_seen": 727515136 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039374122367101305, + "loss": 1.3239, + "theoretical_loss": 3.7647042439778504, + "tokens_seen": 727580672 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039373119358074223, + "loss": 1.3073, + "theoretical_loss": 3.7646700515646363, + "tokens_seen": 727646208 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003937211634904714, + "loss": 1.4141, + "theoretical_loss": 3.7646358630930385, + "tokens_seen": 727711744 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003937111334002006, + "loss": 1.4829, + "theoretical_loss": 3.764601678562248, + "tokens_seen": 727777280 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039370110330992983, + "loss": 1.3318, + "theoretical_loss": 3.7645674979714547, + "tokens_seen": 727842816 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039369107321965896, + "loss": 1.4041, + "theoretical_loss": 3.764533321319851, + "tokens_seen": 727908352 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003936810431293882, + "loss": 1.4803, + "theoretical_loss": 3.764499148606628, + "tokens_seen": 727973888 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003936710130391173, + "loss": 1.416, + "theoretical_loss": 3.7644649798309775, + "tokens_seen": 728039424 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039366098294884655, + "loss": 1.5212, + "theoretical_loss": 3.7644308149920906, + "tokens_seen": 728104960 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039365095285857573, + "loss": 1.4314, + "theoretical_loss": 3.7643966540891602, + "tokens_seen": 728170496 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003936409227683049, + "loss": 1.4844, + "theoretical_loss": 3.764362497121379, + "tokens_seen": 728236032 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003936308926780341, + "loss": 1.5311, + "theoretical_loss": 3.7643283440879385, + "tokens_seen": 728301568 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039362086258776333, + "loss": 1.3028, + "theoretical_loss": 3.764294194988033, + "tokens_seen": 728367104 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039361083249749246, + "loss": 1.4668, + "theoretical_loss": 3.7642600498208543, + "tokens_seen": 728432640 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003936008024072217, + "loss": 1.4568, + "theoretical_loss": 3.764225908585596, + "tokens_seen": 728498176 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003935907723169508, + "loss": 1.4938, + "theoretical_loss": 3.7641917712814523, + "tokens_seen": 728563712 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039358074222668006, + "loss": 1.4987, + "theoretical_loss": 3.7641576379076165, + "tokens_seen": 728629248 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039357071213640924, + "loss": 1.3538, + "theoretical_loss": 3.764123508463282, + "tokens_seen": 728694784 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003935606820461384, + "loss": 1.3141, + "theoretical_loss": 3.7640893829476445, + "tokens_seen": 728760320 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003935506519558676, + "loss": 1.2126, + "theoretical_loss": 3.7640552613598977, + "tokens_seen": 728825856 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003935406218655968, + "loss": 1.5062, + "theoretical_loss": 3.764021143699237, + "tokens_seen": 728891392 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039353059177532596, + "loss": 1.548, + "theoretical_loss": 3.763987029964856, + "tokens_seen": 728956928 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003935205616850552, + "loss": 1.4386, + "theoretical_loss": 3.763952920155951, + "tokens_seen": 729022464 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0006290096207521856, + "objective/train/docs_used": 421294, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.957242250442505, + "objective/train/original_loss": 2.957242012023926, + "objective/train/theoretical_loss": 3.763918814271718, + "objective/train/tokens_used": 749548000, + "objective/train/value_avg": -0.006809234619140625, + "objective/train/value_loss": 9.040231816470623e-05, + "objective/train/value_max": -0.00037550926208496094, + "objective/train/value_min": -0.0701904296875, + "objective/train/value_reward_corr": 0.47309862186482887, + "objective/train/value_std": 0.0059356689453125, + "objective/train/weight_avg": 0.9979191422462463, + "objective/train/weighted_lm_loss": 2.942451000213623, + "objective/train/weights_max": 1.653180718421936, + "objective/train/weights_min": 0.339802086353302, + "theoretical_loss": 3.763918814271718, + "tokens_seen": 729088000 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003935105315947843, + "loss": 1.4467, + "theoretical_loss": 3.763918814271718, + "tokens_seen": 729088000 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039350050150451356, + "loss": 1.4718, + "theoretical_loss": 3.763884712311352, + "tokens_seen": 729153536 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003934904714142427, + "loss": 1.4494, + "theoretical_loss": 3.763850614274048, + "tokens_seen": 729219072 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003934804413239719, + "loss": 1.3891, + "theoretical_loss": 3.763816520159004, + "tokens_seen": 729284608 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003934704112337011, + "loss": 1.5116, + "theoretical_loss": 3.7637824299654152, + "tokens_seen": 729350144 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003934603811434303, + "loss": 1.2575, + "theoretical_loss": 3.763748343692479, + "tokens_seen": 729415680 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039345035105315946, + "loss": 1.4245, + "theoretical_loss": 3.763714261339392, + "tokens_seen": 729481216 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003934403209628887, + "loss": 1.4346, + "theoretical_loss": 3.7636801829053503, + "tokens_seen": 729546752 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039343029087261783, + "loss": 1.3882, + "theoretical_loss": 3.763646108389553, + "tokens_seen": 729612288 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039342026078234706, + "loss": 1.4402, + "theoretical_loss": 3.763612037791197, + "tokens_seen": 729677824 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003934102306920762, + "loss": 1.3646, + "theoretical_loss": 3.7635779711094797, + "tokens_seen": 729743360 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003934002006018054, + "loss": 1.4237, + "theoretical_loss": 3.7635439083435998, + "tokens_seen": 729808896 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039339017051153466, + "loss": 1.421, + "theoretical_loss": 3.7635098494927552, + "tokens_seen": 729874432 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003933801404212638, + "loss": 1.3489, + "theoretical_loss": 3.7634757945561446, + "tokens_seen": 729939968 + }, + { + "epoch": 0.22, + "learning_rate": 0.000393370110330993, + "loss": 1.3907, + "theoretical_loss": 3.763441743532967, + "tokens_seen": 730005504 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039336008024072215, + "loss": 1.527, + "theoretical_loss": 3.763407696422421, + "tokens_seen": 730071040 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003933500501504514, + "loss": 1.3779, + "theoretical_loss": 3.763373653223706, + "tokens_seen": 730136576 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039334002006018057, + "loss": 1.4272, + "theoretical_loss": 3.7633396139360213, + "tokens_seen": 730202112 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039332998996990975, + "loss": 1.4426, + "theoretical_loss": 3.763305578558567, + "tokens_seen": 730267648 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039331995987963893, + "loss": 1.4441, + "theoretical_loss": 3.763271547090543, + "tokens_seen": 730333184 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039330992978936816, + "loss": 1.4778, + "theoretical_loss": 3.76323751953115, + "tokens_seen": 730398720 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003932998996990973, + "loss": 1.4674, + "theoretical_loss": 3.763203495879587, + "tokens_seen": 730464256 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003932898696088265, + "loss": 1.4934, + "theoretical_loss": 3.7631694761350563, + "tokens_seen": 730529792 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039327983951855565, + "loss": 1.3682, + "theoretical_loss": 3.7631354602967573, + "tokens_seen": 730595328 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003932698094282849, + "loss": 1.3864, + "theoretical_loss": 3.7631014483638925, + "tokens_seen": 730660864 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": 0.0003900799201801419, + "objective/train/docs_used": 422072, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.750411033630371, + "objective/train/original_loss": 2.750410795211792, + "objective/train/theoretical_loss": 3.7630674403356625, + "objective/train/tokens_used": 751186400, + "objective/train/value_avg": -0.005096435546875, + "objective/train/value_loss": 5.7363995438208804e-05, + "objective/train/value_max": -0.0004355907440185547, + "objective/train/value_min": -0.1373291015625, + "objective/train/value_reward_corr": 0.43140815471503025, + "objective/train/value_std": 0.004695892333984375, + "objective/train/weight_avg": 1.006587028503418, + "objective/train/weighted_lm_loss": 2.769063949584961, + "objective/train/weights_max": 2.674281120300293, + "objective/train/weights_min": 0.4172251522541046, + "theoretical_loss": 3.7630674403356625, + "tokens_seen": 730726400 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039325977933801407, + "loss": 1.3944, + "theoretical_loss": 3.7630674403356625, + "tokens_seen": 730726400 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039324974924774325, + "loss": 1.3515, + "theoretical_loss": 3.763033436211269, + "tokens_seen": 730791936 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039323971915747243, + "loss": 1.5041, + "theoretical_loss": 3.762999435989914, + "tokens_seen": 730857472 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003932296890672016, + "loss": 1.4028, + "theoretical_loss": 3.7629654396707997, + "tokens_seen": 730923008 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003932196589769308, + "loss": 1.5128, + "theoretical_loss": 3.7629314472531274, + "tokens_seen": 730988544 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039320962888666003, + "loss": 1.3586, + "theoretical_loss": 3.7628974587361013, + "tokens_seen": 731054080 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039319959879638916, + "loss": 1.6096, + "theoretical_loss": 3.762863474118923, + "tokens_seen": 731119616 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003931895687061184, + "loss": 1.4066, + "theoretical_loss": 3.7628294934007958, + "tokens_seen": 731185152 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003931795386158475, + "loss": 1.4837, + "theoretical_loss": 3.7627955165809226, + "tokens_seen": 731250688 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039316950852557675, + "loss": 1.4421, + "theoretical_loss": 3.762761543658508, + "tokens_seen": 731316224 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039315947843530593, + "loss": 1.429, + "theoretical_loss": 3.7627275746327546, + "tokens_seen": 731381760 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003931494483450351, + "loss": 1.5458, + "theoretical_loss": 3.7626936095028674, + "tokens_seen": 731447296 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003931394182547643, + "loss": 1.4086, + "theoretical_loss": 3.762659648268049, + "tokens_seen": 731512832 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039312938816449353, + "loss": 1.4726, + "theoretical_loss": 3.7626256909275053, + "tokens_seen": 731578368 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039311935807422266, + "loss": 1.4641, + "theoretical_loss": 3.7625917374804407, + "tokens_seen": 731643904 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003931093279839519, + "loss": 1.343, + "theoretical_loss": 3.762557787926059, + "tokens_seen": 731709440 + }, + { + "epoch": 0.22, + "learning_rate": 0.000393099297893681, + "loss": 1.4734, + "theoretical_loss": 3.7625238422635667, + "tokens_seen": 731774976 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039308926780341026, + "loss": 1.5352, + "theoretical_loss": 3.7624899004921684, + "tokens_seen": 731840512 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039307923771313944, + "loss": 1.4314, + "theoretical_loss": 3.76245596261107, + "tokens_seen": 731906048 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003930692076228686, + "loss": 1.3811, + "theoretical_loss": 3.7624220286194774, + "tokens_seen": 731971584 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003930591775325978, + "loss": 1.3863, + "theoretical_loss": 3.7623880985165963, + "tokens_seen": 732037120 + }, + { + "epoch": 0.22, + "learning_rate": 0.000393049147442327, + "loss": 1.3647, + "theoretical_loss": 3.7623541723016327, + "tokens_seen": 732102656 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039303911735205616, + "loss": 1.4237, + "theoretical_loss": 3.7623202499737944, + "tokens_seen": 732168192 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003930290872617854, + "loss": 1.4431, + "theoretical_loss": 3.7622863315322865, + "tokens_seen": 732233728 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003930190571715145, + "loss": 1.4006, + "theoretical_loss": 3.762252416976317, + "tokens_seen": 732299264 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": 2.6109999453183264e-05, + "objective/train/docs_used": 423458, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.406873941421509, + "objective/train/original_loss": 2.4068734645843506, + "objective/train/theoretical_loss": 3.7622185063050937, + "objective/train/tokens_used": 752824800, + "objective/train/value_avg": -0.004657745361328125, + "objective/train/value_loss": 5.620590673061088e-05, + "objective/train/value_max": -0.0004582405090332031, + "objective/train/value_min": -0.04443359375, + "objective/train/value_reward_corr": 0.2922363418392947, + "objective/train/value_std": 0.0032901763916015625, + "objective/train/weight_avg": 1.0028551816940308, + "objective/train/weighted_lm_loss": 2.4154179096221924, + "objective/train/weights_max": 1.4689112901687622, + "objective/train/weights_min": 0.4170580804347992, + "theoretical_loss": 3.7622185063050937, + "tokens_seen": 732364800 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039300902708124376, + "loss": 1.3041, + "theoretical_loss": 3.7622185063050937, + "tokens_seen": 732364800 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003929989969909729, + "loss": 1.4167, + "theoretical_loss": 3.7621845995178225, + "tokens_seen": 732430336 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003929889669007021, + "loss": 1.4742, + "theoretical_loss": 3.762150696613712, + "tokens_seen": 732495872 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003929789368104313, + "loss": 1.4803, + "theoretical_loss": 3.76211679759197, + "tokens_seen": 732561408 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003929689067201605, + "loss": 1.4149, + "theoretical_loss": 3.7620829024518048, + "tokens_seen": 732626944 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039295887662988967, + "loss": 1.4482, + "theoretical_loss": 3.7620490111924245, + "tokens_seen": 732692480 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003929488465396189, + "loss": 1.3641, + "theoretical_loss": 3.762015123813038, + "tokens_seen": 732758016 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039293881644934803, + "loss": 1.3759, + "theoretical_loss": 3.7619812403128536, + "tokens_seen": 732823552 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039292878635907726, + "loss": 1.4645, + "theoretical_loss": 3.761947360691081, + "tokens_seen": 732889088 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003929187562688064, + "loss": 1.4188, + "theoretical_loss": 3.7619134849469296, + "tokens_seen": 732954624 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003929087261785356, + "loss": 1.4876, + "theoretical_loss": 3.761879613079608, + "tokens_seen": 733020160 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003928986960882648, + "loss": 1.5394, + "theoretical_loss": 3.761845745088327, + "tokens_seen": 733085696 + }, + { + "epoch": 0.22, + "learning_rate": 0.000392888665997994, + "loss": 1.4872, + "theoretical_loss": 3.761811880972296, + "tokens_seen": 733151232 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039287863590772317, + "loss": 1.4766, + "theoretical_loss": 3.7617780207307256, + "tokens_seen": 733216768 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039286860581745235, + "loss": 1.3871, + "theoretical_loss": 3.7617441643628258, + "tokens_seen": 733282304 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039285857572718153, + "loss": 1.3919, + "theoretical_loss": 3.761710311867808, + "tokens_seen": 733347840 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039284854563691077, + "loss": 1.4652, + "theoretical_loss": 3.761676463244883, + "tokens_seen": 733413376 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003928385155466399, + "loss": 1.3886, + "theoretical_loss": 3.7616426184932608, + "tokens_seen": 733478912 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039282848545636913, + "loss": 1.474, + "theoretical_loss": 3.761608777612154, + "tokens_seen": 733544448 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039281845536609825, + "loss": 1.5231, + "theoretical_loss": 3.7615749406007746, + "tokens_seen": 733609984 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003928084252758275, + "loss": 1.4689, + "theoretical_loss": 3.761541107458333, + "tokens_seen": 733675520 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039279839518555667, + "loss": 1.4506, + "theoretical_loss": 3.761507278184043, + "tokens_seen": 733741056 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039278836509528585, + "loss": 1.401, + "theoretical_loss": 3.761473452777115, + "tokens_seen": 733806592 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039277833500501503, + "loss": 1.4131, + "theoretical_loss": 3.7614396312367626, + "tokens_seen": 733872128 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039276830491474427, + "loss": 1.4078, + "theoretical_loss": 3.761405813562199, + "tokens_seen": 733937664 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0012120254104956985, + "objective/train/docs_used": 424116, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.130725860595703, + "objective/train/original_loss": 3.130725145339966, + "objective/train/theoretical_loss": 3.7613719997526367, + "objective/train/tokens_used": 754463200, + "objective/train/value_avg": -0.00501251220703125, + "objective/train/value_loss": 7.913057925179601e-05, + "objective/train/value_max": -0.00036406517028808594, + "objective/train/value_min": -0.0633544921875, + "objective/train/value_reward_corr": 0.3388640955471713, + "objective/train/value_std": 0.0036792755126953125, + "objective/train/weight_avg": 0.9915286898612976, + "objective/train/weighted_lm_loss": 3.0921108722686768, + "objective/train/weights_max": 1.884278416633606, + "objective/train/weights_min": 0.39417752623558044, + "theoretical_loss": 3.7613719997526367, + "tokens_seen": 734003200 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003927582748244734, + "loss": 1.4118, + "theoretical_loss": 3.7613719997526367, + "tokens_seen": 734003200 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039274824473420263, + "loss": 1.3795, + "theoretical_loss": 3.7613381898072884, + "tokens_seen": 734068736 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039273821464393176, + "loss": 1.3645, + "theoretical_loss": 3.761304383725369, + "tokens_seen": 734134272 + }, + { + "epoch": 0.22, + "learning_rate": 0.000392728184553661, + "loss": 1.4222, + "theoretical_loss": 3.7612705815060905, + "tokens_seen": 734199808 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003927181544633902, + "loss": 1.4199, + "theoretical_loss": 3.761236783148668, + "tokens_seen": 734265344 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039270812437311936, + "loss": 1.3739, + "theoretical_loss": 3.7612029886523155, + "tokens_seen": 734330880 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039269809428284854, + "loss": 1.3736, + "theoretical_loss": 3.761169198016247, + "tokens_seen": 734396416 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003926880641925777, + "loss": 1.4662, + "theoretical_loss": 3.7611354112396773, + "tokens_seen": 734461952 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003926780341023069, + "loss": 1.4212, + "theoretical_loss": 3.761101628321822, + "tokens_seen": 734527488 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039266800401203613, + "loss": 1.4361, + "theoretical_loss": 3.7610678492618947, + "tokens_seen": 734593024 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039265797392176526, + "loss": 1.3217, + "theoretical_loss": 3.761034074059112, + "tokens_seen": 734658560 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003926479438314945, + "loss": 1.4172, + "theoretical_loss": 3.761000302712689, + "tokens_seen": 734724096 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039263791374122373, + "loss": 1.4757, + "theoretical_loss": 3.760966535221841, + "tokens_seen": 734789632 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039262788365095286, + "loss": 1.2947, + "theoretical_loss": 3.760932771585784, + "tokens_seen": 734855168 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003926178535606821, + "loss": 1.3805, + "theoretical_loss": 3.760899011803735, + "tokens_seen": 734920704 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003926078234704112, + "loss": 1.3851, + "theoretical_loss": 3.7608652558749105, + "tokens_seen": 734986240 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039259779338014046, + "loss": 1.4939, + "theoretical_loss": 3.760831503798527, + "tokens_seen": 735051776 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039258776328986964, + "loss": 1.4295, + "theoretical_loss": 3.7607977555738006, + "tokens_seen": 735117312 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003925777331995988, + "loss": 1.4544, + "theoretical_loss": 3.7607640111999494, + "tokens_seen": 735182848 + }, + { + "epoch": 0.22, + "learning_rate": 0.000392567703109328, + "loss": 1.3803, + "theoretical_loss": 3.7607302706761896, + "tokens_seen": 735248384 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003925576730190572, + "loss": 1.3208, + "theoretical_loss": 3.760696534001741, + "tokens_seen": 735313920 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039254764292878636, + "loss": 1.4724, + "theoretical_loss": 3.760662801175819, + "tokens_seen": 735379456 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003925376128385156, + "loss": 1.4343, + "theoretical_loss": 3.760629072197643, + "tokens_seen": 735444992 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003925275827482447, + "loss": 1.4227, + "theoretical_loss": 3.760595347066431, + "tokens_seen": 735510528 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039251755265797396, + "loss": 1.4131, + "theoretical_loss": 3.760561625781401, + "tokens_seen": 735576064 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0003765590372495353, + "objective/train/docs_used": 424834, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5772857666015625, + "objective/train/original_loss": 2.5772860050201416, + "objective/train/theoretical_loss": 3.760527908341773, + "objective/train/tokens_used": 756101600, + "objective/train/value_avg": -0.00562286376953125, + "objective/train/value_loss": 5.624250843538903e-05, + "objective/train/value_max": -0.000293731689453125, + "objective/train/value_min": -0.12322998046875, + "objective/train/value_reward_corr": 0.40485526317814413, + "objective/train/value_std": 0.00458526611328125, + "objective/train/weight_avg": 0.9989231824874878, + "objective/train/weighted_lm_loss": 2.5714967250823975, + "objective/train/weights_max": 3.168138265609741, + "objective/train/weights_min": 0.4247325658798218, + "theoretical_loss": 3.760527908341773, + "tokens_seen": 735641600 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003925075225677031, + "loss": 1.3533, + "theoretical_loss": 3.760527908341773, + "tokens_seen": 735641600 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003924974924774323, + "loss": 1.3633, + "theoretical_loss": 3.760494194746765, + "tokens_seen": 735707136 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003924874623871615, + "loss": 1.4169, + "theoretical_loss": 3.7604604849955967, + "tokens_seen": 735772672 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003924774322968907, + "loss": 1.3983, + "theoretical_loss": 3.760426779087487, + "tokens_seen": 735838208 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039246740220661987, + "loss": 1.5259, + "theoretical_loss": 3.760393077021656, + "tokens_seen": 735903744 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003924573721163491, + "loss": 1.4221, + "theoretical_loss": 3.760359378797323, + "tokens_seen": 735969280 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039244734202607823, + "loss": 1.4662, + "theoretical_loss": 3.7603256844137087, + "tokens_seen": 736034816 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039243731193580746, + "loss": 1.4726, + "theoretical_loss": 3.760291993870034, + "tokens_seen": 736100352 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003924272818455366, + "loss": 1.4972, + "theoretical_loss": 3.760258307165518, + "tokens_seen": 736165888 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003924172517552658, + "loss": 1.5201, + "theoretical_loss": 3.7602246242993824, + "tokens_seen": 736231424 + }, + { + "epoch": 0.22, + "learning_rate": 0.000392407221664995, + "loss": 1.6076, + "theoretical_loss": 3.760190945270848, + "tokens_seen": 736296960 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003923971915747242, + "loss": 1.6237, + "theoretical_loss": 3.7601572700791364, + "tokens_seen": 736362496 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039238716148445337, + "loss": 1.5131, + "theoretical_loss": 3.7601235987234682, + "tokens_seen": 736428032 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039237713139418255, + "loss": 1.4093, + "theoretical_loss": 3.7600899312030664, + "tokens_seen": 736493568 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039236710130391173, + "loss": 1.4164, + "theoretical_loss": 3.760056267517152, + "tokens_seen": 736559104 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039235707121364097, + "loss": 1.4236, + "theoretical_loss": 3.760022607664947, + "tokens_seen": 736624640 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003923470411233701, + "loss": 1.4356, + "theoretical_loss": 3.7599889516456746, + "tokens_seen": 736690176 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039233701103309933, + "loss": 1.3863, + "theoretical_loss": 3.759955299458557, + "tokens_seen": 736755712 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039232698094282846, + "loss": 1.4682, + "theoretical_loss": 3.7599216511028173, + "tokens_seen": 736821248 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003923169508525577, + "loss": 1.5311, + "theoretical_loss": 3.7598880065776776, + "tokens_seen": 736886784 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039230692076228687, + "loss": 1.541, + "theoretical_loss": 3.7598543658823624, + "tokens_seen": 736952320 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039229689067201605, + "loss": 1.4052, + "theoretical_loss": 3.7598207290160945, + "tokens_seen": 737017856 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039228686058174523, + "loss": 1.4623, + "theoretical_loss": 3.7597870959780977, + "tokens_seen": 737083392 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039227683049147447, + "loss": 1.4978, + "theoretical_loss": 3.759753466767597, + "tokens_seen": 737148928 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003922668004012036, + "loss": 1.4492, + "theoretical_loss": 3.759719841383815, + "tokens_seen": 737214464 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0003744152782019228, + "objective/train/docs_used": 426204, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4784717559814453, + "objective/train/original_loss": 3.478471517562866, + "objective/train/theoretical_loss": 3.7596862198259773, + "objective/train/tokens_used": 757740000, + "objective/train/value_avg": -0.00580596923828125, + "objective/train/value_loss": 7.766007183818147e-05, + "objective/train/value_max": -0.0003998279571533203, + "objective/train/value_min": -0.11181640625, + "objective/train/value_reward_corr": 0.42939837804663156, + "objective/train/value_std": 0.004863739013671875, + "objective/train/weight_avg": 0.9998762607574463, + "objective/train/weighted_lm_loss": 3.484679698944092, + "objective/train/weights_max": 2.6866471767425537, + "objective/train/weights_min": 0.4101046025753021, + "theoretical_loss": 3.7596862198259773, + "tokens_seen": 737280000 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039225677031093283, + "loss": 1.5636, + "theoretical_loss": 3.7596862198259773, + "tokens_seen": 737280000 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039224674022066196, + "loss": 1.4272, + "theoretical_loss": 3.7596526020933076, + "tokens_seen": 737345536 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003922367101303912, + "loss": 1.4573, + "theoretical_loss": 3.7596189881850313, + "tokens_seen": 737411072 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003922266800401204, + "loss": 1.525, + "theoretical_loss": 3.7595853781003745, + "tokens_seen": 737476608 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039221664994984956, + "loss": 1.4523, + "theoretical_loss": 3.7595517718385603, + "tokens_seen": 737542144 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039220661985957874, + "loss": 1.3537, + "theoretical_loss": 3.7595181693988162, + "tokens_seen": 737607680 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003921965897693079, + "loss": 1.3995, + "theoretical_loss": 3.7594845707803675, + "tokens_seen": 737673216 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003921865596790371, + "loss": 1.5679, + "theoretical_loss": 3.7594509759824395, + "tokens_seen": 737738752 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039217652958876633, + "loss": 1.4062, + "theoretical_loss": 3.7594173850042587, + "tokens_seen": 737804288 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039216649949849546, + "loss": 1.4593, + "theoretical_loss": 3.7593837978450524, + "tokens_seen": 737869824 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003921564694082247, + "loss": 1.2724, + "theoretical_loss": 3.7593502145040465, + "tokens_seen": 737935360 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003921464393179539, + "loss": 1.427, + "theoretical_loss": 3.7593166349804683, + "tokens_seen": 738000896 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039213640922768306, + "loss": 1.4216, + "theoretical_loss": 3.7592830592735442, + "tokens_seen": 738066432 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039212637913741224, + "loss": 1.5617, + "theoretical_loss": 3.759249487382502, + "tokens_seen": 738131968 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003921163490471414, + "loss": 1.4124, + "theoretical_loss": 3.7592159193065697, + "tokens_seen": 738197504 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003921063189568706, + "loss": 1.4992, + "theoretical_loss": 3.7591823550449748, + "tokens_seen": 738263040 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039209628886659984, + "loss": 1.4183, + "theoretical_loss": 3.7591487945969453, + "tokens_seen": 738328576 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039208625877632896, + "loss": 1.4537, + "theoretical_loss": 3.759115237961709, + "tokens_seen": 738394112 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003920762286860582, + "loss": 1.3681, + "theoretical_loss": 3.7590816851384954, + "tokens_seen": 738459648 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003920661985957873, + "loss": 1.4166, + "theoretical_loss": 3.759048136126532, + "tokens_seen": 738525184 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039205616850551656, + "loss": 1.3946, + "theoretical_loss": 3.7590145909250494, + "tokens_seen": 738590720 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039204613841524574, + "loss": 1.3068, + "theoretical_loss": 3.7589810495332747, + "tokens_seen": 738656256 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003920361083249749, + "loss": 1.3496, + "theoretical_loss": 3.7589475119504394, + "tokens_seen": 738721792 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003920260782347041, + "loss": 1.4801, + "theoretical_loss": 3.758913978175771, + "tokens_seen": 738787328 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003920160481444333, + "loss": 1.423, + "theoretical_loss": 3.758880448208501, + "tokens_seen": 738852864 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0005848518339917064, + "objective/train/docs_used": 426747, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7717583179473877, + "objective/train/original_loss": 2.771758556365967, + "objective/train/theoretical_loss": 3.7588469220478586, + "objective/train/tokens_used": 759378400, + "objective/train/value_avg": -0.0053863525390625, + "objective/train/value_loss": 5.6675649830140173e-05, + "objective/train/value_max": -0.00047469139099121094, + "objective/train/value_min": -0.09930419921875, + "objective/train/value_reward_corr": 0.4225493419094201, + "objective/train/value_std": 0.004669189453125, + "objective/train/weight_avg": 0.9967314004898071, + "objective/train/weighted_lm_loss": 2.7601163387298584, + "objective/train/weights_max": 1.9330474138259888, + "objective/train/weights_min": 0.13775864243507385, + "theoretical_loss": 3.7588469220478586, + "tokens_seen": 738918400 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039200601805416247, + "loss": 1.4882, + "theoretical_loss": 3.7588469220478586, + "tokens_seen": 738918400 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003919959879638917, + "loss": 1.3987, + "theoretical_loss": 3.758813399693074, + "tokens_seen": 738983936 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039198595787362083, + "loss": 1.289, + "theoretical_loss": 3.7587798811433784, + "tokens_seen": 739049472 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039197592778335007, + "loss": 1.3563, + "theoretical_loss": 3.758746366398002, + "tokens_seen": 739115008 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039196589769307925, + "loss": 1.4186, + "theoretical_loss": 3.7587128554561753, + "tokens_seen": 739180544 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039195586760280843, + "loss": 1.3078, + "theoretical_loss": 3.758679348317131, + "tokens_seen": 739246080 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003919458375125376, + "loss": 1.4441, + "theoretical_loss": 3.758645844980099, + "tokens_seen": 739311616 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003919358074222668, + "loss": 1.4594, + "theoretical_loss": 3.7586123454443117, + "tokens_seen": 739377152 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039192577733199597, + "loss": 1.453, + "theoretical_loss": 3.758578849709, + "tokens_seen": 739442688 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003919157472417252, + "loss": 1.4083, + "theoretical_loss": 3.7585453577733974, + "tokens_seen": 739508224 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039190571715145433, + "loss": 1.3969, + "theoretical_loss": 3.7585118696367354, + "tokens_seen": 739573760 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039189568706118357, + "loss": 1.4917, + "theoretical_loss": 3.758478385298246, + "tokens_seen": 739639296 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039188565697091275, + "loss": 1.3599, + "theoretical_loss": 3.7584449047571633, + "tokens_seen": 739704832 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039187562688064193, + "loss": 1.4868, + "theoretical_loss": 3.758411428012719, + "tokens_seen": 739770368 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039186559679037117, + "loss": 1.412, + "theoretical_loss": 3.7583779550641467, + "tokens_seen": 739835904 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003918555667001003, + "loss": 1.4443, + "theoretical_loss": 3.7583444859106803, + "tokens_seen": 739901440 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039184553660982953, + "loss": 1.4024, + "theoretical_loss": 3.7583110205515524, + "tokens_seen": 739966976 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039183550651955866, + "loss": 1.5371, + "theoretical_loss": 3.7582775589859976, + "tokens_seen": 740032512 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003918254764292879, + "loss": 1.4559, + "theoretical_loss": 3.7582441012132497, + "tokens_seen": 740098048 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039181544633901707, + "loss": 1.4815, + "theoretical_loss": 3.758210647232543, + "tokens_seen": 740163584 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039180541624874625, + "loss": 1.5553, + "theoretical_loss": 3.7581771970431124, + "tokens_seen": 740229120 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039179538615847543, + "loss": 1.441, + "theoretical_loss": 3.7581437506441926, + "tokens_seen": 740294656 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039178535606820467, + "loss": 1.4113, + "theoretical_loss": 3.758110308035018, + "tokens_seen": 740360192 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003917753259779338, + "loss": 1.3099, + "theoretical_loss": 3.7580768692148236, + "tokens_seen": 740425728 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039176529588766303, + "loss": 1.4136, + "theoretical_loss": 3.758043434182846, + "tokens_seen": 740491264 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": -0.0006371609633788466, + "objective/train/docs_used": 427723, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.755448818206787, + "objective/train/original_loss": 2.755448579788208, + "objective/train/theoretical_loss": 3.75801000293832, + "objective/train/tokens_used": 761016800, + "objective/train/value_avg": -0.004119873046875, + "objective/train/value_loss": 5.27717056684196e-05, + "objective/train/value_max": -0.00036406517028808594, + "objective/train/value_min": -0.043853759765625, + "objective/train/value_reward_corr": 0.2912960608944865, + "objective/train/value_std": 0.003231048583984375, + "objective/train/weight_avg": 0.9961222410202026, + "objective/train/weighted_lm_loss": 2.752345085144043, + "objective/train/weights_max": 1.3819372653961182, + "objective/train/weights_min": 0.589179515838623, + "theoretical_loss": 3.75801000293832, + "tokens_seen": 740556800 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039175526579739216, + "loss": 1.4727, + "theoretical_loss": 3.75801000293832, + "tokens_seen": 740556800 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003917452357071214, + "loss": 1.338, + "theoretical_loss": 3.7579765754804813, + "tokens_seen": 740622336 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003917352056168506, + "loss": 1.4944, + "theoretical_loss": 3.7579431518085666, + "tokens_seen": 740687872 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039172517552657976, + "loss": 1.4789, + "theoretical_loss": 3.7579097319218113, + "tokens_seen": 740753408 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039171514543630894, + "loss": 1.5778, + "theoretical_loss": 3.757876315819453, + "tokens_seen": 740818944 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003917051153460381, + "loss": 1.3017, + "theoretical_loss": 3.7578429035007277, + "tokens_seen": 740884480 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003916950852557673, + "loss": 1.4641, + "theoretical_loss": 3.7578094949648726, + "tokens_seen": 740950016 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039168505516549653, + "loss": 1.5032, + "theoretical_loss": 3.7577760902111246, + "tokens_seen": 741015552 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039167502507522566, + "loss": 1.3744, + "theoretical_loss": 3.7577426892387216, + "tokens_seen": 741081088 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003916649949849549, + "loss": 1.5213, + "theoretical_loss": 3.7577092920469006, + "tokens_seen": 741146624 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003916549648946841, + "loss": 1.351, + "theoretical_loss": 3.7576758986348997, + "tokens_seen": 741212160 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039164493480441326, + "loss": 1.3569, + "theoretical_loss": 3.7576425090019576, + "tokens_seen": 741277696 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039163490471414244, + "loss": 1.4671, + "theoretical_loss": 3.7576091231473114, + "tokens_seen": 741343232 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003916248746238716, + "loss": 1.4808, + "theoretical_loss": 3.757575741070201, + "tokens_seen": 741408768 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003916148445336008, + "loss": 1.4428, + "theoretical_loss": 3.757542362769864, + "tokens_seen": 741474304 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039160481444333004, + "loss": 1.4912, + "theoretical_loss": 3.7575089882455392, + "tokens_seen": 741539840 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039159478435305916, + "loss": 1.475, + "theoretical_loss": 3.7574756174964667, + "tokens_seen": 741605376 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003915847542627884, + "loss": 1.5359, + "theoretical_loss": 3.757442250521886, + "tokens_seen": 741670912 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003915747241725175, + "loss": 1.3751, + "theoretical_loss": 3.7574088873210356, + "tokens_seen": 741736448 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039156469408224676, + "loss": 1.5533, + "theoretical_loss": 3.757375527893156, + "tokens_seen": 741801984 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039155466399197594, + "loss": 1.5552, + "theoretical_loss": 3.757342172237487, + "tokens_seen": 741867520 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003915446339017051, + "loss": 1.4438, + "theoretical_loss": 3.7573088203532694, + "tokens_seen": 741933056 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003915346038114343, + "loss": 1.3726, + "theoretical_loss": 3.7572754722397432, + "tokens_seen": 741998592 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003915245737211635, + "loss": 1.4592, + "theoretical_loss": 3.757242127896149, + "tokens_seen": 742064128 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039151454363089267, + "loss": 1.3755, + "theoretical_loss": 3.757208787321728, + "tokens_seen": 742129664 + }, + { + "epoch": 0.22, + "objective/train/advantage_avg": 0.0007752607925795019, + "objective/train/docs_used": 429091, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.860340118408203, + "objective/train/original_loss": 2.8603403568267822, + "objective/train/theoretical_loss": 3.7571754505157218, + "objective/train/tokens_used": 762655200, + "objective/train/value_avg": -0.004787445068359375, + "objective/train/value_loss": 3.546662992448546e-05, + "objective/train/value_max": -0.00037860870361328125, + "objective/train/value_min": -0.0386962890625, + "objective/train/value_reward_corr": 0.2942893262364646, + "objective/train/value_std": 0.0031909942626953125, + "objective/train/weight_avg": 1.0094810724258423, + "objective/train/weighted_lm_loss": 2.8810174465179443, + "objective/train/weights_max": 1.462648868560791, + "objective/train/weights_min": 0.6165716052055359, + "theoretical_loss": 3.7571754505157218, + "tokens_seen": 742195200 + }, + { + "epoch": 0.22, + "learning_rate": 0.0003915045135406219, + "loss": 1.3868, + "theoretical_loss": 3.7571754505157218, + "tokens_seen": 742195200 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039149448345035103, + "loss": 1.3492, + "theoretical_loss": 3.7571421174773705, + "tokens_seen": 742260736 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039148445336008027, + "loss": 1.3858, + "theoretical_loss": 3.7571087882059166, + "tokens_seen": 742326272 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039147442326980945, + "loss": 1.3955, + "theoretical_loss": 3.7570754627006018, + "tokens_seen": 742391808 + }, + { + "epoch": 0.22, + "learning_rate": 0.00039146439317953863, + "loss": 1.4941, + "theoretical_loss": 3.7570421409606682, + "tokens_seen": 742457344 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003914543630892678, + "loss": 1.3563, + "theoretical_loss": 3.757008822985358, + "tokens_seen": 742522880 + }, + { + "epoch": 0.23, + "learning_rate": 0.000391444332998997, + "loss": 1.4512, + "theoretical_loss": 3.756975508773914, + "tokens_seen": 742588416 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039143430290872617, + "loss": 1.5436, + "theoretical_loss": 3.7569421983255777, + "tokens_seen": 742653952 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003914242728184554, + "loss": 1.4585, + "theoretical_loss": 3.7569088916395934, + "tokens_seen": 742719488 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039141424272818453, + "loss": 1.4033, + "theoretical_loss": 3.756875588715203, + "tokens_seen": 742785024 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039140421263791377, + "loss": 1.4636, + "theoretical_loss": 3.756842289551651, + "tokens_seen": 742850560 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003913941825476429, + "loss": 1.2628, + "theoretical_loss": 3.7568089941481806, + "tokens_seen": 742916096 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039138415245737213, + "loss": 1.5221, + "theoretical_loss": 3.7567757025040347, + "tokens_seen": 742981632 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003913741223671013, + "loss": 1.3601, + "theoretical_loss": 3.7567424146184587, + "tokens_seen": 743047168 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003913640922768305, + "loss": 1.45, + "theoretical_loss": 3.7567091304906954, + "tokens_seen": 743112704 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003913540621865597, + "loss": 1.4688, + "theoretical_loss": 3.756675850119991, + "tokens_seen": 743178240 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039134403209628886, + "loss": 1.4417, + "theoretical_loss": 3.756642573505588, + "tokens_seen": 743243776 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039133400200601804, + "loss": 1.3988, + "theoretical_loss": 3.7566093006467334, + "tokens_seen": 743309312 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039132397191574727, + "loss": 1.3674, + "theoretical_loss": 3.7565760315426706, + "tokens_seen": 743374848 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003913139418254764, + "loss": 1.5358, + "theoretical_loss": 3.756542766192646, + "tokens_seen": 743440384 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039130391173520563, + "loss": 1.3257, + "theoretical_loss": 3.756509504595904, + "tokens_seen": 743505920 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003912938816449348, + "loss": 1.4906, + "theoretical_loss": 3.7564762467516917, + "tokens_seen": 743571456 + }, + { + "epoch": 0.23, + "learning_rate": 0.000391283851554664, + "loss": 1.414, + "theoretical_loss": 3.756442992659254, + "tokens_seen": 743636992 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003912738214643932, + "loss": 1.362, + "theoretical_loss": 3.756409742317838, + "tokens_seen": 743702528 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039126379137412236, + "loss": 1.5706, + "theoretical_loss": 3.7563764957266894, + "tokens_seen": 743768064 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.00015810556942597032, + "objective/train/docs_used": 429832, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.988060474395752, + "objective/train/original_loss": 2.988060235977173, + "objective/train/theoretical_loss": 3.756343252885055, + "objective/train/tokens_used": 764293600, + "objective/train/value_avg": -0.00499725341796875, + "objective/train/value_loss": 4.850095501751639e-05, + "objective/train/value_max": -0.0003998279571533203, + "objective/train/value_min": -0.045257568359375, + "objective/train/value_reward_corr": 0.35753848464764604, + "objective/train/value_std": 0.00377655029296875, + "objective/train/weight_avg": 1.000756025314331, + "objective/train/weighted_lm_loss": 3.0018343925476074, + "objective/train/weights_max": 1.505674123764038, + "objective/train/weights_min": 0.5403850078582764, + "theoretical_loss": 3.756343252885055, + "tokens_seen": 743833600 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039125376128385154, + "loss": 1.463, + "theoretical_loss": 3.756343252885055, + "tokens_seen": 743833600 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003912437311935808, + "loss": 1.4354, + "theoretical_loss": 3.7563100137921817, + "tokens_seen": 743899136 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003912337011033099, + "loss": 1.4608, + "theoretical_loss": 3.756276778447316, + "tokens_seen": 743964672 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039122367101303914, + "loss": 1.4059, + "theoretical_loss": 3.7562435468497064, + "tokens_seen": 744030208 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039121364092276826, + "loss": 1.5281, + "theoretical_loss": 3.7562103189985994, + "tokens_seen": 744095744 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003912036108324975, + "loss": 1.4499, + "theoretical_loss": 3.756177094893243, + "tokens_seen": 744161280 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003911935807422267, + "loss": 1.4803, + "theoretical_loss": 3.7561438745328855, + "tokens_seen": 744226816 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039118355065195586, + "loss": 1.6333, + "theoretical_loss": 3.756110657916774, + "tokens_seen": 744292352 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039117352056168504, + "loss": 1.4648, + "theoretical_loss": 3.7560774450441583, + "tokens_seen": 744357888 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003911634904714143, + "loss": 1.3767, + "theoretical_loss": 3.7560442359142856, + "tokens_seen": 744423424 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003911534603811434, + "loss": 1.48, + "theoretical_loss": 3.7560110305264054, + "tokens_seen": 744488960 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039114343029087264, + "loss": 1.4812, + "theoretical_loss": 3.7559778288797663, + "tokens_seen": 744554496 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003911334002006018, + "loss": 1.367, + "theoretical_loss": 3.755944630973618, + "tokens_seen": 744620032 + }, + { + "epoch": 0.23, + "learning_rate": 0.000391123370110331, + "loss": 1.4582, + "theoretical_loss": 3.75591143680721, + "tokens_seen": 744685568 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039111334002006024, + "loss": 1.3997, + "theoretical_loss": 3.7558782463797917, + "tokens_seen": 744751104 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039110330992978936, + "loss": 1.4348, + "theoretical_loss": 3.755845059690613, + "tokens_seen": 744816640 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003910932798395186, + "loss": 1.4697, + "theoretical_loss": 3.7558118767389237, + "tokens_seen": 744882176 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039108324974924773, + "loss": 1.4669, + "theoretical_loss": 3.755778697523975, + "tokens_seen": 744947712 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039107321965897696, + "loss": 1.4845, + "theoretical_loss": 3.755745522045016, + "tokens_seen": 745013248 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039106318956870614, + "loss": 1.3266, + "theoretical_loss": 3.7557123503012986, + "tokens_seen": 745078784 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003910531594784353, + "loss": 1.4482, + "theoretical_loss": 3.755679182292073, + "tokens_seen": 745144320 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003910431293881645, + "loss": 1.3513, + "theoretical_loss": 3.7556460180165914, + "tokens_seen": 745209856 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003910330992978937, + "loss": 1.4386, + "theoretical_loss": 3.7556128574741043, + "tokens_seen": 745275392 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039102306920762287, + "loss": 1.4433, + "theoretical_loss": 3.755579700663864, + "tokens_seen": 745340928 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003910130391173521, + "loss": 1.4826, + "theoretical_loss": 3.755546547585121, + "tokens_seen": 745406464 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": 0.0003699743829201907, + "objective/train/docs_used": 430489, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5472586154937744, + "objective/train/original_loss": 2.5472583770751953, + "objective/train/theoretical_loss": 3.755513398237129, + "objective/train/tokens_used": 765932000, + "objective/train/value_avg": -0.004192352294921875, + "objective/train/value_loss": 3.291424582130276e-05, + "objective/train/value_max": -0.00031495094299316406, + "objective/train/value_min": -0.033477783203125, + "objective/train/value_reward_corr": 0.24012112402003338, + "objective/train/value_std": 0.0030117034912109375, + "objective/train/weight_avg": 1.0053038597106934, + "objective/train/weighted_lm_loss": 2.561643600463867, + "objective/train/weights_max": 1.3318381309509277, + "objective/train/weights_min": 0.5637206435203552, + "theoretical_loss": 3.755513398237129, + "tokens_seen": 745472000 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039100300902708123, + "loss": 1.3577, + "theoretical_loss": 3.755513398237129, + "tokens_seen": 745472000 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039099297893681047, + "loss": 1.5294, + "theoretical_loss": 3.7554802526191393, + "tokens_seen": 745537536 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039098294884653965, + "loss": 1.4438, + "theoretical_loss": 3.7554471107304046, + "tokens_seen": 745603072 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039097291875626883, + "loss": 1.412, + "theoretical_loss": 3.755413972570177, + "tokens_seen": 745668608 + }, + { + "epoch": 0.23, + "learning_rate": 0.000390962888665998, + "loss": 1.4654, + "theoretical_loss": 3.7553808381377105, + "tokens_seen": 745734144 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003909528585757272, + "loss": 1.2173, + "theoretical_loss": 3.7553477074322577, + "tokens_seen": 745799680 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039094282848545637, + "loss": 1.3982, + "theoretical_loss": 3.7553145804530716, + "tokens_seen": 745865216 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003909327983951856, + "loss": 1.4538, + "theoretical_loss": 3.7552814571994064, + "tokens_seen": 745930752 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039092276830491473, + "loss": 1.4505, + "theoretical_loss": 3.755248337670515, + "tokens_seen": 745996288 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039091273821464397, + "loss": 1.4568, + "theoretical_loss": 3.755215221865652, + "tokens_seen": 746061824 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003909027081243731, + "loss": 1.3817, + "theoretical_loss": 3.755182109784071, + "tokens_seen": 746127360 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039089267803410233, + "loss": 1.4012, + "theoretical_loss": 3.755149001425028, + "tokens_seen": 746192896 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003908826479438315, + "loss": 1.4715, + "theoretical_loss": 3.7551158967877756, + "tokens_seen": 746258432 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003908726178535607, + "loss": 1.3062, + "theoretical_loss": 3.75508279587157, + "tokens_seen": 746323968 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003908625877632899, + "loss": 1.4202, + "theoretical_loss": 3.7550496986756654, + "tokens_seen": 746389504 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039085255767301906, + "loss": 1.5442, + "theoretical_loss": 3.7550166051993172, + "tokens_seen": 746455040 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039084252758274824, + "loss": 1.3322, + "theoretical_loss": 3.7549835154417814, + "tokens_seen": 746520576 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039083249749247747, + "loss": 1.4269, + "theoretical_loss": 3.7549504294023137, + "tokens_seen": 746586112 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003908224674022066, + "loss": 1.5518, + "theoretical_loss": 3.7549173470801693, + "tokens_seen": 746651648 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039081243731193583, + "loss": 1.5051, + "theoretical_loss": 3.754884268474605, + "tokens_seen": 746717184 + }, + { + "epoch": 0.23, + "learning_rate": 0.000390802407221665, + "loss": 1.4909, + "theoretical_loss": 3.7548511935848765, + "tokens_seen": 746782720 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003907923771313942, + "loss": 1.5152, + "theoretical_loss": 3.7548181224102404, + "tokens_seen": 746848256 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003907823470411234, + "loss": 1.5009, + "theoretical_loss": 3.754785054949954, + "tokens_seen": 746913792 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039077231695085256, + "loss": 1.4364, + "theoretical_loss": 3.7547519912032747, + "tokens_seen": 746979328 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039076228686058174, + "loss": 1.4694, + "theoretical_loss": 3.7547189311694584, + "tokens_seen": 747044864 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0013004184002056718, + "objective/train/docs_used": 431396, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.732736349105835, + "objective/train/original_loss": 2.732736587524414, + "objective/train/theoretical_loss": 3.7546858748477634, + "objective/train/tokens_used": 767570400, + "objective/train/value_avg": -0.00583648681640625, + "objective/train/value_loss": 7.102457311702892e-05, + "objective/train/value_max": -0.0004601478576660156, + "objective/train/value_min": -0.05987548828125, + "objective/train/value_reward_corr": 0.5277923962834653, + "objective/train/value_std": 0.005084991455078125, + "objective/train/weight_avg": 0.9903489947319031, + "objective/train/weighted_lm_loss": 2.7037084102630615, + "objective/train/weights_max": 1.479728102684021, + "objective/train/weights_min": 0.44928932189941406, + "theoretical_loss": 3.7546858748477634, + "tokens_seen": 747110400 + }, + { + "epoch": 0.23, + "learning_rate": 0.000390752256770311, + "loss": 1.4027, + "theoretical_loss": 3.7546858748477634, + "tokens_seen": 747110400 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003907422266800401, + "loss": 1.4378, + "theoretical_loss": 3.7546528222374467, + "tokens_seen": 747175936 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039073219658976934, + "loss": 1.5278, + "theoretical_loss": 3.7546197733377666, + "tokens_seen": 747241472 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039072216649949846, + "loss": 1.4722, + "theoretical_loss": 3.7545867281479812, + "tokens_seen": 747307008 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003907121364092277, + "loss": 1.4576, + "theoretical_loss": 3.7545536866673483, + "tokens_seen": 747372544 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003907021063189569, + "loss": 1.5295, + "theoretical_loss": 3.754520648895127, + "tokens_seen": 747438080 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039069207622868606, + "loss": 1.4761, + "theoretical_loss": 3.754487614830575, + "tokens_seen": 747503616 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039068204613841524, + "loss": 1.4072, + "theoretical_loss": 3.7544545844729527, + "tokens_seen": 747569152 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003906720160481445, + "loss": 1.4073, + "theoretical_loss": 3.7544215578215177, + "tokens_seen": 747634688 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003906619859578736, + "loss": 1.4682, + "theoretical_loss": 3.75438853487553, + "tokens_seen": 747700224 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039065195586760284, + "loss": 1.5451, + "theoretical_loss": 3.7543555156342494, + "tokens_seen": 747765760 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039064192577733197, + "loss": 1.5369, + "theoretical_loss": 3.7543225000969356, + "tokens_seen": 747831296 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003906318956870612, + "loss": 1.4522, + "theoretical_loss": 3.7542894882628475, + "tokens_seen": 747896832 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003906218655967904, + "loss": 1.4506, + "theoretical_loss": 3.7542564801312466, + "tokens_seen": 747962368 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039061183550651957, + "loss": 1.4244, + "theoretical_loss": 3.7542234757013926, + "tokens_seen": 748027904 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039060180541624875, + "loss": 1.4125, + "theoretical_loss": 3.7541904749725465, + "tokens_seen": 748093440 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039059177532597793, + "loss": 1.4458, + "theoretical_loss": 3.7541574779439686, + "tokens_seen": 748158976 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003905817452357071, + "loss": 1.4036, + "theoretical_loss": 3.7541244846149207, + "tokens_seen": 748224512 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039057171514543634, + "loss": 1.4603, + "theoretical_loss": 3.754091494984663, + "tokens_seen": 748290048 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039056168505516547, + "loss": 1.398, + "theoretical_loss": 3.7540585090524576, + "tokens_seen": 748355584 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003905516549648947, + "loss": 1.3807, + "theoretical_loss": 3.7540255268175664, + "tokens_seen": 748421120 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039054162487462383, + "loss": 1.4535, + "theoretical_loss": 3.7539925482792507, + "tokens_seen": 748486656 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039053159478435307, + "loss": 1.4558, + "theoretical_loss": 3.753959573436773, + "tokens_seen": 748552192 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039052156469408225, + "loss": 1.3958, + "theoretical_loss": 3.753926602289395, + "tokens_seen": 748617728 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039051153460381143, + "loss": 1.4842, + "theoretical_loss": 3.75389363483638, + "tokens_seen": 748683264 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": 8.694343705428764e-05, + "objective/train/docs_used": 432599, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9626576900482178, + "objective/train/original_loss": 2.9626574516296387, + "objective/train/theoretical_loss": 3.7538606710769904, + "objective/train/tokens_used": 769208800, + "objective/train/value_avg": -0.00492095947265625, + "objective/train/value_loss": 6.643614324275404e-05, + "objective/train/value_max": -0.0002491474151611328, + "objective/train/value_min": -0.0743408203125, + "objective/train/value_reward_corr": 0.3939703126259623, + "objective/train/value_std": 0.004344940185546875, + "objective/train/weight_avg": 1.0039050579071045, + "objective/train/weighted_lm_loss": 2.963059663772583, + "objective/train/weights_max": 1.7275199890136719, + "objective/train/weights_min": 0.2593618929386139, + "theoretical_loss": 3.7538606710769904, + "tokens_seen": 748748800 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003905015045135406, + "loss": 1.3928, + "theoretical_loss": 3.7538606710769904, + "tokens_seen": 748748800 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039049147442326985, + "loss": 1.5383, + "theoretical_loss": 3.753827711010489, + "tokens_seen": 748814336 + }, + { + "epoch": 0.23, + "learning_rate": 0.000390481444332999, + "loss": 1.5506, + "theoretical_loss": 3.7537947546361385, + "tokens_seen": 748879872 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003904714142427282, + "loss": 1.5004, + "theoretical_loss": 3.7537618019532033, + "tokens_seen": 748945408 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039046138415245734, + "loss": 1.5359, + "theoretical_loss": 3.753728852960947, + "tokens_seen": 749010944 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039045135406218657, + "loss": 1.3765, + "theoretical_loss": 3.7536959076586314, + "tokens_seen": 749076480 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039044132397191575, + "loss": 1.5154, + "theoretical_loss": 3.753662966045523, + "tokens_seen": 749142016 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039043129388164493, + "loss": 1.6158, + "theoretical_loss": 3.7536300281208845, + "tokens_seen": 749207552 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003904212637913741, + "loss": 1.5237, + "theoretical_loss": 3.753597093883981, + "tokens_seen": 749273088 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003904112337011033, + "loss": 1.4257, + "theoretical_loss": 3.753564163334077, + "tokens_seen": 749338624 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003904012036108325, + "loss": 1.4212, + "theoretical_loss": 3.7535312364704367, + "tokens_seen": 749404160 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003903911735205617, + "loss": 1.4071, + "theoretical_loss": 3.753498313292326, + "tokens_seen": 749469696 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003903811434302909, + "loss": 1.5281, + "theoretical_loss": 3.7534653937990097, + "tokens_seen": 749535232 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003903711133400201, + "loss": 1.531, + "theoretical_loss": 3.7534324779897528, + "tokens_seen": 749600768 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039036108324974926, + "loss": 1.5591, + "theoretical_loss": 3.7533995658638224, + "tokens_seen": 749666304 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039035105315947844, + "loss": 1.4946, + "theoretical_loss": 3.753366657420483, + "tokens_seen": 749731840 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039034102306920767, + "loss": 1.425, + "theoretical_loss": 3.7533337526590014, + "tokens_seen": 749797376 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003903309929789368, + "loss": 1.4444, + "theoretical_loss": 3.7533008515786435, + "tokens_seen": 749862912 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039032096288866603, + "loss": 1.5037, + "theoretical_loss": 3.753267954178676, + "tokens_seen": 749928448 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003903109327983952, + "loss": 1.5231, + "theoretical_loss": 3.7532350604583664, + "tokens_seen": 749993984 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003903009027081244, + "loss": 1.4836, + "theoretical_loss": 3.75320217041698, + "tokens_seen": 750059520 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003902908726178536, + "loss": 1.4286, + "theoretical_loss": 3.7531692840537856, + "tokens_seen": 750125056 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039028084252758276, + "loss": 1.3418, + "theoretical_loss": 3.7531364013680495, + "tokens_seen": 750190592 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039027081243731194, + "loss": 1.4774, + "theoretical_loss": 3.7531035223590394, + "tokens_seen": 750256128 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003902607823470412, + "loss": 1.3935, + "theoretical_loss": 3.7530706470260236, + "tokens_seen": 750321664 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0007170581957325339, + "objective/train/docs_used": 433353, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.954293966293335, + "objective/train/original_loss": 2.9542934894561768, + "objective/train/theoretical_loss": 3.7530377753682695, + "objective/train/tokens_used": 770847200, + "objective/train/value_avg": -0.005062103271484375, + "objective/train/value_loss": 5.808454443467781e-05, + "objective/train/value_max": -0.0003998279571533203, + "objective/train/value_min": -0.0416259765625, + "objective/train/value_reward_corr": 0.3027583198614816, + "objective/train/value_std": 0.0035839080810546875, + "objective/train/weight_avg": 0.9954973459243774, + "objective/train/weighted_lm_loss": 2.9460232257843018, + "objective/train/weights_max": 1.435672402381897, + "objective/train/weights_min": 0.1950109452009201, + "theoretical_loss": 3.7530377753682695, + "tokens_seen": 750387200 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003902507522567703, + "loss": 1.5191, + "theoretical_loss": 3.7530377753682695, + "tokens_seen": 750387200 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039024072216649954, + "loss": 1.4631, + "theoretical_loss": 3.7530049073850456, + "tokens_seen": 750452736 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039023069207622866, + "loss": 1.3952, + "theoretical_loss": 3.75297204307562, + "tokens_seen": 750518272 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003902206619859579, + "loss": 1.4114, + "theoretical_loss": 3.7529391824392624, + "tokens_seen": 750583808 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003902106318956871, + "loss": 1.4416, + "theoretical_loss": 3.75290632547524, + "tokens_seen": 750649344 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039020060180541626, + "loss": 1.5194, + "theoretical_loss": 3.7528734721828227, + "tokens_seen": 750714880 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039019057171514544, + "loss": 1.4831, + "theoretical_loss": 3.75284062256128, + "tokens_seen": 750780416 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003901805416248747, + "loss": 1.4739, + "theoretical_loss": 3.752807776609881, + "tokens_seen": 750845952 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003901705115346038, + "loss": 1.4301, + "theoretical_loss": 3.7527749343278956, + "tokens_seen": 750911488 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039016048144433304, + "loss": 1.4465, + "theoretical_loss": 3.752742095714593, + "tokens_seen": 750977024 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039015045135406217, + "loss": 1.4766, + "theoretical_loss": 3.7527092607692443, + "tokens_seen": 751042560 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003901404212637914, + "loss": 1.4368, + "theoretical_loss": 3.752676429491119, + "tokens_seen": 751108096 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003901303911735206, + "loss": 1.4287, + "theoretical_loss": 3.7526436018794875, + "tokens_seen": 751173632 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039012036108324977, + "loss": 1.4448, + "theoretical_loss": 3.7526107779336213, + "tokens_seen": 751239168 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039011033099297895, + "loss": 1.4371, + "theoretical_loss": 3.752577957652791, + "tokens_seen": 751304704 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039010030090270813, + "loss": 1.4455, + "theoretical_loss": 3.752545141036267, + "tokens_seen": 751370240 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003900902708124373, + "loss": 1.5725, + "theoretical_loss": 3.7525123280833217, + "tokens_seen": 751435776 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039008024072216654, + "loss": 1.5415, + "theoretical_loss": 3.7524795187932263, + "tokens_seen": 751501312 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039007021063189567, + "loss": 1.4015, + "theoretical_loss": 3.752446713165252, + "tokens_seen": 751566848 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003900601805416249, + "loss": 1.4128, + "theoretical_loss": 3.7524139111986714, + "tokens_seen": 751632384 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039005015045135403, + "loss": 1.4393, + "theoretical_loss": 3.7523811128927567, + "tokens_seen": 751697920 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039004012036108327, + "loss": 1.3981, + "theoretical_loss": 3.7523483182467796, + "tokens_seen": 751763456 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039003009027081245, + "loss": 1.3376, + "theoretical_loss": 3.7523155272600137, + "tokens_seen": 751828992 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039002006018054163, + "loss": 1.4139, + "theoretical_loss": 3.7522827399317307, + "tokens_seen": 751894528 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003900100300902708, + "loss": 1.5158, + "theoretical_loss": 3.752249956261204, + "tokens_seen": 751960064 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0011542283464223146, + "objective/train/docs_used": 434593, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6631977558135986, + "objective/train/original_loss": 2.6631975173950195, + "objective/train/theoretical_loss": 3.752217176247707, + "objective/train/tokens_used": 772485600, + "objective/train/value_avg": -0.005218505859375, + "objective/train/value_loss": 0.00010632596240611747, + "objective/train/value_max": -0.00033283233642578125, + "objective/train/value_min": -0.058135986328125, + "objective/train/value_reward_corr": 0.4415595765371465, + "objective/train/value_std": 0.0051116943359375, + "objective/train/weight_avg": 0.9931341409683228, + "objective/train/weighted_lm_loss": 2.6433424949645996, + "objective/train/weights_max": 1.6300104856491089, + "objective/train/weights_min": 0.19219349324703217, + "theoretical_loss": 3.752217176247707, + "tokens_seen": 752025600 + }, + { + "epoch": 0.23, + "learning_rate": 0.00039000000000000005, + "loss": 1.4981, + "theoretical_loss": 3.752217176247707, + "tokens_seen": 752025600 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003899899699097292, + "loss": 1.4163, + "theoretical_loss": 3.7521843998905133, + "tokens_seen": 752091136 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003899799398194584, + "loss": 1.4278, + "theoretical_loss": 3.7521516271888964, + "tokens_seen": 752156672 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038996990972918754, + "loss": 1.3971, + "theoretical_loss": 3.752118858142129, + "tokens_seen": 752222208 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038995987963891677, + "loss": 1.4413, + "theoretical_loss": 3.7520860927494875, + "tokens_seen": 752287744 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038994984954864595, + "loss": 1.3049, + "theoretical_loss": 3.752053331010244, + "tokens_seen": 752353280 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038993981945837513, + "loss": 1.5, + "theoretical_loss": 3.7520205729236737, + "tokens_seen": 752418816 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003899297893681043, + "loss": 1.3203, + "theoretical_loss": 3.7519878184890514, + "tokens_seen": 752484352 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003899197592778335, + "loss": 1.4328, + "theoretical_loss": 3.7519550677056523, + "tokens_seen": 752549888 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003899097291875627, + "loss": 1.4994, + "theoretical_loss": 3.75192232057275, + "tokens_seen": 752615424 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003898996990972919, + "loss": 1.4107, + "theoretical_loss": 3.7518895770896217, + "tokens_seen": 752680960 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038988966900702104, + "loss": 1.4084, + "theoretical_loss": 3.7518568372555414, + "tokens_seen": 752746496 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003898796389167503, + "loss": 1.4994, + "theoretical_loss": 3.7518241010697864, + "tokens_seen": 752812032 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003898696088264794, + "loss": 1.3867, + "theoretical_loss": 3.751791368531631, + "tokens_seen": 752877568 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038985957873620864, + "loss": 1.4532, + "theoretical_loss": 3.751758639640351, + "tokens_seen": 752943104 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003898495486459378, + "loss": 1.568, + "theoretical_loss": 3.751725914395225, + "tokens_seen": 753008640 + }, + { + "epoch": 0.23, + "learning_rate": 0.000389839518555667, + "loss": 1.4357, + "theoretical_loss": 3.751693192795527, + "tokens_seen": 753074176 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003898294884653962, + "loss": 1.5658, + "theoretical_loss": 3.7516604748405356, + "tokens_seen": 753139712 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003898194583751254, + "loss": 1.4709, + "theoretical_loss": 3.7516277605295265, + "tokens_seen": 753205248 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038980942828485454, + "loss": 1.5165, + "theoretical_loss": 3.751595049861778, + "tokens_seen": 753270784 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003897993981945838, + "loss": 1.4296, + "theoretical_loss": 3.751562342836566, + "tokens_seen": 753336320 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003897893681043129, + "loss": 1.4493, + "theoretical_loss": 3.751529639453169, + "tokens_seen": 753401856 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038977933801404214, + "loss": 1.4414, + "theoretical_loss": 3.751496939710865, + "tokens_seen": 753467392 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003897693079237713, + "loss": 1.5725, + "theoretical_loss": 3.7514642436089316, + "tokens_seen": 753532928 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003897592778335005, + "loss": 1.3772, + "theoretical_loss": 3.7514315511466467, + "tokens_seen": 753598464 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": 0.0006734259659424424, + "objective/train/docs_used": 435245, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.764052391052246, + "objective/train/original_loss": 2.764052152633667, + "objective/train/theoretical_loss": 3.7513988623232883, + "objective/train/tokens_used": 774124000, + "objective/train/value_avg": -0.004650115966796875, + "objective/train/value_loss": 4.373448609840125e-05, + "objective/train/value_max": -0.00037550926208496094, + "objective/train/value_min": -0.056549072265625, + "objective/train/value_reward_corr": 0.3514459071555547, + "objective/train/value_std": 0.0032787322998046875, + "objective/train/weight_avg": 1.0087909698486328, + "objective/train/weighted_lm_loss": 2.783090829849243, + "objective/train/weights_max": 1.676503300666809, + "objective/train/weights_min": 0.30571192502975464, + "theoretical_loss": 3.7513988623232883, + "tokens_seen": 753664000 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003897492477432297, + "loss": 1.3736, + "theoretical_loss": 3.7513988623232883, + "tokens_seen": 753664000 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038973921765295886, + "loss": 1.5745, + "theoretical_loss": 3.7513661771381366, + "tokens_seen": 753729536 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038972918756268805, + "loss": 1.3445, + "theoretical_loss": 3.751333495590469, + "tokens_seen": 753795072 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003897191574724173, + "loss": 1.4072, + "theoretical_loss": 3.7513008176795646, + "tokens_seen": 753860608 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003897091273821464, + "loss": 1.4856, + "theoretical_loss": 3.7512681434047033, + "tokens_seen": 753926144 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038969909729187564, + "loss": 1.4306, + "theoretical_loss": 3.7512354727651642, + "tokens_seen": 753991680 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038968906720160477, + "loss": 1.4928, + "theoretical_loss": 3.7512028057602267, + "tokens_seen": 754057216 + }, + { + "epoch": 0.23, + "learning_rate": 0.000389679037111334, + "loss": 1.4326, + "theoretical_loss": 3.7511701423891703, + "tokens_seen": 754122752 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003896690070210632, + "loss": 1.3187, + "theoretical_loss": 3.7511374826512753, + "tokens_seen": 754188288 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038965897693079237, + "loss": 1.3242, + "theoretical_loss": 3.7511048265458227, + "tokens_seen": 754253824 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038964894684052155, + "loss": 1.4326, + "theoretical_loss": 3.751072174072092, + "tokens_seen": 754319360 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003896389167502508, + "loss": 1.4528, + "theoretical_loss": 3.751039525229364, + "tokens_seen": 754384896 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038962888665997997, + "loss": 1.4207, + "theoretical_loss": 3.75100688001692, + "tokens_seen": 754450432 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038961885656970915, + "loss": 1.4904, + "theoretical_loss": 3.7509742384340403, + "tokens_seen": 754515968 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038960882647943833, + "loss": 1.4546, + "theoretical_loss": 3.750941600480007, + "tokens_seen": 754581504 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003895987963891675, + "loss": 1.415, + "theoretical_loss": 3.7509089661541006, + "tokens_seen": 754647040 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038958876629889674, + "loss": 1.4653, + "theoretical_loss": 3.750876335455604, + "tokens_seen": 754712576 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038957873620862587, + "loss": 1.4725, + "theoretical_loss": 3.750843708383798, + "tokens_seen": 754778112 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003895687061183551, + "loss": 1.4607, + "theoretical_loss": 3.750811084937965, + "tokens_seen": 754843648 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038955867602808423, + "loss": 1.4072, + "theoretical_loss": 3.750778465117387, + "tokens_seen": 754909184 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038954864593781347, + "loss": 1.4401, + "theoretical_loss": 3.7507458489213477, + "tokens_seen": 754974720 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038953861584754265, + "loss": 1.3629, + "theoretical_loss": 3.750713236349128, + "tokens_seen": 755040256 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038952858575727183, + "loss": 1.4771, + "theoretical_loss": 3.750680627400012, + "tokens_seen": 755105792 + }, + { + "epoch": 0.23, + "learning_rate": 0.000389518555667001, + "loss": 1.3909, + "theoretical_loss": 3.7506480220732823, + "tokens_seen": 755171328 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038950852557673025, + "loss": 1.4175, + "theoretical_loss": 3.7506154203682223, + "tokens_seen": 755236864 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0011791138676926494, + "objective/train/docs_used": 436337, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.627865791320801, + "objective/train/original_loss": 2.6278655529022217, + "objective/train/theoretical_loss": 3.7505828222841155, + "objective/train/tokens_used": 775762400, + "objective/train/value_avg": -0.006664276123046875, + "objective/train/value_loss": 8.39463755255565e-05, + "objective/train/value_max": -0.00042557716369628906, + "objective/train/value_min": -0.0855712890625, + "objective/train/value_reward_corr": 0.30094147865485754, + "objective/train/value_std": 0.00527191162109375, + "objective/train/weight_avg": 0.9921740293502808, + "objective/train/weighted_lm_loss": 2.6159675121307373, + "objective/train/weights_max": 2.1202759742736816, + "objective/train/weights_min": 0.37932270765304565, + "theoretical_loss": 3.7505828222841155, + "tokens_seen": 755302400 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003894984954864594, + "loss": 1.3533, + "theoretical_loss": 3.7505828222841155, + "tokens_seen": 755302400 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003894884653961886, + "loss": 1.4477, + "theoretical_loss": 3.750550227820246, + "tokens_seen": 755367936 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038947843530591774, + "loss": 1.5101, + "theoretical_loss": 3.7505176369758972, + "tokens_seen": 755433472 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038946840521564697, + "loss": 1.309, + "theoretical_loss": 3.750485049750353, + "tokens_seen": 755499008 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038945837512537615, + "loss": 1.3593, + "theoretical_loss": 3.7504524661428986, + "tokens_seen": 755564544 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038944834503510533, + "loss": 1.5292, + "theoretical_loss": 3.7504198861528177, + "tokens_seen": 755630080 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003894383149448345, + "loss": 1.4721, + "theoretical_loss": 3.750387309779395, + "tokens_seen": 755695616 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003894282848545637, + "loss": 1.484, + "theoretical_loss": 3.750354737021916, + "tokens_seen": 755761152 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003894182547642929, + "loss": 1.4403, + "theoretical_loss": 3.750322167879666, + "tokens_seen": 755826688 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003894082246740221, + "loss": 1.4752, + "theoretical_loss": 3.7502896023519297, + "tokens_seen": 755892224 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038939819458375124, + "loss": 1.3975, + "theoretical_loss": 3.7502570404379925, + "tokens_seen": 755957760 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003893881644934805, + "loss": 1.5023, + "theoretical_loss": 3.7502244821371407, + "tokens_seen": 756023296 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003893781344032096, + "loss": 1.4614, + "theoretical_loss": 3.75019192744866, + "tokens_seen": 756088832 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038936810431293884, + "loss": 1.3584, + "theoretical_loss": 3.750159376371837, + "tokens_seen": 756154368 + }, + { + "epoch": 0.23, + "learning_rate": 0.000389358074222668, + "loss": 1.497, + "theoretical_loss": 3.7501268289059566, + "tokens_seen": 756219904 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003893480441323972, + "loss": 1.5317, + "theoretical_loss": 3.7500942850503067, + "tokens_seen": 756285440 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003893380140421264, + "loss": 1.3692, + "theoretical_loss": 3.7500617448041744, + "tokens_seen": 756350976 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003893279839518556, + "loss": 1.4016, + "theoretical_loss": 3.750029208166845, + "tokens_seen": 756416512 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038931795386158474, + "loss": 1.3917, + "theoretical_loss": 3.7499966751376075, + "tokens_seen": 756482048 + }, + { + "epoch": 0.23, + "learning_rate": 0.000389307923771314, + "loss": 1.4903, + "theoretical_loss": 3.749964145715748, + "tokens_seen": 756547584 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003892978936810431, + "loss": 1.5575, + "theoretical_loss": 3.7499316199005546, + "tokens_seen": 756613120 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038928786359077234, + "loss": 1.5221, + "theoretical_loss": 3.7498990976913147, + "tokens_seen": 756678656 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003892778335005015, + "loss": 1.4508, + "theoretical_loss": 3.7498665790873167, + "tokens_seen": 756744192 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003892678034102307, + "loss": 1.4416, + "theoretical_loss": 3.7498340640878487, + "tokens_seen": 756809728 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003892577733199599, + "loss": 1.4659, + "theoretical_loss": 3.7498015526921984, + "tokens_seen": 756875264 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0005488164606504142, + "objective/train/docs_used": 436904, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.182206630706787, + "objective/train/original_loss": 3.182206630706787, + "objective/train/theoretical_loss": 3.7497690448996552, + "objective/train/tokens_used": 777400800, + "objective/train/value_avg": -0.00656890869140625, + "objective/train/value_loss": 6.879382999613881e-05, + "objective/train/value_max": -0.0003554821014404297, + "objective/train/value_min": -0.090087890625, + "objective/train/value_reward_corr": 0.5066786544819706, + "objective/train/value_std": 0.006298065185546875, + "objective/train/weight_avg": 0.9977819919586182, + "objective/train/weighted_lm_loss": 3.183567523956299, + "objective/train/weights_max": 2.4218029975891113, + "objective/train/weights_min": 0.23275382816791534, + "theoretical_loss": 3.7497690448996552, + "tokens_seen": 756940800 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038924774322968906, + "loss": 1.4436, + "theoretical_loss": 3.7497690448996552, + "tokens_seen": 756940800 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038923771313941825, + "loss": 1.4167, + "theoretical_loss": 3.749736540709508, + "tokens_seen": 757006336 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003892276830491475, + "loss": 1.4381, + "theoretical_loss": 3.7497040401210446, + "tokens_seen": 757071872 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003892176529588766, + "loss": 1.4095, + "theoretical_loss": 3.749671543133555, + "tokens_seen": 757137408 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038920762286860584, + "loss": 1.5324, + "theoretical_loss": 3.749639049746329, + "tokens_seen": 757202944 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038919759277833497, + "loss": 1.3839, + "theoretical_loss": 3.7496065599586554, + "tokens_seen": 757268480 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003891875626880642, + "loss": 1.3764, + "theoretical_loss": 3.7495740737698244, + "tokens_seen": 757334016 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003891775325977934, + "loss": 1.395, + "theoretical_loss": 3.749541591179126, + "tokens_seen": 757399552 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038916750250752257, + "loss": 1.392, + "theoretical_loss": 3.7495091121858497, + "tokens_seen": 757465088 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038915747241725175, + "loss": 1.607, + "theoretical_loss": 3.7494766367892876, + "tokens_seen": 757530624 + }, + { + "epoch": 0.23, + "learning_rate": 0.000389147442326981, + "loss": 1.4255, + "theoretical_loss": 3.749444164988728, + "tokens_seen": 757596160 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003891374122367101, + "loss": 1.3319, + "theoretical_loss": 3.749411696783463, + "tokens_seen": 757661696 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038912738214643935, + "loss": 1.6697, + "theoretical_loss": 3.7493792321727843, + "tokens_seen": 757727232 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003891173520561685, + "loss": 1.4723, + "theoretical_loss": 3.749346771155982, + "tokens_seen": 757792768 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003891073219658977, + "loss": 1.3485, + "theoretical_loss": 3.749314313732347, + "tokens_seen": 757858304 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003890972918756269, + "loss": 1.4598, + "theoretical_loss": 3.749281859901172, + "tokens_seen": 757923840 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038908726178535607, + "loss": 1.3858, + "theoretical_loss": 3.749249409661748, + "tokens_seen": 757989376 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038907723169508525, + "loss": 1.4988, + "theoretical_loss": 3.7492169630133683, + "tokens_seen": 758054912 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038906720160481443, + "loss": 1.3701, + "theoretical_loss": 3.7491845199553238, + "tokens_seen": 758120448 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003890571715145436, + "loss": 1.4658, + "theoretical_loss": 3.7491520804869074, + "tokens_seen": 758185984 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038904714142427285, + "loss": 1.491, + "theoretical_loss": 3.7491196446074113, + "tokens_seen": 758251520 + }, + { + "epoch": 0.23, + "learning_rate": 0.000389037111334002, + "loss": 1.4534, + "theoretical_loss": 3.7490872123161285, + "tokens_seen": 758317056 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003890270812437312, + "loss": 1.4272, + "theoretical_loss": 3.749054783612353, + "tokens_seen": 758382592 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003890170511534604, + "loss": 1.4283, + "theoretical_loss": 3.7490223584953766, + "tokens_seen": 758448128 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003890070210631896, + "loss": 1.4377, + "theoretical_loss": 3.748989936964493, + "tokens_seen": 758513664 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0018802577396854758, + "objective/train/docs_used": 438157, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.802485227584839, + "objective/train/original_loss": 2.802485227584839, + "objective/train/theoretical_loss": 3.7489575190189965, + "objective/train/tokens_used": 779039200, + "objective/train/value_avg": -0.004730224609375, + "objective/train/value_loss": 6.148443935671821e-05, + "objective/train/value_max": -0.0004673004150390625, + "objective/train/value_min": -0.062225341796875, + "objective/train/value_reward_corr": 0.47396864085448254, + "objective/train/value_std": 0.003948211669921875, + "objective/train/weight_avg": 0.9840568900108337, + "objective/train/weighted_lm_loss": 2.756725788116455, + "objective/train/weights_max": 1.5294145345687866, + "objective/train/weights_min": 0.4193553626537323, + "theoretical_loss": 3.7489575190189965, + "tokens_seen": 758579200 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038899699097291876, + "loss": 1.3906, + "theoretical_loss": 3.7489575190189965, + "tokens_seen": 758579200 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038898696088264794, + "loss": 1.3822, + "theoretical_loss": 3.7489251046581797, + "tokens_seen": 758644736 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003889769307923771, + "loss": 1.4234, + "theoretical_loss": 3.7488926938813383, + "tokens_seen": 758710272 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038896690070210635, + "loss": 1.4472, + "theoretical_loss": 3.748860286687765, + "tokens_seen": 758775808 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003889568706118355, + "loss": 1.3991, + "theoretical_loss": 3.748827883076755, + "tokens_seen": 758841344 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003889468405215647, + "loss": 1.3589, + "theoretical_loss": 3.7487954830476022, + "tokens_seen": 758906880 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038893681043129384, + "loss": 1.3316, + "theoretical_loss": 3.7487630865996024, + "tokens_seen": 758972416 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003889267803410231, + "loss": 1.3648, + "theoretical_loss": 3.7487306937320497, + "tokens_seen": 759037952 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038891675025075226, + "loss": 1.4633, + "theoretical_loss": 3.7486983044442397, + "tokens_seen": 759103488 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038890672016048144, + "loss": 1.4699, + "theoretical_loss": 3.748665918735468, + "tokens_seen": 759169024 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003888966900702106, + "loss": 1.4221, + "theoretical_loss": 3.74863353660503, + "tokens_seen": 759234560 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003888866599799398, + "loss": 1.4468, + "theoretical_loss": 3.7486011580522214, + "tokens_seen": 759300096 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038887662988966904, + "loss": 1.5607, + "theoretical_loss": 3.7485687830763377, + "tokens_seen": 759365632 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003888665997993982, + "loss": 1.421, + "theoretical_loss": 3.7485364116766764, + "tokens_seen": 759431168 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003888565697091274, + "loss": 1.4472, + "theoretical_loss": 3.7485040438525328, + "tokens_seen": 759496704 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003888465396188566, + "loss": 1.4427, + "theoretical_loss": 3.748471679603204, + "tokens_seen": 759562240 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003888365095285858, + "loss": 1.4612, + "theoretical_loss": 3.7484393189279865, + "tokens_seen": 759627776 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038882647943831494, + "loss": 1.4647, + "theoretical_loss": 3.7484069618261775, + "tokens_seen": 759693312 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003888164493480442, + "loss": 1.4608, + "theoretical_loss": 3.748374608297074, + "tokens_seen": 759758848 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003888064192577733, + "loss": 1.4164, + "theoretical_loss": 3.7483422583399735, + "tokens_seen": 759824384 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038879638916750254, + "loss": 1.4427, + "theoretical_loss": 3.7483099119541743, + "tokens_seen": 759889920 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003887863590772317, + "loss": 1.3953, + "theoretical_loss": 3.748277569138973, + "tokens_seen": 759955456 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003887763289869609, + "loss": 1.4143, + "theoretical_loss": 3.7482452298936684, + "tokens_seen": 760020992 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003887662988966901, + "loss": 1.4607, + "theoretical_loss": 3.7482128942175574, + "tokens_seen": 760086528 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038875626880641927, + "loss": 1.3886, + "theoretical_loss": 3.7481805621099404, + "tokens_seen": 760152064 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0007740834844298661, + "objective/train/docs_used": 438864, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.707430362701416, + "objective/train/original_loss": 2.707430362701416, + "objective/train/theoretical_loss": 3.748148233570115, + "objective/train/tokens_used": 780677600, + "objective/train/value_avg": -0.005550384521484375, + "objective/train/value_loss": 6.663044769084081e-05, + "objective/train/value_max": -0.00022172927856445312, + "objective/train/value_min": -0.06427001953125, + "objective/train/value_reward_corr": 0.43293535357023416, + "objective/train/value_std": 0.004604339599609375, + "objective/train/weight_avg": 0.9954140186309814, + "objective/train/weighted_lm_loss": 2.6935977935791016, + "objective/train/weights_max": 1.4942305088043213, + "objective/train/weights_min": 0.3076539635658264, + "theoretical_loss": 3.748148233570115, + "tokens_seen": 760217600 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038874623871614845, + "loss": 1.3541, + "theoretical_loss": 3.748148233570115, + "tokens_seen": 760217600 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003887362086258777, + "loss": 1.3799, + "theoretical_loss": 3.7481159085973794, + "tokens_seen": 760283136 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003887261785356068, + "loss": 1.5534, + "theoretical_loss": 3.7480835871910334, + "tokens_seen": 760348672 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038871614844533604, + "loss": 1.5441, + "theoretical_loss": 3.748051269350376, + "tokens_seen": 760414208 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038870611835506517, + "loss": 1.4501, + "theoretical_loss": 3.7480189550747065, + "tokens_seen": 760479744 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003886960882647944, + "loss": 1.4001, + "theoretical_loss": 3.7479866443633245, + "tokens_seen": 760545280 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003886860581745236, + "loss": 1.4338, + "theoretical_loss": 3.7479543372155293, + "tokens_seen": 760610816 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038867602808425277, + "loss": 1.5092, + "theoretical_loss": 3.747922033630622, + "tokens_seen": 760676352 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038866599799398195, + "loss": 1.4212, + "theoretical_loss": 3.7478897336079013, + "tokens_seen": 760741888 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003886559679037112, + "loss": 1.4075, + "theoretical_loss": 3.747857437146669, + "tokens_seen": 760807424 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003886459378134403, + "loss": 1.5217, + "theoretical_loss": 3.7478251442462245, + "tokens_seen": 760872960 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038863590772316955, + "loss": 1.4485, + "theoretical_loss": 3.747792854905869, + "tokens_seen": 760938496 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003886258776328987, + "loss": 1.3845, + "theoretical_loss": 3.7477605691249045, + "tokens_seen": 761004032 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003886158475426279, + "loss": 1.4015, + "theoretical_loss": 3.7477282869026305, + "tokens_seen": 761069568 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003886058174523571, + "loss": 1.4714, + "theoretical_loss": 3.747696008238349, + "tokens_seen": 761135104 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038859578736208627, + "loss": 1.3571, + "theoretical_loss": 3.7476637331313616, + "tokens_seen": 761200640 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038858575727181545, + "loss": 1.4203, + "theoretical_loss": 3.74763146158097, + "tokens_seen": 761266176 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038857572718154463, + "loss": 1.5326, + "theoretical_loss": 3.7475991935864768, + "tokens_seen": 761331712 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003885656970912738, + "loss": 1.3667, + "theoretical_loss": 3.747566929147183, + "tokens_seen": 761397248 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038855566700100305, + "loss": 1.5645, + "theoretical_loss": 3.747534668262391, + "tokens_seen": 761462784 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003885456369107322, + "loss": 1.3944, + "theoretical_loss": 3.7475024109314043, + "tokens_seen": 761528320 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003885356068204614, + "loss": 1.4346, + "theoretical_loss": 3.7474701571535256, + "tokens_seen": 761593856 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003885255767301906, + "loss": 1.4109, + "theoretical_loss": 3.747437906928057, + "tokens_seen": 761659392 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003885155466399198, + "loss": 1.4147, + "theoretical_loss": 3.747405660254302, + "tokens_seen": 761724928 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038850551654964896, + "loss": 1.4096, + "theoretical_loss": 3.747373417131564, + "tokens_seen": 761790464 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.00039207213558256626, + "objective/train/docs_used": 439620, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.074498414993286, + "objective/train/original_loss": 3.074497938156128, + "objective/train/theoretical_loss": 3.7473411775591465, + "objective/train/tokens_used": 782316000, + "objective/train/value_avg": -0.00568389892578125, + "objective/train/value_loss": 6.12423536949791e-05, + "objective/train/value_max": -0.00019407272338867188, + "objective/train/value_min": -0.09844970703125, + "objective/train/value_reward_corr": 0.5605614554659318, + "objective/train/value_std": 0.005008697509765625, + "objective/train/weight_avg": 0.9988999962806702, + "objective/train/weighted_lm_loss": 3.059377908706665, + "objective/train/weights_max": 1.8254045248031616, + "objective/train/weights_min": 0.3011974096298218, + "theoretical_loss": 3.7473411775591465, + "tokens_seen": 761856000 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038849548645937814, + "loss": 1.4248, + "theoretical_loss": 3.7473411775591465, + "tokens_seen": 761856000 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003884854563691073, + "loss": 1.5768, + "theoretical_loss": 3.747308941536353, + "tokens_seen": 761921536 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038847542627883655, + "loss": 1.3405, + "theoretical_loss": 3.7472767090624877, + "tokens_seen": 761987072 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003884653961885657, + "loss": 1.452, + "theoretical_loss": 3.747244480136854, + "tokens_seen": 762052608 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003884553660982949, + "loss": 1.4728, + "theoretical_loss": 3.7472122547587574, + "tokens_seen": 762118144 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038844533600802404, + "loss": 1.3872, + "theoretical_loss": 3.7471800329275022, + "tokens_seen": 762183680 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003884353059177533, + "loss": 1.3535, + "theoretical_loss": 3.747147814642392, + "tokens_seen": 762249216 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038842527582748246, + "loss": 1.4823, + "theoretical_loss": 3.747115599902733, + "tokens_seen": 762314752 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038841524573721164, + "loss": 1.3938, + "theoretical_loss": 3.747083388707829, + "tokens_seen": 762380288 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003884052156469408, + "loss": 1.457, + "theoretical_loss": 3.747051181056986, + "tokens_seen": 762445824 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038839518555667, + "loss": 1.4637, + "theoretical_loss": 3.7470189769495104, + "tokens_seen": 762511360 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003883851554663992, + "loss": 1.3488, + "theoretical_loss": 3.746986776384706, + "tokens_seen": 762576896 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003883751253761284, + "loss": 1.4479, + "theoretical_loss": 3.74695457936188, + "tokens_seen": 762642432 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038836509528585755, + "loss": 1.3006, + "theoretical_loss": 3.746922385880338, + "tokens_seen": 762707968 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003883550651955868, + "loss": 1.4535, + "theoretical_loss": 3.746890195939386, + "tokens_seen": 762773504 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038834503510531596, + "loss": 1.409, + "theoretical_loss": 3.7468580095383315, + "tokens_seen": 762839040 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038833500501504514, + "loss": 1.4104, + "theoretical_loss": 3.7468258266764796, + "tokens_seen": 762904576 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003883249749247743, + "loss": 1.4189, + "theoretical_loss": 3.746793647353139, + "tokens_seen": 762970112 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003883149448345035, + "loss": 1.5008, + "theoretical_loss": 3.7467614715676154, + "tokens_seen": 763035648 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003883049147442327, + "loss": 1.4404, + "theoretical_loss": 3.746729299319216, + "tokens_seen": 763101184 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003882948846539619, + "loss": 1.4247, + "theoretical_loss": 3.7466971306072487, + "tokens_seen": 763166720 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038828485456369105, + "loss": 1.5083, + "theoretical_loss": 3.7466649654310213, + "tokens_seen": 763232256 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003882748244734203, + "loss": 1.3932, + "theoretical_loss": 3.7466328037898418, + "tokens_seen": 763297792 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003882647943831494, + "loss": 1.3811, + "theoretical_loss": 3.746600645683017, + "tokens_seen": 763363328 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038825476429287865, + "loss": 1.3781, + "theoretical_loss": 3.7465684911098567, + "tokens_seen": 763428864 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0004375773132778704, + "objective/train/docs_used": 440741, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7715656757354736, + "objective/train/original_loss": 2.7715654373168945, + "objective/train/theoretical_loss": 3.7465363400696683, + "objective/train/tokens_used": 783954400, + "objective/train/value_avg": -0.005199432373046875, + "objective/train/value_loss": 5.9086818509968e-05, + "objective/train/value_max": -0.0005092620849609375, + "objective/train/value_min": -0.06427001953125, + "objective/train/value_reward_corr": 0.4459853858519888, + "objective/train/value_std": 0.00423431396484375, + "objective/train/weight_avg": 0.9982956647872925, + "objective/train/weighted_lm_loss": 2.772562026977539, + "objective/train/weights_max": 1.5826791524887085, + "objective/train/weights_min": 0.1457751840353012, + "theoretical_loss": 3.7465363400696683, + "tokens_seen": 763494400 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038824473420260783, + "loss": 1.4019, + "theoretical_loss": 3.7465363400696683, + "tokens_seen": 763494400 + }, + { + "epoch": 0.23, + "learning_rate": 0.000388234704112337, + "loss": 1.4476, + "theoretical_loss": 3.74650419256176, + "tokens_seen": 763559936 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003882246740220662, + "loss": 1.4253, + "theoretical_loss": 3.746472048585442, + "tokens_seen": 763625472 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038821464393179537, + "loss": 1.2603, + "theoretical_loss": 3.7464399081400224, + "tokens_seen": 763691008 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038820461384152455, + "loss": 1.4863, + "theoretical_loss": 3.7464077712248107, + "tokens_seen": 763756544 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003881945837512538, + "loss": 1.4234, + "theoretical_loss": 3.7463756378391158, + "tokens_seen": 763822080 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003881845536609829, + "loss": 1.3862, + "theoretical_loss": 3.746343507982248, + "tokens_seen": 763887616 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038817452357071215, + "loss": 1.393, + "theoretical_loss": 3.7463113816535163, + "tokens_seen": 763953152 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038816449348044133, + "loss": 1.5604, + "theoretical_loss": 3.746279258852231, + "tokens_seen": 764018688 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003881544633901705, + "loss": 1.2887, + "theoretical_loss": 3.7462471395777026, + "tokens_seen": 764084224 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003881444332998997, + "loss": 1.4632, + "theoretical_loss": 3.746215023829241, + "tokens_seen": 764149760 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003881344032096289, + "loss": 1.4375, + "theoretical_loss": 3.746182911606157, + "tokens_seen": 764215296 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003881243731193581, + "loss": 1.361, + "theoretical_loss": 3.7461508029077613, + "tokens_seen": 764280832 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003881143430290873, + "loss": 1.4479, + "theoretical_loss": 3.7461186977333645, + "tokens_seen": 764346368 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038810431293881647, + "loss": 1.4218, + "theoretical_loss": 3.7460865960822782, + "tokens_seen": 764411904 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038809428284854565, + "loss": 1.3653, + "theoretical_loss": 3.7460544979538137, + "tokens_seen": 764477440 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038808425275827483, + "loss": 1.4591, + "theoretical_loss": 3.7460224033472826, + "tokens_seen": 764542976 + }, + { + "epoch": 0.23, + "learning_rate": 0.000388074222668004, + "loss": 1.4803, + "theoretical_loss": 3.7459903122619957, + "tokens_seen": 764608512 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038806419257773325, + "loss": 1.4582, + "theoretical_loss": 3.745958224697266, + "tokens_seen": 764674048 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003880541624874624, + "loss": 1.3645, + "theoretical_loss": 3.745926140652405, + "tokens_seen": 764739584 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003880441323971916, + "loss": 1.3831, + "theoretical_loss": 3.7458940601267257, + "tokens_seen": 764805120 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003880341023069208, + "loss": 1.4191, + "theoretical_loss": 3.745861983119539, + "tokens_seen": 764870656 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038802407221665, + "loss": 1.5098, + "theoretical_loss": 3.7458299096301597, + "tokens_seen": 764936192 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038801404212637916, + "loss": 1.3361, + "theoretical_loss": 3.7457978396578993, + "tokens_seen": 765001728 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038800401203610834, + "loss": 1.4193, + "theoretical_loss": 3.7457657732020713, + "tokens_seen": 765067264 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0004996819770894945, + "objective/train/docs_used": 441342, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8368518352508545, + "objective/train/original_loss": 2.8368513584136963, + "objective/train/theoretical_loss": 3.7457337102619888, + "objective/train/tokens_used": 785592800, + "objective/train/value_avg": -0.005748748779296875, + "objective/train/value_loss": 5.693536877515726e-05, + "objective/train/value_max": -0.00034332275390625, + "objective/train/value_min": -0.060760498046875, + "objective/train/value_reward_corr": 0.4508354003493356, + "objective/train/value_std": 0.00494384765625, + "objective/train/weight_avg": 0.9976741075515747, + "objective/train/weighted_lm_loss": 2.8293402194976807, + "objective/train/weights_max": 1.6959775686264038, + "objective/train/weights_min": 0.294359415769577, + "theoretical_loss": 3.7457337102619888, + "tokens_seen": 765132800 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003879939819458375, + "loss": 1.4108, + "theoretical_loss": 3.7457337102619888, + "tokens_seen": 765132800 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038798395185556675, + "loss": 1.444, + "theoretical_loss": 3.745701650836965, + "tokens_seen": 765198336 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003879739217652959, + "loss": 1.4067, + "theoretical_loss": 3.745669594926314, + "tokens_seen": 765263872 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003879638916750251, + "loss": 1.2624, + "theoretical_loss": 3.7456375425293493, + "tokens_seen": 765329408 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038795386158475424, + "loss": 1.432, + "theoretical_loss": 3.7456054936453853, + "tokens_seen": 765394944 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003879438314944835, + "loss": 1.3344, + "theoretical_loss": 3.745573448273736, + "tokens_seen": 765460480 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038793380140421266, + "loss": 1.4169, + "theoretical_loss": 3.745541406413716, + "tokens_seen": 765526016 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038792377131394184, + "loss": 1.5553, + "theoretical_loss": 3.7455093680646394, + "tokens_seen": 765591552 + }, + { + "epoch": 0.23, + "learning_rate": 0.000387913741223671, + "loss": 1.4428, + "theoretical_loss": 3.7454773332258213, + "tokens_seen": 765657088 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003879037111334002, + "loss": 1.4194, + "theoretical_loss": 3.745445301896577, + "tokens_seen": 765722624 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003878936810431294, + "loss": 1.5412, + "theoretical_loss": 3.7454132740762214, + "tokens_seen": 765788160 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003878836509528586, + "loss": 1.4182, + "theoretical_loss": 3.74538124976407, + "tokens_seen": 765853696 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038787362086258775, + "loss": 1.4239, + "theoretical_loss": 3.745349228959438, + "tokens_seen": 765919232 + }, + { + "epoch": 0.23, + "learning_rate": 0.000387863590772317, + "loss": 1.4194, + "theoretical_loss": 3.745317211661642, + "tokens_seen": 765984768 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038785356068204616, + "loss": 1.4499, + "theoretical_loss": 3.7452851978699964, + "tokens_seen": 766050304 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038784353059177534, + "loss": 1.3088, + "theoretical_loss": 3.7452531875838186, + "tokens_seen": 766115840 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003878335005015045, + "loss": 1.5393, + "theoretical_loss": 3.7452211808024254, + "tokens_seen": 766181376 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003878234704112337, + "loss": 1.363, + "theoretical_loss": 3.7451891775251323, + "tokens_seen": 766246912 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003878134403209629, + "loss": 1.3769, + "theoretical_loss": 3.7451571777512553, + "tokens_seen": 766312448 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003878034102306921, + "loss": 1.443, + "theoretical_loss": 3.7451251814801134, + "tokens_seen": 766377984 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038779338014042125, + "loss": 1.4062, + "theoretical_loss": 3.7450931887110226, + "tokens_seen": 766443520 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003877833500501505, + "loss": 1.4015, + "theoretical_loss": 3.7450611994433, + "tokens_seen": 766509056 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003877733199598796, + "loss": 1.4489, + "theoretical_loss": 3.745029213676263, + "tokens_seen": 766574592 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038776328986960885, + "loss": 1.4355, + "theoretical_loss": 3.7449972314092297, + "tokens_seen": 766640128 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038775325977933803, + "loss": 1.2893, + "theoretical_loss": 3.744965252641518, + "tokens_seen": 766705664 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0009724645060487092, + "objective/train/docs_used": 442736, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.306094169616699, + "objective/train/original_loss": 3.3060944080352783, + "objective/train/theoretical_loss": 3.7449332773724455, + "objective/train/tokens_used": 787231200, + "objective/train/value_avg": -0.004894256591796875, + "objective/train/value_loss": 4.553080361802131e-05, + "objective/train/value_max": -0.0003981590270996094, + "objective/train/value_min": -0.053192138671875, + "objective/train/value_reward_corr": 0.4460542242332949, + "objective/train/value_std": 0.0038967132568359375, + "objective/train/weight_avg": 0.992472231388092, + "objective/train/weighted_lm_loss": 3.2825405597686768, + "objective/train/weights_max": 1.5856579542160034, + "objective/train/weights_min": 0.5725141763687134, + "theoretical_loss": 3.7449332773724455, + "tokens_seen": 766771200 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003877432296890672, + "loss": 1.4866, + "theoretical_loss": 3.7449332773724455, + "tokens_seen": 766771200 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003877331995987964, + "loss": 1.3954, + "theoretical_loss": 3.7449013056013314, + "tokens_seen": 766836736 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038772316950852557, + "loss": 1.3879, + "theoretical_loss": 3.744869337327493, + "tokens_seen": 766902272 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038771313941825475, + "loss": 1.4714, + "theoretical_loss": 3.7448373725502497, + "tokens_seen": 766967808 + }, + { + "epoch": 0.23, + "learning_rate": 0.000387703109327984, + "loss": 1.5465, + "theoretical_loss": 3.7448054112689197, + "tokens_seen": 767033344 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003876930792377131, + "loss": 1.4179, + "theoretical_loss": 3.7447734534828223, + "tokens_seen": 767098880 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038768304914744235, + "loss": 1.5153, + "theoretical_loss": 3.744741499191277, + "tokens_seen": 767164416 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038767301905717153, + "loss": 1.4418, + "theoretical_loss": 3.7447095483936037, + "tokens_seen": 767229952 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003876629889669007, + "loss": 1.3716, + "theoretical_loss": 3.74467760108912, + "tokens_seen": 767295488 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003876529588766299, + "loss": 1.3688, + "theoretical_loss": 3.7446456572771476, + "tokens_seen": 767361024 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003876429287863591, + "loss": 1.3702, + "theoretical_loss": 3.744613716957006, + "tokens_seen": 767426560 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038763289869608826, + "loss": 1.366, + "theoretical_loss": 3.744581780128015, + "tokens_seen": 767492096 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003876228686058175, + "loss": 1.3736, + "theoretical_loss": 3.7445498467894947, + "tokens_seen": 767557632 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003876128385155466, + "loss": 1.4397, + "theoretical_loss": 3.7445179169407665, + "tokens_seen": 767623168 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038760280842527585, + "loss": 1.4996, + "theoretical_loss": 3.7444859905811505, + "tokens_seen": 767688704 + }, + { + "epoch": 0.23, + "learning_rate": 0.000387592778335005, + "loss": 1.4414, + "theoretical_loss": 3.7444540677099676, + "tokens_seen": 767754240 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003875827482447342, + "loss": 1.4169, + "theoretical_loss": 3.7444221483265396, + "tokens_seen": 767819776 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003875727181544634, + "loss": 1.3157, + "theoretical_loss": 3.744390232430187, + "tokens_seen": 767885312 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003875626880641926, + "loss": 1.3797, + "theoretical_loss": 3.7443583200202317, + "tokens_seen": 767950848 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038755265797392176, + "loss": 1.4523, + "theoretical_loss": 3.7443264110959955, + "tokens_seen": 768016384 + }, + { + "epoch": 0.23, + "learning_rate": 0.000387542627883651, + "loss": 1.4166, + "theoretical_loss": 3.7442945056568, + "tokens_seen": 768081920 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003875325977933801, + "loss": 1.2827, + "theoretical_loss": 3.7442626037019675, + "tokens_seen": 768147456 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038752256770310936, + "loss": 1.4057, + "theoretical_loss": 3.7442307052308195, + "tokens_seen": 768212992 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003875125376128385, + "loss": 1.5264, + "theoretical_loss": 3.744198810242679, + "tokens_seen": 768278528 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003875025075225677, + "loss": 1.356, + "theoretical_loss": 3.744166918736869, + "tokens_seen": 768344064 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0003047626814804971, + "objective/train/docs_used": 443303, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5189547538757324, + "objective/train/original_loss": 2.5189547538757324, + "objective/train/theoretical_loss": 3.7441350307127115, + "objective/train/tokens_used": 788869600, + "objective/train/value_avg": -0.005237579345703125, + "objective/train/value_loss": 6.13098090980202e-05, + "objective/train/value_max": -0.00031495094299316406, + "objective/train/value_min": -0.0843505859375, + "objective/train/value_reward_corr": 0.31678438835268286, + "objective/train/value_std": 0.004543304443359375, + "objective/train/weight_avg": 0.999779462814331, + "objective/train/weighted_lm_loss": 2.5234153270721436, + "objective/train/weights_max": 2.164295196533203, + "objective/train/weights_min": 0.254656046628952, + "theoretical_loss": 3.7441350307127115, + "tokens_seen": 768409600 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003874924774322969, + "loss": 1.3749, + "theoretical_loss": 3.7441350307127115, + "tokens_seen": 768409600 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003874824473420261, + "loss": 1.5371, + "theoretical_loss": 3.7441031461695307, + "tokens_seen": 768475136 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038747241725175526, + "loss": 1.3911, + "theoretical_loss": 3.7440712651066486, + "tokens_seen": 768540672 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038746238716148444, + "loss": 1.4431, + "theoretical_loss": 3.7440393875233893, + "tokens_seen": 768606208 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003874523570712136, + "loss": 1.377, + "theoretical_loss": 3.744007513419076, + "tokens_seen": 768671744 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038744232698094286, + "loss": 1.4012, + "theoretical_loss": 3.743975642793032, + "tokens_seen": 768737280 + }, + { + "epoch": 0.23, + "learning_rate": 0.000387432296890672, + "loss": 1.3555, + "theoretical_loss": 3.743943775644582, + "tokens_seen": 768802816 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003874222668004012, + "loss": 1.4709, + "theoretical_loss": 3.7439119119730506, + "tokens_seen": 768868352 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038741223671013035, + "loss": 1.3656, + "theoretical_loss": 3.7438800517777615, + "tokens_seen": 768933888 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003874022066198596, + "loss": 1.5493, + "theoretical_loss": 3.743848195058039, + "tokens_seen": 768999424 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038739217652958876, + "loss": 1.4205, + "theoretical_loss": 3.7438163418132078, + "tokens_seen": 769064960 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038738214643931795, + "loss": 1.4392, + "theoretical_loss": 3.743784492042593, + "tokens_seen": 769130496 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003873721163490472, + "loss": 1.421, + "theoretical_loss": 3.7437526457455204, + "tokens_seen": 769196032 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038736208625877636, + "loss": 1.3662, + "theoretical_loss": 3.743720802921314, + "tokens_seen": 769261568 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038735205616850554, + "loss": 1.2665, + "theoretical_loss": 3.7436889635693005, + "tokens_seen": 769327104 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003873420260782347, + "loss": 1.3068, + "theoretical_loss": 3.743657127688804, + "tokens_seen": 769392640 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003873319959879639, + "loss": 1.2781, + "theoretical_loss": 3.743625295279152, + "tokens_seen": 769458176 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003873219658976931, + "loss": 1.3687, + "theoretical_loss": 3.7435934663396693, + "tokens_seen": 769523712 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003873119358074223, + "loss": 1.3709, + "theoretical_loss": 3.743561640869683, + "tokens_seen": 769589248 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038730190571715145, + "loss": 1.4376, + "theoretical_loss": 3.7435298188685184, + "tokens_seen": 769654784 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003872918756268807, + "loss": 1.387, + "theoretical_loss": 3.743498000335504, + "tokens_seen": 769720320 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003872818455366098, + "loss": 1.4473, + "theoretical_loss": 3.7434661852699644, + "tokens_seen": 769785856 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038727181544633905, + "loss": 1.5611, + "theoretical_loss": 3.743434373671228, + "tokens_seen": 769851392 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038726178535606823, + "loss": 1.5494, + "theoretical_loss": 3.7434025655386214, + "tokens_seen": 769916928 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003872517552657974, + "loss": 1.3979, + "theoretical_loss": 3.743370760871472, + "tokens_seen": 769982464 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0018917550332844257, + "objective/train/docs_used": 444724, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.731442451477051, + "objective/train/original_loss": 2.7314419746398926, + "objective/train/theoretical_loss": 3.7433389596691073, + "objective/train/tokens_used": 790508000, + "objective/train/value_avg": -0.006664276123046875, + "objective/train/value_loss": 9.447104821447283e-05, + "objective/train/value_max": -0.0003447532653808594, + "objective/train/value_min": -0.09368896484375, + "objective/train/value_reward_corr": 0.47456963845642997, + "objective/train/value_std": 0.006542205810546875, + "objective/train/weight_avg": 0.9854757785797119, + "objective/train/weighted_lm_loss": 2.7074532508850098, + "objective/train/weights_max": 1.926220417022705, + "objective/train/weights_min": 0.25442299246788025, + "theoretical_loss": 3.7433389596691073, + "tokens_seen": 770048000 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003872417251755266, + "loss": 1.4796, + "theoretical_loss": 3.7433389596691073, + "tokens_seen": 770048000 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038723169508525577, + "loss": 1.3438, + "theoretical_loss": 3.7433071619308556, + "tokens_seen": 770113536 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038722166499498495, + "loss": 1.3881, + "theoretical_loss": 3.743275367656044, + "tokens_seen": 770179072 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003872116349047142, + "loss": 1.4078, + "theoretical_loss": 3.7432435768440007, + "tokens_seen": 770244608 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003872016048144433, + "loss": 1.3504, + "theoretical_loss": 3.7432117894940546, + "tokens_seen": 770310144 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038719157472417255, + "loss": 1.4623, + "theoretical_loss": 3.7431800056055335, + "tokens_seen": 770375680 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038718154463390173, + "loss": 1.5518, + "theoretical_loss": 3.743148225177766, + "tokens_seen": 770441216 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003871715145436309, + "loss": 1.3979, + "theoretical_loss": 3.743116448210082, + "tokens_seen": 770506752 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003871614844533601, + "loss": 1.3568, + "theoretical_loss": 3.7430846747018096, + "tokens_seen": 770572288 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003871514543630893, + "loss": 1.4881, + "theoretical_loss": 3.743052904652278, + "tokens_seen": 770637824 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038714142427281846, + "loss": 1.39, + "theoretical_loss": 3.7430211380608167, + "tokens_seen": 770703360 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003871313941825477, + "loss": 1.3659, + "theoretical_loss": 3.742989374926756, + "tokens_seen": 770768896 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003871213640922768, + "loss": 1.3572, + "theoretical_loss": 3.7429576152494244, + "tokens_seen": 770834432 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038711133400200605, + "loss": 1.3885, + "theoretical_loss": 3.742925859028153, + "tokens_seen": 770899968 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003871013039117352, + "loss": 1.4049, + "theoretical_loss": 3.7428941062622716, + "tokens_seen": 770965504 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003870912738214644, + "loss": 1.5609, + "theoretical_loss": 3.74286235695111, + "tokens_seen": 771031040 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003870812437311936, + "loss": 1.4399, + "theoretical_loss": 3.742830611094, + "tokens_seen": 771096576 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003870712136409228, + "loss": 1.3823, + "theoretical_loss": 3.7427988686902713, + "tokens_seen": 771162112 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038706118355065196, + "loss": 1.6132, + "theoretical_loss": 3.7427671297392546, + "tokens_seen": 771227648 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003870511534603812, + "loss": 1.3302, + "theoretical_loss": 3.742735394240282, + "tokens_seen": 771293184 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003870411233701103, + "loss": 1.5016, + "theoretical_loss": 3.7427036621926835, + "tokens_seen": 771358720 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038703109327983956, + "loss": 1.4058, + "theoretical_loss": 3.7426719335957914, + "tokens_seen": 771424256 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003870210631895687, + "loss": 1.4349, + "theoretical_loss": 3.742640208448938, + "tokens_seen": 771489792 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003870110330992979, + "loss": 1.3873, + "theoretical_loss": 3.742608486751453, + "tokens_seen": 771555328 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003870010030090271, + "loss": 1.375, + "theoretical_loss": 3.742576768502671, + "tokens_seen": 771620864 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": 0.0004299404681660235, + "objective/train/docs_used": 445347, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8350563049316406, + "objective/train/original_loss": 2.8350560665130615, + "objective/train/theoretical_loss": 3.7425450537019223, + "objective/train/tokens_used": 792146400, + "objective/train/value_avg": -0.005146026611328125, + "objective/train/value_loss": 5.0654110964387655e-05, + "objective/train/value_max": -0.0003981590270996094, + "objective/train/value_min": -0.09271240234375, + "objective/train/value_reward_corr": 0.3736466760238711, + "objective/train/value_std": 0.003978729248046875, + "objective/train/weight_avg": 1.0066877603530884, + "objective/train/weighted_lm_loss": 2.8525631427764893, + "objective/train/weights_max": 1.847956657409668, + "objective/train/weights_min": 0.29873722791671753, + "theoretical_loss": 3.7425450537019223, + "tokens_seen": 771686400 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003869909729187563, + "loss": 1.382, + "theoretical_loss": 3.7425450537019223, + "tokens_seen": 771686400 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038698094282848546, + "loss": 1.2614, + "theoretical_loss": 3.74251334234854, + "tokens_seen": 771751936 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038697091273821464, + "loss": 1.4965, + "theoretical_loss": 3.742481634441857, + "tokens_seen": 771817472 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003869608826479438, + "loss": 1.4252, + "theoretical_loss": 3.742449929981205, + "tokens_seen": 771883008 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038695085255767306, + "loss": 1.4587, + "theoretical_loss": 3.742418228965918, + "tokens_seen": 771948544 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003869408224674022, + "loss": 1.4155, + "theoretical_loss": 3.742386531395329, + "tokens_seen": 772014080 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003869307923771314, + "loss": 1.3682, + "theoretical_loss": 3.7423548372687705, + "tokens_seen": 772079616 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038692076228686055, + "loss": 1.3351, + "theoretical_loss": 3.7423231465855773, + "tokens_seen": 772145152 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003869107321965898, + "loss": 1.5521, + "theoretical_loss": 3.742291459345082, + "tokens_seen": 772210688 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038690070210631896, + "loss": 1.4565, + "theoretical_loss": 3.7422597755466187, + "tokens_seen": 772276224 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038689067201604815, + "loss": 1.5505, + "theoretical_loss": 3.742228095189522, + "tokens_seen": 772341760 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038688064192577733, + "loss": 1.3882, + "theoretical_loss": 3.7421964182731253, + "tokens_seen": 772407296 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038687061183550656, + "loss": 1.3519, + "theoretical_loss": 3.7421647447967636, + "tokens_seen": 772472832 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003868605817452357, + "loss": 1.5574, + "theoretical_loss": 3.7421330747597716, + "tokens_seen": 772538368 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003868505516549649, + "loss": 1.379, + "theoretical_loss": 3.742101408161484, + "tokens_seen": 772603904 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038684052156469405, + "loss": 1.5902, + "theoretical_loss": 3.7420697450012357, + "tokens_seen": 772669440 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003868304914744233, + "loss": 1.3981, + "theoretical_loss": 3.742038085278362, + "tokens_seen": 772734976 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038682046138415247, + "loss": 1.4474, + "theoretical_loss": 3.742006428992198, + "tokens_seen": 772800512 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038681043129388165, + "loss": 1.3619, + "theoretical_loss": 3.7419747761420794, + "tokens_seen": 772866048 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038680040120361083, + "loss": 1.3682, + "theoretical_loss": 3.7419431267273415, + "tokens_seen": 772931584 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038679037111334, + "loss": 1.459, + "theoretical_loss": 3.741911480747321, + "tokens_seen": 772997120 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003867803410230692, + "loss": 1.4694, + "theoretical_loss": 3.7418798382013536, + "tokens_seen": 773062656 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038677031093279843, + "loss": 1.41, + "theoretical_loss": 3.741848199088776, + "tokens_seen": 773128192 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038676028084252755, + "loss": 1.4637, + "theoretical_loss": 3.741816563408924, + "tokens_seen": 773193728 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003867502507522568, + "loss": 1.4918, + "theoretical_loss": 3.7417849311611344, + "tokens_seen": 773259264 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.00010128489520866424, + "objective/train/docs_used": 446623, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.984121561050415, + "objective/train/original_loss": 2.984121322631836, + "objective/train/theoretical_loss": 3.7417533023447445, + "objective/train/tokens_used": 793784800, + "objective/train/value_avg": -0.006038665771484375, + "objective/train/value_loss": 8.345668902620673e-05, + "objective/train/value_max": -0.0003654956817626953, + "objective/train/value_min": -0.07891845703125, + "objective/train/value_reward_corr": 0.4830468967552865, + "objective/train/value_std": 0.005794525146484375, + "objective/train/weight_avg": 1.002816081047058, + "objective/train/weighted_lm_loss": 3.0010457038879395, + "objective/train/weights_max": 1.6315034627914429, + "objective/train/weights_min": 0.19996781647205353, + "theoretical_loss": 3.7417533023447445, + "tokens_seen": 773324800 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003867402206619859, + "loss": 1.4075, + "theoretical_loss": 3.7417533023447445, + "tokens_seen": 773324800 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038673019057171515, + "loss": 1.2957, + "theoretical_loss": 3.7417216769590906, + "tokens_seen": 773390336 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038672016048144433, + "loss": 1.3229, + "theoretical_loss": 3.7416900550035104, + "tokens_seen": 773455872 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003867101303911735, + "loss": 1.3935, + "theoretical_loss": 3.741658436477341, + "tokens_seen": 773521408 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003867001003009027, + "loss": 1.3369, + "theoretical_loss": 3.7416268213799206, + "tokens_seen": 773586944 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038669007021063193, + "loss": 1.3654, + "theoretical_loss": 3.741595209710586, + "tokens_seen": 773652480 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038668004012036106, + "loss": 1.4372, + "theoretical_loss": 3.741563601468676, + "tokens_seen": 773718016 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003866700100300903, + "loss": 1.4652, + "theoretical_loss": 3.7415319966535288, + "tokens_seen": 773783552 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003866599799398194, + "loss": 1.3732, + "theoretical_loss": 3.741500395264481, + "tokens_seen": 773849088 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038664994984954866, + "loss": 1.4256, + "theoretical_loss": 3.7414687973008736, + "tokens_seen": 773914624 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038663991975927784, + "loss": 1.3705, + "theoretical_loss": 3.741437202762043, + "tokens_seen": 773980160 + }, + { + "epoch": 0.23, + "learning_rate": 0.000386629889669007, + "loss": 1.3491, + "theoretical_loss": 3.74140561164733, + "tokens_seen": 774045696 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038661985957873625, + "loss": 1.2467, + "theoretical_loss": 3.7413740239560727, + "tokens_seen": 774111232 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003866098294884654, + "loss": 1.3713, + "theoretical_loss": 3.7413424396876103, + "tokens_seen": 774176768 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003865997993981946, + "loss": 1.4143, + "theoretical_loss": 3.741310858841282, + "tokens_seen": 774242304 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003865897693079238, + "loss": 1.3475, + "theoretical_loss": 3.7412792814164275, + "tokens_seen": 774307840 + }, + { + "epoch": 0.23, + "learning_rate": 0.000386579739217653, + "loss": 1.5195, + "theoretical_loss": 3.741247707412387, + "tokens_seen": 774373376 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038656970912738216, + "loss": 1.4206, + "theoretical_loss": 3.7412161368285, + "tokens_seen": 774438912 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003865596790371114, + "loss": 1.3327, + "theoretical_loss": 3.7411845696641075, + "tokens_seen": 774504448 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003865496489468405, + "loss": 1.4175, + "theoretical_loss": 3.741153005918549, + "tokens_seen": 774569984 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038653961885656976, + "loss": 1.5618, + "theoretical_loss": 3.7411214455911646, + "tokens_seen": 774635520 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003865295887662989, + "loss": 1.5218, + "theoretical_loss": 3.741089888681296, + "tokens_seen": 774701056 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003865195586760281, + "loss": 1.5081, + "theoretical_loss": 3.741058335188284, + "tokens_seen": 774766592 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003865095285857573, + "loss": 1.4136, + "theoretical_loss": 3.741026785111469, + "tokens_seen": 774832128 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003864994984954865, + "loss": 1.304, + "theoretical_loss": 3.7409952384501923, + "tokens_seen": 774897664 + }, + { + "epoch": 0.23, + "objective/train/advantage_avg": -0.0007445212104357779, + "objective/train/docs_used": 447277, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1752610206604004, + "objective/train/original_loss": 3.1752614974975586, + "objective/train/theoretical_loss": 3.7409636952037957, + "objective/train/tokens_used": 795423200, + "objective/train/value_avg": -0.005245208740234375, + "objective/train/value_loss": 6.398245750460774e-05, + "objective/train/value_max": -0.000293731689453125, + "objective/train/value_min": -0.04931640625, + "objective/train/value_reward_corr": 0.31997368695690653, + "objective/train/value_std": 0.004016876220703125, + "objective/train/weight_avg": 0.9956161975860596, + "objective/train/weighted_lm_loss": 3.180593252182007, + "objective/train/weights_max": 1.4435805082321167, + "objective/train/weights_min": 0.504240095615387, + "theoretical_loss": 3.7409636952037957, + "tokens_seen": 774963200 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038648946840521566, + "loss": 1.4918, + "theoretical_loss": 3.7409636952037957, + "tokens_seen": 774963200 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038647943831494484, + "loss": 1.3935, + "theoretical_loss": 3.740932155371621, + "tokens_seen": 775028736 + }, + { + "epoch": 0.23, + "learning_rate": 0.000386469408224674, + "loss": 1.3521, + "theoretical_loss": 3.7409006189530096, + "tokens_seen": 775094272 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038645937813440326, + "loss": 1.3396, + "theoretical_loss": 3.7408690859473035, + "tokens_seen": 775159808 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003864493480441324, + "loss": 1.4375, + "theoretical_loss": 3.7408375563538447, + "tokens_seen": 775225344 + }, + { + "epoch": 0.23, + "learning_rate": 0.0003864393179538616, + "loss": 1.3553, + "theoretical_loss": 3.740806030171976, + "tokens_seen": 775290880 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038642928786359075, + "loss": 1.3854, + "theoretical_loss": 3.7407745074010395, + "tokens_seen": 775356416 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038641925777332, + "loss": 1.3706, + "theoretical_loss": 3.7407429880403775, + "tokens_seen": 775421952 + }, + { + "epoch": 0.23, + "learning_rate": 0.00038640922768304917, + "loss": 1.3763, + "theoretical_loss": 3.7407114720893344, + "tokens_seen": 775487488 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038639919759277835, + "loss": 1.3838, + "theoretical_loss": 3.740679959547252, + "tokens_seen": 775553024 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038638916750250753, + "loss": 1.427, + "theoretical_loss": 3.740648450413474, + "tokens_seen": 775618560 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038637913741223676, + "loss": 1.373, + "theoretical_loss": 3.7406169446873436, + "tokens_seen": 775684096 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003863691073219659, + "loss": 1.3886, + "theoretical_loss": 3.740585442368204, + "tokens_seen": 775749632 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003863590772316951, + "loss": 1.4709, + "theoretical_loss": 3.7405539434554003, + "tokens_seen": 775815168 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038634904714142425, + "loss": 1.2924, + "theoretical_loss": 3.7405224479482753, + "tokens_seen": 775880704 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003863390170511535, + "loss": 1.3251, + "theoretical_loss": 3.740490955846173, + "tokens_seen": 775946240 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038632898696088267, + "loss": 1.3731, + "theoretical_loss": 3.740459467148439, + "tokens_seen": 776011776 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038631895687061185, + "loss": 1.3666, + "theoretical_loss": 3.740427981854417, + "tokens_seen": 776077312 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038630892678034103, + "loss": 1.3593, + "theoretical_loss": 3.7403964999634516, + "tokens_seen": 776142848 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003862988966900702, + "loss": 1.2822, + "theoretical_loss": 3.7403650214748874, + "tokens_seen": 776208384 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003862888665997994, + "loss": 1.2963, + "theoretical_loss": 3.7403335463880705, + "tokens_seen": 776273920 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038627883650952863, + "loss": 1.535, + "theoretical_loss": 3.7403020747023454, + "tokens_seen": 776339456 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038626880641925775, + "loss": 1.4706, + "theoretical_loss": 3.740270606417058, + "tokens_seen": 776404992 + }, + { + "epoch": 0.24, + "learning_rate": 0.000386258776328987, + "loss": 1.2699, + "theoretical_loss": 3.7402391415315535, + "tokens_seen": 776470528 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003862487462387161, + "loss": 1.445, + "theoretical_loss": 3.740207680045178, + "tokens_seen": 776536064 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": 0.0007070557912811637, + "objective/train/docs_used": 447858, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4551992416381836, + "objective/train/original_loss": 2.4551994800567627, + "objective/train/theoretical_loss": 3.7401762219572765, + "objective/train/tokens_used": 797061600, + "objective/train/value_avg": -0.005161285400390625, + "objective/train/value_loss": 4.468537008506246e-05, + "objective/train/value_max": -0.00014889240264892578, + "objective/train/value_min": -0.1171875, + "objective/train/value_reward_corr": 0.25012271909534667, + "objective/train/value_std": 0.00402069091796875, + "objective/train/weight_avg": 1.0092222690582275, + "objective/train/weighted_lm_loss": 2.482053756713867, + "objective/train/weights_max": 3.228039503097534, + "objective/train/weights_min": 0.2987942099571228, + "theoretical_loss": 3.7401762219572765, + "tokens_seen": 776601600 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038623871614844535, + "loss": 1.4287, + "theoretical_loss": 3.7401762219572765, + "tokens_seen": 776601600 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038622868605817453, + "loss": 1.4628, + "theoretical_loss": 3.740144767267197, + "tokens_seen": 776667136 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003862186559679037, + "loss": 1.3758, + "theoretical_loss": 3.740113315974284, + "tokens_seen": 776732672 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003862086258776329, + "loss": 1.3966, + "theoretical_loss": 3.7400818680778847, + "tokens_seen": 776798208 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038619859578736213, + "loss": 1.4825, + "theoretical_loss": 3.740050423577346, + "tokens_seen": 776863744 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038618856569709126, + "loss": 1.5134, + "theoretical_loss": 3.740018982472015, + "tokens_seen": 776929280 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003861785356068205, + "loss": 1.3662, + "theoretical_loss": 3.739987544761238, + "tokens_seen": 776994816 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003861685055165496, + "loss": 1.4066, + "theoretical_loss": 3.739956110444363, + "tokens_seen": 777060352 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038615847542627886, + "loss": 1.4745, + "theoretical_loss": 3.739924679520737, + "tokens_seen": 777125888 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038614844533600804, + "loss": 1.3638, + "theoretical_loss": 3.739893251989707, + "tokens_seen": 777191424 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003861384152457372, + "loss": 1.3955, + "theoretical_loss": 3.739861827850622, + "tokens_seen": 777256960 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003861283851554664, + "loss": 1.4935, + "theoretical_loss": 3.739830407102829, + "tokens_seen": 777322496 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003861183550651956, + "loss": 1.4215, + "theoretical_loss": 3.739798989745677, + "tokens_seen": 777388032 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038610832497492476, + "loss": 1.5037, + "theoretical_loss": 3.7397675757785134, + "tokens_seen": 777453568 + }, + { + "epoch": 0.24, + "learning_rate": 0.000386098294884654, + "loss": 1.3815, + "theoretical_loss": 3.7397361652006875, + "tokens_seen": 777519104 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003860882647943831, + "loss": 1.4676, + "theoretical_loss": 3.7397047580115474, + "tokens_seen": 777584640 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038607823470411236, + "loss": 1.5192, + "theoretical_loss": 3.739673354210442, + "tokens_seen": 777650176 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003860682046138415, + "loss": 1.4788, + "theoretical_loss": 3.7396419537967214, + "tokens_seen": 777715712 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003860581745235707, + "loss": 1.423, + "theoretical_loss": 3.739610556769733, + "tokens_seen": 777781248 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003860481444332999, + "loss": 1.3643, + "theoretical_loss": 3.7395791631288278, + "tokens_seen": 777846784 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003860381143430291, + "loss": 1.4084, + "theoretical_loss": 3.7395477728733537, + "tokens_seen": 777912320 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038602808425275826, + "loss": 1.4284, + "theoretical_loss": 3.7395163860026623, + "tokens_seen": 777977856 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003860180541624875, + "loss": 1.3525, + "theoretical_loss": 3.7394850025161026, + "tokens_seen": 778043392 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003860080240722166, + "loss": 1.3761, + "theoretical_loss": 3.7394536224130244, + "tokens_seen": 778108928 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038599799398194586, + "loss": 1.3948, + "theoretical_loss": 3.7394222456927793, + "tokens_seen": 778174464 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -0.0009485802729614079, + "objective/train/docs_used": 448758, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0303947925567627, + "objective/train/original_loss": 3.0303945541381836, + "objective/train/theoretical_loss": 3.739390872354716, + "objective/train/tokens_used": 798700000, + "objective/train/value_avg": -0.005802154541015625, + "objective/train/value_loss": 5.03429546370171e-05, + "objective/train/value_max": -0.00036835670471191406, + "objective/train/value_min": -0.0531005859375, + "objective/train/value_reward_corr": 0.4565626498644021, + "objective/train/value_std": 0.0047454833984375, + "objective/train/weight_avg": 0.9929587244987488, + "objective/train/weighted_lm_loss": 3.005985975265503, + "objective/train/weights_max": 1.5667698383331299, + "objective/train/weights_min": 0.647672712802887, + "theoretical_loss": 3.739390872354716, + "tokens_seen": 778240000 + }, + { + "epoch": 0.24, + "learning_rate": 0.000385987963891675, + "loss": 1.3543, + "theoretical_loss": 3.739390872354716, + "tokens_seen": 778240000 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003859779338014042, + "loss": 1.4209, + "theoretical_loss": 3.7393595023981865, + "tokens_seen": 778305536 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003859679037111334, + "loss": 1.443, + "theoretical_loss": 3.739328135822541, + "tokens_seen": 778371072 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003859578736208626, + "loss": 1.2665, + "theoretical_loss": 3.7392967726271307, + "tokens_seen": 778436608 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038594784353059177, + "loss": 1.4678, + "theoretical_loss": 3.7392654128113074, + "tokens_seen": 778502144 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038593781344032095, + "loss": 1.447, + "theoretical_loss": 3.7392340563744213, + "tokens_seen": 778567680 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038592778335005013, + "loss": 1.4029, + "theoretical_loss": 3.7392027033158244, + "tokens_seen": 778633216 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038591775325977937, + "loss": 1.4094, + "theoretical_loss": 3.7391713536348687, + "tokens_seen": 778698752 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003859077231695085, + "loss": 1.3048, + "theoretical_loss": 3.739140007330906, + "tokens_seen": 778764288 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038589769307923773, + "loss": 1.4122, + "theoretical_loss": 3.7391086644032887, + "tokens_seen": 778829824 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038588766298896685, + "loss": 1.4777, + "theoretical_loss": 3.739077324851368, + "tokens_seen": 778895360 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003858776328986961, + "loss": 1.439, + "theoretical_loss": 3.7390459886744973, + "tokens_seen": 778960896 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003858676028084253, + "loss": 1.4134, + "theoretical_loss": 3.7390146558720296, + "tokens_seen": 779026432 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038585757271815445, + "loss": 1.3125, + "theoretical_loss": 3.738983326443316, + "tokens_seen": 779091968 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003858475426278837, + "loss": 1.4169, + "theoretical_loss": 3.7389520003877115, + "tokens_seen": 779157504 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038583751253761287, + "loss": 1.4832, + "theoretical_loss": 3.7389206777045683, + "tokens_seen": 779223040 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038582748244734205, + "loss": 1.3627, + "theoretical_loss": 3.738889358393239, + "tokens_seen": 779288576 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038581745235707123, + "loss": 1.4173, + "theoretical_loss": 3.738858042453079, + "tokens_seen": 779354112 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003858074222668004, + "loss": 1.4742, + "theoretical_loss": 3.7388267298834403, + "tokens_seen": 779419648 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003857973921765296, + "loss": 1.4933, + "theoretical_loss": 3.7387954206836773, + "tokens_seen": 779485184 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038578736208625883, + "loss": 1.502, + "theoretical_loss": 3.738764114853144, + "tokens_seen": 779550720 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038577733199598796, + "loss": 1.5012, + "theoretical_loss": 3.738732812391195, + "tokens_seen": 779616256 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003857673019057172, + "loss": 1.4516, + "theoretical_loss": 3.738701513297184, + "tokens_seen": 779681792 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003857572718154463, + "loss": 1.4842, + "theoretical_loss": 3.7386702175704665, + "tokens_seen": 779747328 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038574724172517555, + "loss": 1.4294, + "theoretical_loss": 3.738638925210397, + "tokens_seen": 779812864 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -5.091354069008958e-06, + "objective/train/docs_used": 449318, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.899625301361084, + "objective/train/original_loss": 2.899625301361084, + "objective/train/theoretical_loss": 3.73860763621633, + "objective/train/tokens_used": 800338400, + "objective/train/value_avg": -0.0045623779296875, + "objective/train/value_loss": 4.307795825297944e-05, + "objective/train/value_max": -0.00051116943359375, + "objective/train/value_min": -0.10687255859375, + "objective/train/value_reward_corr": 0.4254258015683253, + "objective/train/value_std": 0.004055023193359375, + "objective/train/weight_avg": 1.001926064491272, + "objective/train/weighted_lm_loss": 2.918891668319702, + "objective/train/weights_max": 2.0915701389312744, + "objective/train/weights_min": 0.20498809218406677, + "theoretical_loss": 3.73860763621633, + "tokens_seen": 779878400 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038573721163490473, + "loss": 1.4914, + "theoretical_loss": 3.73860763621633, + "tokens_seen": 779878400 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003857271815446339, + "loss": 1.4118, + "theoretical_loss": 3.738576350587621, + "tokens_seen": 779943936 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003857171514543631, + "loss": 1.4028, + "theoretical_loss": 3.7385450683236248, + "tokens_seen": 780009472 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038570712136409233, + "loss": 1.3692, + "theoretical_loss": 3.7385137894236973, + "tokens_seen": 780075008 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038569709127382146, + "loss": 1.4077, + "theoretical_loss": 3.7384825138871944, + "tokens_seen": 780140544 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003856870611835507, + "loss": 1.4541, + "theoretical_loss": 3.738451241713472, + "tokens_seen": 780206080 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003856770310932798, + "loss": 1.5241, + "theoretical_loss": 3.7384199729018848, + "tokens_seen": 780271616 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038566700100300906, + "loss": 1.3935, + "theoretical_loss": 3.73838870745179, + "tokens_seen": 780337152 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038565697091273824, + "loss": 1.439, + "theoretical_loss": 3.7383574453625448, + "tokens_seen": 780402688 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003856469408224674, + "loss": 1.4971, + "theoretical_loss": 3.738326186633504, + "tokens_seen": 780468224 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003856369107321966, + "loss": 1.4773, + "theoretical_loss": 3.7382949312640252, + "tokens_seen": 780533760 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003856268806419258, + "loss": 1.2749, + "theoretical_loss": 3.7382636792534654, + "tokens_seen": 780599296 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038561685055165496, + "loss": 1.3123, + "theoretical_loss": 3.7382324306011814, + "tokens_seen": 780664832 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003856068204613842, + "loss": 1.3862, + "theoretical_loss": 3.7382011853065307, + "tokens_seen": 780730368 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003855967903711133, + "loss": 1.3473, + "theoretical_loss": 3.73816994336887, + "tokens_seen": 780795904 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038558676028084256, + "loss": 1.4471, + "theoretical_loss": 3.7381387047875583, + "tokens_seen": 780861440 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003855767301905717, + "loss": 1.4131, + "theoretical_loss": 3.7381074695619523, + "tokens_seen": 780926976 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003855667001003009, + "loss": 1.5255, + "theoretical_loss": 3.73807623769141, + "tokens_seen": 780992512 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003855566700100301, + "loss": 1.3452, + "theoretical_loss": 3.7380450091752904, + "tokens_seen": 781058048 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003855466399197593, + "loss": 1.4781, + "theoretical_loss": 3.73801378401295, + "tokens_seen": 781123584 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038553660982948846, + "loss": 1.4441, + "theoretical_loss": 3.737982562203749, + "tokens_seen": 781189120 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003855265797392177, + "loss": 1.4082, + "theoretical_loss": 3.7379513437470457, + "tokens_seen": 781254656 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003855165496489468, + "loss": 1.4755, + "theoretical_loss": 3.737920128642199, + "tokens_seen": 781320192 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038550651955867606, + "loss": 1.4139, + "theoretical_loss": 3.737888916888567, + "tokens_seen": 781385728 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003854964894684052, + "loss": 1.3496, + "theoretical_loss": 3.7378577084855102, + "tokens_seen": 781451264 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": 0.0004314843099564314, + "objective/train/docs_used": 450726, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.369527578353882, + "objective/train/original_loss": 2.369527578353882, + "objective/train/theoretical_loss": 3.7378265034323865, + "objective/train/tokens_used": 801976800, + "objective/train/value_avg": -0.004730224609375, + "objective/train/value_loss": 4.397968223202042e-05, + "objective/train/value_max": -0.0003447532653808594, + "objective/train/value_min": -0.09173583984375, + "objective/train/value_reward_corr": 0.29965449774078384, + "objective/train/value_std": 0.004085540771484375, + "objective/train/weight_avg": 1.0064195394515991, + "objective/train/weighted_lm_loss": 2.387387990951538, + "objective/train/weights_max": 1.9064627885818481, + "objective/train/weights_min": 0.3805619180202484, + "theoretical_loss": 3.7378265034323865, + "tokens_seen": 781516800 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003854864593781344, + "loss": 1.426, + "theoretical_loss": 3.7378265034323865, + "tokens_seen": 781516800 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003854764292878636, + "loss": 1.4506, + "theoretical_loss": 3.737795301728557, + "tokens_seen": 781582336 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003854663991975928, + "loss": 1.3679, + "theoretical_loss": 3.7377641033733804, + "tokens_seen": 781647872 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038545636910732197, + "loss": 1.5287, + "theoretical_loss": 3.7377329083662167, + "tokens_seen": 781713408 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038544633901705115, + "loss": 1.4854, + "theoretical_loss": 3.737701716706426, + "tokens_seen": 781778944 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038543630892678033, + "loss": 1.3562, + "theoretical_loss": 3.7376705283933696, + "tokens_seen": 781844480 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038542627883650957, + "loss": 1.5234, + "theoretical_loss": 3.737639343426406, + "tokens_seen": 781910016 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003854162487462387, + "loss": 1.4185, + "theoretical_loss": 3.7376081618048973, + "tokens_seen": 781975552 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038540621865596793, + "loss": 1.4182, + "theoretical_loss": 3.7375769835282036, + "tokens_seen": 782041088 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038539618856569705, + "loss": 1.5499, + "theoretical_loss": 3.7375458085956863, + "tokens_seen": 782106624 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003853861584754263, + "loss": 1.486, + "theoretical_loss": 3.737514637006706, + "tokens_seen": 782172160 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038537612838515547, + "loss": 1.3339, + "theoretical_loss": 3.737483468760624, + "tokens_seen": 782237696 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038536609829488465, + "loss": 1.3601, + "theoretical_loss": 3.737452303856803, + "tokens_seen": 782303232 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038535606820461383, + "loss": 1.3735, + "theoretical_loss": 3.737421142294603, + "tokens_seen": 782368768 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038534603811434307, + "loss": 1.4241, + "theoretical_loss": 3.7373899840733866, + "tokens_seen": 782434304 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003853360080240722, + "loss": 1.6176, + "theoretical_loss": 3.737358829192516, + "tokens_seen": 782499840 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038532597793380143, + "loss": 1.4267, + "theoretical_loss": 3.7373276776513533, + "tokens_seen": 782565376 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038531594784353056, + "loss": 1.4095, + "theoretical_loss": 3.73729652944926, + "tokens_seen": 782630912 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003853059177532598, + "loss": 1.3816, + "theoretical_loss": 3.7372653845855996, + "tokens_seen": 782696448 + }, + { + "epoch": 0.24, + "learning_rate": 0.000385295887662989, + "loss": 1.387, + "theoretical_loss": 3.737234243059735, + "tokens_seen": 782761984 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038528585757271816, + "loss": 1.3651, + "theoretical_loss": 3.7372031048710275, + "tokens_seen": 782827520 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038527582748244734, + "loss": 1.452, + "theoretical_loss": 3.737171970018842, + "tokens_seen": 782893056 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003852657973921765, + "loss": 1.4868, + "theoretical_loss": 3.737140838502541, + "tokens_seen": 782958592 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003852557673019057, + "loss": 1.4136, + "theoretical_loss": 3.7371097103214876, + "tokens_seen": 783024128 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038524573721163493, + "loss": 1.399, + "theoretical_loss": 3.7370785854750457, + "tokens_seen": 783089664 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -0.0011608386412262917, + "objective/train/docs_used": 451239, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8440730571746826, + "objective/train/original_loss": 2.8440732955932617, + "objective/train/theoretical_loss": 3.737047463962579, + "objective/train/tokens_used": 803615200, + "objective/train/value_avg": -0.005321502685546875, + "objective/train/value_loss": 0.00011010930757038295, + "objective/train/value_max": -0.0005421638488769531, + "objective/train/value_min": -0.0938720703125, + "objective/train/value_reward_corr": 0.660846444415234, + "objective/train/value_std": 0.005828857421875, + "objective/train/weight_avg": 0.993308961391449, + "objective/train/weighted_lm_loss": 2.823720693588257, + "objective/train/weights_max": 1.83995521068573, + "objective/train/weights_min": 0.3664967715740204, + "theoretical_loss": 3.737047463962579, + "tokens_seen": 783155200 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038523570712136406, + "loss": 1.5545, + "theoretical_loss": 3.737047463962579, + "tokens_seen": 783155200 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003852256770310933, + "loss": 1.4174, + "theoretical_loss": 3.7370163457834513, + "tokens_seen": 783220736 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003852156469408225, + "loss": 1.5592, + "theoretical_loss": 3.7369852309370275, + "tokens_seen": 783286272 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038520561685055166, + "loss": 1.4868, + "theoretical_loss": 3.7369541194226708, + "tokens_seen": 783351808 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038519558676028084, + "loss": 1.4151, + "theoretical_loss": 3.736923011239746, + "tokens_seen": 783417344 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038518555667001, + "loss": 1.3212, + "theoretical_loss": 3.7368919063876174, + "tokens_seen": 783482880 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003851755265797392, + "loss": 1.4085, + "theoretical_loss": 3.7368608048656506, + "tokens_seen": 783548416 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038516549648946844, + "loss": 1.5098, + "theoretical_loss": 3.7368297066732103, + "tokens_seen": 783613952 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038515546639919756, + "loss": 1.448, + "theoretical_loss": 3.7367986118096614, + "tokens_seen": 783679488 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003851454363089268, + "loss": 1.3695, + "theoretical_loss": 3.7367675202743698, + "tokens_seen": 783745024 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003851354062186559, + "loss": 1.493, + "theoretical_loss": 3.7367364320666994, + "tokens_seen": 783810560 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038512537612838516, + "loss": 1.3623, + "theoretical_loss": 3.7367053471860183, + "tokens_seen": 783876096 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003851153460381144, + "loss": 1.4467, + "theoretical_loss": 3.73667426563169, + "tokens_seen": 783941632 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003851053159478435, + "loss": 1.3555, + "theoretical_loss": 3.7366431874030823, + "tokens_seen": 784007168 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038509528585757276, + "loss": 1.2697, + "theoretical_loss": 3.73661211249956, + "tokens_seen": 784072704 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003850852557673019, + "loss": 1.4939, + "theoretical_loss": 3.7365810409204903, + "tokens_seen": 784138240 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003850752256770311, + "loss": 1.4118, + "theoretical_loss": 3.73654997266524, + "tokens_seen": 784203776 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003850651955867603, + "loss": 1.4678, + "theoretical_loss": 3.736518907733175, + "tokens_seen": 784269312 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003850551654964895, + "loss": 1.4415, + "theoretical_loss": 3.736487846123663, + "tokens_seen": 784334848 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038504513540621866, + "loss": 1.3388, + "theoretical_loss": 3.7364567878360706, + "tokens_seen": 784400384 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003850351053159479, + "loss": 1.482, + "theoretical_loss": 3.7364257328697645, + "tokens_seen": 784465920 + }, + { + "epoch": 0.24, + "learning_rate": 0.000385025075225677, + "loss": 1.4382, + "theoretical_loss": 3.7363946812241133, + "tokens_seen": 784531456 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038501504513540626, + "loss": 1.3363, + "theoretical_loss": 3.7363636328984837, + "tokens_seen": 784596992 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003850050150451354, + "loss": 1.3906, + "theoretical_loss": 3.7363325878922433, + "tokens_seen": 784662528 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003849949849548646, + "loss": 1.4835, + "theoretical_loss": 3.7363015462047606, + "tokens_seen": 784728064 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -0.00024765782291069627, + "objective/train/docs_used": 452482, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9573893547058105, + "objective/train/original_loss": 2.9573893547058105, + "objective/train/theoretical_loss": 3.736270507835404, + "objective/train/tokens_used": 805253600, + "objective/train/value_avg": -0.005977630615234375, + "objective/train/value_loss": 7.32621192582883e-05, + "objective/train/value_max": -0.0004897117614746094, + "objective/train/value_min": -0.1007080078125, + "objective/train/value_reward_corr": 0.3678118342174366, + "objective/train/value_std": 0.00551605224609375, + "objective/train/weight_avg": 1.001020073890686, + "objective/train/weighted_lm_loss": 2.9772050380706787, + "objective/train/weights_max": 1.992957592010498, + "objective/train/weights_min": 0.22038033604621887, + "theoretical_loss": 3.736270507835404, + "tokens_seen": 784793600 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003849849548645938, + "loss": 1.4805, + "theoretical_loss": 3.736270507835404, + "tokens_seen": 784793600 + }, + { + "epoch": 0.24, + "learning_rate": 0.000384974924774323, + "loss": 1.4042, + "theoretical_loss": 3.736239472783541, + "tokens_seen": 784859136 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038496489468405217, + "loss": 1.4447, + "theoretical_loss": 3.7362084410485403, + "tokens_seen": 784924672 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038495486459378135, + "loss": 1.4309, + "theoretical_loss": 3.736177412629771, + "tokens_seen": 784990208 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038494483450351053, + "loss": 1.3651, + "theoretical_loss": 3.736146387526601, + "tokens_seen": 785055744 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038493480441323977, + "loss": 1.4963, + "theoretical_loss": 3.7361153657384003, + "tokens_seen": 785121280 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003849247743229689, + "loss": 1.4605, + "theoretical_loss": 3.736084347264537, + "tokens_seen": 785186816 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038491474423269813, + "loss": 1.3809, + "theoretical_loss": 3.7360533321043805, + "tokens_seen": 785252352 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038490471414242725, + "loss": 1.4409, + "theoretical_loss": 3.7360223202573013, + "tokens_seen": 785317888 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003848946840521565, + "loss": 1.306, + "theoretical_loss": 3.7359913117226684, + "tokens_seen": 785383424 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038488465396188567, + "loss": 1.3465, + "theoretical_loss": 3.7359603064998517, + "tokens_seen": 785448960 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038487462387161485, + "loss": 1.4771, + "theoretical_loss": 3.735929304588221, + "tokens_seen": 785514496 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038486459378134403, + "loss": 1.4162, + "theoretical_loss": 3.7358983059871465, + "tokens_seen": 785580032 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038485456369107327, + "loss": 1.5085, + "theoretical_loss": 3.7358673106959985, + "tokens_seen": 785645568 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003848445336008024, + "loss": 1.4493, + "theoretical_loss": 3.7358363187141483, + "tokens_seen": 785711104 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038483450351053163, + "loss": 1.4855, + "theoretical_loss": 3.7358053300409653, + "tokens_seen": 785776640 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038482447342026076, + "loss": 1.3301, + "theoretical_loss": 3.7357743446758214, + "tokens_seen": 785842176 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038481444332999, + "loss": 1.4526, + "theoretical_loss": 3.735743362618088, + "tokens_seen": 785907712 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003848044132397192, + "loss": 1.4924, + "theoretical_loss": 3.735712383867135, + "tokens_seen": 785973248 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038479438314944836, + "loss": 1.3341, + "theoretical_loss": 3.7356814084223338, + "tokens_seen": 786038784 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038478435305917754, + "loss": 1.3046, + "theoretical_loss": 3.735650436283057, + "tokens_seen": 786104320 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003847743229689067, + "loss": 1.3873, + "theoretical_loss": 3.735619467448676, + "tokens_seen": 786169856 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003847642928786359, + "loss": 1.4462, + "theoretical_loss": 3.735588501918562, + "tokens_seen": 786235392 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038475426278836513, + "loss": 1.3081, + "theoretical_loss": 3.7355575396920884, + "tokens_seen": 786300928 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038474423269809426, + "loss": 1.5265, + "theoretical_loss": 3.7355265807686258, + "tokens_seen": 786366464 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": 8.537209214409813e-05, + "objective/train/docs_used": 453141, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.644871950149536, + "objective/train/original_loss": 2.644871711730957, + "objective/train/theoretical_loss": 3.735495625147548, + "objective/train/tokens_used": 806892000, + "objective/train/value_avg": -0.005115509033203125, + "objective/train/value_loss": 4.513575549935922e-05, + "objective/train/value_max": -0.0002491474151611328, + "objective/train/value_min": -0.049224853515625, + "objective/train/value_reward_corr": 0.33756152461060784, + "objective/train/value_std": 0.00421142578125, + "objective/train/weight_avg": 1.0030461549758911, + "objective/train/weighted_lm_loss": 2.6578779220581055, + "objective/train/weights_max": 1.5713974237442017, + "objective/train/weights_min": 0.5361347198486328, + "theoretical_loss": 3.735495625147548, + "tokens_seen": 786432000 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003847342026078235, + "loss": 1.3707, + "theoretical_loss": 3.735495625147548, + "tokens_seen": 786432000 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003847241725175527, + "loss": 1.3098, + "theoretical_loss": 3.7354646728282264, + "tokens_seen": 786497536 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038471414242728186, + "loss": 1.4241, + "theoretical_loss": 3.735433723810035, + "tokens_seen": 786563072 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038470411233701104, + "loss": 1.3212, + "theoretical_loss": 3.7354027780923458, + "tokens_seen": 786628608 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003846940822467402, + "loss": 1.3596, + "theoretical_loss": 3.7353718356745325, + "tokens_seen": 786694144 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003846840521564694, + "loss": 1.4019, + "theoretical_loss": 3.735340896555968, + "tokens_seen": 786759680 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038467402206619864, + "loss": 1.3676, + "theoretical_loss": 3.735309960736026, + "tokens_seen": 786825216 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038466399197592776, + "loss": 1.3423, + "theoretical_loss": 3.7352790282140793, + "tokens_seen": 786890752 + }, + { + "epoch": 0.24, + "learning_rate": 0.000384653961885657, + "loss": 1.3354, + "theoretical_loss": 3.7352480989895023, + "tokens_seen": 786956288 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003846439317953861, + "loss": 1.4494, + "theoretical_loss": 3.7352171730616694, + "tokens_seen": 787021824 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038463390170511536, + "loss": 1.2919, + "theoretical_loss": 3.735186250429954, + "tokens_seen": 787087360 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038462387161484454, + "loss": 1.5088, + "theoretical_loss": 3.735155331093731, + "tokens_seen": 787152896 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003846138415245737, + "loss": 1.4008, + "theoretical_loss": 3.7351244150523737, + "tokens_seen": 787218432 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003846038114343029, + "loss": 1.3183, + "theoretical_loss": 3.7350935023052583, + "tokens_seen": 787283968 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003845937813440321, + "loss": 1.4501, + "theoretical_loss": 3.735062592851759, + "tokens_seen": 787349504 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038458375125376127, + "loss": 1.4313, + "theoretical_loss": 3.73503168669125, + "tokens_seen": 787415040 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003845737211634905, + "loss": 1.4863, + "theoretical_loss": 3.735000783823107, + "tokens_seen": 787480576 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038456369107321963, + "loss": 1.4635, + "theoretical_loss": 3.7349698842467056, + "tokens_seen": 787546112 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038455366098294886, + "loss": 1.4107, + "theoretical_loss": 3.7349389879614217, + "tokens_seen": 787611648 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038454363089267805, + "loss": 1.3845, + "theoretical_loss": 3.7349080949666296, + "tokens_seen": 787677184 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038453360080240723, + "loss": 1.3216, + "theoretical_loss": 3.7348772052617063, + "tokens_seen": 787742720 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003845235707121364, + "loss": 1.4708, + "theoretical_loss": 3.734846318846027, + "tokens_seen": 787808256 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003845135406218656, + "loss": 1.3239, + "theoretical_loss": 3.7348154357189682, + "tokens_seen": 787873792 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038450351053159477, + "loss": 1.4593, + "theoretical_loss": 3.7347845558799064, + "tokens_seen": 787939328 + }, + { + "epoch": 0.24, + "learning_rate": 0.000384493480441324, + "loss": 1.5159, + "theoretical_loss": 3.734753679328218, + "tokens_seen": 788004864 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -0.0018956036074087024, + "objective/train/docs_used": 453953, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.505749464035034, + "objective/train/original_loss": 2.5057497024536133, + "objective/train/theoretical_loss": 3.7347228060632798, + "objective/train/tokens_used": 808530400, + "objective/train/value_avg": -0.004791259765625, + "objective/train/value_loss": 7.628709863638505e-05, + "objective/train/value_max": -0.0003542900085449219, + "objective/train/value_min": -0.04620361328125, + "objective/train/value_reward_corr": 0.3532058810866956, + "objective/train/value_std": 0.003993988037109375, + "objective/train/weight_avg": 0.9846086502075195, + "objective/train/weighted_lm_loss": 2.459601402282715, + "objective/train/weights_max": 1.4475507736206055, + "objective/train/weights_min": 0.5114023089408875, + "theoretical_loss": 3.7347228060632798, + "tokens_seen": 788070400 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038448345035105313, + "loss": 1.2914, + "theoretical_loss": 3.7347228060632798, + "tokens_seen": 788070400 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038447342026078237, + "loss": 1.4103, + "theoretical_loss": 3.7346919360844684, + "tokens_seen": 788135936 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003844633901705115, + "loss": 1.2709, + "theoretical_loss": 3.734661069391161, + "tokens_seen": 788201472 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038445336008024073, + "loss": 1.4689, + "theoretical_loss": 3.7346302059827345, + "tokens_seen": 788267008 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003844433299899699, + "loss": 1.3393, + "theoretical_loss": 3.734599345858567, + "tokens_seen": 788332544 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003844332998996991, + "loss": 1.3939, + "theoretical_loss": 3.7345684890180344, + "tokens_seen": 788398080 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003844232698094283, + "loss": 1.4641, + "theoretical_loss": 3.7345376354605166, + "tokens_seen": 788463616 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038441323971915745, + "loss": 1.3713, + "theoretical_loss": 3.7345067851853897, + "tokens_seen": 788529152 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038440320962888664, + "loss": 1.3364, + "theoretical_loss": 3.734475938192033, + "tokens_seen": 788594688 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038439317953861587, + "loss": 1.3805, + "theoretical_loss": 3.734445094479824, + "tokens_seen": 788660224 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038438314944834505, + "loss": 1.3532, + "theoretical_loss": 3.734414254048141, + "tokens_seen": 788725760 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038437311935807423, + "loss": 1.3643, + "theoretical_loss": 3.734383416896362, + "tokens_seen": 788791296 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038436308926780347, + "loss": 1.3469, + "theoretical_loss": 3.7343525830238673, + "tokens_seen": 788856832 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003843530591775326, + "loss": 1.4698, + "theoretical_loss": 3.7343217524300347, + "tokens_seen": 788922368 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038434302908726183, + "loss": 1.4167, + "theoretical_loss": 3.7342909251142435, + "tokens_seen": 788987904 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038433299899699096, + "loss": 1.3615, + "theoretical_loss": 3.734260101075873, + "tokens_seen": 789053440 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003843229689067202, + "loss": 1.4002, + "theoretical_loss": 3.7342292803143025, + "tokens_seen": 789118976 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003843129388164494, + "loss": 1.2995, + "theoretical_loss": 3.7341984628289118, + "tokens_seen": 789184512 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038430290872617856, + "loss": 1.3455, + "theoretical_loss": 3.73416764861908, + "tokens_seen": 789250048 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038429287863590774, + "loss": 1.4547, + "theoretical_loss": 3.7341368376841872, + "tokens_seen": 789315584 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003842828485456369, + "loss": 1.4606, + "theoretical_loss": 3.7341060300236144, + "tokens_seen": 789381120 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003842728184553661, + "loss": 1.3709, + "theoretical_loss": 3.73407522563674, + "tokens_seen": 789446656 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038426278836509533, + "loss": 1.3828, + "theoretical_loss": 3.7340444245229465, + "tokens_seen": 789512192 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038425275827482446, + "loss": 1.5576, + "theoretical_loss": 3.7340136266816133, + "tokens_seen": 789577728 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003842427281845537, + "loss": 1.4166, + "theoretical_loss": 3.733982832112121, + "tokens_seen": 789643264 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": 0.00021245848620310426, + "objective/train/docs_used": 455438, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.908259868621826, + "objective/train/original_loss": 2.908259868621826, + "objective/train/theoretical_loss": 3.7339520408138513, + "objective/train/tokens_used": 810168800, + "objective/train/value_avg": -0.004852294921875, + "objective/train/value_loss": 4.068120688316412e-05, + "objective/train/value_max": -0.0004458427429199219, + "objective/train/value_min": -0.07421875, + "objective/train/value_reward_corr": 0.4449770898692714, + "objective/train/value_std": 0.00373077392578125, + "objective/train/weight_avg": 1.004081130027771, + "objective/train/weighted_lm_loss": 2.917440414428711, + "objective/train/weights_max": 2.100525379180908, + "objective/train/weights_min": 0.41828492283821106, + "theoretical_loss": 3.7339520408138513, + "tokens_seen": 789708800 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003842326980942829, + "loss": 1.3588, + "theoretical_loss": 3.7339520408138513, + "tokens_seen": 789708800 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038422266800401206, + "loss": 1.4228, + "theoretical_loss": 3.7339212527861845, + "tokens_seen": 789774336 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038421263791374124, + "loss": 1.3789, + "theoretical_loss": 3.733890468028502, + "tokens_seen": 789839872 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003842026078234704, + "loss": 1.3837, + "theoretical_loss": 3.7338596865401854, + "tokens_seen": 789905408 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003841925777331996, + "loss": 1.3979, + "theoretical_loss": 3.733828908320616, + "tokens_seen": 789970944 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038418254764292884, + "loss": 1.3993, + "theoretical_loss": 3.7337981333691763, + "tokens_seen": 790036480 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038417251755265796, + "loss": 1.4971, + "theoretical_loss": 3.733767361685248, + "tokens_seen": 790102016 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003841624874623872, + "loss": 1.4106, + "theoretical_loss": 3.7337365932682123, + "tokens_seen": 790167552 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003841524573721163, + "loss": 1.3949, + "theoretical_loss": 3.7337058281174524, + "tokens_seen": 790233088 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038414242728184556, + "loss": 1.3448, + "theoretical_loss": 3.73367506623235, + "tokens_seen": 790298624 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038413239719157474, + "loss": 1.4477, + "theoretical_loss": 3.7336443076122885, + "tokens_seen": 790364160 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003841223671013039, + "loss": 1.4247, + "theoretical_loss": 3.7336135522566503, + "tokens_seen": 790429696 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003841123370110331, + "loss": 1.3603, + "theoretical_loss": 3.733582800164818, + "tokens_seen": 790495232 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003841023069207623, + "loss": 1.3982, + "theoretical_loss": 3.7335520513361753, + "tokens_seen": 790560768 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038409227683049147, + "loss": 1.4282, + "theoretical_loss": 3.733521305770105, + "tokens_seen": 790626304 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003840822467402207, + "loss": 1.4258, + "theoretical_loss": 3.73349056346599, + "tokens_seen": 790691840 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038407221664994983, + "loss": 1.5609, + "theoretical_loss": 3.733459824423216, + "tokens_seen": 790757376 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038406218655967907, + "loss": 1.3639, + "theoretical_loss": 3.7334290886411643, + "tokens_seen": 790822912 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038405215646940825, + "loss": 1.3674, + "theoretical_loss": 3.7333983561192206, + "tokens_seen": 790888448 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038404212637913743, + "loss": 1.3627, + "theoretical_loss": 3.7333676268567677, + "tokens_seen": 790953984 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003840320962888666, + "loss": 1.4646, + "theoretical_loss": 3.7333369008531907, + "tokens_seen": 791019520 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003840220661985958, + "loss": 1.3781, + "theoretical_loss": 3.733306178107873, + "tokens_seen": 791085056 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038401203610832497, + "loss": 1.3745, + "theoretical_loss": 3.7332754586202013, + "tokens_seen": 791150592 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003840020060180542, + "loss": 1.3384, + "theoretical_loss": 3.7332447423895583, + "tokens_seen": 791216128 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038399197592778333, + "loss": 1.355, + "theoretical_loss": 3.73321402941533, + "tokens_seen": 791281664 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -8.058904495555907e-05, + "objective/train/docs_used": 456184, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.01324200630188, + "objective/train/original_loss": 3.01324200630188, + "objective/train/theoretical_loss": 3.733183319696901, + "objective/train/tokens_used": 811807200, + "objective/train/value_avg": -0.004665374755859375, + "objective/train/value_loss": 4.806047218153253e-05, + "objective/train/value_max": -0.00021147727966308594, + "objective/train/value_min": -0.05242919921875, + "objective/train/value_reward_corr": 0.33720346064227646, + "objective/train/value_std": 0.003345489501953125, + "objective/train/weight_avg": 1.0014474391937256, + "objective/train/weighted_lm_loss": 3.0291600227355957, + "objective/train/weights_max": 1.396285891532898, + "objective/train/weights_min": 0.2983158826828003, + "theoretical_loss": 3.733183319696901, + "tokens_seen": 791347200 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038398194583751257, + "loss": 1.4038, + "theoretical_loss": 3.733183319696901, + "tokens_seen": 791347200 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003839719157472417, + "loss": 1.4126, + "theoretical_loss": 3.7331526132336568, + "tokens_seen": 791412736 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038396188565697093, + "loss": 1.3705, + "theoretical_loss": 3.733121910024983, + "tokens_seen": 791478272 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003839518555667001, + "loss": 1.3305, + "theoretical_loss": 3.7330912100702647, + "tokens_seen": 791543808 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003839418254764293, + "loss": 1.382, + "theoretical_loss": 3.7330605133688883, + "tokens_seen": 791609344 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003839317953861585, + "loss": 1.4138, + "theoretical_loss": 3.7330298199202394, + "tokens_seen": 791674880 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038392176529588765, + "loss": 1.3689, + "theoretical_loss": 3.732999129723704, + "tokens_seen": 791740416 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038391173520561684, + "loss": 1.3458, + "theoretical_loss": 3.7329684427786685, + "tokens_seen": 791805952 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038390170511534607, + "loss": 1.486, + "theoretical_loss": 3.73293775908452, + "tokens_seen": 791871488 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003838916750250752, + "loss": 1.2834, + "theoretical_loss": 3.732907078640644, + "tokens_seen": 791937024 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038388164493480443, + "loss": 1.4431, + "theoretical_loss": 3.7328764014464273, + "tokens_seen": 792002560 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003838716148445336, + "loss": 1.4167, + "theoretical_loss": 3.732845727501258, + "tokens_seen": 792068096 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003838615847542628, + "loss": 1.3724, + "theoretical_loss": 3.7328150568045224, + "tokens_seen": 792133632 + }, + { + "epoch": 0.24, + "learning_rate": 0.000383851554663992, + "loss": 1.5549, + "theoretical_loss": 3.732784389355608, + "tokens_seen": 792199168 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038384152457372116, + "loss": 1.3159, + "theoretical_loss": 3.7327537251539016, + "tokens_seen": 792264704 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038383149448345034, + "loss": 1.3839, + "theoretical_loss": 3.732723064198791, + "tokens_seen": 792330240 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003838214643931796, + "loss": 1.583, + "theoretical_loss": 3.732692406489665, + "tokens_seen": 792395776 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003838114343029087, + "loss": 1.404, + "theoretical_loss": 3.732661752025911, + "tokens_seen": 792461312 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038380140421263794, + "loss": 1.3114, + "theoretical_loss": 3.732631100806916, + "tokens_seen": 792526848 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038379137412236706, + "loss": 1.3489, + "theoretical_loss": 3.73260045283207, + "tokens_seen": 792592384 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003837813440320963, + "loss": 1.388, + "theoretical_loss": 3.7325698081007603, + "tokens_seen": 792657920 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003837713139418255, + "loss": 1.3704, + "theoretical_loss": 3.7325391666123764, + "tokens_seen": 792723456 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038376128385155466, + "loss": 1.3292, + "theoretical_loss": 3.732508528366306, + "tokens_seen": 792788992 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038375125376128384, + "loss": 1.3526, + "theoretical_loss": 3.7324778933619385, + "tokens_seen": 792854528 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003837412236710131, + "loss": 1.4082, + "theoretical_loss": 3.7324472615986632, + "tokens_seen": 792920064 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -0.0010877458844333887, + "objective/train/docs_used": 457386, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.378549337387085, + "objective/train/original_loss": 2.378549337387085, + "objective/train/theoretical_loss": 3.732416633075869, + "objective/train/tokens_used": 813445600, + "objective/train/value_avg": -0.00457763671875, + "objective/train/value_loss": 3.619615381467156e-05, + "objective/train/value_max": -0.0002913475036621094, + "objective/train/value_min": -0.04302978515625, + "objective/train/value_reward_corr": 0.38662487111851995, + "objective/train/value_std": 0.0031890869140625, + "objective/train/weight_avg": 0.9908692836761475, + "objective/train/weighted_lm_loss": 2.3667876720428467, + "objective/train/weights_max": 1.5377155542373657, + "objective/train/weights_min": 0.6532655954360962, + "theoretical_loss": 3.732416633075869, + "tokens_seen": 792985600 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003837311935807422, + "loss": 1.3661, + "theoretical_loss": 3.732416633075869, + "tokens_seen": 792985600 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038372116349047144, + "loss": 1.4186, + "theoretical_loss": 3.732386007792946, + "tokens_seen": 793051136 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038371113340020057, + "loss": 1.4021, + "theoretical_loss": 3.7323553857492824, + "tokens_seen": 793116672 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003837011033099298, + "loss": 1.3109, + "theoretical_loss": 3.732324766944269, + "tokens_seen": 793182208 + }, + { + "epoch": 0.24, + "learning_rate": 0.000383691073219659, + "loss": 1.3585, + "theoretical_loss": 3.732294151377296, + "tokens_seen": 793247744 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038368104312938816, + "loss": 1.378, + "theoretical_loss": 3.7322635390477537, + "tokens_seen": 793313280 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038367101303911735, + "loss": 1.3815, + "theoretical_loss": 3.7322329299550305, + "tokens_seen": 793378816 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003836609829488465, + "loss": 1.3845, + "theoretical_loss": 3.7322023240985187, + "tokens_seen": 793444352 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003836509528585757, + "loss": 1.4588, + "theoretical_loss": 3.7321717214776085, + "tokens_seen": 793509888 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038364092276830494, + "loss": 1.3175, + "theoretical_loss": 3.73214112209169, + "tokens_seen": 793575424 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003836308926780341, + "loss": 1.4684, + "theoretical_loss": 3.7321105259401546, + "tokens_seen": 793640960 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003836208625877633, + "loss": 1.4327, + "theoretical_loss": 3.7320799330223933, + "tokens_seen": 793706496 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003836108324974925, + "loss": 1.4122, + "theoretical_loss": 3.7320493433377973, + "tokens_seen": 793772032 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038360080240722167, + "loss": 1.5243, + "theoretical_loss": 3.7320187568857586, + "tokens_seen": 793837568 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003835907723169509, + "loss": 1.5332, + "theoretical_loss": 3.731988173665668, + "tokens_seen": 793903104 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038358074222668003, + "loss": 1.3344, + "theoretical_loss": 3.731957593676917, + "tokens_seen": 793968640 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038357071213640927, + "loss": 1.2706, + "theoretical_loss": 3.731927016918898, + "tokens_seen": 794034176 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038356068204613845, + "loss": 1.4418, + "theoretical_loss": 3.7318964433910033, + "tokens_seen": 794099712 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038355065195586763, + "loss": 1.3202, + "theoretical_loss": 3.731865873092625, + "tokens_seen": 794165248 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003835406218655968, + "loss": 1.4456, + "theoretical_loss": 3.7318353060231555, + "tokens_seen": 794230784 + }, + { + "epoch": 0.24, + "learning_rate": 0.000383530591775326, + "loss": 1.3681, + "theoretical_loss": 3.7318047421819864, + "tokens_seen": 794296320 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038352056168505517, + "loss": 1.4251, + "theoretical_loss": 3.731774181568512, + "tokens_seen": 794361856 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003835105315947844, + "loss": 1.498, + "theoretical_loss": 3.7317436241821245, + "tokens_seen": 794427392 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038350050150451353, + "loss": 1.3841, + "theoretical_loss": 3.7317130700222174, + "tokens_seen": 794492928 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038349047141424277, + "loss": 1.55, + "theoretical_loss": 3.7316825190881824, + "tokens_seen": 794558464 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -0.000763515243306756, + "objective/train/docs_used": 458050, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.556380033493042, + "objective/train/original_loss": 2.556379795074463, + "objective/train/theoretical_loss": 3.7316519713794145, + "objective/train/tokens_used": 815084000, + "objective/train/value_avg": -0.004917144775390625, + "objective/train/value_loss": 5.8936300774803385e-05, + "objective/train/value_max": -0.0003006458282470703, + "objective/train/value_min": -0.07501220703125, + "objective/train/value_reward_corr": 0.46764253328405114, + "objective/train/value_std": 0.004486083984375, + "objective/train/weight_avg": 0.9951719641685486, + "objective/train/weighted_lm_loss": 2.540766954421997, + "objective/train/weights_max": 2.036457061767578, + "objective/train/weights_min": 0.569997251033783, + "theoretical_loss": 3.7316519713794145, + "tokens_seen": 794624000 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003834804413239719, + "loss": 1.5084, + "theoretical_loss": 3.7316519713794145, + "tokens_seen": 794624000 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038347041123370113, + "loss": 1.3597, + "theoretical_loss": 3.7316214268953063, + "tokens_seen": 794689536 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003834603811434303, + "loss": 1.3048, + "theoretical_loss": 3.7315908856352524, + "tokens_seen": 794755072 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003834503510531595, + "loss": 1.3839, + "theoretical_loss": 3.731560347598646, + "tokens_seen": 794820608 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003834403209628887, + "loss": 1.3757, + "theoretical_loss": 3.7315298127848813, + "tokens_seen": 794886144 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038343029087261786, + "loss": 1.2526, + "theoretical_loss": 3.7314992811933525, + "tokens_seen": 794951680 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038342026078234704, + "loss": 1.3177, + "theoretical_loss": 3.731468752823454, + "tokens_seen": 795017216 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038341023069207627, + "loss": 1.4157, + "theoretical_loss": 3.73143822767458, + "tokens_seen": 795082752 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003834002006018054, + "loss": 1.5415, + "theoretical_loss": 3.7314077057461255, + "tokens_seen": 795148288 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038339017051153463, + "loss": 1.3974, + "theoretical_loss": 3.731377187037486, + "tokens_seen": 795213824 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003833801404212638, + "loss": 1.4128, + "theoretical_loss": 3.7313466715480557, + "tokens_seen": 795279360 + }, + { + "epoch": 0.24, + "learning_rate": 0.000383370110330993, + "loss": 1.3872, + "theoretical_loss": 3.7313161592772293, + "tokens_seen": 795344896 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003833600802407222, + "loss": 1.2655, + "theoretical_loss": 3.731285650224404, + "tokens_seen": 795410432 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038335005015045136, + "loss": 1.3116, + "theoretical_loss": 3.7312551443889728, + "tokens_seen": 795475968 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038334002006018054, + "loss": 1.4147, + "theoretical_loss": 3.731224641770334, + "tokens_seen": 795541504 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003833299899699098, + "loss": 1.3442, + "theoretical_loss": 3.7311941423678814, + "tokens_seen": 795607040 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003833199598796389, + "loss": 1.3665, + "theoretical_loss": 3.731163646181012, + "tokens_seen": 795672576 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038330992978936814, + "loss": 1.4254, + "theoretical_loss": 3.7311331532091216, + "tokens_seen": 795738112 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038329989969909726, + "loss": 1.5138, + "theoretical_loss": 3.7311026634516065, + "tokens_seen": 795803648 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003832898696088265, + "loss": 1.3207, + "theoretical_loss": 3.7310721769078636, + "tokens_seen": 795869184 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003832798395185557, + "loss": 1.4267, + "theoretical_loss": 3.731041693577289, + "tokens_seen": 795934720 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038326980942828486, + "loss": 1.4022, + "theoretical_loss": 3.7310112134592806, + "tokens_seen": 796000256 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038325977933801404, + "loss": 1.458, + "theoretical_loss": 3.730980736553234, + "tokens_seen": 796065792 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003832497492477433, + "loss": 1.47, + "theoretical_loss": 3.7309502628585474, + "tokens_seen": 796131328 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003832397191574724, + "loss": 1.3596, + "theoretical_loss": 3.7309197923746176, + "tokens_seen": 796196864 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -0.0015226717805489898, + "objective/train/docs_used": 459137, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9786407947540283, + "objective/train/original_loss": 2.97864031791687, + "objective/train/theoretical_loss": 3.7308893251008413, + "objective/train/tokens_used": 816722400, + "objective/train/value_avg": -0.005016326904296875, + "objective/train/value_loss": 7.816913421265781e-05, + "objective/train/value_max": -0.00042891502380371094, + "objective/train/value_min": -0.05291748046875, + "objective/train/value_reward_corr": 0.3719385494142836, + "objective/train/value_std": 0.003971099853515625, + "objective/train/weight_avg": 0.9884165525436401, + "objective/train/weighted_lm_loss": 2.9643537998199463, + "objective/train/weights_max": 1.5003794431686401, + "objective/train/weights_min": 0.5413753986358643, + "theoretical_loss": 3.7308893251008413, + "tokens_seen": 796262400 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038322968906720164, + "loss": 1.4668, + "theoretical_loss": 3.7308893251008413, + "tokens_seen": 796262400 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038321965897693077, + "loss": 1.2624, + "theoretical_loss": 3.7308588610366176, + "tokens_seen": 796327936 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038320962888666, + "loss": 1.2726, + "theoretical_loss": 3.730828400181344, + "tokens_seen": 796393472 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003831995987963892, + "loss": 1.2973, + "theoretical_loss": 3.7307979425344175, + "tokens_seen": 796459008 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038318956870611836, + "loss": 1.3624, + "theoretical_loss": 3.7307674880952373, + "tokens_seen": 796524544 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038317953861584755, + "loss": 1.3414, + "theoretical_loss": 3.730737036863201, + "tokens_seen": 796590080 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003831695085255767, + "loss": 1.4593, + "theoretical_loss": 3.730706588837707, + "tokens_seen": 796655616 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003831594784353059, + "loss": 1.3957, + "theoretical_loss": 3.730676144018154, + "tokens_seen": 796721152 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038314944834503514, + "loss": 1.3871, + "theoretical_loss": 3.7306457024039417, + "tokens_seen": 796786688 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038313941825476427, + "loss": 1.3232, + "theoretical_loss": 3.730615263994468, + "tokens_seen": 796852224 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003831293881644935, + "loss": 1.4396, + "theoretical_loss": 3.730584828789132, + "tokens_seen": 796917760 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038311935807422263, + "loss": 1.3843, + "theoretical_loss": 3.7305543967873334, + "tokens_seen": 796983296 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038310932798395187, + "loss": 1.245, + "theoretical_loss": 3.7305239679884714, + "tokens_seen": 797048832 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038309929789368105, + "loss": 1.3057, + "theoretical_loss": 3.7304935423919456, + "tokens_seen": 797114368 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038308926780341023, + "loss": 1.3393, + "theoretical_loss": 3.730463119997156, + "tokens_seen": 797179904 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003830792377131394, + "loss": 1.3064, + "theoretical_loss": 3.730432700803502, + "tokens_seen": 797245440 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038306920762286865, + "loss": 1.3783, + "theoretical_loss": 3.730402284810384, + "tokens_seen": 797310976 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003830591775325978, + "loss": 1.3173, + "theoretical_loss": 3.7303718720172023, + "tokens_seen": 797376512 + }, + { + "epoch": 0.24, + "learning_rate": 0.000383049147442327, + "loss": 1.3284, + "theoretical_loss": 3.730341462423357, + "tokens_seen": 797442048 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038303911735205614, + "loss": 1.3739, + "theoretical_loss": 3.7303110560282486, + "tokens_seen": 797507584 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038302908726178537, + "loss": 1.4147, + "theoretical_loss": 3.7302806528312784, + "tokens_seen": 797573120 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038301905717151455, + "loss": 1.2578, + "theoretical_loss": 3.730250252831847, + "tokens_seen": 797638656 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038300902708124373, + "loss": 1.3798, + "theoretical_loss": 3.730219856029355, + "tokens_seen": 797704192 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003829989969909729, + "loss": 1.3915, + "theoretical_loss": 3.730189462423204, + "tokens_seen": 797769728 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003829889669007021, + "loss": 1.3639, + "theoretical_loss": 3.7301590720127953, + "tokens_seen": 797835264 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": 0.00016628798039164394, + "objective/train/docs_used": 459607, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.706113338470459, + "objective/train/original_loss": 2.706113576889038, + "objective/train/theoretical_loss": 3.7301286847975295, + "objective/train/tokens_used": 818360800, + "objective/train/value_avg": -0.00489044189453125, + "objective/train/value_loss": 6.270970334298909e-05, + "objective/train/value_max": -0.0004172325134277344, + "objective/train/value_min": -0.132080078125, + "objective/train/value_reward_corr": 0.44086993302041505, + "objective/train/value_std": 0.00429534912109375, + "objective/train/weight_avg": 1.0045372247695923, + "objective/train/weighted_lm_loss": 2.712780475616455, + "objective/train/weights_max": 3.3162994384765625, + "objective/train/weights_min": 0.2956310212612152, + "theoretical_loss": 3.7301286847975295, + "tokens_seen": 797900800 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003829789368104313, + "loss": 1.4337, + "theoretical_loss": 3.7301286847975295, + "tokens_seen": 797900800 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003829689067201605, + "loss": 1.4069, + "theoretical_loss": 3.7300983007768105, + "tokens_seen": 797966336 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038295887662988964, + "loss": 1.4338, + "theoretical_loss": 3.730067919950038, + "tokens_seen": 798031872 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003829488465396189, + "loss": 1.2607, + "theoretical_loss": 3.730037542316615, + "tokens_seen": 798097408 + }, + { + "epoch": 0.24, + "learning_rate": 0.000382938816449348, + "loss": 1.3122, + "theoretical_loss": 3.730007167875944, + "tokens_seen": 798162944 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038292878635907724, + "loss": 1.3815, + "theoretical_loss": 3.729976796627426, + "tokens_seen": 798228480 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003829187562688064, + "loss": 1.4444, + "theoretical_loss": 3.7299464285704644, + "tokens_seen": 798294016 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003829087261785356, + "loss": 1.6433, + "theoretical_loss": 3.7299160637044624, + "tokens_seen": 798359552 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003828986960882648, + "loss": 1.4436, + "theoretical_loss": 3.7298857020288216, + "tokens_seen": 798425088 + }, + { + "epoch": 0.24, + "learning_rate": 0.000382888665997994, + "loss": 1.3863, + "theoretical_loss": 3.7298553435429453, + "tokens_seen": 798490624 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003828786359077232, + "loss": 1.4402, + "theoretical_loss": 3.7298249882462375, + "tokens_seen": 798556160 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003828686058174524, + "loss": 1.401, + "theoretical_loss": 3.7297946361381005, + "tokens_seen": 798621696 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038285857572718156, + "loss": 1.3134, + "theoretical_loss": 3.7297642872179377, + "tokens_seen": 798687232 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038284854563691074, + "loss": 1.377, + "theoretical_loss": 3.7297339414851534, + "tokens_seen": 798752768 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038283851554664, + "loss": 1.352, + "theoretical_loss": 3.7297035989391514, + "tokens_seen": 798818304 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003828284854563691, + "loss": 1.2479, + "theoretical_loss": 3.729673259579335, + "tokens_seen": 798883840 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038281845536609834, + "loss": 1.4825, + "theoretical_loss": 3.729642923405108, + "tokens_seen": 798949376 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038280842527582746, + "loss": 1.3764, + "theoretical_loss": 3.729612590415876, + "tokens_seen": 799014912 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003827983951855567, + "loss": 1.4528, + "theoretical_loss": 3.7295822606110423, + "tokens_seen": 799080448 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003827883650952859, + "loss": 1.4065, + "theoretical_loss": 3.7295519339900123, + "tokens_seen": 799145984 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038277833500501506, + "loss": 1.4765, + "theoretical_loss": 3.7295216105521893, + "tokens_seen": 799211520 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038276830491474424, + "loss": 1.3338, + "theoretical_loss": 3.72949129029698, + "tokens_seen": 799277056 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003827582748244735, + "loss": 1.3829, + "theoretical_loss": 3.7294609732237882, + "tokens_seen": 799342592 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003827482447342026, + "loss": 1.4279, + "theoretical_loss": 3.72943065933202, + "tokens_seen": 799408128 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038273821464393184, + "loss": 1.4292, + "theoretical_loss": 3.7294003486210796, + "tokens_seen": 799473664 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -0.0007528486312367022, + "objective/train/docs_used": 460885, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.45682692527771, + "objective/train/original_loss": 2.456827163696289, + "objective/train/theoretical_loss": 3.729370041090373, + "objective/train/tokens_used": 819999200, + "objective/train/value_avg": -0.004924774169921875, + "objective/train/value_loss": 4.0731989429332316e-05, + "objective/train/value_max": -0.0003712177276611328, + "objective/train/value_min": -0.052337646484375, + "objective/train/value_reward_corr": 0.4300162841733819, + "objective/train/value_std": 0.00417327880859375, + "objective/train/weight_avg": 0.994431734085083, + "objective/train/weighted_lm_loss": 2.44399356842041, + "objective/train/weights_max": 1.5000108480453491, + "objective/train/weights_min": 0.41107177734375, + "theoretical_loss": 3.729370041090373, + "tokens_seen": 799539200 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038272818455366097, + "loss": 1.2742, + "theoretical_loss": 3.729370041090373, + "tokens_seen": 799539200 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003827181544633902, + "loss": 1.3654, + "theoretical_loss": 3.7293397367393073, + "tokens_seen": 799604736 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003827081243731194, + "loss": 1.4233, + "theoretical_loss": 3.729309435567286, + "tokens_seen": 799670272 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038269809428284856, + "loss": 1.3061, + "theoretical_loss": 3.729279137573717, + "tokens_seen": 799735808 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038268806419257775, + "loss": 1.3155, + "theoretical_loss": 3.7292488427580053, + "tokens_seen": 799801344 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003826780341023069, + "loss": 1.4381, + "theoretical_loss": 3.729218551119558, + "tokens_seen": 799866880 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003826680040120361, + "loss": 1.3298, + "theoretical_loss": 3.729188262657781, + "tokens_seen": 799932416 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038265797392176534, + "loss": 1.3996, + "theoretical_loss": 3.7291579773720813, + "tokens_seen": 799997952 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038264794383149447, + "loss": 1.3309, + "theoretical_loss": 3.7291276952618655, + "tokens_seen": 800063488 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003826379137412237, + "loss": 1.3552, + "theoretical_loss": 3.7290974163265407, + "tokens_seen": 800129024 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038262788365095283, + "loss": 1.2253, + "theoretical_loss": 3.7290671405655145, + "tokens_seen": 800194560 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038261785356068207, + "loss": 1.3952, + "theoretical_loss": 3.729036867978193, + "tokens_seen": 800260096 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038260782347041125, + "loss": 1.3302, + "theoretical_loss": 3.7290065985639846, + "tokens_seen": 800325632 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038259779338014043, + "loss": 1.4497, + "theoretical_loss": 3.728976332322297, + "tokens_seen": 800391168 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003825877632898696, + "loss": 1.4913, + "theoretical_loss": 3.728946069252537, + "tokens_seen": 800456704 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038257773319959885, + "loss": 1.4509, + "theoretical_loss": 3.7289158093541133, + "tokens_seen": 800522240 + }, + { + "epoch": 0.24, + "learning_rate": 0.000382567703109328, + "loss": 1.3718, + "theoretical_loss": 3.7288855526264344, + "tokens_seen": 800587776 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003825576730190572, + "loss": 1.2903, + "theoretical_loss": 3.7288552990689077, + "tokens_seen": 800653312 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038254764292878634, + "loss": 1.3473, + "theoretical_loss": 3.7288250486809416, + "tokens_seen": 800718848 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038253761283851557, + "loss": 1.4213, + "theoretical_loss": 3.7287948014619454, + "tokens_seen": 800784384 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038252758274824475, + "loss": 1.3624, + "theoretical_loss": 3.728764557411327, + "tokens_seen": 800849920 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038251755265797393, + "loss": 1.3119, + "theoretical_loss": 3.728734316528495, + "tokens_seen": 800915456 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003825075225677031, + "loss": 1.5075, + "theoretical_loss": 3.72870407881286, + "tokens_seen": 800980992 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003824974924774323, + "loss": 1.4259, + "theoretical_loss": 3.72867384426383, + "tokens_seen": 801046528 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003824874623871615, + "loss": 1.2881, + "theoretical_loss": 3.7286436128808145, + "tokens_seen": 801112064 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": 0.000864318513777107, + "objective/train/docs_used": 461399, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9298744201660156, + "objective/train/original_loss": 2.929874897003174, + "objective/train/theoretical_loss": 3.7286133846632232, + "objective/train/tokens_used": 821637600, + "objective/train/value_avg": -0.005321502685546875, + "objective/train/value_loss": 5.7478911912767217e-05, + "objective/train/value_max": -0.0003800392150878906, + "objective/train/value_min": -0.055511474609375, + "objective/train/value_reward_corr": 0.32739394502977254, + "objective/train/value_std": 0.0042724609375, + "objective/train/weight_avg": 1.01140296459198, + "objective/train/weighted_lm_loss": 2.969003438949585, + "objective/train/weights_max": 1.7074023485183716, + "objective/train/weights_min": 0.502540647983551, + "theoretical_loss": 3.7286133846632232, + "tokens_seen": 801177600 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003824774322968907, + "loss": 1.4477, + "theoretical_loss": 3.7286133846632232, + "tokens_seen": 801177600 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038246740220661984, + "loss": 1.3415, + "theoretical_loss": 3.7285831596104657, + "tokens_seen": 801243136 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003824573721163491, + "loss": 1.3782, + "theoretical_loss": 3.7285529377219517, + "tokens_seen": 801308672 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003824473420260782, + "loss": 1.4063, + "theoretical_loss": 3.7285227189970915, + "tokens_seen": 801374208 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038243731193580744, + "loss": 1.345, + "theoretical_loss": 3.728492503435295, + "tokens_seen": 801439744 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003824272818455366, + "loss": 1.4229, + "theoretical_loss": 3.7284622910359726, + "tokens_seen": 801505280 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003824172517552658, + "loss": 1.3206, + "theoretical_loss": 3.728432081798535, + "tokens_seen": 801570816 + }, + { + "epoch": 0.24, + "learning_rate": 0.000382407221664995, + "loss": 1.4081, + "theoretical_loss": 3.7284018757223913, + "tokens_seen": 801636352 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003823971915747242, + "loss": 1.3878, + "theoretical_loss": 3.7283716728069547, + "tokens_seen": 801701888 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038238716148445334, + "loss": 1.3003, + "theoretical_loss": 3.7283414730516347, + "tokens_seen": 801767424 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003823771313941826, + "loss": 1.3933, + "theoretical_loss": 3.7283112764558424, + "tokens_seen": 801832960 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003823671013039117, + "loss": 1.4571, + "theoretical_loss": 3.7282810830189894, + "tokens_seen": 801898496 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038235707121364094, + "loss": 1.4189, + "theoretical_loss": 3.728250892740487, + "tokens_seen": 801964032 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003823470411233701, + "loss": 1.4062, + "theoretical_loss": 3.728220705619747, + "tokens_seen": 802029568 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003823370110330993, + "loss": 1.2692, + "theoretical_loss": 3.72819052165618, + "tokens_seen": 802095104 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003823269809428285, + "loss": 1.3715, + "theoretical_loss": 3.7281603408491995, + "tokens_seen": 802160640 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038231695085255766, + "loss": 1.4273, + "theoretical_loss": 3.728130163198217, + "tokens_seen": 802226176 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038230692076228685, + "loss": 1.3617, + "theoretical_loss": 3.728099988702644, + "tokens_seen": 802291712 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003822968906720161, + "loss": 1.4243, + "theoretical_loss": 3.728069817361894, + "tokens_seen": 802357248 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003822868605817452, + "loss": 1.4475, + "theoretical_loss": 3.7280396491753782, + "tokens_seen": 802422784 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038227683049147444, + "loss": 1.2924, + "theoretical_loss": 3.72800948414251, + "tokens_seen": 802488320 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038226680040120357, + "loss": 1.4002, + "theoretical_loss": 3.7279793222627022, + "tokens_seen": 802553856 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003822567703109328, + "loss": 1.3852, + "theoretical_loss": 3.727949163535368, + "tokens_seen": 802619392 + }, + { + "epoch": 0.24, + "learning_rate": 0.000382246740220662, + "loss": 1.4417, + "theoretical_loss": 3.7279190079599203, + "tokens_seen": 802684928 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038223671013039117, + "loss": 1.4009, + "theoretical_loss": 3.727888855535772, + "tokens_seen": 802750464 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -0.0012744329869747162, + "objective/train/docs_used": 462084, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.041008710861206, + "objective/train/original_loss": 3.0410079956054688, + "objective/train/theoretical_loss": 3.727858706262338, + "objective/train/tokens_used": 823276000, + "objective/train/value_avg": -0.00604248046875, + "objective/train/value_loss": 0.00010020803892984986, + "objective/train/value_max": -0.00028014183044433594, + "objective/train/value_min": -0.0592041015625, + "objective/train/value_reward_corr": 0.3810192495544001, + "objective/train/value_std": 0.004604339599609375, + "objective/train/weight_avg": 0.9917526245117188, + "objective/train/weighted_lm_loss": 3.0160741806030273, + "objective/train/weights_max": 1.7470197677612305, + "objective/train/weights_min": 0.24185286462306976, + "theoretical_loss": 3.727858706262338, + "tokens_seen": 802816000 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038222668004012035, + "loss": 1.4876, + "theoretical_loss": 3.727858706262338, + "tokens_seen": 802816000 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003822166499498496, + "loss": 1.4007, + "theoretical_loss": 3.7278285601390295, + "tokens_seen": 802881536 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003822066198595787, + "loss": 1.4453, + "theoretical_loss": 3.7277984171652623, + "tokens_seen": 802947072 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038219658976930795, + "loss": 1.4097, + "theoretical_loss": 3.7277682773404495, + "tokens_seen": 803012608 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003821865596790371, + "loss": 1.3851, + "theoretical_loss": 3.7277381406640053, + "tokens_seen": 803078144 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003821765295887663, + "loss": 1.4474, + "theoretical_loss": 3.7277080071353437, + "tokens_seen": 803143680 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003821664994984955, + "loss": 1.4614, + "theoretical_loss": 3.72767787675388, + "tokens_seen": 803209216 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038215646940822467, + "loss": 1.4249, + "theoretical_loss": 3.727647749519028, + "tokens_seen": 803274752 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038214643931795385, + "loss": 1.4026, + "theoretical_loss": 3.7276176254302023, + "tokens_seen": 803340288 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038213640922768303, + "loss": 1.4035, + "theoretical_loss": 3.727587504486818, + "tokens_seen": 803405824 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038212637913741227, + "loss": 1.2849, + "theoretical_loss": 3.7275573866882903, + "tokens_seen": 803471360 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038211634904714145, + "loss": 1.4235, + "theoretical_loss": 3.727527272034034, + "tokens_seen": 803536896 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038210631895687063, + "loss": 1.5186, + "theoretical_loss": 3.7274971605234644, + "tokens_seen": 803602432 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003820962888665998, + "loss": 1.4535, + "theoretical_loss": 3.7274670521559976, + "tokens_seen": 803667968 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038208625877632905, + "loss": 1.4084, + "theoretical_loss": 3.727436946931049, + "tokens_seen": 803733504 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003820762286860582, + "loss": 1.4553, + "theoretical_loss": 3.727406844848034, + "tokens_seen": 803799040 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003820661985957874, + "loss": 1.4224, + "theoretical_loss": 3.7273767459063687, + "tokens_seen": 803864576 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038205616850551654, + "loss": 1.4723, + "theoretical_loss": 3.7273466501054697, + "tokens_seen": 803930112 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038204613841524577, + "loss": 1.4003, + "theoretical_loss": 3.7273165574447527, + "tokens_seen": 803995648 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038203610832497495, + "loss": 1.4127, + "theoretical_loss": 3.7272864679236344, + "tokens_seen": 804061184 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038202607823470413, + "loss": 1.3739, + "theoretical_loss": 3.7272563815415314, + "tokens_seen": 804126720 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003820160481444333, + "loss": 1.3786, + "theoretical_loss": 3.727226298297861, + "tokens_seen": 804192256 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003820060180541625, + "loss": 1.4204, + "theoretical_loss": 3.727196218192039, + "tokens_seen": 804257792 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003819959879638917, + "loss": 1.437, + "theoretical_loss": 3.727166141223482, + "tokens_seen": 804323328 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003819859578736209, + "loss": 1.2929, + "theoretical_loss": 3.7271360673916094, + "tokens_seen": 804388864 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -5.756829705205746e-05, + "objective/train/docs_used": 463313, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1441853046417236, + "objective/train/original_loss": 3.1441850662231445, + "objective/train/theoretical_loss": 3.727105996695837, + "objective/train/tokens_used": 824914400, + "objective/train/value_avg": -0.006626129150390625, + "objective/train/value_loss": 0.00011311288108117878, + "objective/train/value_max": -0.0005593299865722656, + "objective/train/value_min": -0.06927490234375, + "objective/train/value_reward_corr": 0.41718870585092827, + "objective/train/value_std": 0.005779266357421875, + "objective/train/weight_avg": 1.004387378692627, + "objective/train/weighted_lm_loss": 3.1599416732788086, + "objective/train/weights_max": 1.6255396604537964, + "objective/train/weights_min": 0.30360838770866394, + "theoretical_loss": 3.727105996695837, + "tokens_seen": 804454400 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038197592778335004, + "loss": 1.4054, + "theoretical_loss": 3.727105996695837, + "tokens_seen": 804454400 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003819658976930793, + "loss": 1.4199, + "theoretical_loss": 3.727075929135583, + "tokens_seen": 804519936 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003819558676028084, + "loss": 1.283, + "theoretical_loss": 3.727045864710264, + "tokens_seen": 804585472 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038194583751253764, + "loss": 1.4875, + "theoretical_loss": 3.727015803419299, + "tokens_seen": 804651008 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003819358074222668, + "loss": 1.264, + "theoretical_loss": 3.7269857452621054, + "tokens_seen": 804716544 + }, + { + "epoch": 0.24, + "learning_rate": 0.000381925777331996, + "loss": 1.3175, + "theoretical_loss": 3.7269556902381007, + "tokens_seen": 804782080 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003819157472417252, + "loss": 1.4531, + "theoretical_loss": 3.726925638346705, + "tokens_seen": 804847616 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003819057171514544, + "loss": 1.4203, + "theoretical_loss": 3.7268955895873352, + "tokens_seen": 804913152 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038189568706118354, + "loss": 1.3737, + "theoretical_loss": 3.726865543959411, + "tokens_seen": 804978688 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003818856569709128, + "loss": 1.5263, + "theoretical_loss": 3.72683550146235, + "tokens_seen": 805044224 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003818756268806419, + "loss": 1.3427, + "theoretical_loss": 3.726805462095572, + "tokens_seen": 805109760 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038186559679037114, + "loss": 1.412, + "theoretical_loss": 3.726775425858496, + "tokens_seen": 805175296 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003818555667001003, + "loss": 1.4677, + "theoretical_loss": 3.7267453927505407, + "tokens_seen": 805240832 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003818455366098295, + "loss": 1.6029, + "theoretical_loss": 3.7267153627711256, + "tokens_seen": 805306368 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003818355065195587, + "loss": 1.4403, + "theoretical_loss": 3.726685335919671, + "tokens_seen": 805371904 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038182547642928786, + "loss": 1.4505, + "theoretical_loss": 3.7266553121955956, + "tokens_seen": 805437440 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038181544633901705, + "loss": 1.4519, + "theoretical_loss": 3.72662529159832, + "tokens_seen": 805502976 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003818054162487463, + "loss": 1.5089, + "theoretical_loss": 3.7265952741272637, + "tokens_seen": 805568512 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003817953861584754, + "loss": 1.3892, + "theoretical_loss": 3.726565259781847, + "tokens_seen": 805634048 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038178535606820464, + "loss": 1.4028, + "theoretical_loss": 3.7265352485614907, + "tokens_seen": 805699584 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038177532597793377, + "loss": 1.4192, + "theoretical_loss": 3.726505240465614, + "tokens_seen": 805765120 + }, + { + "epoch": 0.24, + "learning_rate": 0.000381765295887663, + "loss": 1.455, + "theoretical_loss": 3.726475235493639, + "tokens_seen": 805830656 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003817552657973922, + "loss": 1.394, + "theoretical_loss": 3.726445233644986, + "tokens_seen": 805896192 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038174523570712137, + "loss": 1.4828, + "theoretical_loss": 3.726415234919075, + "tokens_seen": 805961728 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038173520561685055, + "loss": 1.3261, + "theoretical_loss": 3.7263852393153285, + "tokens_seen": 806027264 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -0.001847331179305911, + "objective/train/docs_used": 464591, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5792036056518555, + "objective/train/original_loss": 2.5792033672332764, + "objective/train/theoretical_loss": 3.7263552468331667, + "objective/train/tokens_used": 826552800, + "objective/train/value_avg": -0.004894256591796875, + "objective/train/value_loss": 9.817386307986453e-05, + "objective/train/value_max": -0.0002892017364501953, + "objective/train/value_min": -0.07122802734375, + "objective/train/value_reward_corr": 0.3941271126787699, + "objective/train/value_std": 0.004291534423828125, + "objective/train/weight_avg": 0.9859204888343811, + "objective/train/weighted_lm_loss": 2.553748846054077, + "objective/train/weights_max": 1.9115272760391235, + "objective/train/weights_min": 0.37204688787460327, + "theoretical_loss": 3.7263552468331667, + "tokens_seen": 806092800 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003817251755265798, + "loss": 1.4591, + "theoretical_loss": 3.7263552468331667, + "tokens_seen": 806092800 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003817151454363089, + "loss": 1.3571, + "theoretical_loss": 3.726325257472012, + "tokens_seen": 806158336 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038170511534603815, + "loss": 1.4466, + "theoretical_loss": 3.7262952712312845, + "tokens_seen": 806223872 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003816950852557673, + "loss": 1.4007, + "theoretical_loss": 3.7262652881104072, + "tokens_seen": 806289408 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003816850551654965, + "loss": 1.3712, + "theoretical_loss": 3.7262353081088015, + "tokens_seen": 806354944 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003816750250752257, + "loss": 1.4443, + "theoretical_loss": 3.726205331225889, + "tokens_seen": 806420480 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038166499498495487, + "loss": 1.3954, + "theoretical_loss": 3.726175357461093, + "tokens_seen": 806486016 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038165496489468405, + "loss": 1.4037, + "theoretical_loss": 3.7261453868138346, + "tokens_seen": 806551552 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038164493480441323, + "loss": 1.4705, + "theoretical_loss": 3.726115419283537, + "tokens_seen": 806617088 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003816349047141424, + "loss": 1.4448, + "theoretical_loss": 3.7260854548696223, + "tokens_seen": 806682624 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038162487462387165, + "loss": 1.345, + "theoretical_loss": 3.7260554935715136, + "tokens_seen": 806748160 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003816148445336008, + "loss": 1.3715, + "theoretical_loss": 3.726025535388634, + "tokens_seen": 806813696 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038160481444333, + "loss": 1.3818, + "theoretical_loss": 3.725995580320406, + "tokens_seen": 806879232 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038159478435305914, + "loss": 1.3913, + "theoretical_loss": 3.7259656283662537, + "tokens_seen": 806944768 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003815847542627884, + "loss": 1.4565, + "theoretical_loss": 3.7259356795256, + "tokens_seen": 807010304 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038157472417251756, + "loss": 1.3997, + "theoretical_loss": 3.7259057337978687, + "tokens_seen": 807075840 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038156469408224674, + "loss": 1.5548, + "theoretical_loss": 3.7258757911824825, + "tokens_seen": 807141376 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003815546639919759, + "loss": 1.3276, + "theoretical_loss": 3.7258458516788666, + "tokens_seen": 807206912 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038154463390170515, + "loss": 1.4519, + "theoretical_loss": 3.7258159152864447, + "tokens_seen": 807272448 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003815346038114343, + "loss": 1.3157, + "theoretical_loss": 3.7257859820046404, + "tokens_seen": 807337984 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003815245737211635, + "loss": 1.3589, + "theoretical_loss": 3.725756051832878, + "tokens_seen": 807403520 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038151454363089264, + "loss": 1.4225, + "theoretical_loss": 3.725726124770583, + "tokens_seen": 807469056 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003815045135406219, + "loss": 1.4341, + "theoretical_loss": 3.725696200817179, + "tokens_seen": 807534592 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038149448345035106, + "loss": 1.5412, + "theoretical_loss": 3.725666279972091, + "tokens_seen": 807600128 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038148445336008024, + "loss": 1.3569, + "theoretical_loss": 3.725636362234744, + "tokens_seen": 807665664 + }, + { + "epoch": 0.24, + "objective/train/advantage_avg": -0.0009310911991633475, + "objective/train/docs_used": 465345, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4455678462982178, + "objective/train/original_loss": 2.4455676078796387, + "objective/train/theoretical_loss": 3.725606447604563, + "objective/train/tokens_used": 828191200, + "objective/train/value_avg": -0.00429534912109375, + "objective/train/value_loss": 5.144343958818354e-05, + "objective/train/value_max": -0.00028252601623535156, + "objective/train/value_min": -0.06671142578125, + "objective/train/value_reward_corr": 0.33819317929775994, + "objective/train/value_std": 0.0033626556396484375, + "objective/train/weight_avg": 0.9930817484855652, + "objective/train/weighted_lm_loss": 2.432213544845581, + "objective/train/weights_max": 1.948606014251709, + "objective/train/weights_min": 0.2631087899208069, + "theoretical_loss": 3.725606447604563, + "tokens_seen": 807731200 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003814744232698094, + "loss": 1.3948, + "theoretical_loss": 3.725606447604563, + "tokens_seen": 807731200 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003814643931795386, + "loss": 1.3745, + "theoretical_loss": 3.725576536080974, + "tokens_seen": 807796736 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003814543630892678, + "loss": 1.3642, + "theoretical_loss": 3.7255466276634013, + "tokens_seen": 807862272 + }, + { + "epoch": 0.24, + "learning_rate": 0.000381444332998997, + "loss": 1.4704, + "theoretical_loss": 3.725516722351271, + "tokens_seen": 807927808 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038143430290872614, + "loss": 1.4322, + "theoretical_loss": 3.725486820144009, + "tokens_seen": 807993344 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003814242728184554, + "loss": 1.3381, + "theoretical_loss": 3.7254569210410398, + "tokens_seen": 808058880 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038141424272818456, + "loss": 1.4041, + "theoretical_loss": 3.725427025041791, + "tokens_seen": 808124416 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038140421263791374, + "loss": 1.3682, + "theoretical_loss": 3.7253971321456882, + "tokens_seen": 808189952 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003813941825476429, + "loss": 1.3318, + "theoretical_loss": 3.7253672423521578, + "tokens_seen": 808255488 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003813841524573721, + "loss": 1.4248, + "theoretical_loss": 3.725337355660626, + "tokens_seen": 808321024 + }, + { + "epoch": 0.24, + "learning_rate": 0.00038137412236710134, + "loss": 1.4674, + "theoretical_loss": 3.7253074720705195, + "tokens_seen": 808386560 + }, + { + "epoch": 0.24, + "learning_rate": 0.0003813640922768305, + "loss": 1.4344, + "theoretical_loss": 3.7252775915812655, + "tokens_seen": 808452096 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003813540621865597, + "loss": 1.3254, + "theoretical_loss": 3.72524771419229, + "tokens_seen": 808517632 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003813440320962889, + "loss": 1.4644, + "theoretical_loss": 3.725217839903021, + "tokens_seen": 808583168 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038133400200601806, + "loss": 1.3537, + "theoretical_loss": 3.725187968712885, + "tokens_seen": 808648704 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038132397191574725, + "loss": 1.435, + "theoretical_loss": 3.72515810062131, + "tokens_seen": 808714240 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003813139418254765, + "loss": 1.48, + "theoretical_loss": 3.7251282356277233, + "tokens_seen": 808779776 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003813039117352056, + "loss": 1.4694, + "theoretical_loss": 3.725098373731552, + "tokens_seen": 808845312 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038129388164493484, + "loss": 1.4702, + "theoretical_loss": 3.7250685149322247, + "tokens_seen": 808910848 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038128385155466397, + "loss": 1.4386, + "theoretical_loss": 3.7250386592291695, + "tokens_seen": 808976384 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003812738214643932, + "loss": 1.4674, + "theoretical_loss": 3.725008806621813, + "tokens_seen": 809041920 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003812637913741224, + "loss": 1.372, + "theoretical_loss": 3.7249789571095855, + "tokens_seen": 809107456 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038125376128385157, + "loss": 1.5306, + "theoretical_loss": 3.7249491106919144, + "tokens_seen": 809172992 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038124373119358075, + "loss": 1.3512, + "theoretical_loss": 3.7249192673682283, + "tokens_seen": 809238528 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038123370110331, + "loss": 1.5831, + "theoretical_loss": 3.7248894271379562, + "tokens_seen": 809304064 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.001057670102454722, + "objective/train/docs_used": 466409, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8903310298919678, + "objective/train/original_loss": 2.8903310298919678, + "objective/train/theoretical_loss": 3.724859590000527, + "objective/train/tokens_used": 829829600, + "objective/train/value_avg": -0.005146026611328125, + "objective/train/value_loss": 6.938383012311533e-05, + "objective/train/value_max": -0.00019109249114990234, + "objective/train/value_min": -0.057281494140625, + "objective/train/value_reward_corr": 0.46249111345525407, + "objective/train/value_std": 0.0047454833984375, + "objective/train/weight_avg": 0.9926656484603882, + "objective/train/weighted_lm_loss": 2.8749990463256836, + "objective/train/weights_max": 1.6101146936416626, + "objective/train/weights_min": 0.32276561856269836, + "theoretical_loss": 3.724859590000527, + "tokens_seen": 809369600 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003812236710130391, + "loss": 1.3668, + "theoretical_loss": 3.724859590000527, + "tokens_seen": 809369600 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038121364092276835, + "loss": 1.4214, + "theoretical_loss": 3.7248297559553696, + "tokens_seen": 809435136 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003812036108324975, + "loss": 1.4389, + "theoretical_loss": 3.724799925001913, + "tokens_seen": 809500672 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003811935807422267, + "loss": 1.3424, + "theoretical_loss": 3.7247700971395865, + "tokens_seen": 809566208 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003811835506519559, + "loss": 1.4728, + "theoretical_loss": 3.7247402723678205, + "tokens_seen": 809631744 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038117352056168507, + "loss": 1.4618, + "theoretical_loss": 3.7247104506860436, + "tokens_seen": 809697280 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038116349047141425, + "loss": 1.464, + "theoretical_loss": 3.724680632093686, + "tokens_seen": 809762816 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038115346038114343, + "loss": 1.315, + "theoretical_loss": 3.7246508165901777, + "tokens_seen": 809828352 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003811434302908726, + "loss": 1.3775, + "theoretical_loss": 3.7246210041749492, + "tokens_seen": 809893888 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038113340020060185, + "loss": 1.4617, + "theoretical_loss": 3.7245911948474295, + "tokens_seen": 809959424 + }, + { + "epoch": 0.25, + "learning_rate": 0.000381123370110331, + "loss": 1.4257, + "theoretical_loss": 3.72456138860705, + "tokens_seen": 810024960 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003811133400200602, + "loss": 1.4033, + "theoretical_loss": 3.7245315854532417, + "tokens_seen": 810090496 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038110330992978934, + "loss": 1.442, + "theoretical_loss": 3.724501785385434, + "tokens_seen": 810156032 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003810932798395186, + "loss": 1.3441, + "theoretical_loss": 3.7244719884030593, + "tokens_seen": 810221568 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038108324974924776, + "loss": 1.4754, + "theoretical_loss": 3.7244421945055466, + "tokens_seen": 810287104 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038107321965897694, + "loss": 1.4439, + "theoretical_loss": 3.7244124036923294, + "tokens_seen": 810352640 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003810631895687061, + "loss": 1.2826, + "theoretical_loss": 3.7243826159628366, + "tokens_seen": 810418176 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038105315947843535, + "loss": 1.3703, + "theoretical_loss": 3.7243528313165015, + "tokens_seen": 810483712 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003810431293881645, + "loss": 1.3773, + "theoretical_loss": 3.7243230497527553, + "tokens_seen": 810549248 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003810330992978937, + "loss": 1.3815, + "theoretical_loss": 3.724293271271029, + "tokens_seen": 810614784 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038102306920762284, + "loss": 1.3957, + "theoretical_loss": 3.7242634958707557, + "tokens_seen": 810680320 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003810130391173521, + "loss": 1.4165, + "theoretical_loss": 3.724233723551367, + "tokens_seen": 810745856 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038100300902708126, + "loss": 1.4076, + "theoretical_loss": 3.7242039543122942, + "tokens_seen": 810811392 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038099297893681044, + "loss": 1.4289, + "theoretical_loss": 3.724174188152971, + "tokens_seen": 810876928 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003809829488465396, + "loss": 1.5026, + "theoretical_loss": 3.7241444250728293, + "tokens_seen": 810942464 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0002955579257104546, + "objective/train/docs_used": 466961, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7383389472961426, + "objective/train/original_loss": 2.7383391857147217, + "objective/train/theoretical_loss": 3.7241146650713013, + "objective/train/tokens_used": 831468000, + "objective/train/value_avg": -0.00516510009765625, + "objective/train/value_loss": 7.998975343070924e-05, + "objective/train/value_max": -0.0003101825714111328, + "objective/train/value_min": -0.040618896484375, + "objective/train/value_reward_corr": 0.3448163319501015, + "objective/train/value_std": 0.004241943359375, + "objective/train/weight_avg": 1.0005390644073486, + "objective/train/weighted_lm_loss": 2.7347660064697266, + "objective/train/weights_max": 1.5010861158370972, + "objective/train/weights_min": 0.08906028419733047, + "theoretical_loss": 3.7241146650713013, + "tokens_seen": 811008000 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003809729187562688, + "loss": 1.3641, + "theoretical_loss": 3.7241146650713013, + "tokens_seen": 811008000 + }, + { + "epoch": 0.25, + "learning_rate": 0.000380962888665998, + "loss": 1.5072, + "theoretical_loss": 3.724084908147821, + "tokens_seen": 811073536 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003809528585757272, + "loss": 1.34, + "theoretical_loss": 3.72405515430182, + "tokens_seen": 811139072 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038094282848545635, + "loss": 1.501, + "theoretical_loss": 3.7240254035327327, + "tokens_seen": 811204608 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003809327983951856, + "loss": 1.5097, + "theoretical_loss": 3.7239956558399916, + "tokens_seen": 811270144 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038092276830491476, + "loss": 1.4482, + "theoretical_loss": 3.7239659112230306, + "tokens_seen": 811335680 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038091273821464394, + "loss": 1.4582, + "theoretical_loss": 3.7239361696812825, + "tokens_seen": 811401216 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003809027081243731, + "loss": 1.4588, + "theoretical_loss": 3.7239064312141816, + "tokens_seen": 811466752 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003808926780341023, + "loss": 1.5298, + "theoretical_loss": 3.723876695821162, + "tokens_seen": 811532288 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003808826479438315, + "loss": 1.4002, + "theoretical_loss": 3.723846963501657, + "tokens_seen": 811597824 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003808726178535607, + "loss": 1.417, + "theoretical_loss": 3.723817234255101, + "tokens_seen": 811663360 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038086258776328985, + "loss": 1.4729, + "theoretical_loss": 3.723787508080929, + "tokens_seen": 811728896 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003808525576730191, + "loss": 1.4242, + "theoretical_loss": 3.7237577849785746, + "tokens_seen": 811794432 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003808425275827482, + "loss": 1.5763, + "theoretical_loss": 3.7237280649474727, + "tokens_seen": 811859968 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038083249749247745, + "loss": 1.4163, + "theoretical_loss": 3.7236983479870585, + "tokens_seen": 811925504 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003808224674022066, + "loss": 1.374, + "theoretical_loss": 3.723668634096766, + "tokens_seen": 811991040 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003808124373119358, + "loss": 1.2963, + "theoretical_loss": 3.7236389232760314, + "tokens_seen": 812056576 + }, + { + "epoch": 0.25, + "learning_rate": 0.000380802407221665, + "loss": 1.4442, + "theoretical_loss": 3.7236092155242893, + "tokens_seen": 812122112 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038079237713139417, + "loss": 1.4647, + "theoretical_loss": 3.723579510840975, + "tokens_seen": 812187648 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038078234704112335, + "loss": 1.4448, + "theoretical_loss": 3.7235498092255237, + "tokens_seen": 812253184 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003807723169508526, + "loss": 1.3161, + "theoretical_loss": 3.723520110677372, + "tokens_seen": 812318720 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003807622868605817, + "loss": 1.4025, + "theoretical_loss": 3.723490415195955, + "tokens_seen": 812384256 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038075225677031095, + "loss": 1.4703, + "theoretical_loss": 3.7234607227807084, + "tokens_seen": 812449792 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038074222668004013, + "loss": 1.3847, + "theoretical_loss": 3.723431033431069, + "tokens_seen": 812515328 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003807321965897693, + "loss": 1.4108, + "theoretical_loss": 3.7234013471464733, + "tokens_seen": 812580864 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.001644170843064785, + "objective/train/docs_used": 468484, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.112196922302246, + "objective/train/original_loss": 3.112196445465088, + "objective/train/theoretical_loss": 3.7233716639263568, + "objective/train/tokens_used": 833106400, + "objective/train/value_avg": -0.0056915283203125, + "objective/train/value_loss": 0.00012102723121643066, + "objective/train/value_max": -0.0002613067626953125, + "objective/train/value_min": -0.050811767578125, + "objective/train/value_reward_corr": 0.3295320720763483, + "objective/train/value_std": 0.004085540771484375, + "objective/train/weight_avg": 0.9889106154441833, + "objective/train/weighted_lm_loss": 3.0898594856262207, + "objective/train/weights_max": 1.6621595621109009, + "objective/train/weights_min": 0.29665902256965637, + "theoretical_loss": 3.7233716639263568, + "tokens_seen": 812646400 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003807221664994985, + "loss": 1.3905, + "theoretical_loss": 3.7233716639263568, + "tokens_seen": 812646400 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003807121364092277, + "loss": 1.3745, + "theoretical_loss": 3.7233419837701565, + "tokens_seen": 812711936 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038070210631895685, + "loss": 1.4991, + "theoretical_loss": 3.7233123066773097, + "tokens_seen": 812777472 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003806920762286861, + "loss": 1.3684, + "theoretical_loss": 3.723282632647252, + "tokens_seen": 812843008 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003806820461384152, + "loss": 1.3842, + "theoretical_loss": 3.723252961679421, + "tokens_seen": 812908544 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038067201604814445, + "loss": 1.3418, + "theoretical_loss": 3.7232232937732546, + "tokens_seen": 812974080 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003806619859578736, + "loss": 1.4, + "theoretical_loss": 3.723193628928189, + "tokens_seen": 813039616 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003806519558676028, + "loss": 1.4498, + "theoretical_loss": 3.723163967143662, + "tokens_seen": 813105152 + }, + { + "epoch": 0.25, + "learning_rate": 0.000380641925777332, + "loss": 1.4766, + "theoretical_loss": 3.7231343084191115, + "tokens_seen": 813170688 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003806318956870612, + "loss": 1.4393, + "theoretical_loss": 3.723104652753975, + "tokens_seen": 813236224 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003806218655967904, + "loss": 1.4827, + "theoretical_loss": 3.7230750001476904, + "tokens_seen": 813301760 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038061183550651954, + "loss": 1.4559, + "theoretical_loss": 3.723045350599696, + "tokens_seen": 813367296 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003806018054162488, + "loss": 1.3838, + "theoretical_loss": 3.7230157041094296, + "tokens_seen": 813432832 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038059177532597796, + "loss": 1.4624, + "theoretical_loss": 3.7229860606763294, + "tokens_seen": 813498368 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038058174523570714, + "loss": 1.426, + "theoretical_loss": 3.722956420299835, + "tokens_seen": 813563904 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003805717151454363, + "loss": 1.3216, + "theoretical_loss": 3.7229267829793837, + "tokens_seen": 813629440 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038056168505516555, + "loss": 1.4753, + "theoretical_loss": 3.7228971487144147, + "tokens_seen": 813694976 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003805516549648947, + "loss": 1.4793, + "theoretical_loss": 3.7228675175043673, + "tokens_seen": 813760512 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003805416248746239, + "loss": 1.4609, + "theoretical_loss": 3.7228378893486807, + "tokens_seen": 813826048 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038053159478435304, + "loss": 1.3252, + "theoretical_loss": 3.7228082642467935, + "tokens_seen": 813891584 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003805215646940823, + "loss": 1.3754, + "theoretical_loss": 3.7227786421981452, + "tokens_seen": 813957120 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038051153460381146, + "loss": 1.3687, + "theoretical_loss": 3.722749023202176, + "tokens_seen": 814022656 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038050150451354064, + "loss": 1.4811, + "theoretical_loss": 3.722719407258325, + "tokens_seen": 814088192 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003804914744232698, + "loss": 1.4592, + "theoretical_loss": 3.7226897943660315, + "tokens_seen": 814153728 + }, + { + "epoch": 0.25, + "learning_rate": 0.000380481444332999, + "loss": 1.4086, + "theoretical_loss": 3.722660184524737, + "tokens_seen": 814219264 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": 0.0008986633038148284, + "objective/train/docs_used": 469177, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9180748462677, + "objective/train/original_loss": 2.9180750846862793, + "objective/train/theoretical_loss": 3.722630577733881, + "objective/train/tokens_used": 834744800, + "objective/train/value_avg": -0.005084991455078125, + "objective/train/value_loss": 5.24194874742534e-05, + "objective/train/value_max": -0.00033545494079589844, + "objective/train/value_min": -0.05126953125, + "objective/train/value_reward_corr": 0.27634730678144315, + "objective/train/value_std": 0.0038700103759765625, + "objective/train/weight_avg": 1.0112322568893433, + "objective/train/weighted_lm_loss": 2.951798677444458, + "objective/train/weights_max": 1.5330299139022827, + "objective/train/weights_min": 0.14775708317756653, + "theoretical_loss": 3.722630577733881, + "tokens_seen": 814284800 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003804714142427282, + "loss": 1.4899, + "theoretical_loss": 3.722630577733881, + "tokens_seen": 814284800 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003804613841524574, + "loss": 1.4035, + "theoretical_loss": 3.7226009739929022, + "tokens_seen": 814350336 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038045135406218655, + "loss": 1.4542, + "theoretical_loss": 3.7225713733012435, + "tokens_seen": 814415872 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003804413239719158, + "loss": 1.4238, + "theoretical_loss": 3.7225417756583434, + "tokens_seen": 814481408 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038043129388164496, + "loss": 1.3894, + "theoretical_loss": 3.722512181063644, + "tokens_seen": 814546944 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038042126379137414, + "loss": 1.2962, + "theoretical_loss": 3.722482589516585, + "tokens_seen": 814612480 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003804112337011033, + "loss": 1.3636, + "theoretical_loss": 3.722453001016609, + "tokens_seen": 814678016 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003804012036108325, + "loss": 1.273, + "theoretical_loss": 3.722423415563156, + "tokens_seen": 814743552 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003803911735205617, + "loss": 1.359, + "theoretical_loss": 3.7223938331556674, + "tokens_seen": 814809088 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003803811434302909, + "loss": 1.4912, + "theoretical_loss": 3.722364253793585, + "tokens_seen": 814874624 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038037111334002005, + "loss": 1.487, + "theoretical_loss": 3.7223346774763497, + "tokens_seen": 814940160 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003803610832497493, + "loss": 1.3918, + "theoretical_loss": 3.7223051042034037, + "tokens_seen": 815005696 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003803510531594784, + "loss": 1.4334, + "theoretical_loss": 3.7222755339741895, + "tokens_seen": 815071232 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038034102306920765, + "loss": 1.4905, + "theoretical_loss": 3.7222459667881482, + "tokens_seen": 815136768 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038033099297893683, + "loss": 1.4868, + "theoretical_loss": 3.7222164026447224, + "tokens_seen": 815202304 + }, + { + "epoch": 0.25, + "learning_rate": 0.000380320962888666, + "loss": 1.5147, + "theoretical_loss": 3.7221868415433543, + "tokens_seen": 815267840 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003803109327983952, + "loss": 1.2979, + "theoretical_loss": 3.7221572834834866, + "tokens_seen": 815333376 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038030090270812437, + "loss": 1.5378, + "theoretical_loss": 3.7221277284645615, + "tokens_seen": 815398912 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038029087261785355, + "loss": 1.3964, + "theoretical_loss": 3.722098176486022, + "tokens_seen": 815464448 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003802808425275828, + "loss": 1.4887, + "theoretical_loss": 3.7220686275473116, + "tokens_seen": 815529984 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003802708124373119, + "loss": 1.4854, + "theoretical_loss": 3.722039081647873, + "tokens_seen": 815595520 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038026078234704115, + "loss": 1.4062, + "theoretical_loss": 3.7220095387871486, + "tokens_seen": 815661056 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038025075225677033, + "loss": 1.3757, + "theoretical_loss": 3.7219799989645828, + "tokens_seen": 815726592 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003802407221664995, + "loss": 1.3243, + "theoretical_loss": 3.7219504621796187, + "tokens_seen": 815792128 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003802306920762287, + "loss": 1.47, + "theoretical_loss": 3.7219209284317, + "tokens_seen": 815857664 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0009110237006098032, + "objective/train/docs_used": 469961, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8797571659088135, + "objective/train/original_loss": 2.8797569274902344, + "objective/train/theoretical_loss": 3.7218913977202703, + "objective/train/tokens_used": 836383200, + "objective/train/value_avg": -0.00514984130859375, + "objective/train/value_loss": 6.19627462583594e-05, + "objective/train/value_max": -0.0002694129943847656, + "objective/train/value_min": -0.0703125, + "objective/train/value_reward_corr": 0.4814326407386729, + "objective/train/value_std": 0.00482940673828125, + "objective/train/weight_avg": 0.9938156604766846, + "objective/train/weighted_lm_loss": 2.8676586151123047, + "objective/train/weights_max": 1.6779588460922241, + "objective/train/weights_min": 0.38558143377304077, + "theoretical_loss": 3.7218913977202703, + "tokens_seen": 815923200 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003802206619859579, + "loss": 1.4006, + "theoretical_loss": 3.7218913977202703, + "tokens_seen": 815923200 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038021063189568705, + "loss": 1.4592, + "theoretical_loss": 3.7218618700447736, + "tokens_seen": 815988736 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003802006018054163, + "loss": 1.5327, + "theoretical_loss": 3.7218323454046547, + "tokens_seen": 816054272 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003801905717151454, + "loss": 1.3897, + "theoretical_loss": 3.721802823799357, + "tokens_seen": 816119808 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038018054162487465, + "loss": 1.4897, + "theoretical_loss": 3.721773305228325, + "tokens_seen": 816185344 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003801705115346038, + "loss": 1.4532, + "theoretical_loss": 3.7217437896910037, + "tokens_seen": 816250880 + }, + { + "epoch": 0.25, + "learning_rate": 0.000380160481444333, + "loss": 1.428, + "theoretical_loss": 3.7217142771868374, + "tokens_seen": 816316416 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003801504513540622, + "loss": 1.4549, + "theoretical_loss": 3.721684767715271, + "tokens_seen": 816381952 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003801404212637914, + "loss": 1.3726, + "theoretical_loss": 3.7216552612757496, + "tokens_seen": 816447488 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038013039117352056, + "loss": 1.4735, + "theoretical_loss": 3.721625757867718, + "tokens_seen": 816513024 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038012036108324974, + "loss": 1.3597, + "theoretical_loss": 3.7215962574906216, + "tokens_seen": 816578560 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003801103309929789, + "loss": 1.471, + "theoretical_loss": 3.721566760143906, + "tokens_seen": 816644096 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038010030090270816, + "loss": 1.3427, + "theoretical_loss": 3.721537265827017, + "tokens_seen": 816709632 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003800902708124373, + "loss": 1.4405, + "theoretical_loss": 3.7215077745393996, + "tokens_seen": 816775168 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003800802407221665, + "loss": 1.5484, + "theoretical_loss": 3.7214782862805, + "tokens_seen": 816840704 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003800702106318957, + "loss": 1.4485, + "theoretical_loss": 3.721448801049764, + "tokens_seen": 816906240 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003800601805416249, + "loss": 1.4337, + "theoretical_loss": 3.721419318846638, + "tokens_seen": 816971776 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038005015045135406, + "loss": 1.4684, + "theoretical_loss": 3.721389839670568, + "tokens_seen": 817037312 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038004012036108324, + "loss": 1.3747, + "theoretical_loss": 3.7213603635210015, + "tokens_seen": 817102848 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003800300902708124, + "loss": 1.4163, + "theoretical_loss": 3.7213308903973834, + "tokens_seen": 817168384 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038002006018054166, + "loss": 1.3712, + "theoretical_loss": 3.7213014202991612, + "tokens_seen": 817233920 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003800100300902708, + "loss": 1.427, + "theoretical_loss": 3.7212719532257825, + "tokens_seen": 817299456 + }, + { + "epoch": 0.25, + "learning_rate": 0.00038, + "loss": 1.4275, + "theoretical_loss": 3.721242489176693, + "tokens_seen": 817364992 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037998996990972915, + "loss": 1.4187, + "theoretical_loss": 3.721213028151341, + "tokens_seen": 817430528 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003799799398194584, + "loss": 1.4087, + "theoretical_loss": 3.721183570149173, + "tokens_seen": 817496064 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0002934815420303494, + "objective/train/docs_used": 471203, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7157368659973145, + "objective/train/original_loss": 2.7157366275787354, + "objective/train/theoretical_loss": 3.721154115169637, + "objective/train/tokens_used": 838021600, + "objective/train/value_avg": -0.005481719970703125, + "objective/train/value_loss": 7.021182682365179e-05, + "objective/train/value_max": -0.00026726722717285156, + "objective/train/value_min": -0.06475830078125, + "objective/train/value_reward_corr": 0.40886419143151714, + "objective/train/value_std": 0.006183624267578125, + "objective/train/weight_avg": 1.0003459453582764, + "objective/train/weighted_lm_loss": 2.7234318256378174, + "objective/train/weights_max": 1.7021996974945068, + "objective/train/weights_min": 0.34541529417037964, + "theoretical_loss": 3.721154115169637, + "tokens_seen": 817561600 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037996990972918756, + "loss": 1.3513, + "theoretical_loss": 3.721154115169637, + "tokens_seen": 817561600 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037995987963891675, + "loss": 1.4479, + "theoretical_loss": 3.72112466321218, + "tokens_seen": 817627136 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003799498495486459, + "loss": 1.4368, + "theoretical_loss": 3.72109521427625, + "tokens_seen": 817692672 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037993981945837516, + "loss": 1.4523, + "theoretical_loss": 3.721065768361295, + "tokens_seen": 817758208 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003799297893681043, + "loss": 1.4149, + "theoretical_loss": 3.721036325466763, + "tokens_seen": 817823744 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003799197592778335, + "loss": 1.4351, + "theoretical_loss": 3.7210068855921024, + "tokens_seen": 817889280 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037990972918756265, + "loss": 1.4368, + "theoretical_loss": 3.720977448736761, + "tokens_seen": 817954816 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003798996990972919, + "loss": 1.4788, + "theoretical_loss": 3.7209480149001877, + "tokens_seen": 818020352 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037988966900702107, + "loss": 1.3397, + "theoretical_loss": 3.720918584081831, + "tokens_seen": 818085888 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037987963891675025, + "loss": 1.4562, + "theoretical_loss": 3.72088915628114, + "tokens_seen": 818151424 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003798696088264795, + "loss": 1.5226, + "theoretical_loss": 3.720859731497563, + "tokens_seen": 818216960 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003798595787362086, + "loss": 1.3814, + "theoretical_loss": 3.720830309730549, + "tokens_seen": 818282496 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037984954864593785, + "loss": 1.4356, + "theoretical_loss": 3.7208008909795476, + "tokens_seen": 818348032 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037983951855566703, + "loss": 1.4653, + "theoretical_loss": 3.7207714752440078, + "tokens_seen": 818413568 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003798294884653962, + "loss": 1.3672, + "theoretical_loss": 3.72074206252338, + "tokens_seen": 818479104 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003798194583751254, + "loss": 1.536, + "theoretical_loss": 3.720712652817112, + "tokens_seen": 818544640 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037980942828485457, + "loss": 1.4763, + "theoretical_loss": 3.720683246124655, + "tokens_seen": 818610176 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037979939819458375, + "loss": 1.4331, + "theoretical_loss": 3.7206538424454587, + "tokens_seen": 818675712 + }, + { + "epoch": 0.25, + "learning_rate": 0.000379789368104313, + "loss": 1.4392, + "theoretical_loss": 3.7206244417789733, + "tokens_seen": 818741248 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003797793380140421, + "loss": 1.3288, + "theoretical_loss": 3.720595044124649, + "tokens_seen": 818806784 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037976930792377135, + "loss": 1.4501, + "theoretical_loss": 3.7205656494819355, + "tokens_seen": 818872320 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037975927783350053, + "loss": 1.3226, + "theoretical_loss": 3.7205362578502834, + "tokens_seen": 818937856 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003797492477432297, + "loss": 1.3631, + "theoretical_loss": 3.720506869229144, + "tokens_seen": 819003392 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003797392176529589, + "loss": 1.4127, + "theoretical_loss": 3.7204774836179673, + "tokens_seen": 819068928 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003797291875626881, + "loss": 1.4747, + "theoretical_loss": 3.7204481010162045, + "tokens_seen": 819134464 + }, + { + "debugging/Self-BLEU-5": 0.41481118177622384, + "debugging/distinct-1-grams": 0.740269797661071, + "debugging/distinct-2-grams": 0.9420104450835436, + "debugging/entropy-1-grams": 5.634551829652006, + "debugging/entropy-2-grams": 6.391772523795091, + "debugging/length": 516.875, + "debugging/num_segments": 8, + "debugging/raw_token_scores_avg": 0.00791973527520895, + "debugging/raw_token_scores_std": 0.012947010807693005, + "debugging/score": 0.006249142059687382, + "debugging/score_std": 0.003488928891414518, + "epoch": 0.25, + "objective/train/advantage_avg": -0.0025930465199053288, + "objective/train/docs_used": 471797, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3540198802948, + "objective/train/original_loss": 2.3540198802948, + "objective/train/theoretical_loss": 3.7204187214233073, + "objective/train/tokens_used": 839660000, + "objective/train/value_avg": -0.0053253173828125, + "objective/train/value_loss": 0.0001318306167377159, + "objective/train/value_max": -0.0003077983856201172, + "objective/train/value_min": -0.056243896484375, + "objective/train/value_reward_corr": 0.5418826309971637, + "objective/train/value_std": 0.00443267822265625, + "objective/train/weight_avg": 0.979795515537262, + "objective/train/weighted_lm_loss": 2.3398773670196533, + "objective/train/weights_max": 1.6743781566619873, + "objective/train/weights_min": 0.35319676995277405, + "theoretical_loss": 3.7204187214233073, + "tokens_seen": 819200000 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037971915747241725, + "loss": 1.4639, + "theoretical_loss": 3.7204187214233073, + "tokens_seen": 819200000 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003797091273821465, + "loss": 1.4389, + "theoretical_loss": 3.7203893448387264, + "tokens_seen": 819265536 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003796990972918756, + "loss": 1.376, + "theoretical_loss": 3.720359971261913, + "tokens_seen": 819331072 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037968906720160485, + "loss": 1.5584, + "theoretical_loss": 3.7203306006923187, + "tokens_seen": 819396608 + }, + { + "epoch": 0.25, + "learning_rate": 0.000379679037111334, + "loss": 1.2792, + "theoretical_loss": 3.720301233129395, + "tokens_seen": 819462144 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003796690070210632, + "loss": 1.4437, + "theoretical_loss": 3.720271868572594, + "tokens_seen": 819527680 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003796589769307924, + "loss": 1.2072, + "theoretical_loss": 3.720242507021368, + "tokens_seen": 819593216 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003796489468405216, + "loss": 1.3144, + "theoretical_loss": 3.720213148475168, + "tokens_seen": 819658752 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037963891675025076, + "loss": 1.3066, + "theoretical_loss": 3.7201837929334465, + "tokens_seen": 819724288 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037962888665997994, + "loss": 1.3498, + "theoretical_loss": 3.720154440395657, + "tokens_seen": 819789824 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003796188565697091, + "loss": 1.4245, + "theoretical_loss": 3.7201250908612504, + "tokens_seen": 819855360 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037960882647943836, + "loss": 1.3208, + "theoretical_loss": 3.7200957443296803, + "tokens_seen": 819920896 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003795987963891675, + "loss": 1.4102, + "theoretical_loss": 3.7200664008004, + "tokens_seen": 819986432 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003795887662988967, + "loss": 1.3868, + "theoretical_loss": 3.720037060272861, + "tokens_seen": 820051968 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003795787362086259, + "loss": 1.4523, + "theoretical_loss": 3.720007722746517, + "tokens_seen": 820117504 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003795687061183551, + "loss": 1.1887, + "theoretical_loss": 3.719978388220822, + "tokens_seen": 820183040 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037955867602808426, + "loss": 1.336, + "theoretical_loss": 3.7199490566952287, + "tokens_seen": 820248576 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037954864593781344, + "loss": 1.4817, + "theoretical_loss": 3.71991972816919, + "tokens_seen": 820314112 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003795386158475426, + "loss": 1.348, + "theoretical_loss": 3.7198904026421604, + "tokens_seen": 820379648 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037952858575727186, + "loss": 1.3494, + "theoretical_loss": 3.719861080113594, + "tokens_seen": 820445184 + }, + { + "epoch": 0.25, + "learning_rate": 0.000379518555667001, + "loss": 1.4555, + "theoretical_loss": 3.7198317605829434, + "tokens_seen": 820510720 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003795085255767302, + "loss": 1.3303, + "theoretical_loss": 3.7198024440496638, + "tokens_seen": 820576256 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037949849548645935, + "loss": 1.4702, + "theoretical_loss": 3.7197731305132087, + "tokens_seen": 820641792 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003794884653961886, + "loss": 1.3395, + "theoretical_loss": 3.7197438199730337, + "tokens_seen": 820707328 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037947843530591776, + "loss": 1.4361, + "theoretical_loss": 3.7197145124285917, + "tokens_seen": 820772864 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0007303787861019373, + "objective/train/docs_used": 473281, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.873579978942871, + "objective/train/original_loss": 2.873579978942871, + "objective/train/theoretical_loss": 3.7196852078793388, + "objective/train/tokens_used": 841298400, + "objective/train/value_avg": -0.004024505615234375, + "objective/train/value_loss": 3.698689033626579e-05, + "objective/train/value_max": -0.00021827220916748047, + "objective/train/value_min": -0.041839599609375, + "objective/train/value_reward_corr": 0.35281984044649906, + "objective/train/value_std": 0.003040313720703125, + "objective/train/weight_avg": 0.9944557547569275, + "objective/train/weighted_lm_loss": 2.862072229385376, + "objective/train/weights_max": 1.402180552482605, + "objective/train/weights_min": 0.372824102640152, + "theoretical_loss": 3.7196852078793388, + "tokens_seen": 820838400 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037946840521564695, + "loss": 1.3908, + "theoretical_loss": 3.7196852078793388, + "tokens_seen": 820838400 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003794583751253761, + "loss": 1.4183, + "theoretical_loss": 3.7196559063247285, + "tokens_seen": 820903936 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037944834503510536, + "loss": 1.507, + "theoretical_loss": 3.719626607764217, + "tokens_seen": 820969472 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003794383149448345, + "loss": 1.3204, + "theoretical_loss": 3.7195973121972585, + "tokens_seen": 821035008 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003794282848545637, + "loss": 1.4449, + "theoretical_loss": 3.719568019623308, + "tokens_seen": 821100544 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037941825476429285, + "loss": 1.3483, + "theoretical_loss": 3.7195387300418217, + "tokens_seen": 821166080 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003794082246740221, + "loss": 1.3918, + "theoretical_loss": 3.719509443452255, + "tokens_seen": 821231616 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037939819458375127, + "loss": 1.3198, + "theoretical_loss": 3.719480159854063, + "tokens_seen": 821297152 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037938816449348045, + "loss": 1.4183, + "theoretical_loss": 3.7194508792467023, + "tokens_seen": 821362688 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037937813440320963, + "loss": 1.4858, + "theoretical_loss": 3.719421601629628, + "tokens_seen": 821428224 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003793681043129388, + "loss": 1.3761, + "theoretical_loss": 3.7193923270022964, + "tokens_seen": 821493760 + }, + { + "epoch": 0.25, + "learning_rate": 0.000379358074222668, + "loss": 1.3527, + "theoretical_loss": 3.7193630553641643, + "tokens_seen": 821559296 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037934804413239723, + "loss": 1.3955, + "theoretical_loss": 3.719333786714687, + "tokens_seen": 821624832 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037933801404212635, + "loss": 1.5055, + "theoretical_loss": 3.7193045210533215, + "tokens_seen": 821690368 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003793279839518556, + "loss": 1.3174, + "theoretical_loss": 3.7192752583795254, + "tokens_seen": 821755904 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003793179538615847, + "loss": 1.2474, + "theoretical_loss": 3.719245998692754, + "tokens_seen": 821821440 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037930792377131395, + "loss": 1.3125, + "theoretical_loss": 3.719216741992465, + "tokens_seen": 821886976 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037929789368104313, + "loss": 1.4981, + "theoretical_loss": 3.7191874882781155, + "tokens_seen": 821952512 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003792878635907723, + "loss": 1.5013, + "theoretical_loss": 3.7191582375491623, + "tokens_seen": 822018048 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003792778335005015, + "loss": 1.3726, + "theoretical_loss": 3.7191289898050632, + "tokens_seen": 822083584 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037926780341023073, + "loss": 1.472, + "theoretical_loss": 3.719099745045275, + "tokens_seen": 822149120 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037925777331995986, + "loss": 1.3185, + "theoretical_loss": 3.719070503269257, + "tokens_seen": 822214656 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003792477432296891, + "loss": 1.4312, + "theoretical_loss": 3.719041264476465, + "tokens_seen": 822280192 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003792377131394182, + "loss": 1.3635, + "theoretical_loss": 3.719012028666358, + "tokens_seen": 822345728 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037922768304914746, + "loss": 1.4745, + "theoretical_loss": 3.718982795838394, + "tokens_seen": 822411264 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": 0.00038628288893960416, + "objective/train/docs_used": 473825, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.831556797027588, + "objective/train/original_loss": 2.831557035446167, + "objective/train/theoretical_loss": 3.718953565992031, + "objective/train/tokens_used": 842936800, + "objective/train/value_avg": -0.0043487548828125, + "objective/train/value_loss": 3.695869600051083e-05, + "objective/train/value_max": -0.00034737586975097656, + "objective/train/value_min": -0.06671142578125, + "objective/train/value_reward_corr": 0.32753498751044513, + "objective/train/value_std": 0.00323486328125, + "objective/train/weight_avg": 1.0056116580963135, + "objective/train/weighted_lm_loss": 2.855339527130127, + "objective/train/weights_max": 1.8943308591842651, + "objective/train/weights_min": 0.3780442774295807, + "theoretical_loss": 3.718953565992031, + "tokens_seen": 822476800 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037921765295887664, + "loss": 1.4744, + "theoretical_loss": 3.718953565992031, + "tokens_seen": 822476800 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003792076228686058, + "loss": 1.4338, + "theoretical_loss": 3.718924339126727, + "tokens_seen": 822542336 + }, + { + "epoch": 0.25, + "learning_rate": 0.000379197592778335, + "loss": 1.4194, + "theoretical_loss": 3.718895115241941, + "tokens_seen": 822607872 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003791875626880642, + "loss": 1.3715, + "theoretical_loss": 3.7188658943371316, + "tokens_seen": 822673408 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037917753259779336, + "loss": 1.4658, + "theoretical_loss": 3.7188366764117577, + "tokens_seen": 822738944 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003791675025075226, + "loss": 1.4497, + "theoretical_loss": 3.718807461465278, + "tokens_seen": 822804480 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003791574724172517, + "loss": 1.5107, + "theoretical_loss": 3.7187782494971517, + "tokens_seen": 822870016 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037914744232698096, + "loss": 1.3711, + "theoretical_loss": 3.7187490405068377, + "tokens_seen": 822935552 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003791374122367101, + "loss": 1.4736, + "theoretical_loss": 3.718719834493795, + "tokens_seen": 823001088 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003791273821464393, + "loss": 1.4693, + "theoretical_loss": 3.718690631457484, + "tokens_seen": 823066624 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037911735205616856, + "loss": 1.4694, + "theoretical_loss": 3.7186614313973645, + "tokens_seen": 823132160 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003791073219658977, + "loss": 1.3546, + "theoretical_loss": 3.7186322343128952, + "tokens_seen": 823197696 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003790972918756269, + "loss": 1.4281, + "theoretical_loss": 3.7186030402035364, + "tokens_seen": 823263232 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003790872617853561, + "loss": 1.383, + "theoretical_loss": 3.7185738490687483, + "tokens_seen": 823328768 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003790772316950853, + "loss": 1.3859, + "theoretical_loss": 3.718544660907991, + "tokens_seen": 823394304 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037906720160481446, + "loss": 1.3238, + "theoretical_loss": 3.718515475720725, + "tokens_seen": 823459840 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037905717151454364, + "loss": 1.5113, + "theoretical_loss": 3.7184862935064102, + "tokens_seen": 823525376 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003790471414242728, + "loss": 1.3408, + "theoretical_loss": 3.718457114264508, + "tokens_seen": 823590912 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037903711133400206, + "loss": 1.3553, + "theoretical_loss": 3.7184279379944787, + "tokens_seen": 823656448 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003790270812437312, + "loss": 1.4626, + "theoretical_loss": 3.7183987646957832, + "tokens_seen": 823721984 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003790170511534604, + "loss": 1.2561, + "theoretical_loss": 3.718369594367883, + "tokens_seen": 823787520 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037900702106318955, + "loss": 1.4434, + "theoretical_loss": 3.7183404270102383, + "tokens_seen": 823853056 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003789969909729188, + "loss": 1.355, + "theoretical_loss": 3.7183112626223114, + "tokens_seen": 823918592 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037898696088264796, + "loss": 1.3197, + "theoretical_loss": 3.718282101203563, + "tokens_seen": 823984128 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037897693079237715, + "loss": 1.3861, + "theoretical_loss": 3.718252942753455, + "tokens_seen": 824049664 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0013182146940380335, + "objective/train/docs_used": 475209, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.741726875305176, + "objective/train/original_loss": 2.7417266368865967, + "objective/train/theoretical_loss": 3.718223787271449, + "objective/train/tokens_used": 844575200, + "objective/train/value_avg": -0.005939483642578125, + "objective/train/value_loss": 9.874683746602386e-05, + "objective/train/value_max": -0.00023233890533447266, + "objective/train/value_min": -0.07574462890625, + "objective/train/value_reward_corr": 0.5117690502242687, + "objective/train/value_std": 0.00568389892578125, + "objective/train/weight_avg": 0.9913066625595093, + "objective/train/weighted_lm_loss": 2.7196273803710938, + "objective/train/weights_max": 2.1328225135803223, + "objective/train/weights_min": 0.23672960698604584, + "theoretical_loss": 3.718223787271449, + "tokens_seen": 824115200 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003789669007021063, + "loss": 1.3963, + "theoretical_loss": 3.718223787271449, + "tokens_seen": 824115200 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037895687061183556, + "loss": 1.5081, + "theoretical_loss": 3.7181946347570074, + "tokens_seen": 824180736 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003789468405215647, + "loss": 1.3988, + "theoretical_loss": 3.7181654852095924, + "tokens_seen": 824246272 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003789368104312939, + "loss": 1.3353, + "theoretical_loss": 3.718136338628665, + "tokens_seen": 824311808 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037892678034102305, + "loss": 1.3984, + "theoretical_loss": 3.7181071950136877, + "tokens_seen": 824377344 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003789167502507523, + "loss": 1.347, + "theoretical_loss": 3.718078054364124, + "tokens_seen": 824442880 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037890672016048147, + "loss": 1.2772, + "theoretical_loss": 3.7180489166794355, + "tokens_seen": 824508416 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037889669007021065, + "loss": 1.3546, + "theoretical_loss": 3.718019781959085, + "tokens_seen": 824573952 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037888665997993983, + "loss": 1.3849, + "theoretical_loss": 3.717990650202536, + "tokens_seen": 824639488 + }, + { + "epoch": 0.25, + "learning_rate": 0.000378876629889669, + "loss": 1.2754, + "theoretical_loss": 3.717961521409251, + "tokens_seen": 824705024 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003788665997993982, + "loss": 1.3348, + "theoretical_loss": 3.7179323955786936, + "tokens_seen": 824770560 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037885656970912743, + "loss": 1.4429, + "theoretical_loss": 3.717903272710326, + "tokens_seen": 824836096 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037884653961885655, + "loss": 1.468, + "theoretical_loss": 3.7178741528036126, + "tokens_seen": 824901632 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003788365095285858, + "loss": 1.4636, + "theoretical_loss": 3.7178450358580166, + "tokens_seen": 824967168 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003788264794383149, + "loss": 1.5335, + "theoretical_loss": 3.717815921873002, + "tokens_seen": 825032704 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037881644934804415, + "loss": 1.4191, + "theoretical_loss": 3.7177868108480316, + "tokens_seen": 825098240 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037880641925777333, + "loss": 1.3425, + "theoretical_loss": 3.717757702782571, + "tokens_seen": 825163776 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003787963891675025, + "loss": 1.3553, + "theoretical_loss": 3.7177285976760834, + "tokens_seen": 825229312 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003787863590772317, + "loss": 1.3934, + "theoretical_loss": 3.7176994955280334, + "tokens_seen": 825294848 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037877632898696093, + "loss": 1.3729, + "theoretical_loss": 3.7176703963378843, + "tokens_seen": 825360384 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037876629889669006, + "loss": 1.4353, + "theoretical_loss": 3.717641300105102, + "tokens_seen": 825425920 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003787562688064193, + "loss": 1.366, + "theoretical_loss": 3.71761220682915, + "tokens_seen": 825491456 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003787462387161484, + "loss": 1.3557, + "theoretical_loss": 3.717583116509495, + "tokens_seen": 825556992 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037873620862587766, + "loss": 1.3275, + "theoretical_loss": 3.7175540291455995, + "tokens_seen": 825622528 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037872617853560684, + "loss": 1.4585, + "theoretical_loss": 3.7175249447369305, + "tokens_seen": 825688064 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0007065586978569627, + "objective/train/docs_used": 475848, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.778744697570801, + "objective/train/original_loss": 2.778744697570801, + "objective/train/theoretical_loss": 3.7174958632829522, + "objective/train/tokens_used": 846213600, + "objective/train/value_avg": -0.0046539306640625, + "objective/train/value_loss": 4.3141251808265224e-05, + "objective/train/value_max": -0.0003407001495361328, + "objective/train/value_min": -0.08758544921875, + "objective/train/value_reward_corr": 0.3660680723237594, + "objective/train/value_std": 0.00399017333984375, + "objective/train/weight_avg": 0.9950064420700073, + "objective/train/weighted_lm_loss": 2.770004987716675, + "objective/train/weights_max": 1.710646629333496, + "objective/train/weights_min": 0.5036153793334961, + "theoretical_loss": 3.7174958632829522, + "tokens_seen": 825753600 + }, + { + "epoch": 0.25, + "learning_rate": 0.000378716148445336, + "loss": 1.4878, + "theoretical_loss": 3.7174958632829522, + "tokens_seen": 825753600 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003787061183550652, + "loss": 1.313, + "theoretical_loss": 3.71746678478313, + "tokens_seen": 825819136 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003786960882647944, + "loss": 1.438, + "theoretical_loss": 3.71743770923693, + "tokens_seen": 825884672 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037868605817452356, + "loss": 1.3094, + "theoretical_loss": 3.7174086366438175, + "tokens_seen": 825950208 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003786760280842528, + "loss": 1.4208, + "theoretical_loss": 3.7173795670032588, + "tokens_seen": 826015744 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003786659979939819, + "loss": 1.4907, + "theoretical_loss": 3.717350500314719, + "tokens_seen": 826081280 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037865596790371116, + "loss": 1.3963, + "theoretical_loss": 3.7173214365776643, + "tokens_seen": 826146816 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003786459378134403, + "loss": 1.3761, + "theoretical_loss": 3.7172923757915615, + "tokens_seen": 826212352 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003786359077231695, + "loss": 1.3956, + "theoretical_loss": 3.7172633179558763, + "tokens_seen": 826277888 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003786258776328987, + "loss": 1.4579, + "theoretical_loss": 3.7172342630700754, + "tokens_seen": 826343424 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003786158475426279, + "loss": 1.3309, + "theoretical_loss": 3.7172052111336256, + "tokens_seen": 826408960 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037860581745235706, + "loss": 1.3341, + "theoretical_loss": 3.7171761621459933, + "tokens_seen": 826474496 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003785957873620863, + "loss": 1.2592, + "theoretical_loss": 3.717147116106646, + "tokens_seen": 826540032 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003785857572718154, + "loss": 1.4212, + "theoretical_loss": 3.71711807301505, + "tokens_seen": 826605568 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037857572718154466, + "loss": 1.3976, + "theoretical_loss": 3.717089032870673, + "tokens_seen": 826671104 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003785656970912738, + "loss": 1.5314, + "theoretical_loss": 3.717059995672982, + "tokens_seen": 826736640 + }, + { + "epoch": 0.25, + "learning_rate": 0.000378555667001003, + "loss": 1.5094, + "theoretical_loss": 3.717030961421445, + "tokens_seen": 826802176 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003785456369107322, + "loss": 1.4934, + "theoretical_loss": 3.717001930115529, + "tokens_seen": 826867712 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003785356068204614, + "loss": 1.5702, + "theoretical_loss": 3.7169729017547013, + "tokens_seen": 826933248 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037852557673019057, + "loss": 1.4403, + "theoretical_loss": 3.716943876338431, + "tokens_seen": 826998784 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037851554663991975, + "loss": 1.4882, + "theoretical_loss": 3.716914853866185, + "tokens_seen": 827064320 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037850551654964893, + "loss": 1.4791, + "theoretical_loss": 3.7168858343374325, + "tokens_seen": 827129856 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037849548645937816, + "loss": 1.3459, + "theoretical_loss": 3.7168568177516415, + "tokens_seen": 827195392 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003784854563691073, + "loss": 1.5075, + "theoretical_loss": 3.716827804108279, + "tokens_seen": 827260928 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003784754262788365, + "loss": 1.3947, + "theoretical_loss": 3.7167987934068156, + "tokens_seen": 827326464 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0012872805818915367, + "objective/train/docs_used": 477214, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8008198738098145, + "objective/train/original_loss": 2.8008193969726562, + "objective/train/theoretical_loss": 3.7167697856467186, + "objective/train/tokens_used": 847852000, + "objective/train/value_avg": -0.005252838134765625, + "objective/train/value_loss": 9.685193072073162e-05, + "objective/train/value_max": -0.00023412704467773438, + "objective/train/value_min": -0.093017578125, + "objective/train/value_reward_corr": 0.5777813572326268, + "objective/train/value_std": 0.005641937255859375, + "objective/train/weight_avg": 0.9913302063941956, + "objective/train/weighted_lm_loss": 2.7689523696899414, + "objective/train/weights_max": 1.5959460735321045, + "objective/train/weights_min": 0.13705755770206451, + "theoretical_loss": 3.7167697856467186, + "tokens_seen": 827392000 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037846539618856565, + "loss": 1.2525, + "theoretical_loss": 3.7167697856467186, + "tokens_seen": 827392000 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003784553660982949, + "loss": 1.3557, + "theoretical_loss": 3.7167407808274575, + "tokens_seen": 827457536 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037844533600802407, + "loss": 1.3292, + "theoretical_loss": 3.716711778948501, + "tokens_seen": 827523072 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037843530591775325, + "loss": 1.4321, + "theoretical_loss": 3.716682780009318, + "tokens_seen": 827588608 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037842527582748243, + "loss": 1.2887, + "theoretical_loss": 3.7166537840093783, + "tokens_seen": 827654144 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037841524573721167, + "loss": 1.4312, + "theoretical_loss": 3.7166247909481513, + "tokens_seen": 827719680 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003784052156469408, + "loss": 1.487, + "theoretical_loss": 3.716595800825106, + "tokens_seen": 827785216 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037839518555667003, + "loss": 1.4253, + "theoretical_loss": 3.7165668136397123, + "tokens_seen": 827850752 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037838515546639916, + "loss": 1.3943, + "theoretical_loss": 3.7165378293914397, + "tokens_seen": 827916288 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003783751253761284, + "loss": 1.3332, + "theoretical_loss": 3.7165088480797595, + "tokens_seen": 827981824 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037836509528585763, + "loss": 1.3353, + "theoretical_loss": 3.7164798697041395, + "tokens_seen": 828047360 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037835506519558675, + "loss": 1.3801, + "theoretical_loss": 3.716450894264052, + "tokens_seen": 828112896 + }, + { + "epoch": 0.25, + "learning_rate": 0.000378345035105316, + "loss": 1.4112, + "theoretical_loss": 3.7164219217589656, + "tokens_seen": 828178432 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003783350050150451, + "loss": 1.4101, + "theoretical_loss": 3.716392952188352, + "tokens_seen": 828243968 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037832497492477435, + "loss": 1.3919, + "theoretical_loss": 3.716363985551682, + "tokens_seen": 828309504 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037831494483450353, + "loss": 1.3031, + "theoretical_loss": 3.7163350218484252, + "tokens_seen": 828375040 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003783049147442327, + "loss": 1.2982, + "theoretical_loss": 3.716306061078054, + "tokens_seen": 828440576 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003782948846539619, + "loss": 1.4931, + "theoretical_loss": 3.7162771032400377, + "tokens_seen": 828506112 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037828485456369113, + "loss": 1.4178, + "theoretical_loss": 3.7162481483338485, + "tokens_seen": 828571648 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037827482447342026, + "loss": 1.3356, + "theoretical_loss": 3.716219196358958, + "tokens_seen": 828637184 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003782647943831495, + "loss": 1.4646, + "theoretical_loss": 3.716190247314837, + "tokens_seen": 828702720 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003782547642928786, + "loss": 1.3642, + "theoretical_loss": 3.716161301200957, + "tokens_seen": 828768256 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037824473420260786, + "loss": 1.4618, + "theoretical_loss": 3.7161323580167904, + "tokens_seen": 828833792 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037823470411233704, + "loss": 1.3811, + "theoretical_loss": 3.7161034177618086, + "tokens_seen": 828899328 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003782246740220662, + "loss": 1.3935, + "theoretical_loss": 3.7160744804354833, + "tokens_seen": 828964864 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -6.951454270165414e-05, + "objective/train/docs_used": 478014, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.93367338180542, + "objective/train/original_loss": 2.93367338180542, + "objective/train/theoretical_loss": 3.716045546037287, + "objective/train/tokens_used": 849490400, + "objective/train/value_avg": -0.004947662353515625, + "objective/train/value_loss": 4.213188367430121e-05, + "objective/train/value_max": -0.0003712177276611328, + "objective/train/value_min": -0.06903076171875, + "objective/train/value_reward_corr": 0.3853669333521878, + "objective/train/value_std": 0.0039215087890625, + "objective/train/weight_avg": 1.0012308359146118, + "objective/train/weighted_lm_loss": 2.9399263858795166, + "objective/train/weights_max": 1.6730226278305054, + "objective/train/weights_min": 0.19450204074382782, + "theoretical_loss": 3.716045546037287, + "tokens_seen": 829030400 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003782146439317954, + "loss": 1.4261, + "theoretical_loss": 3.716045546037287, + "tokens_seen": 829030400 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003782046138415246, + "loss": 1.3026, + "theoretical_loss": 3.7160166145666924, + "tokens_seen": 829095936 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037819458375125376, + "loss": 1.4483, + "theoretical_loss": 3.7159876860231713, + "tokens_seen": 829161472 + }, + { + "epoch": 0.25, + "learning_rate": 0.000378184553660983, + "loss": 1.4047, + "theoretical_loss": 3.7159587604061963, + "tokens_seen": 829227008 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003781745235707121, + "loss": 1.4931, + "theoretical_loss": 3.7159298377152403, + "tokens_seen": 829292544 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037816449348044136, + "loss": 1.3402, + "theoretical_loss": 3.715900917949776, + "tokens_seen": 829358080 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003781544633901705, + "loss": 1.4052, + "theoretical_loss": 3.7158720011092767, + "tokens_seen": 829423616 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003781444332998997, + "loss": 1.4492, + "theoretical_loss": 3.715843087193215, + "tokens_seen": 829489152 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003781344032096289, + "loss": 1.3608, + "theoretical_loss": 3.715814176201064, + "tokens_seen": 829554688 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003781243731193581, + "loss": 1.4216, + "theoretical_loss": 3.715785268132297, + "tokens_seen": 829620224 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037811434302908726, + "loss": 1.437, + "theoretical_loss": 3.715756362986389, + "tokens_seen": 829685760 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003781043129388165, + "loss": 1.3961, + "theoretical_loss": 3.7157274607628117, + "tokens_seen": 829751296 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003780942828485456, + "loss": 1.3614, + "theoretical_loss": 3.71569856146104, + "tokens_seen": 829816832 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037808425275827486, + "loss": 1.3216, + "theoretical_loss": 3.7156696650805467, + "tokens_seen": 829882368 + }, + { + "epoch": 0.25, + "learning_rate": 0.000378074222668004, + "loss": 1.4136, + "theoretical_loss": 3.7156407716208073, + "tokens_seen": 829947904 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003780641925777332, + "loss": 1.4064, + "theoretical_loss": 3.715611881081295, + "tokens_seen": 830013440 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003780541624874624, + "loss": 1.382, + "theoretical_loss": 3.7155829934614846, + "tokens_seen": 830078976 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003780441323971916, + "loss": 1.3786, + "theoretical_loss": 3.71555410876085, + "tokens_seen": 830144512 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037803410230692077, + "loss": 1.4619, + "theoretical_loss": 3.7155252269788663, + "tokens_seen": 830210048 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037802407221664995, + "loss": 1.4231, + "theoretical_loss": 3.7154963481150087, + "tokens_seen": 830275584 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037801404212637913, + "loss": 1.3267, + "theoretical_loss": 3.7154674721687506, + "tokens_seen": 830341120 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037800401203610836, + "loss": 1.3947, + "theoretical_loss": 3.715438599139568, + "tokens_seen": 830406656 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003779939819458375, + "loss": 1.3542, + "theoretical_loss": 3.715409729026936, + "tokens_seen": 830472192 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037798395185556673, + "loss": 1.3861, + "theoretical_loss": 3.7153808618303295, + "tokens_seen": 830537728 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037797392176529585, + "loss": 1.4696, + "theoretical_loss": 3.7153519975492246, + "tokens_seen": 830603264 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0006605027010664344, + "objective/train/docs_used": 479315, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.673999547958374, + "objective/train/original_loss": 2.673999786376953, + "objective/train/theoretical_loss": 3.7153231361830956, + "objective/train/tokens_used": 851128800, + "objective/train/value_avg": -0.00475311279296875, + "objective/train/value_loss": 6.275265332078561e-05, + "objective/train/value_max": -0.00042057037353515625, + "objective/train/value_min": -0.0374755859375, + "objective/train/value_reward_corr": 0.21529593476231976, + "objective/train/value_std": 0.003398895263671875, + "objective/train/weight_avg": 0.9963363409042358, + "objective/train/weighted_lm_loss": 2.6728951930999756, + "objective/train/weights_max": 1.4546362161636353, + "objective/train/weights_min": 0.4104567766189575, + "theoretical_loss": 3.7153231361830956, + "tokens_seen": 830668800 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003779638916750251, + "loss": 1.4207, + "theoretical_loss": 3.7153231361830956, + "tokens_seen": 830668800 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037795386158475427, + "loss": 1.3863, + "theoretical_loss": 3.7152942777314197, + "tokens_seen": 830734336 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037794383149448345, + "loss": 1.2936, + "theoretical_loss": 3.7152654221936716, + "tokens_seen": 830799872 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037793380140421263, + "loss": 1.3191, + "theoretical_loss": 3.7152365695693277, + "tokens_seen": 830865408 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037792377131394187, + "loss": 1.391, + "theoretical_loss": 3.715207719857864, + "tokens_seen": 830930944 + }, + { + "epoch": 0.25, + "learning_rate": 0.000377913741223671, + "loss": 1.2804, + "theoretical_loss": 3.7151788730587567, + "tokens_seen": 830996480 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037790371113340023, + "loss": 1.4522, + "theoretical_loss": 3.715150029171482, + "tokens_seen": 831062016 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037789368104312936, + "loss": 1.4693, + "theoretical_loss": 3.715121188195517, + "tokens_seen": 831127552 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003778836509528586, + "loss": 1.3316, + "theoretical_loss": 3.7150923501303375, + "tokens_seen": 831193088 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003778736208625878, + "loss": 1.4275, + "theoretical_loss": 3.7150635149754203, + "tokens_seen": 831258624 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037786359077231695, + "loss": 1.4449, + "theoretical_loss": 3.7150346827302427, + "tokens_seen": 831324160 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037785356068204614, + "loss": 1.497, + "theoretical_loss": 3.715005853394282, + "tokens_seen": 831389696 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003778435305917753, + "loss": 1.3378, + "theoretical_loss": 3.7149770269670146, + "tokens_seen": 831455232 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003778335005015045, + "loss": 1.4749, + "theoretical_loss": 3.714948203447919, + "tokens_seen": 831520768 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037782347041123373, + "loss": 1.3014, + "theoretical_loss": 3.714919382836471, + "tokens_seen": 831586304 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037781344032096286, + "loss": 1.3962, + "theoretical_loss": 3.7148905651321495, + "tokens_seen": 831651840 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003778034102306921, + "loss": 1.514, + "theoretical_loss": 3.714861750334432, + "tokens_seen": 831717376 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003777933801404213, + "loss": 1.4937, + "theoretical_loss": 3.714832938442796, + "tokens_seen": 831782912 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037778335005015046, + "loss": 1.3469, + "theoretical_loss": 3.7148041294567187, + "tokens_seen": 831848448 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037777331995987964, + "loss": 1.4249, + "theoretical_loss": 3.7147753233756795, + "tokens_seen": 831913984 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003777632898696088, + "loss": 1.2992, + "theoretical_loss": 3.7147465201991565, + "tokens_seen": 831979520 + }, + { + "epoch": 0.25, + "learning_rate": 0.000377753259779338, + "loss": 1.4504, + "theoretical_loss": 3.7147177199266275, + "tokens_seen": 832045056 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037774322968906724, + "loss": 1.5245, + "theoretical_loss": 3.7146889225575714, + "tokens_seen": 832110592 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037773319959879636, + "loss": 1.3439, + "theoretical_loss": 3.7146601280914666, + "tokens_seen": 832176128 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003777231695085256, + "loss": 1.3142, + "theoretical_loss": 3.714631336527792, + "tokens_seen": 832241664 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0006110891699790955, + "objective/train/docs_used": 479893, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6715619564056396, + "objective/train/original_loss": 2.6715619564056396, + "objective/train/theoretical_loss": 3.7146025478660274, + "objective/train/tokens_used": 852767200, + "objective/train/value_avg": -0.005474090576171875, + "objective/train/value_loss": 6.431693327613175e-05, + "objective/train/value_max": -0.00033020973205566406, + "objective/train/value_min": -0.054595947265625, + "objective/train/value_reward_corr": 0.43071043220718247, + "objective/train/value_std": 0.004913330078125, + "objective/train/weight_avg": 0.9966827630996704, + "objective/train/weighted_lm_loss": 2.65999436378479, + "objective/train/weights_max": 1.5588037967681885, + "objective/train/weights_min": 0.08331215381622314, + "theoretical_loss": 3.7146025478660274, + "tokens_seen": 832307200 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003777131394182547, + "loss": 1.2806, + "theoretical_loss": 3.7146025478660274, + "tokens_seen": 832307200 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037770310932798396, + "loss": 1.4236, + "theoretical_loss": 3.7145737621056503, + "tokens_seen": 832372736 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037769307923771314, + "loss": 1.4306, + "theoretical_loss": 3.7145449792461402, + "tokens_seen": 832438272 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003776830491474423, + "loss": 1.3862, + "theoretical_loss": 3.7145161992869773, + "tokens_seen": 832503808 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003776730190571715, + "loss": 1.309, + "theoretical_loss": 3.7144874222276405, + "tokens_seen": 832569344 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003776629889669007, + "loss": 1.4766, + "theoretical_loss": 3.71445864806761, + "tokens_seen": 832634880 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037765295887662987, + "loss": 1.4325, + "theoretical_loss": 3.714429876806364, + "tokens_seen": 832700416 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003776429287863591, + "loss": 1.3338, + "theoretical_loss": 3.7144011084433837, + "tokens_seen": 832765952 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037763289869608823, + "loss": 1.3517, + "theoretical_loss": 3.714372342978149, + "tokens_seen": 832831488 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037762286860581746, + "loss": 1.3852, + "theoretical_loss": 3.71434358041014, + "tokens_seen": 832897024 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003776128385155467, + "loss": 1.3595, + "theoretical_loss": 3.714314820738836, + "tokens_seen": 832962560 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003776028084252758, + "loss": 1.2537, + "theoretical_loss": 3.7142860639637187, + "tokens_seen": 833028096 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037759277833500506, + "loss": 1.2941, + "theoretical_loss": 3.7142573100842675, + "tokens_seen": 833093632 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003775827482447342, + "loss": 1.3245, + "theoretical_loss": 3.714228559099964, + "tokens_seen": 833159168 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003775727181544634, + "loss": 1.3305, + "theoretical_loss": 3.714199811010289, + "tokens_seen": 833224704 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003775626880641926, + "loss": 1.424, + "theoretical_loss": 3.7141710658147224, + "tokens_seen": 833290240 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003775526579739218, + "loss": 1.3612, + "theoretical_loss": 3.714142323512746, + "tokens_seen": 833355776 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037754262788365097, + "loss": 1.4112, + "theoretical_loss": 3.714113584103841, + "tokens_seen": 833421312 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037753259779338015, + "loss": 1.4615, + "theoretical_loss": 3.7140848475874892, + "tokens_seen": 833486848 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037752256770310933, + "loss": 1.5079, + "theoretical_loss": 3.714056113963171, + "tokens_seen": 833552384 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037751253761283857, + "loss": 1.3807, + "theoretical_loss": 3.714027383230369, + "tokens_seen": 833617920 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003775025075225677, + "loss": 1.3969, + "theoretical_loss": 3.713998655388564, + "tokens_seen": 833683456 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037749247743229693, + "loss": 1.4002, + "theoretical_loss": 3.7139699304372384, + "tokens_seen": 833748992 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037748244734202605, + "loss": 1.2969, + "theoretical_loss": 3.713941208375874, + "tokens_seen": 833814528 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003774724172517553, + "loss": 1.3642, + "theoretical_loss": 3.7139124892039534, + "tokens_seen": 833880064 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0008187235216610134, + "objective/train/docs_used": 480416, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.860668897628784, + "objective/train/original_loss": 2.860668897628784, + "objective/train/theoretical_loss": 3.7138837729209584, + "objective/train/tokens_used": 854405600, + "objective/train/value_avg": -0.00494384765625, + "objective/train/value_loss": 4.8649231757735834e-05, + "objective/train/value_max": -0.00039505958557128906, + "objective/train/value_min": -0.04876708984375, + "objective/train/value_reward_corr": 0.5353015546113755, + "objective/train/value_std": 0.0043792724609375, + "objective/train/weight_avg": 0.9940635561943054, + "objective/train/weighted_lm_loss": 2.8509552478790283, + "objective/train/weights_max": 1.4917525053024292, + "objective/train/weights_min": 0.3762243688106537, + "theoretical_loss": 3.7138837729209584, + "tokens_seen": 833945600 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037746238716148447, + "loss": 1.4641, + "theoretical_loss": 3.7138837729209584, + "tokens_seen": 833945600 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037745235707121365, + "loss": 1.2385, + "theoretical_loss": 3.7138550595263715, + "tokens_seen": 834011136 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037744232698094283, + "loss": 1.3234, + "theoretical_loss": 3.7138263490196755, + "tokens_seen": 834076672 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037743229689067207, + "loss": 1.3508, + "theoretical_loss": 3.713797641400353, + "tokens_seen": 834142208 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003774222668004012, + "loss": 1.3702, + "theoretical_loss": 3.7137689366678863, + "tokens_seen": 834207744 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037741223671013043, + "loss": 1.3729, + "theoretical_loss": 3.7137402348217585, + "tokens_seen": 834273280 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037740220661985956, + "loss": 1.3651, + "theoretical_loss": 3.713711535861453, + "tokens_seen": 834338816 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003773921765295888, + "loss": 1.4674, + "theoretical_loss": 3.713682839786453, + "tokens_seen": 834404352 + }, + { + "epoch": 0.25, + "learning_rate": 0.000377382146439318, + "loss": 1.288, + "theoretical_loss": 3.713654146596242, + "tokens_seen": 834469888 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037737211634904715, + "loss": 1.4074, + "theoretical_loss": 3.713625456290303, + "tokens_seen": 834535424 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037736208625877634, + "loss": 1.3675, + "theoretical_loss": 3.7135967688681193, + "tokens_seen": 834600960 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003773520561685055, + "loss": 1.3822, + "theoretical_loss": 3.713568084329175, + "tokens_seen": 834666496 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003773420260782347, + "loss": 1.2951, + "theoretical_loss": 3.7135394026729553, + "tokens_seen": 834732032 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037733199598796393, + "loss": 1.4325, + "theoretical_loss": 3.713510723898942, + "tokens_seen": 834797568 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037732196589769306, + "loss": 1.2414, + "theoretical_loss": 3.71348204800662, + "tokens_seen": 834863104 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003773119358074223, + "loss": 1.4379, + "theoretical_loss": 3.7134533749954746, + "tokens_seen": 834928640 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003773019057171515, + "loss": 1.3489, + "theoretical_loss": 3.713424704864989, + "tokens_seen": 834994176 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037729187562688066, + "loss": 1.3632, + "theoretical_loss": 3.7133960376146478, + "tokens_seen": 835059712 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037728184553660984, + "loss": 1.4108, + "theoretical_loss": 3.7133673732439365, + "tokens_seen": 835125248 + }, + { + "epoch": 0.25, + "learning_rate": 0.000377271815446339, + "loss": 1.4369, + "theoretical_loss": 3.713338711752339, + "tokens_seen": 835190784 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003772617853560682, + "loss": 1.491, + "theoretical_loss": 3.713310053139341, + "tokens_seen": 835256320 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037725175526579744, + "loss": 1.3902, + "theoretical_loss": 3.713281397404427, + "tokens_seen": 835321856 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037724172517552656, + "loss": 1.3962, + "theoretical_loss": 3.7132527445470815, + "tokens_seen": 835387392 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003772316950852558, + "loss": 1.3307, + "theoretical_loss": 3.713224094566792, + "tokens_seen": 835452928 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003772216649949849, + "loss": 1.3878, + "theoretical_loss": 3.7131954474630424, + "tokens_seen": 835518464 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": 0.0003179203486070037, + "objective/train/docs_used": 481156, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6695475578308105, + "objective/train/original_loss": 2.6695473194122314, + "objective/train/theoretical_loss": 3.713166803235318, + "objective/train/tokens_used": 856044000, + "objective/train/value_avg": -0.004863739013671875, + "objective/train/value_loss": 5.458140731207095e-05, + "objective/train/value_max": -0.00036263465881347656, + "objective/train/value_min": -0.07977294921875, + "objective/train/value_reward_corr": 0.3965151782970697, + "objective/train/value_std": 0.004344940185546875, + "objective/train/weight_avg": 1.0057209730148315, + "objective/train/weighted_lm_loss": 2.684976577758789, + "objective/train/weights_max": 1.9433166980743408, + "objective/train/weights_min": 0.29006510972976685, + "theoretical_loss": 3.713166803235318, + "tokens_seen": 835584000 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037721163490471416, + "loss": 1.2717, + "theoretical_loss": 3.713166803235318, + "tokens_seen": 835584000 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037720160481444334, + "loss": 1.4184, + "theoretical_loss": 3.713138161883105, + "tokens_seen": 835649536 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003771915747241725, + "loss": 1.3922, + "theoretical_loss": 3.7131095234058895, + "tokens_seen": 835715072 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003771815446339017, + "loss": 1.4363, + "theoretical_loss": 3.713080887803158, + "tokens_seen": 835780608 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003771715145436309, + "loss": 1.503, + "theoretical_loss": 3.7130522550743947, + "tokens_seen": 835846144 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037716148445336007, + "loss": 1.3069, + "theoretical_loss": 3.7130236252190882, + "tokens_seen": 835911680 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003771514543630893, + "loss": 1.4727, + "theoretical_loss": 3.7129949982367236, + "tokens_seen": 835977216 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037714142427281843, + "loss": 1.4304, + "theoretical_loss": 3.712966374126787, + "tokens_seen": 836042752 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037713139418254766, + "loss": 1.375, + "theoretical_loss": 3.7129377528887657, + "tokens_seen": 836108288 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037712136409227685, + "loss": 1.4266, + "theoretical_loss": 3.712909134522147, + "tokens_seen": 836173824 + }, + { + "epoch": 0.25, + "learning_rate": 0.000377111334002006, + "loss": 1.4595, + "theoretical_loss": 3.7128805190264167, + "tokens_seen": 836239360 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003771013039117352, + "loss": 1.4374, + "theoretical_loss": 3.7128519064010628, + "tokens_seen": 836304896 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003770912738214644, + "loss": 1.4373, + "theoretical_loss": 3.712823296645572, + "tokens_seen": 836370432 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037708124373119357, + "loss": 1.3798, + "theoretical_loss": 3.7127946897594315, + "tokens_seen": 836435968 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003770712136409228, + "loss": 1.3639, + "theoretical_loss": 3.7127660857421287, + "tokens_seen": 836501504 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037706118355065193, + "loss": 1.3872, + "theoretical_loss": 3.712737484593152, + "tokens_seen": 836567040 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037705115346038117, + "loss": 1.2697, + "theoretical_loss": 3.712708886311988, + "tokens_seen": 836632576 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003770411233701103, + "loss": 1.3083, + "theoretical_loss": 3.712680290898125, + "tokens_seen": 836698112 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037703109327983953, + "loss": 1.4036, + "theoretical_loss": 3.712651698351051, + "tokens_seen": 836763648 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003770210631895687, + "loss": 1.4734, + "theoretical_loss": 3.7126231086702544, + "tokens_seen": 836829184 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003770110330992979, + "loss": 1.36, + "theoretical_loss": 3.7125945218552228, + "tokens_seen": 836894720 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003770010030090271, + "loss": 1.4058, + "theoretical_loss": 3.7125659379054454, + "tokens_seen": 836960256 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037699097291875625, + "loss": 1.3714, + "theoretical_loss": 3.71253735682041, + "tokens_seen": 837025792 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037698094282848544, + "loss": 1.3164, + "theoretical_loss": 3.7125087785996045, + "tokens_seen": 837091328 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037697091273821467, + "loss": 1.3636, + "theoretical_loss": 3.712480203242519, + "tokens_seen": 837156864 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -7.06085775163956e-05, + "objective/train/docs_used": 482509, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.115107774734497, + "objective/train/original_loss": 3.115108013153076, + "objective/train/theoretical_loss": 3.7124516307486424, + "objective/train/tokens_used": 857682400, + "objective/train/value_avg": -0.006359100341796875, + "objective/train/value_loss": 6.371740892063826e-05, + "objective/train/value_max": -0.0002846717834472656, + "objective/train/value_min": -0.057586669921875, + "objective/train/value_reward_corr": 0.45053297719490626, + "objective/train/value_std": 0.00510406494140625, + "objective/train/weight_avg": 1.002297282218933, + "objective/train/weighted_lm_loss": 3.1259765625, + "objective/train/weights_max": 1.7110536098480225, + "objective/train/weights_min": 0.254957377910614, + "theoretical_loss": 3.7124516307486424, + "tokens_seen": 837222400 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003769608826479438, + "loss": 1.4433, + "theoretical_loss": 3.7124516307486424, + "tokens_seen": 837222400 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037695085255767303, + "loss": 1.4495, + "theoretical_loss": 3.7124230611174633, + "tokens_seen": 837287936 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003769408224674022, + "loss": 1.4064, + "theoretical_loss": 3.7123944943484704, + "tokens_seen": 837353472 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003769307923771314, + "loss": 1.3513, + "theoretical_loss": 3.7123659304411536, + "tokens_seen": 837419008 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003769207622868606, + "loss": 1.392, + "theoretical_loss": 3.7123373693950024, + "tokens_seen": 837484544 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037691073219658976, + "loss": 1.3552, + "theoretical_loss": 3.712308811209505, + "tokens_seen": 837550080 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037690070210631894, + "loss": 1.321, + "theoretical_loss": 3.7122802558841532, + "tokens_seen": 837615616 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003768906720160482, + "loss": 1.5196, + "theoretical_loss": 3.7122517034184352, + "tokens_seen": 837681152 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003768806419257773, + "loss": 1.3496, + "theoretical_loss": 3.7122231538118413, + "tokens_seen": 837746688 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037687061183550654, + "loss": 1.3817, + "theoretical_loss": 3.7121946070638625, + "tokens_seen": 837812224 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003768605817452357, + "loss": 1.4079, + "theoretical_loss": 3.7121660631739877, + "tokens_seen": 837877760 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003768505516549649, + "loss": 1.3812, + "theoretical_loss": 3.712137522141708, + "tokens_seen": 837943296 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037684052156469413, + "loss": 1.3804, + "theoretical_loss": 3.7121089839665133, + "tokens_seen": 838008832 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037683049147442326, + "loss": 1.4166, + "theoretical_loss": 3.7120804486478947, + "tokens_seen": 838074368 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003768204613841525, + "loss": 1.4683, + "theoretical_loss": 3.7120519161853425, + "tokens_seen": 838139904 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003768104312938817, + "loss": 1.3783, + "theoretical_loss": 3.7120233865783483, + "tokens_seen": 838205440 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037680040120361086, + "loss": 1.4314, + "theoretical_loss": 3.711994859826402, + "tokens_seen": 838270976 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037679037111334004, + "loss": 1.3749, + "theoretical_loss": 3.711966335928995, + "tokens_seen": 838336512 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003767803410230692, + "loss": 1.4534, + "theoretical_loss": 3.7119378148856192, + "tokens_seen": 838402048 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003767703109327984, + "loss": 1.478, + "theoretical_loss": 3.711909296695766, + "tokens_seen": 838467584 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037676028084252764, + "loss": 1.3922, + "theoretical_loss": 3.7118807813589263, + "tokens_seen": 838533120 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037675025075225676, + "loss": 1.3114, + "theoretical_loss": 3.7118522688745914, + "tokens_seen": 838598656 + }, + { + "epoch": 0.25, + "learning_rate": 0.000376740220661986, + "loss": 1.4115, + "theoretical_loss": 3.7118237592422543, + "tokens_seen": 838664192 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003767301905717151, + "loss": 1.443, + "theoretical_loss": 3.711795252461406, + "tokens_seen": 838729728 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037672016048144436, + "loss": 1.4866, + "theoretical_loss": 3.711766748531538, + "tokens_seen": 838795264 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0011613731039687991, + "objective/train/docs_used": 483094, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0759902000427246, + "objective/train/original_loss": 3.0759904384613037, + "objective/train/theoretical_loss": 3.7117382474521436, + "objective/train/tokens_used": 859320800, + "objective/train/value_avg": -0.005474090576171875, + "objective/train/value_loss": 8.969875489128754e-05, + "objective/train/value_max": -0.00038743019104003906, + "objective/train/value_min": -0.0638427734375, + "objective/train/value_reward_corr": 0.4313102917589946, + "objective/train/value_std": 0.004169464111328125, + "objective/train/weight_avg": 0.9924256801605225, + "objective/train/weighted_lm_loss": 3.0592424869537354, + "objective/train/weights_max": 1.6833412647247314, + "objective/train/weights_min": 0.08394700288772583, + "theoretical_loss": 3.7117382474521436, + "tokens_seen": 838860800 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037671013039117354, + "loss": 1.4242, + "theoretical_loss": 3.7117382474521436, + "tokens_seen": 838860800 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003767001003009027, + "loss": 1.4524, + "theoretical_loss": 3.711709749222715, + "tokens_seen": 838926336 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003766900702106319, + "loss": 1.4726, + "theoretical_loss": 3.7116812538427437, + "tokens_seen": 838991872 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003766800401203611, + "loss": 1.4839, + "theoretical_loss": 3.7116527613117234, + "tokens_seen": 839057408 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037667001003009027, + "loss": 1.3819, + "theoretical_loss": 3.7116242716291454, + "tokens_seen": 839122944 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003766599799398195, + "loss": 1.3997, + "theoretical_loss": 3.711595784794504, + "tokens_seen": 839188480 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037664994984954863, + "loss": 1.3941, + "theoretical_loss": 3.711567300807291, + "tokens_seen": 839254016 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037663991975927786, + "loss": 1.3274, + "theoretical_loss": 3.7115388196670005, + "tokens_seen": 839319552 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037662988966900705, + "loss": 1.4063, + "theoretical_loss": 3.711510341373124, + "tokens_seen": 839385088 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003766198595787362, + "loss": 1.3471, + "theoretical_loss": 3.711481865925157, + "tokens_seen": 839450624 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003766098294884654, + "loss": 1.3662, + "theoretical_loss": 3.7114533933225915, + "tokens_seen": 839516160 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003765997993981946, + "loss": 1.4349, + "theoretical_loss": 3.711424923564921, + "tokens_seen": 839581696 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037658976930792377, + "loss": 1.3824, + "theoretical_loss": 3.71139645665164, + "tokens_seen": 839647232 + }, + { + "epoch": 0.25, + "learning_rate": 0.000376579739217653, + "loss": 1.3363, + "theoretical_loss": 3.7113679925822423, + "tokens_seen": 839712768 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037656970912738213, + "loss": 1.4361, + "theoretical_loss": 3.711339531356221, + "tokens_seen": 839778304 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037655967903711137, + "loss": 1.4747, + "theoretical_loss": 3.7113110729730714, + "tokens_seen": 839843840 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003765496489468405, + "loss": 1.3875, + "theoretical_loss": 3.7112826174322864, + "tokens_seen": 839909376 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037653961885656973, + "loss": 1.4694, + "theoretical_loss": 3.7112541647333614, + "tokens_seen": 839974912 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003765295887662989, + "loss": 1.4342, + "theoretical_loss": 3.7112257148757903, + "tokens_seen": 840040448 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003765195586760281, + "loss": 1.3262, + "theoretical_loss": 3.7111972678590686, + "tokens_seen": 840105984 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003765095285857573, + "loss": 1.4331, + "theoretical_loss": 3.71116882368269, + "tokens_seen": 840171520 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037649949849548645, + "loss": 1.4445, + "theoretical_loss": 3.7111403823461497, + "tokens_seen": 840237056 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037648946840521564, + "loss": 1.3791, + "theoretical_loss": 3.711111943848943, + "tokens_seen": 840302592 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037647943831494487, + "loss": 1.3016, + "theoretical_loss": 3.7110835081905647, + "tokens_seen": 840368128 + }, + { + "epoch": 0.25, + "learning_rate": 0.000376469408224674, + "loss": 1.4996, + "theoretical_loss": 3.7110550753705103, + "tokens_seen": 840433664 + }, + { + "epoch": 0.25, + "objective/train/advantage_avg": -0.0012425751192495227, + "objective/train/docs_used": 483926, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.162062168121338, + "objective/train/original_loss": 3.162062168121338, + "objective/train/theoretical_loss": 3.711026645388275, + "objective/train/tokens_used": 860959200, + "objective/train/value_avg": -0.006488800048828125, + "objective/train/value_loss": 7.610340253449976e-05, + "objective/train/value_max": -0.00047850608825683594, + "objective/train/value_min": -0.06903076171875, + "objective/train/value_reward_corr": 0.5699394694522335, + "objective/train/value_std": 0.006465911865234375, + "objective/train/weight_avg": 0.9911224246025085, + "objective/train/weighted_lm_loss": 3.1351404190063477, + "objective/train/weights_max": 1.5811976194381714, + "objective/train/weights_min": 0.34238487482070923, + "theoretical_loss": 3.711026645388275, + "tokens_seen": 840499200 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037645937813440323, + "loss": 1.5171, + "theoretical_loss": 3.711026645388275, + "tokens_seen": 840499200 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003764493480441324, + "loss": 1.4539, + "theoretical_loss": 3.710998218243355, + "tokens_seen": 840564736 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003764393179538616, + "loss": 1.3688, + "theoretical_loss": 3.7109697939352446, + "tokens_seen": 840630272 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003764292878635908, + "loss": 1.444, + "theoretical_loss": 3.7109413724634406, + "tokens_seen": 840695808 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037641925777331996, + "loss": 1.3998, + "theoretical_loss": 3.710912953827439, + "tokens_seen": 840761344 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037640922768304914, + "loss": 1.4002, + "theoretical_loss": 3.7108845380267352, + "tokens_seen": 840826880 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003763991975927784, + "loss": 1.2931, + "theoretical_loss": 3.7108561250608254, + "tokens_seen": 840892416 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003763891675025075, + "loss": 1.3532, + "theoretical_loss": 3.7108277149292066, + "tokens_seen": 840957952 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037637913741223674, + "loss": 1.3771, + "theoretical_loss": 3.7107993076313748, + "tokens_seen": 841023488 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037636910732196586, + "loss": 1.4414, + "theoretical_loss": 3.710770903166827, + "tokens_seen": 841089024 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003763590772316951, + "loss": 1.3835, + "theoretical_loss": 3.7107425015350586, + "tokens_seen": 841154560 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003763490471414243, + "loss": 1.366, + "theoretical_loss": 3.710714102735568, + "tokens_seen": 841220096 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037633901705115346, + "loss": 1.3996, + "theoretical_loss": 3.7106857067678503, + "tokens_seen": 841285632 + }, + { + "epoch": 0.25, + "learning_rate": 0.00037632898696088264, + "loss": 1.401, + "theoretical_loss": 3.7106573136314047, + "tokens_seen": 841351168 + }, + { + "epoch": 0.25, + "learning_rate": 0.0003763189568706119, + "loss": 1.3276, + "theoretical_loss": 3.710628923325727, + "tokens_seen": 841416704 + }, + { + "epoch": 0.25, + "learning_rate": 0.000376308926780341, + "loss": 1.5064, + "theoretical_loss": 3.7106005358503147, + "tokens_seen": 841482240 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037629889669007024, + "loss": 1.4401, + "theoretical_loss": 3.7105721512046657, + "tokens_seen": 841547776 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037628886659979937, + "loss": 1.3661, + "theoretical_loss": 3.7105437693882775, + "tokens_seen": 841613312 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003762788365095286, + "loss": 1.3959, + "theoretical_loss": 3.7105153904006474, + "tokens_seen": 841678848 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003762688064192578, + "loss": 1.3471, + "theoretical_loss": 3.7104870142412736, + "tokens_seen": 841744384 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037625877632898696, + "loss": 1.5, + "theoretical_loss": 3.710458640909654, + "tokens_seen": 841809920 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037624874623871615, + "loss": 1.266, + "theoretical_loss": 3.7104302704052863, + "tokens_seen": 841875456 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003762387161484453, + "loss": 1.3693, + "theoretical_loss": 3.7104019027276696, + "tokens_seen": 841940992 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003762286860581745, + "loss": 1.4894, + "theoretical_loss": 3.7103735378763014, + "tokens_seen": 842006528 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037621865596790374, + "loss": 1.3415, + "theoretical_loss": 3.71034517585068, + "tokens_seen": 842072064 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": -0.0003986908122897148, + "objective/train/docs_used": 484283, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0953471660614014, + "objective/train/original_loss": 3.0953471660614014, + "objective/train/theoretical_loss": 3.7103168166503053, + "objective/train/tokens_used": 862597600, + "objective/train/value_avg": -0.006420135498046875, + "objective/train/value_loss": 9.420266724191606e-05, + "objective/train/value_max": -0.0005745887756347656, + "objective/train/value_min": -0.0970458984375, + "objective/train/value_reward_corr": 0.3553812506013309, + "objective/train/value_std": 0.005218505859375, + "objective/train/weight_avg": 1.0002610683441162, + "objective/train/weighted_lm_loss": 3.109797477722168, + "objective/train/weights_max": 2.3041512966156006, + "objective/train/weights_min": 0.08548197150230408, + "theoretical_loss": 3.7103168166503053, + "tokens_seen": 842137600 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037620862587763287, + "loss": 1.5317, + "theoretical_loss": 3.7103168166503053, + "tokens_seen": 842137600 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003761985957873621, + "loss": 1.5343, + "theoretical_loss": 3.7102884602746746, + "tokens_seen": 842203136 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037618856569709123, + "loss": 1.5406, + "theoretical_loss": 3.710260106723288, + "tokens_seen": 842268672 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037617853560682047, + "loss": 1.4133, + "theoretical_loss": 3.7102317559956437, + "tokens_seen": 842334208 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037616850551654965, + "loss": 1.4449, + "theoretical_loss": 3.7102034080912407, + "tokens_seen": 842399744 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037615847542627883, + "loss": 1.3607, + "theoretical_loss": 3.7101750630095793, + "tokens_seen": 842465280 + }, + { + "epoch": 0.26, + "learning_rate": 0.000376148445336008, + "loss": 1.4251, + "theoretical_loss": 3.7101467207501573, + "tokens_seen": 842530816 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037613841524573725, + "loss": 1.3716, + "theoretical_loss": 3.7101183813124754, + "tokens_seen": 842596352 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037612838515546637, + "loss": 1.3573, + "theoretical_loss": 3.7100900446960328, + "tokens_seen": 842661888 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003761183550651956, + "loss": 1.3399, + "theoretical_loss": 3.7100617109003293, + "tokens_seen": 842727424 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003761083249749248, + "loss": 1.428, + "theoretical_loss": 3.710033379924865, + "tokens_seen": 842792960 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037609829488465397, + "loss": 1.4725, + "theoretical_loss": 3.7100050517691403, + "tokens_seen": 842858496 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003760882647943832, + "loss": 1.2781, + "theoretical_loss": 3.7099767264326537, + "tokens_seen": 842924032 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037607823470411233, + "loss": 1.4097, + "theoretical_loss": 3.7099484039149075, + "tokens_seen": 842989568 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037606820461384157, + "loss": 1.3366, + "theoretical_loss": 3.7099200842154003, + "tokens_seen": 843055104 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003760581745235707, + "loss": 1.4879, + "theoretical_loss": 3.709891767333634, + "tokens_seen": 843120640 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037604814443329993, + "loss": 1.4835, + "theoretical_loss": 3.709863453269109, + "tokens_seen": 843186176 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003760381143430291, + "loss": 1.4775, + "theoretical_loss": 3.709835142021325, + "tokens_seen": 843251712 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003760280842527583, + "loss": 1.4629, + "theoretical_loss": 3.7098068335897842, + "tokens_seen": 843317248 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003760180541624875, + "loss": 1.4605, + "theoretical_loss": 3.7097785279739868, + "tokens_seen": 843382784 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037600802407221665, + "loss": 1.3162, + "theoretical_loss": 3.7097502251734342, + "tokens_seen": 843448320 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037599799398194584, + "loss": 1.4809, + "theoretical_loss": 3.7097219251876283, + "tokens_seen": 843513856 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037598796389167507, + "loss": 1.3327, + "theoretical_loss": 3.7096936280160695, + "tokens_seen": 843579392 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003759779338014042, + "loss": 1.3952, + "theoretical_loss": 3.7096653336582603, + "tokens_seen": 843644928 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037596790371113343, + "loss": 1.4652, + "theoretical_loss": 3.7096370421137017, + "tokens_seen": 843710464 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": -0.000297591119306162, + "objective/train/docs_used": 485567, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.567457437515259, + "objective/train/original_loss": 2.5674571990966797, + "objective/train/theoretical_loss": 3.7096087533818953, + "objective/train/tokens_used": 864236000, + "objective/train/value_avg": -0.005046844482421875, + "objective/train/value_loss": 7.434578583342955e-05, + "objective/train/value_max": -0.00031256675720214844, + "objective/train/value_min": -0.0810546875, + "objective/train/value_reward_corr": 0.24319851704556325, + "objective/train/value_std": 0.0034732818603515625, + "objective/train/weight_avg": 1.0001896619796753, + "objective/train/weighted_lm_loss": 2.572587013244629, + "objective/train/weights_max": 2.2491376399993896, + "objective/train/weights_min": 0.08878213912248611, + "theoretical_loss": 3.7096087533818953, + "tokens_seen": 843776000 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003759578736208626, + "loss": 1.408, + "theoretical_loss": 3.7096087533818953, + "tokens_seen": 843776000 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003759478435305918, + "loss": 1.4556, + "theoretical_loss": 3.7095804674623443, + "tokens_seen": 843841536 + }, + { + "epoch": 0.26, + "learning_rate": 0.000375937813440321, + "loss": 1.4546, + "theoretical_loss": 3.709552184354549, + "tokens_seen": 843907072 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037592778335005016, + "loss": 1.4218, + "theoretical_loss": 3.709523904058013, + "tokens_seen": 843972608 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037591775325977934, + "loss": 1.454, + "theoretical_loss": 3.709495626572237, + "tokens_seen": 844038144 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003759077231695086, + "loss": 1.2601, + "theoretical_loss": 3.709467351896726, + "tokens_seen": 844103680 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003758976930792377, + "loss": 1.4452, + "theoretical_loss": 3.70943908003098, + "tokens_seen": 844169216 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037588766298896694, + "loss": 1.362, + "theoretical_loss": 3.709410810974503, + "tokens_seen": 844234752 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037587763289869606, + "loss": 1.4061, + "theoretical_loss": 3.7093825447267976, + "tokens_seen": 844300288 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003758676028084253, + "loss": 1.3515, + "theoretical_loss": 3.709354281287367, + "tokens_seen": 844365824 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003758575727181545, + "loss": 1.3027, + "theoretical_loss": 3.7093260206557135, + "tokens_seen": 844431360 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037584754262788366, + "loss": 1.4043, + "theoretical_loss": 3.7092977628313406, + "tokens_seen": 844496896 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037583751253761284, + "loss": 1.3572, + "theoretical_loss": 3.709269507813752, + "tokens_seen": 844562432 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003758274824473421, + "loss": 1.388, + "theoretical_loss": 3.7092412556024503, + "tokens_seen": 844627968 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003758174523570712, + "loss": 1.4828, + "theoretical_loss": 3.70921300619694, + "tokens_seen": 844693504 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037580742226680044, + "loss": 1.4606, + "theoretical_loss": 3.7091847595967247, + "tokens_seen": 844759040 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037579739217652957, + "loss": 1.3856, + "theoretical_loss": 3.7091565158013076, + "tokens_seen": 844824576 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003757873620862588, + "loss": 1.5151, + "theoretical_loss": 3.709128274810193, + "tokens_seen": 844890112 + }, + { + "epoch": 0.26, + "learning_rate": 0.000375777331995988, + "loss": 1.2637, + "theoretical_loss": 3.709100036622885, + "tokens_seen": 844955648 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037576730190571716, + "loss": 1.4525, + "theoretical_loss": 3.709071801238888, + "tokens_seen": 845021184 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037575727181544635, + "loss": 1.4589, + "theoretical_loss": 3.709043568657706, + "tokens_seen": 845086720 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003757472417251755, + "loss": 1.4023, + "theoretical_loss": 3.709015338878843, + "tokens_seen": 845152256 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003757372116349047, + "loss": 1.5041, + "theoretical_loss": 3.7089871119018047, + "tokens_seen": 845217792 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037572718154463394, + "loss": 1.4192, + "theoretical_loss": 3.7089588877260953, + "tokens_seen": 845283328 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037571715145436307, + "loss": 1.4266, + "theoretical_loss": 3.708930666351219, + "tokens_seen": 845348864 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": 0.0006659977370873094, + "objective/train/docs_used": 486205, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.789411783218384, + "objective/train/original_loss": 2.7894115447998047, + "objective/train/theoretical_loss": 3.7089024477766817, + "objective/train/tokens_used": 865874400, + "objective/train/value_avg": -0.0050506591796875, + "objective/train/value_loss": 6.064239642000757e-05, + "objective/train/value_max": -0.0003669261932373047, + "objective/train/value_min": -0.09588623046875, + "objective/train/value_reward_corr": 0.26936060997349887, + "objective/train/value_std": 0.0050506591796875, + "objective/train/weight_avg": 1.009461760520935, + "objective/train/weighted_lm_loss": 2.8205373287200928, + "objective/train/weights_max": 2.3530514240264893, + "objective/train/weights_min": 0.1449710875749588, + "theoretical_loss": 3.7089024477766817, + "tokens_seen": 845414400 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003757071213640923, + "loss": 1.3938, + "theoretical_loss": 3.7089024477766817, + "tokens_seen": 845414400 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037569709127382143, + "loss": 1.4929, + "theoretical_loss": 3.708874232001988, + "tokens_seen": 845479936 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037568706118355067, + "loss": 1.3047, + "theoretical_loss": 3.7088460190266437, + "tokens_seen": 845545472 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037567703109327985, + "loss": 1.2748, + "theoretical_loss": 3.708817808850153, + "tokens_seen": 845611008 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037566700100300903, + "loss": 1.3436, + "theoretical_loss": 3.708789601472022, + "tokens_seen": 845676544 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003756569709127382, + "loss": 1.4553, + "theoretical_loss": 3.7087613968917563, + "tokens_seen": 845742080 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037564694082246745, + "loss": 1.2398, + "theoretical_loss": 3.7087331951088616, + "tokens_seen": 845807616 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003756369107321966, + "loss": 1.4542, + "theoretical_loss": 3.7087049961228438, + "tokens_seen": 845873152 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003756268806419258, + "loss": 1.3083, + "theoretical_loss": 3.7086767999332086, + "tokens_seen": 845938688 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037561685055165494, + "loss": 1.4835, + "theoretical_loss": 3.7086486065394624, + "tokens_seen": 846004224 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037560682046138417, + "loss": 1.3895, + "theoretical_loss": 3.7086204159411116, + "tokens_seen": 846069760 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037559679037111335, + "loss": 1.3822, + "theoretical_loss": 3.708592228137662, + "tokens_seen": 846135296 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037558676028084253, + "loss": 1.3065, + "theoretical_loss": 3.7085640431286198, + "tokens_seen": 846200832 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003755767301905717, + "loss": 1.393, + "theoretical_loss": 3.708535860913493, + "tokens_seen": 846266368 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003755667001003009, + "loss": 1.4674, + "theoretical_loss": 3.7085076814917866, + "tokens_seen": 846331904 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003755566700100301, + "loss": 1.2928, + "theoretical_loss": 3.708479504863008, + "tokens_seen": 846397440 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003755466399197593, + "loss": 1.4602, + "theoretical_loss": 3.708451331026665, + "tokens_seen": 846462976 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037553660982948844, + "loss": 1.485, + "theoretical_loss": 3.7084231599822637, + "tokens_seen": 846528512 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003755265797392177, + "loss": 1.2951, + "theoretical_loss": 3.7083949917293118, + "tokens_seen": 846594048 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003755165496489468, + "loss": 1.4718, + "theoretical_loss": 3.7083668262673166, + "tokens_seen": 846659584 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037550651955867604, + "loss": 1.4144, + "theoretical_loss": 3.7083386635957853, + "tokens_seen": 846725120 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003754964894684052, + "loss": 1.3181, + "theoretical_loss": 3.708310503714226, + "tokens_seen": 846790656 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003754864593781344, + "loss": 1.4394, + "theoretical_loss": 3.7082823466221457, + "tokens_seen": 846856192 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003754764292878636, + "loss": 1.4035, + "theoretical_loss": 3.708254192319053, + "tokens_seen": 846921728 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003754663991975928, + "loss": 1.4385, + "theoretical_loss": 3.7082260408044547, + "tokens_seen": 846987264 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": 0.0003360224945936352, + "objective/train/docs_used": 487507, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0192646980285645, + "objective/train/original_loss": 3.0192649364471436, + "objective/train/theoretical_loss": 3.7081978920778607, + "objective/train/tokens_used": 867512800, + "objective/train/value_avg": -0.004894256591796875, + "objective/train/value_loss": 4.5056967792334035e-05, + "objective/train/value_max": -0.000469207763671875, + "objective/train/value_min": -0.044525146484375, + "objective/train/value_reward_corr": 0.3421011773228369, + "objective/train/value_std": 0.0038909912109375, + "objective/train/weight_avg": 1.005489706993103, + "objective/train/weighted_lm_loss": 3.0291061401367188, + "objective/train/weights_max": 1.4743003845214844, + "objective/train/weights_min": 0.3436541259288788, + "theoretical_loss": 3.7081978920778607, + "tokens_seen": 847052800 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037545636910732194, + "loss": 1.4107, + "theoretical_loss": 3.7081978920778607, + "tokens_seen": 847052800 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003754463390170512, + "loss": 1.4091, + "theoretical_loss": 3.7081697461387777, + "tokens_seen": 847118336 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003754363089267803, + "loss": 1.3893, + "theoretical_loss": 3.7081416029867147, + "tokens_seen": 847183872 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037542627883650954, + "loss": 1.3804, + "theoretical_loss": 3.7081134626211796, + "tokens_seen": 847249408 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003754162487462387, + "loss": 1.3724, + "theoretical_loss": 3.708085325041682, + "tokens_seen": 847314944 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003754062186559679, + "loss": 1.2944, + "theoretical_loss": 3.7080571902477297, + "tokens_seen": 847380480 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003753961885656971, + "loss": 1.389, + "theoretical_loss": 3.7080290582388318, + "tokens_seen": 847446016 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037538615847542626, + "loss": 1.3474, + "theoretical_loss": 3.7080009290144975, + "tokens_seen": 847511552 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037537612838515544, + "loss": 1.363, + "theoretical_loss": 3.7079728025742353, + "tokens_seen": 847577088 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003753660982948847, + "loss": 1.4427, + "theoretical_loss": 3.707944678917555, + "tokens_seen": 847642624 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037535606820461386, + "loss": 1.4323, + "theoretical_loss": 3.7079165580439657, + "tokens_seen": 847708160 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037534603811434304, + "loss": 1.4338, + "theoretical_loss": 3.707888439952977, + "tokens_seen": 847773696 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003753360080240723, + "loss": 1.3619, + "theoretical_loss": 3.7078603246440984, + "tokens_seen": 847839232 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003753259779338014, + "loss": 1.4681, + "theoretical_loss": 3.7078322121168394, + "tokens_seen": 847904768 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037531594784353064, + "loss": 1.385, + "theoretical_loss": 3.70780410237071, + "tokens_seen": 847970304 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037530591775325977, + "loss": 1.442, + "theoretical_loss": 3.7077759954052203, + "tokens_seen": 848035840 + }, + { + "epoch": 0.26, + "learning_rate": 0.000375295887662989, + "loss": 1.2916, + "theoretical_loss": 3.7077478912198805, + "tokens_seen": 848101376 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003752858575727182, + "loss": 1.4422, + "theoretical_loss": 3.7077197898142, + "tokens_seen": 848166912 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037527582748244736, + "loss": 1.3418, + "theoretical_loss": 3.70769169118769, + "tokens_seen": 848232448 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037526579739217655, + "loss": 1.4934, + "theoretical_loss": 3.7076635953398607, + "tokens_seen": 848297984 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003752557673019057, + "loss": 1.3134, + "theoretical_loss": 3.7076355022702225, + "tokens_seen": 848363520 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003752457372116349, + "loss": 1.3472, + "theoretical_loss": 3.7076074119782865, + "tokens_seen": 848429056 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037523570712136414, + "loss": 1.4148, + "theoretical_loss": 3.7075793244635626, + "tokens_seen": 848494592 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037522567703109327, + "loss": 1.4829, + "theoretical_loss": 3.707551239725563, + "tokens_seen": 848560128 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003752156469408225, + "loss": 1.3777, + "theoretical_loss": 3.707523157763798, + "tokens_seen": 848625664 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": -0.00016662615234963596, + "objective/train/docs_used": 487917, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8231136798858643, + "objective/train/original_loss": 2.8231139183044434, + "objective/train/theoretical_loss": 3.7074950785777787, + "objective/train/tokens_used": 869151200, + "objective/train/value_avg": -0.004940032958984375, + "objective/train/value_loss": 3.776865924010053e-05, + "objective/train/value_max": -0.0004239082336425781, + "objective/train/value_min": -0.0615234375, + "objective/train/value_reward_corr": 0.26837050833547876, + "objective/train/value_std": 0.0035572052001953125, + "objective/train/weight_avg": 1.0001682043075562, + "objective/train/weighted_lm_loss": 2.8257858753204346, + "objective/train/weights_max": 1.6845263242721558, + "objective/train/weights_min": 0.4828398823738098, + "theoretical_loss": 3.7074950785777787, + "tokens_seen": 848691200 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037520561685055163, + "loss": 1.3689, + "theoretical_loss": 3.7074950785777787, + "tokens_seen": 848691200 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037519558676028087, + "loss": 1.5824, + "theoretical_loss": 3.707467002167017, + "tokens_seen": 848756736 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037518555667001005, + "loss": 1.4084, + "theoretical_loss": 3.7074389285310243, + "tokens_seen": 848822272 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037517552657973923, + "loss": 1.3521, + "theoretical_loss": 3.707410857669312, + "tokens_seen": 848887808 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003751654964894684, + "loss": 1.3785, + "theoretical_loss": 3.707382789581391, + "tokens_seen": 848953344 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037515546639919765, + "loss": 1.4107, + "theoretical_loss": 3.707354724266774, + "tokens_seen": 849018880 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003751454363089268, + "loss": 1.3839, + "theoretical_loss": 3.707326661724973, + "tokens_seen": 849084416 + }, + { + "epoch": 0.26, + "learning_rate": 0.000375135406218656, + "loss": 1.4201, + "theoretical_loss": 3.7072986019554994, + "tokens_seen": 849149952 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037512537612838514, + "loss": 1.4254, + "theoretical_loss": 3.7072705449578667, + "tokens_seen": 849215488 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037511534603811437, + "loss": 1.3997, + "theoretical_loss": 3.7072424907315855, + "tokens_seen": 849281024 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037510531594784355, + "loss": 1.4144, + "theoretical_loss": 3.7072144392761697, + "tokens_seen": 849346560 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037509528585757273, + "loss": 1.3928, + "theoretical_loss": 3.70718639059113, + "tokens_seen": 849412096 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003750852557673019, + "loss": 1.4941, + "theoretical_loss": 3.7071583446759817, + "tokens_seen": 849477632 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003750752256770311, + "loss": 1.2488, + "theoretical_loss": 3.7071303015302357, + "tokens_seen": 849543168 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003750651955867603, + "loss": 1.4057, + "theoretical_loss": 3.707102261153405, + "tokens_seen": 849608704 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003750551654964895, + "loss": 1.2103, + "theoretical_loss": 3.7070742235450034, + "tokens_seen": 849674240 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037504513540621864, + "loss": 1.4272, + "theoretical_loss": 3.7070461887045436, + "tokens_seen": 849739776 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003750351053159479, + "loss": 1.4056, + "theoretical_loss": 3.7070181566315394, + "tokens_seen": 849805312 + }, + { + "epoch": 0.26, + "learning_rate": 0.000375025075225677, + "loss": 1.4071, + "theoretical_loss": 3.7069901273255037, + "tokens_seen": 849870848 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037501504513540624, + "loss": 1.3415, + "theoretical_loss": 3.70696210078595, + "tokens_seen": 849936384 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003750050150451354, + "loss": 1.435, + "theoretical_loss": 3.7069340770123924, + "tokens_seen": 850001920 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003749949849548646, + "loss": 1.3486, + "theoretical_loss": 3.7069060560043443, + "tokens_seen": 850067456 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003749849548645938, + "loss": 1.408, + "theoretical_loss": 3.70687803776132, + "tokens_seen": 850132992 + }, + { + "epoch": 0.26, + "learning_rate": 0.000374974924774323, + "loss": 1.369, + "theoretical_loss": 3.7068500222828327, + "tokens_seen": 850198528 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037496489468405214, + "loss": 1.4378, + "theoretical_loss": 3.706822009568398, + "tokens_seen": 850264064 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": -0.00013586098793894053, + "objective/train/docs_used": 489068, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.81002140045166, + "objective/train/original_loss": 2.810021162033081, + "objective/train/theoretical_loss": 3.706793999617529, + "objective/train/tokens_used": 870789600, + "objective/train/value_avg": -0.0052490234375, + "objective/train/value_loss": 7.162934343796223e-05, + "objective/train/value_max": -0.00014770030975341797, + "objective/train/value_min": -0.063232421875, + "objective/train/value_reward_corr": 0.39842467371719775, + "objective/train/value_std": 0.004352569580078125, + "objective/train/weight_avg": 1.0017304420471191, + "objective/train/weighted_lm_loss": 2.8119359016418457, + "objective/train/weights_max": 1.771628975868225, + "objective/train/weights_min": 0.14095745980739594, + "theoretical_loss": 3.706793999617529, + "tokens_seen": 850329600 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003749548645937814, + "loss": 1.4175, + "theoretical_loss": 3.706793999617529, + "tokens_seen": 850329600 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003749448345035105, + "loss": 1.516, + "theoretical_loss": 3.70676599242974, + "tokens_seen": 850395136 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037493480441323974, + "loss": 1.3255, + "theoretical_loss": 3.7067379880045466, + "tokens_seen": 850460672 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003749247743229689, + "loss": 1.4847, + "theoretical_loss": 3.7067099863414628, + "tokens_seen": 850526208 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003749147442326981, + "loss": 1.3971, + "theoretical_loss": 3.7066819874400028, + "tokens_seen": 850591744 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003749047141424273, + "loss": 1.4387, + "theoretical_loss": 3.7066539912996825, + "tokens_seen": 850657280 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037489468405215646, + "loss": 1.4466, + "theoretical_loss": 3.7066259979200167, + "tokens_seen": 850722816 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037488465396188564, + "loss": 1.3914, + "theoretical_loss": 3.70659800730052, + "tokens_seen": 850788352 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003748746238716149, + "loss": 1.4351, + "theoretical_loss": 3.706570019440708, + "tokens_seen": 850853888 + }, + { + "epoch": 0.26, + "learning_rate": 0.000374864593781344, + "loss": 1.4354, + "theoretical_loss": 3.706542034340096, + "tokens_seen": 850919424 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037485456369107324, + "loss": 1.4239, + "theoretical_loss": 3.7065140519982, + "tokens_seen": 850984960 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037484453360080237, + "loss": 1.4412, + "theoretical_loss": 3.7064860724145348, + "tokens_seen": 851050496 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003748345035105316, + "loss": 1.4112, + "theoretical_loss": 3.7064580955886166, + "tokens_seen": 851116032 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003748244734202608, + "loss": 1.4124, + "theoretical_loss": 3.7064301215199613, + "tokens_seen": 851181568 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037481444332998997, + "loss": 1.4287, + "theoretical_loss": 3.7064021502080853, + "tokens_seen": 851247104 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037480441323971915, + "loss": 1.4592, + "theoretical_loss": 3.706374181652504, + "tokens_seen": 851312640 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003747943831494484, + "loss": 1.3576, + "theoretical_loss": 3.706346215852734, + "tokens_seen": 851378176 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003747843530591775, + "loss": 1.3158, + "theoretical_loss": 3.706318252808291, + "tokens_seen": 851443712 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037477432296890675, + "loss": 1.4431, + "theoretical_loss": 3.706290292518692, + "tokens_seen": 851509248 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037476429287863587, + "loss": 1.4167, + "theoretical_loss": 3.706262334983454, + "tokens_seen": 851574784 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003747542627883651, + "loss": 1.4189, + "theoretical_loss": 3.7062343802020936, + "tokens_seen": 851640320 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003747442326980943, + "loss": 1.4493, + "theoretical_loss": 3.706206428174127, + "tokens_seen": 851705856 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037473420260782347, + "loss": 1.3377, + "theoretical_loss": 3.706178478899072, + "tokens_seen": 851771392 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037472417251755265, + "loss": 1.3871, + "theoretical_loss": 3.706150532376445, + "tokens_seen": 851836928 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037471414242728183, + "loss": 1.4848, + "theoretical_loss": 3.706122588605764, + "tokens_seen": 851902464 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": 4.089896901859902e-05, + "objective/train/docs_used": 489617, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7969110012054443, + "objective/train/original_loss": 2.7969110012054443, + "objective/train/theoretical_loss": 3.7060946475865455, + "objective/train/tokens_used": 872428000, + "objective/train/value_avg": -0.004810333251953125, + "objective/train/value_loss": 4.761000309372321e-05, + "objective/train/value_max": -0.000415802001953125, + "objective/train/value_min": -0.058441162109375, + "objective/train/value_reward_corr": 0.3256240571848714, + "objective/train/value_std": 0.0038051605224609375, + "objective/train/weight_avg": 1.002602458000183, + "objective/train/weighted_lm_loss": 2.8207833766937256, + "objective/train/weights_max": 1.5632661581039429, + "objective/train/weights_min": 0.38026443123817444, + "theoretical_loss": 3.7060946475865455, + "tokens_seen": 851968000 + }, + { + "epoch": 0.26, + "learning_rate": 0.000374704112337011, + "loss": 1.3802, + "theoretical_loss": 3.7060946475865455, + "tokens_seen": 851968000 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037469408224674025, + "loss": 1.4266, + "theoretical_loss": 3.7060667093183075, + "tokens_seen": 852033536 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003746840521564694, + "loss": 1.4358, + "theoretical_loss": 3.706038773800567, + "tokens_seen": 852099072 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003746740220661986, + "loss": 1.3788, + "theoretical_loss": 3.7060108410328425, + "tokens_seen": 852164608 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037466399197592774, + "loss": 1.3571, + "theoretical_loss": 3.705982911014651, + "tokens_seen": 852230144 + }, + { + "epoch": 0.26, + "learning_rate": 0.000374653961885657, + "loss": 1.3946, + "theoretical_loss": 3.705954983745511, + "tokens_seen": 852295680 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037464393179538615, + "loss": 1.5775, + "theoretical_loss": 3.7059270592249405, + "tokens_seen": 852361216 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037463390170511534, + "loss": 1.3914, + "theoretical_loss": 3.705899137452458, + "tokens_seen": 852426752 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037462387161484457, + "loss": 1.4118, + "theoretical_loss": 3.705871218427581, + "tokens_seen": 852492288 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037461384152457375, + "loss": 1.3678, + "theoretical_loss": 3.7058433021498285, + "tokens_seen": 852557824 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037460381143430293, + "loss": 1.4023, + "theoretical_loss": 3.705815388618719, + "tokens_seen": 852623360 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003745937813440321, + "loss": 1.5437, + "theoretical_loss": 3.7057874778337707, + "tokens_seen": 852688896 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003745837512537613, + "loss": 1.5146, + "theoretical_loss": 3.7057595697945027, + "tokens_seen": 852754432 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003745737211634905, + "loss": 1.4776, + "theoretical_loss": 3.7057316645004343, + "tokens_seen": 852819968 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003745636910732197, + "loss": 1.4691, + "theoretical_loss": 3.705703761951084, + "tokens_seen": 852885504 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037455366098294884, + "loss": 1.4538, + "theoretical_loss": 3.7056758621459718, + "tokens_seen": 852951040 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003745436308926781, + "loss": 1.2789, + "theoretical_loss": 3.705647965084616, + "tokens_seen": 853016576 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003745336008024072, + "loss": 1.3026, + "theoretical_loss": 3.7056200707665363, + "tokens_seen": 853082112 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037452357071213644, + "loss": 1.4882, + "theoretical_loss": 3.7055921791912527, + "tokens_seen": 853147648 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003745135406218656, + "loss": 1.4871, + "theoretical_loss": 3.705564290358284, + "tokens_seen": 853213184 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003745035105315948, + "loss": 1.2923, + "theoretical_loss": 3.705536404267151, + "tokens_seen": 853278720 + }, + { + "epoch": 0.26, + "learning_rate": 0.000374493480441324, + "loss": 1.3804, + "theoretical_loss": 3.7055085209173724, + "tokens_seen": 853344256 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003744834503510532, + "loss": 1.4002, + "theoretical_loss": 3.705480640308469, + "tokens_seen": 853409792 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037447342026078234, + "loss": 1.4914, + "theoretical_loss": 3.705452762439961, + "tokens_seen": 853475328 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003744633901705116, + "loss": 1.4295, + "theoretical_loss": 3.705424887311368, + "tokens_seen": 853540864 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": -0.0009263426181860268, + "objective/train/docs_used": 491051, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.265573024749756, + "objective/train/original_loss": 2.2655727863311768, + "objective/train/theoretical_loss": 3.705397014922211, + "objective/train/tokens_used": 874066400, + "objective/train/value_avg": -0.00615692138671875, + "objective/train/value_loss": 6.841315189376473e-05, + "objective/train/value_max": -0.00075531005859375, + "objective/train/value_min": -0.0509033203125, + "objective/train/value_reward_corr": 0.4509165807604683, + "objective/train/value_std": 0.00466156005859375, + "objective/train/weight_avg": 0.9938395023345947, + "objective/train/weighted_lm_loss": 2.2479028701782227, + "objective/train/weights_max": 1.663681983947754, + "objective/train/weights_min": 0.1397687792778015, + "theoretical_loss": 3.705397014922211, + "tokens_seen": 853606400 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003744533600802407, + "loss": 1.3149, + "theoretical_loss": 3.705397014922211, + "tokens_seen": 853606400 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037444332998996994, + "loss": 1.3143, + "theoretical_loss": 3.7053691452720106, + "tokens_seen": 853671936 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003744332998996991, + "loss": 1.452, + "theoretical_loss": 3.7053412783602866, + "tokens_seen": 853737472 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003744232698094283, + "loss": 1.4472, + "theoretical_loss": 3.70531341418656, + "tokens_seen": 853803008 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003744132397191575, + "loss": 1.4751, + "theoretical_loss": 3.705285552750352, + "tokens_seen": 853868544 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037440320962888666, + "loss": 1.4163, + "theoretical_loss": 3.7052576940511837, + "tokens_seen": 853934080 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037439317953861585, + "loss": 1.4926, + "theoretical_loss": 3.7052298380885755, + "tokens_seen": 853999616 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003743831494483451, + "loss": 1.3692, + "theoretical_loss": 3.7052019848620494, + "tokens_seen": 854065152 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003743731193580742, + "loss": 1.3991, + "theoretical_loss": 3.7051741343711266, + "tokens_seen": 854130688 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037436308926780344, + "loss": 1.3766, + "theoretical_loss": 3.705146286615328, + "tokens_seen": 854196224 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037435305917753257, + "loss": 1.3469, + "theoretical_loss": 3.705118441594175, + "tokens_seen": 854261760 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003743430290872618, + "loss": 1.4483, + "theoretical_loss": 3.7050905993071908, + "tokens_seen": 854327296 + }, + { + "epoch": 0.26, + "learning_rate": 0.000374332998996991, + "loss": 1.3643, + "theoretical_loss": 3.7050627597538957, + "tokens_seen": 854392832 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037432296890672017, + "loss": 1.4491, + "theoretical_loss": 3.705034922933812, + "tokens_seen": 854458368 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037431293881644935, + "loss": 1.3978, + "theoretical_loss": 3.705007088846462, + "tokens_seen": 854523904 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003743029087261786, + "loss": 1.3489, + "theoretical_loss": 3.704979257491368, + "tokens_seen": 854589440 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003742928786359077, + "loss": 1.4885, + "theoretical_loss": 3.7049514288680516, + "tokens_seen": 854654976 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037428284854563695, + "loss": 1.4139, + "theoretical_loss": 3.704923602976036, + "tokens_seen": 854720512 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037427281845536607, + "loss": 1.4114, + "theoretical_loss": 3.704895779814844, + "tokens_seen": 854786048 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003742627883650953, + "loss": 1.4821, + "theoretical_loss": 3.704867959383997, + "tokens_seen": 854851584 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003742527582748245, + "loss": 1.2931, + "theoretical_loss": 3.7048401416830186, + "tokens_seen": 854917120 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037424272818455367, + "loss": 1.4888, + "theoretical_loss": 3.7048123267114317, + "tokens_seen": 854982656 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037423269809428285, + "loss": 1.4051, + "theoretical_loss": 3.7047845144687592, + "tokens_seen": 855048192 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037422266800401203, + "loss": 1.3055, + "theoretical_loss": 3.704756704954524, + "tokens_seen": 855113728 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003742126379137412, + "loss": 1.3683, + "theoretical_loss": 3.7047288981682502, + "tokens_seen": 855179264 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": 8.914316276786849e-05, + "objective/train/docs_used": 491675, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5916757583618164, + "objective/train/original_loss": 2.5916762351989746, + "objective/train/theoretical_loss": 3.70470109410946, + "objective/train/tokens_used": 875704800, + "objective/train/value_avg": -0.00617218017578125, + "objective/train/value_loss": 6.632770237047225e-05, + "objective/train/value_max": -0.00026726722717285156, + "objective/train/value_min": -0.0894775390625, + "objective/train/value_reward_corr": 0.3781147950855316, + "objective/train/value_std": 0.00499725341796875, + "objective/train/weight_avg": 1.0040827989578247, + "objective/train/weighted_lm_loss": 2.6150646209716797, + "objective/train/weights_max": 2.1253371238708496, + "objective/train/weights_min": 0.4598228335380554, + "theoretical_loss": 3.70470109410946, + "tokens_seen": 855244800 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037420260782347045, + "loss": 1.2748, + "theoretical_loss": 3.70470109410946, + "tokens_seen": 855244800 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003741925777331996, + "loss": 1.3861, + "theoretical_loss": 3.7046732927776773, + "tokens_seen": 855310336 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003741825476429288, + "loss": 1.379, + "theoretical_loss": 3.7046454941724263, + "tokens_seen": 855375872 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037417251755265794, + "loss": 1.4757, + "theoretical_loss": 3.70461769829323, + "tokens_seen": 855441408 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003741624874623872, + "loss": 1.3329, + "theoretical_loss": 3.7045899051396125, + "tokens_seen": 855506944 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037415245737211635, + "loss": 1.3509, + "theoretical_loss": 3.7045621147110985, + "tokens_seen": 855572480 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037414242728184554, + "loss": 1.3829, + "theoretical_loss": 3.704534327007211, + "tokens_seen": 855638016 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003741323971915747, + "loss": 1.3413, + "theoretical_loss": 3.7045065420274748, + "tokens_seen": 855703552 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037412236710130395, + "loss": 1.4128, + "theoretical_loss": 3.7044787597714137, + "tokens_seen": 855769088 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003741123370110331, + "loss": 1.3313, + "theoretical_loss": 3.704450980238553, + "tokens_seen": 855834624 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003741023069207623, + "loss": 1.5033, + "theoretical_loss": 3.7044232034284166, + "tokens_seen": 855900160 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037409227683049144, + "loss": 1.3933, + "theoretical_loss": 3.7043954293405292, + "tokens_seen": 855965696 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003740822467402207, + "loss": 1.4279, + "theoretical_loss": 3.7043676579744163, + "tokens_seen": 856031232 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037407221664994986, + "loss": 1.289, + "theoretical_loss": 3.704339889329602, + "tokens_seen": 856096768 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037406218655967904, + "loss": 1.4415, + "theoretical_loss": 3.704312123405612, + "tokens_seen": 856162304 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003740521564694082, + "loss": 1.4109, + "theoretical_loss": 3.704284360201971, + "tokens_seen": 856227840 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003740421263791374, + "loss": 1.4451, + "theoretical_loss": 3.704256599718205, + "tokens_seen": 856293376 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003740320962888666, + "loss": 1.4396, + "theoretical_loss": 3.704228841953838, + "tokens_seen": 856358912 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003740220661985958, + "loss": 1.4779, + "theoretical_loss": 3.704201086908397, + "tokens_seen": 856424448 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037401203610832494, + "loss": 1.4347, + "theoretical_loss": 3.704173334581407, + "tokens_seen": 856489984 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003740020060180542, + "loss": 1.4062, + "theoretical_loss": 3.7041455849723937, + "tokens_seen": 856555520 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037399197592778336, + "loss": 1.3196, + "theoretical_loss": 3.704117838080883, + "tokens_seen": 856621056 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037398194583751254, + "loss": 1.4387, + "theoretical_loss": 3.7040900939064008, + "tokens_seen": 856686592 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003739719157472417, + "loss": 1.5705, + "theoretical_loss": 3.704062352448474, + "tokens_seen": 856752128 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003739618856569709, + "loss": 1.5525, + "theoretical_loss": 3.704034613706628, + "tokens_seen": 856817664 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": -0.0013087779516354203, + "objective/train/docs_used": 492132, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.1923091411590576, + "objective/train/original_loss": 2.1923093795776367, + "objective/train/theoretical_loss": 3.7040068776803894, + "objective/train/tokens_used": 877343200, + "objective/train/value_avg": -0.00543212890625, + "objective/train/value_loss": 6.382058199960738e-05, + "objective/train/value_max": -0.0004673004150390625, + "objective/train/value_min": -0.060638427734375, + "objective/train/value_reward_corr": 0.2835776662073643, + "objective/train/value_std": 0.00431060791015625, + "objective/train/weight_avg": 0.9899634122848511, + "objective/train/weighted_lm_loss": 2.1806116104125977, + "objective/train/weights_max": 1.8337888717651367, + "objective/train/weights_min": 0.4513409435749054, + "theoretical_loss": 3.7040068776803894, + "tokens_seen": 856883200 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003739518555667001, + "loss": 1.3344, + "theoretical_loss": 3.7040068776803894, + "tokens_seen": 856883200 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003739418254764293, + "loss": 1.4382, + "theoretical_loss": 3.7039791443692844, + "tokens_seen": 856948736 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037393179538615845, + "loss": 1.4527, + "theoretical_loss": 3.703951413772841, + "tokens_seen": 857014272 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003739217652958877, + "loss": 1.4255, + "theoretical_loss": 3.7039236858905835, + "tokens_seen": 857079808 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003739117352056168, + "loss": 1.3762, + "theoretical_loss": 3.7038959607220407, + "tokens_seen": 857145344 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037390170511534605, + "loss": 1.526, + "theoretical_loss": 3.7038682382667387, + "tokens_seen": 857210880 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003738916750250752, + "loss": 1.3977, + "theoretical_loss": 3.7038405185242054, + "tokens_seen": 857276416 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003738816449348044, + "loss": 1.4855, + "theoretical_loss": 3.7038128014939664, + "tokens_seen": 857341952 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037387161484453364, + "loss": 1.4877, + "theoretical_loss": 3.703785087175551, + "tokens_seen": 857407488 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037386158475426277, + "loss": 1.4462, + "theoretical_loss": 3.703757375568485, + "tokens_seen": 857473024 + }, + { + "epoch": 0.26, + "learning_rate": 0.000373851554663992, + "loss": 1.4323, + "theoretical_loss": 3.703729666672297, + "tokens_seen": 857538560 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003738415245737212, + "loss": 1.4363, + "theoretical_loss": 3.703701960486514, + "tokens_seen": 857604096 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037383149448345037, + "loss": 1.5189, + "theoretical_loss": 3.7036742570106638, + "tokens_seen": 857669632 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037382146439317955, + "loss": 1.4462, + "theoretical_loss": 3.7036465562442746, + "tokens_seen": 857735168 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003738114343029088, + "loss": 1.4031, + "theoretical_loss": 3.703618858186875, + "tokens_seen": 857800704 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003738014042126379, + "loss": 1.5188, + "theoretical_loss": 3.7035911628379914, + "tokens_seen": 857866240 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037379137412236715, + "loss": 1.5775, + "theoretical_loss": 3.703563470197154, + "tokens_seen": 857931776 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037378134403209627, + "loss": 1.4224, + "theoretical_loss": 3.7035357802638895, + "tokens_seen": 857997312 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003737713139418255, + "loss": 1.4367, + "theoretical_loss": 3.703508093037728, + "tokens_seen": 858062848 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003737612838515547, + "loss": 1.4407, + "theoretical_loss": 3.703480408518197, + "tokens_seen": 858128384 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037375125376128387, + "loss": 1.3523, + "theoretical_loss": 3.7034527267048256, + "tokens_seen": 858193920 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037374122367101305, + "loss": 1.4845, + "theoretical_loss": 3.7034250475971424, + "tokens_seen": 858259456 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037373119358074223, + "loss": 1.2758, + "theoretical_loss": 3.7033973711946766, + "tokens_seen": 858324992 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003737211634904714, + "loss": 1.4546, + "theoretical_loss": 3.7033696974969574, + "tokens_seen": 858390528 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037371113340020065, + "loss": 1.4016, + "theoretical_loss": 3.703342026503514, + "tokens_seen": 858456064 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": 0.0005319549236446619, + "objective/train/docs_used": 493247, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6858315467834473, + "objective/train/original_loss": 2.685831069946289, + "objective/train/theoretical_loss": 3.7033143582138752, + "objective/train/tokens_used": 878981600, + "objective/train/value_avg": -0.00701904296875, + "objective/train/value_loss": 0.00010567898425506428, + "objective/train/value_max": -0.00032258033752441406, + "objective/train/value_min": -0.08404541015625, + "objective/train/value_reward_corr": 0.46132115860744033, + "objective/train/value_std": 0.006938934326171875, + "objective/train/weight_avg": 1.0102251768112183, + "objective/train/weighted_lm_loss": 2.7104177474975586, + "objective/train/weights_max": 2.139247417449951, + "objective/train/weights_min": 0.2922336161136627, + "theoretical_loss": 3.7033143582138752, + "tokens_seen": 858521600 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003737011033099298, + "loss": 1.3121, + "theoretical_loss": 3.7033143582138752, + "tokens_seen": 858521600 + }, + { + "epoch": 0.26, + "learning_rate": 0.000373691073219659, + "loss": 1.4933, + "theoretical_loss": 3.7032866926275707, + "tokens_seen": 858587136 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037368104312938814, + "loss": 1.4048, + "theoretical_loss": 3.703259029744131, + "tokens_seen": 858652672 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003736710130391174, + "loss": 1.4064, + "theoretical_loss": 3.703231369563084, + "tokens_seen": 858718208 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037366098294884655, + "loss": 1.5436, + "theoretical_loss": 3.703203712083961, + "tokens_seen": 858783744 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037365095285857574, + "loss": 1.3502, + "theoretical_loss": 3.703176057306291, + "tokens_seen": 858849280 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003736409227683049, + "loss": 1.5145, + "theoretical_loss": 3.703148405229604, + "tokens_seen": 858914816 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037363089267803415, + "loss": 1.4284, + "theoretical_loss": 3.7031207558534316, + "tokens_seen": 858980352 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003736208625877633, + "loss": 1.4383, + "theoretical_loss": 3.703093109177302, + "tokens_seen": 859045888 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003736108324974925, + "loss": 1.4101, + "theoretical_loss": 3.703065465200747, + "tokens_seen": 859111424 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037360080240722164, + "loss": 1.325, + "theoretical_loss": 3.703037823923296, + "tokens_seen": 859176960 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003735907723169509, + "loss": 1.4249, + "theoretical_loss": 3.703010185344481, + "tokens_seen": 859242496 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037358074222668006, + "loss": 1.4705, + "theoretical_loss": 3.7029825494638313, + "tokens_seen": 859308032 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037357071213640924, + "loss": 1.444, + "theoretical_loss": 3.702954916280879, + "tokens_seen": 859373568 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003735606820461384, + "loss": 1.4411, + "theoretical_loss": 3.7029272857951545, + "tokens_seen": 859439104 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003735506519558676, + "loss": 1.5088, + "theoretical_loss": 3.7028996580061886, + "tokens_seen": 859504640 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003735406218655968, + "loss": 1.4587, + "theoretical_loss": 3.7028720329135125, + "tokens_seen": 859570176 + }, + { + "epoch": 0.26, + "learning_rate": 0.000373530591775326, + "loss": 1.4563, + "theoretical_loss": 3.702844410516658, + "tokens_seen": 859635712 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037352056168505514, + "loss": 1.3995, + "theoretical_loss": 3.702816790815157, + "tokens_seen": 859701248 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003735105315947844, + "loss": 1.3281, + "theoretical_loss": 3.7027891738085392, + "tokens_seen": 859766784 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037350050150451356, + "loss": 1.38, + "theoretical_loss": 3.702761559496338, + "tokens_seen": 859832320 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037349047141424274, + "loss": 1.4427, + "theoretical_loss": 3.7027339478780847, + "tokens_seen": 859897856 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003734804413239719, + "loss": 1.3018, + "theoretical_loss": 3.7027063389533112, + "tokens_seen": 859963392 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003734704112337011, + "loss": 1.402, + "theoretical_loss": 3.7026787327215493, + "tokens_seen": 860028928 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003734603811434303, + "loss": 1.407, + "theoretical_loss": 3.7026511291823314, + "tokens_seen": 860094464 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": -0.00028626201674342155, + "objective/train/docs_used": 493898, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7297286987304688, + "objective/train/original_loss": 2.7297282218933105, + "objective/train/theoretical_loss": 3.70262352833519, + "objective/train/tokens_used": 880620000, + "objective/train/value_avg": -0.006061553955078125, + "objective/train/value_loss": 5.553428127313964e-05, + "objective/train/value_max": -0.00047278404235839844, + "objective/train/value_min": -0.11163330078125, + "objective/train/value_reward_corr": 0.37114077289306724, + "objective/train/value_std": 0.0050811767578125, + "objective/train/weight_avg": 0.9997788667678833, + "objective/train/weighted_lm_loss": 2.722656726837158, + "objective/train/weights_max": 2.748540163040161, + "objective/train/weights_min": 0.19562561810016632, + "theoretical_loss": 3.70262352833519, + "tokens_seen": 860160000 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003734503510531595, + "loss": 1.4372, + "theoretical_loss": 3.70262352833519, + "tokens_seen": 860160000 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037344032096288865, + "loss": 1.5659, + "theoretical_loss": 3.7025959301796565, + "tokens_seen": 860225536 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003734302908726179, + "loss": 1.4038, + "theoretical_loss": 3.702568334715264, + "tokens_seen": 860291072 + }, + { + "epoch": 0.26, + "learning_rate": 0.000373420260782347, + "loss": 1.4244, + "theoretical_loss": 3.7025407419415455, + "tokens_seen": 860356608 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037341023069207625, + "loss": 1.4536, + "theoretical_loss": 3.7025131518580325, + "tokens_seen": 860422144 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003734002006018054, + "loss": 1.2814, + "theoretical_loss": 3.7024855644642596, + "tokens_seen": 860487680 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003733901705115346, + "loss": 1.3569, + "theoretical_loss": 3.7024579797597585, + "tokens_seen": 860553216 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003733801404212638, + "loss": 1.4397, + "theoretical_loss": 3.7024303977440627, + "tokens_seen": 860618752 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037337011033099297, + "loss": 1.4545, + "theoretical_loss": 3.7024028184167053, + "tokens_seen": 860684288 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037336008024072215, + "loss": 1.4931, + "theoretical_loss": 3.7023752417772195, + "tokens_seen": 860749824 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003733500501504514, + "loss": 1.4154, + "theoretical_loss": 3.702347667825139, + "tokens_seen": 860815360 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003733400200601805, + "loss": 1.4896, + "theoretical_loss": 3.7023200965599967, + "tokens_seen": 860880896 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037332998996990975, + "loss": 1.4038, + "theoretical_loss": 3.7022925279813266, + "tokens_seen": 860946432 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037331995987963893, + "loss": 1.4384, + "theoretical_loss": 3.702264962088663, + "tokens_seen": 861011968 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003733099297893681, + "loss": 1.5325, + "theoretical_loss": 3.702237398881539, + "tokens_seen": 861077504 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003732998996990973, + "loss": 1.3872, + "theoretical_loss": 3.7022098383594892, + "tokens_seen": 861143040 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003732898696088265, + "loss": 1.5122, + "theoretical_loss": 3.702182280522048, + "tokens_seen": 861208576 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037327983951855565, + "loss": 1.4916, + "theoretical_loss": 3.7021547253687483, + "tokens_seen": 861274112 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003732698094282849, + "loss": 1.4533, + "theoretical_loss": 3.702127172899125, + "tokens_seen": 861339648 + }, + { + "epoch": 0.26, + "learning_rate": 0.000373259779338014, + "loss": 1.4948, + "theoretical_loss": 3.702099623112714, + "tokens_seen": 861405184 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037324974924774325, + "loss": 1.4099, + "theoretical_loss": 3.7020720760090478, + "tokens_seen": 861470720 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003732397191574724, + "loss": 1.4598, + "theoretical_loss": 3.7020445315876622, + "tokens_seen": 861536256 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003732296890672016, + "loss": 1.3557, + "theoretical_loss": 3.702016989848092, + "tokens_seen": 861601792 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003732196589769308, + "loss": 1.5, + "theoretical_loss": 3.701989450789872, + "tokens_seen": 861667328 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037320962888666, + "loss": 1.3336, + "theoretical_loss": 3.701961914412537, + "tokens_seen": 861732864 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": 0.00037320886622183025, + "objective/train/docs_used": 495406, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.729614019393921, + "objective/train/original_loss": 2.7296149730682373, + "objective/train/theoretical_loss": 3.701934380715622, + "objective/train/tokens_used": 882258400, + "objective/train/value_avg": -0.00539398193359375, + "objective/train/value_loss": 6.13092488492839e-05, + "objective/train/value_max": -0.00031256675720214844, + "objective/train/value_min": -0.093505859375, + "objective/train/value_reward_corr": 0.40326351263055016, + "objective/train/value_std": 0.0049896240234375, + "objective/train/weight_avg": 1.0066372156143188, + "objective/train/weighted_lm_loss": 2.74674129486084, + "objective/train/weights_max": 2.3220860958099365, + "objective/train/weights_min": 0.4482279419898987, + "theoretical_loss": 3.701934380715622, + "tokens_seen": 861798400 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037319959879638916, + "loss": 1.4493, + "theoretical_loss": 3.701934380715622, + "tokens_seen": 861798400 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037318956870611834, + "loss": 1.3985, + "theoretical_loss": 3.7019068496986636, + "tokens_seen": 861863936 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003731795386158475, + "loss": 1.4412, + "theoretical_loss": 3.7018793213611954, + "tokens_seen": 861929472 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037316950852557675, + "loss": 1.4689, + "theoretical_loss": 3.7018517957027544, + "tokens_seen": 861995008 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003731594784353059, + "loss": 1.333, + "theoretical_loss": 3.7018242727228756, + "tokens_seen": 862060544 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003731494483450351, + "loss": 1.4519, + "theoretical_loss": 3.701796752421094, + "tokens_seen": 862126080 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003731394182547643, + "loss": 1.4074, + "theoretical_loss": 3.701769234796947, + "tokens_seen": 862191616 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003731293881644935, + "loss": 1.4557, + "theoretical_loss": 3.7017417198499696, + "tokens_seen": 862257152 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003731193580742227, + "loss": 1.4036, + "theoretical_loss": 3.701714207579698, + "tokens_seen": 862322688 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037310932798395184, + "loss": 1.4144, + "theoretical_loss": 3.701686697985668, + "tokens_seen": 862388224 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003730992978936811, + "loss": 1.3882, + "theoretical_loss": 3.7016591910674173, + "tokens_seen": 862453760 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037308926780341026, + "loss": 1.4866, + "theoretical_loss": 3.7016316868244807, + "tokens_seen": 862519296 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037307923771313944, + "loss": 1.4821, + "theoretical_loss": 3.7016041852563957, + "tokens_seen": 862584832 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003730692076228686, + "loss": 1.3485, + "theoretical_loss": 3.701576686362699, + "tokens_seen": 862650368 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003730591775325978, + "loss": 1.382, + "theoretical_loss": 3.7015491901429263, + "tokens_seen": 862715904 + }, + { + "epoch": 0.26, + "learning_rate": 0.000373049147442327, + "loss": 1.3528, + "theoretical_loss": 3.7015216965966165, + "tokens_seen": 862781440 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003730391173520562, + "loss": 1.4147, + "theoretical_loss": 3.7014942057233045, + "tokens_seen": 862846976 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037302908726178534, + "loss": 1.5354, + "theoretical_loss": 3.701466717522529, + "tokens_seen": 862912512 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003730190571715146, + "loss": 1.4163, + "theoretical_loss": 3.7014392319938265, + "tokens_seen": 862978048 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037300902708124376, + "loss": 1.4713, + "theoretical_loss": 3.701411749136734, + "tokens_seen": 863043584 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037299899699097294, + "loss": 1.556, + "theoretical_loss": 3.7013842689507896, + "tokens_seen": 863109120 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003729889669007021, + "loss": 1.4932, + "theoretical_loss": 3.701356791435532, + "tokens_seen": 863174656 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003729789368104313, + "loss": 1.391, + "theoretical_loss": 3.7013293165904964, + "tokens_seen": 863240192 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003729689067201605, + "loss": 1.3591, + "theoretical_loss": 3.701301844415222, + "tokens_seen": 863305728 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003729588766298897, + "loss": 1.4466, + "theoretical_loss": 3.7012743749092465, + "tokens_seen": 863371264 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": -0.0007112006423994899, + "objective/train/docs_used": 496136, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0594470500946045, + "objective/train/original_loss": 3.0594470500946045, + "objective/train/theoretical_loss": 3.7012469080721084, + "objective/train/tokens_used": 883896800, + "objective/train/value_avg": -0.005733489990234375, + "objective/train/value_loss": 8.58526982483454e-05, + "objective/train/value_max": -0.0004546642303466797, + "objective/train/value_min": -0.10504150390625, + "objective/train/value_reward_corr": 0.557035074837396, + "objective/train/value_std": 0.00591278076171875, + "objective/train/weight_avg": 0.9968644976615906, + "objective/train/weighted_lm_loss": 3.053791046142578, + "objective/train/weights_max": 1.7006419897079468, + "objective/train/weights_min": 0.49728965759277344, + "theoretical_loss": 3.7012469080721084, + "tokens_seen": 863436800 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037294884653961885, + "loss": 1.4795, + "theoretical_loss": 3.7012469080721084, + "tokens_seen": 863436800 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003729388164493481, + "loss": 1.4132, + "theoretical_loss": 3.7012194439033457, + "tokens_seen": 863502336 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003729287863590772, + "loss": 1.465, + "theoretical_loss": 3.7011919824024964, + "tokens_seen": 863567872 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037291875626880645, + "loss": 1.4415, + "theoretical_loss": 3.7011645235690986, + "tokens_seen": 863633408 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003729087261785356, + "loss": 1.4018, + "theoretical_loss": 3.701137067402692, + "tokens_seen": 863698944 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003728986960882648, + "loss": 1.4714, + "theoretical_loss": 3.7011096139028137, + "tokens_seen": 863764480 + }, + { + "epoch": 0.26, + "learning_rate": 0.000372888665997994, + "loss": 1.3331, + "theoretical_loss": 3.701082163069004, + "tokens_seen": 863830016 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037287863590772317, + "loss": 1.4181, + "theoretical_loss": 3.7010547149008, + "tokens_seen": 863895552 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037286860581745235, + "loss": 1.3714, + "theoretical_loss": 3.701027269397742, + "tokens_seen": 863961088 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003728585757271816, + "loss": 1.344, + "theoretical_loss": 3.700999826559369, + "tokens_seen": 864026624 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003728485456369107, + "loss": 1.3959, + "theoretical_loss": 3.70097238638522, + "tokens_seen": 864092160 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037283851554663995, + "loss": 1.3723, + "theoretical_loss": 3.700944948874834, + "tokens_seen": 864157696 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037282848545636913, + "loss": 1.398, + "theoretical_loss": 3.7009175140277506, + "tokens_seen": 864223232 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003728184553660983, + "loss": 1.4169, + "theoretical_loss": 3.700890081843509, + "tokens_seen": 864288768 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003728084252758275, + "loss": 1.3575, + "theoretical_loss": 3.70086265232165, + "tokens_seen": 864354304 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003727983951855567, + "loss": 1.5025, + "theoretical_loss": 3.700835225461712, + "tokens_seen": 864419840 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037278836509528585, + "loss": 1.501, + "theoretical_loss": 3.7008078012632364, + "tokens_seen": 864485376 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003727783350050151, + "loss": 1.329, + "theoretical_loss": 3.700780379725761, + "tokens_seen": 864550912 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003727683049147442, + "loss": 1.3654, + "theoretical_loss": 3.7007529608488285, + "tokens_seen": 864616448 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037275827482447345, + "loss": 1.5273, + "theoretical_loss": 3.700725544631977, + "tokens_seen": 864681984 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003727482447342026, + "loss": 1.3967, + "theoretical_loss": 3.7006981310747484, + "tokens_seen": 864747520 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003727382146439318, + "loss": 1.3952, + "theoretical_loss": 3.700670720176682, + "tokens_seen": 864813056 + }, + { + "epoch": 0.26, + "learning_rate": 0.000372728184553661, + "loss": 1.3752, + "theoretical_loss": 3.700643311937319, + "tokens_seen": 864878592 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003727181544633902, + "loss": 1.4864, + "theoretical_loss": 3.7006159063562, + "tokens_seen": 864944128 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037270812437311936, + "loss": 1.6076, + "theoretical_loss": 3.700588503432866, + "tokens_seen": 865009664 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": 0.0004241900460328907, + "objective/train/docs_used": 497267, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9942450523376465, + "objective/train/original_loss": 2.9942450523376465, + "objective/train/theoretical_loss": 3.700561103166857, + "objective/train/tokens_used": 885535200, + "objective/train/value_avg": -0.006134033203125, + "objective/train/value_loss": 4.8231064283754677e-05, + "objective/train/value_max": -0.00026726722717285156, + "objective/train/value_min": -0.06744384765625, + "objective/train/value_reward_corr": 0.4159454477132716, + "objective/train/value_std": 0.004711151123046875, + "objective/train/weight_avg": 1.0065975189208984, + "objective/train/weighted_lm_loss": 3.0107460021972656, + "objective/train/weights_max": 1.6668528318405151, + "objective/train/weights_min": 0.5188010334968567, + "theoretical_loss": 3.700561103166857, + "tokens_seen": 865075200 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037269809428284854, + "loss": 1.4784, + "theoretical_loss": 3.700561103166857, + "tokens_seen": 865075200 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003726880641925777, + "loss": 1.3641, + "theoretical_loss": 3.7005337055577145, + "tokens_seen": 865140736 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037267803410230696, + "loss": 1.3861, + "theoretical_loss": 3.7005063106049807, + "tokens_seen": 865206272 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003726680040120361, + "loss": 1.4267, + "theoretical_loss": 3.700478918308195, + "tokens_seen": 865271808 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003726579739217653, + "loss": 1.4078, + "theoretical_loss": 3.7004515286669006, + "tokens_seen": 865337344 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003726479438314945, + "loss": 1.4546, + "theoretical_loss": 3.700424141680638, + "tokens_seen": 865402880 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003726379137412237, + "loss": 1.4431, + "theoretical_loss": 3.7003967573489485, + "tokens_seen": 865468416 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037262788365095286, + "loss": 1.4758, + "theoretical_loss": 3.7003693756713747, + "tokens_seen": 865533952 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037261785356068204, + "loss": 1.2598, + "theoretical_loss": 3.7003419966474578, + "tokens_seen": 865599488 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003726078234704112, + "loss": 1.4369, + "theoretical_loss": 3.70031462027674, + "tokens_seen": 865665024 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037259779338014046, + "loss": 1.3501, + "theoretical_loss": 3.700287246558763, + "tokens_seen": 865730560 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003725877632898696, + "loss": 1.4346, + "theoretical_loss": 3.700259875493069, + "tokens_seen": 865796096 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003725777331995988, + "loss": 1.3827, + "theoretical_loss": 3.7002325070792015, + "tokens_seen": 865861632 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037256770310932795, + "loss": 1.4499, + "theoretical_loss": 3.700205141316701, + "tokens_seen": 865927168 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003725576730190572, + "loss": 1.3906, + "theoretical_loss": 3.700177778205111, + "tokens_seen": 865992704 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037254764292878636, + "loss": 1.3476, + "theoretical_loss": 3.7001504177439744, + "tokens_seen": 866058240 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037253761283851554, + "loss": 1.4921, + "theoretical_loss": 3.7001230599328334, + "tokens_seen": 866123776 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003725275827482447, + "loss": 1.4851, + "theoretical_loss": 3.7000957047712317, + "tokens_seen": 866189312 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037251755265797396, + "loss": 1.3512, + "theoretical_loss": 3.700068352258711, + "tokens_seen": 866254848 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003725075225677031, + "loss": 1.413, + "theoretical_loss": 3.7000410023948147, + "tokens_seen": 866320384 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003724974924774323, + "loss": 1.4651, + "theoretical_loss": 3.7000136551790868, + "tokens_seen": 866385920 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037248746238716145, + "loss": 1.3426, + "theoretical_loss": 3.6999863106110698, + "tokens_seen": 866451456 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003724774322968907, + "loss": 1.3825, + "theoretical_loss": 3.699958968690307, + "tokens_seen": 866516992 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037246740220661987, + "loss": 1.317, + "theoretical_loss": 3.6999316294163433, + "tokens_seen": 866582528 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037245737211634905, + "loss": 1.3588, + "theoretical_loss": 3.6999042927887205, + "tokens_seen": 866648064 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": -0.00024049580679275095, + "objective/train/docs_used": 497986, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.56120228767395, + "objective/train/original_loss": 2.56120228767395, + "objective/train/theoretical_loss": 3.6998769588069838, + "objective/train/tokens_used": 887173600, + "objective/train/value_avg": -0.00466156005859375, + "objective/train/value_loss": 4.9177346227224916e-05, + "objective/train/value_max": -0.0002694129943847656, + "objective/train/value_min": -0.050048828125, + "objective/train/value_reward_corr": 0.3363071616312242, + "objective/train/value_std": 0.0035419464111328125, + "objective/train/weight_avg": 0.9998930096626282, + "objective/train/weighted_lm_loss": 2.567042589187622, + "objective/train/weights_max": 1.4830387830734253, + "objective/train/weights_min": 0.5110052227973938, + "theoretical_loss": 3.6998769588069838, + "tokens_seen": 866713600 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037244734202607823, + "loss": 1.4904, + "theoretical_loss": 3.6998769588069838, + "tokens_seen": 866713600 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003724373119358074, + "loss": 1.3256, + "theoretical_loss": 3.699849627470676, + "tokens_seen": 866779136 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003724272818455366, + "loss": 1.3672, + "theoretical_loss": 3.699822298779342, + "tokens_seen": 866844672 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003724172517552658, + "loss": 1.405, + "theoretical_loss": 3.699794972732525, + "tokens_seen": 866910208 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037240722166499495, + "loss": 1.3961, + "theoretical_loss": 3.69976764932977, + "tokens_seen": 866975744 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003723971915747242, + "loss": 1.4396, + "theoretical_loss": 3.6997403285706207, + "tokens_seen": 867041280 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003723871614844533, + "loss": 1.3671, + "theoretical_loss": 3.699713010454622, + "tokens_seen": 867106816 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037237713139418255, + "loss": 1.5087, + "theoretical_loss": 3.6996856949813184, + "tokens_seen": 867172352 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003723671013039118, + "loss": 1.5436, + "theoretical_loss": 3.699658382150254, + "tokens_seen": 867237888 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003723570712136409, + "loss": 1.4142, + "theoretical_loss": 3.699631071960974, + "tokens_seen": 867303424 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037234704112337015, + "loss": 1.5185, + "theoretical_loss": 3.699603764413024, + "tokens_seen": 867368960 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037233701103309933, + "loss": 1.3862, + "theoretical_loss": 3.6995764595059475, + "tokens_seen": 867434496 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003723269809428285, + "loss": 1.4834, + "theoretical_loss": 3.699549157239291, + "tokens_seen": 867500032 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003723169508525577, + "loss": 1.52, + "theoretical_loss": 3.699521857612599, + "tokens_seen": 867565568 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003723069207622869, + "loss": 1.382, + "theoretical_loss": 3.699494560625417, + "tokens_seen": 867631104 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037229689067201605, + "loss": 1.4487, + "theoretical_loss": 3.6994672662772903, + "tokens_seen": 867696640 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003722868605817453, + "loss": 1.3912, + "theoretical_loss": 3.699439974567764, + "tokens_seen": 867762176 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003722768304914744, + "loss": 1.4723, + "theoretical_loss": 3.6994126854963856, + "tokens_seen": 867827712 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037226680040120365, + "loss": 1.3061, + "theoretical_loss": 3.699385399062699, + "tokens_seen": 867893248 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003722567703109328, + "loss": 1.4296, + "theoretical_loss": 3.69935811526625, + "tokens_seen": 867958784 + }, + { + "epoch": 0.26, + "learning_rate": 0.000372246740220662, + "loss": 1.4337, + "theoretical_loss": 3.6993308341065863, + "tokens_seen": 868024320 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003722367101303912, + "loss": 1.447, + "theoretical_loss": 3.6993035555832527, + "tokens_seen": 868089856 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003722266800401204, + "loss": 1.4241, + "theoretical_loss": 3.699276279695796, + "tokens_seen": 868155392 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037221664994984956, + "loss": 1.4665, + "theoretical_loss": 3.6992490064437624, + "tokens_seen": 868220928 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037220661985957874, + "loss": 1.4028, + "theoretical_loss": 3.6992217358266983, + "tokens_seen": 868286464 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": -0.0008690840913914144, + "objective/train/docs_used": 499247, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.939300060272217, + "objective/train/original_loss": 2.939300060272217, + "objective/train/theoretical_loss": 3.6991944678441504, + "objective/train/tokens_used": 888812000, + "objective/train/value_avg": -0.005344390869140625, + "objective/train/value_loss": 8.469414751743898e-05, + "objective/train/value_max": -0.00030541419982910156, + "objective/train/value_min": -0.0794677734375, + "objective/train/value_reward_corr": 0.3591380411381957, + "objective/train/value_std": 0.003948211669921875, + "objective/train/weight_avg": 0.9952071309089661, + "objective/train/weighted_lm_loss": 2.927234172821045, + "objective/train/weights_max": 1.787829875946045, + "objective/train/weights_min": 0.30267173051834106, + "theoretical_loss": 3.6991944678441504, + "tokens_seen": 868352000 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003721965897693079, + "loss": 1.4481, + "theoretical_loss": 3.6991944678441504, + "tokens_seen": 868352000 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037218655967903716, + "loss": 1.4436, + "theoretical_loss": 3.6991672024956648, + "tokens_seen": 868417536 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003721765295887663, + "loss": 1.3784, + "theoretical_loss": 3.699139939780789, + "tokens_seen": 868483072 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003721664994984955, + "loss": 1.2711, + "theoretical_loss": 3.69911267969907, + "tokens_seen": 868548608 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003721564694082247, + "loss": 1.4395, + "theoretical_loss": 3.699085422250054, + "tokens_seen": 868614144 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003721464393179539, + "loss": 1.4378, + "theoretical_loss": 3.6990581674332894, + "tokens_seen": 868679680 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037213640922768306, + "loss": 1.4546, + "theoretical_loss": 3.699030915248322, + "tokens_seen": 868745216 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037212637913741224, + "loss": 1.4579, + "theoretical_loss": 3.6990036656947005, + "tokens_seen": 868810752 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003721163490471414, + "loss": 1.452, + "theoretical_loss": 3.698976418771971, + "tokens_seen": 868876288 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037210631895687066, + "loss": 1.35, + "theoretical_loss": 3.698949174479682, + "tokens_seen": 868941824 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003720962888665998, + "loss": 1.4608, + "theoretical_loss": 3.698921932817381, + "tokens_seen": 869007360 + }, + { + "epoch": 0.26, + "learning_rate": 0.000372086258776329, + "loss": 1.3737, + "theoretical_loss": 3.6988946937846165, + "tokens_seen": 869072896 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037207622868605815, + "loss": 1.415, + "theoretical_loss": 3.698867457380935, + "tokens_seen": 869138432 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003720661985957874, + "loss": 1.1824, + "theoretical_loss": 3.6988402236058855, + "tokens_seen": 869203968 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037205616850551656, + "loss": 1.4927, + "theoretical_loss": 3.6988129924590156, + "tokens_seen": 869269504 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037204613841524575, + "loss": 1.3609, + "theoretical_loss": 3.698785763939874, + "tokens_seen": 869335040 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003720361083249749, + "loss": 1.4141, + "theoretical_loss": 3.698758538048009, + "tokens_seen": 869400576 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037202607823470416, + "loss": 1.4157, + "theoretical_loss": 3.698731314782969, + "tokens_seen": 869466112 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003720160481444333, + "loss": 1.3543, + "theoretical_loss": 3.698704094144303, + "tokens_seen": 869531648 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003720060180541625, + "loss": 1.4114, + "theoretical_loss": 3.698676876131559, + "tokens_seen": 869597184 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037199598796389165, + "loss": 1.4321, + "theoretical_loss": 3.6986496607442856, + "tokens_seen": 869662720 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003719859578736209, + "loss": 1.3899, + "theoretical_loss": 3.698622447982033, + "tokens_seen": 869728256 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037197592778335007, + "loss": 1.5175, + "theoretical_loss": 3.6985952378443487, + "tokens_seen": 869793792 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037196589769307925, + "loss": 1.5049, + "theoretical_loss": 3.698568030330783, + "tokens_seen": 869859328 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037195586760280843, + "loss": 1.4831, + "theoretical_loss": 3.698540825440885, + "tokens_seen": 869924864 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": -0.00030114094261080027, + "objective/train/docs_used": 499921, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2538347244262695, + "objective/train/original_loss": 3.2538349628448486, + "objective/train/theoretical_loss": 3.6985136231742035, + "objective/train/tokens_used": 890450400, + "objective/train/value_avg": -0.005077362060546875, + "objective/train/value_loss": 4.309868018026464e-05, + "objective/train/value_max": -0.00051116943359375, + "objective/train/value_min": -0.04742431640625, + "objective/train/value_reward_corr": 0.2829643882402502, + "objective/train/value_std": 0.0035648345947265625, + "objective/train/weight_avg": 0.9990850687026978, + "objective/train/weighted_lm_loss": 3.2407801151275635, + "objective/train/weights_max": 1.4109147787094116, + "objective/train/weights_min": 0.5293207168579102, + "theoretical_loss": 3.6985136231742035, + "tokens_seen": 869990400 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003719458375125376, + "loss": 1.3911, + "theoretical_loss": 3.6985136231742035, + "tokens_seen": 869990400 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003719358074222668, + "loss": 1.4147, + "theoretical_loss": 3.6984864235302886, + "tokens_seen": 870055936 + }, + { + "epoch": 0.26, + "learning_rate": 0.000371925777331996, + "loss": 1.2222, + "theoretical_loss": 3.6984592265086893, + "tokens_seen": 870121472 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037191574724172515, + "loss": 1.5029, + "theoretical_loss": 3.698432032108956, + "tokens_seen": 870187008 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003719057171514544, + "loss": 1.4288, + "theoretical_loss": 3.6984048403306375, + "tokens_seen": 870252544 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003718956870611835, + "loss": 1.311, + "theoretical_loss": 3.698377651173285, + "tokens_seen": 870318080 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037188565697091275, + "loss": 1.4488, + "theoretical_loss": 3.698350464636447, + "tokens_seen": 870383616 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037187562688064193, + "loss": 1.5218, + "theoretical_loss": 3.6983232807196753, + "tokens_seen": 870449152 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003718655967903711, + "loss": 1.5484, + "theoretical_loss": 3.698296099422519, + "tokens_seen": 870514688 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003718555667001003, + "loss": 1.3929, + "theoretical_loss": 3.698268920744529, + "tokens_seen": 870580224 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037184553660982953, + "loss": 1.4972, + "theoretical_loss": 3.698241744685256, + "tokens_seen": 870645760 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037183550651955866, + "loss": 1.3955, + "theoretical_loss": 3.6982145712442502, + "tokens_seen": 870711296 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003718254764292879, + "loss": 1.438, + "theoretical_loss": 3.6981874004210615, + "tokens_seen": 870776832 + }, + { + "epoch": 0.26, + "learning_rate": 0.000371815446339017, + "loss": 1.381, + "theoretical_loss": 3.6981602322152423, + "tokens_seen": 870842368 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037180541624874625, + "loss": 1.3635, + "theoretical_loss": 3.698133066626343, + "tokens_seen": 870907904 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037179538615847544, + "loss": 1.4043, + "theoretical_loss": 3.6981059036539135, + "tokens_seen": 870973440 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003717853560682046, + "loss": 1.4435, + "theoretical_loss": 3.698078743297507, + "tokens_seen": 871038976 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003717753259779338, + "loss": 1.4552, + "theoretical_loss": 3.698051585556673, + "tokens_seen": 871104512 + }, + { + "epoch": 0.26, + "learning_rate": 0.000371765295887663, + "loss": 1.5185, + "theoretical_loss": 3.698024430430963, + "tokens_seen": 871170048 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037175526579739216, + "loss": 1.3502, + "theoretical_loss": 3.6979972779199297, + "tokens_seen": 871235584 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003717452357071214, + "loss": 1.4468, + "theoretical_loss": 3.6979701280231234, + "tokens_seen": 871301120 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003717352056168505, + "loss": 1.3768, + "theoretical_loss": 3.6979429807400965, + "tokens_seen": 871366656 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037172517552657976, + "loss": 1.2213, + "theoretical_loss": 3.697915836070401, + "tokens_seen": 871432192 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003717151454363089, + "loss": 1.5834, + "theoretical_loss": 3.697888694013588, + "tokens_seen": 871497728 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003717051153460381, + "loss": 1.4276, + "theoretical_loss": 3.6978615545692106, + "tokens_seen": 871563264 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": 0.00022990281286183745, + "objective/train/docs_used": 501092, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0684781074523926, + "objective/train/original_loss": 3.0684776306152344, + "objective/train/theoretical_loss": 3.69783441773682, + "objective/train/tokens_used": 892088800, + "objective/train/value_avg": -0.005340576171875, + "objective/train/value_loss": 6.304252747213468e-05, + "objective/train/value_max": -0.0003814697265625, + "objective/train/value_min": -0.058135986328125, + "objective/train/value_reward_corr": 0.3574374515977072, + "objective/train/value_std": 0.00426483154296875, + "objective/train/weight_avg": 1.005097508430481, + "objective/train/weighted_lm_loss": 3.0865254402160645, + "objective/train/weights_max": 1.6112169027328491, + "objective/train/weights_min": 0.1414206475019455, + "theoretical_loss": 3.69783441773682, + "tokens_seen": 871628800 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003716950852557673, + "loss": 1.4594, + "theoretical_loss": 3.69783441773682, + "tokens_seen": 871628800 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003716850551654965, + "loss": 1.4876, + "theoretical_loss": 3.697807283515969, + "tokens_seen": 871694336 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037167502507522566, + "loss": 1.4546, + "theoretical_loss": 3.6977801519062092, + "tokens_seen": 871759872 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003716649949849549, + "loss": 1.4574, + "theoretical_loss": 3.6977530229070945, + "tokens_seen": 871825408 + }, + { + "epoch": 0.26, + "learning_rate": 0.000371654964894684, + "loss": 1.4785, + "theoretical_loss": 3.6977258965181763, + "tokens_seen": 871890944 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037164493480441326, + "loss": 1.3845, + "theoretical_loss": 3.697698772739007, + "tokens_seen": 871956480 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003716349047141424, + "loss": 1.4038, + "theoretical_loss": 3.6976716515691406, + "tokens_seen": 872022016 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003716248746238716, + "loss": 1.4641, + "theoretical_loss": 3.6976445330081296, + "tokens_seen": 872087552 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037161484453360086, + "loss": 1.3755, + "theoretical_loss": 3.6976174170555263, + "tokens_seen": 872153088 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037160481444333, + "loss": 1.4147, + "theoretical_loss": 3.697590303710885, + "tokens_seen": 872218624 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003715947843530592, + "loss": 1.4504, + "theoretical_loss": 3.6975631929737576, + "tokens_seen": 872284160 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037158475426278835, + "loss": 1.4856, + "theoretical_loss": 3.6975360848436987, + "tokens_seen": 872349696 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003715747241725176, + "loss": 1.3573, + "theoretical_loss": 3.6975089793202613, + "tokens_seen": 872415232 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037156469408224676, + "loss": 1.3457, + "theoretical_loss": 3.697481876402999, + "tokens_seen": 872480768 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037155466399197595, + "loss": 1.4349, + "theoretical_loss": 3.6974547760914653, + "tokens_seen": 872546304 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003715446339017051, + "loss": 1.3645, + "theoretical_loss": 3.697427678385214, + "tokens_seen": 872611840 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037153460381143436, + "loss": 1.5018, + "theoretical_loss": 3.697400583283799, + "tokens_seen": 872677376 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003715245737211635, + "loss": 1.5262, + "theoretical_loss": 3.6973734907867746, + "tokens_seen": 872742912 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003715145436308927, + "loss": 1.36, + "theoretical_loss": 3.697346400893695, + "tokens_seen": 872808448 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037150451354062185, + "loss": 1.4207, + "theoretical_loss": 3.697319313604114, + "tokens_seen": 872873984 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003714944834503511, + "loss": 1.3695, + "theoretical_loss": 3.6972922289175862, + "tokens_seen": 872939520 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037148445336008027, + "loss": 1.3918, + "theoretical_loss": 3.697265146833666, + "tokens_seen": 873005056 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037147442326980945, + "loss": 1.3905, + "theoretical_loss": 3.6972380673519076, + "tokens_seen": 873070592 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037146439317953863, + "loss": 1.5442, + "theoretical_loss": 3.6972109904718664, + "tokens_seen": 873136128 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003714543630892678, + "loss": 1.3972, + "theoretical_loss": 3.6971839161930964, + "tokens_seen": 873201664 + }, + { + "epoch": 0.26, + "objective/train/advantage_avg": 0.00023514853091910481, + "objective/train/docs_used": 501627, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3969969749450684, + "objective/train/original_loss": 3.396996259689331, + "objective/train/theoretical_loss": 3.6971568445151535, + "objective/train/tokens_used": 893727200, + "objective/train/value_avg": -0.00537109375, + "objective/train/value_loss": 5.911157495575026e-05, + "objective/train/value_max": -0.0004458427429199219, + "objective/train/value_min": -0.060302734375, + "objective/train/value_reward_corr": 0.3667503605223568, + "objective/train/value_std": 0.004062652587890625, + "objective/train/weight_avg": 1.0049047470092773, + "objective/train/weighted_lm_loss": 3.4155068397521973, + "objective/train/weights_max": 1.4788486957550049, + "objective/train/weights_min": 0.13900046050548553, + "theoretical_loss": 3.6971568445151535, + "tokens_seen": 873267200 + }, + { + "epoch": 0.26, + "learning_rate": 0.000371444332998997, + "loss": 1.4713, + "theoretical_loss": 3.6971568445151535, + "tokens_seen": 873267200 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037143430290872623, + "loss": 1.4416, + "theoretical_loss": 3.697129775437592, + "tokens_seen": 873332736 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037142427281845535, + "loss": 1.424, + "theoretical_loss": 3.697102708959967, + "tokens_seen": 873398272 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003714142427281846, + "loss": 1.4102, + "theoretical_loss": 3.697075645081833, + "tokens_seen": 873463808 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003714042126379137, + "loss": 1.4003, + "theoretical_loss": 3.697048583802747, + "tokens_seen": 873529344 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037139418254764295, + "loss": 1.493, + "theoretical_loss": 3.697021525122264, + "tokens_seen": 873594880 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037138415245737213, + "loss": 1.3772, + "theoretical_loss": 3.6969944690399386, + "tokens_seen": 873660416 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003713741223671013, + "loss": 1.3827, + "theoretical_loss": 3.696967415555327, + "tokens_seen": 873725952 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003713640922768305, + "loss": 1.5813, + "theoretical_loss": 3.6969403646679853, + "tokens_seen": 873791488 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037135406218655973, + "loss": 1.4233, + "theoretical_loss": 3.696913316377469, + "tokens_seen": 873857024 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037134403209628886, + "loss": 1.4178, + "theoretical_loss": 3.6968862706833336, + "tokens_seen": 873922560 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003713340020060181, + "loss": 1.4655, + "theoretical_loss": 3.6968592275851364, + "tokens_seen": 873988096 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003713239719157472, + "loss": 1.4051, + "theoretical_loss": 3.6968321870824328, + "tokens_seen": 874053632 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037131394182547645, + "loss": 1.5096, + "theoretical_loss": 3.696805149174779, + "tokens_seen": 874119168 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037130391173520564, + "loss": 1.3772, + "theoretical_loss": 3.6967781138617317, + "tokens_seen": 874184704 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003712938816449348, + "loss": 1.3257, + "theoretical_loss": 3.696751081142848, + "tokens_seen": 874250240 + }, + { + "epoch": 0.26, + "learning_rate": 0.000371283851554664, + "loss": 1.4083, + "theoretical_loss": 3.696724051017683, + "tokens_seen": 874315776 + }, + { + "epoch": 0.26, + "learning_rate": 0.0003712738214643932, + "loss": 1.3222, + "theoretical_loss": 3.696697023485795, + "tokens_seen": 874381312 + }, + { + "epoch": 0.26, + "learning_rate": 0.00037126379137412236, + "loss": 1.3385, + "theoretical_loss": 3.6966699985467404, + "tokens_seen": 874446848 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003712537612838516, + "loss": 1.4008, + "theoretical_loss": 3.6966429762000756, + "tokens_seen": 874512384 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003712437311935807, + "loss": 1.3349, + "theoretical_loss": 3.6966159564453585, + "tokens_seen": 874577920 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037123370110330996, + "loss": 1.5639, + "theoretical_loss": 3.6965889392821456, + "tokens_seen": 874643456 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003712236710130391, + "loss": 1.441, + "theoretical_loss": 3.696561924709995, + "tokens_seen": 874708992 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003712136409227683, + "loss": 1.4532, + "theoretical_loss": 3.6965349127284632, + "tokens_seen": 874774528 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003712036108324975, + "loss": 1.5102, + "theoretical_loss": 3.6965079033371078, + "tokens_seen": 874840064 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0015037530101835728, + "objective/train/docs_used": 502963, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.793501853942871, + "objective/train/original_loss": 2.793501853942871, + "objective/train/theoretical_loss": 3.696480896535487, + "objective/train/tokens_used": 895365600, + "objective/train/value_avg": -0.006153106689453125, + "objective/train/value_loss": 6.925177876837552e-05, + "objective/train/value_max": -0.0004107952117919922, + "objective/train/value_min": -0.06646728515625, + "objective/train/value_reward_corr": 0.47785667377590135, + "objective/train/value_std": 0.00553131103515625, + "objective/train/weight_avg": 0.9882100224494934, + "objective/train/weighted_lm_loss": 2.7526447772979736, + "objective/train/weights_max": 1.755286455154419, + "objective/train/weights_min": 0.3067399263381958, + "theoretical_loss": 3.696480896535487, + "tokens_seen": 874905600 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003711935807422267, + "loss": 1.4149, + "theoretical_loss": 3.696480896535487, + "tokens_seen": 874905600 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037118355065195586, + "loss": 1.4786, + "theoretical_loss": 3.6964538923231585, + "tokens_seen": 874971136 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003711735205616851, + "loss": 1.3667, + "theoretical_loss": 3.696426890699679, + "tokens_seen": 875036672 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003711634904714142, + "loss": 1.3545, + "theoretical_loss": 3.6963998916646084, + "tokens_seen": 875102208 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037115346038114346, + "loss": 1.355, + "theoretical_loss": 3.6963728952175035, + "tokens_seen": 875167744 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003711434302908726, + "loss": 1.4544, + "theoretical_loss": 3.6963459013579225, + "tokens_seen": 875233280 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003711334002006018, + "loss": 1.4229, + "theoretical_loss": 3.6963189100854237, + "tokens_seen": 875298816 + }, + { + "epoch": 0.27, + "learning_rate": 0.000371123370110331, + "loss": 1.4725, + "theoretical_loss": 3.696291921399566, + "tokens_seen": 875364352 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003711133400200602, + "loss": 1.3808, + "theoretical_loss": 3.696264935299907, + "tokens_seen": 875429888 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037110330992978937, + "loss": 1.4197, + "theoretical_loss": 3.696237951786006, + "tokens_seen": 875495424 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037109327983951855, + "loss": 1.3817, + "theoretical_loss": 3.696210970857422, + "tokens_seen": 875560960 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037108324974924773, + "loss": 1.5238, + "theoretical_loss": 3.6961839925137125, + "tokens_seen": 875626496 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037107321965897696, + "loss": 1.3604, + "theoretical_loss": 3.6961570167544373, + "tokens_seen": 875692032 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003710631895687061, + "loss": 1.5118, + "theoretical_loss": 3.696130043579156, + "tokens_seen": 875757568 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003710531594784353, + "loss": 1.2635, + "theoretical_loss": 3.6961030729874267, + "tokens_seen": 875823104 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037104312938816445, + "loss": 1.3714, + "theoretical_loss": 3.6960761049788093, + "tokens_seen": 875888640 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003710330992978937, + "loss": 1.3833, + "theoretical_loss": 3.696049139552863, + "tokens_seen": 875954176 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037102306920762287, + "loss": 1.461, + "theoretical_loss": 3.6960221767091466, + "tokens_seen": 876019712 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037101303911735205, + "loss": 1.2811, + "theoretical_loss": 3.695995216447221, + "tokens_seen": 876085248 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037100300902708123, + "loss": 1.3434, + "theoretical_loss": 3.6959682587666443, + "tokens_seen": 876150784 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037099297893681047, + "loss": 1.28, + "theoretical_loss": 3.6959413036669777, + "tokens_seen": 876216320 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003709829488465396, + "loss": 1.4965, + "theoretical_loss": 3.6959143511477803, + "tokens_seen": 876281856 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037097291875626883, + "loss": 1.4174, + "theoretical_loss": 3.6958874012086125, + "tokens_seen": 876347392 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037096288866599796, + "loss": 1.4034, + "theoretical_loss": 3.695860453849034, + "tokens_seen": 876412928 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003709528585757272, + "loss": 1.4736, + "theoretical_loss": 3.6958335090686054, + "tokens_seen": 876478464 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.000832165649626404, + "objective/train/docs_used": 503455, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.910149097442627, + "objective/train/original_loss": 2.9101486206054688, + "objective/train/theoretical_loss": 3.695806566866887, + "objective/train/tokens_used": 897004000, + "objective/train/value_avg": -0.006114959716796875, + "objective/train/value_loss": 5.754265657742508e-05, + "objective/train/value_max": -0.000396728515625, + "objective/train/value_min": -0.060302734375, + "objective/train/value_reward_corr": 0.5069901842360044, + "objective/train/value_std": 0.005138397216796875, + "objective/train/weight_avg": 0.9944601655006409, + "objective/train/weighted_lm_loss": 2.898165464401245, + "objective/train/weights_max": 1.5638178586959839, + "objective/train/weights_min": 0.5266782641410828, + "theoretical_loss": 3.695806566866887, + "tokens_seen": 876544000 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003709428284854564, + "loss": 1.4493, + "theoretical_loss": 3.695806566866887, + "tokens_seen": 876544000 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037093279839518555, + "loss": 1.3787, + "theoretical_loss": 3.695779627243439, + "tokens_seen": 876609536 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037092276830491474, + "loss": 1.4466, + "theoretical_loss": 3.695752690197822, + "tokens_seen": 876675072 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003709127382146439, + "loss": 1.3806, + "theoretical_loss": 3.6957257557295966, + "tokens_seen": 876740608 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003709027081243731, + "loss": 1.5443, + "theoretical_loss": 3.695698823838324, + "tokens_seen": 876806144 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037089267803410233, + "loss": 1.4633, + "theoretical_loss": 3.6956718945235645, + "tokens_seen": 876871680 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037088264794383146, + "loss": 1.3435, + "theoretical_loss": 3.6956449677848795, + "tokens_seen": 876937216 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003708726178535607, + "loss": 1.3742, + "theoretical_loss": 3.6956180436218293, + "tokens_seen": 877002752 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037086258776328993, + "loss": 1.531, + "theoretical_loss": 3.695591122033976, + "tokens_seen": 877068288 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037085255767301906, + "loss": 1.4954, + "theoretical_loss": 3.695564203020881, + "tokens_seen": 877133824 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003708425275827483, + "loss": 1.4057, + "theoretical_loss": 3.695537286582105, + "tokens_seen": 877199360 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003708324974924774, + "loss": 1.3357, + "theoretical_loss": 3.6955103727172096, + "tokens_seen": 877264896 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037082246740220665, + "loss": 1.4659, + "theoretical_loss": 3.695483461425757, + "tokens_seen": 877330432 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037081243731193584, + "loss": 1.3778, + "theoretical_loss": 3.6954565527073084, + "tokens_seen": 877395968 + }, + { + "epoch": 0.27, + "learning_rate": 0.000370802407221665, + "loss": 1.5088, + "theoretical_loss": 3.6954296465614256, + "tokens_seen": 877461504 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003707923771313942, + "loss": 1.4358, + "theoretical_loss": 3.695402742987671, + "tokens_seen": 877527040 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003707823470411234, + "loss": 1.3455, + "theoretical_loss": 3.695375841985606, + "tokens_seen": 877592576 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037077231695085256, + "loss": 1.4226, + "theoretical_loss": 3.695348943554793, + "tokens_seen": 877658112 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003707622868605818, + "loss": 1.2526, + "theoretical_loss": 3.6953220476947948, + "tokens_seen": 877723648 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003707522567703109, + "loss": 1.3666, + "theoretical_loss": 3.695295154405173, + "tokens_seen": 877789184 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037074222668004016, + "loss": 1.4814, + "theoretical_loss": 3.6952682636854908, + "tokens_seen": 877854720 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003707321965897693, + "loss": 1.3749, + "theoretical_loss": 3.69524137553531, + "tokens_seen": 877920256 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003707221664994985, + "loss": 1.3634, + "theoretical_loss": 3.695214489954194, + "tokens_seen": 877985792 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003707121364092277, + "loss": 1.4769, + "theoretical_loss": 3.6951876069417056, + "tokens_seen": 878051328 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003707021063189569, + "loss": 1.4546, + "theoretical_loss": 3.6951607264974067, + "tokens_seen": 878116864 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": 0.0003863504098262638, + "objective/train/docs_used": 504719, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8297622203826904, + "objective/train/original_loss": 2.8297624588012695, + "objective/train/theoretical_loss": 3.695133848620862, + "objective/train/tokens_used": 898642400, + "objective/train/value_avg": -0.00516510009765625, + "objective/train/value_loss": 3.814995943685062e-05, + "objective/train/value_max": -0.000469207763671875, + "objective/train/value_min": -0.04931640625, + "objective/train/value_reward_corr": 0.3288488307543286, + "objective/train/value_std": 0.0037689208984375, + "objective/train/weight_avg": 1.0057374238967896, + "objective/train/weighted_lm_loss": 2.8498904705047607, + "objective/train/weights_max": 1.5686973333358765, + "objective/train/weights_min": 0.5570511221885681, + "theoretical_loss": 3.695133848620862, + "tokens_seen": 878182400 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037069207622868606, + "loss": 1.3395, + "theoretical_loss": 3.695133848620862, + "tokens_seen": 878182400 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003706820461384153, + "loss": 1.4828, + "theoretical_loss": 3.695106973311632, + "tokens_seen": 878247936 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003706720160481444, + "loss": 1.3803, + "theoretical_loss": 3.695080100569283, + "tokens_seen": 878313472 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037066198595787366, + "loss": 1.4636, + "theoretical_loss": 3.695053230393376, + "tokens_seen": 878379008 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003706519558676028, + "loss": 1.3999, + "theoretical_loss": 3.6950263627834765, + "tokens_seen": 878444544 + }, + { + "epoch": 0.27, + "learning_rate": 0.000370641925777332, + "loss": 1.5081, + "theoretical_loss": 3.694999497739146, + "tokens_seen": 878510080 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003706318956870612, + "loss": 1.3447, + "theoretical_loss": 3.6949726352599495, + "tokens_seen": 878575616 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003706218655967904, + "loss": 1.3308, + "theoretical_loss": 3.6949457753454498, + "tokens_seen": 878641152 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037061183550651957, + "loss": 1.3729, + "theoretical_loss": 3.6949189179952113, + "tokens_seen": 878706688 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037060180541624875, + "loss": 1.5702, + "theoretical_loss": 3.694892063208798, + "tokens_seen": 878772224 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037059177532597793, + "loss": 1.3459, + "theoretical_loss": 3.694865210985774, + "tokens_seen": 878837760 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037058174523570716, + "loss": 1.4266, + "theoretical_loss": 3.6948383613257034, + "tokens_seen": 878903296 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003705717151454363, + "loss": 1.4535, + "theoretical_loss": 3.6948115142281504, + "tokens_seen": 878968832 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003705616850551655, + "loss": 1.4525, + "theoretical_loss": 3.6947846696926803, + "tokens_seen": 879034368 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037055165496489465, + "loss": 1.3516, + "theoretical_loss": 3.6947578277188557, + "tokens_seen": 879099904 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003705416248746239, + "loss": 1.5199, + "theoretical_loss": 3.694730988306243, + "tokens_seen": 879165440 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037053159478435307, + "loss": 1.4389, + "theoretical_loss": 3.694704151454406, + "tokens_seen": 879230976 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037052156469408225, + "loss": 1.5579, + "theoretical_loss": 3.6946773171629097, + "tokens_seen": 879296512 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037051153460381143, + "loss": 1.452, + "theoretical_loss": 3.6946504854313194, + "tokens_seen": 879362048 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037050150451354067, + "loss": 1.408, + "theoretical_loss": 3.6946236562591994, + "tokens_seen": 879427584 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003704914744232698, + "loss": 1.4483, + "theoretical_loss": 3.6945968296461156, + "tokens_seen": 879493120 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037048144433299903, + "loss": 1.3806, + "theoretical_loss": 3.6945700055916326, + "tokens_seen": 879558656 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037047141424272816, + "loss": 1.2955, + "theoretical_loss": 3.6945431840953162, + "tokens_seen": 879624192 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003704613841524574, + "loss": 1.4248, + "theoretical_loss": 3.694516365156732, + "tokens_seen": 879689728 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003704513540621866, + "loss": 1.4412, + "theoretical_loss": 3.6944895487754454, + "tokens_seen": 879755264 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.00034843271714635193, + "objective/train/docs_used": 505361, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1568210124969482, + "objective/train/original_loss": 3.1568210124969482, + "objective/train/theoretical_loss": 3.694462734951022, + "objective/train/tokens_used": 900280800, + "objective/train/value_avg": -0.004947662353515625, + "objective/train/value_loss": 5.0296530389459804e-05, + "objective/train/value_max": -0.00045299530029296875, + "objective/train/value_min": -0.09283447265625, + "objective/train/value_reward_corr": 0.3564467704342003, + "objective/train/value_std": 0.003940582275390625, + "objective/train/weight_avg": 0.9988802671432495, + "objective/train/weighted_lm_loss": 3.161625623703003, + "objective/train/weights_max": 2.371978282928467, + "objective/train/weights_min": 0.41007721424102783, + "theoretical_loss": 3.694462734951022, + "tokens_seen": 879820800 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037044132397191575, + "loss": 1.4817, + "theoretical_loss": 3.694462734951022, + "tokens_seen": 879820800 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037043129388164494, + "loss": 1.4049, + "theoretical_loss": 3.6944359236830273, + "tokens_seen": 879886336 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003704212637913741, + "loss": 1.3311, + "theoretical_loss": 3.6944091149710276, + "tokens_seen": 879951872 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003704112337011033, + "loss": 1.3329, + "theoretical_loss": 3.694382308814589, + "tokens_seen": 880017408 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037040120361083253, + "loss": 1.4995, + "theoretical_loss": 3.6943555052132773, + "tokens_seen": 880082944 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037039117352056166, + "loss": 1.4074, + "theoretical_loss": 3.6943287041666584, + "tokens_seen": 880148480 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003703811434302909, + "loss": 1.4379, + "theoretical_loss": 3.6943019056742994, + "tokens_seen": 880214016 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003703711133400201, + "loss": 1.3368, + "theoretical_loss": 3.694275109735766, + "tokens_seen": 880279552 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037036108324974926, + "loss": 1.4175, + "theoretical_loss": 3.694248316350625, + "tokens_seen": 880345088 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037035105315947844, + "loss": 1.3961, + "theoretical_loss": 3.6942215255184436, + "tokens_seen": 880410624 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003703410230692076, + "loss": 1.3472, + "theoretical_loss": 3.6941947372387873, + "tokens_seen": 880476160 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003703309929789368, + "loss": 1.4404, + "theoretical_loss": 3.694167951511224, + "tokens_seen": 880541696 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037032096288866604, + "loss": 1.3628, + "theoretical_loss": 3.69414116833532, + "tokens_seen": 880607232 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037031093279839516, + "loss": 1.4199, + "theoretical_loss": 3.694114387710643, + "tokens_seen": 880672768 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003703009027081244, + "loss": 1.4363, + "theoretical_loss": 3.69408760963676, + "tokens_seen": 880738304 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003702908726178535, + "loss": 1.3538, + "theoretical_loss": 3.6940608341132375, + "tokens_seen": 880803840 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037028084252758276, + "loss": 1.344, + "theoretical_loss": 3.694034061139644, + "tokens_seen": 880869376 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037027081243731194, + "loss": 1.4515, + "theoretical_loss": 3.694007290715546, + "tokens_seen": 880934912 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003702607823470411, + "loss": 1.4808, + "theoretical_loss": 3.693980522840512, + "tokens_seen": 881000448 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003702507522567703, + "loss": 1.4594, + "theoretical_loss": 3.693953757514109, + "tokens_seen": 881065984 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003702407221664995, + "loss": 1.435, + "theoretical_loss": 3.693926994735904, + "tokens_seen": 881131520 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037023069207622867, + "loss": 1.3514, + "theoretical_loss": 3.693900234505467, + "tokens_seen": 881197056 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003702206619859579, + "loss": 1.2924, + "theoretical_loss": 3.6938734768223647, + "tokens_seen": 881262592 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037021063189568703, + "loss": 1.4464, + "theoretical_loss": 3.693846721686165, + "tokens_seen": 881328128 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037020060180541626, + "loss": 1.457, + "theoretical_loss": 3.693819969096437, + "tokens_seen": 881393664 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.00039015995571389794, + "objective/train/docs_used": 505856, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0157384872436523, + "objective/train/original_loss": 3.0157382488250732, + "objective/train/theoretical_loss": 3.693793219052748, + "objective/train/tokens_used": 901919200, + "objective/train/value_avg": -0.00464630126953125, + "objective/train/value_loss": 3.874042158713564e-05, + "objective/train/value_max": -0.00035834312438964844, + "objective/train/value_min": -0.07879638671875, + "objective/train/value_reward_corr": 0.5250025446742201, + "objective/train/value_std": 0.004302978515625, + "objective/train/weight_avg": 0.9979839324951172, + "objective/train/weighted_lm_loss": 3.019076347351074, + "objective/train/weights_max": 1.7572474479675293, + "objective/train/weights_min": 0.5580294728279114, + "theoretical_loss": 3.693793219052748, + "tokens_seen": 881459200 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037019057171514544, + "loss": 1.4231, + "theoretical_loss": 3.693793219052748, + "tokens_seen": 881459200 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003701805416248746, + "loss": 1.3227, + "theoretical_loss": 3.693766471554667, + "tokens_seen": 881524736 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003701705115346038, + "loss": 1.5282, + "theoretical_loss": 3.6937397266017626, + "tokens_seen": 881590272 + }, + { + "epoch": 0.27, + "learning_rate": 0.000370160481444333, + "loss": 1.2965, + "theoretical_loss": 3.693712984193603, + "tokens_seen": 881655808 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037015045135406217, + "loss": 1.506, + "theoretical_loss": 3.693686244329758, + "tokens_seen": 881721344 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003701404212637914, + "loss": 1.4187, + "theoretical_loss": 3.6936595070097953, + "tokens_seen": 881786880 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037013039117352053, + "loss": 1.3736, + "theoretical_loss": 3.693632772233284, + "tokens_seen": 881852416 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037012036108324977, + "loss": 1.3488, + "theoretical_loss": 3.6936060399997936, + "tokens_seen": 881917952 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037011033099297895, + "loss": 1.4089, + "theoretical_loss": 3.693579310308893, + "tokens_seen": 881983488 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037010030090270813, + "loss": 1.3972, + "theoretical_loss": 3.6935525831601517, + "tokens_seen": 882049024 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037009027081243736, + "loss": 1.3926, + "theoretical_loss": 3.6935258585531385, + "tokens_seen": 882114560 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003700802407221665, + "loss": 1.3345, + "theoretical_loss": 3.6934991364874232, + "tokens_seen": 882180096 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003700702106318957, + "loss": 1.433, + "theoretical_loss": 3.6934724169625763, + "tokens_seen": 882245632 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037006018054162485, + "loss": 1.2912, + "theoretical_loss": 3.6934456999781657, + "tokens_seen": 882311168 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003700501504513541, + "loss": 1.3587, + "theoretical_loss": 3.6934189855337625, + "tokens_seen": 882376704 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037004012036108327, + "loss": 1.4991, + "theoretical_loss": 3.693392273628936, + "tokens_seen": 882442240 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037003009027081245, + "loss": 1.4854, + "theoretical_loss": 3.6933655642632566, + "tokens_seen": 882507776 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037002006018054163, + "loss": 1.4295, + "theoretical_loss": 3.693338857436294, + "tokens_seen": 882573312 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037001003009027087, + "loss": 1.4207, + "theoretical_loss": 3.693312153147619, + "tokens_seen": 882638848 + }, + { + "epoch": 0.27, + "learning_rate": 0.00037, + "loss": 1.4916, + "theoretical_loss": 3.693285451396801, + "tokens_seen": 882704384 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036998996990972923, + "loss": 1.5154, + "theoretical_loss": 3.6932587521834117, + "tokens_seen": 882769920 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036997993981945836, + "loss": 1.3296, + "theoretical_loss": 3.693232055507021, + "tokens_seen": 882835456 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003699699097291876, + "loss": 1.4039, + "theoretical_loss": 3.6932053613671982, + "tokens_seen": 882900992 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003699598796389168, + "loss": 1.5196, + "theoretical_loss": 3.6931786697635163, + "tokens_seen": 882966528 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036994984954864595, + "loss": 1.4585, + "theoretical_loss": 3.6931519806955446, + "tokens_seen": 883032064 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0003729608142748475, + "objective/train/docs_used": 506939, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7033450603485107, + "objective/train/original_loss": 2.7033450603485107, + "objective/train/theoretical_loss": 3.6931252941628547, + "objective/train/tokens_used": 903557600, + "objective/train/value_avg": -0.00496673583984375, + "objective/train/value_loss": 5.9156984207220376e-05, + "objective/train/value_max": -0.00023055076599121094, + "objective/train/value_min": -0.06683349609375, + "objective/train/value_reward_corr": 0.22659604927555285, + "objective/train/value_std": 0.003704071044921875, + "objective/train/weight_avg": 0.9991013407707214, + "objective/train/weighted_lm_loss": 2.696384906768799, + "objective/train/weights_max": 1.9509861469268799, + "objective/train/weights_min": 0.5185285806655884, + "theoretical_loss": 3.6931252941628547, + "tokens_seen": 883097600 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036993981945837514, + "loss": 1.2456, + "theoretical_loss": 3.6931252941628547, + "tokens_seen": 883097600 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003699297893681043, + "loss": 1.3125, + "theoretical_loss": 3.693098610165017, + "tokens_seen": 883163136 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003699197592778335, + "loss": 1.4454, + "theoretical_loss": 3.693071928701604, + "tokens_seen": 883228672 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036990972918756273, + "loss": 1.4373, + "theoretical_loss": 3.6930452497721857, + "tokens_seen": 883294208 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036989969909729186, + "loss": 1.4767, + "theoretical_loss": 3.6930185733763334, + "tokens_seen": 883359744 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003698896690070211, + "loss": 1.4322, + "theoretical_loss": 3.6929918995136193, + "tokens_seen": 883425280 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003698796389167503, + "loss": 1.3404, + "theoretical_loss": 3.6929652281836143, + "tokens_seen": 883490816 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036986960882647946, + "loss": 1.4196, + "theoretical_loss": 3.692938559385891, + "tokens_seen": 883556352 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036985957873620864, + "loss": 1.4329, + "theoretical_loss": 3.69291189312002, + "tokens_seen": 883621888 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003698495486459378, + "loss": 1.3166, + "theoretical_loss": 3.6928852293855745, + "tokens_seen": 883687424 + }, + { + "epoch": 0.27, + "learning_rate": 0.000369839518555667, + "loss": 1.4337, + "theoretical_loss": 3.692858568182125, + "tokens_seen": 883752960 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036982948846539624, + "loss": 1.4472, + "theoretical_loss": 3.692831909509245, + "tokens_seen": 883818496 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036981945837512536, + "loss": 1.4476, + "theoretical_loss": 3.6928052533665054, + "tokens_seen": 883884032 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003698094282848546, + "loss": 1.4968, + "theoretical_loss": 3.6927785997534794, + "tokens_seen": 883949568 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003697993981945837, + "loss": 1.3341, + "theoretical_loss": 3.6927519486697395, + "tokens_seen": 884015104 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036978936810431296, + "loss": 1.4684, + "theoretical_loss": 3.6927253001148577, + "tokens_seen": 884080640 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036977933801404214, + "loss": 1.5105, + "theoretical_loss": 3.6926986540884066, + "tokens_seen": 884146176 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003697693079237713, + "loss": 1.4492, + "theoretical_loss": 3.6926720105899586, + "tokens_seen": 884211712 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003697592778335005, + "loss": 1.4942, + "theoretical_loss": 3.6926453696190875, + "tokens_seen": 884277248 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003697492477432297, + "loss": 1.3426, + "theoretical_loss": 3.6926187311753655, + "tokens_seen": 884342784 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036973921765295887, + "loss": 1.4349, + "theoretical_loss": 3.692592095258366, + "tokens_seen": 884408320 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003697291875626881, + "loss": 1.4456, + "theoretical_loss": 3.692565461867662, + "tokens_seen": 884473856 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036971915747241723, + "loss": 1.5149, + "theoretical_loss": 3.6925388310028264, + "tokens_seen": 884539392 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036970912738214646, + "loss": 1.4178, + "theoretical_loss": 3.692512202663433, + "tokens_seen": 884604928 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036969909729187565, + "loss": 1.275, + "theoretical_loss": 3.692485576849055, + "tokens_seen": 884670464 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0008300432236865163, + "objective/train/docs_used": 507654, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5888113975524902, + "objective/train/original_loss": 2.5888116359710693, + "objective/train/theoretical_loss": 3.6924589535592656, + "objective/train/tokens_used": 905196000, + "objective/train/value_avg": -0.005443572998046875, + "objective/train/value_loss": 7.280692079802975e-05, + "objective/train/value_max": -0.000293731689453125, + "objective/train/value_min": -0.06842041015625, + "objective/train/value_reward_corr": 0.5444924585981109, + "objective/train/value_std": 0.0054473876953125, + "objective/train/weight_avg": 0.9949708580970764, + "objective/train/weighted_lm_loss": 2.577789545059204, + "objective/train/weights_max": 1.8021658658981323, + "objective/train/weights_min": 0.15137417614459991, + "theoretical_loss": 3.6924589535592656, + "tokens_seen": 884736000 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003696890672016048, + "loss": 1.3243, + "theoretical_loss": 3.6924589535592656, + "tokens_seen": 884736000 + }, + { + "epoch": 0.27, + "learning_rate": 0.000369679037111334, + "loss": 1.3951, + "theoretical_loss": 3.6924323327936395, + "tokens_seen": 884801536 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003696690070210632, + "loss": 1.4082, + "theoretical_loss": 3.6924057145517493, + "tokens_seen": 884867072 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036965897693079237, + "loss": 1.3949, + "theoretical_loss": 3.6923790988331695, + "tokens_seen": 884932608 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003696489468405216, + "loss": 1.451, + "theoretical_loss": 3.692352485637474, + "tokens_seen": 884998144 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036963891675025073, + "loss": 1.3777, + "theoretical_loss": 3.6923258749642365, + "tokens_seen": 885063680 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036962888665997997, + "loss": 1.4686, + "theoretical_loss": 3.6922992668130314, + "tokens_seen": 885129216 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003696188565697091, + "loss": 1.4597, + "theoretical_loss": 3.6922726611834333, + "tokens_seen": 885194752 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036960882647943833, + "loss": 1.438, + "theoretical_loss": 3.692246058075016, + "tokens_seen": 885260288 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003695987963891675, + "loss": 1.466, + "theoretical_loss": 3.6922194574873544, + "tokens_seen": 885325824 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003695887662988967, + "loss": 1.3372, + "theoretical_loss": 3.6921928594200226, + "tokens_seen": 885391360 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036957873620862587, + "loss": 1.4129, + "theoretical_loss": 3.6921662638725956, + "tokens_seen": 885456896 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036956870611835505, + "loss": 1.3743, + "theoretical_loss": 3.6921396708446483, + "tokens_seen": 885522432 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036955867602808424, + "loss": 1.3748, + "theoretical_loss": 3.692113080335755, + "tokens_seen": 885587968 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036954864593781347, + "loss": 1.3834, + "theoretical_loss": 3.6920864923454917, + "tokens_seen": 885653504 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003695386158475426, + "loss": 1.3133, + "theoretical_loss": 3.6920599068734328, + "tokens_seen": 885719040 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036952858575727183, + "loss": 1.2728, + "theoretical_loss": 3.6920333239191536, + "tokens_seen": 885784576 + }, + { + "epoch": 0.27, + "learning_rate": 0.000369518555667001, + "loss": 1.3626, + "theoretical_loss": 3.6920067434822297, + "tokens_seen": 885850112 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003695085255767302, + "loss": 1.5582, + "theoretical_loss": 3.6919801655622355, + "tokens_seen": 885915648 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003694984954864594, + "loss": 1.3718, + "theoretical_loss": 3.6919535901587475, + "tokens_seen": 885981184 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036948846539618856, + "loss": 1.2955, + "theoretical_loss": 3.6919270172713414, + "tokens_seen": 886046720 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036947843530591774, + "loss": 1.3151, + "theoretical_loss": 3.6919004468995924, + "tokens_seen": 886112256 + }, + { + "epoch": 0.27, + "learning_rate": 0.000369468405215647, + "loss": 1.465, + "theoretical_loss": 3.6918738790430763, + "tokens_seen": 886177792 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003694583751253761, + "loss": 1.4268, + "theoretical_loss": 3.691847313701369, + "tokens_seen": 886243328 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036944834503510534, + "loss": 1.3811, + "theoretical_loss": 3.6918207508740473, + "tokens_seen": 886308864 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0012716156197711825, + "objective/train/docs_used": 509172, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.573035478591919, + "objective/train/original_loss": 2.573035478591919, + "objective/train/theoretical_loss": 3.691794190560686, + "objective/train/tokens_used": 906834400, + "objective/train/value_avg": -0.005474090576171875, + "objective/train/value_loss": 9.226159454556182e-05, + "objective/train/value_max": -0.0002415180206298828, + "objective/train/value_min": -0.054290771484375, + "objective/train/value_reward_corr": 0.39283837533287563, + "objective/train/value_std": 0.00461578369140625, + "objective/train/weight_avg": 0.991241991519928, + "objective/train/weighted_lm_loss": 2.5332703590393066, + "objective/train/weights_max": 1.6624720096588135, + "objective/train/weights_min": 0.13790717720985413, + "theoretical_loss": 3.691794190560686, + "tokens_seen": 886374400 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036943831494483446, + "loss": 1.3201, + "theoretical_loss": 3.691794190560686, + "tokens_seen": 886374400 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003694282848545637, + "loss": 1.2924, + "theoretical_loss": 3.691767632760863, + "tokens_seen": 886439936 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003694182547642929, + "loss": 1.4508, + "theoretical_loss": 3.6917410774741524, + "tokens_seen": 886505472 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036940822467402206, + "loss": 1.4231, + "theoretical_loss": 3.6917145247001324, + "tokens_seen": 886571008 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036939819458375124, + "loss": 1.5301, + "theoretical_loss": 3.6916879744383793, + "tokens_seen": 886636544 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003693881644934805, + "loss": 1.4955, + "theoretical_loss": 3.6916614266884693, + "tokens_seen": 886702080 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003693781344032096, + "loss": 1.3467, + "theoretical_loss": 3.6916348814499793, + "tokens_seen": 886767616 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036936810431293884, + "loss": 1.484, + "theoretical_loss": 3.691608338722486, + "tokens_seen": 886833152 + }, + { + "epoch": 0.27, + "learning_rate": 0.000369358074222668, + "loss": 1.3794, + "theoretical_loss": 3.691581798505567, + "tokens_seen": 886898688 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003693480441323972, + "loss": 1.3871, + "theoretical_loss": 3.6915552607987983, + "tokens_seen": 886964224 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036933801404212644, + "loss": 1.525, + "theoretical_loss": 3.691528725601758, + "tokens_seen": 887029760 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036932798395185556, + "loss": 1.3186, + "theoretical_loss": 3.6915021929140224, + "tokens_seen": 887095296 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003693179538615848, + "loss": 1.367, + "theoretical_loss": 3.6914756627351695, + "tokens_seen": 887160832 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003693079237713139, + "loss": 1.3642, + "theoretical_loss": 3.6914491350647767, + "tokens_seen": 887226368 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036929789368104316, + "loss": 1.3258, + "theoretical_loss": 3.6914226099024217, + "tokens_seen": 887291904 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036928786359077234, + "loss": 1.3928, + "theoretical_loss": 3.691396087247682, + "tokens_seen": 887357440 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003692778335005015, + "loss": 1.3347, + "theoretical_loss": 3.691369567100135, + "tokens_seen": 887422976 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003692678034102307, + "loss": 1.3267, + "theoretical_loss": 3.691343049459359, + "tokens_seen": 887488512 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003692577733199599, + "loss": 1.4901, + "theoretical_loss": 3.6913165343249323, + "tokens_seen": 887554048 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036924774322968907, + "loss": 1.4316, + "theoretical_loss": 3.6912900216964317, + "tokens_seen": 887619584 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003692377131394183, + "loss": 1.4069, + "theoretical_loss": 3.6912635115734367, + "tokens_seen": 887685120 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036922768304914743, + "loss": 1.4653, + "theoretical_loss": 3.691237003955525, + "tokens_seen": 887750656 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036921765295887666, + "loss": 1.3206, + "theoretical_loss": 3.691210498842275, + "tokens_seen": 887816192 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036920762286860585, + "loss": 1.4423, + "theoretical_loss": 3.6911839962332653, + "tokens_seen": 887881728 + }, + { + "epoch": 0.27, + "learning_rate": 0.000369197592778335, + "loss": 1.4378, + "theoretical_loss": 3.691157496128074, + "tokens_seen": 887947264 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": 8.100773993646726e-05, + "objective/train/docs_used": 509747, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6930172443389893, + "objective/train/original_loss": 2.6930172443389893, + "objective/train/theoretical_loss": 3.691130998526281, + "objective/train/tokens_used": 908472800, + "objective/train/value_avg": -0.005565643310546875, + "objective/train/value_loss": 6.43728271825239e-05, + "objective/train/value_max": -0.0004012584686279297, + "objective/train/value_min": -0.07147216796875, + "objective/train/value_reward_corr": 0.3275183447137406, + "objective/train/value_std": 0.004062652587890625, + "objective/train/weight_avg": 1.0036238431930542, + "objective/train/weighted_lm_loss": 2.6890015602111816, + "objective/train/weights_max": 1.83995521068573, + "objective/train/weights_min": 0.0842389240860939, + "theoretical_loss": 3.691130998526281, + "tokens_seen": 888012800 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003691875626880642, + "loss": 1.4018, + "theoretical_loss": 3.691130998526281, + "tokens_seen": 888012800 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003691775325977934, + "loss": 1.3373, + "theoretical_loss": 3.6911045034274634, + "tokens_seen": 888078336 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036916750250752257, + "loss": 1.4334, + "theoretical_loss": 3.691078010831202, + "tokens_seen": 888143872 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003691574724172518, + "loss": 1.3891, + "theoretical_loss": 3.6910515207370738, + "tokens_seen": 888209408 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036914744232698093, + "loss": 1.4359, + "theoretical_loss": 3.6910250331446592, + "tokens_seen": 888274944 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036913741223671017, + "loss": 1.4373, + "theoretical_loss": 3.6909985480535368, + "tokens_seen": 888340480 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003691273821464393, + "loss": 1.3653, + "theoretical_loss": 3.6909720654632863, + "tokens_seen": 888406016 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036911735205616853, + "loss": 1.545, + "theoretical_loss": 3.6909455853734867, + "tokens_seen": 888471552 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003691073219658977, + "loss": 1.307, + "theoretical_loss": 3.690919107783718, + "tokens_seen": 888537088 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003690972918756269, + "loss": 1.5638, + "theoretical_loss": 3.6908926326935596, + "tokens_seen": 888602624 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003690872617853561, + "loss": 1.3288, + "theoretical_loss": 3.6908661601025905, + "tokens_seen": 888668160 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036907723169508525, + "loss": 1.3983, + "theoretical_loss": 3.6908396900103915, + "tokens_seen": 888733696 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036906720160481444, + "loss": 1.4322, + "theoretical_loss": 3.6908132224165424, + "tokens_seen": 888799232 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036905717151454367, + "loss": 1.4562, + "theoretical_loss": 3.6907867573206223, + "tokens_seen": 888864768 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003690471414242728, + "loss": 1.3368, + "theoretical_loss": 3.6907602947222125, + "tokens_seen": 888930304 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036903711133400203, + "loss": 1.5021, + "theoretical_loss": 3.690733834620892, + "tokens_seen": 888995840 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003690270812437312, + "loss": 1.4759, + "theoretical_loss": 3.6907073770162424, + "tokens_seen": 889061376 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003690170511534604, + "loss": 1.4152, + "theoretical_loss": 3.690680921907843, + "tokens_seen": 889126912 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003690070210631896, + "loss": 1.3278, + "theoretical_loss": 3.690654469295275, + "tokens_seen": 889192448 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036899699097291876, + "loss": 1.4503, + "theoretical_loss": 3.690628019178119, + "tokens_seen": 889257984 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036898696088264794, + "loss": 1.3832, + "theoretical_loss": 3.6906015715559555, + "tokens_seen": 889323520 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003689769307923772, + "loss": 1.4926, + "theoretical_loss": 3.6905751264283646, + "tokens_seen": 889389056 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003689669007021063, + "loss": 1.4428, + "theoretical_loss": 3.6905486837949284, + "tokens_seen": 889454592 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036895687061183554, + "loss": 1.3574, + "theoretical_loss": 3.690522243655227, + "tokens_seen": 889520128 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036894684052156466, + "loss": 1.4541, + "theoretical_loss": 3.690495806008843, + "tokens_seen": 889585664 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0012635949533432722, + "objective/train/docs_used": 511200, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7751097679138184, + "objective/train/original_loss": 2.7751097679138184, + "objective/train/theoretical_loss": 3.690469370855356, + "objective/train/tokens_used": 910111200, + "objective/train/value_avg": -0.0056610107421875, + "objective/train/value_loss": 5.863578917342238e-05, + "objective/train/value_max": -0.0002002716064453125, + "objective/train/value_min": -0.047882080078125, + "objective/train/value_reward_corr": 0.49644431762622726, + "objective/train/value_std": 0.00461578369140625, + "objective/train/weight_avg": 0.9901456236839294, + "objective/train/weighted_lm_loss": 2.754092216491699, + "objective/train/weights_max": 1.5490193367004395, + "objective/train/weights_min": 0.3401457667350769, + "theoretical_loss": 3.690469370855356, + "tokens_seen": 889651200 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003689368104312939, + "loss": 1.3283, + "theoretical_loss": 3.690469370855356, + "tokens_seen": 889651200 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003689267803410231, + "loss": 1.4372, + "theoretical_loss": 3.690442938194348, + "tokens_seen": 889716736 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036891675025075226, + "loss": 1.3908, + "theoretical_loss": 3.6904165080254003, + "tokens_seen": 889782272 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036890672016048144, + "loss": 1.3455, + "theoretical_loss": 3.6903900803480942, + "tokens_seen": 889847808 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003688966900702107, + "loss": 1.3495, + "theoretical_loss": 3.690363655162012, + "tokens_seen": 889913344 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003688866599799398, + "loss": 1.3599, + "theoretical_loss": 3.6903372324667347, + "tokens_seen": 889978880 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036887662988966904, + "loss": 1.3682, + "theoretical_loss": 3.6903108122618447, + "tokens_seen": 890044416 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036886659979939817, + "loss": 1.3331, + "theoretical_loss": 3.6902843945469233, + "tokens_seen": 890109952 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003688565697091274, + "loss": 1.5077, + "theoretical_loss": 3.6902579793215535, + "tokens_seen": 890175488 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003688465396188566, + "loss": 1.3798, + "theoretical_loss": 3.6902315665853163, + "tokens_seen": 890241024 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036883650952858576, + "loss": 1.4967, + "theoretical_loss": 3.690205156337795, + "tokens_seen": 890306560 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036882647943831494, + "loss": 1.3572, + "theoretical_loss": 3.6901787485785706, + "tokens_seen": 890372096 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003688164493480441, + "loss": 1.4636, + "theoretical_loss": 3.690152343307227, + "tokens_seen": 890437632 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003688064192577733, + "loss": 1.4162, + "theoretical_loss": 3.690125940523346, + "tokens_seen": 890503168 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036879638916750254, + "loss": 1.3557, + "theoretical_loss": 3.6900995402265098, + "tokens_seen": 890568704 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036878635907723167, + "loss": 1.3191, + "theoretical_loss": 3.6900731424163027, + "tokens_seen": 890634240 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003687763289869609, + "loss": 1.3633, + "theoretical_loss": 3.690046747092305, + "tokens_seen": 890699776 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036876629889669003, + "loss": 1.3915, + "theoretical_loss": 3.690020354254102, + "tokens_seen": 890765312 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036875626880641927, + "loss": 1.372, + "theoretical_loss": 3.6899939639012764, + "tokens_seen": 890830848 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036874623871614845, + "loss": 1.3615, + "theoretical_loss": 3.68996757603341, + "tokens_seen": 890896384 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036873620862587763, + "loss": 1.3227, + "theoretical_loss": 3.689941190650087, + "tokens_seen": 890961920 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003687261785356068, + "loss": 1.3744, + "theoretical_loss": 3.689914807750891, + "tokens_seen": 891027456 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036871614844533605, + "loss": 1.4255, + "theoretical_loss": 3.6898884273354042, + "tokens_seen": 891092992 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036870611835506517, + "loss": 1.417, + "theoretical_loss": 3.689862049403212, + "tokens_seen": 891158528 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003686960882647944, + "loss": 1.3547, + "theoretical_loss": 3.689835673953896, + "tokens_seen": 891224064 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0010065669193863869, + "objective/train/docs_used": 511885, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7052531242370605, + "objective/train/original_loss": 2.7052526473999023, + "objective/train/theoretical_loss": 3.689809300987042, + "objective/train/tokens_used": 911749600, + "objective/train/value_avg": -0.00537109375, + "objective/train/value_loss": 7.892166468081996e-05, + "objective/train/value_max": -0.0003769397735595703, + "objective/train/value_min": -0.05865478515625, + "objective/train/value_reward_corr": 0.3515710545403447, + "objective/train/value_std": 0.00437164306640625, + "objective/train/weight_avg": 0.993448793888092, + "objective/train/weighted_lm_loss": 2.694650650024414, + "objective/train/weights_max": 1.7977715730667114, + "objective/train/weights_min": 0.1975039690732956, + "theoretical_loss": 3.689809300987042, + "tokens_seen": 891289600 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036868605817452353, + "loss": 1.3656, + "theoretical_loss": 3.689809300987042, + "tokens_seen": 891289600 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036867602808425277, + "loss": 1.4436, + "theoretical_loss": 3.689782930502232, + "tokens_seen": 891355136 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036866599799398195, + "loss": 1.53, + "theoretical_loss": 3.6897565624990505, + "tokens_seen": 891420672 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036865596790371113, + "loss": 1.3661, + "theoretical_loss": 3.6897301969770817, + "tokens_seen": 891486208 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003686459378134403, + "loss": 1.4131, + "theoretical_loss": 3.6897038339359107, + "tokens_seen": 891551744 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003686359077231695, + "loss": 1.4163, + "theoretical_loss": 3.6896774733751205, + "tokens_seen": 891617280 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003686258776328987, + "loss": 1.4615, + "theoretical_loss": 3.6896511152942955, + "tokens_seen": 891682816 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003686158475426279, + "loss": 1.4576, + "theoretical_loss": 3.6896247596930207, + "tokens_seen": 891748352 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003686058174523571, + "loss": 1.4076, + "theoretical_loss": 3.6895984065708807, + "tokens_seen": 891813888 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003685957873620863, + "loss": 1.4463, + "theoretical_loss": 3.689572055927459, + "tokens_seen": 891879424 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036858575727181545, + "loss": 1.3416, + "theoretical_loss": 3.689545707762342, + "tokens_seen": 891944960 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036857572718154464, + "loss": 1.3248, + "theoretical_loss": 3.689519362075113, + "tokens_seen": 892010496 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036856569709127387, + "loss": 1.4487, + "theoretical_loss": 3.6894930188653587, + "tokens_seen": 892076032 + }, + { + "epoch": 0.27, + "learning_rate": 0.000368555667001003, + "loss": 1.523, + "theoretical_loss": 3.6894666781326624, + "tokens_seen": 892141568 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036854563691073223, + "loss": 1.4712, + "theoretical_loss": 3.68944033987661, + "tokens_seen": 892207104 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003685356068204614, + "loss": 1.4892, + "theoretical_loss": 3.6894140040967867, + "tokens_seen": 892272640 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003685255767301906, + "loss": 1.5638, + "theoretical_loss": 3.6893876707927777, + "tokens_seen": 892338176 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003685155466399198, + "loss": 1.4866, + "theoretical_loss": 3.6893613399641687, + "tokens_seen": 892403712 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036850551654964896, + "loss": 1.4169, + "theoretical_loss": 3.6893350116105452, + "tokens_seen": 892469248 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036849548645937814, + "loss": 1.3869, + "theoretical_loss": 3.689308685731493, + "tokens_seen": 892534784 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003684854563691074, + "loss": 1.376, + "theoretical_loss": 3.6892823623265976, + "tokens_seen": 892600320 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003684754262788365, + "loss": 1.403, + "theoretical_loss": 3.6892560413954447, + "tokens_seen": 892665856 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036846539618856574, + "loss": 1.3814, + "theoretical_loss": 3.68922972293762, + "tokens_seen": 892731392 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036845536609829486, + "loss": 1.4146, + "theoretical_loss": 3.6892034069527107, + "tokens_seen": 892796928 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003684453360080241, + "loss": 1.3485, + "theoretical_loss": 3.6891770934403016, + "tokens_seen": 892862464 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.000685345265083015, + "objective/train/docs_used": 512575, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.420750856399536, + "objective/train/original_loss": 2.420750617980957, + "objective/train/theoretical_loss": 3.68915078239998, + "objective/train/tokens_used": 913388000, + "objective/train/value_avg": -0.006023406982421875, + "objective/train/value_loss": 7.55677538109012e-05, + "objective/train/value_max": -0.0002982616424560547, + "objective/train/value_min": -0.09234619140625, + "objective/train/value_reward_corr": 0.4975610706637417, + "objective/train/value_std": 0.006557464599609375, + "objective/train/weight_avg": 0.9966380000114441, + "objective/train/weighted_lm_loss": 2.412728786468506, + "objective/train/weights_max": 1.8385759592056274, + "objective/train/weights_min": 0.26200705766677856, + "theoretical_loss": 3.68915078239998, + "tokens_seen": 892928000 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003684353059177533, + "loss": 1.3015, + "theoretical_loss": 3.68915078239998, + "tokens_seen": 892928000 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036842527582748246, + "loss": 1.3734, + "theoretical_loss": 3.689124473831331, + "tokens_seen": 892993536 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036841524573721164, + "loss": 1.2917, + "theoretical_loss": 3.6890981677339427, + "tokens_seen": 893059072 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003684052156469409, + "loss": 1.4685, + "theoretical_loss": 3.6890718641074005, + "tokens_seen": 893124608 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036839518555667, + "loss": 1.3762, + "theoretical_loss": 3.689045562951291, + "tokens_seen": 893190144 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036838515546639924, + "loss": 1.3633, + "theoretical_loss": 3.6890192642652018, + "tokens_seen": 893255680 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036837512537612837, + "loss": 1.3816, + "theoretical_loss": 3.688992968048719, + "tokens_seen": 893321216 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003683650952858576, + "loss": 1.356, + "theoretical_loss": 3.6889666743014295, + "tokens_seen": 893386752 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003683550651955868, + "loss": 1.3503, + "theoretical_loss": 3.6889403830229206, + "tokens_seen": 893452288 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036834503510531596, + "loss": 1.3202, + "theoretical_loss": 3.6889140942127794, + "tokens_seen": 893517824 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036833500501504514, + "loss": 1.3955, + "theoretical_loss": 3.688887807870593, + "tokens_seen": 893583360 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003683249749247743, + "loss": 1.4272, + "theoretical_loss": 3.688861523995949, + "tokens_seen": 893648896 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003683149448345035, + "loss": 1.4156, + "theoretical_loss": 3.6888352425884348, + "tokens_seen": 893714432 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036830491474423274, + "loss": 1.4281, + "theoretical_loss": 3.6888089636476376, + "tokens_seen": 893779968 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036829488465396187, + "loss": 1.3766, + "theoretical_loss": 3.6887826871731453, + "tokens_seen": 893845504 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003682848545636911, + "loss": 1.4974, + "theoretical_loss": 3.688756413164546, + "tokens_seen": 893911040 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036827482447342023, + "loss": 1.4557, + "theoretical_loss": 3.688730141621427, + "tokens_seen": 893976576 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036826479438314947, + "loss": 1.339, + "theoretical_loss": 3.688703872543376, + "tokens_seen": 894042112 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036825476429287865, + "loss": 1.4841, + "theoretical_loss": 3.6886776059299815, + "tokens_seen": 894107648 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036824473420260783, + "loss": 1.4605, + "theoretical_loss": 3.6886513417808318, + "tokens_seen": 894173184 + }, + { + "epoch": 0.27, + "learning_rate": 0.000368234704112337, + "loss": 1.4305, + "theoretical_loss": 3.6886250800955143, + "tokens_seen": 894238720 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036822467402206625, + "loss": 1.413, + "theoretical_loss": 3.6885988208736187, + "tokens_seen": 894304256 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036821464393179537, + "loss": 1.3518, + "theoretical_loss": 3.6885725641147316, + "tokens_seen": 894369792 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003682046138415246, + "loss": 1.372, + "theoretical_loss": 3.6885463098184434, + "tokens_seen": 894435328 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036819458375125373, + "loss": 1.3989, + "theoretical_loss": 3.6885200579843413, + "tokens_seen": 894500864 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.002276606159284711, + "objective/train/docs_used": 513630, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.958585023880005, + "objective/train/original_loss": 2.9585845470428467, + "objective/train/theoretical_loss": 3.688493808612015, + "objective/train/tokens_used": 915026400, + "objective/train/value_avg": -0.00614166259765625, + "objective/train/value_loss": 8.430451271124184e-05, + "objective/train/value_max": -0.00021827220916748047, + "objective/train/value_min": -0.08929443359375, + "objective/train/value_reward_corr": 0.6864215892169767, + "objective/train/value_std": 0.006984710693359375, + "objective/train/weight_avg": 0.9811862707138062, + "objective/train/weighted_lm_loss": 2.8957903385162354, + "objective/train/weights_max": 1.6959775686264038, + "objective/train/weights_min": 0.3431956171989441, + "theoretical_loss": 3.688493808612015, + "tokens_seen": 894566400 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036818455366098297, + "loss": 1.4343, + "theoretical_loss": 3.688493808612015, + "tokens_seen": 894566400 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036817452357071215, + "loss": 1.4637, + "theoretical_loss": 3.688467561701053, + "tokens_seen": 894631936 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036816449348044133, + "loss": 1.4769, + "theoretical_loss": 3.6884413172510437, + "tokens_seen": 894697472 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003681544633901705, + "loss": 1.3651, + "theoretical_loss": 3.6884150752615765, + "tokens_seen": 894763008 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003681444332998997, + "loss": 1.3768, + "theoretical_loss": 3.688388835732241, + "tokens_seen": 894828544 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003681344032096289, + "loss": 1.4563, + "theoretical_loss": 3.688362598662626, + "tokens_seen": 894894080 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003681243731193581, + "loss": 1.3411, + "theoretical_loss": 3.6883363640523203, + "tokens_seen": 894959616 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036811434302908724, + "loss": 1.433, + "theoretical_loss": 3.6883101319009137, + "tokens_seen": 895025152 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003681043129388165, + "loss": 1.3947, + "theoretical_loss": 3.6882839022079965, + "tokens_seen": 895090688 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003680942828485456, + "loss": 1.4191, + "theoretical_loss": 3.6882576749731575, + "tokens_seen": 895156224 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036808425275827484, + "loss": 1.4484, + "theoretical_loss": 3.6882314501959863, + "tokens_seen": 895221760 + }, + { + "epoch": 0.27, + "learning_rate": 0.000368074222668004, + "loss": 1.4092, + "theoretical_loss": 3.6882052278760726, + "tokens_seen": 895287296 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003680641925777332, + "loss": 1.3648, + "theoretical_loss": 3.688179008013007, + "tokens_seen": 895352832 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003680541624874624, + "loss": 1.5759, + "theoretical_loss": 3.688152790606379, + "tokens_seen": 895418368 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003680441323971916, + "loss": 1.3831, + "theoretical_loss": 3.6881265756557795, + "tokens_seen": 895483904 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036803410230692074, + "loss": 1.3949, + "theoretical_loss": 3.688100363160797, + "tokens_seen": 895549440 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036802407221665, + "loss": 1.4867, + "theoretical_loss": 3.6880741531210237, + "tokens_seen": 895614976 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003680140421263791, + "loss": 1.3316, + "theoretical_loss": 3.6880479455360486, + "tokens_seen": 895680512 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036800401203610834, + "loss": 1.3374, + "theoretical_loss": 3.688021740405463, + "tokens_seen": 895746048 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003679939819458375, + "loss": 1.4033, + "theoretical_loss": 3.687995537728858, + "tokens_seen": 895811584 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003679839518555667, + "loss": 1.4218, + "theoretical_loss": 3.6879693375058222, + "tokens_seen": 895877120 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003679739217652959, + "loss": 1.4841, + "theoretical_loss": 3.687943139735949, + "tokens_seen": 895942656 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036796389167502506, + "loss": 1.4702, + "theoretical_loss": 3.687916944418827, + "tokens_seen": 896008192 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036795386158475424, + "loss": 1.4208, + "theoretical_loss": 3.6878907515540487, + "tokens_seen": 896073728 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003679438314944835, + "loss": 1.422, + "theoretical_loss": 3.687864561141205, + "tokens_seen": 896139264 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0009397021494805813, + "objective/train/docs_used": 514378, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4741315841674805, + "objective/train/original_loss": 3.4741318225860596, + "objective/train/theoretical_loss": 3.6878383731798863, + "objective/train/tokens_used": 916664800, + "objective/train/value_avg": -0.0052490234375, + "objective/train/value_loss": 5.683863491867669e-05, + "objective/train/value_max": -0.0004673004150390625, + "objective/train/value_min": -0.053985595703125, + "objective/train/value_reward_corr": 0.361422858152832, + "objective/train/value_std": 0.0038814544677734375, + "objective/train/weight_avg": 0.993232250213623, + "objective/train/weighted_lm_loss": 3.461693525314331, + "objective/train/weights_max": 1.6075469255447388, + "objective/train/weights_min": 0.4148921072483063, + "theoretical_loss": 3.6878383731798863, + "tokens_seen": 896204800 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003679338014042126, + "loss": 1.5297, + "theoretical_loss": 3.6878383731798863, + "tokens_seen": 896204800 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036792377131394184, + "loss": 1.3605, + "theoretical_loss": 3.6878121876696843, + "tokens_seen": 896270336 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036791374122367097, + "loss": 1.3596, + "theoretical_loss": 3.6877860046101905, + "tokens_seen": 896335872 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003679037111334002, + "loss": 1.3596, + "theoretical_loss": 3.6877598240009966, + "tokens_seen": 896401408 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003678936810431294, + "loss": 1.2574, + "theoretical_loss": 3.6877336458416936, + "tokens_seen": 896466944 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036788365095285857, + "loss": 1.3552, + "theoretical_loss": 3.6877074701318735, + "tokens_seen": 896532480 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036787362086258775, + "loss": 1.3258, + "theoretical_loss": 3.6876812968711286, + "tokens_seen": 896598016 + }, + { + "epoch": 0.27, + "learning_rate": 0.000367863590772317, + "loss": 1.4043, + "theoretical_loss": 3.6876551260590498, + "tokens_seen": 896663552 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036785356068204616, + "loss": 1.4485, + "theoretical_loss": 3.6876289576952295, + "tokens_seen": 896729088 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036784353059177535, + "loss": 1.402, + "theoretical_loss": 3.68760279177926, + "tokens_seen": 896794624 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003678335005015045, + "loss": 1.3582, + "theoretical_loss": 3.6875766283107327, + "tokens_seen": 896860160 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003678234704112337, + "loss": 1.3737, + "theoretical_loss": 3.6875504672892405, + "tokens_seen": 896925696 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036781344032096294, + "loss": 1.4142, + "theoretical_loss": 3.6875243087143756, + "tokens_seen": 896991232 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036780341023069207, + "loss": 1.4694, + "theoretical_loss": 3.687498152585731, + "tokens_seen": 897056768 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003677933801404213, + "loss": 1.3816, + "theoretical_loss": 3.687471998902898, + "tokens_seen": 897122304 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036778335005015043, + "loss": 1.3746, + "theoretical_loss": 3.6874458476654706, + "tokens_seen": 897187840 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036777331995987967, + "loss": 1.3075, + "theoretical_loss": 3.6874196988730406, + "tokens_seen": 897253376 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036776328986960885, + "loss": 1.4331, + "theoretical_loss": 3.6873935525252017, + "tokens_seen": 897318912 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036775325977933803, + "loss": 1.3758, + "theoretical_loss": 3.6873674086215456, + "tokens_seen": 897384448 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003677432296890672, + "loss": 1.4823, + "theoretical_loss": 3.687341267161667, + "tokens_seen": 897449984 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036773319959879645, + "loss": 1.3713, + "theoretical_loss": 3.6873151281451575, + "tokens_seen": 897515520 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036772316950852557, + "loss": 1.4265, + "theoretical_loss": 3.6872889915716107, + "tokens_seen": 897581056 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003677131394182548, + "loss": 1.3707, + "theoretical_loss": 3.6872628574406208, + "tokens_seen": 897646592 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036770310932798393, + "loss": 1.4206, + "theoretical_loss": 3.68723672575178, + "tokens_seen": 897712128 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036769307923771317, + "loss": 1.3459, + "theoretical_loss": 3.687210596504683, + "tokens_seen": 897777664 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0002606348425615579, + "objective/train/docs_used": 515835, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7933313846588135, + "objective/train/original_loss": 2.7933316230773926, + "objective/train/theoretical_loss": 3.6871844696989227, + "objective/train/tokens_used": 918303200, + "objective/train/value_avg": -0.004302978515625, + "objective/train/value_loss": 3.2565247238380834e-05, + "objective/train/value_max": -0.0002551078796386719, + "objective/train/value_min": -0.0390625, + "objective/train/value_reward_corr": 0.319373273401145, + "objective/train/value_std": 0.003063201904296875, + "objective/train/weight_avg": 0.9989808797836304, + "objective/train/weighted_lm_loss": 2.796635627746582, + "objective/train/weights_max": 1.3676658868789673, + "objective/train/weights_min": 0.6958411931991577, + "theoretical_loss": 3.6871844696989227, + "tokens_seen": 897843200 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036768304914744235, + "loss": 1.3697, + "theoretical_loss": 3.6871844696989227, + "tokens_seen": 897843200 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036767301905717153, + "loss": 1.4173, + "theoretical_loss": 3.687158345334093, + "tokens_seen": 897908736 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003676629889669007, + "loss": 1.3679, + "theoretical_loss": 3.6871322234097876, + "tokens_seen": 897974272 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003676529588766299, + "loss": 1.4639, + "theoretical_loss": 3.6871061039256006, + "tokens_seen": 898039808 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003676429287863591, + "loss": 1.3683, + "theoretical_loss": 3.687079986881126, + "tokens_seen": 898105344 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003676328986960883, + "loss": 1.2791, + "theoretical_loss": 3.687053872275958, + "tokens_seen": 898170880 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036762286860581744, + "loss": 1.4359, + "theoretical_loss": 3.6870277601096904, + "tokens_seen": 898236416 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003676128385155467, + "loss": 1.4097, + "theoretical_loss": 3.6870016503819176, + "tokens_seen": 898301952 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003676028084252758, + "loss": 1.4527, + "theoretical_loss": 3.6869755430922346, + "tokens_seen": 898367488 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036759277833500504, + "loss": 1.2967, + "theoretical_loss": 3.6869494382402355, + "tokens_seen": 898433024 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003675827482447342, + "loss": 1.405, + "theoretical_loss": 3.6869233358255147, + "tokens_seen": 898498560 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003675727181544634, + "loss": 1.364, + "theoretical_loss": 3.686897235847667, + "tokens_seen": 898564096 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003675626880641926, + "loss": 1.4714, + "theoretical_loss": 3.6868711383062873, + "tokens_seen": 898629632 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003675526579739218, + "loss": 1.3865, + "theoretical_loss": 3.6868450432009707, + "tokens_seen": 898695168 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036754262788365094, + "loss": 1.2634, + "theoretical_loss": 3.6868189505313116, + "tokens_seen": 898760704 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003675325977933802, + "loss": 1.4434, + "theoretical_loss": 3.686792860296906, + "tokens_seen": 898826240 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003675225677031093, + "loss": 1.412, + "theoretical_loss": 3.686766772497348, + "tokens_seen": 898891776 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036751253761283854, + "loss": 1.3675, + "theoretical_loss": 3.6867406871322332, + "tokens_seen": 898957312 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003675025075225677, + "loss": 1.3643, + "theoretical_loss": 3.686714604201158, + "tokens_seen": 899022848 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003674924774322969, + "loss": 1.4179, + "theoretical_loss": 3.686688523703716, + "tokens_seen": 899088384 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003674824473420261, + "loss": 1.3273, + "theoretical_loss": 3.686662445639504, + "tokens_seen": 899153920 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036747241725175526, + "loss": 1.418, + "theoretical_loss": 3.686636370008118, + "tokens_seen": 899219456 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036746238716148444, + "loss": 1.31, + "theoretical_loss": 3.686610296809153, + "tokens_seen": 899284992 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003674523570712137, + "loss": 1.4516, + "theoretical_loss": 3.6865842260422053, + "tokens_seen": 899350528 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003674423269809428, + "loss": 1.3771, + "theoretical_loss": 3.68655815770687, + "tokens_seen": 899416064 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": 0.00044684053864330053, + "objective/train/docs_used": 516375, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.194222927093506, + "objective/train/original_loss": 2.194223165512085, + "objective/train/theoretical_loss": 3.6865320918027438, + "objective/train/tokens_used": 919941600, + "objective/train/value_avg": -0.00455474853515625, + "objective/train/value_loss": 3.9751223084749654e-05, + "objective/train/value_max": -0.0004897117614746094, + "objective/train/value_min": -0.06243896484375, + "objective/train/value_reward_corr": 0.22132080882295357, + "objective/train/value_std": 0.0037689208984375, + "objective/train/weight_avg": 1.006399393081665, + "objective/train/weighted_lm_loss": 2.2063119411468506, + "objective/train/weights_max": 1.7426800727844238, + "objective/train/weights_min": 0.5018414258956909, + "theoretical_loss": 3.6865320918027438, + "tokens_seen": 899481600 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036743229689067204, + "loss": 1.3782, + "theoretical_loss": 3.6865320918027438, + "tokens_seen": 899481600 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036742226680040117, + "loss": 1.2547, + "theoretical_loss": 3.6865060283294238, + "tokens_seen": 899547136 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003674122367101304, + "loss": 1.459, + "theoretical_loss": 3.686479967286504, + "tokens_seen": 899612672 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003674022066198596, + "loss": 1.4035, + "theoretical_loss": 3.686453908673583, + "tokens_seen": 899678208 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036739217652958877, + "loss": 1.4279, + "theoretical_loss": 3.6864278524902554, + "tokens_seen": 899743744 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036738214643931795, + "loss": 1.3437, + "theoretical_loss": 3.686401798736119, + "tokens_seen": 899809280 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003673721163490472, + "loss": 1.4334, + "theoretical_loss": 3.6863757474107697, + "tokens_seen": 899874816 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003673620862587763, + "loss": 1.3235, + "theoretical_loss": 3.686349698513805, + "tokens_seen": 899940352 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036735205616850555, + "loss": 1.5522, + "theoretical_loss": 3.686323652044821, + "tokens_seen": 900005888 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036734202607823467, + "loss": 1.3166, + "theoretical_loss": 3.686297608003415, + "tokens_seen": 900071424 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003673319959879639, + "loss": 1.3019, + "theoretical_loss": 3.6862715663891836, + "tokens_seen": 900136960 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003673219658976931, + "loss": 1.4524, + "theoretical_loss": 3.6862455272017245, + "tokens_seen": 900202496 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036731193580742227, + "loss": 1.4363, + "theoretical_loss": 3.6862194904406342, + "tokens_seen": 900268032 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036730190571715145, + "loss": 1.4424, + "theoretical_loss": 3.6861934561055105, + "tokens_seen": 900333568 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036729187562688063, + "loss": 1.4071, + "theoretical_loss": 3.6861674241959506, + "tokens_seen": 900399104 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003672818455366098, + "loss": 1.4225, + "theoretical_loss": 3.6861413947115524, + "tokens_seen": 900464640 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036727181544633905, + "loss": 1.4126, + "theoretical_loss": 3.686115367651913, + "tokens_seen": 900530176 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003672617853560682, + "loss": 1.2709, + "theoretical_loss": 3.68608934301663, + "tokens_seen": 900595712 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003672517552657974, + "loss": 1.5009, + "theoretical_loss": 3.686063320805302, + "tokens_seen": 900661248 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036724172517552654, + "loss": 1.436, + "theoretical_loss": 3.6860373010175262, + "tokens_seen": 900726784 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036723169508525577, + "loss": 1.3329, + "theoretical_loss": 3.6860112836529, + "tokens_seen": 900792320 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036722166499498495, + "loss": 1.4575, + "theoretical_loss": 3.685985268711023, + "tokens_seen": 900857856 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036721163490471414, + "loss": 1.4408, + "theoretical_loss": 3.685959256191492, + "tokens_seen": 900923392 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003672016048144433, + "loss": 1.4737, + "theoretical_loss": 3.685933246093906, + "tokens_seen": 900988928 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036719157472417255, + "loss": 1.3583, + "theoretical_loss": 3.685907238417863, + "tokens_seen": 901054464 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0008443820988759398, + "objective/train/docs_used": 517599, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.62571382522583, + "objective/train/original_loss": 2.625714063644409, + "objective/train/theoretical_loss": 3.685881233162962, + "objective/train/tokens_used": 921580000, + "objective/train/value_avg": -0.004978179931640625, + "objective/train/value_loss": 4.0283906855620444e-05, + "objective/train/value_max": -0.00031495094299316406, + "objective/train/value_min": -0.04296875, + "objective/train/value_reward_corr": 0.2702817531269597, + "objective/train/value_std": 0.003326416015625, + "objective/train/weight_avg": 0.9934871792793274, + "objective/train/weighted_lm_loss": 2.619575023651123, + "objective/train/weights_max": 1.3751686811447144, + "objective/train/weights_min": 0.4460531771183014, + "theoretical_loss": 3.685881233162962, + "tokens_seen": 901120000 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003671815446339017, + "loss": 1.451, + "theoretical_loss": 3.685881233162962, + "tokens_seen": 901120000 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003671715145436309, + "loss": 1.3217, + "theoretical_loss": 3.6858552303288006, + "tokens_seen": 901185536 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036716148445336004, + "loss": 1.5272, + "theoretical_loss": 3.6858292299149786, + "tokens_seen": 901251072 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003671514543630893, + "loss": 1.5228, + "theoretical_loss": 3.685803231921094, + "tokens_seen": 901316608 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036714142427281846, + "loss": 1.1859, + "theoretical_loss": 3.6857772363467447, + "tokens_seen": 901382144 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036713139418254764, + "loss": 1.4324, + "theoretical_loss": 3.685751243191532, + "tokens_seen": 901447680 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003671213640922768, + "loss": 1.3848, + "theoretical_loss": 3.685725252455053, + "tokens_seen": 901513216 + }, + { + "epoch": 0.27, + "learning_rate": 0.000367111334002006, + "loss": 1.4022, + "theoretical_loss": 3.6856992641369075, + "tokens_seen": 901578752 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036710130391173524, + "loss": 1.3484, + "theoretical_loss": 3.685673278236695, + "tokens_seen": 901644288 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003670912738214644, + "loss": 1.3902, + "theoretical_loss": 3.6856472947540144, + "tokens_seen": 901709824 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003670812437311936, + "loss": 1.2906, + "theoretical_loss": 3.685621313688465, + "tokens_seen": 901775360 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003670712136409228, + "loss": 1.4268, + "theoretical_loss": 3.685595335039647, + "tokens_seen": 901840896 + }, + { + "epoch": 0.27, + "learning_rate": 0.000367061183550652, + "loss": 1.378, + "theoretical_loss": 3.6855693588071587, + "tokens_seen": 901906432 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036705115346038114, + "loss": 1.3301, + "theoretical_loss": 3.685543384990601, + "tokens_seen": 901971968 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003670411233701104, + "loss": 1.3466, + "theoretical_loss": 3.685517413589573, + "tokens_seen": 902037504 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003670310932798395, + "loss": 1.3645, + "theoretical_loss": 3.6854914446036755, + "tokens_seen": 902103040 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036702106318956874, + "loss": 1.3921, + "theoretical_loss": 3.6854654780325076, + "tokens_seen": 902168576 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003670110330992979, + "loss": 1.4371, + "theoretical_loss": 3.6854395138756693, + "tokens_seen": 902234112 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003670010030090271, + "loss": 1.4766, + "theoretical_loss": 3.6854135521327613, + "tokens_seen": 902299648 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003669909729187563, + "loss": 1.4842, + "theoretical_loss": 3.6853875928033837, + "tokens_seen": 902365184 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036698094282848546, + "loss": 1.4266, + "theoretical_loss": 3.685361635887137, + "tokens_seen": 902430720 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036697091273821464, + "loss": 1.4081, + "theoretical_loss": 3.6853356813836218, + "tokens_seen": 902496256 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003669608826479439, + "loss": 1.346, + "theoretical_loss": 3.6853097292924377, + "tokens_seen": 902561792 + }, + { + "epoch": 0.27, + "learning_rate": 0.000366950852557673, + "loss": 1.4296, + "theoretical_loss": 3.6852837796131865, + "tokens_seen": 902627328 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036694082246740224, + "loss": 1.2686, + "theoretical_loss": 3.685257832345468, + "tokens_seen": 902692864 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0011622003512457013, + "objective/train/docs_used": 518319, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.810044288635254, + "objective/train/original_loss": 2.8100435733795166, + "objective/train/theoretical_loss": 3.6852318874888836, + "objective/train/tokens_used": 923218400, + "objective/train/value_avg": -0.0052490234375, + "objective/train/value_loss": 8.044153946684673e-05, + "objective/train/value_max": -0.00018668174743652344, + "objective/train/value_min": -0.058990478515625, + "objective/train/value_reward_corr": 0.3439786251242787, + "objective/train/value_std": 0.004253387451171875, + "objective/train/weight_avg": 0.9920457005500793, + "objective/train/weighted_lm_loss": 2.7909932136535645, + "objective/train/weights_max": 1.6813820600509644, + "objective/train/weights_min": 0.338327556848526, + "theoretical_loss": 3.6852318874888836, + "tokens_seen": 902758400 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036693079237713137, + "loss": 1.3567, + "theoretical_loss": 3.6852318874888836, + "tokens_seen": 902758400 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003669207622868606, + "loss": 1.3894, + "theoretical_loss": 3.6852059450430343, + "tokens_seen": 902823936 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003669107321965898, + "loss": 1.365, + "theoretical_loss": 3.685180005007521, + "tokens_seen": 902889472 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036690070210631897, + "loss": 1.3963, + "theoretical_loss": 3.6851540673819443, + "tokens_seen": 902955008 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036689067201604815, + "loss": 1.3982, + "theoretical_loss": 3.6851281321659064, + "tokens_seen": 903020544 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003668806419257774, + "loss": 1.2203, + "theoretical_loss": 3.685102199359008, + "tokens_seen": 903086080 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003668706118355065, + "loss": 1.3649, + "theoretical_loss": 3.6850762689608505, + "tokens_seen": 903151616 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036686058174523575, + "loss": 1.3942, + "theoretical_loss": 3.685050340971036, + "tokens_seen": 903217152 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036685055165496487, + "loss": 1.3801, + "theoretical_loss": 3.685024415389165, + "tokens_seen": 903282688 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003668405215646941, + "loss": 1.3938, + "theoretical_loss": 3.6849984922148398, + "tokens_seen": 903348224 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003668304914744233, + "loss": 1.4687, + "theoretical_loss": 3.6849725714476627, + "tokens_seen": 903413760 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036682046138415247, + "loss": 1.3717, + "theoretical_loss": 3.684946653087235, + "tokens_seen": 903479296 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036681043129388165, + "loss": 1.426, + "theoretical_loss": 3.6849207371331585, + "tokens_seen": 903544832 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036680040120361083, + "loss": 1.2829, + "theoretical_loss": 3.6848948235850356, + "tokens_seen": 903610368 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036679037111334, + "loss": 1.4223, + "theoretical_loss": 3.6848689124424685, + "tokens_seen": 903675904 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036678034102306925, + "loss": 1.3585, + "theoretical_loss": 3.6848430037050592, + "tokens_seen": 903741440 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003667703109327984, + "loss": 1.4725, + "theoretical_loss": 3.6848170973724104, + "tokens_seen": 903806976 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003667602808425276, + "loss": 1.3228, + "theoretical_loss": 3.6847911934441244, + "tokens_seen": 903872512 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036675025075225674, + "loss": 1.3146, + "theoretical_loss": 3.6847652919198035, + "tokens_seen": 903938048 + }, + { + "epoch": 0.27, + "learning_rate": 0.000366740220661986, + "loss": 1.4295, + "theoretical_loss": 3.6847393927990506, + "tokens_seen": 904003584 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036673019057171515, + "loss": 1.3031, + "theoretical_loss": 3.6847134960814687, + "tokens_seen": 904069120 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036672016048144434, + "loss": 1.3159, + "theoretical_loss": 3.6846876017666603, + "tokens_seen": 904134656 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003667101303911735, + "loss": 1.3291, + "theoretical_loss": 3.684661709854228, + "tokens_seen": 904200192 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036670010030090275, + "loss": 1.2766, + "theoretical_loss": 3.6846358203437752, + "tokens_seen": 904265728 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003666900702106319, + "loss": 1.3889, + "theoretical_loss": 3.6846099332349054, + "tokens_seen": 904331264 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0003405045426916331, + "objective/train/docs_used": 519704, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9578680992126465, + "objective/train/original_loss": 2.9578680992126465, + "objective/train/theoretical_loss": 3.6845840485272205, + "objective/train/tokens_used": 924856800, + "objective/train/value_avg": -0.006298065185546875, + "objective/train/value_loss": 6.851766374893486e-05, + "objective/train/value_max": -0.000469207763671875, + "objective/train/value_min": -0.09722900390625, + "objective/train/value_reward_corr": 0.4565725585131605, + "objective/train/value_std": 0.005023956298828125, + "objective/train/weight_avg": 0.9997472167015076, + "objective/train/weighted_lm_loss": 2.9480154514312744, + "objective/train/weights_max": 2.159465789794922, + "objective/train/weights_min": 0.08644936233758926, + "theoretical_loss": 3.6845840485272205, + "tokens_seen": 904396800 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003666800401203611, + "loss": 1.4705, + "theoretical_loss": 3.6845840485272205, + "tokens_seen": 904396800 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036667001003009024, + "loss": 1.2965, + "theoretical_loss": 3.6845581662203255, + "tokens_seen": 904462336 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003666599799398195, + "loss": 1.3975, + "theoretical_loss": 3.6845322863138232, + "tokens_seen": 904527872 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036664994984954866, + "loss": 1.3763, + "theoretical_loss": 3.6845064088073163, + "tokens_seen": 904593408 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036663991975927784, + "loss": 1.306, + "theoretical_loss": 3.684480533700409, + "tokens_seen": 904658944 + }, + { + "epoch": 0.27, + "learning_rate": 0.000366629889669007, + "loss": 1.4302, + "theoretical_loss": 3.6844546609927047, + "tokens_seen": 904724480 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003666198595787362, + "loss": 1.4195, + "theoretical_loss": 3.684428790683808, + "tokens_seen": 904790016 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003666098294884654, + "loss": 1.3253, + "theoretical_loss": 3.684402922773322, + "tokens_seen": 904855552 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003665997993981946, + "loss": 1.2909, + "theoretical_loss": 3.6843770572608507, + "tokens_seen": 904921088 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036658976930792374, + "loss": 1.2346, + "theoretical_loss": 3.6843511941459983, + "tokens_seen": 904986624 + }, + { + "epoch": 0.27, + "learning_rate": 0.000366579739217653, + "loss": 1.4975, + "theoretical_loss": 3.6843253334283688, + "tokens_seen": 905052160 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036656970912738216, + "loss": 1.2943, + "theoretical_loss": 3.684299475107567, + "tokens_seen": 905117696 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036655967903711134, + "loss": 1.4278, + "theoretical_loss": 3.684273619183196, + "tokens_seen": 905183232 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003665496489468405, + "loss": 1.3446, + "theoretical_loss": 3.684247765654862, + "tokens_seen": 905248768 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003665396188565697, + "loss": 1.3125, + "theoretical_loss": 3.6842219145221677, + "tokens_seen": 905314304 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003665295887662989, + "loss": 1.3738, + "theoretical_loss": 3.684196065784719, + "tokens_seen": 905379840 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003665195586760281, + "loss": 1.3488, + "theoretical_loss": 3.6841702194421204, + "tokens_seen": 905445376 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036650952858575725, + "loss": 1.3956, + "theoretical_loss": 3.684144375493976, + "tokens_seen": 905510912 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003664994984954865, + "loss": 1.3637, + "theoretical_loss": 3.684118533939891, + "tokens_seen": 905576448 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003664894684052156, + "loss": 1.439, + "theoretical_loss": 3.6840926947794714, + "tokens_seen": 905641984 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036647943831494484, + "loss": 1.4274, + "theoretical_loss": 3.6840668580123213, + "tokens_seen": 905707520 + }, + { + "epoch": 0.27, + "learning_rate": 0.000366469408224674, + "loss": 1.4921, + "theoretical_loss": 3.684041023638046, + "tokens_seen": 905773056 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003664593781344032, + "loss": 1.3201, + "theoretical_loss": 3.6840151916562505, + "tokens_seen": 905838592 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003664493480441324, + "loss": 1.3809, + "theoretical_loss": 3.6839893620665407, + "tokens_seen": 905904128 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036643931795386157, + "loss": 1.3702, + "theoretical_loss": 3.6839635348685222, + "tokens_seen": 905969664 + }, + { + "epoch": 0.27, + "objective/train/advantage_avg": -0.0005605893675237894, + "objective/train/docs_used": 520483, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 1.9179019927978516, + "objective/train/original_loss": 1.917901873588562, + "objective/train/theoretical_loss": 3.6839377100618, + "objective/train/tokens_used": 926495200, + "objective/train/value_avg": -0.004581451416015625, + "objective/train/value_loss": 3.8593010685872287e-05, + "objective/train/value_max": -0.0002472400665283203, + "objective/train/value_min": -0.04638671875, + "objective/train/value_reward_corr": 0.32689074670501495, + "objective/train/value_std": 0.0035839080810546875, + "objective/train/weight_avg": 0.996265172958374, + "objective/train/weighted_lm_loss": 1.9182608127593994, + "objective/train/weights_max": 1.447691798210144, + "objective/train/weights_min": 0.6098842024803162, + "theoretical_loss": 3.6839377100618, + "tokens_seen": 906035200 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036642928786359075, + "loss": 1.3433, + "theoretical_loss": 3.6839377100618, + "tokens_seen": 906035200 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036641925777332, + "loss": 1.328, + "theoretical_loss": 3.6839118876459795, + "tokens_seen": 906100736 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003664092276830491, + "loss": 1.3556, + "theoretical_loss": 3.6838860676206675, + "tokens_seen": 906166272 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036639919759277835, + "loss": 1.3342, + "theoretical_loss": 3.683860249985469, + "tokens_seen": 906231808 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036638916750250753, + "loss": 1.2572, + "theoretical_loss": 3.68383443473999, + "tokens_seen": 906297344 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003663791374122367, + "loss": 1.4696, + "theoretical_loss": 3.683808621883837, + "tokens_seen": 906362880 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003663691073219659, + "loss": 1.3756, + "theoretical_loss": 3.6837828114166156, + "tokens_seen": 906428416 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036635907723169507, + "loss": 1.3914, + "theoretical_loss": 3.683757003337932, + "tokens_seen": 906493952 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003663490471414243, + "loss": 1.3258, + "theoretical_loss": 3.6837311976473934, + "tokens_seen": 906559488 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003663390170511535, + "loss": 1.3585, + "theoretical_loss": 3.6837053943446048, + "tokens_seen": 906625024 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036632898696088267, + "loss": 1.4118, + "theoretical_loss": 3.683679593429174, + "tokens_seen": 906690560 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036631895687061185, + "loss": 1.2991, + "theoretical_loss": 3.6836537949007067, + "tokens_seen": 906756096 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036630892678034103, + "loss": 1.407, + "theoretical_loss": 3.6836279987588103, + "tokens_seen": 906821632 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003662988966900702, + "loss": 1.3724, + "theoretical_loss": 3.6836022050030905, + "tokens_seen": 906887168 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036628886659979945, + "loss": 1.3876, + "theoretical_loss": 3.683576413633155, + "tokens_seen": 906952704 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003662788365095286, + "loss": 1.4, + "theoretical_loss": 3.6835506246486105, + "tokens_seen": 907018240 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003662688064192578, + "loss": 1.4988, + "theoretical_loss": 3.6835248380490646, + "tokens_seen": 907083776 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036625877632898694, + "loss": 1.3658, + "theoretical_loss": 3.6834990538341237, + "tokens_seen": 907149312 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003662487462387162, + "loss": 1.4726, + "theoretical_loss": 3.683473272003395, + "tokens_seen": 907214848 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036623871614844535, + "loss": 1.322, + "theoretical_loss": 3.6834474925564864, + "tokens_seen": 907280384 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036622868605817454, + "loss": 1.5176, + "theoretical_loss": 3.683421715493005, + "tokens_seen": 907345920 + }, + { + "epoch": 0.27, + "learning_rate": 0.0003662186559679037, + "loss": 1.4393, + "theoretical_loss": 3.683395940812558, + "tokens_seen": 907411456 + }, + { + "epoch": 0.27, + "learning_rate": 0.00036620862587763295, + "loss": 1.3755, + "theoretical_loss": 3.683370168514754, + "tokens_seen": 907476992 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003661985957873621, + "loss": 1.4165, + "theoretical_loss": 3.6833443985992, + "tokens_seen": 907542528 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003661885656970913, + "loss": 1.4422, + "theoretical_loss": 3.6833186310655033, + "tokens_seen": 907608064 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.0004232425708323717, + "objective/train/docs_used": 521740, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.548290729522705, + "objective/train/original_loss": 2.5482912063598633, + "objective/train/theoretical_loss": 3.6832928659132724, + "objective/train/tokens_used": 928133600, + "objective/train/value_avg": -0.004558563232421875, + "objective/train/value_loss": 4.6149129047989845e-05, + "objective/train/value_max": -0.0003726482391357422, + "objective/train/value_min": -0.0535888671875, + "objective/train/value_reward_corr": 0.3056799973016389, + "objective/train/value_std": 0.0035800933837890625, + "objective/train/weight_avg": 0.9979376792907715, + "objective/train/weighted_lm_loss": 2.5474066734313965, + "objective/train/weights_max": 1.61049485206604, + "objective/train/weights_min": 0.3137578070163727, + "theoretical_loss": 3.6832928659132724, + "tokens_seen": 907673600 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036617853560682044, + "loss": 1.3549, + "theoretical_loss": 3.6832928659132724, + "tokens_seen": 907673600 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003661685055165497, + "loss": 1.4125, + "theoretical_loss": 3.6832671031421156, + "tokens_seen": 907739136 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036615847542627886, + "loss": 1.4346, + "theoretical_loss": 3.6832413427516406, + "tokens_seen": 907804672 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036614844533600804, + "loss": 1.3111, + "theoretical_loss": 3.683215584741455, + "tokens_seen": 907870208 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003661384152457372, + "loss": 1.376, + "theoretical_loss": 3.683189829111168, + "tokens_seen": 907935744 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003661283851554664, + "loss": 1.3952, + "theoretical_loss": 3.683164075860388, + "tokens_seen": 908001280 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003661183550651956, + "loss": 1.3585, + "theoretical_loss": 3.6831383249887226, + "tokens_seen": 908066816 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003661083249749248, + "loss": 1.4264, + "theoretical_loss": 3.6831125764957804, + "tokens_seen": 908132352 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036609829488465394, + "loss": 1.3336, + "theoretical_loss": 3.683086830381171, + "tokens_seen": 908197888 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003660882647943832, + "loss": 1.3267, + "theoretical_loss": 3.6830610866445017, + "tokens_seen": 908263424 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036607823470411236, + "loss": 1.426, + "theoretical_loss": 3.683035345285383, + "tokens_seen": 908328960 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036606820461384154, + "loss": 1.4076, + "theoretical_loss": 3.6830096063034223, + "tokens_seen": 908394496 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003660581745235707, + "loss": 1.4833, + "theoretical_loss": 3.682983869698229, + "tokens_seen": 908460032 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003660481444332999, + "loss": 1.3572, + "theoretical_loss": 3.6829581354694128, + "tokens_seen": 908525568 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003660381143430291, + "loss": 1.4288, + "theoretical_loss": 3.682932403616583, + "tokens_seen": 908591104 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003660280842527583, + "loss": 1.2648, + "theoretical_loss": 3.6829066741393475, + "tokens_seen": 908656640 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036601805416248745, + "loss": 1.3769, + "theoretical_loss": 3.682880947037317, + "tokens_seen": 908722176 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003660080240722167, + "loss": 1.4439, + "theoretical_loss": 3.6828552223101, + "tokens_seen": 908787712 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003659979939819458, + "loss": 1.4687, + "theoretical_loss": 3.6828294999573075, + "tokens_seen": 908853248 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036598796389167504, + "loss": 1.518, + "theoretical_loss": 3.6828037799785474, + "tokens_seen": 908918784 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003659779338014042, + "loss": 1.3901, + "theoretical_loss": 3.68277806237343, + "tokens_seen": 908984320 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003659679037111334, + "loss": 1.3091, + "theoretical_loss": 3.682752347141566, + "tokens_seen": 909049856 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003659578736208626, + "loss": 1.3637, + "theoretical_loss": 3.682726634282564, + "tokens_seen": 909115392 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036594784353059177, + "loss": 1.3328, + "theoretical_loss": 3.6827009237960353, + "tokens_seen": 909180928 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036593781344032095, + "loss": 1.3793, + "theoretical_loss": 3.682675215681589, + "tokens_seen": 909246464 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": 0.0001747280330164358, + "objective/train/docs_used": 522301, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.707522392272949, + "objective/train/original_loss": 2.707521915435791, + "objective/train/theoretical_loss": 3.682649509938836, + "objective/train/tokens_used": 929772000, + "objective/train/value_avg": -0.004856109619140625, + "objective/train/value_loss": 5.235121352598071e-05, + "objective/train/value_max": -0.0004107952117919922, + "objective/train/value_min": -0.2158203125, + "objective/train/value_reward_corr": 0.29825574920864695, + "objective/train/value_std": 0.004245758056640625, + "objective/train/weight_avg": 1.0044596195220947, + "objective/train/weighted_lm_loss": 2.7134103775024414, + "objective/train/weights_max": 8.655570983886719, + "objective/train/weights_min": 0.4356282949447632, + "theoretical_loss": 3.682649509938836, + "tokens_seen": 909312000 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003659277833500502, + "loss": 1.3928, + "theoretical_loss": 3.682649509938836, + "tokens_seen": 909312000 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003659177532597793, + "loss": 1.3213, + "theoretical_loss": 3.6826238065673866, + "tokens_seen": 909377536 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036590772316950855, + "loss": 1.3032, + "theoretical_loss": 3.68259810556685, + "tokens_seen": 909443072 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036589769307923773, + "loss": 1.4417, + "theoretical_loss": 3.6825724069368384, + "tokens_seen": 909508608 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003658876629889669, + "loss": 1.4428, + "theoretical_loss": 3.6825467106769607, + "tokens_seen": 909574144 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003658776328986961, + "loss": 1.4411, + "theoretical_loss": 3.682521016786829, + "tokens_seen": 909639680 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036586760280842527, + "loss": 1.3498, + "theoretical_loss": 3.682495325266053, + "tokens_seen": 909705216 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036585757271815445, + "loss": 1.3522, + "theoretical_loss": 3.682469636114244, + "tokens_seen": 909770752 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003658475426278837, + "loss": 1.3365, + "theoretical_loss": 3.6824439493310135, + "tokens_seen": 909836288 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003658375125376128, + "loss": 1.4051, + "theoretical_loss": 3.682418264915971, + "tokens_seen": 909901824 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036582748244734205, + "loss": 1.4728, + "theoretical_loss": 3.6823925828687294, + "tokens_seen": 909967360 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003658174523570712, + "loss": 1.2601, + "theoretical_loss": 3.682366903188899, + "tokens_seen": 910032896 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003658074222668004, + "loss": 1.5202, + "theoretical_loss": 3.6823412258760904, + "tokens_seen": 910098432 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003657973921765296, + "loss": 1.3763, + "theoretical_loss": 3.682315550929917, + "tokens_seen": 910163968 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003657873620862588, + "loss": 1.5098, + "theoretical_loss": 3.6822898783499873, + "tokens_seen": 910229504 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036577733199598796, + "loss": 1.3888, + "theoretical_loss": 3.682264208135916, + "tokens_seen": 910295040 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036576730190571714, + "loss": 1.2745, + "theoretical_loss": 3.682238540287313, + "tokens_seen": 910360576 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003657572718154463, + "loss": 1.3237, + "theoretical_loss": 3.6822128748037906, + "tokens_seen": 910426112 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036574724172517555, + "loss": 1.4431, + "theoretical_loss": 3.68218721168496, + "tokens_seen": 910491648 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003657372116349047, + "loss": 1.4499, + "theoretical_loss": 3.6821615509304344, + "tokens_seen": 910557184 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003657271815446339, + "loss": 1.4708, + "theoretical_loss": 3.682135892539825, + "tokens_seen": 910622720 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003657171514543631, + "loss": 1.2436, + "theoretical_loss": 3.6821102365127434, + "tokens_seen": 910688256 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003657071213640923, + "loss": 1.3982, + "theoretical_loss": 3.6820845828488027, + "tokens_seen": 910753792 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036569709127382146, + "loss": 1.5226, + "theoretical_loss": 3.6820589315476147, + "tokens_seen": 910819328 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036568706118355064, + "loss": 1.4406, + "theoretical_loss": 3.6820332826087925, + "tokens_seen": 910884864 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.0011136906687170267, + "objective/train/docs_used": 523753, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7234208583831787, + "objective/train/original_loss": 2.723421335220337, + "objective/train/theoretical_loss": 3.6820076360319485, + "objective/train/tokens_used": 931410400, + "objective/train/value_avg": -0.00484466552734375, + "objective/train/value_loss": 8.61793159856461e-05, + "objective/train/value_max": -0.000507354736328125, + "objective/train/value_min": -0.06622314453125, + "objective/train/value_reward_corr": 0.4316888484894328, + "objective/train/value_std": 0.0037517547607421875, + "objective/train/weight_avg": 0.9927460551261902, + "objective/train/weighted_lm_loss": 2.729816198348999, + "objective/train/weights_max": 1.477904200553894, + "objective/train/weights_min": 0.20917470753192902, + "theoretical_loss": 3.6820076360319485, + "tokens_seen": 910950400 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003656770310932798, + "loss": 1.4352, + "theoretical_loss": 3.6820076360319485, + "tokens_seen": 910950400 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036566700100300906, + "loss": 1.445, + "theoretical_loss": 3.681981991816694, + "tokens_seen": 911015936 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003656569709127382, + "loss": 1.4183, + "theoretical_loss": 3.6819563499626433, + "tokens_seen": 911081472 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003656469408224674, + "loss": 1.5123, + "theoretical_loss": 3.681930710469408, + "tokens_seen": 911147008 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036563691073219655, + "loss": 1.3821, + "theoretical_loss": 3.6819050733366017, + "tokens_seen": 911212544 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003656268806419258, + "loss": 1.4239, + "theoretical_loss": 3.681879438563837, + "tokens_seen": 911278080 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036561685055165496, + "loss": 1.3211, + "theoretical_loss": 3.681853806150727, + "tokens_seen": 911343616 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036560682046138414, + "loss": 1.4499, + "theoretical_loss": 3.681828176096885, + "tokens_seen": 911409152 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003655967903711134, + "loss": 1.3259, + "theoretical_loss": 3.681802548401924, + "tokens_seen": 911474688 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036558676028084256, + "loss": 1.3669, + "theoretical_loss": 3.681776923065457, + "tokens_seen": 911540224 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036557673019057174, + "loss": 1.4165, + "theoretical_loss": 3.6817513000870985, + "tokens_seen": 911605760 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003655667001003009, + "loss": 1.4309, + "theoretical_loss": 3.681725679466461, + "tokens_seen": 911671296 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003655566700100301, + "loss": 1.3304, + "theoretical_loss": 3.6817000612031583, + "tokens_seen": 911736832 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003655466399197593, + "loss": 1.4495, + "theoretical_loss": 3.681674445296804, + "tokens_seen": 911802368 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003655366098294885, + "loss": 1.3689, + "theoretical_loss": 3.6816488317470126, + "tokens_seen": 911867904 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036552657973921765, + "loss": 1.368, + "theoretical_loss": 3.681623220553397, + "tokens_seen": 911933440 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003655165496489469, + "loss": 1.3823, + "theoretical_loss": 3.6815976117155715, + "tokens_seen": 911998976 + }, + { + "epoch": 0.28, + "learning_rate": 0.000365506519558676, + "loss": 1.3478, + "theoretical_loss": 3.681572005233151, + "tokens_seen": 912064512 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036549648946840525, + "loss": 1.4985, + "theoretical_loss": 3.6815464011057477, + "tokens_seen": 912130048 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003654864593781344, + "loss": 1.3246, + "theoretical_loss": 3.681520799332978, + "tokens_seen": 912195584 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003654764292878636, + "loss": 1.3872, + "theoretical_loss": 3.6814951999144547, + "tokens_seen": 912261120 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003654663991975928, + "loss": 1.4478, + "theoretical_loss": 3.681469602849793, + "tokens_seen": 912326656 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036545636910732197, + "loss": 1.3491, + "theoretical_loss": 3.681444008138606, + "tokens_seen": 912392192 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036544633901705115, + "loss": 1.403, + "theoretical_loss": 3.6814184157805108, + "tokens_seen": 912457728 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003654363089267804, + "loss": 1.4053, + "theoretical_loss": 3.68139282577512, + "tokens_seen": 912523264 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.0001744900073390454, + "objective/train/docs_used": 524575, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.722566604614258, + "objective/train/original_loss": 2.7225663661956787, + "objective/train/theoretical_loss": 3.681367238122049, + "objective/train/tokens_used": 933048800, + "objective/train/value_avg": -0.005481719970703125, + "objective/train/value_loss": 5.847991269547492e-05, + "objective/train/value_max": -0.00030541419982910156, + "objective/train/value_min": -0.06915283203125, + "objective/train/value_reward_corr": 0.2790619330303106, + "objective/train/value_std": 0.0040130615234375, + "objective/train/weight_avg": 1.001047968864441, + "objective/train/weighted_lm_loss": 2.717750072479248, + "objective/train/weights_max": 1.996764898300171, + "objective/train/weights_min": 0.555554986000061, + "theoretical_loss": 3.681367238122049, + "tokens_seen": 912588800 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003654262788365095, + "loss": 1.4333, + "theoretical_loss": 3.681367238122049, + "tokens_seen": 912588800 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036541624874623875, + "loss": 1.1793, + "theoretical_loss": 3.6813416528209126, + "tokens_seen": 912654336 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036540621865596793, + "loss": 1.356, + "theoretical_loss": 3.6813160698713263, + "tokens_seen": 912719872 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003653961885656971, + "loss": 1.4132, + "theoretical_loss": 3.6812904892729046, + "tokens_seen": 912785408 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003653861584754263, + "loss": 1.2597, + "theoretical_loss": 3.681264911025263, + "tokens_seen": 912850944 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036537612838515547, + "loss": 1.4785, + "theoretical_loss": 3.6812393351280157, + "tokens_seen": 912916480 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036536609829488465, + "loss": 1.444, + "theoretical_loss": 3.681213761580779, + "tokens_seen": 912982016 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003653560682046139, + "loss": 1.4517, + "theoretical_loss": 3.6811881903831685, + "tokens_seen": 913047552 + }, + { + "epoch": 0.28, + "learning_rate": 0.000365346038114343, + "loss": 1.4409, + "theoretical_loss": 3.681162621534799, + "tokens_seen": 913113088 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036533600802407225, + "loss": 1.3549, + "theoretical_loss": 3.681137055035286, + "tokens_seen": 913178624 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003653259779338014, + "loss": 1.4033, + "theoretical_loss": 3.6811114908842457, + "tokens_seen": 913244160 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003653159478435306, + "loss": 1.3967, + "theoretical_loss": 3.681085929081294, + "tokens_seen": 913309696 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003653059177532598, + "loss": 1.3571, + "theoretical_loss": 3.6810603696260467, + "tokens_seen": 913375232 + }, + { + "epoch": 0.28, + "learning_rate": 0.000365295887662989, + "loss": 1.4025, + "theoretical_loss": 3.681034812518119, + "tokens_seen": 913440768 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036528585757271816, + "loss": 1.3233, + "theoretical_loss": 3.6810092577571276, + "tokens_seen": 913506304 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036527582748244734, + "loss": 1.4485, + "theoretical_loss": 3.6809837053426886, + "tokens_seen": 913571840 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003652657973921765, + "loss": 1.4128, + "theoretical_loss": 3.6809581552744177, + "tokens_seen": 913637376 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036525576730190575, + "loss": 1.3616, + "theoretical_loss": 3.680932607551932, + "tokens_seen": 913702912 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003652457372116349, + "loss": 1.3232, + "theoretical_loss": 3.6809070621748474, + "tokens_seen": 913768448 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003652357071213641, + "loss": 1.3293, + "theoretical_loss": 3.6808815191427806, + "tokens_seen": 913833984 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003652256770310933, + "loss": 1.3834, + "theoretical_loss": 3.680855978455348, + "tokens_seen": 913899520 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003652156469408225, + "loss": 1.3848, + "theoretical_loss": 3.6808304401121665, + "tokens_seen": 913965056 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036520561685055166, + "loss": 1.4545, + "theoretical_loss": 3.6808049041128523, + "tokens_seen": 914030592 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036519558676028084, + "loss": 1.4649, + "theoretical_loss": 3.6807793704570235, + "tokens_seen": 914096128 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036518555667001, + "loss": 1.4833, + "theoretical_loss": 3.6807538391442955, + "tokens_seen": 914161664 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": 0.0004240590787958354, + "objective/train/docs_used": 525722, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6915409564971924, + "objective/train/original_loss": 2.691540479660034, + "objective/train/theoretical_loss": 3.6807283101742865, + "objective/train/tokens_used": 934687200, + "objective/train/value_avg": -0.0064544677734375, + "objective/train/value_loss": 7.275151438079774e-05, + "objective/train/value_max": -0.0003380775451660156, + "objective/train/value_min": -0.07354736328125, + "objective/train/value_reward_corr": 0.41698208070853926, + "objective/train/value_std": 0.006542205810546875, + "objective/train/weight_avg": 1.007786512374878, + "objective/train/weighted_lm_loss": 2.7048981189727783, + "objective/train/weights_max": 1.8298331499099731, + "objective/train/weights_min": 0.34676191210746765, + "theoretical_loss": 3.6807283101742865, + "tokens_seen": 914227200 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036517552657973926, + "loss": 1.4439, + "theoretical_loss": 3.6807283101742865, + "tokens_seen": 914227200 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003651654964894684, + "loss": 1.4073, + "theoretical_loss": 3.6807027835466126, + "tokens_seen": 914292736 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003651554663991976, + "loss": 1.3257, + "theoretical_loss": 3.680677259260892, + "tokens_seen": 914358272 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036514543630892675, + "loss": 1.3901, + "theoretical_loss": 3.680651737316741, + "tokens_seen": 914423808 + }, + { + "epoch": 0.28, + "learning_rate": 0.000365135406218656, + "loss": 1.4415, + "theoretical_loss": 3.6806262177137787, + "tokens_seen": 914489344 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036512537612838516, + "loss": 1.4007, + "theoretical_loss": 3.680600700451621, + "tokens_seen": 914554880 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036511534603811434, + "loss": 1.2934, + "theoretical_loss": 3.6805751855298854, + "tokens_seen": 914620416 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003651053159478435, + "loss": 1.4421, + "theoretical_loss": 3.6805496729481906, + "tokens_seen": 914685952 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036509528585757276, + "loss": 1.3928, + "theoretical_loss": 3.6805241627061536, + "tokens_seen": 914751488 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003650852557673019, + "loss": 1.4916, + "theoretical_loss": 3.6804986548033924, + "tokens_seen": 914817024 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003650752256770311, + "loss": 1.4498, + "theoretical_loss": 3.680473149239525, + "tokens_seen": 914882560 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036506519558676025, + "loss": 1.2772, + "theoretical_loss": 3.6804476460141697, + "tokens_seen": 914948096 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003650551654964895, + "loss": 1.3435, + "theoretical_loss": 3.6804221451269443, + "tokens_seen": 915013632 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036504513540621867, + "loss": 1.5091, + "theoretical_loss": 3.680396646577467, + "tokens_seen": 915079168 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036503510531594785, + "loss": 1.3703, + "theoretical_loss": 3.680371150365356, + "tokens_seen": 915144704 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036502507522567703, + "loss": 1.3039, + "theoretical_loss": 3.6803456564902297, + "tokens_seen": 915210240 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003650150451354062, + "loss": 1.4142, + "theoretical_loss": 3.680320164951707, + "tokens_seen": 915275776 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003650050150451354, + "loss": 1.4713, + "theoretical_loss": 3.6802946757494057, + "tokens_seen": 915341312 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003649949849548646, + "loss": 1.4436, + "theoretical_loss": 3.6802691888829453, + "tokens_seen": 915406848 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036498495486459375, + "loss": 1.3875, + "theoretical_loss": 3.680243704351944, + "tokens_seen": 915472384 + }, + { + "epoch": 0.28, + "learning_rate": 0.000364974924774323, + "loss": 1.3508, + "theoretical_loss": 3.6802182221560207, + "tokens_seen": 915537920 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003649648946840521, + "loss": 1.2984, + "theoretical_loss": 3.6801927422947944, + "tokens_seen": 915603456 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036495486459378135, + "loss": 1.306, + "theoretical_loss": 3.680167264767884, + "tokens_seen": 915668992 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036494483450351053, + "loss": 1.4879, + "theoretical_loss": 3.680141789574909, + "tokens_seen": 915734528 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003649348044132397, + "loss": 1.4295, + "theoretical_loss": 3.680116316715488, + "tokens_seen": 915800064 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": 2.961280188173987e-05, + "objective/train/docs_used": 526330, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7830185890197754, + "objective/train/original_loss": 2.7830188274383545, + "objective/train/theoretical_loss": 3.6800908461892403, + "objective/train/tokens_used": 936325600, + "objective/train/value_avg": -0.006649017333984375, + "objective/train/value_loss": 7.363792974501848e-05, + "objective/train/value_max": -0.00033664703369140625, + "objective/train/value_min": -0.10467529296875, + "objective/train/value_reward_corr": 0.46614937684696633, + "objective/train/value_std": 0.006214141845703125, + "objective/train/weight_avg": 1.0037078857421875, + "objective/train/weighted_lm_loss": 2.800755739212036, + "objective/train/weights_max": 2.0274667739868164, + "objective/train/weights_min": 0.4082237780094147, + "theoretical_loss": 3.6800908461892403, + "tokens_seen": 915865600 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003649247743229689, + "loss": 1.4615, + "theoretical_loss": 3.6800908461892403, + "tokens_seen": 915865600 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036491474423269813, + "loss": 1.4023, + "theoretical_loss": 3.6800653779957857, + "tokens_seen": 915931136 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036490471414242726, + "loss": 1.3706, + "theoretical_loss": 3.6800399121347436, + "tokens_seen": 915996672 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003648946840521565, + "loss": 1.4229, + "theoretical_loss": 3.680014448605733, + "tokens_seen": 916062208 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003648846539618856, + "loss": 1.3574, + "theoretical_loss": 3.6799889874083744, + "tokens_seen": 916127744 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036487462387161485, + "loss": 1.3632, + "theoretical_loss": 3.6799635285422867, + "tokens_seen": 916193280 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003648645937813441, + "loss": 1.3089, + "theoretical_loss": 3.67993807200709, + "tokens_seen": 916258816 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003648545636910732, + "loss": 1.423, + "theoretical_loss": 3.6799126178024046, + "tokens_seen": 916324352 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036484453360080245, + "loss": 1.3415, + "theoretical_loss": 3.67988716592785, + "tokens_seen": 916389888 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003648345035105316, + "loss": 1.43, + "theoretical_loss": 3.679861716383046, + "tokens_seen": 916455424 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003648244734202608, + "loss": 1.4107, + "theoretical_loss": 3.6798362691676134, + "tokens_seen": 916520960 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036481444332999, + "loss": 1.493, + "theoretical_loss": 3.6798108242811725, + "tokens_seen": 916586496 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003648044132397192, + "loss": 1.3671, + "theoretical_loss": 3.6797853817233435, + "tokens_seen": 916652032 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036479438314944836, + "loss": 1.3016, + "theoretical_loss": 3.6797599414937463, + "tokens_seen": 916717568 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036478435305917754, + "loss": 1.469, + "theoretical_loss": 3.6797345035920017, + "tokens_seen": 916783104 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003647743229689067, + "loss": 1.4135, + "theoretical_loss": 3.679709068017731, + "tokens_seen": 916848640 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036476429287863595, + "loss": 1.3886, + "theoretical_loss": 3.679683634770554, + "tokens_seen": 916914176 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003647542627883651, + "loss": 1.4054, + "theoretical_loss": 3.6796582038500922, + "tokens_seen": 916979712 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003647442326980943, + "loss": 1.4662, + "theoretical_loss": 3.6796327752559654, + "tokens_seen": 917045248 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003647342026078235, + "loss": 1.2301, + "theoretical_loss": 3.679607348987796, + "tokens_seen": 917110784 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003647241725175527, + "loss": 1.3005, + "theoretical_loss": 3.6795819250452038, + "tokens_seen": 917176320 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036471414242728186, + "loss": 1.5104, + "theoretical_loss": 3.679556503427811, + "tokens_seen": 917241856 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036470411233701104, + "loss": 1.4052, + "theoretical_loss": 3.6795310841352378, + "tokens_seen": 917307392 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003646940822467402, + "loss": 1.3952, + "theoretical_loss": 3.6795056671671063, + "tokens_seen": 917372928 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036468405215646946, + "loss": 1.3835, + "theoretical_loss": 3.679480252523038, + "tokens_seen": 917438464 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.00041235118987970054, + "objective/train/docs_used": 527026, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9281959533691406, + "objective/train/original_loss": 2.9281954765319824, + "objective/train/theoretical_loss": 3.6794548402026535, + "objective/train/tokens_used": 937964000, + "objective/train/value_avg": -0.005702972412109375, + "objective/train/value_loss": 6.461872544605285e-05, + "objective/train/value_max": -0.0005617141723632812, + "objective/train/value_min": -0.0496826171875, + "objective/train/value_reward_corr": 0.421712885977449, + "objective/train/value_std": 0.004650115966796875, + "objective/train/weight_avg": 0.9988665580749512, + "objective/train/weighted_lm_loss": 2.926548957824707, + "objective/train/weights_max": 1.4555243253707886, + "objective/train/weights_min": 0.20558325946331024, + "theoretical_loss": 3.6794548402026535, + "tokens_seen": 917504000 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003646740220661986, + "loss": 1.4777, + "theoretical_loss": 3.6794548402026535, + "tokens_seen": 917504000 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003646639919759278, + "loss": 1.3735, + "theoretical_loss": 3.679429430205575, + "tokens_seen": 917569536 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036465396188565695, + "loss": 1.387, + "theoretical_loss": 3.6794040225314246, + "tokens_seen": 917635072 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003646439317953862, + "loss": 1.388, + "theoretical_loss": 3.679378617179823, + "tokens_seen": 917700608 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036463390170511536, + "loss": 1.4722, + "theoretical_loss": 3.679353214150393, + "tokens_seen": 917766144 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036462387161484454, + "loss": 1.4382, + "theoretical_loss": 3.679327813442756, + "tokens_seen": 917831680 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003646138415245737, + "loss": 1.2387, + "theoretical_loss": 3.6793024150565343, + "tokens_seen": 917897216 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036460381143430296, + "loss": 1.329, + "theoretical_loss": 3.6792770189913497, + "tokens_seen": 917962752 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003645937813440321, + "loss": 1.4011, + "theoretical_loss": 3.679251625246825, + "tokens_seen": 918028288 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003645837512537613, + "loss": 1.4049, + "theoretical_loss": 3.6792262338225825, + "tokens_seen": 918093824 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036457372116349045, + "loss": 1.3285, + "theoretical_loss": 3.6792008447182436, + "tokens_seen": 918159360 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003645636910732197, + "loss": 1.3351, + "theoretical_loss": 3.6791754579334315, + "tokens_seen": 918224896 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036455366098294887, + "loss": 1.3609, + "theoretical_loss": 3.6791500734677687, + "tokens_seen": 918290432 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036454363089267805, + "loss": 1.3387, + "theoretical_loss": 3.6791246913208777, + "tokens_seen": 918355968 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036453360080240723, + "loss": 1.297, + "theoretical_loss": 3.6790993114923816, + "tokens_seen": 918421504 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003645235707121364, + "loss": 1.2876, + "theoretical_loss": 3.6790739339819027, + "tokens_seen": 918487040 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003645135406218656, + "loss": 1.3777, + "theoretical_loss": 3.6790485587890642, + "tokens_seen": 918552576 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003645035105315948, + "loss": 1.4043, + "theoretical_loss": 3.6790231859134894, + "tokens_seen": 918618112 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036449348044132395, + "loss": 1.2527, + "theoretical_loss": 3.6789978153548013, + "tokens_seen": 918683648 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003644834503510532, + "loss": 1.3209, + "theoretical_loss": 3.6789724471126224, + "tokens_seen": 918749184 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003644734202607823, + "loss": 1.3116, + "theoretical_loss": 3.6789470811865765, + "tokens_seen": 918814720 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036446339017051155, + "loss": 1.3986, + "theoretical_loss": 3.6789217175762867, + "tokens_seen": 918880256 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036445336008024073, + "loss": 1.3216, + "theoretical_loss": 3.678896356281377, + "tokens_seen": 918945792 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003644433299899699, + "loss": 1.4023, + "theoretical_loss": 3.6788709973014706, + "tokens_seen": 919011328 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003644332998996991, + "loss": 1.4196, + "theoretical_loss": 3.678845640636191, + "tokens_seen": 919076864 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.0008217859431169927, + "objective/train/docs_used": 528220, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.906470537185669, + "objective/train/original_loss": 2.906470537185669, + "objective/train/theoretical_loss": 3.678820286285162, + "objective/train/tokens_used": 939602400, + "objective/train/value_avg": -0.00518035888671875, + "objective/train/value_loss": 5.440719178295694e-05, + "objective/train/value_max": -0.0001398324966430664, + "objective/train/value_min": -0.0546875, + "objective/train/value_reward_corr": 0.449344689939682, + "objective/train/value_std": 0.004589080810546875, + "objective/train/weight_avg": 0.994381308555603, + "objective/train/weighted_lm_loss": 2.883211135864258, + "objective/train/weights_max": 1.535370945930481, + "objective/train/weights_min": 0.5037210583686829, + "theoretical_loss": 3.678820286285162, + "tokens_seen": 919142400 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036442326980942833, + "loss": 1.4415, + "theoretical_loss": 3.678820286285162, + "tokens_seen": 919142400 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036441323971915746, + "loss": 1.5728, + "theoretical_loss": 3.6787949342480073, + "tokens_seen": 919207936 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003644032096288867, + "loss": 1.4836, + "theoretical_loss": 3.6787695845243507, + "tokens_seen": 919273472 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003643931795386158, + "loss": 1.4345, + "theoretical_loss": 3.6787442371138166, + "tokens_seen": 919339008 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036438314944834505, + "loss": 1.391, + "theoretical_loss": 3.678718892016029, + "tokens_seen": 919404544 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036437311935807424, + "loss": 1.3514, + "theoretical_loss": 3.6786935492306116, + "tokens_seen": 919470080 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003643630892678034, + "loss": 1.4666, + "theoretical_loss": 3.6786682087571894, + "tokens_seen": 919535616 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003643530591775326, + "loss": 1.4108, + "theoretical_loss": 3.6786428705953855, + "tokens_seen": 919601152 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003643430290872618, + "loss": 1.4085, + "theoretical_loss": 3.6786175347448253, + "tokens_seen": 919666688 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036433299899699096, + "loss": 1.4241, + "theoretical_loss": 3.678592201205133, + "tokens_seen": 919732224 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003643229689067202, + "loss": 1.5381, + "theoretical_loss": 3.6785668699759335, + "tokens_seen": 919797760 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003643129388164493, + "loss": 1.3817, + "theoretical_loss": 3.678541541056851, + "tokens_seen": 919863296 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036430290872617856, + "loss": 1.4147, + "theoretical_loss": 3.6785162144475105, + "tokens_seen": 919928832 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003642928786359077, + "loss": 1.3291, + "theoretical_loss": 3.6784908901475366, + "tokens_seen": 919994368 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003642828485456369, + "loss": 1.4413, + "theoretical_loss": 3.678465568156555, + "tokens_seen": 920059904 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003642728184553661, + "loss": 1.3421, + "theoretical_loss": 3.67844024847419, + "tokens_seen": 920125440 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003642627883650953, + "loss": 1.2924, + "theoretical_loss": 3.6784149311000665, + "tokens_seen": 920190976 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036425275827482446, + "loss": 1.5008, + "theoretical_loss": 3.67838961603381, + "tokens_seen": 920256512 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003642427281845537, + "loss": 1.3956, + "theoretical_loss": 3.678364303275046, + "tokens_seen": 920322048 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003642326980942828, + "loss": 1.2804, + "theoretical_loss": 3.6783389928234005, + "tokens_seen": 920387584 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036422266800401206, + "loss": 1.4167, + "theoretical_loss": 3.6783136846784976, + "tokens_seen": 920453120 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003642126379137412, + "loss": 1.4863, + "theoretical_loss": 3.678288378839963, + "tokens_seen": 920518656 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003642026078234704, + "loss": 1.4164, + "theoretical_loss": 3.6782630753074237, + "tokens_seen": 920584192 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003641925777331996, + "loss": 1.3631, + "theoretical_loss": 3.6782377740805043, + "tokens_seen": 920649728 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003641825476429288, + "loss": 1.4144, + "theoretical_loss": 3.6782124751588308, + "tokens_seen": 920715264 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.002098310273140669, + "objective/train/docs_used": 528886, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9024817943573, + "objective/train/original_loss": 2.902482271194458, + "objective/train/theoretical_loss": 3.678187178542029, + "objective/train/tokens_used": 941240800, + "objective/train/value_avg": -0.00492095947265625, + "objective/train/value_loss": 8.944925502873957e-05, + "objective/train/value_max": -0.0003726482391357422, + "objective/train/value_min": -0.046722412109375, + "objective/train/value_reward_corr": 0.34411557912534385, + "objective/train/value_std": 0.0036182403564453125, + "objective/train/weight_avg": 0.9830686450004578, + "objective/train/weighted_lm_loss": 2.8553664684295654, + "objective/train/weights_max": 1.426936388015747, + "objective/train/weights_min": 0.2929255962371826, + "theoretical_loss": 3.678187178542029, + "tokens_seen": 920780800 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036417251755265797, + "loss": 1.4035, + "theoretical_loss": 3.678187178542029, + "tokens_seen": 920780800 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036416248746238715, + "loss": 1.4502, + "theoretical_loss": 3.6781618842297252, + "tokens_seen": 920846336 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036415245737211633, + "loss": 1.4125, + "theoretical_loss": 3.6781365922215454, + "tokens_seen": 920911872 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036414242728184556, + "loss": 1.4593, + "theoretical_loss": 3.6781113025171157, + "tokens_seen": 920977408 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003641323971915747, + "loss": 1.3996, + "theoretical_loss": 3.678086015116062, + "tokens_seen": 921042944 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003641223671013039, + "loss": 1.3498, + "theoretical_loss": 3.6780607300180113, + "tokens_seen": 921108480 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036411233701103316, + "loss": 1.5012, + "theoretical_loss": 3.6780354472225896, + "tokens_seen": 921174016 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003641023069207623, + "loss": 1.3246, + "theoretical_loss": 3.678010166729423, + "tokens_seen": 921239552 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003640922768304915, + "loss": 1.582, + "theoretical_loss": 3.6779848885381394, + "tokens_seen": 921305088 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036408224674022065, + "loss": 1.3426, + "theoretical_loss": 3.6779596126483645, + "tokens_seen": 921370624 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003640722166499499, + "loss": 1.4511, + "theoretical_loss": 3.6779343390597248, + "tokens_seen": 921436160 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036406218655967907, + "loss": 1.4103, + "theoretical_loss": 3.677909067771848, + "tokens_seen": 921501696 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036405215646940825, + "loss": 1.3882, + "theoretical_loss": 3.6778837987843604, + "tokens_seen": 921567232 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036404212637913743, + "loss": 1.3246, + "theoretical_loss": 3.6778585320968893, + "tokens_seen": 921632768 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003640320962888666, + "loss": 1.4162, + "theoretical_loss": 3.6778332677090617, + "tokens_seen": 921698304 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003640220661985958, + "loss": 1.423, + "theoretical_loss": 3.6778080056205056, + "tokens_seen": 921763840 + }, + { + "epoch": 0.28, + "learning_rate": 0.000364012036108325, + "loss": 1.3727, + "theoretical_loss": 3.6777827458308474, + "tokens_seen": 921829376 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036400200601805415, + "loss": 1.26, + "theoretical_loss": 3.677757488339714, + "tokens_seen": 921894912 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003639919759277834, + "loss": 1.3946, + "theoretical_loss": 3.6777322331467346, + "tokens_seen": 921960448 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003639819458375125, + "loss": 1.3721, + "theoretical_loss": 3.677706980251535, + "tokens_seen": 922025984 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036397191574724175, + "loss": 1.4484, + "theoretical_loss": 3.677681729653744, + "tokens_seen": 922091520 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036396188565697093, + "loss": 1.4826, + "theoretical_loss": 3.6776564813529884, + "tokens_seen": 922157056 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003639518555667001, + "loss": 1.4792, + "theoretical_loss": 3.6776312353488967, + "tokens_seen": 922222592 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003639418254764293, + "loss": 1.3639, + "theoretical_loss": 3.677605991641097, + "tokens_seen": 922288128 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036393179538615853, + "loss": 1.2845, + "theoretical_loss": 3.6775807502292164, + "tokens_seen": 922353664 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.0013419799506664276, + "objective/train/docs_used": 530064, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.304598331451416, + "objective/train/original_loss": 3.3045976161956787, + "objective/train/theoretical_loss": 3.6775555111128835, + "objective/train/tokens_used": 942879200, + "objective/train/value_avg": -0.00519561767578125, + "objective/train/value_loss": 8.52946104714647e-05, + "objective/train/value_max": -0.0002033710479736328, + "objective/train/value_min": -0.051666259765625, + "objective/train/value_reward_corr": 0.460149922680163, + "objective/train/value_std": 0.004436492919921875, + "objective/train/weight_avg": 0.9903239011764526, + "objective/train/weighted_lm_loss": 3.2560338973999023, + "objective/train/weights_max": 1.5005913972854614, + "objective/train/weights_min": 0.20108462870121002, + "theoretical_loss": 3.6775555111128835, + "tokens_seen": 922419200 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036392176529588766, + "loss": 1.442, + "theoretical_loss": 3.6775555111128835, + "tokens_seen": 922419200 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003639117352056169, + "loss": 1.3012, + "theoretical_loss": 3.677530274291727, + "tokens_seen": 922484736 + }, + { + "epoch": 0.28, + "learning_rate": 0.000363901705115346, + "loss": 1.4396, + "theoretical_loss": 3.677505039765374, + "tokens_seen": 922550272 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036389167502507525, + "loss": 1.2765, + "theoretical_loss": 3.6774798075334543, + "tokens_seen": 922615808 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036388164493480444, + "loss": 1.4966, + "theoretical_loss": 3.6774545775955945, + "tokens_seen": 922681344 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003638716148445336, + "loss": 1.2149, + "theoretical_loss": 3.6774293499514243, + "tokens_seen": 922746880 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003638615847542628, + "loss": 1.4472, + "theoretical_loss": 3.6774041246005726, + "tokens_seen": 922812416 + }, + { + "epoch": 0.28, + "learning_rate": 0.000363851554663992, + "loss": 1.5424, + "theoretical_loss": 3.677378901542667, + "tokens_seen": 922877952 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036384152457372116, + "loss": 1.3762, + "theoretical_loss": 3.677353680777337, + "tokens_seen": 922943488 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003638314944834504, + "loss": 1.4785, + "theoretical_loss": 3.677328462304211, + "tokens_seen": 923009024 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003638214643931795, + "loss": 1.4272, + "theoretical_loss": 3.6773032461229183, + "tokens_seen": 923074560 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036381143430290876, + "loss": 1.4173, + "theoretical_loss": 3.6772780322330876, + "tokens_seen": 923140096 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003638014042126379, + "loss": 1.3951, + "theoretical_loss": 3.677252820634349, + "tokens_seen": 923205632 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003637913741223671, + "loss": 1.4373, + "theoretical_loss": 3.6772276113263302, + "tokens_seen": 923271168 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003637813440320963, + "loss": 1.4544, + "theoretical_loss": 3.6772024043086615, + "tokens_seen": 923336704 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003637713139418255, + "loss": 1.3082, + "theoretical_loss": 3.677177199580972, + "tokens_seen": 923402240 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036376128385155466, + "loss": 1.3966, + "theoretical_loss": 3.677151997142891, + "tokens_seen": 923467776 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003637512537612839, + "loss": 1.3475, + "theoretical_loss": 3.677126796994049, + "tokens_seen": 923533312 + }, + { + "epoch": 0.28, + "learning_rate": 0.000363741223671013, + "loss": 1.4306, + "theoretical_loss": 3.677101599134074, + "tokens_seen": 923598848 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036373119358074226, + "loss": 1.278, + "theoretical_loss": 3.6770764035625967, + "tokens_seen": 923664384 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003637211634904714, + "loss": 1.429, + "theoretical_loss": 3.677051210279247, + "tokens_seen": 923729920 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003637111334002006, + "loss": 1.3682, + "theoretical_loss": 3.6770260192836544, + "tokens_seen": 923795456 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003637011033099298, + "loss": 1.4337, + "theoretical_loss": 3.677000830575449, + "tokens_seen": 923860992 + }, + { + "epoch": 0.28, + "learning_rate": 0.000363691073219659, + "loss": 1.3677, + "theoretical_loss": 3.676975644154261, + "tokens_seen": 923926528 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036368104312938817, + "loss": 1.4304, + "theoretical_loss": 3.6769504600197203, + "tokens_seen": 923992064 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.0006551070837303996, + "objective/train/docs_used": 530632, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.901052236557007, + "objective/train/original_loss": 2.901052713394165, + "objective/train/theoretical_loss": 3.6769252781714576, + "objective/train/tokens_used": 944517600, + "objective/train/value_avg": -0.005489349365234375, + "objective/train/value_loss": 6.273984763538465e-05, + "objective/train/value_max": -0.00044417381286621094, + "objective/train/value_min": -0.07763671875, + "objective/train/value_reward_corr": 0.37445717158293895, + "objective/train/value_std": 0.004688262939453125, + "objective/train/weight_avg": 0.9964196085929871, + "objective/train/weighted_lm_loss": 2.8821685314178467, + "objective/train/weights_max": 1.7375683784484863, + "objective/train/weights_min": 0.31992170214653015, + "theoretical_loss": 3.6769252781714576, + "tokens_seen": 924057600 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036367101303911735, + "loss": 1.3618, + "theoretical_loss": 3.6769252781714576, + "tokens_seen": 924057600 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036366098294884653, + "loss": 1.3706, + "theoretical_loss": 3.6769000986091025, + "tokens_seen": 924123136 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036365095285857576, + "loss": 1.2706, + "theoretical_loss": 3.6768749213322858, + "tokens_seen": 924188672 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003636409227683049, + "loss": 1.2676, + "theoretical_loss": 3.676849746340639, + "tokens_seen": 924254208 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003636308926780341, + "loss": 1.4588, + "theoretical_loss": 3.67682457363379, + "tokens_seen": 924319744 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036362086258776325, + "loss": 1.3608, + "theoretical_loss": 3.6767994032113727, + "tokens_seen": 924385280 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003636108324974925, + "loss": 1.3613, + "theoretical_loss": 3.6767742350730153, + "tokens_seen": 924450816 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036360080240722167, + "loss": 1.4663, + "theoretical_loss": 3.67674906921835, + "tokens_seen": 924516352 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036359077231695085, + "loss": 1.5014, + "theoretical_loss": 3.676723905647007, + "tokens_seen": 924581888 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036358074222668003, + "loss": 1.5127, + "theoretical_loss": 3.6766987443586183, + "tokens_seen": 924647424 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036357071213640927, + "loss": 1.3945, + "theoretical_loss": 3.6766735853528143, + "tokens_seen": 924712960 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003635606820461384, + "loss": 1.4122, + "theoretical_loss": 3.6766484286292256, + "tokens_seen": 924778496 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036355065195586763, + "loss": 1.3571, + "theoretical_loss": 3.6766232741874845, + "tokens_seen": 924844032 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036354062186559676, + "loss": 1.2462, + "theoretical_loss": 3.676598122027222, + "tokens_seen": 924909568 + }, + { + "epoch": 0.28, + "learning_rate": 0.000363530591775326, + "loss": 1.4319, + "theoretical_loss": 3.6765729721480693, + "tokens_seen": 924975104 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036352056168505517, + "loss": 1.3192, + "theoretical_loss": 3.676547824549658, + "tokens_seen": 925040640 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036351053159478435, + "loss": 1.306, + "theoretical_loss": 3.6765226792316206, + "tokens_seen": 925106176 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036350050150451353, + "loss": 1.417, + "theoretical_loss": 3.676497536193587, + "tokens_seen": 925171712 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003634904714142427, + "loss": 1.3617, + "theoretical_loss": 3.6764723954351908, + "tokens_seen": 925237248 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003634804413239719, + "loss": 1.4079, + "theoretical_loss": 3.6764472569560622, + "tokens_seen": 925302784 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036347041123370113, + "loss": 1.3992, + "theoretical_loss": 3.6764221207558343, + "tokens_seen": 925368320 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036346038114343026, + "loss": 1.4124, + "theoretical_loss": 3.6763969868341393, + "tokens_seen": 925433856 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003634503510531595, + "loss": 1.3071, + "theoretical_loss": 3.6763718551906086, + "tokens_seen": 925499392 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003634403209628886, + "loss": 1.3671, + "theoretical_loss": 3.6763467258248745, + "tokens_seen": 925564928 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036343029087261786, + "loss": 1.3875, + "theoretical_loss": 3.67632159873657, + "tokens_seen": 925630464 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.0011427938006818295, + "objective/train/docs_used": 532017, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.702497720718384, + "objective/train/original_loss": 2.7024974822998047, + "objective/train/theoretical_loss": 3.6762964739253263, + "objective/train/tokens_used": 946156000, + "objective/train/value_avg": -0.0052642822265625, + "objective/train/value_loss": 6.405996100511402e-05, + "objective/train/value_max": -0.00027370452880859375, + "objective/train/value_min": -0.09185791015625, + "objective/train/value_reward_corr": 0.5397648128801549, + "objective/train/value_std": 0.0051116943359375, + "objective/train/weight_avg": 0.9915583729743958, + "objective/train/weighted_lm_loss": 2.686950922012329, + "objective/train/weights_max": 1.8572407960891724, + "objective/train/weights_min": 0.20558325946331024, + "theoretical_loss": 3.6762964739253263, + "tokens_seen": 925696000 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036342026078234704, + "loss": 1.3013, + "theoretical_loss": 3.6762964739253263, + "tokens_seen": 925696000 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003634102306920762, + "loss": 1.3834, + "theoretical_loss": 3.6762713513907768, + "tokens_seen": 925761536 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003634002006018054, + "loss": 1.4489, + "theoretical_loss": 3.676246231132554, + "tokens_seen": 925827072 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036339017051153464, + "loss": 1.2985, + "theoretical_loss": 3.67622111315029, + "tokens_seen": 925892608 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036338014042126376, + "loss": 1.3933, + "theoretical_loss": 3.6761959974436174, + "tokens_seen": 925958144 + }, + { + "epoch": 0.28, + "learning_rate": 0.000363370110330993, + "loss": 1.4166, + "theoretical_loss": 3.67617088401217, + "tokens_seen": 926023680 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003633600802407222, + "loss": 1.3838, + "theoretical_loss": 3.6761457728555795, + "tokens_seen": 926089216 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036335005015045136, + "loss": 1.4527, + "theoretical_loss": 3.67612066397348, + "tokens_seen": 926154752 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003633400200601806, + "loss": 1.346, + "theoretical_loss": 3.676095557365504, + "tokens_seen": 926220288 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003633299899699097, + "loss": 1.411, + "theoretical_loss": 3.6760704530312847, + "tokens_seen": 926285824 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036331995987963896, + "loss": 1.3998, + "theoretical_loss": 3.6760453509704547, + "tokens_seen": 926351360 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003633099297893681, + "loss": 1.5013, + "theoretical_loss": 3.6760202511826483, + "tokens_seen": 926416896 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003632998996990973, + "loss": 1.3735, + "theoretical_loss": 3.6759951536674986, + "tokens_seen": 926482432 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003632898696088265, + "loss": 1.3092, + "theoretical_loss": 3.6759700584246393, + "tokens_seen": 926547968 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003632798395185557, + "loss": 1.3041, + "theoretical_loss": 3.675944965453703, + "tokens_seen": 926613504 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036326980942828486, + "loss": 1.3209, + "theoretical_loss": 3.6759198747543245, + "tokens_seen": 926679040 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003632597793380141, + "loss": 1.3417, + "theoretical_loss": 3.675894786326137, + "tokens_seen": 926744576 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003632497492477432, + "loss": 1.3582, + "theoretical_loss": 3.675869700168774, + "tokens_seen": 926810112 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036323971915747246, + "loss": 1.358, + "theoretical_loss": 3.6758446162818705, + "tokens_seen": 926875648 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003632296890672016, + "loss": 1.3609, + "theoretical_loss": 3.6758195346650595, + "tokens_seen": 926941184 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003632196589769308, + "loss": 1.3802, + "theoretical_loss": 3.675794455317975, + "tokens_seen": 927006720 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036320962888666, + "loss": 1.3995, + "theoretical_loss": 3.675769378240252, + "tokens_seen": 927072256 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003631995987963892, + "loss": 1.3868, + "theoretical_loss": 3.6757443034315243, + "tokens_seen": 927137792 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036318956870611837, + "loss": 1.4445, + "theoretical_loss": 3.675719230891426, + "tokens_seen": 927203328 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036317953861584755, + "loss": 1.3505, + "theoretical_loss": 3.6756941606195914, + "tokens_seen": 927268864 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.001273985835723579, + "objective/train/docs_used": 532753, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.927586317062378, + "objective/train/original_loss": 2.927586078643799, + "objective/train/theoretical_loss": 3.6756690926156557, + "objective/train/tokens_used": 947794400, + "objective/train/value_avg": -0.006542205810546875, + "objective/train/value_loss": 0.00011795214959420264, + "objective/train/value_max": -0.00030541419982910156, + "objective/train/value_min": -0.12200927734375, + "objective/train/value_reward_corr": 0.5911987584750253, + "objective/train/value_std": 0.007305145263671875, + "objective/train/weight_avg": 0.9924706816673279, + "objective/train/weighted_lm_loss": 2.897669553756714, + "objective/train/weights_max": 2.2014284133911133, + "objective/train/weights_min": 0.3023601174354553, + "theoretical_loss": 3.6756690926156557, + "tokens_seen": 927334400 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036316950852557673, + "loss": 1.3591, + "theoretical_loss": 3.6756690926156557, + "tokens_seen": 927334400 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036315947843530596, + "loss": 1.4065, + "theoretical_loss": 3.6756440268792536, + "tokens_seen": 927399936 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003631494483450351, + "loss": 1.368, + "theoretical_loss": 3.6756189634100185, + "tokens_seen": 927465472 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003631394182547643, + "loss": 1.4651, + "theoretical_loss": 3.6755939022075865, + "tokens_seen": 927531008 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036312938816449345, + "loss": 1.372, + "theoretical_loss": 3.6755688432715914, + "tokens_seen": 927596544 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003631193580742227, + "loss": 1.4485, + "theoretical_loss": 3.6755437866016694, + "tokens_seen": 927662080 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036310932798395187, + "loss": 1.3964, + "theoretical_loss": 3.675518732197454, + "tokens_seen": 927727616 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036309929789368105, + "loss": 1.4172, + "theoretical_loss": 3.6754936800585813, + "tokens_seen": 927793152 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036308926780341023, + "loss": 1.5038, + "theoretical_loss": 3.675468630184686, + "tokens_seen": 927858688 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036307923771313947, + "loss": 1.3227, + "theoretical_loss": 3.6754435825754044, + "tokens_seen": 927924224 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003630692076228686, + "loss": 1.3877, + "theoretical_loss": 3.6754185372303705, + "tokens_seen": 927989760 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036305917753259783, + "loss": 1.4281, + "theoretical_loss": 3.6753934941492203, + "tokens_seen": 928055296 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036304914744232696, + "loss": 1.3581, + "theoretical_loss": 3.6753684533315893, + "tokens_seen": 928120832 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003630391173520562, + "loss": 1.4438, + "theoretical_loss": 3.675343414777113, + "tokens_seen": 928186368 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036302908726178537, + "loss": 1.3742, + "theoretical_loss": 3.675318378485427, + "tokens_seen": 928251904 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036301905717151455, + "loss": 1.365, + "theoretical_loss": 3.6752933444561675, + "tokens_seen": 928317440 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036300902708124374, + "loss": 1.2861, + "theoretical_loss": 3.6752683126889703, + "tokens_seen": 928382976 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003629989969909729, + "loss": 1.4877, + "theoretical_loss": 3.675243283183471, + "tokens_seen": 928448512 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003629889669007021, + "loss": 1.3903, + "theoretical_loss": 3.6752182559393054, + "tokens_seen": 928514048 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036297893681043133, + "loss": 1.3573, + "theoretical_loss": 3.6751932309561104, + "tokens_seen": 928579584 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036296890672016046, + "loss": 1.4645, + "theoretical_loss": 3.6751682082335217, + "tokens_seen": 928645120 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003629588766298897, + "loss": 1.3633, + "theoretical_loss": 3.6751431877711758, + "tokens_seen": 928710656 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003629488465396188, + "loss": 1.351, + "theoretical_loss": 3.6751181695687087, + "tokens_seen": 928776192 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036293881644934806, + "loss": 1.3043, + "theoretical_loss": 3.675093153625757, + "tokens_seen": 928841728 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036292878635907724, + "loss": 1.463, + "theoretical_loss": 3.675068139941957, + "tokens_seen": 928907264 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.0008341668290086091, + "objective/train/docs_used": 534166, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.917280435562134, + "objective/train/original_loss": 2.917280673980713, + "objective/train/theoretical_loss": 3.6750431285169465, + "objective/train/tokens_used": 949432800, + "objective/train/value_avg": -0.00569915771484375, + "objective/train/value_loss": 6.777307135052979e-05, + "objective/train/value_max": -0.0002779960632324219, + "objective/train/value_min": -0.057586669921875, + "objective/train/value_reward_corr": 0.48181481792037706, + "objective/train/value_std": 0.004756927490234375, + "objective/train/weight_avg": 0.9948642253875732, + "objective/train/weighted_lm_loss": 2.892822265625, + "objective/train/weights_max": 1.6563613414764404, + "objective/train/weights_min": 0.4994446337223053, + "theoretical_loss": 3.6750431285169465, + "tokens_seen": 928972800 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003629187562688064, + "loss": 1.3038, + "theoretical_loss": 3.6750431285169465, + "tokens_seen": 928972800 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003629087261785356, + "loss": 1.4476, + "theoretical_loss": 3.6750181193503604, + "tokens_seen": 929038336 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036289869608826484, + "loss": 1.3684, + "theoretical_loss": 3.674993112441837, + "tokens_seen": 929103872 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036288866599799396, + "loss": 1.3721, + "theoretical_loss": 3.6749681077910124, + "tokens_seen": 929169408 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003628786359077232, + "loss": 1.4485, + "theoretical_loss": 3.6749431053975234, + "tokens_seen": 929234944 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003628686058174523, + "loss": 1.3737, + "theoretical_loss": 3.674918105261008, + "tokens_seen": 929300480 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036285857572718156, + "loss": 1.3463, + "theoretical_loss": 3.674893107381102, + "tokens_seen": 929366016 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036284854563691074, + "loss": 1.346, + "theoretical_loss": 3.6748681117574433, + "tokens_seen": 929431552 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003628385155466399, + "loss": 1.2713, + "theoretical_loss": 3.6748431183896693, + "tokens_seen": 929497088 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003628284854563691, + "loss": 1.4234, + "theoretical_loss": 3.674818127277417, + "tokens_seen": 929562624 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003628184553660983, + "loss": 1.3534, + "theoretical_loss": 3.6747931384203243, + "tokens_seen": 929628160 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036280842527582747, + "loss": 1.3384, + "theoretical_loss": 3.6747681518180286, + "tokens_seen": 929693696 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003627983951855567, + "loss": 1.3774, + "theoretical_loss": 3.6747431674701674, + "tokens_seen": 929759232 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036278836509528583, + "loss": 1.3197, + "theoretical_loss": 3.6747181853763786, + "tokens_seen": 929824768 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036277833500501506, + "loss": 1.3551, + "theoretical_loss": 3.6746932055362995, + "tokens_seen": 929890304 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036276830491474424, + "loss": 1.2979, + "theoretical_loss": 3.674668227949568, + "tokens_seen": 929955840 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003627582748244734, + "loss": 1.4134, + "theoretical_loss": 3.674643252615823, + "tokens_seen": 930021376 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003627482447342026, + "loss": 1.3578, + "theoretical_loss": 3.6746182795347013, + "tokens_seen": 930086912 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003627382146439318, + "loss": 1.2925, + "theoretical_loss": 3.674593308705842, + "tokens_seen": 930152448 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036272818455366097, + "loss": 1.4023, + "theoretical_loss": 3.6745683401288827, + "tokens_seen": 930217984 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003627181544633902, + "loss": 1.3587, + "theoretical_loss": 3.674543373803462, + "tokens_seen": 930283520 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036270812437311933, + "loss": 1.2961, + "theoretical_loss": 3.674518409729218, + "tokens_seen": 930349056 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036269809428284857, + "loss": 1.4042, + "theoretical_loss": 3.67449344790579, + "tokens_seen": 930414592 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003626880641925777, + "loss": 1.4562, + "theoretical_loss": 3.674468488332815, + "tokens_seen": 930480128 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036267803410230693, + "loss": 1.4623, + "theoretical_loss": 3.674443531009933, + "tokens_seen": 930545664 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.0015077635180205107, + "objective/train/docs_used": 534638, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1421191692352295, + "objective/train/original_loss": 3.1421196460723877, + "objective/train/theoretical_loss": 3.674418575936782, + "objective/train/tokens_used": 951071200, + "objective/train/value_avg": -0.005275726318359375, + "objective/train/value_loss": 7.65531076467596e-05, + "objective/train/value_max": -0.0004458427429199219, + "objective/train/value_min": -0.0726318359375, + "objective/train/value_reward_corr": 0.4483341726431059, + "objective/train/value_std": 0.0042266845703125, + "objective/train/weight_avg": 0.9884366393089294, + "objective/train/weighted_lm_loss": 3.1141672134399414, + "objective/train/weights_max": 1.6540632247924805, + "objective/train/weights_min": 0.2699933350086212, + "theoretical_loss": 3.674418575936782, + "tokens_seen": 930611200 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003626680040120361, + "loss": 1.382, + "theoretical_loss": 3.674418575936782, + "tokens_seen": 930611200 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003626579739217653, + "loss": 1.3701, + "theoretical_loss": 3.674393623113001, + "tokens_seen": 930676736 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036264794383149447, + "loss": 1.3416, + "theoretical_loss": 3.674368672538229, + "tokens_seen": 930742272 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036263791374122365, + "loss": 1.357, + "theoretical_loss": 3.674343724212105, + "tokens_seen": 930807808 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036262788365095283, + "loss": 1.382, + "theoretical_loss": 3.6743187781342677, + "tokens_seen": 930873344 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036261785356068207, + "loss": 1.2245, + "theoretical_loss": 3.674293834304356, + "tokens_seen": 930938880 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036260782347041125, + "loss": 1.4315, + "theoretical_loss": 3.67426889272201, + "tokens_seen": 931004416 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036259779338014043, + "loss": 1.4005, + "theoretical_loss": 3.6742439533868687, + "tokens_seen": 931069952 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036258776328986967, + "loss": 1.4068, + "theoretical_loss": 3.674219016298571, + "tokens_seen": 931135488 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003625777331995988, + "loss": 1.3878, + "theoretical_loss": 3.6741940814567564, + "tokens_seen": 931201024 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036256770310932803, + "loss": 1.3545, + "theoretical_loss": 3.674169148861065, + "tokens_seen": 931266560 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036255767301905716, + "loss": 1.3569, + "theoretical_loss": 3.674144218511136, + "tokens_seen": 931332096 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003625476429287864, + "loss": 1.4166, + "theoretical_loss": 3.674119290406609, + "tokens_seen": 931397632 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003625376128385156, + "loss": 1.3797, + "theoretical_loss": 3.6740943645471242, + "tokens_seen": 931463168 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036252758274824475, + "loss": 1.273, + "theoretical_loss": 3.6740694409323216, + "tokens_seen": 931528704 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036251755265797394, + "loss": 1.3333, + "theoretical_loss": 3.6740445195618405, + "tokens_seen": 931594240 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003625075225677031, + "loss": 1.3574, + "theoretical_loss": 3.6740196004353214, + "tokens_seen": 931659776 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003624974924774323, + "loss": 1.3184, + "theoretical_loss": 3.6739946835524044, + "tokens_seen": 931725312 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036248746238716153, + "loss": 1.2683, + "theoretical_loss": 3.6739697689127295, + "tokens_seen": 931790848 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036247743229689066, + "loss": 1.4991, + "theoretical_loss": 3.673944856515937, + "tokens_seen": 931856384 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003624674022066199, + "loss": 1.3215, + "theoretical_loss": 3.673919946361667, + "tokens_seen": 931921920 + }, + { + "epoch": 0.28, + "learning_rate": 0.000362457372116349, + "loss": 1.3703, + "theoretical_loss": 3.673895038449561, + "tokens_seen": 931987456 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036244734202607826, + "loss": 1.2681, + "theoretical_loss": 3.6738701327792587, + "tokens_seen": 932052992 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036243731193580744, + "loss": 1.3523, + "theoretical_loss": 3.673845229350401, + "tokens_seen": 932118528 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003624272818455366, + "loss": 1.3223, + "theoretical_loss": 3.673820328162628, + "tokens_seen": 932184064 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.000623123487457633, + "objective/train/docs_used": 535899, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4999566078186035, + "objective/train/original_loss": 2.4999563694000244, + "objective/train/theoretical_loss": 3.6737954292155814, + "objective/train/tokens_used": 952709600, + "objective/train/value_avg": -0.005481719970703125, + "objective/train/value_loss": 4.215823719277978e-05, + "objective/train/value_max": -0.0005769729614257812, + "objective/train/value_min": -0.0592041015625, + "objective/train/value_reward_corr": 0.48310716172074103, + "objective/train/value_std": 0.00434112548828125, + "objective/train/weight_avg": 0.9958279132843018, + "objective/train/weighted_lm_loss": 2.4976110458374023, + "objective/train/weights_max": 1.5350786447525024, + "objective/train/weights_min": 0.7104644179344177, + "theoretical_loss": 3.6737954292155814, + "tokens_seen": 932249600 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003624172517552658, + "loss": 1.3372, + "theoretical_loss": 3.6737954292155814, + "tokens_seen": 932249600 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036240722166499504, + "loss": 1.4567, + "theoretical_loss": 3.673770532508901, + "tokens_seen": 932315136 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036239719157472416, + "loss": 1.3313, + "theoretical_loss": 3.673745638042229, + "tokens_seen": 932380672 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003623871614844534, + "loss": 1.4058, + "theoretical_loss": 3.673720745815206, + "tokens_seen": 932446208 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003623771313941825, + "loss": 1.3517, + "theoretical_loss": 3.673695855827473, + "tokens_seen": 932511744 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036236710130391176, + "loss": 1.3547, + "theoretical_loss": 3.6736709680786706, + "tokens_seen": 932577280 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036235707121364094, + "loss": 1.3781, + "theoretical_loss": 3.6736460825684407, + "tokens_seen": 932642816 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003623470411233701, + "loss": 1.411, + "theoretical_loss": 3.6736211992964254, + "tokens_seen": 932708352 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003623370110330993, + "loss": 1.3878, + "theoretical_loss": 3.6735963182622644, + "tokens_seen": 932773888 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003623269809428285, + "loss": 1.5145, + "theoretical_loss": 3.6735714394656007, + "tokens_seen": 932839424 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036231695085255767, + "loss": 1.5056, + "theoretical_loss": 3.673546562906075, + "tokens_seen": 932904960 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003623069207622869, + "loss": 1.4709, + "theoretical_loss": 3.67352168858333, + "tokens_seen": 932970496 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036229689067201603, + "loss": 1.3232, + "theoretical_loss": 3.673496816497006, + "tokens_seen": 933036032 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036228686058174526, + "loss": 1.4275, + "theoretical_loss": 3.6734719466467465, + "tokens_seen": 933101568 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036227683049147444, + "loss": 1.2771, + "theoretical_loss": 3.6734470790321927, + "tokens_seen": 933167104 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003622668004012036, + "loss": 1.3701, + "theoretical_loss": 3.673422213652986, + "tokens_seen": 933232640 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003622567703109328, + "loss": 1.3505, + "theoretical_loss": 3.6733973505087696, + "tokens_seen": 933298176 + }, + { + "epoch": 0.28, + "learning_rate": 0.000362246740220662, + "loss": 1.3338, + "theoretical_loss": 3.673372489599185, + "tokens_seen": 933363712 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036223671013039117, + "loss": 1.3266, + "theoretical_loss": 3.6733476309238746, + "tokens_seen": 933429248 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003622266800401204, + "loss": 1.3264, + "theoretical_loss": 3.6733227744824815, + "tokens_seen": 933494784 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036221664994984953, + "loss": 1.416, + "theoretical_loss": 3.6732979202746465, + "tokens_seen": 933560320 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036220661985957877, + "loss": 1.3341, + "theoretical_loss": 3.6732730683000137, + "tokens_seen": 933625856 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003621965897693079, + "loss": 1.418, + "theoretical_loss": 3.673248218558225, + "tokens_seen": 933691392 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036218655967903713, + "loss": 1.4116, + "theoretical_loss": 3.6732233710489233, + "tokens_seen": 933756928 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003621765295887663, + "loss": 1.3475, + "theoretical_loss": 3.6731985257717508, + "tokens_seen": 933822464 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": 0.0006840224377810955, + "objective/train/docs_used": 536516, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.121621608734131, + "objective/train/original_loss": 3.121621608734131, + "objective/train/theoretical_loss": 3.6731736827263513, + "objective/train/tokens_used": 954348000, + "objective/train/value_avg": -0.004634857177734375, + "objective/train/value_loss": 4.8011082981247455e-05, + "objective/train/value_max": -0.00034999847412109375, + "objective/train/value_min": -0.050994873046875, + "objective/train/value_reward_corr": 0.32704141023854294, + "objective/train/value_std": 0.00384521484375, + "objective/train/weight_avg": 1.0090957880020142, + "objective/train/weighted_lm_loss": 3.1510169506073, + "objective/train/weights_max": 1.4680149555206299, + "objective/train/weights_min": 0.3876757323741913, + "theoretical_loss": 3.6731736827263513, + "tokens_seen": 933888000 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003621664994984955, + "loss": 1.4718, + "theoretical_loss": 3.6731736827263513, + "tokens_seen": 933888000 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036215646940822467, + "loss": 1.327, + "theoretical_loss": 3.6731488419123672, + "tokens_seen": 933953536 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036214643931795385, + "loss": 1.3395, + "theoretical_loss": 3.6731240033294412, + "tokens_seen": 934019072 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036213640922768303, + "loss": 1.3091, + "theoretical_loss": 3.673099166977217, + "tokens_seen": 934084608 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036212637913741227, + "loss": 1.438, + "theoretical_loss": 3.6730743328553377, + "tokens_seen": 934150144 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003621163490471414, + "loss": 1.3883, + "theoretical_loss": 3.6730495009634465, + "tokens_seen": 934215680 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036210631895687063, + "loss": 1.3511, + "theoretical_loss": 3.673024671301186, + "tokens_seen": 934281216 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003620962888665998, + "loss": 1.3524, + "theoretical_loss": 3.6729998438682006, + "tokens_seen": 934346752 + }, + { + "epoch": 0.28, + "learning_rate": 0.000362086258776329, + "loss": 1.3244, + "theoretical_loss": 3.6729750186641335, + "tokens_seen": 934412288 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003620762286860582, + "loss": 1.2826, + "theoretical_loss": 3.672950195688628, + "tokens_seen": 934477824 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036206619859578736, + "loss": 1.3547, + "theoretical_loss": 3.6729253749413284, + "tokens_seen": 934543360 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036205616850551654, + "loss": 1.462, + "theoretical_loss": 3.6729005564218777, + "tokens_seen": 934608896 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003620461384152458, + "loss": 1.3892, + "theoretical_loss": 3.6728757401299204, + "tokens_seen": 934674432 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003620361083249749, + "loss": 1.4383, + "theoretical_loss": 3.6728509260651006, + "tokens_seen": 934739968 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036202607823470414, + "loss": 1.4582, + "theoretical_loss": 3.6728261142270613, + "tokens_seen": 934805504 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036201604814443326, + "loss": 1.3325, + "theoretical_loss": 3.672801304615447, + "tokens_seen": 934871040 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003620060180541625, + "loss": 1.3264, + "theoretical_loss": 3.672776497229902, + "tokens_seen": 934936576 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003619959879638917, + "loss": 1.2767, + "theoretical_loss": 3.6727516920700705, + "tokens_seen": 935002112 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036198595787362086, + "loss": 1.5503, + "theoretical_loss": 3.672726889135597, + "tokens_seen": 935067648 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036197592778335004, + "loss": 1.3735, + "theoretical_loss": 3.6727020884261257, + "tokens_seen": 935133184 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003619658976930792, + "loss": 1.325, + "theoretical_loss": 3.6726772899413014, + "tokens_seen": 935198720 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003619558676028084, + "loss": 1.295, + "theoretical_loss": 3.6726524936807676, + "tokens_seen": 935264256 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036194583751253764, + "loss": 1.3693, + "theoretical_loss": 3.6726276996441705, + "tokens_seen": 935329792 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036193580742226677, + "loss": 1.4272, + "theoretical_loss": 3.672602907831154, + "tokens_seen": 935395328 + }, + { + "epoch": 0.28, + "learning_rate": 0.000361925777331996, + "loss": 1.3274, + "theoretical_loss": 3.672578118241363, + "tokens_seen": 935460864 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.001239054137840867, + "objective/train/docs_used": 537462, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3712713718414307, + "objective/train/original_loss": 2.3712713718414307, + "objective/train/theoretical_loss": 3.6725533308744422, + "objective/train/tokens_used": 955986400, + "objective/train/value_avg": -0.004425048828125, + "objective/train/value_loss": 6.067777576390654e-05, + "objective/train/value_max": -0.00034737586975097656, + "objective/train/value_min": -0.037384033203125, + "objective/train/value_reward_corr": 0.4004911860810898, + "objective/train/value_std": 0.003208160400390625, + "objective/train/weight_avg": 0.9904459118843079, + "objective/train/weighted_lm_loss": 2.3512938022613525, + "objective/train/weights_max": 1.3466453552246094, + "objective/train/weights_min": 0.5904394388198853, + "theoretical_loss": 3.6725533308744422, + "tokens_seen": 935526400 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003619157472417252, + "loss": 1.3042, + "theoretical_loss": 3.6725533308744422, + "tokens_seen": 935526400 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036190571715145436, + "loss": 1.3911, + "theoretical_loss": 3.6725285457300374, + "tokens_seen": 935591936 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036189568706118354, + "loss": 1.5359, + "theoretical_loss": 3.672503762807793, + "tokens_seen": 935657472 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003618856569709127, + "loss": 1.2773, + "theoretical_loss": 3.6724789821073536, + "tokens_seen": 935723008 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003618756268806419, + "loss": 1.4513, + "theoretical_loss": 3.672454203628365, + "tokens_seen": 935788544 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036186559679037114, + "loss": 1.3577, + "theoretical_loss": 3.6724294273704725, + "tokens_seen": 935854080 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003618555667001003, + "loss": 1.41, + "theoretical_loss": 3.672404653333322, + "tokens_seen": 935919616 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003618455366098295, + "loss": 1.367, + "theoretical_loss": 3.6723798815165583, + "tokens_seen": 935985152 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003618355065195587, + "loss": 1.3695, + "theoretical_loss": 3.672355111919827, + "tokens_seen": 936050688 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036182547642928787, + "loss": 1.4107, + "theoretical_loss": 3.672330344542774, + "tokens_seen": 936116224 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003618154463390171, + "loss": 1.4225, + "theoretical_loss": 3.672305579385045, + "tokens_seen": 936181760 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036180541624874623, + "loss": 1.4436, + "theoretical_loss": 3.6722808164462855, + "tokens_seen": 936247296 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036179538615847546, + "loss": 1.3258, + "theoretical_loss": 3.672256055726142, + "tokens_seen": 936312832 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036178535606820464, + "loss": 1.4369, + "theoretical_loss": 3.6722312972242594, + "tokens_seen": 936378368 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003617753259779338, + "loss": 1.419, + "theoretical_loss": 3.672206540940284, + "tokens_seen": 936443904 + }, + { + "epoch": 0.28, + "learning_rate": 0.000361765295887663, + "loss": 1.3882, + "theoretical_loss": 3.672181786873863, + "tokens_seen": 936509440 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003617552657973922, + "loss": 1.3329, + "theoretical_loss": 3.6721570350246417, + "tokens_seen": 936574976 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036174523570712137, + "loss": 1.3319, + "theoretical_loss": 3.6721322853922667, + "tokens_seen": 936640512 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003617352056168506, + "loss": 1.3123, + "theoretical_loss": 3.672107537976384, + "tokens_seen": 936706048 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036172517552657973, + "loss": 1.4198, + "theoretical_loss": 3.67208279277664, + "tokens_seen": 936771584 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036171514543630897, + "loss": 1.372, + "theoretical_loss": 3.672058049792682, + "tokens_seen": 936837120 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003617051153460381, + "loss": 1.3912, + "theoretical_loss": 3.6720333090241555, + "tokens_seen": 936902656 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036169508525576733, + "loss": 1.3639, + "theoretical_loss": 3.672008570470708, + "tokens_seen": 936968192 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003616850551654965, + "loss": 1.4311, + "theoretical_loss": 3.6719838341319866, + "tokens_seen": 937033728 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003616750250752257, + "loss": 1.3776, + "theoretical_loss": 3.671959100007637, + "tokens_seen": 937099264 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.0007029033149592578, + "objective/train/docs_used": 538123, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8850114345550537, + "objective/train/original_loss": 2.8850114345550537, + "objective/train/theoretical_loss": 3.6719343680973067, + "objective/train/tokens_used": 957624800, + "objective/train/value_avg": -0.0047454833984375, + "objective/train/value_loss": 5.632851753034629e-05, + "objective/train/value_max": -0.00043892860412597656, + "objective/train/value_min": -0.049591064453125, + "objective/train/value_reward_corr": 0.3946850652847342, + "objective/train/value_std": 0.00370025634765625, + "objective/train/weight_avg": 0.9956126809120178, + "objective/train/weighted_lm_loss": 2.8613078594207764, + "objective/train/weights_max": 1.6419928073883057, + "objective/train/weights_min": 0.389573335647583, + "theoretical_loss": 3.6719343680973067, + "tokens_seen": 937164800 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036166499498495487, + "loss": 1.4032, + "theoretical_loss": 3.6719343680973067, + "tokens_seen": 937164800 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036165496489468405, + "loss": 1.46, + "theoretical_loss": 3.671909638400643, + "tokens_seen": 937230336 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036164493480441323, + "loss": 1.3963, + "theoretical_loss": 3.671884910917292, + "tokens_seen": 937295872 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036163490471414247, + "loss": 1.206, + "theoretical_loss": 3.6718601856469024, + "tokens_seen": 937361408 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003616248746238716, + "loss": 1.4979, + "theoretical_loss": 3.6718354625891205, + "tokens_seen": 937426944 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036161484453360083, + "loss": 1.4139, + "theoretical_loss": 3.671810741743594, + "tokens_seen": 937492480 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036160481444333, + "loss": 1.3982, + "theoretical_loss": 3.6717860231099695, + "tokens_seen": 937558016 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003615947843530592, + "loss": 1.2788, + "theoretical_loss": 3.6717613066878956, + "tokens_seen": 937623552 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003615847542627884, + "loss": 1.3981, + "theoretical_loss": 3.671736592477019, + "tokens_seen": 937689088 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036157472417251756, + "loss": 1.4182, + "theoretical_loss": 3.6717118804769875, + "tokens_seen": 937754624 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036156469408224674, + "loss": 1.4583, + "theoretical_loss": 3.67168717068745, + "tokens_seen": 937820160 + }, + { + "epoch": 0.28, + "learning_rate": 0.000361554663991976, + "loss": 1.3829, + "theoretical_loss": 3.6716624631080528, + "tokens_seen": 937885696 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003615446339017051, + "loss": 1.3027, + "theoretical_loss": 3.671637757738444, + "tokens_seen": 937951232 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036153460381143434, + "loss": 1.3646, + "theoretical_loss": 3.671613054578273, + "tokens_seen": 938016768 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036152457372116346, + "loss": 1.3729, + "theoretical_loss": 3.671588353627186, + "tokens_seen": 938082304 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003615145436308927, + "loss": 1.4012, + "theoretical_loss": 3.6715636548848325, + "tokens_seen": 938147840 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003615045135406219, + "loss": 1.4534, + "theoretical_loss": 3.6715389583508595, + "tokens_seen": 938213376 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036149448345035106, + "loss": 1.5102, + "theoretical_loss": 3.6715142640249168, + "tokens_seen": 938278912 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036148445336008024, + "loss": 1.4344, + "theoretical_loss": 3.6714895719066516, + "tokens_seen": 938344448 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003614744232698094, + "loss": 1.407, + "theoretical_loss": 3.6714648819957123, + "tokens_seen": 938409984 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003614643931795386, + "loss": 1.5233, + "theoretical_loss": 3.6714401942917485, + "tokens_seen": 938475520 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036145436308926784, + "loss": 1.3483, + "theoretical_loss": 3.6714155087944085, + "tokens_seen": 938541056 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036144433299899697, + "loss": 1.3343, + "theoretical_loss": 3.67139082550334, + "tokens_seen": 938606592 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003614343029087262, + "loss": 1.4038, + "theoretical_loss": 3.671366144418193, + "tokens_seen": 938672128 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003614242728184554, + "loss": 1.3699, + "theoretical_loss": 3.6713414655386156, + "tokens_seen": 938737664 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -3.207070039934479e-05, + "objective/train/docs_used": 539283, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8074305057525635, + "objective/train/original_loss": 2.8074302673339844, + "objective/train/theoretical_loss": 3.6713167888642575, + "objective/train/tokens_used": 959263200, + "objective/train/value_avg": -0.00499725341796875, + "objective/train/value_loss": 5.507756213773973e-05, + "objective/train/value_max": -0.00035834312438964844, + "objective/train/value_min": -0.062225341796875, + "objective/train/value_reward_corr": 0.46306306321744695, + "objective/train/value_std": 0.004444122314453125, + "objective/train/weight_avg": 1.0023173093795776, + "objective/train/weighted_lm_loss": 2.8159565925598145, + "objective/train/weights_max": 1.6651142835617065, + "objective/train/weights_min": 0.2997286915779114, + "theoretical_loss": 3.6713167888642575, + "tokens_seen": 938803200 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036141424272818456, + "loss": 1.3701, + "theoretical_loss": 3.6713167888642575, + "tokens_seen": 938803200 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036140421263791374, + "loss": 1.4067, + "theoretical_loss": 3.6712921143947668, + "tokens_seen": 938868736 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003613941825476429, + "loss": 1.3941, + "theoretical_loss": 3.671267442129793, + "tokens_seen": 938934272 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003613841524573721, + "loss": 1.2778, + "theoretical_loss": 3.6712427720689855, + "tokens_seen": 938999808 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036137412236710134, + "loss": 1.3166, + "theoretical_loss": 3.6712181042119933, + "tokens_seen": 939065344 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036136409227683047, + "loss": 1.3613, + "theoretical_loss": 3.6711934385584666, + "tokens_seen": 939130880 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003613540621865597, + "loss": 1.3388, + "theoretical_loss": 3.671168775108053, + "tokens_seen": 939196416 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036134403209628883, + "loss": 1.5002, + "theoretical_loss": 3.6711441138604037, + "tokens_seen": 939261952 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036133400200601807, + "loss": 1.339, + "theoretical_loss": 3.6711194548151678, + "tokens_seen": 939327488 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036132397191574725, + "loss": 1.4507, + "theoretical_loss": 3.671094797971995, + "tokens_seen": 939393024 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036131394182547643, + "loss": 1.2497, + "theoretical_loss": 3.6710701433305344, + "tokens_seen": 939458560 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003613039117352056, + "loss": 1.451, + "theoretical_loss": 3.6710454908904366, + "tokens_seen": 939524096 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036129388164493484, + "loss": 1.4093, + "theoretical_loss": 3.671020840651351, + "tokens_seen": 939589632 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036128385155466397, + "loss": 1.3476, + "theoretical_loss": 3.6709961926129284, + "tokens_seen": 939655168 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003612738214643932, + "loss": 1.3867, + "theoretical_loss": 3.670971546774818, + "tokens_seen": 939720704 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036126379137412233, + "loss": 1.3519, + "theoretical_loss": 3.67094690313667, + "tokens_seen": 939786240 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036125376128385157, + "loss": 1.3644, + "theoretical_loss": 3.6709222616981356, + "tokens_seen": 939851776 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036124373119358075, + "loss": 1.4031, + "theoretical_loss": 3.670897622458864, + "tokens_seen": 939917312 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036123370110330993, + "loss": 1.4926, + "theoretical_loss": 3.670872985418506, + "tokens_seen": 939982848 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003612236710130391, + "loss": 1.4048, + "theoretical_loss": 3.670848350576712, + "tokens_seen": 940048384 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003612136409227683, + "loss": 1.3986, + "theoretical_loss": 3.670823717933133, + "tokens_seen": 940113920 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003612036108324975, + "loss": 1.4652, + "theoretical_loss": 3.6707990874874192, + "tokens_seen": 940179456 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003611935807422267, + "loss": 1.3587, + "theoretical_loss": 3.670774459239221, + "tokens_seen": 940244992 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036118355065195584, + "loss": 1.4773, + "theoretical_loss": 3.6707498331881903, + "tokens_seen": 940310528 + }, + { + "epoch": 0.28, + "learning_rate": 0.00036117352056168507, + "loss": 1.4834, + "theoretical_loss": 3.670725209333976, + "tokens_seen": 940376064 + }, + { + "epoch": 0.28, + "objective/train/advantage_avg": -0.0002911534102167934, + "objective/train/docs_used": 539932, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.100034713745117, + "objective/train/original_loss": 3.100034475326538, + "objective/train/theoretical_loss": 3.6707005876762313, + "objective/train/tokens_used": 960901600, + "objective/train/value_avg": -0.00478363037109375, + "objective/train/value_loss": 5.601146040135063e-05, + "objective/train/value_max": -0.0003726482391357422, + "objective/train/value_min": -0.049041748046875, + "objective/train/value_reward_corr": 0.31877716287515945, + "objective/train/value_std": 0.0035457611083984375, + "objective/train/weight_avg": 0.9997032880783081, + "objective/train/weighted_lm_loss": 3.100522518157959, + "objective/train/weights_max": 1.4165233373641968, + "objective/train/weights_min": 0.3792358934879303, + "theoretical_loss": 3.6707005876762313, + "tokens_seen": 940441600 + }, + { + "epoch": 0.28, + "learning_rate": 0.0003611634904714142, + "loss": 1.4149, + "theoretical_loss": 3.6707005876762313, + "tokens_seen": 940441600 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036115346038114343, + "loss": 1.3219, + "theoretical_loss": 3.670675968214606, + "tokens_seen": 940507136 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003611434302908726, + "loss": 1.323, + "theoretical_loss": 3.6706513509487513, + "tokens_seen": 940572672 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003611334002006018, + "loss": 1.3962, + "theoretical_loss": 3.6706267358783187, + "tokens_seen": 940638208 + }, + { + "epoch": 0.29, + "learning_rate": 0.000361123370110331, + "loss": 1.3335, + "theoretical_loss": 3.6706021230029586, + "tokens_seen": 940703744 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003611133400200602, + "loss": 1.3539, + "theoretical_loss": 3.670577512322324, + "tokens_seen": 940769280 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003611033099297894, + "loss": 1.378, + "theoretical_loss": 3.6705529038360645, + "tokens_seen": 940834816 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003610932798395186, + "loss": 1.4135, + "theoretical_loss": 3.670528297543833, + "tokens_seen": 940900352 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036108324974924776, + "loss": 1.4166, + "theoretical_loss": 3.6705036934452804, + "tokens_seen": 940965888 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036107321965897694, + "loss": 1.4393, + "theoretical_loss": 3.6704790915400585, + "tokens_seen": 941031424 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003610631895687062, + "loss": 1.5701, + "theoretical_loss": 3.6704544918278192, + "tokens_seen": 941096960 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003610531594784353, + "loss": 1.4402, + "theoretical_loss": 3.670429894308214, + "tokens_seen": 941162496 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036104312938816454, + "loss": 1.4461, + "theoretical_loss": 3.670405298980895, + "tokens_seen": 941228032 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036103309929789366, + "loss": 1.2713, + "theoretical_loss": 3.6703807058455142, + "tokens_seen": 941293568 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003610230692076229, + "loss": 1.4039, + "theoretical_loss": 3.6703561149017236, + "tokens_seen": 941359104 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003610130391173521, + "loss": 1.4035, + "theoretical_loss": 3.670331526149176, + "tokens_seen": 941424640 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036100300902708126, + "loss": 1.3212, + "theoretical_loss": 3.670306939587522, + "tokens_seen": 941490176 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036099297893681044, + "loss": 1.3108, + "theoretical_loss": 3.6702823552164157, + "tokens_seen": 941555712 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003609829488465396, + "loss": 1.3695, + "theoretical_loss": 3.6702577730355084, + "tokens_seen": 941621248 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003609729187562688, + "loss": 1.4135, + "theoretical_loss": 3.670233193044453, + "tokens_seen": 941686784 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036096288866599804, + "loss": 1.4199, + "theoretical_loss": 3.6702086152429017, + "tokens_seen": 941752320 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036095285857572717, + "loss": 1.3367, + "theoretical_loss": 3.670184039630507, + "tokens_seen": 941817856 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003609428284854564, + "loss": 1.4875, + "theoretical_loss": 3.670159466206923, + "tokens_seen": 941883392 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003609327983951856, + "loss": 1.4132, + "theoretical_loss": 3.6701348949718002, + "tokens_seen": 941948928 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036092276830491476, + "loss": 1.368, + "theoretical_loss": 3.670110325924793, + "tokens_seen": 942014464 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": 0.00010685604502214119, + "objective/train/docs_used": 541311, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7775285243988037, + "objective/train/original_loss": 2.777528762817383, + "objective/train/theoretical_loss": 3.6700857590655547, + "objective/train/tokens_used": 962540000, + "objective/train/value_avg": -0.0048828125, + "objective/train/value_loss": 4.834992068936117e-05, + "objective/train/value_max": -0.0002982616424560547, + "objective/train/value_min": -0.0638427734375, + "objective/train/value_reward_corr": 0.28158043417687023, + "objective/train/value_std": 0.004119873046875, + "objective/train/weight_avg": 1.0033016204833984, + "objective/train/weighted_lm_loss": 2.7828657627105713, + "objective/train/weights_max": 1.6614800691604614, + "objective/train/weights_min": 0.08931885659694672, + "theoretical_loss": 3.6700857590655547, + "tokens_seen": 942080000 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036091273821464394, + "loss": 1.3362, + "theoretical_loss": 3.6700857590655547, + "tokens_seen": 942080000 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003609027081243731, + "loss": 1.3581, + "theoretical_loss": 3.6700611943937367, + "tokens_seen": 942145536 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003608926780341023, + "loss": 1.3989, + "theoretical_loss": 3.670036631908993, + "tokens_seen": 942211072 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036088264794383154, + "loss": 1.2186, + "theoretical_loss": 3.6700120716109774, + "tokens_seen": 942276608 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036087261785356067, + "loss": 1.444, + "theoretical_loss": 3.6699875134993425, + "tokens_seen": 942342144 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003608625877632899, + "loss": 1.3588, + "theoretical_loss": 3.6699629575737416, + "tokens_seen": 942407680 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036085255767301903, + "loss": 1.1787, + "theoretical_loss": 3.669938403833828, + "tokens_seen": 942473216 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036084252758274827, + "loss": 1.4371, + "theoretical_loss": 3.6699138522792554, + "tokens_seen": 942538752 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036083249749247745, + "loss": 1.4967, + "theoretical_loss": 3.6698893029096773, + "tokens_seen": 942604288 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036082246740220663, + "loss": 1.3457, + "theoretical_loss": 3.6698647557247472, + "tokens_seen": 942669824 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003608124373119358, + "loss": 1.435, + "theoretical_loss": 3.6698402107241193, + "tokens_seen": 942735360 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036080240722166505, + "loss": 1.386, + "theoretical_loss": 3.6698156679074474, + "tokens_seen": 942800896 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036079237713139417, + "loss": 1.363, + "theoretical_loss": 3.6697911272743844, + "tokens_seen": 942866432 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003607823470411234, + "loss": 1.3645, + "theoretical_loss": 3.6697665888245856, + "tokens_seen": 942931968 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036077231695085253, + "loss": 1.4939, + "theoretical_loss": 3.6697420525577042, + "tokens_seen": 942997504 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036076228686058177, + "loss": 1.3862, + "theoretical_loss": 3.6697175184733943, + "tokens_seen": 943063040 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036075225677031095, + "loss": 1.3422, + "theoretical_loss": 3.6696929865713104, + "tokens_seen": 943128576 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036074222668004013, + "loss": 1.3974, + "theoretical_loss": 3.669668456851107, + "tokens_seen": 943194112 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003607321965897693, + "loss": 1.4023, + "theoretical_loss": 3.6696439293124383, + "tokens_seen": 943259648 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003607221664994985, + "loss": 1.2668, + "theoretical_loss": 3.6696194039549583, + "tokens_seen": 943325184 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003607121364092277, + "loss": 1.2648, + "theoretical_loss": 3.669594880778322, + "tokens_seen": 943390720 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003607021063189569, + "loss": 1.3344, + "theoretical_loss": 3.669570359782184, + "tokens_seen": 943456256 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036069207622868604, + "loss": 1.3878, + "theoretical_loss": 3.6695458409661983, + "tokens_seen": 943521792 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036068204613841527, + "loss": 1.3566, + "theoretical_loss": 3.66952132433002, + "tokens_seen": 943587328 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003606720160481444, + "loss": 1.2495, + "theoretical_loss": 3.669496809873305, + "tokens_seen": 943652864 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.0012557520531117916, + "objective/train/docs_used": 542104, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9883530139923096, + "objective/train/original_loss": 2.9883527755737305, + "objective/train/theoretical_loss": 3.6694722975957066, + "objective/train/tokens_used": 964178400, + "objective/train/value_avg": -0.00481414794921875, + "objective/train/value_loss": 5.859800148755312e-05, + "objective/train/value_max": -0.0003514289855957031, + "objective/train/value_min": -0.0743408203125, + "objective/train/value_reward_corr": 0.38084092297629585, + "objective/train/value_std": 0.00366973876953125, + "objective/train/weight_avg": 0.9902141094207764, + "objective/train/weighted_lm_loss": 2.979642152786255, + "objective/train/weights_max": 2.103091239929199, + "objective/train/weights_min": 0.42347070574760437, + "theoretical_loss": 3.6694722975957066, + "tokens_seen": 943718400 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036066198595787364, + "loss": 1.4614, + "theoretical_loss": 3.6694722975957066, + "tokens_seen": 943718400 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003606519558676028, + "loss": 1.3538, + "theoretical_loss": 3.6694477874968805, + "tokens_seen": 943783936 + }, + { + "epoch": 0.29, + "learning_rate": 0.000360641925777332, + "loss": 1.1962, + "theoretical_loss": 3.669423279576482, + "tokens_seen": 943849472 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003606318956870612, + "loss": 1.3771, + "theoretical_loss": 3.669398773834166, + "tokens_seen": 943915008 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003606218655967904, + "loss": 1.3683, + "theoretical_loss": 3.6693742702695875, + "tokens_seen": 943980544 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036061183550651954, + "loss": 1.3384, + "theoretical_loss": 3.6693497688824026, + "tokens_seen": 944046080 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003606018054162488, + "loss": 1.3334, + "theoretical_loss": 3.6693252696722656, + "tokens_seen": 944111616 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003605917753259779, + "loss": 1.385, + "theoretical_loss": 3.6693007726388327, + "tokens_seen": 944177152 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036058174523570714, + "loss": 1.4183, + "theoretical_loss": 3.669276277781759, + "tokens_seen": 944242688 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003605717151454363, + "loss": 1.3504, + "theoretical_loss": 3.6692517851007, + "tokens_seen": 944308224 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003605616850551655, + "loss": 1.4096, + "theoretical_loss": 3.6692272945953124, + "tokens_seen": 944373760 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003605516549648947, + "loss": 1.4068, + "theoretical_loss": 3.669202806265251, + "tokens_seen": 944439296 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036054162487462386, + "loss": 1.3638, + "theoretical_loss": 3.6691783201101718, + "tokens_seen": 944504832 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036053159478435304, + "loss": 1.3325, + "theoretical_loss": 3.6691538361297313, + "tokens_seen": 944570368 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003605215646940823, + "loss": 1.491, + "theoretical_loss": 3.6691293543235846, + "tokens_seen": 944635904 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003605115346038114, + "loss": 1.2766, + "theoretical_loss": 3.6691048746913886, + "tokens_seen": 944701440 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036050150451354064, + "loss": 1.3512, + "theoretical_loss": 3.6690803972327988, + "tokens_seen": 944766976 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036049147442326977, + "loss": 1.4278, + "theoretical_loss": 3.669055921947472, + "tokens_seen": 944832512 + }, + { + "epoch": 0.29, + "learning_rate": 0.000360481444332999, + "loss": 1.2956, + "theoretical_loss": 3.669031448835064, + "tokens_seen": 944898048 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003604714142427282, + "loss": 1.2392, + "theoretical_loss": 3.669006977895232, + "tokens_seen": 944963584 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036046138415245737, + "loss": 1.2382, + "theoretical_loss": 3.668982509127632, + "tokens_seen": 945029120 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036045135406218655, + "loss": 1.3897, + "theoretical_loss": 3.66895804253192, + "tokens_seen": 945094656 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003604413239719158, + "loss": 1.4081, + "theoretical_loss": 3.6689335781077537, + "tokens_seen": 945160192 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003604312938816449, + "loss": 1.3437, + "theoretical_loss": 3.6689091158547886, + "tokens_seen": 945225728 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036042126379137414, + "loss": 1.3947, + "theoretical_loss": 3.6688846557726826, + "tokens_seen": 945291264 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.0003184693050570786, + "objective/train/docs_used": 543471, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.85571026802063, + "objective/train/original_loss": 2.85571026802063, + "objective/train/theoretical_loss": 3.6688601978610915, + "objective/train/tokens_used": 965816800, + "objective/train/value_avg": -0.00472259521484375, + "objective/train/value_loss": 7.156668289098889e-05, + "objective/train/value_max": -0.0002491474151611328, + "objective/train/value_min": -0.07330322265625, + "objective/train/value_reward_corr": 0.4060170373363072, + "objective/train/value_std": 0.00437164306640625, + "objective/train/weight_avg": 0.9999953508377075, + "objective/train/weighted_lm_loss": 2.86010479927063, + "objective/train/weights_max": 1.7246841192245483, + "objective/train/weights_min": 0.19863732159137726, + "theoretical_loss": 3.6688601978610915, + "tokens_seen": 945356800 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036041123370110327, + "loss": 1.3992, + "theoretical_loss": 3.6688601978610915, + "tokens_seen": 945356800 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003604012036108325, + "loss": 1.3534, + "theoretical_loss": 3.668835742119674, + "tokens_seen": 945422336 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003603911735205617, + "loss": 1.283, + "theoretical_loss": 3.6688112885480857, + "tokens_seen": 945487872 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036038114343029087, + "loss": 1.4287, + "theoretical_loss": 3.6687868371459835, + "tokens_seen": 945553408 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036037111334002005, + "loss": 1.3177, + "theoretical_loss": 3.6687623879130253, + "tokens_seen": 945618944 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036036108324974923, + "loss": 1.3602, + "theoretical_loss": 3.6687379408488683, + "tokens_seen": 945684480 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036035105315947847, + "loss": 1.41, + "theoretical_loss": 3.668713495953169, + "tokens_seen": 945750016 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036034102306920765, + "loss": 1.3797, + "theoretical_loss": 3.6686890532255862, + "tokens_seen": 945815552 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036033099297893683, + "loss": 1.4536, + "theoretical_loss": 3.6686646126657765, + "tokens_seen": 945881088 + }, + { + "epoch": 0.29, + "learning_rate": 0.000360320962888666, + "loss": 1.4236, + "theoretical_loss": 3.6686401742733974, + "tokens_seen": 945946624 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036031093279839525, + "loss": 1.4153, + "theoretical_loss": 3.668615738048107, + "tokens_seen": 946012160 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036030090270812437, + "loss": 1.3955, + "theoretical_loss": 3.6685913039895635, + "tokens_seen": 946077696 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003602908726178536, + "loss": 1.4905, + "theoretical_loss": 3.668566872097423, + "tokens_seen": 946143232 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036028084252758273, + "loss": 1.3524, + "theoretical_loss": 3.6685424423713444, + "tokens_seen": 946208768 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036027081243731197, + "loss": 1.3221, + "theoretical_loss": 3.6685180148109864, + "tokens_seen": 946274304 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036026078234704115, + "loss": 1.3759, + "theoretical_loss": 3.6684935894160056, + "tokens_seen": 946339840 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036025075225677033, + "loss": 1.543, + "theoretical_loss": 3.6684691661860613, + "tokens_seen": 946405376 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003602407221664995, + "loss": 1.4708, + "theoretical_loss": 3.6684447451208113, + "tokens_seen": 946470912 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003602306920762287, + "loss": 1.2895, + "theoretical_loss": 3.6684203262199135, + "tokens_seen": 946536448 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003602206619859579, + "loss": 1.4004, + "theoretical_loss": 3.668395909483026, + "tokens_seen": 946601984 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003602106318956871, + "loss": 1.2541, + "theoretical_loss": 3.6683714949098087, + "tokens_seen": 946667520 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036020060180541624, + "loss": 1.314, + "theoretical_loss": 3.6683470824999187, + "tokens_seen": 946733056 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003601905717151455, + "loss": 1.3094, + "theoretical_loss": 3.668322672253015, + "tokens_seen": 946798592 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003601805416248746, + "loss": 1.4117, + "theoretical_loss": 3.6682982641687563, + "tokens_seen": 946864128 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036017051153460384, + "loss": 1.512, + "theoretical_loss": 3.6682738582468013, + "tokens_seen": 946929664 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.0012964954366907477, + "objective/train/docs_used": 544140, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6068849563598633, + "objective/train/original_loss": 2.6068849563598633, + "objective/train/theoretical_loss": 3.668249454486809, + "objective/train/tokens_used": 967455200, + "objective/train/value_avg": -0.005931854248046875, + "objective/train/value_loss": 6.854951789136976e-05, + "objective/train/value_max": -0.00033926963806152344, + "objective/train/value_min": -0.058135986328125, + "objective/train/value_reward_corr": 0.3023901497464335, + "objective/train/value_std": 0.004474639892578125, + "objective/train/weight_avg": 0.9901732802391052, + "objective/train/weighted_lm_loss": 2.587177038192749, + "objective/train/weights_max": 1.7884689569473267, + "objective/train/weights_min": 0.2018454670906067, + "theoretical_loss": 3.668249454486809, + "tokens_seen": 946995200 + }, + { + "epoch": 0.29, + "learning_rate": 0.000360160481444333, + "loss": 1.4215, + "theoretical_loss": 3.668249454486809, + "tokens_seen": 946995200 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003601504513540622, + "loss": 1.4683, + "theoretical_loss": 3.6682250528884377, + "tokens_seen": 947060736 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003601404212637914, + "loss": 1.4505, + "theoretical_loss": 3.668200653451347, + "tokens_seen": 947126272 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003601303911735206, + "loss": 1.3741, + "theoretical_loss": 3.6681762561751956, + "tokens_seen": 947191808 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036012036108324974, + "loss": 1.3454, + "theoretical_loss": 3.668151861059643, + "tokens_seen": 947257344 + }, + { + "epoch": 0.29, + "learning_rate": 0.000360110330992979, + "loss": 1.3725, + "theoretical_loss": 3.6681274681043483, + "tokens_seen": 947322880 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003601003009027081, + "loss": 1.4798, + "theoretical_loss": 3.6681030773089702, + "tokens_seen": 947388416 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036009027081243734, + "loss": 1.373, + "theoretical_loss": 3.6680786886731687, + "tokens_seen": 947453952 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003600802407221665, + "loss": 1.3738, + "theoretical_loss": 3.6680543021966026, + "tokens_seen": 947519488 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003600702106318957, + "loss": 1.455, + "theoretical_loss": 3.668029917878932, + "tokens_seen": 947585024 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003600601805416249, + "loss": 1.4879, + "theoretical_loss": 3.6680055357198165, + "tokens_seen": 947650560 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036005015045135406, + "loss": 1.3076, + "theoretical_loss": 3.6679811557189153, + "tokens_seen": 947716096 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036004012036108324, + "loss": 1.3399, + "theoretical_loss": 3.667956777875889, + "tokens_seen": 947781632 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003600300902708125, + "loss": 1.3254, + "theoretical_loss": 3.667932402190396, + "tokens_seen": 947847168 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003600200601805416, + "loss": 1.4725, + "theoretical_loss": 3.6679080286620973, + "tokens_seen": 947912704 + }, + { + "epoch": 0.29, + "learning_rate": 0.00036001003009027084, + "loss": 1.4596, + "theoretical_loss": 3.667883657290653, + "tokens_seen": 947978240 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035999999999999997, + "loss": 1.3695, + "theoretical_loss": 3.6678592880757224, + "tokens_seen": 948043776 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003599899699097292, + "loss": 1.4308, + "theoretical_loss": 3.667834921016966, + "tokens_seen": 948109312 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003599799398194584, + "loss": 1.4284, + "theoretical_loss": 3.6678105561140435, + "tokens_seen": 948174848 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035996990972918757, + "loss": 1.3323, + "theoretical_loss": 3.6677861933666156, + "tokens_seen": 948240384 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035995987963891675, + "loss": 1.3823, + "theoretical_loss": 3.6677618327743433, + "tokens_seen": 948305920 + }, + { + "epoch": 0.29, + "learning_rate": 0.000359949849548646, + "loss": 1.4385, + "theoretical_loss": 3.667737474336886, + "tokens_seen": 948371456 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003599398194583751, + "loss": 1.4468, + "theoretical_loss": 3.667713118053905, + "tokens_seen": 948436992 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035992978936810434, + "loss": 1.4222, + "theoretical_loss": 3.66768876392506, + "tokens_seen": 948502528 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035991975927783347, + "loss": 1.4824, + "theoretical_loss": 3.6676644119500126, + "tokens_seen": 948568064 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.0011453505139797926, + "objective/train/docs_used": 544543, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.428448438644409, + "objective/train/original_loss": 2.428448438644409, + "objective/train/theoretical_loss": 3.667640062128423, + "objective/train/tokens_used": 969093600, + "objective/train/value_avg": -0.00572967529296875, + "objective/train/value_loss": 5.8053403336089104e-05, + "objective/train/value_max": -0.0003845691680908203, + "objective/train/value_min": -0.05560302734375, + "objective/train/value_reward_corr": 0.41790419484561503, + "objective/train/value_std": 0.00501251220703125, + "objective/train/weight_avg": 0.9912968277931213, + "objective/train/weighted_lm_loss": 2.426102638244629, + "objective/train/weights_max": 1.640490174293518, + "objective/train/weights_min": 0.3626672029495239, + "theoretical_loss": 3.667640062128423, + "tokens_seen": 948633600 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003599097291875627, + "loss": 1.2708, + "theoretical_loss": 3.667640062128423, + "tokens_seen": 948633600 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003598996990972919, + "loss": 1.4554, + "theoretical_loss": 3.667615714459952, + "tokens_seen": 948699136 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035988966900702107, + "loss": 1.4322, + "theoretical_loss": 3.6675913689442607, + "tokens_seen": 948764672 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035987963891675025, + "loss": 1.4373, + "theoretical_loss": 3.66756702558101, + "tokens_seen": 948830208 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035986960882647943, + "loss": 1.4157, + "theoretical_loss": 3.6675426843698613, + "tokens_seen": 948895744 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003598595787362086, + "loss": 1.423, + "theoretical_loss": 3.6675183453104747, + "tokens_seen": 948961280 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035984954864593785, + "loss": 1.3896, + "theoretical_loss": 3.667494008402513, + "tokens_seen": 949026816 + }, + { + "epoch": 0.29, + "learning_rate": 0.000359839518555667, + "loss": 1.4329, + "theoretical_loss": 3.667469673645636, + "tokens_seen": 949092352 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003598294884653962, + "loss": 1.4446, + "theoretical_loss": 3.667445341039505, + "tokens_seen": 949157888 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035981945837512534, + "loss": 1.5377, + "theoretical_loss": 3.667421010583783, + "tokens_seen": 949223424 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035980942828485457, + "loss": 1.3488, + "theoretical_loss": 3.66739668227813, + "tokens_seen": 949288960 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035979939819458375, + "loss": 1.4533, + "theoretical_loss": 3.6673723561222085, + "tokens_seen": 949354496 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035978936810431293, + "loss": 1.3305, + "theoretical_loss": 3.6673480321156795, + "tokens_seen": 949420032 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003597793380140421, + "loss": 1.2738, + "theoretical_loss": 3.667323710258205, + "tokens_seen": 949485568 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035976930792377135, + "loss": 1.4107, + "theoretical_loss": 3.667299390549447, + "tokens_seen": 949551104 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003597592778335005, + "loss": 1.4439, + "theoretical_loss": 3.6672750729890673, + "tokens_seen": 949616640 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003597492477432297, + "loss": 1.4081, + "theoretical_loss": 3.6672507575767277, + "tokens_seen": 949682176 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035973921765295884, + "loss": 1.3776, + "theoretical_loss": 3.6672264443120905, + "tokens_seen": 949747712 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003597291875626881, + "loss": 1.3836, + "theoretical_loss": 3.6672021331948175, + "tokens_seen": 949813248 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035971915747241726, + "loss": 1.4509, + "theoretical_loss": 3.667177824224571, + "tokens_seen": 949878784 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035970912738214644, + "loss": 1.2834, + "theoretical_loss": 3.6671535174010135, + "tokens_seen": 949944320 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003596990972918756, + "loss": 1.4314, + "theoretical_loss": 3.6671292127238067, + "tokens_seen": 950009856 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003596890672016048, + "loss": 1.3827, + "theoretical_loss": 3.667104910192614, + "tokens_seen": 950075392 + }, + { + "epoch": 0.29, + "learning_rate": 0.000359679037111334, + "loss": 1.3991, + "theoretical_loss": 3.6670806098070976, + "tokens_seen": 950140928 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003596690070210632, + "loss": 1.3635, + "theoretical_loss": 3.6670563115669195, + "tokens_seen": 950206464 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.0005420548259280622, + "objective/train/docs_used": 545942, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9442925453186035, + "objective/train/original_loss": 2.9442927837371826, + "objective/train/theoretical_loss": 3.667032015471743, + "objective/train/tokens_used": 970732000, + "objective/train/value_avg": -0.004253387451171875, + "objective/train/value_loss": 6.39367790427059e-05, + "objective/train/value_max": -0.00034880638122558594, + "objective/train/value_min": -0.046722412109375, + "objective/train/value_reward_corr": 0.2431635143561949, + "objective/train/value_std": 0.00301361083984375, + "objective/train/weight_avg": 0.9974822402000427, + "objective/train/weighted_lm_loss": 2.935971736907959, + "objective/train/weights_max": 1.3676724433898926, + "objective/train/weights_min": 0.32014790177345276, + "theoretical_loss": 3.667032015471743, + "tokens_seen": 950272000 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035965897693079234, + "loss": 1.4728, + "theoretical_loss": 3.667032015471743, + "tokens_seen": 950272000 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003596489468405216, + "loss": 1.5141, + "theoretical_loss": 3.6670077215212302, + "tokens_seen": 950337536 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035963891675025076, + "loss": 1.4099, + "theoretical_loss": 3.666983429715045, + "tokens_seen": 950403072 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035962888665997994, + "loss": 1.4693, + "theoretical_loss": 3.6669591400528487, + "tokens_seen": 950468608 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003596188565697091, + "loss": 1.3075, + "theoretical_loss": 3.6669348525343057, + "tokens_seen": 950534144 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003596088264794383, + "loss": 1.4038, + "theoretical_loss": 3.6669105671590785, + "tokens_seen": 950599680 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035959879638916754, + "loss": 1.4257, + "theoretical_loss": 3.6668862839268304, + "tokens_seen": 950665216 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003595887662988967, + "loss": 1.474, + "theoretical_loss": 3.666862002837224, + "tokens_seen": 950730752 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003595787362086259, + "loss": 1.3709, + "theoretical_loss": 3.666837723889923, + "tokens_seen": 950796288 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003595687061183551, + "loss": 1.3897, + "theoretical_loss": 3.66681344708459, + "tokens_seen": 950861824 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035955867602808426, + "loss": 1.3949, + "theoretical_loss": 3.66678917242089, + "tokens_seen": 950927360 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035954864593781344, + "loss": 1.425, + "theoretical_loss": 3.6667648998984856, + "tokens_seen": 950992896 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003595386158475427, + "loss": 1.481, + "theoretical_loss": 3.66674062951704, + "tokens_seen": 951058432 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003595285857572718, + "loss": 1.3076, + "theoretical_loss": 3.6667163612762175, + "tokens_seen": 951123968 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035951855566700104, + "loss": 1.3885, + "theoretical_loss": 3.666692095175682, + "tokens_seen": 951189504 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035950852557673017, + "loss": 1.3679, + "theoretical_loss": 3.6666678312150958, + "tokens_seen": 951255040 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003594984954864594, + "loss": 1.3131, + "theoretical_loss": 3.6666435693941244, + "tokens_seen": 951320576 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003594884653961886, + "loss": 1.473, + "theoretical_loss": 3.6666193097124307, + "tokens_seen": 951386112 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035947843530591777, + "loss": 1.3674, + "theoretical_loss": 3.6665950521696793, + "tokens_seen": 951451648 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035946840521564695, + "loss": 1.3763, + "theoretical_loss": 3.6665707967655345, + "tokens_seen": 951517184 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003594583751253762, + "loss": 1.3655, + "theoretical_loss": 3.66654654349966, + "tokens_seen": 951582720 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003594483450351053, + "loss": 1.5071, + "theoretical_loss": 3.6665222923717202, + "tokens_seen": 951648256 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035943831494483454, + "loss": 1.299, + "theoretical_loss": 3.66649804338138, + "tokens_seen": 951713792 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035942828485456367, + "loss": 1.434, + "theoretical_loss": 3.6664737965283027, + "tokens_seen": 951779328 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003594182547642929, + "loss": 1.4069, + "theoretical_loss": 3.6664495518121534, + "tokens_seen": 951844864 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.0005722944624722004, + "objective/train/docs_used": 546639, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0417282581329346, + "objective/train/original_loss": 3.0417280197143555, + "objective/train/theoretical_loss": 3.6664253092325962, + "objective/train/tokens_used": 972370400, + "objective/train/value_avg": -0.0052642822265625, + "objective/train/value_loss": 7.844670471968129e-05, + "objective/train/value_max": -0.00046563148498535156, + "objective/train/value_min": -0.051666259765625, + "objective/train/value_reward_corr": 0.357379191267588, + "objective/train/value_std": 0.00411224365234375, + "objective/train/weight_avg": 0.9978433847427368, + "objective/train/weighted_lm_loss": 3.027359962463379, + "objective/train/weights_max": 1.4542386531829834, + "objective/train/weights_min": 0.34243711829185486, + "theoretical_loss": 3.6664253092325962, + "tokens_seen": 951910400 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003594082246740221, + "loss": 1.2719, + "theoretical_loss": 3.6664253092325962, + "tokens_seen": 951910400 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035939819458375127, + "loss": 1.5044, + "theoretical_loss": 3.6664010687892965, + "tokens_seen": 951975936 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035938816449348045, + "loss": 1.4045, + "theoretical_loss": 3.666376830481919, + "tokens_seen": 952041472 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035937813440320963, + "loss": 1.2756, + "theoretical_loss": 3.666352594310127, + "tokens_seen": 952107008 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003593681043129388, + "loss": 1.3325, + "theoretical_loss": 3.666328360273588, + "tokens_seen": 952172544 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035935807422266805, + "loss": 1.4479, + "theoretical_loss": 3.6663041283719644, + "tokens_seen": 952238080 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003593480441323972, + "loss": 1.4357, + "theoretical_loss": 3.6662798986049223, + "tokens_seen": 952303616 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003593380140421264, + "loss": 1.4107, + "theoretical_loss": 3.666255670972127, + "tokens_seen": 952369152 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035932798395185554, + "loss": 1.2805, + "theoretical_loss": 3.666231445473243, + "tokens_seen": 952434688 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035931795386158477, + "loss": 1.4089, + "theoretical_loss": 3.666207222107936, + "tokens_seen": 952500224 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035930792377131395, + "loss": 1.3209, + "theoretical_loss": 3.666183000875871, + "tokens_seen": 952565760 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035929789368104313, + "loss": 1.3889, + "theoretical_loss": 3.666158781776714, + "tokens_seen": 952631296 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003592878635907723, + "loss": 1.425, + "theoretical_loss": 3.66613456481013, + "tokens_seen": 952696832 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035927783350050155, + "loss": 1.39, + "theoretical_loss": 3.6661103499757846, + "tokens_seen": 952762368 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003592678034102307, + "loss": 1.3542, + "theoretical_loss": 3.6660861372733438, + "tokens_seen": 952827904 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003592577733199599, + "loss": 1.3649, + "theoretical_loss": 3.666061926702472, + "tokens_seen": 952893440 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035924774322968904, + "loss": 1.4519, + "theoretical_loss": 3.6660377182628365, + "tokens_seen": 952958976 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003592377131394183, + "loss": 1.2274, + "theoretical_loss": 3.6660135119541026, + "tokens_seen": 953024512 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035922768304914746, + "loss": 1.503, + "theoretical_loss": 3.6659893077759356, + "tokens_seen": 953090048 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035921765295887664, + "loss": 1.4535, + "theoretical_loss": 3.6659651057280023, + "tokens_seen": 953155584 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003592076228686058, + "loss": 1.4144, + "theoretical_loss": 3.665940905809969, + "tokens_seen": 953221120 + }, + { + "epoch": 0.29, + "learning_rate": 0.000359197592778335, + "loss": 1.408, + "theoretical_loss": 3.6659167080215003, + "tokens_seen": 953286656 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003591875626880642, + "loss": 1.4202, + "theoretical_loss": 3.665892512362264, + "tokens_seen": 953352192 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003591775325977934, + "loss": 1.4106, + "theoretical_loss": 3.6658683188319263, + "tokens_seen": 953417728 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035916750250752254, + "loss": 1.3517, + "theoretical_loss": 3.6658441274301525, + "tokens_seen": 953483264 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.0010829208185896277, + "objective/train/docs_used": 547886, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.114253044128418, + "objective/train/original_loss": 3.114253520965576, + "objective/train/theoretical_loss": 3.66581993815661, + "objective/train/tokens_used": 974008800, + "objective/train/value_avg": -0.00490570068359375, + "objective/train/value_loss": 7.639366958756e-05, + "objective/train/value_max": -0.0005054473876953125, + "objective/train/value_min": -0.05291748046875, + "objective/train/value_reward_corr": 0.40984625000095004, + "objective/train/value_std": 0.0036640167236328125, + "objective/train/weight_avg": 0.9924160242080688, + "objective/train/weighted_lm_loss": 3.089439868927002, + "objective/train/weights_max": 1.463249921798706, + "objective/train/weights_min": 0.08366327732801437, + "theoretical_loss": 3.66581993815661, + "tokens_seen": 953548800 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003591574724172518, + "loss": 1.4338, + "theoretical_loss": 3.66581993815661, + "tokens_seen": 953548800 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035914744232698096, + "loss": 1.2356, + "theoretical_loss": 3.6657957510109647, + "tokens_seen": 953614336 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035913741223671014, + "loss": 1.4813, + "theoretical_loss": 3.6657715659928836, + "tokens_seen": 953679872 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003591273821464393, + "loss": 1.352, + "theoretical_loss": 3.6657473831020333, + "tokens_seen": 953745408 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003591173520561685, + "loss": 1.4656, + "theoretical_loss": 3.6657232023380804, + "tokens_seen": 953810944 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003591073219658977, + "loss": 1.3322, + "theoretical_loss": 3.665699023700692, + "tokens_seen": 953876480 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003590972918756269, + "loss": 1.2809, + "theoretical_loss": 3.6656748471895346, + "tokens_seen": 953942016 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035908726178535605, + "loss": 1.4179, + "theoretical_loss": 3.6656506728042753, + "tokens_seen": 954007552 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003590772316950853, + "loss": 1.4453, + "theoretical_loss": 3.6656265005445814, + "tokens_seen": 954073088 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003590672016048144, + "loss": 1.4799, + "theoretical_loss": 3.6656023304101195, + "tokens_seen": 954138624 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035905717151454364, + "loss": 1.4986, + "theoretical_loss": 3.665578162400558, + "tokens_seen": 954204160 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003590471414242728, + "loss": 1.3743, + "theoretical_loss": 3.665553996515562, + "tokens_seen": 954269696 + }, + { + "epoch": 0.29, + "learning_rate": 0.000359037111334002, + "loss": 1.4, + "theoretical_loss": 3.6655298327548014, + "tokens_seen": 954335232 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003590270812437312, + "loss": 1.4126, + "theoretical_loss": 3.6655056711179412, + "tokens_seen": 954400768 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035901705115346037, + "loss": 1.4693, + "theoretical_loss": 3.665481511604651, + "tokens_seen": 954466304 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035900702106318955, + "loss": 1.345, + "theoretical_loss": 3.665457354214597, + "tokens_seen": 954531840 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003589969909729188, + "loss": 1.4967, + "theoretical_loss": 3.665433198947447, + "tokens_seen": 954597376 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003589869608826479, + "loss": 1.3184, + "theoretical_loss": 3.6654090458028694, + "tokens_seen": 954662912 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035897693079237715, + "loss": 1.3524, + "theoretical_loss": 3.665384894780532, + "tokens_seen": 954728448 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035896690070210633, + "loss": 1.3769, + "theoretical_loss": 3.6653607458801014, + "tokens_seen": 954793984 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003589568706118355, + "loss": 1.3082, + "theoretical_loss": 3.6653365991012468, + "tokens_seen": 954859520 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003589468405215647, + "loss": 1.4328, + "theoretical_loss": 3.6653124544436357, + "tokens_seen": 954925056 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035893681043129387, + "loss": 1.4021, + "theoretical_loss": 3.665288311906936, + "tokens_seen": 954990592 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035892678034102305, + "loss": 1.3871, + "theoretical_loss": 3.6652641714908167, + "tokens_seen": 955056128 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003589167502507523, + "loss": 1.3701, + "theoretical_loss": 3.6652400331949453, + "tokens_seen": 955121664 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": 0.0011815432226285338, + "objective/train/docs_used": 548540, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5849084854125977, + "objective/train/original_loss": 2.5849084854125977, + "objective/train/theoretical_loss": 3.66521589701899, + "objective/train/tokens_used": 975647200, + "objective/train/value_avg": -0.004650115966796875, + "objective/train/value_loss": 4.293138408684172e-05, + "objective/train/value_max": -0.00019562244415283203, + "objective/train/value_min": -0.0894775390625, + "objective/train/value_reward_corr": 0.30880938218288406, + "objective/train/value_std": 0.003803253173828125, + "objective/train/weight_avg": 1.0138909816741943, + "objective/train/weighted_lm_loss": 2.626004695892334, + "objective/train/weights_max": 2.4467861652374268, + "objective/train/weights_min": 0.45854413509368896, + "theoretical_loss": 3.66521589701899, + "tokens_seen": 955187200 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003589067201604814, + "loss": 1.3621, + "theoretical_loss": 3.66521589701899, + "tokens_seen": 955187200 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035889669007021065, + "loss": 1.3507, + "theoretical_loss": 3.66519176296262, + "tokens_seen": 955252736 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003588866599799398, + "loss": 1.4095, + "theoretical_loss": 3.6651676310255032, + "tokens_seen": 955318272 + }, + { + "epoch": 0.29, + "learning_rate": 0.000358876629889669, + "loss": 1.4501, + "theoretical_loss": 3.665143501207308, + "tokens_seen": 955383808 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003588665997993982, + "loss": 1.3193, + "theoretical_loss": 3.6651193735077032, + "tokens_seen": 955449344 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003588565697091274, + "loss": 1.3067, + "theoretical_loss": 3.6650952479263577, + "tokens_seen": 955514880 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003588465396188566, + "loss": 1.3807, + "theoretical_loss": 3.6650711244629406, + "tokens_seen": 955580416 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035883650952858574, + "loss": 1.3623, + "theoretical_loss": 3.6650470031171194, + "tokens_seen": 955645952 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035882647943831497, + "loss": 1.3772, + "theoretical_loss": 3.665022883888564, + "tokens_seen": 955711488 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035881644934804415, + "loss": 1.3196, + "theoretical_loss": 3.6649987667769435, + "tokens_seen": 955777024 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035880641925777333, + "loss": 1.2743, + "theoretical_loss": 3.6649746517819266, + "tokens_seen": 955842560 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003587963891675025, + "loss": 1.4799, + "theoretical_loss": 3.6649505389031827, + "tokens_seen": 955908096 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035878635907723175, + "loss": 1.356, + "theoretical_loss": 3.6649264281403804, + "tokens_seen": 955973632 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003587763289869609, + "loss": 1.3276, + "theoretical_loss": 3.66490231949319, + "tokens_seen": 956039168 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003587662988966901, + "loss": 1.4147, + "theoretical_loss": 3.6648782129612796, + "tokens_seen": 956104704 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035875626880641924, + "loss": 1.4182, + "theoretical_loss": 3.66485410854432, + "tokens_seen": 956170240 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003587462387161485, + "loss": 1.391, + "theoretical_loss": 3.6648300062419796, + "tokens_seen": 956235776 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035873620862587766, + "loss": 1.4004, + "theoretical_loss": 3.664805906053928, + "tokens_seen": 956301312 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035872617853560684, + "loss": 1.4412, + "theoretical_loss": 3.6647818079798364, + "tokens_seen": 956366848 + }, + { + "epoch": 0.29, + "learning_rate": 0.000358716148445336, + "loss": 1.3713, + "theoretical_loss": 3.6647577120193726, + "tokens_seen": 956432384 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003587061183550652, + "loss": 1.4704, + "theoretical_loss": 3.6647336181722077, + "tokens_seen": 956497920 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003586960882647944, + "loss": 1.482, + "theoretical_loss": 3.6647095264380107, + "tokens_seen": 956563456 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003586860581745236, + "loss": 1.2684, + "theoretical_loss": 3.664685436816452, + "tokens_seen": 956628992 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035867602808425274, + "loss": 1.425, + "theoretical_loss": 3.6646613493072016, + "tokens_seen": 956694528 + }, + { + "epoch": 0.29, + "learning_rate": 0.000358665997993982, + "loss": 1.4572, + "theoretical_loss": 3.6646372639099294, + "tokens_seen": 956760064 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": 0.0011263970518484712, + "objective/train/docs_used": 549880, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6034812927246094, + "objective/train/original_loss": 2.6034812927246094, + "objective/train/theoretical_loss": 3.664613180624306, + "objective/train/tokens_used": 977285600, + "objective/train/value_avg": -0.004878997802734375, + "objective/train/value_loss": 4.0346189052797854e-05, + "objective/train/value_max": -0.00036406517028808594, + "objective/train/value_min": -0.061767578125, + "objective/train/value_reward_corr": 0.3995976585438842, + "objective/train/value_std": 0.004058837890625, + "objective/train/weight_avg": 1.0132006406784058, + "objective/train/weighted_lm_loss": 2.6362102031707764, + "objective/train/weights_max": 1.5098458528518677, + "objective/train/weights_min": 0.294359415769577, + "theoretical_loss": 3.664613180624306, + "tokens_seen": 956825600 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035865596790371116, + "loss": 1.4733, + "theoretical_loss": 3.664613180624306, + "tokens_seen": 956825600 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035864593781344034, + "loss": 1.3257, + "theoretical_loss": 3.6645890994500014, + "tokens_seen": 956891136 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003586359077231695, + "loss": 1.3613, + "theoretical_loss": 3.6645650203866857, + "tokens_seen": 956956672 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003586258776328987, + "loss": 1.3524, + "theoretical_loss": 3.6645409434340293, + "tokens_seen": 957022208 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003586158475426279, + "loss": 1.3241, + "theoretical_loss": 3.6645168685917033, + "tokens_seen": 957087744 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003586058174523571, + "loss": 1.3346, + "theoretical_loss": 3.6644927958593776, + "tokens_seen": 957153280 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035859578736208625, + "loss": 1.3287, + "theoretical_loss": 3.6644687252367234, + "tokens_seen": 957218816 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003585857572718155, + "loss": 1.3569, + "theoretical_loss": 3.6644446567234112, + "tokens_seen": 957284352 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003585757271815446, + "loss": 1.4648, + "theoretical_loss": 3.6644205903191107, + "tokens_seen": 957349888 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035856569709127384, + "loss": 1.3471, + "theoretical_loss": 3.6643965260234945, + "tokens_seen": 957415424 + }, + { + "epoch": 0.29, + "learning_rate": 0.000358555667001003, + "loss": 1.4285, + "theoretical_loss": 3.6643724638362327, + "tokens_seen": 957480960 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003585456369107322, + "loss": 1.3937, + "theoretical_loss": 3.6643484037569958, + "tokens_seen": 957546496 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003585356068204614, + "loss": 1.4594, + "theoretical_loss": 3.6643243457854555, + "tokens_seen": 957612032 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035852557673019057, + "loss": 1.4239, + "theoretical_loss": 3.6643002899212824, + "tokens_seen": 957677568 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035851554663991975, + "loss": 1.387, + "theoretical_loss": 3.6642762361641488, + "tokens_seen": 957743104 + }, + { + "epoch": 0.29, + "learning_rate": 0.000358505516549649, + "loss": 1.307, + "theoretical_loss": 3.6642521845137246, + "tokens_seen": 957808640 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003584954864593781, + "loss": 1.2955, + "theoretical_loss": 3.664228134969682, + "tokens_seen": 957874176 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035848545636910735, + "loss": 1.3589, + "theoretical_loss": 3.664204087531693, + "tokens_seen": 957939712 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035847542627883653, + "loss": 1.3728, + "theoretical_loss": 3.664180042199427, + "tokens_seen": 958005248 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003584653961885657, + "loss": 1.4301, + "theoretical_loss": 3.664155998972558, + "tokens_seen": 958070784 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003584553660982949, + "loss": 1.3619, + "theoretical_loss": 3.6641319578507563, + "tokens_seen": 958136320 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035844533600802407, + "loss": 1.4131, + "theoretical_loss": 3.664107918833694, + "tokens_seen": 958201856 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035843530591775325, + "loss": 1.4439, + "theoretical_loss": 3.6640838819210426, + "tokens_seen": 958267392 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003584252758274825, + "loss": 1.2797, + "theoretical_loss": 3.6640598471124743, + "tokens_seen": 958332928 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003584152457372116, + "loss": 1.3739, + "theoretical_loss": 3.664035814407661, + "tokens_seen": 958398464 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.00147747911978513, + "objective/train/docs_used": 550706, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4884982109069824, + "objective/train/original_loss": 2.4884979724884033, + "objective/train/theoretical_loss": 3.6640117838062745, + "objective/train/tokens_used": 978924000, + "objective/train/value_avg": -0.00531005859375, + "objective/train/value_loss": 6.473714165622368e-05, + "objective/train/value_max": -0.0004012584686279297, + "objective/train/value_min": -0.10125732421875, + "objective/train/value_reward_corr": 0.39517412050525863, + "objective/train/value_std": 0.0039520263671875, + "objective/train/weight_avg": 0.9882991313934326, + "objective/train/weighted_lm_loss": 2.4656264781951904, + "objective/train/weights_max": 1.8146753311157227, + "objective/train/weights_min": 0.529013991355896, + "theoretical_loss": 3.6640117838062745, + "tokens_seen": 958464000 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035840521564694085, + "loss": 1.3402, + "theoretical_loss": 3.6640117838062745, + "tokens_seen": 958464000 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035839518555667, + "loss": 1.407, + "theoretical_loss": 3.6639877553079874, + "tokens_seen": 958529536 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003583851554663992, + "loss": 1.3861, + "theoretical_loss": 3.6639637289124716, + "tokens_seen": 958595072 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003583751253761284, + "loss": 1.3745, + "theoretical_loss": 3.6639397046193984, + "tokens_seen": 958660608 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003583650952858576, + "loss": 1.499, + "theoretical_loss": 3.663915682428442, + "tokens_seen": 958726144 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035835506519558676, + "loss": 1.3535, + "theoretical_loss": 3.663891662339273, + "tokens_seen": 958791680 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035834503510531594, + "loss": 1.3644, + "theoretical_loss": 3.663867644351565, + "tokens_seen": 958857216 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003583350050150451, + "loss": 1.3992, + "theoretical_loss": 3.6638436284649902, + "tokens_seen": 958922752 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035832497492477435, + "loss": 1.3172, + "theoretical_loss": 3.6638196146792215, + "tokens_seen": 958988288 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003583149448345035, + "loss": 1.5983, + "theoretical_loss": 3.6637956029939307, + "tokens_seen": 959053824 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003583049147442327, + "loss": 1.4188, + "theoretical_loss": 3.6637715934087915, + "tokens_seen": 959119360 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003582948846539619, + "loss": 1.3402, + "theoretical_loss": 3.6637475859234763, + "tokens_seen": 959184896 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003582848545636911, + "loss": 1.3583, + "theoretical_loss": 3.663723580537658, + "tokens_seen": 959250432 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035827482447342026, + "loss": 1.3781, + "theoretical_loss": 3.66369957725101, + "tokens_seen": 959315968 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035826479438314944, + "loss": 1.2845, + "theoretical_loss": 3.6636755760632043, + "tokens_seen": 959381504 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003582547642928786, + "loss": 1.3658, + "theoretical_loss": 3.663651576973915, + "tokens_seen": 959447040 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035824473420260786, + "loss": 1.4013, + "theoretical_loss": 3.6636275799828155, + "tokens_seen": 959512576 + }, + { + "epoch": 0.29, + "learning_rate": 0.000358234704112337, + "loss": 1.3779, + "theoretical_loss": 3.663603585089578, + "tokens_seen": 959578112 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003582246740220662, + "loss": 1.3375, + "theoretical_loss": 3.6635795922938765, + "tokens_seen": 959643648 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035821464393179535, + "loss": 1.2987, + "theoretical_loss": 3.663555601595384, + "tokens_seen": 959709184 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003582046138415246, + "loss": 1.3501, + "theoretical_loss": 3.6635316129937747, + "tokens_seen": 959774720 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035819458375125376, + "loss": 1.3274, + "theoretical_loss": 3.6635076264887223, + "tokens_seen": 959840256 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035818455366098294, + "loss": 1.4436, + "theoretical_loss": 3.663483642079899, + "tokens_seen": 959905792 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003581745235707121, + "loss": 1.347, + "theoretical_loss": 3.66345965976698, + "tokens_seen": 959971328 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035816449348044136, + "loss": 1.4767, + "theoretical_loss": 3.6634356795496386, + "tokens_seen": 960036864 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.0008587418124079704, + "objective/train/docs_used": 551132, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.275871753692627, + "objective/train/original_loss": 2.2758712768554688, + "objective/train/theoretical_loss": 3.663411701427548, + "objective/train/tokens_used": 980562400, + "objective/train/value_avg": -0.00516510009765625, + "objective/train/value_loss": 5.478889579535462e-05, + "objective/train/value_max": -0.00034880638122558594, + "objective/train/value_min": -0.06768798828125, + "objective/train/value_reward_corr": 0.5110958982646808, + "objective/train/value_std": 0.00485992431640625, + "objective/train/weight_avg": 0.9939701557159424, + "objective/train/weighted_lm_loss": 2.2665858268737793, + "objective/train/weights_max": 1.6416807174682617, + "objective/train/weights_min": 0.3666645884513855, + "theoretical_loss": 3.663411701427548, + "tokens_seen": 960102400 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003581544633901705, + "loss": 1.3512, + "theoretical_loss": 3.663411701427548, + "tokens_seen": 960102400 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003581444332998997, + "loss": 1.4622, + "theoretical_loss": 3.6633877254003835, + "tokens_seen": 960167936 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035813440320962885, + "loss": 1.4627, + "theoretical_loss": 3.6633637514678177, + "tokens_seen": 960233472 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003581243731193581, + "loss": 1.4515, + "theoretical_loss": 3.663339779629525, + "tokens_seen": 960299008 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035811434302908727, + "loss": 1.4187, + "theoretical_loss": 3.6633158098851806, + "tokens_seen": 960364544 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035810431293881645, + "loss": 1.3764, + "theoretical_loss": 3.6632918422344574, + "tokens_seen": 960430080 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003580942828485457, + "loss": 1.3263, + "theoretical_loss": 3.66326787667703, + "tokens_seen": 960495616 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003580842527582748, + "loss": 1.3565, + "theoretical_loss": 3.6632439132125736, + "tokens_seen": 960561152 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035807422266800404, + "loss": 1.3031, + "theoretical_loss": 3.6632199518407615, + "tokens_seen": 960626688 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003580641925777332, + "loss": 1.4232, + "theoretical_loss": 3.6631959925612696, + "tokens_seen": 960692224 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003580541624874624, + "loss": 1.3947, + "theoretical_loss": 3.663172035373771, + "tokens_seen": 960757760 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003580441323971916, + "loss": 1.3352, + "theoretical_loss": 3.663148080277941, + "tokens_seen": 960823296 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035803410230692077, + "loss": 1.3475, + "theoretical_loss": 3.663124127273454, + "tokens_seen": 960888832 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035802407221664995, + "loss": 1.3712, + "theoretical_loss": 3.6631001763599853, + "tokens_seen": 960954368 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003580140421263792, + "loss": 1.4203, + "theoretical_loss": 3.6630762275372097, + "tokens_seen": 961019904 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003580040120361083, + "loss": 1.353, + "theoretical_loss": 3.663052280804802, + "tokens_seen": 961085440 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035799398194583755, + "loss": 1.4163, + "theoretical_loss": 3.663028336162437, + "tokens_seen": 961150976 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035798395185556673, + "loss": 1.2801, + "theoretical_loss": 3.663004393609791, + "tokens_seen": 961216512 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003579739217652959, + "loss": 1.3383, + "theoretical_loss": 3.6629804531465373, + "tokens_seen": 961282048 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003579638916750251, + "loss": 1.3493, + "theoretical_loss": 3.662956514772352, + "tokens_seen": 961347584 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035795386158475427, + "loss": 1.2793, + "theoretical_loss": 3.6629325784869113, + "tokens_seen": 961413120 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035794383149448345, + "loss": 1.2748, + "theoretical_loss": 3.662908644289889, + "tokens_seen": 961478656 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003579338014042127, + "loss": 1.3517, + "theoretical_loss": 3.6628847121809613, + "tokens_seen": 961544192 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003579237713139418, + "loss": 1.358, + "theoretical_loss": 3.662860782159804, + "tokens_seen": 961609728 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035791374122367105, + "loss": 1.375, + "theoretical_loss": 3.662836854226092, + "tokens_seen": 961675264 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": 0.0003039479779545218, + "objective/train/docs_used": 552317, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5133979320526123, + "objective/train/original_loss": 2.513397693634033, + "objective/train/theoretical_loss": 3.662812928379502, + "objective/train/tokens_used": 982200800, + "objective/train/value_avg": -0.004547119140625, + "objective/train/value_loss": 4.519113645073958e-05, + "objective/train/value_max": -0.00019562244415283203, + "objective/train/value_min": -0.11456298828125, + "objective/train/value_reward_corr": 0.3692602363333201, + "objective/train/value_std": 0.00392913818359375, + "objective/train/weight_avg": 1.0052202939987183, + "objective/train/weighted_lm_loss": 2.5377063751220703, + "objective/train/weights_max": 2.3025453090667725, + "objective/train/weights_min": 0.4618092477321625, + "theoretical_loss": 3.662812928379502, + "tokens_seen": 961740800 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003579037111334002, + "loss": 1.3596, + "theoretical_loss": 3.662812928379502, + "tokens_seen": 961740800 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003578936810431294, + "loss": 1.4134, + "theoretical_loss": 3.662789004619709, + "tokens_seen": 961806336 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003578836509528586, + "loss": 1.493, + "theoretical_loss": 3.6627650829463887, + "tokens_seen": 961871872 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003578736208625878, + "loss": 1.4026, + "theoretical_loss": 3.6627411633592173, + "tokens_seen": 961937408 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035786359077231696, + "loss": 1.4987, + "theoretical_loss": 3.662717245857871, + "tokens_seen": 962002944 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035785356068204614, + "loss": 1.3496, + "theoretical_loss": 3.6626933304420257, + "tokens_seen": 962068480 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003578435305917753, + "loss": 1.3714, + "theoretical_loss": 3.6626694171113567, + "tokens_seen": 962134016 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035783350050150455, + "loss": 1.4447, + "theoretical_loss": 3.662645505865541, + "tokens_seen": 962199552 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003578234704112337, + "loss": 1.3825, + "theoretical_loss": 3.662621596704255, + "tokens_seen": 962265088 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003578134403209629, + "loss": 1.4888, + "theoretical_loss": 3.662597689627175, + "tokens_seen": 962330624 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003578034102306921, + "loss": 1.4961, + "theoretical_loss": 3.6625737846339765, + "tokens_seen": 962396160 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003577933801404213, + "loss": 1.4345, + "theoretical_loss": 3.6625498817243365, + "tokens_seen": 962461696 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035778335005015046, + "loss": 1.4786, + "theoretical_loss": 3.6625259808979322, + "tokens_seen": 962527232 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035777331995987964, + "loss": 1.3407, + "theoretical_loss": 3.662502082154439, + "tokens_seen": 962592768 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003577632898696088, + "loss": 1.4572, + "theoretical_loss": 3.6624781854935344, + "tokens_seen": 962658304 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035775325977933806, + "loss": 1.466, + "theoretical_loss": 3.6624542909148956, + "tokens_seen": 962723840 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003577432296890672, + "loss": 1.334, + "theoretical_loss": 3.6624303984181985, + "tokens_seen": 962789376 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003577331995987964, + "loss": 1.3533, + "theoretical_loss": 3.6624065080031203, + "tokens_seen": 962854912 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035772316950852555, + "loss": 1.4828, + "theoretical_loss": 3.662382619669338, + "tokens_seen": 962920448 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003577131394182548, + "loss": 1.4459, + "theoretical_loss": 3.6623587334165286, + "tokens_seen": 962985984 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035770310932798396, + "loss": 1.3164, + "theoretical_loss": 3.6623348492443686, + "tokens_seen": 963051520 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035769307923771314, + "loss": 1.34, + "theoretical_loss": 3.6623109671525365, + "tokens_seen": 963117056 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003576830491474423, + "loss": 1.4177, + "theoretical_loss": 3.662287087140709, + "tokens_seen": 963182592 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035767301905717156, + "loss": 1.4238, + "theoretical_loss": 3.6622632092085636, + "tokens_seen": 963248128 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003576629889669007, + "loss": 1.4276, + "theoretical_loss": 3.662239333355777, + "tokens_seen": 963313664 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": 0.0010390268871560693, + "objective/train/docs_used": 552873, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6848971843719482, + "objective/train/original_loss": 2.6848976612091064, + "objective/train/theoretical_loss": 3.662215459582027, + "objective/train/tokens_used": 983839200, + "objective/train/value_avg": -0.00508880615234375, + "objective/train/value_loss": 4.780441304319538e-05, + "objective/train/value_max": -0.0004673004150390625, + "objective/train/value_min": -0.048583984375, + "objective/train/value_reward_corr": 0.31467018569701527, + "objective/train/value_std": 0.004215240478515625, + "objective/train/weight_avg": 1.0125640630722046, + "objective/train/weighted_lm_loss": 2.719092845916748, + "objective/train/weights_max": 1.5010861158370972, + "objective/train/weights_min": 0.25009283423423767, + "theoretical_loss": 3.662215459582027, + "tokens_seen": 963379200 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003576529588766299, + "loss": 1.4109, + "theoretical_loss": 3.662215459582027, + "tokens_seen": 963379200 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035764292878635905, + "loss": 1.3675, + "theoretical_loss": 3.6621915878869915, + "tokens_seen": 963444736 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003576328986960883, + "loss": 1.2634, + "theoretical_loss": 3.662167718270348, + "tokens_seen": 963510272 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035762286860581747, + "loss": 1.3108, + "theoretical_loss": 3.6621438507317743, + "tokens_seen": 963575808 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035761283851554665, + "loss": 1.3558, + "theoretical_loss": 3.662119985270947, + "tokens_seen": 963641344 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035760280842527583, + "loss": 1.3736, + "theoretical_loss": 3.6620961218875463, + "tokens_seen": 963706880 + }, + { + "epoch": 0.29, + "learning_rate": 0.000357592778335005, + "loss": 1.441, + "theoretical_loss": 3.662072260581248, + "tokens_seen": 963772416 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003575827482447342, + "loss": 1.3925, + "theoretical_loss": 3.6620484013517305, + "tokens_seen": 963837952 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003575727181544634, + "loss": 1.2557, + "theoretical_loss": 3.6620245441986725, + "tokens_seen": 963903488 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035756268806419255, + "loss": 1.4521, + "theoretical_loss": 3.6620006891217516, + "tokens_seen": 963969024 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003575526579739218, + "loss": 1.4125, + "theoretical_loss": 3.6619768361206466, + "tokens_seen": 964034560 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003575426278836509, + "loss": 1.4367, + "theoretical_loss": 3.661952985195035, + "tokens_seen": 964100096 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035753259779338015, + "loss": 1.3006, + "theoretical_loss": 3.661929136344596, + "tokens_seen": 964165632 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035752256770310933, + "loss": 1.2992, + "theoretical_loss": 3.661905289569007, + "tokens_seen": 964231168 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003575125376128385, + "loss": 1.4, + "theoretical_loss": 3.661881444867947, + "tokens_seen": 964296704 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003575025075225677, + "loss": 1.3544, + "theoretical_loss": 3.6618576022410947, + "tokens_seen": 964362240 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035749247743229693, + "loss": 1.2843, + "theoretical_loss": 3.6618337616881282, + "tokens_seen": 964427776 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035748244734202606, + "loss": 1.3905, + "theoretical_loss": 3.661809923208727, + "tokens_seen": 964493312 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003574724172517553, + "loss": 1.3993, + "theoretical_loss": 3.66178608680257, + "tokens_seen": 964558848 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003574623871614844, + "loss": 1.3145, + "theoretical_loss": 3.6617622524693347, + "tokens_seen": 964624384 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035745235707121365, + "loss": 1.5091, + "theoretical_loss": 3.6617384202087004, + "tokens_seen": 964689920 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035744232698094283, + "loss": 1.4098, + "theoretical_loss": 3.661714590020347, + "tokens_seen": 964755456 + }, + { + "epoch": 0.29, + "learning_rate": 0.000357432296890672, + "loss": 1.4097, + "theoretical_loss": 3.6616907619039534, + "tokens_seen": 964820992 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003574222668004012, + "loss": 1.4892, + "theoretical_loss": 3.6616669358591984, + "tokens_seen": 964886528 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003574122367101304, + "loss": 1.4784, + "theoretical_loss": 3.661643111885761, + "tokens_seen": 964952064 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.00014124797598924488, + "objective/train/docs_used": 554313, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3458240032196045, + "objective/train/original_loss": 2.3458237648010254, + "objective/train/theoretical_loss": 3.66161928998332, + "objective/train/tokens_used": 985477600, + "objective/train/value_avg": -0.0047760009765625, + "objective/train/value_loss": 3.768566966755316e-05, + "objective/train/value_max": -0.0003859996795654297, + "objective/train/value_min": -0.05865478515625, + "objective/train/value_reward_corr": 0.37499864454097775, + "objective/train/value_std": 0.004093170166015625, + "objective/train/weight_avg": 1.0004123449325562, + "objective/train/weighted_lm_loss": 2.365036725997925, + "objective/train/weights_max": 1.663681983947754, + "objective/train/weights_min": 0.5001433491706848, + "theoretical_loss": 3.66161928998332, + "tokens_seen": 965017600 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035740220661985956, + "loss": 1.338, + "theoretical_loss": 3.66161928998332, + "tokens_seen": 965017600 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003573921765295888, + "loss": 1.3511, + "theoretical_loss": 3.661595470151556, + "tokens_seen": 965083136 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003573821464393179, + "loss": 1.2907, + "theoretical_loss": 3.6615716523901476, + "tokens_seen": 965148672 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035737211634904716, + "loss": 1.3044, + "theoretical_loss": 3.661547836698775, + "tokens_seen": 965214208 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003573620862587763, + "loss": 1.3466, + "theoretical_loss": 3.6615240230771167, + "tokens_seen": 965279744 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003573520561685055, + "loss": 1.3445, + "theoretical_loss": 3.661500211524853, + "tokens_seen": 965345280 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035734202607823475, + "loss": 1.3528, + "theoretical_loss": 3.6614764020416635, + "tokens_seen": 965410816 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003573319959879639, + "loss": 1.3085, + "theoretical_loss": 3.661452594627228, + "tokens_seen": 965476352 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003573219658976931, + "loss": 1.4322, + "theoretical_loss": 3.6614287892812265, + "tokens_seen": 965541888 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003573119358074223, + "loss": 1.4419, + "theoretical_loss": 3.6614049860033386, + "tokens_seen": 965607424 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003573019057171515, + "loss": 1.2885, + "theoretical_loss": 3.6613811847932443, + "tokens_seen": 965672960 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035729187562688066, + "loss": 1.4437, + "theoretical_loss": 3.6613573856506236, + "tokens_seen": 965738496 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035728184553660984, + "loss": 1.4717, + "theoretical_loss": 3.661333588575157, + "tokens_seen": 965804032 + }, + { + "epoch": 0.29, + "learning_rate": 0.000357271815446339, + "loss": 1.4399, + "theoretical_loss": 3.6613097935665246, + "tokens_seen": 965869568 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035726178535606826, + "loss": 1.3467, + "theoretical_loss": 3.661286000624406, + "tokens_seen": 965935104 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003572517552657974, + "loss": 1.3487, + "theoretical_loss": 3.6612622097484824, + "tokens_seen": 966000640 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003572417251755266, + "loss": 1.3474, + "theoretical_loss": 3.6612384209384343, + "tokens_seen": 966066176 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035723169508525575, + "loss": 1.4473, + "theoretical_loss": 3.661214634193941, + "tokens_seen": 966131712 + }, + { + "epoch": 0.29, + "learning_rate": 0.000357221664994985, + "loss": 1.2986, + "theoretical_loss": 3.6611908495146848, + "tokens_seen": 966197248 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035721163490471416, + "loss": 1.4215, + "theoretical_loss": 3.6611670669003447, + "tokens_seen": 966262784 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035720160481444334, + "loss": 1.2496, + "theoretical_loss": 3.661143286350602, + "tokens_seen": 966328320 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003571915747241725, + "loss": 1.4671, + "theoretical_loss": 3.661119507865138, + "tokens_seen": 966393856 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035718154463390176, + "loss": 1.3848, + "theoretical_loss": 3.661095731443633, + "tokens_seen": 966459392 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003571715145436309, + "loss": 1.2906, + "theoretical_loss": 3.6610719570857677, + "tokens_seen": 966524928 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003571614844533601, + "loss": 1.3973, + "theoretical_loss": 3.6610481847912233, + "tokens_seen": 966590464 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.00025003927294164896, + "objective/train/docs_used": 554946, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.006551742553711, + "objective/train/original_loss": 3.006551504135132, + "objective/train/theoretical_loss": 3.661024414559681, + "objective/train/tokens_used": 987116000, + "objective/train/value_avg": -0.005435943603515625, + "objective/train/value_loss": 5.927092570345849e-05, + "objective/train/value_max": -0.0003814697265625, + "objective/train/value_min": -0.0582275390625, + "objective/train/value_reward_corr": 0.4344280489110461, + "objective/train/value_std": 0.004344940185546875, + "objective/train/weight_avg": 1.0002937316894531, + "objective/train/weighted_lm_loss": 3.0156726837158203, + "objective/train/weights_max": 1.5618356466293335, + "objective/train/weights_min": 0.2550254464149475, + "theoretical_loss": 3.661024414559681, + "tokens_seen": 966656000 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035715145436308925, + "loss": 1.3167, + "theoretical_loss": 3.661024414559681, + "tokens_seen": 966656000 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003571414242728185, + "loss": 1.4022, + "theoretical_loss": 3.6610006463908222, + "tokens_seen": 966721536 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035713139418254767, + "loss": 1.3852, + "theoretical_loss": 3.6609768802843274, + "tokens_seen": 966787072 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035712136409227685, + "loss": 1.3552, + "theoretical_loss": 3.6609531162398783, + "tokens_seen": 966852608 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035711133400200603, + "loss": 1.4005, + "theoretical_loss": 3.6609293542571564, + "tokens_seen": 966918144 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003571013039117352, + "loss": 1.3463, + "theoretical_loss": 3.6609055943358424, + "tokens_seen": 966983680 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003570912738214644, + "loss": 1.4583, + "theoretical_loss": 3.6608818364756184, + "tokens_seen": 967049216 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003570812437311936, + "loss": 1.3918, + "theoretical_loss": 3.6608580806761655, + "tokens_seen": 967114752 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035707121364092275, + "loss": 1.3577, + "theoretical_loss": 3.6608343269371657, + "tokens_seen": 967180288 + }, + { + "epoch": 0.29, + "learning_rate": 0.000357061183550652, + "loss": 1.2636, + "theoretical_loss": 3.6608105752583007, + "tokens_seen": 967245824 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003570511534603811, + "loss": 1.2709, + "theoretical_loss": 3.660786825639252, + "tokens_seen": 967311360 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035704112337011035, + "loss": 1.4837, + "theoretical_loss": 3.6607630780797016, + "tokens_seen": 967376896 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035703109327983953, + "loss": 1.4431, + "theoretical_loss": 3.6607393325793316, + "tokens_seen": 967442432 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003570210631895687, + "loss": 1.2899, + "theoretical_loss": 3.6607155891378236, + "tokens_seen": 967507968 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003570110330992979, + "loss": 1.4577, + "theoretical_loss": 3.6606918477548596, + "tokens_seen": 967573504 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035700100300902713, + "loss": 1.4984, + "theoretical_loss": 3.660668108430122, + "tokens_seen": 967639040 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035699097291875626, + "loss": 1.3108, + "theoretical_loss": 3.6606443711632926, + "tokens_seen": 967704576 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003569809428284855, + "loss": 1.4836, + "theoretical_loss": 3.6606206359540545, + "tokens_seen": 967770112 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003569709127382146, + "loss": 1.5445, + "theoretical_loss": 3.660596902802089, + "tokens_seen": 967835648 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035696088264794385, + "loss": 1.357, + "theoretical_loss": 3.660573171707079, + "tokens_seen": 967901184 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035695085255767303, + "loss": 1.4386, + "theoretical_loss": 3.6605494426687075, + "tokens_seen": 967966720 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003569408224674022, + "loss": 1.2399, + "theoretical_loss": 3.6605257156866555, + "tokens_seen": 968032256 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003569307923771314, + "loss": 1.3423, + "theoretical_loss": 3.6605019907606073, + "tokens_seen": 968097792 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003569207622868606, + "loss": 1.3203, + "theoretical_loss": 3.6604782678902446, + "tokens_seen": 968163328 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035691073219658976, + "loss": 1.3319, + "theoretical_loss": 3.6604545470752505, + "tokens_seen": 968228864 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": 8.703691128175706e-05, + "objective/train/docs_used": 556500, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5161614418029785, + "objective/train/original_loss": 2.5161609649658203, + "objective/train/theoretical_loss": 3.6604308283153073, + "objective/train/tokens_used": 988754400, + "objective/train/value_avg": -0.00438690185546875, + "objective/train/value_loss": 4.214517684886232e-05, + "objective/train/value_max": -0.00038743019104003906, + "objective/train/value_min": -0.083740234375, + "objective/train/value_reward_corr": 0.4361917613620867, + "objective/train/value_std": 0.0038852691650390625, + "objective/train/weight_avg": 1.0028403997421265, + "objective/train/weighted_lm_loss": 2.5205459594726562, + "objective/train/weights_max": 1.595559000968933, + "objective/train/weights_min": 0.2858809232711792, + "theoretical_loss": 3.6604308283153073, + "tokens_seen": 968294400 + }, + { + "epoch": 0.29, + "learning_rate": 0.000356900702106319, + "loss": 1.3658, + "theoretical_loss": 3.6604308283153073, + "tokens_seen": 968294400 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003568906720160481, + "loss": 1.3677, + "theoretical_loss": 3.6604071116100987, + "tokens_seen": 968359936 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035688064192577736, + "loss": 1.4403, + "theoretical_loss": 3.6603833969593067, + "tokens_seen": 968425472 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003568706118355065, + "loss": 1.3398, + "theoretical_loss": 3.6603596843626156, + "tokens_seen": 968491008 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003568605817452357, + "loss": 1.3079, + "theoretical_loss": 3.6603359738197074, + "tokens_seen": 968556544 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003568505516549649, + "loss": 1.4813, + "theoretical_loss": 3.660312265330266, + "tokens_seen": 968622080 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003568405215646941, + "loss": 1.3429, + "theoretical_loss": 3.660288558893974, + "tokens_seen": 968687616 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035683049147442326, + "loss": 1.3736, + "theoretical_loss": 3.660264854510515, + "tokens_seen": 968753152 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003568204613841525, + "loss": 1.3042, + "theoretical_loss": 3.660241152179572, + "tokens_seen": 968818688 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003568104312938816, + "loss": 1.3864, + "theoretical_loss": 3.66021745190083, + "tokens_seen": 968884224 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035680040120361086, + "loss": 1.3354, + "theoretical_loss": 3.6601937536739704, + "tokens_seen": 968949760 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035679037111334, + "loss": 1.3118, + "theoretical_loss": 3.6601700574986777, + "tokens_seen": 969015296 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003567803410230692, + "loss": 1.4476, + "theoretical_loss": 3.6601463633746363, + "tokens_seen": 969080832 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003567703109327984, + "loss": 1.4085, + "theoretical_loss": 3.6601226713015285, + "tokens_seen": 969146368 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003567602808425276, + "loss": 1.3594, + "theoretical_loss": 3.660098981279039, + "tokens_seen": 969211904 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035675025075225677, + "loss": 1.3629, + "theoretical_loss": 3.6600752933068517, + "tokens_seen": 969277440 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035674022066198595, + "loss": 1.4101, + "theoretical_loss": 3.6600516073846503, + "tokens_seen": 969342976 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035673019057171513, + "loss": 1.3053, + "theoretical_loss": 3.6600279235121196, + "tokens_seen": 969408512 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035672016048144436, + "loss": 1.3375, + "theoretical_loss": 3.660004241688942, + "tokens_seen": 969474048 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003567101303911735, + "loss": 1.3459, + "theoretical_loss": 3.6599805619148027, + "tokens_seen": 969539584 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003567001003009027, + "loss": 1.3852, + "theoretical_loss": 3.6599568841893864, + "tokens_seen": 969605120 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035669007021063185, + "loss": 1.2774, + "theoretical_loss": 3.6599332085123764, + "tokens_seen": 969670656 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003566800401203611, + "loss": 1.2961, + "theoretical_loss": 3.6599095348834574, + "tokens_seen": 969736192 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035667001003009027, + "loss": 1.3835, + "theoretical_loss": 3.659885863302314, + "tokens_seen": 969801728 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035665997993981945, + "loss": 1.4646, + "theoretical_loss": 3.659862193768631, + "tokens_seen": 969867264 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": -0.0006214274908415973, + "objective/train/docs_used": 557201, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.851456880569458, + "objective/train/original_loss": 2.851457118988037, + "objective/train/theoretical_loss": 3.6598385262820923, + "objective/train/tokens_used": 990392800, + "objective/train/value_avg": -0.005809783935546875, + "objective/train/value_loss": 9.302605758421123e-05, + "objective/train/value_max": -0.0002453327178955078, + "objective/train/value_min": -0.04351806640625, + "objective/train/value_reward_corr": 0.2662075778051855, + "objective/train/value_std": 0.00431060791015625, + "objective/train/weight_avg": 0.9979054927825928, + "objective/train/weighted_lm_loss": 2.850625514984131, + "objective/train/weights_max": 1.4221640825271606, + "objective/train/weights_min": 0.37836888432502747, + "theoretical_loss": 3.6598385262820923, + "tokens_seen": 969932800 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035664994984954863, + "loss": 1.4468, + "theoretical_loss": 3.6598385262820923, + "tokens_seen": 969932800 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035663991975927787, + "loss": 1.4206, + "theoretical_loss": 3.659814860842383, + "tokens_seen": 969998336 + }, + { + "epoch": 0.29, + "learning_rate": 0.000356629889669007, + "loss": 1.438, + "theoretical_loss": 3.659791197449188, + "tokens_seen": 970063872 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035661985957873623, + "loss": 1.4578, + "theoretical_loss": 3.6597675361021906, + "tokens_seen": 970129408 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035660982948846536, + "loss": 1.2365, + "theoretical_loss": 3.6597438768010777, + "tokens_seen": 970194944 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003565997993981946, + "loss": 1.3227, + "theoretical_loss": 3.659720219545533, + "tokens_seen": 970260480 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003565897693079238, + "loss": 1.3622, + "theoretical_loss": 3.659696564335242, + "tokens_seen": 970326016 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035657973921765295, + "loss": 1.3868, + "theoretical_loss": 3.6596729111698894, + "tokens_seen": 970391552 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003565697091273822, + "loss": 1.3922, + "theoretical_loss": 3.659649260049161, + "tokens_seen": 970457088 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003565596790371113, + "loss": 1.3916, + "theoretical_loss": 3.659625610972741, + "tokens_seen": 970522624 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035654964894684055, + "loss": 1.325, + "theoretical_loss": 3.6596019639403146, + "tokens_seen": 970588160 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035653961885656973, + "loss": 1.4231, + "theoretical_loss": 3.659578318951569, + "tokens_seen": 970653696 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003565295887662989, + "loss": 1.3883, + "theoretical_loss": 3.6595546760061874, + "tokens_seen": 970719232 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003565195586760281, + "loss": 1.3814, + "theoretical_loss": 3.6595310351038566, + "tokens_seen": 970784768 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035650952858575733, + "loss": 1.3598, + "theoretical_loss": 3.6595073962442615, + "tokens_seen": 970850304 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035649949849548646, + "loss": 1.299, + "theoretical_loss": 3.659483759427088, + "tokens_seen": 970915840 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003564894684052157, + "loss": 1.3157, + "theoretical_loss": 3.659460124652022, + "tokens_seen": 970981376 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003564794383149448, + "loss": 1.3636, + "theoretical_loss": 3.6594364919187488, + "tokens_seen": 971046912 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035646940822467405, + "loss": 1.1929, + "theoretical_loss": 3.6594128612269543, + "tokens_seen": 971112448 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035645937813440323, + "loss": 1.3697, + "theoretical_loss": 3.659389232576325, + "tokens_seen": 971177984 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003564493480441324, + "loss": 1.2891, + "theoretical_loss": 3.6593656059665465, + "tokens_seen": 971243520 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003564393179538616, + "loss": 1.4029, + "theoretical_loss": 3.6593419813973043, + "tokens_seen": 971309056 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003564292878635908, + "loss": 1.4077, + "theoretical_loss": 3.6593183588682847, + "tokens_seen": 971374592 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035641925777331996, + "loss": 1.4185, + "theoretical_loss": 3.6592947383791747, + "tokens_seen": 971440128 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003564092276830492, + "loss": 1.3977, + "theoretical_loss": 3.6592711199296595, + "tokens_seen": 971505664 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": 0.00025169082800857723, + "objective/train/docs_used": 558372, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.565185785293579, + "objective/train/original_loss": 2.5651862621307373, + "objective/train/theoretical_loss": 3.6592475035194267, + "objective/train/tokens_used": 992031200, + "objective/train/value_avg": -0.005870819091796875, + "objective/train/value_loss": 4.985735358786769e-05, + "objective/train/value_max": -0.00037550926208496094, + "objective/train/value_min": -0.105224609375, + "objective/train/value_reward_corr": 0.4389899594448866, + "objective/train/value_std": 0.004604339599609375, + "objective/train/weight_avg": 1.0049490928649902, + "objective/train/weighted_lm_loss": 2.5687198638916016, + "objective/train/weights_max": 2.0979528427124023, + "objective/train/weights_min": 0.4770635664463043, + "theoretical_loss": 3.6592475035194267, + "tokens_seen": 971571200 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003563991975927783, + "loss": 1.3692, + "theoretical_loss": 3.6592475035194267, + "tokens_seen": 971571200 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035638916750250756, + "loss": 1.4303, + "theoretical_loss": 3.659223889148161, + "tokens_seen": 971636736 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003563791374122367, + "loss": 1.2952, + "theoretical_loss": 3.6592002768155503, + "tokens_seen": 971702272 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003563691073219659, + "loss": 1.3781, + "theoretical_loss": 3.6591766665212804, + "tokens_seen": 971767808 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003563590772316951, + "loss": 1.3002, + "theoretical_loss": 3.659153058265038, + "tokens_seen": 971833344 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003563490471414243, + "loss": 1.367, + "theoretical_loss": 3.6591294520465096, + "tokens_seen": 971898880 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035633901705115346, + "loss": 1.4801, + "theoretical_loss": 3.6591058478653817, + "tokens_seen": 971964416 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003563289869608827, + "loss": 1.4529, + "theoretical_loss": 3.6590822457213426, + "tokens_seen": 972029952 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003563189568706118, + "loss": 1.2973, + "theoretical_loss": 3.6590586456140777, + "tokens_seen": 972095488 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035630892678034106, + "loss": 1.4691, + "theoretical_loss": 3.659035047543274, + "tokens_seen": 972161024 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003562988966900702, + "loss": 1.3982, + "theoretical_loss": 3.659011451508619, + "tokens_seen": 972226560 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003562888665997994, + "loss": 1.178, + "theoretical_loss": 3.6589878575098, + "tokens_seen": 972292096 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003562788365095286, + "loss": 1.2681, + "theoretical_loss": 3.658964265546503, + "tokens_seen": 972357632 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003562688064192578, + "loss": 1.4716, + "theoretical_loss": 3.6589406756184166, + "tokens_seen": 972423168 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035625877632898697, + "loss": 1.3556, + "theoretical_loss": 3.6589170877252277, + "tokens_seen": 972488704 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035624874623871615, + "loss": 1.4289, + "theoretical_loss": 3.658893501866623, + "tokens_seen": 972554240 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035623871614844533, + "loss": 1.3065, + "theoretical_loss": 3.6588699180422903, + "tokens_seen": 972619776 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035622868605817456, + "loss": 1.3044, + "theoretical_loss": 3.658846336251917, + "tokens_seen": 972685312 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003562186559679037, + "loss": 1.3612, + "theoretical_loss": 3.658822756495191, + "tokens_seen": 972750848 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003562086258776329, + "loss": 1.4175, + "theoretical_loss": 3.6587991787717997, + "tokens_seen": 972816384 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035619859578736205, + "loss": 1.4855, + "theoretical_loss": 3.6587756030814313, + "tokens_seen": 972881920 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003561885656970913, + "loss": 1.3925, + "theoretical_loss": 3.6587520294237725, + "tokens_seen": 972947456 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035617853560682047, + "loss": 1.3714, + "theoretical_loss": 3.658728457798512, + "tokens_seen": 973012992 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035616850551654965, + "loss": 1.3975, + "theoretical_loss": 3.658704888205337, + "tokens_seen": 973078528 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035615847542627883, + "loss": 1.3215, + "theoretical_loss": 3.6586813206439364, + "tokens_seen": 973144064 + }, + { + "epoch": 0.29, + "objective/train/advantage_avg": 0.0002740146592259407, + "objective/train/docs_used": 558840, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6844940185546875, + "objective/train/original_loss": 2.6844942569732666, + "objective/train/theoretical_loss": 3.6586577551139974, + "objective/train/tokens_used": 993669600, + "objective/train/value_avg": -0.006092071533203125, + "objective/train/value_loss": 5.91364259889815e-05, + "objective/train/value_max": -0.00031495094299316406, + "objective/train/value_min": -0.056640625, + "objective/train/value_reward_corr": 0.43636653295825295, + "objective/train/value_std": 0.005237579345703125, + "objective/train/weight_avg": 1.0055345296859741, + "objective/train/weighted_lm_loss": 2.701421022415161, + "objective/train/weights_max": 1.5604102611541748, + "objective/train/weights_min": 0.3591614067554474, + "theoretical_loss": 3.6586577551139974, + "tokens_seen": 973209600 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035614844533600807, + "loss": 1.3815, + "theoretical_loss": 3.6586577551139974, + "tokens_seen": 973209600 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003561384152457372, + "loss": 1.3579, + "theoretical_loss": 3.6586341916152083, + "tokens_seen": 973275136 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035612838515546643, + "loss": 1.3287, + "theoretical_loss": 3.6586106301472583, + "tokens_seen": 973340672 + }, + { + "epoch": 0.29, + "learning_rate": 0.00035611835506519556, + "loss": 1.4511, + "theoretical_loss": 3.6585870707098342, + "tokens_seen": 973406208 + }, + { + "epoch": 0.29, + "learning_rate": 0.0003561083249749248, + "loss": 1.3136, + "theoretical_loss": 3.658563513302625, + "tokens_seen": 973471744 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035609829488465397, + "loss": 1.4294, + "theoretical_loss": 3.658539957925319, + "tokens_seen": 973537280 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035608826479438315, + "loss": 1.2316, + "theoretical_loss": 3.6585164045776044, + "tokens_seen": 973602816 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035607823470411233, + "loss": 1.3279, + "theoretical_loss": 3.6584928532591703, + "tokens_seen": 973668352 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003560682046138415, + "loss": 1.4936, + "theoretical_loss": 3.6584693039697047, + "tokens_seen": 973733888 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003560581745235707, + "loss": 1.4064, + "theoretical_loss": 3.658445756708897, + "tokens_seen": 973799424 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035604814443329993, + "loss": 1.371, + "theoretical_loss": 3.6584222114764353, + "tokens_seen": 973864960 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035603811434302906, + "loss": 1.271, + "theoretical_loss": 3.6583986682720084, + "tokens_seen": 973930496 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003560280842527583, + "loss": 1.3624, + "theoretical_loss": 3.6583751270953053, + "tokens_seen": 973996032 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003560180541624874, + "loss": 1.4153, + "theoretical_loss": 3.6583515879460147, + "tokens_seen": 974061568 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035600802407221666, + "loss": 1.3527, + "theoretical_loss": 3.658328050823826, + "tokens_seen": 974127104 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035599799398194584, + "loss": 1.319, + "theoretical_loss": 3.658304515728429, + "tokens_seen": 974192640 + }, + { + "epoch": 0.3, + "learning_rate": 0.000355987963891675, + "loss": 1.4292, + "theoretical_loss": 3.658280982659511, + "tokens_seen": 974258176 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003559779338014042, + "loss": 1.3926, + "theoretical_loss": 3.6582574516167625, + "tokens_seen": 974323712 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035596790371113344, + "loss": 1.3062, + "theoretical_loss": 3.658233922599873, + "tokens_seen": 974389248 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035595787362086256, + "loss": 1.2864, + "theoretical_loss": 3.658210395608531, + "tokens_seen": 974454784 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003559478435305918, + "loss": 1.4328, + "theoretical_loss": 3.658186870642426, + "tokens_seen": 974520320 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003559378134403209, + "loss": 1.4031, + "theoretical_loss": 3.6581633477012483, + "tokens_seen": 974585856 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035592778335005016, + "loss": 1.3627, + "theoretical_loss": 3.6581398267846863, + "tokens_seen": 974651392 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035591775325977934, + "loss": 1.45, + "theoretical_loss": 3.6581163078924304, + "tokens_seen": 974716928 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003559077231695085, + "loss": 1.5312, + "theoretical_loss": 3.65809279102417, + "tokens_seen": 974782464 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -2.250070610898547e-05, + "objective/train/docs_used": 560118, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.755704164505005, + "objective/train/original_loss": 2.7557036876678467, + "objective/train/theoretical_loss": 3.658069276179595, + "objective/train/tokens_used": 995308000, + "objective/train/value_avg": -0.00569915771484375, + "objective/train/value_loss": 6.525949720526114e-05, + "objective/train/value_max": -0.0005154609680175781, + "objective/train/value_min": -0.059112548828125, + "objective/train/value_reward_corr": 0.38560305247457183, + "objective/train/value_std": 0.004680633544921875, + "objective/train/weight_avg": 1.0026776790618896, + "objective/train/weighted_lm_loss": 2.7620882987976074, + "objective/train/weights_max": 1.5468634366989136, + "objective/train/weights_min": 0.13754990696907043, + "theoretical_loss": 3.658069276179595, + "tokens_seen": 974848000 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003558976930792377, + "loss": 1.3099, + "theoretical_loss": 3.658069276179595, + "tokens_seen": 974848000 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003558876629889669, + "loss": 1.247, + "theoretical_loss": 3.6580457633583956, + "tokens_seen": 974913536 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035587763289869607, + "loss": 1.394, + "theoretical_loss": 3.658022252560261, + "tokens_seen": 974979072 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003558676028084253, + "loss": 1.4319, + "theoretical_loss": 3.6579987437848813, + "tokens_seen": 975044608 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035585757271815443, + "loss": 1.3561, + "theoretical_loss": 3.657975237031947, + "tokens_seen": 975110144 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035584754262788366, + "loss": 1.4222, + "theoretical_loss": 3.657951732301148, + "tokens_seen": 975175680 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003558375125376129, + "loss": 1.3999, + "theoretical_loss": 3.657928229592174, + "tokens_seen": 975241216 + }, + { + "epoch": 0.3, + "learning_rate": 0.000355827482447342, + "loss": 1.327, + "theoretical_loss": 3.6579047289047155, + "tokens_seen": 975306752 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035581745235707126, + "loss": 1.4903, + "theoretical_loss": 3.6578812302384636, + "tokens_seen": 975372288 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003558074222668004, + "loss": 1.3521, + "theoretical_loss": 3.6578577335931075, + "tokens_seen": 975437824 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003557973921765296, + "loss": 1.2975, + "theoretical_loss": 3.657834238968338, + "tokens_seen": 975503360 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003557873620862588, + "loss": 1.2576, + "theoretical_loss": 3.657810746363846, + "tokens_seen": 975568896 + }, + { + "epoch": 0.3, + "learning_rate": 0.000355777331995988, + "loss": 1.3902, + "theoretical_loss": 3.6577872557793216, + "tokens_seen": 975634432 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035576730190571717, + "loss": 1.408, + "theoretical_loss": 3.657763767214456, + "tokens_seen": 975699968 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035575727181544635, + "loss": 1.3538, + "theoretical_loss": 3.657740280668939, + "tokens_seen": 975765504 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035574724172517553, + "loss": 1.3708, + "theoretical_loss": 3.657716796142463, + "tokens_seen": 975831040 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035573721163490476, + "loss": 1.3311, + "theoretical_loss": 3.657693313634717, + "tokens_seen": 975896576 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003557271815446339, + "loss": 1.4554, + "theoretical_loss": 3.657669833145393, + "tokens_seen": 975962112 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003557171514543631, + "loss": 1.4121, + "theoretical_loss": 3.6576463546741813, + "tokens_seen": 976027648 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035570712136409225, + "loss": 1.4346, + "theoretical_loss": 3.6576228782207743, + "tokens_seen": 976093184 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003556970912738215, + "loss": 1.4796, + "theoretical_loss": 3.6575994037848614, + "tokens_seen": 976158720 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035568706118355067, + "loss": 1.3707, + "theoretical_loss": 3.657575931366135, + "tokens_seen": 976224256 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035567703109327985, + "loss": 1.3829, + "theoretical_loss": 3.6575524609642858, + "tokens_seen": 976289792 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035566700100300903, + "loss": 1.3682, + "theoretical_loss": 3.6575289925790053, + "tokens_seen": 976355328 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035565697091273827, + "loss": 1.3784, + "theoretical_loss": 3.657505526209985, + "tokens_seen": 976420864 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -0.0009873643284663558, + "objective/train/docs_used": 560789, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7401647567749023, + "objective/train/original_loss": 2.7401649951934814, + "objective/train/theoretical_loss": 3.657482061856916, + "objective/train/tokens_used": 996946400, + "objective/train/value_avg": -0.00469207763671875, + "objective/train/value_loss": 6.6398672061041e-05, + "objective/train/value_max": -0.00023055076599121094, + "objective/train/value_min": -0.073974609375, + "objective/train/value_reward_corr": 0.35918739518420506, + "objective/train/value_std": 0.0036468505859375, + "objective/train/weight_avg": 0.9932113289833069, + "objective/train/weighted_lm_loss": 2.7230944633483887, + "objective/train/weights_max": 2.0954034328460693, + "objective/train/weights_min": 0.24794599413871765, + "theoretical_loss": 3.657482061856916, + "tokens_seen": 976486400 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003556469408224674, + "loss": 1.5361, + "theoretical_loss": 3.657482061856916, + "tokens_seen": 976486400 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035563691073219663, + "loss": 1.342, + "theoretical_loss": 3.65745859951949, + "tokens_seen": 976551936 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035562688064192576, + "loss": 1.3405, + "theoretical_loss": 3.657435139197399, + "tokens_seen": 976617472 + }, + { + "epoch": 0.3, + "learning_rate": 0.000355616850551655, + "loss": 1.3315, + "theoretical_loss": 3.6574116808903345, + "tokens_seen": 976683008 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035560682046138417, + "loss": 1.4222, + "theoretical_loss": 3.657388224597988, + "tokens_seen": 976748544 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035559679037111335, + "loss": 1.348, + "theoretical_loss": 3.657364770320051, + "tokens_seen": 976814080 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035558676028084253, + "loss": 1.438, + "theoretical_loss": 3.6573413180562158, + "tokens_seen": 976879616 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003555767301905717, + "loss": 1.3978, + "theoretical_loss": 3.657317867806175, + "tokens_seen": 976945152 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003555667001003009, + "loss": 1.4322, + "theoretical_loss": 3.6572944195696193, + "tokens_seen": 977010688 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035555667001003013, + "loss": 1.3748, + "theoretical_loss": 3.6572709733462414, + "tokens_seen": 977076224 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035554663991975926, + "loss": 1.3885, + "theoretical_loss": 3.6572475291357334, + "tokens_seen": 977141760 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003555366098294885, + "loss": 1.3615, + "theoretical_loss": 3.657224086937788, + "tokens_seen": 977207296 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003555265797392176, + "loss": 1.4798, + "theoretical_loss": 3.6572006467520968, + "tokens_seen": 977272832 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035551654964894686, + "loss": 1.4485, + "theoretical_loss": 3.657177208578352, + "tokens_seen": 977338368 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035550651955867604, + "loss": 1.4257, + "theoretical_loss": 3.6571537724162466, + "tokens_seen": 977403904 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003554964894684052, + "loss": 1.3718, + "theoretical_loss": 3.657130338265473, + "tokens_seen": 977469440 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003554864593781344, + "loss": 1.4278, + "theoretical_loss": 3.6571069061257235, + "tokens_seen": 977534976 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035547642928786364, + "loss": 1.4602, + "theoretical_loss": 3.6570834759966906, + "tokens_seen": 977600512 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035546639919759276, + "loss": 1.39, + "theoretical_loss": 3.6570600478780673, + "tokens_seen": 977666048 + }, + { + "epoch": 0.3, + "learning_rate": 0.000355456369107322, + "loss": 1.3973, + "theoretical_loss": 3.6570366217695467, + "tokens_seen": 977731584 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003554463390170511, + "loss": 1.3651, + "theoretical_loss": 3.6570131976708202, + "tokens_seen": 977797120 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035543630892678036, + "loss": 1.4212, + "theoretical_loss": 3.6569897755815823, + "tokens_seen": 977862656 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035542627883650954, + "loss": 1.4194, + "theoretical_loss": 3.6569663555015257, + "tokens_seen": 977928192 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003554162487462387, + "loss": 1.3419, + "theoretical_loss": 3.656942937430342, + "tokens_seen": 977993728 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003554062186559679, + "loss": 1.4551, + "theoretical_loss": 3.656919521367726, + "tokens_seen": 978059264 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -0.0015108350198715925, + "objective/train/docs_used": 561934, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7491533756256104, + "objective/train/original_loss": 2.7491536140441895, + "objective/train/theoretical_loss": 3.6568961073133703, + "objective/train/tokens_used": 998584800, + "objective/train/value_avg": -0.00514984130859375, + "objective/train/value_loss": 7.743048627162352e-05, + "objective/train/value_max": -0.0003418922424316406, + "objective/train/value_min": -0.06488037109375, + "objective/train/value_reward_corr": 0.4111987522051867, + "objective/train/value_std": 0.0043792724609375, + "objective/train/weight_avg": 0.9885168075561523, + "objective/train/weighted_lm_loss": 2.7370662689208984, + "objective/train/weights_max": 1.7383859157562256, + "objective/train/weights_min": 0.4079357981681824, + "theoretical_loss": 3.6568961073133703, + "tokens_seen": 978124800 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003553961885656971, + "loss": 1.4362, + "theoretical_loss": 3.6568961073133703, + "tokens_seen": 978124800 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035538615847542627, + "loss": 1.3141, + "theoretical_loss": 3.656872695266968, + "tokens_seen": 978190336 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003553761283851555, + "loss": 1.3972, + "theoretical_loss": 3.656849285228212, + "tokens_seen": 978255872 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035536609829488463, + "loss": 1.3853, + "theoretical_loss": 3.6568258771967965, + "tokens_seen": 978321408 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035535606820461386, + "loss": 1.32, + "theoretical_loss": 3.656802471172414, + "tokens_seen": 978386944 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035534603811434304, + "loss": 1.2537, + "theoretical_loss": 3.656779067154759, + "tokens_seen": 978452480 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003553360080240722, + "loss": 1.3215, + "theoretical_loss": 3.656755665143524, + "tokens_seen": 978518016 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003553259779338014, + "loss": 1.3974, + "theoretical_loss": 3.656732265138403, + "tokens_seen": 978583552 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003553159478435306, + "loss": 1.2796, + "theoretical_loss": 3.6567088671390904, + "tokens_seen": 978649088 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035530591775325977, + "loss": 1.3602, + "theoretical_loss": 3.6566854711452796, + "tokens_seen": 978714624 + }, + { + "epoch": 0.3, + "learning_rate": 0.000355295887662989, + "loss": 1.4528, + "theoretical_loss": 3.656662077156664, + "tokens_seen": 978780160 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035528585757271813, + "loss": 1.4293, + "theoretical_loss": 3.656638685172937, + "tokens_seen": 978845696 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035527582748244737, + "loss": 1.3564, + "theoretical_loss": 3.6566152951937942, + "tokens_seen": 978911232 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003552657973921765, + "loss": 1.3652, + "theoretical_loss": 3.656591907218928, + "tokens_seen": 978976768 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035525576730190573, + "loss": 1.4212, + "theoretical_loss": 3.656568521248034, + "tokens_seen": 979042304 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003552457372116349, + "loss": 1.4407, + "theoretical_loss": 3.6565451372808053, + "tokens_seen": 979107840 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003552357071213641, + "loss": 1.3967, + "theoretical_loss": 3.6565217553169367, + "tokens_seen": 979173376 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035522567703109327, + "loss": 1.395, + "theoretical_loss": 3.6564983753561218, + "tokens_seen": 979238912 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035521564694082245, + "loss": 1.4698, + "theoretical_loss": 3.6564749973980555, + "tokens_seen": 979304448 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035520561685055163, + "loss": 1.3239, + "theoretical_loss": 3.6564516214424323, + "tokens_seen": 979369984 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035519558676028087, + "loss": 1.3278, + "theoretical_loss": 3.6564282474889467, + "tokens_seen": 979435520 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035518555667001, + "loss": 1.3684, + "theoretical_loss": 3.6564048755372927, + "tokens_seen": 979501056 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035517552657973923, + "loss": 1.4539, + "theoretical_loss": 3.6563815055871656, + "tokens_seen": 979566592 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003551654964894684, + "loss": 1.4835, + "theoretical_loss": 3.65635813763826, + "tokens_seen": 979632128 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003551554663991976, + "loss": 1.4555, + "theoretical_loss": 3.6563347716902697, + "tokens_seen": 979697664 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -0.0008755629532970488, + "objective/train/docs_used": 562605, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0042402744293213, + "objective/train/original_loss": 3.004240036010742, + "objective/train/theoretical_loss": 3.656311407742891, + "objective/train/tokens_used": 1000223200, + "objective/train/value_avg": -0.005558013916015625, + "objective/train/value_loss": 6.94130576448515e-05, + "objective/train/value_max": -0.00032258033752441406, + "objective/train/value_min": -0.06304931640625, + "objective/train/value_reward_corr": 0.39560514836368055, + "objective/train/value_std": 0.004413604736328125, + "objective/train/weight_avg": 0.994359016418457, + "objective/train/weighted_lm_loss": 2.990288257598877, + "objective/train/weights_max": 1.7042789459228516, + "objective/train/weights_min": 0.25254711508750916, + "theoretical_loss": 3.656311407742891, + "tokens_seen": 979763200 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003551454363089268, + "loss": 1.4511, + "theoretical_loss": 3.656311407742891, + "tokens_seen": 979763200 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035513540621865596, + "loss": 1.3629, + "theoretical_loss": 3.6562880457958182, + "tokens_seen": 979828736 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035512537612838514, + "loss": 1.3747, + "theoretical_loss": 3.656264685848746, + "tokens_seen": 979894272 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035511534603811437, + "loss": 1.364, + "theoretical_loss": 3.6562413279013697, + "tokens_seen": 979959808 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003551053159478435, + "loss": 1.424, + "theoretical_loss": 3.6562179719533843, + "tokens_seen": 980025344 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035509528585757273, + "loss": 1.483, + "theoretical_loss": 3.656194618004485, + "tokens_seen": 980090880 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003550852557673019, + "loss": 1.3892, + "theoretical_loss": 3.6561712660543675, + "tokens_seen": 980156416 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003550752256770311, + "loss": 1.2983, + "theoretical_loss": 3.656147916102726, + "tokens_seen": 980221952 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035506519558676033, + "loss": 1.3767, + "theoretical_loss": 3.656124568149257, + "tokens_seen": 980287488 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035505516549648946, + "loss": 1.3413, + "theoretical_loss": 3.6561012221936555, + "tokens_seen": 980353024 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003550451354062187, + "loss": 1.4761, + "theoretical_loss": 3.656077878235617, + "tokens_seen": 980418560 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003550351053159478, + "loss": 1.3528, + "theoretical_loss": 3.6560545362748367, + "tokens_seen": 980484096 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035502507522567706, + "loss": 1.3836, + "theoretical_loss": 3.6560311963110106, + "tokens_seen": 980549632 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035501504513540624, + "loss": 1.3037, + "theoretical_loss": 3.656007858343835, + "tokens_seen": 980615168 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003550050150451354, + "loss": 1.3483, + "theoretical_loss": 3.6559845223730045, + "tokens_seen": 980680704 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003549949849548646, + "loss": 1.3235, + "theoretical_loss": 3.6559611883982157, + "tokens_seen": 980746240 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035498495486459384, + "loss": 1.2941, + "theoretical_loss": 3.6559378564191647, + "tokens_seen": 980811776 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035497492477432296, + "loss": 1.3692, + "theoretical_loss": 3.655914526435547, + "tokens_seen": 980877312 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003549648946840522, + "loss": 1.4558, + "theoretical_loss": 3.6558911984470583, + "tokens_seen": 980942848 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003549548645937813, + "loss": 1.3396, + "theoretical_loss": 3.655867872453395, + "tokens_seen": 981008384 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035494483450351056, + "loss": 1.4079, + "theoretical_loss": 3.6558445484542537, + "tokens_seen": 981073920 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035493480441323974, + "loss": 1.3964, + "theoretical_loss": 3.6558212264493304, + "tokens_seen": 981139456 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003549247743229689, + "loss": 1.2415, + "theoretical_loss": 3.6557979064383215, + "tokens_seen": 981204992 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003549147442326981, + "loss": 1.3648, + "theoretical_loss": 3.6557745884209227, + "tokens_seen": 981270528 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003549047141424273, + "loss": 1.4049, + "theoretical_loss": 3.6557512723968313, + "tokens_seen": 981336064 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -0.0004640332190319896, + "objective/train/docs_used": 563033, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6978962421417236, + "objective/train/original_loss": 2.6978957653045654, + "objective/train/theoretical_loss": 3.6557279583657434, + "objective/train/tokens_used": 1001861600, + "objective/train/value_avg": -0.0055389404296875, + "objective/train/value_loss": 0.00011271083349129185, + "objective/train/value_max": -0.00034999847412109375, + "objective/train/value_min": -0.097900390625, + "objective/train/value_reward_corr": 0.39923483623943684, + "objective/train/value_std": 0.005046844482421875, + "objective/train/weight_avg": 0.9998717308044434, + "objective/train/weighted_lm_loss": 2.7033274173736572, + "objective/train/weights_max": 2.451082468032837, + "objective/train/weights_min": 0.19187118113040924, + "theoretical_loss": 3.6557279583657434, + "tokens_seen": 981401600 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035489468405215647, + "loss": 1.4781, + "theoretical_loss": 3.6557279583657434, + "tokens_seen": 981401600 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003548846539618857, + "loss": 1.432, + "theoretical_loss": 3.6557046463273557, + "tokens_seen": 981467136 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035487462387161483, + "loss": 1.4574, + "theoretical_loss": 3.655681336281365, + "tokens_seen": 981532672 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035486459378134406, + "loss": 1.4894, + "theoretical_loss": 3.6556580282274673, + "tokens_seen": 981598208 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035485456369107324, + "loss": 1.3937, + "theoretical_loss": 3.6556347221653605, + "tokens_seen": 981663744 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003548445336008024, + "loss": 1.4249, + "theoretical_loss": 3.65561141809474, + "tokens_seen": 981729280 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003548345035105316, + "loss": 1.4465, + "theoretical_loss": 3.655588116015304, + "tokens_seen": 981794816 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003548244734202608, + "loss": 1.3367, + "theoretical_loss": 3.6555648159267484, + "tokens_seen": 981860352 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035481444332998997, + "loss": 1.3633, + "theoretical_loss": 3.6555415178287713, + "tokens_seen": 981925888 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003548044132397192, + "loss": 1.3825, + "theoretical_loss": 3.655518221721069, + "tokens_seen": 981991424 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035479438314944833, + "loss": 1.4466, + "theoretical_loss": 3.655494927603339, + "tokens_seen": 982056960 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035478435305917757, + "loss": 1.4228, + "theoretical_loss": 3.655471635475279, + "tokens_seen": 982122496 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003547743229689067, + "loss": 1.4367, + "theoretical_loss": 3.655448345336585, + "tokens_seen": 982188032 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035476429287863593, + "loss": 1.4359, + "theoretical_loss": 3.6554250571869558, + "tokens_seen": 982253568 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003547542627883651, + "loss": 1.4068, + "theoretical_loss": 3.6554017710260878, + "tokens_seen": 982319104 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003547442326980943, + "loss": 1.4381, + "theoretical_loss": 3.655378486853679, + "tokens_seen": 982384640 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035473420260782347, + "loss": 1.4086, + "theoretical_loss": 3.655355204669427, + "tokens_seen": 982450176 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035472417251755265, + "loss": 1.4874, + "theoretical_loss": 3.655331924473029, + "tokens_seen": 982515712 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035471414242728183, + "loss": 1.4627, + "theoretical_loss": 3.6553086462641833, + "tokens_seen": 982581248 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035470411233701107, + "loss": 1.4256, + "theoretical_loss": 3.6552853700425874, + "tokens_seen": 982646784 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003546940822467402, + "loss": 1.4844, + "theoretical_loss": 3.655262095807939, + "tokens_seen": 982712320 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035468405215646943, + "loss": 1.4775, + "theoretical_loss": 3.6552388235599356, + "tokens_seen": 982777856 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003546740220661986, + "loss": 1.3822, + "theoretical_loss": 3.655215553298276, + "tokens_seen": 982843392 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003546639919759278, + "loss": 1.3406, + "theoretical_loss": 3.655192285022658, + "tokens_seen": 982908928 + }, + { + "epoch": 0.3, + "learning_rate": 0.000354653961885657, + "loss": 1.3022, + "theoretical_loss": 3.655169018732779, + "tokens_seen": 982974464 + }, + { + "debugging/Self-BLEU-5": 0.25837060001996337, + "debugging/distinct-1-grams": 0.6894724120204403, + "debugging/distinct-2-grams": 0.9227700050458512, + "debugging/entropy-1-grams": 5.206518678302481, + "debugging/entropy-2-grams": 5.923409593722085, + "debugging/length": 467.1666666666667, + "debugging/num_segments": 6, + "debugging/raw_token_scores_avg": 0.005825165659189224, + "debugging/raw_token_scores_std": 0.008988859131932259, + "debugging/score": 0.005928728759380126, + "debugging/score_std": 0.004287858360742871, + "epoch": 0.3, + "objective/train/advantage_avg": -0.0007336692069657147, + "objective/train/docs_used": 564216, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6907410621643066, + "objective/train/original_loss": 2.6907408237457275, + "objective/train/theoretical_loss": 3.6551457544283386, + "objective/train/tokens_used": 1003500000, + "objective/train/value_avg": -0.005092620849609375, + "objective/train/value_loss": 7.167224976001307e-05, + "objective/train/value_max": -0.0003249645233154297, + "objective/train/value_min": -0.075439453125, + "objective/train/value_reward_corr": 0.37954376329653966, + "objective/train/value_std": 0.004802703857421875, + "objective/train/weight_avg": 0.995888888835907, + "objective/train/weighted_lm_loss": 2.687798500061035, + "objective/train/weights_max": 1.938281774520874, + "objective/train/weights_min": 0.1973382830619812, + "theoretical_loss": 3.6551457544283386, + "tokens_seen": 983040000 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035464393179538616, + "loss": 1.425, + "theoretical_loss": 3.6551457544283386, + "tokens_seen": 983040000 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035463390170511534, + "loss": 1.3954, + "theoretical_loss": 3.655122492109034, + "tokens_seen": 983105536 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035462387161484457, + "loss": 1.2526, + "theoretical_loss": 3.655099231774563, + "tokens_seen": 983171072 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003546138415245737, + "loss": 1.4401, + "theoretical_loss": 3.6550759734246245, + "tokens_seen": 983236608 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035460381143430293, + "loss": 1.3687, + "theoretical_loss": 3.6550527170589175, + "tokens_seen": 983302144 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035459378134403206, + "loss": 1.4566, + "theoretical_loss": 3.6550294626771405, + "tokens_seen": 983367680 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003545837512537613, + "loss": 1.3892, + "theoretical_loss": 3.655006210278991, + "tokens_seen": 983433216 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003545737211634905, + "loss": 1.3309, + "theoretical_loss": 3.654982959864168, + "tokens_seen": 983498752 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035456369107321966, + "loss": 1.2674, + "theoretical_loss": 3.6549597114323706, + "tokens_seen": 983564288 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035455366098294884, + "loss": 1.4475, + "theoretical_loss": 3.6549364649832974, + "tokens_seen": 983629824 + }, + { + "epoch": 0.3, + "learning_rate": 0.000354543630892678, + "loss": 1.3875, + "theoretical_loss": 3.6549132205166472, + "tokens_seen": 983695360 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003545336008024072, + "loss": 1.4416, + "theoretical_loss": 3.6548899780321187, + "tokens_seen": 983760896 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035452357071213644, + "loss": 1.4793, + "theoretical_loss": 3.654866737529411, + "tokens_seen": 983826432 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035451354062186556, + "loss": 1.411, + "theoretical_loss": 3.6548434990082233, + "tokens_seen": 983891968 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003545035105315948, + "loss": 1.3363, + "theoretical_loss": 3.654820262468254, + "tokens_seen": 983957504 + }, + { + "epoch": 0.3, + "learning_rate": 0.000354493480441324, + "loss": 1.5295, + "theoretical_loss": 3.6547970279092032, + "tokens_seen": 984023040 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035448345035105316, + "loss": 1.466, + "theoretical_loss": 3.6547737953307697, + "tokens_seen": 984088576 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035447342026078234, + "loss": 1.3946, + "theoretical_loss": 3.6547505647326526, + "tokens_seen": 984154112 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003544633901705115, + "loss": 1.3327, + "theoretical_loss": 3.6547273361145507, + "tokens_seen": 984219648 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003544533600802407, + "loss": 1.3264, + "theoretical_loss": 3.6547041094761648, + "tokens_seen": 984285184 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035444332998996994, + "loss": 1.5025, + "theoretical_loss": 3.6546808848171937, + "tokens_seen": 984350720 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035443329989969907, + "loss": 1.4519, + "theoretical_loss": 3.654657662137337, + "tokens_seen": 984416256 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003544232698094283, + "loss": 1.2658, + "theoretical_loss": 3.6546344414362935, + "tokens_seen": 984481792 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035441323971915743, + "loss": 1.3148, + "theoretical_loss": 3.6546112227137644, + "tokens_seen": 984547328 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035440320962888667, + "loss": 1.4775, + "theoretical_loss": 3.6545880059694484, + "tokens_seen": 984612864 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -0.0008746817475184798, + "objective/train/docs_used": 564602, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.873075246810913, + "objective/train/original_loss": 2.873075008392334, + "objective/train/theoretical_loss": 3.654564791203045, + "objective/train/tokens_used": 1005138400, + "objective/train/value_avg": -0.0055084228515625, + "objective/train/value_loss": 9.857625263975933e-05, + "objective/train/value_max": -0.0004494190216064453, + "objective/train/value_min": -0.06829833984375, + "objective/train/value_reward_corr": 0.39956209097881545, + "objective/train/value_std": 0.004791259765625, + "objective/train/weight_avg": 0.9955049157142639, + "objective/train/weighted_lm_loss": 2.8689820766448975, + "objective/train/weights_max": 1.979775309562683, + "objective/train/weights_min": 0.30216410756111145, + "theoretical_loss": 3.654564791203045, + "tokens_seen": 984678400 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035439317953861585, + "loss": 1.3932, + "theoretical_loss": 3.654564791203045, + "tokens_seen": 984678400 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035438314944834503, + "loss": 1.3891, + "theoretical_loss": 3.654541578414255, + "tokens_seen": 984743936 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003543731193580742, + "loss": 1.4134, + "theoretical_loss": 3.654518367602778, + "tokens_seen": 984809472 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035436308926780344, + "loss": 1.4645, + "theoretical_loss": 3.654495158768314, + "tokens_seen": 984875008 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003543530591775326, + "loss": 1.4508, + "theoretical_loss": 3.654471951910563, + "tokens_seen": 984940544 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003543430290872618, + "loss": 1.3368, + "theoretical_loss": 3.6544487470292246, + "tokens_seen": 985006080 + }, + { + "epoch": 0.3, + "learning_rate": 0.000354332998996991, + "loss": 1.447, + "theoretical_loss": 3.6544255441240003, + "tokens_seen": 985071616 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035432296890672017, + "loss": 1.4233, + "theoretical_loss": 3.6544023431945893, + "tokens_seen": 985137152 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003543129388164494, + "loss": 1.2773, + "theoretical_loss": 3.654379144240692, + "tokens_seen": 985202688 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035430290872617853, + "loss": 1.4306, + "theoretical_loss": 3.654355947262009, + "tokens_seen": 985268224 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035429287863590777, + "loss": 1.364, + "theoretical_loss": 3.6543327522582416, + "tokens_seen": 985333760 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003542828485456369, + "loss": 1.4936, + "theoretical_loss": 3.6543095592290884, + "tokens_seen": 985399296 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035427281845536613, + "loss": 1.35, + "theoretical_loss": 3.654286368174252, + "tokens_seen": 985464832 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003542627883650953, + "loss": 1.3348, + "theoretical_loss": 3.654263179093432, + "tokens_seen": 985530368 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003542527582748245, + "loss": 1.4658, + "theoretical_loss": 3.6542399919863287, + "tokens_seen": 985595904 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035424272818455367, + "loss": 1.4393, + "theoretical_loss": 3.6542168068526433, + "tokens_seen": 985661440 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035423269809428285, + "loss": 1.4498, + "theoretical_loss": 3.654193623692078, + "tokens_seen": 985726976 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035422266800401203, + "loss": 1.4512, + "theoretical_loss": 3.6541704425043315, + "tokens_seen": 985792512 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035421263791374127, + "loss": 1.4175, + "theoretical_loss": 3.654147263289106, + "tokens_seen": 985858048 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003542026078234704, + "loss": 1.4347, + "theoretical_loss": 3.654124086046102, + "tokens_seen": 985923584 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035419257773319963, + "loss": 1.3491, + "theoretical_loss": 3.6541009107750217, + "tokens_seen": 985989120 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003541825476429288, + "loss": 1.475, + "theoretical_loss": 3.654077737475565, + "tokens_seen": 986054656 + }, + { + "epoch": 0.3, + "learning_rate": 0.000354172517552658, + "loss": 1.4082, + "theoretical_loss": 3.654054566147434, + "tokens_seen": 986120192 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003541624874623872, + "loss": 1.4419, + "theoretical_loss": 3.65403139679033, + "tokens_seen": 986185728 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035415245737211636, + "loss": 1.3177, + "theoretical_loss": 3.6540082294039538, + "tokens_seen": 986251264 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -0.0003315691719762981, + "objective/train/docs_used": 565772, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0346908569335938, + "objective/train/original_loss": 3.0346908569335938, + "objective/train/theoretical_loss": 3.6539850639880065, + "objective/train/tokens_used": 1006776800, + "objective/train/value_avg": -0.00567626953125, + "objective/train/value_loss": 8.626774797448888e-05, + "objective/train/value_max": -0.0003199577331542969, + "objective/train/value_min": -0.05419921875, + "objective/train/value_reward_corr": 0.345532580180402, + "objective/train/value_std": 0.004207611083984375, + "objective/train/weight_avg": 1.000381350517273, + "objective/train/weighted_lm_loss": 3.034342050552368, + "objective/train/weights_max": 1.5047982931137085, + "objective/train/weights_min": 0.13856633007526398, + "theoretical_loss": 3.6539850639880065, + "tokens_seen": 986316800 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035414242728184554, + "loss": 1.3941, + "theoretical_loss": 3.6539850639880065, + "tokens_seen": 986316800 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035413239719157477, + "loss": 1.3159, + "theoretical_loss": 3.653961900542191, + "tokens_seen": 986382336 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003541223671013039, + "loss": 1.4824, + "theoretical_loss": 3.6539387390662075, + "tokens_seen": 986447872 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035411233701103314, + "loss": 1.4338, + "theoretical_loss": 3.6539155795597584, + "tokens_seen": 986513408 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035410230692076226, + "loss": 1.3238, + "theoretical_loss": 3.6538924220225457, + "tokens_seen": 986578944 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003540922768304915, + "loss": 1.4457, + "theoretical_loss": 3.6538692664542696, + "tokens_seen": 986644480 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003540822467402207, + "loss": 1.3788, + "theoretical_loss": 3.653846112854634, + "tokens_seen": 986710016 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035407221664994986, + "loss": 1.3682, + "theoretical_loss": 3.653822961223339, + "tokens_seen": 986775552 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035406218655967904, + "loss": 1.4022, + "theoretical_loss": 3.653799811560088, + "tokens_seen": 986841088 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003540521564694082, + "loss": 1.3972, + "theoretical_loss": 3.6537766638645817, + "tokens_seen": 986906624 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003540421263791374, + "loss": 1.514, + "theoretical_loss": 3.653753518136523, + "tokens_seen": 986972160 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035403209628886664, + "loss": 1.3395, + "theoretical_loss": 3.6537303743756144, + "tokens_seen": 987037696 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035402206619859577, + "loss": 1.4324, + "theoretical_loss": 3.6537072325815574, + "tokens_seen": 987103232 + }, + { + "epoch": 0.3, + "learning_rate": 0.000354012036108325, + "loss": 1.404, + "theoretical_loss": 3.6536840927540544, + "tokens_seen": 987168768 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003540020060180542, + "loss": 1.3634, + "theoretical_loss": 3.653660954892808, + "tokens_seen": 987234304 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035399197592778336, + "loss": 1.541, + "theoretical_loss": 3.6536378189975203, + "tokens_seen": 987299840 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035398194583751254, + "loss": 1.4409, + "theoretical_loss": 3.653614685067894, + "tokens_seen": 987365376 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003539719157472417, + "loss": 1.3364, + "theoretical_loss": 3.6535915531036314, + "tokens_seen": 987430912 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003539618856569709, + "loss": 1.3785, + "theoretical_loss": 3.6535684231044354, + "tokens_seen": 987496448 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035395185556670014, + "loss": 1.4016, + "theoretical_loss": 3.6535452950700082, + "tokens_seen": 987561984 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035394182547642927, + "loss": 1.3978, + "theoretical_loss": 3.6535221690000537, + "tokens_seen": 987627520 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003539317953861585, + "loss": 1.4047, + "theoretical_loss": 3.6534990448942732, + "tokens_seen": 987693056 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035392176529588763, + "loss": 1.4404, + "theoretical_loss": 3.6534759227523708, + "tokens_seen": 987758592 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035391173520561687, + "loss": 1.448, + "theoretical_loss": 3.6534528025740483, + "tokens_seen": 987824128 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035390170511534605, + "loss": 1.3554, + "theoretical_loss": 3.6534296843590095, + "tokens_seen": 987889664 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -3.9129750803112984e-05, + "objective/train/docs_used": 566398, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.166884660720825, + "objective/train/original_loss": 2.166884183883667, + "objective/train/theoretical_loss": 3.6534065681069574, + "objective/train/tokens_used": 1008415200, + "objective/train/value_avg": -0.00507354736328125, + "objective/train/value_loss": 7.352345710387453e-05, + "objective/train/value_max": -0.0003726482391357422, + "objective/train/value_min": -0.09686279296875, + "objective/train/value_reward_corr": 0.2603494962613876, + "objective/train/value_std": 0.003742218017578125, + "objective/train/weight_avg": 1.0029525756835938, + "objective/train/weighted_lm_loss": 2.166701555252075, + "objective/train/weights_max": 2.6343276500701904, + "objective/train/weights_min": 0.13778755068778992, + "theoretical_loss": 3.6534065681069574, + "tokens_seen": 987955200 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035389167502507523, + "loss": 1.3738, + "theoretical_loss": 3.6534065681069574, + "tokens_seen": 987955200 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003538816449348044, + "loss": 1.3957, + "theoretical_loss": 3.6533834538175944, + "tokens_seen": 988020736 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035387161484453364, + "loss": 1.3103, + "theoretical_loss": 3.653360341490625, + "tokens_seen": 988086272 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035386158475426277, + "loss": 1.4088, + "theoretical_loss": 3.653337231125751, + "tokens_seen": 988151808 + }, + { + "epoch": 0.3, + "learning_rate": 0.000353851554663992, + "loss": 1.3604, + "theoretical_loss": 3.6533141227226764, + "tokens_seen": 988217344 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035384152457372113, + "loss": 1.2954, + "theoretical_loss": 3.653291016281105, + "tokens_seen": 988282880 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035383149448345037, + "loss": 1.4285, + "theoretical_loss": 3.6532679118007403, + "tokens_seen": 988348416 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035382146439317955, + "loss": 1.3354, + "theoretical_loss": 3.6532448092812846, + "tokens_seen": 988413952 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035381143430290873, + "loss": 1.3986, + "theoretical_loss": 3.653221708722443, + "tokens_seen": 988479488 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003538014042126379, + "loss": 1.4434, + "theoretical_loss": 3.6531986101239173, + "tokens_seen": 988545024 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003537913741223671, + "loss": 1.243, + "theoretical_loss": 3.6531755134854125, + "tokens_seen": 988610560 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003537813440320963, + "loss": 1.3852, + "theoretical_loss": 3.6531524188066324, + "tokens_seen": 988676096 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003537713139418255, + "loss": 1.3544, + "theoretical_loss": 3.6531293260872806, + "tokens_seen": 988741632 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035376128385155464, + "loss": 1.4011, + "theoretical_loss": 3.653106235327061, + "tokens_seen": 988807168 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035375125376128387, + "loss": 1.4417, + "theoretical_loss": 3.6530831465256774, + "tokens_seen": 988872704 + }, + { + "epoch": 0.3, + "learning_rate": 0.000353741223671013, + "loss": 1.4638, + "theoretical_loss": 3.653060059682834, + "tokens_seen": 988938240 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035373119358074223, + "loss": 1.4539, + "theoretical_loss": 3.6530369747982347, + "tokens_seen": 989003776 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003537211634904714, + "loss": 1.2331, + "theoretical_loss": 3.653013891871584, + "tokens_seen": 989069312 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003537111334002006, + "loss": 1.4019, + "theoretical_loss": 3.652990810902586, + "tokens_seen": 989134848 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003537011033099298, + "loss": 1.3646, + "theoretical_loss": 3.6529677318909446, + "tokens_seen": 989200384 + }, + { + "epoch": 0.3, + "learning_rate": 0.000353691073219659, + "loss": 1.5059, + "theoretical_loss": 3.6529446548363644, + "tokens_seen": 989265920 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035368104312938814, + "loss": 1.3608, + "theoretical_loss": 3.6529215797385506, + "tokens_seen": 989331456 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003536710130391174, + "loss": 1.4232, + "theoretical_loss": 3.6528985065972064, + "tokens_seen": 989396992 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003536609829488465, + "loss": 1.3548, + "theoretical_loss": 3.6528754354120365, + "tokens_seen": 989462528 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035365095285857574, + "loss": 1.2955, + "theoretical_loss": 3.6528523661827466, + "tokens_seen": 989528064 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -0.0012925436021760106, + "objective/train/docs_used": 567574, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.975478172302246, + "objective/train/original_loss": 2.975478172302246, + "objective/train/theoretical_loss": 3.65282929890904, + "objective/train/tokens_used": 1010053600, + "objective/train/value_avg": -0.006175994873046875, + "objective/train/value_loss": 0.00010050573473563418, + "objective/train/value_max": -0.0004107952117919922, + "objective/train/value_min": -0.06243896484375, + "objective/train/value_reward_corr": 0.4076764029208356, + "objective/train/value_std": 0.005069732666015625, + "objective/train/weight_avg": 0.9917318820953369, + "objective/train/weighted_lm_loss": 2.9536073207855225, + "objective/train/weights_max": 1.6598756313323975, + "objective/train/weights_min": 0.37896835803985596, + "theoretical_loss": 3.65282929890904, + "tokens_seen": 989593600 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003536409227683049, + "loss": 1.4082, + "theoretical_loss": 3.65282929890904, + "tokens_seen": 989593600 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003536308926780341, + "loss": 1.4262, + "theoretical_loss": 3.6528062335906224, + "tokens_seen": 989659136 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003536208625877633, + "loss": 1.4275, + "theoretical_loss": 3.652783170227198, + "tokens_seen": 989724672 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035361083249749246, + "loss": 1.351, + "theoretical_loss": 3.6527601088184722, + "tokens_seen": 989790208 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003536008024072217, + "loss": 1.3696, + "theoretical_loss": 3.6527370493641493, + "tokens_seen": 989855744 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003535907723169509, + "loss": 1.3804, + "theoretical_loss": 3.6527139918639353, + "tokens_seen": 989921280 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035358074222668006, + "loss": 1.3716, + "theoretical_loss": 3.6526909363175344, + "tokens_seen": 989986816 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035357071213640924, + "loss": 1.3712, + "theoretical_loss": 3.6526678827246513, + "tokens_seen": 990052352 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003535606820461384, + "loss": 1.3616, + "theoretical_loss": 3.6526448310849924, + "tokens_seen": 990117888 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003535506519558676, + "loss": 1.2775, + "theoretical_loss": 3.6526217813982624, + "tokens_seen": 990183424 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035354062186559684, + "loss": 1.4456, + "theoretical_loss": 3.652598733664166, + "tokens_seen": 990248960 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035353059177532597, + "loss": 1.3166, + "theoretical_loss": 3.6525756878824094, + "tokens_seen": 990314496 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003535205616850552, + "loss": 1.3251, + "theoretical_loss": 3.652552644052698, + "tokens_seen": 990380032 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003535105315947844, + "loss": 1.2802, + "theoretical_loss": 3.6525296021747367, + "tokens_seen": 990445568 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035350050150451356, + "loss": 1.4766, + "theoretical_loss": 3.652506562248232, + "tokens_seen": 990511104 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035349047141424274, + "loss": 1.4657, + "theoretical_loss": 3.6524835242728884, + "tokens_seen": 990576640 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003534804413239719, + "loss": 1.3218, + "theoretical_loss": 3.652460488248412, + "tokens_seen": 990642176 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003534704112337011, + "loss": 1.3213, + "theoretical_loss": 3.652437454174509, + "tokens_seen": 990707712 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035346038114343034, + "loss": 1.498, + "theoretical_loss": 3.6524144220508847, + "tokens_seen": 990773248 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035345035105315947, + "loss": 1.3287, + "theoretical_loss": 3.6523913918772455, + "tokens_seen": 990838784 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003534403209628887, + "loss": 1.4336, + "theoretical_loss": 3.652368363653297, + "tokens_seen": 990904320 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035343029087261783, + "loss": 1.3868, + "theoretical_loss": 3.652345337378745, + "tokens_seen": 990969856 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035342026078234707, + "loss": 1.4189, + "theoretical_loss": 3.652322313053296, + "tokens_seen": 991035392 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035341023069207625, + "loss": 1.4722, + "theoretical_loss": 3.6522992906766563, + "tokens_seen": 991100928 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035340020060180543, + "loss": 1.2052, + "theoretical_loss": 3.6522762702485316, + "tokens_seen": 991166464 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": 0.00010838128946488723, + "objective/train/docs_used": 568180, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.370918035507202, + "objective/train/original_loss": 2.370917797088623, + "objective/train/theoretical_loss": 3.652253251768628, + "objective/train/tokens_used": 1011692000, + "objective/train/value_avg": -0.004756927490234375, + "objective/train/value_loss": 6.808080797782168e-05, + "objective/train/value_max": -0.0002551078796386719, + "objective/train/value_min": -0.06561279296875, + "objective/train/value_reward_corr": 0.2798324034568481, + "objective/train/value_std": 0.003662109375, + "objective/train/weight_avg": 1.0040992498397827, + "objective/train/weighted_lm_loss": 2.377704620361328, + "objective/train/weights_max": 1.6268349885940552, + "objective/train/weights_min": 0.3051759600639343, + "theoretical_loss": 3.652253251768628, + "tokens_seen": 991232000 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003533901705115346, + "loss": 1.3368, + "theoretical_loss": 3.652253251768628, + "tokens_seen": 991232000 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035338014042126384, + "loss": 1.32, + "theoretical_loss": 3.6522302352366527, + "tokens_seen": 991297536 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035337011033099297, + "loss": 1.3587, + "theoretical_loss": 3.6522072206523113, + "tokens_seen": 991363072 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003533600802407222, + "loss": 1.4502, + "theoretical_loss": 3.6521842080153104, + "tokens_seen": 991428608 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035335005015045133, + "loss": 1.3703, + "theoretical_loss": 3.652161197325357, + "tokens_seen": 991494144 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035334002006018057, + "loss": 1.269, + "theoretical_loss": 3.652138188582157, + "tokens_seen": 991559680 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035332998996990975, + "loss": 1.3905, + "theoretical_loss": 3.652115181785417, + "tokens_seen": 991625216 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035331995987963893, + "loss": 1.394, + "theoretical_loss": 3.652092176934845, + "tokens_seen": 991690752 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003533099297893681, + "loss": 1.4819, + "theoretical_loss": 3.6520691740301463, + "tokens_seen": 991756288 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003532998996990973, + "loss": 1.3411, + "theoretical_loss": 3.6520461730710285, + "tokens_seen": 991821824 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003532898696088265, + "loss": 1.4763, + "theoretical_loss": 3.6520231740571982, + "tokens_seen": 991887360 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003532798395185557, + "loss": 1.2607, + "theoretical_loss": 3.6520001769883628, + "tokens_seen": 991952896 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035326980942828484, + "loss": 1.3957, + "theoretical_loss": 3.651977181864229, + "tokens_seen": 992018432 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035325977933801407, + "loss": 1.3728, + "theoretical_loss": 3.6519541886845035, + "tokens_seen": 992083968 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003532497492477432, + "loss": 1.3801, + "theoretical_loss": 3.651931197448894, + "tokens_seen": 992149504 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035323971915747243, + "loss": 1.358, + "theoretical_loss": 3.6519082081571073, + "tokens_seen": 992215040 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003532296890672016, + "loss": 1.3625, + "theoretical_loss": 3.6518852208088513, + "tokens_seen": 992280576 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003532196589769308, + "loss": 1.3431, + "theoretical_loss": 3.651862235403833, + "tokens_seen": 992346112 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035320962888666, + "loss": 1.328, + "theoretical_loss": 3.6518392519417597, + "tokens_seen": 992411648 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003531995987963892, + "loss": 1.3497, + "theoretical_loss": 3.6518162704223394, + "tokens_seen": 992477184 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035318956870611834, + "loss": 1.4082, + "theoretical_loss": 3.6517932908452786, + "tokens_seen": 992542720 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003531795386158476, + "loss": 1.3794, + "theoretical_loss": 3.651770313210286, + "tokens_seen": 992608256 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003531695085255767, + "loss": 1.458, + "theoretical_loss": 3.651747337517069, + "tokens_seen": 992673792 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035315947843530594, + "loss": 1.3493, + "theoretical_loss": 3.651724363765334, + "tokens_seen": 992739328 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003531494483450351, + "loss": 1.3475, + "theoretical_loss": 3.6517013919547905, + "tokens_seen": 992804864 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -0.000853446894325316, + "objective/train/docs_used": 569645, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.004215717315674, + "objective/train/original_loss": 3.004216194152832, + "objective/train/theoretical_loss": 3.651678422085146, + "objective/train/tokens_used": 1013330400, + "objective/train/value_avg": -0.0059661865234375, + "objective/train/value_loss": 9.533327829558402e-05, + "objective/train/value_max": -0.00042557716369628906, + "objective/train/value_min": -0.06756591796875, + "objective/train/value_reward_corr": 0.3636739054777828, + "objective/train/value_std": 0.0048828125, + "objective/train/weight_avg": 0.9957310557365417, + "objective/train/weighted_lm_loss": 3.00654935836792, + "objective/train/weights_max": 1.9653279781341553, + "objective/train/weights_min": 0.3025447428226471, + "theoretical_loss": 3.651678422085146, + "tokens_seen": 992870400 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003531394182547643, + "loss": 1.4217, + "theoretical_loss": 3.651678422085146, + "tokens_seen": 992870400 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003531293881644935, + "loss": 1.3702, + "theoretical_loss": 3.651655454156108, + "tokens_seen": 992935936 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035311935807422266, + "loss": 1.4953, + "theoretical_loss": 3.651632488167385, + "tokens_seen": 993001472 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035310932798395184, + "loss": 1.4, + "theoretical_loss": 3.6516095241186846, + "tokens_seen": 993067008 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003530992978936811, + "loss": 1.2958, + "theoretical_loss": 3.6515865620097148, + "tokens_seen": 993132544 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003530892678034102, + "loss": 1.3964, + "theoretical_loss": 3.6515636018401842, + "tokens_seen": 993198080 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035307923771313944, + "loss": 1.367, + "theoretical_loss": 3.6515406436098004, + "tokens_seen": 993263616 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035306920762286857, + "loss": 1.4019, + "theoretical_loss": 3.6515176873182726, + "tokens_seen": 993329152 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003530591775325978, + "loss": 1.3817, + "theoretical_loss": 3.651494732965308, + "tokens_seen": 993394688 + }, + { + "epoch": 0.3, + "learning_rate": 0.000353049147442327, + "loss": 1.3202, + "theoretical_loss": 3.651471780550616, + "tokens_seen": 993460224 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035303911735205617, + "loss": 1.4803, + "theoretical_loss": 3.651448830073905, + "tokens_seen": 993525760 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035302908726178535, + "loss": 1.3214, + "theoretical_loss": 3.651425881534883, + "tokens_seen": 993591296 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003530190571715146, + "loss": 1.4539, + "theoretical_loss": 3.6514029349332597, + "tokens_seen": 993656832 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003530090270812437, + "loss": 1.3896, + "theoretical_loss": 3.651379990268742, + "tokens_seen": 993722368 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035299899699097294, + "loss": 1.4804, + "theoretical_loss": 3.65135704754104, + "tokens_seen": 993787904 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035298896690070207, + "loss": 1.4655, + "theoretical_loss": 3.651334106749862, + "tokens_seen": 993853440 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003529789368104313, + "loss": 1.4202, + "theoretical_loss": 3.6513111678949173, + "tokens_seen": 993918976 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003529689067201605, + "loss": 1.4475, + "theoretical_loss": 3.651288230975915, + "tokens_seen": 993984512 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035295887662988967, + "loss": 1.4132, + "theoretical_loss": 3.651265295992563, + "tokens_seen": 994050048 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035294884653961885, + "loss": 1.4213, + "theoretical_loss": 3.6512423629445707, + "tokens_seen": 994115584 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035293881644934803, + "loss": 1.3654, + "theoretical_loss": 3.651219431831648, + "tokens_seen": 994181120 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003529287863590772, + "loss": 1.3532, + "theoretical_loss": 3.6511965026535034, + "tokens_seen": 994246656 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035291875626880645, + "loss": 1.4472, + "theoretical_loss": 3.651173575409846, + "tokens_seen": 994312192 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003529087261785356, + "loss": 1.492, + "theoretical_loss": 3.651150650100386, + "tokens_seen": 994377728 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003528986960882648, + "loss": 1.441, + "theoretical_loss": 3.6511277267248317, + "tokens_seen": 994443264 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": 0.00038005883106961846, + "objective/train/docs_used": 570064, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7726399898529053, + "objective/train/original_loss": 2.7726402282714844, + "objective/train/theoretical_loss": 3.651104805282893, + "objective/train/tokens_used": 1014968800, + "objective/train/value_avg": -0.007175445556640625, + "objective/train/value_loss": 6.52793823974207e-05, + "objective/train/value_max": -0.0004076957702636719, + "objective/train/value_min": -0.059967041015625, + "objective/train/value_reward_corr": 0.45314374137271446, + "objective/train/value_std": 0.00592803955078125, + "objective/train/weight_avg": 1.0070066452026367, + "objective/train/weighted_lm_loss": 2.7899298667907715, + "objective/train/weights_max": 1.8215184211730957, + "objective/train/weights_min": 0.5843630433082581, + "theoretical_loss": 3.651104805282893, + "tokens_seen": 994508800 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035288866599799394, + "loss": 1.4187, + "theoretical_loss": 3.651104805282893, + "tokens_seen": 994508800 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035287863590772317, + "loss": 1.4867, + "theoretical_loss": 3.6510818857742797, + "tokens_seen": 994574336 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035286860581745235, + "loss": 1.4842, + "theoretical_loss": 3.651058968198701, + "tokens_seen": 994639872 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035285857572718153, + "loss": 1.398, + "theoretical_loss": 3.6510360525558667, + "tokens_seen": 994705408 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035284854563691077, + "loss": 1.4048, + "theoretical_loss": 3.651013138845486, + "tokens_seen": 994770944 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035283851554663995, + "loss": 1.2746, + "theoretical_loss": 3.650990227067269, + "tokens_seen": 994836480 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035282848545636913, + "loss": 1.4347, + "theoretical_loss": 3.6509673172209256, + "tokens_seen": 994902016 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003528184553660983, + "loss": 1.4244, + "theoretical_loss": 3.650944409306166, + "tokens_seen": 994967552 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003528084252758275, + "loss": 1.4003, + "theoretical_loss": 3.6509215033226994, + "tokens_seen": 995033088 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003527983951855567, + "loss": 1.3845, + "theoretical_loss": 3.650898599270236, + "tokens_seen": 995098624 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003527883650952859, + "loss": 1.3122, + "theoretical_loss": 3.6508756971484857, + "tokens_seen": 995164160 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035277833500501504, + "loss": 1.4641, + "theoretical_loss": 3.6508527969571594, + "tokens_seen": 995229696 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035276830491474427, + "loss": 1.2769, + "theoretical_loss": 3.6508298986959664, + "tokens_seen": 995295232 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003527582748244734, + "loss": 1.3979, + "theoretical_loss": 3.6508070023646173, + "tokens_seen": 995360768 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035274824473420263, + "loss": 1.3741, + "theoretical_loss": 3.6507841079628225, + "tokens_seen": 995426304 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003527382146439318, + "loss": 1.3128, + "theoretical_loss": 3.6507612154902924, + "tokens_seen": 995491840 + }, + { + "epoch": 0.3, + "learning_rate": 0.000352728184553661, + "loss": 1.3332, + "theoretical_loss": 3.650738324946737, + "tokens_seen": 995557376 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003527181544633902, + "loss": 1.3173, + "theoretical_loss": 3.6507154363318675, + "tokens_seen": 995622912 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003527081243731194, + "loss": 1.3238, + "theoretical_loss": 3.6506925496453935, + "tokens_seen": 995688448 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035269809428284854, + "loss": 1.4234, + "theoretical_loss": 3.650669664887027, + "tokens_seen": 995753984 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003526880641925778, + "loss": 1.4268, + "theoretical_loss": 3.650646782056477, + "tokens_seen": 995819520 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003526780341023069, + "loss": 1.3971, + "theoretical_loss": 3.6506239011534554, + "tokens_seen": 995885056 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035266800401203614, + "loss": 1.3897, + "theoretical_loss": 3.650601022177673, + "tokens_seen": 995950592 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003526579739217653, + "loss": 1.3814, + "theoretical_loss": 3.65057814512884, + "tokens_seen": 996016128 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003526479438314945, + "loss": 1.3425, + "theoretical_loss": 3.650555270006668, + "tokens_seen": 996081664 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -0.0007047561230137944, + "objective/train/docs_used": 571551, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.974527359008789, + "objective/train/original_loss": 2.974527359008789, + "objective/train/theoretical_loss": 3.6505323968108674, + "objective/train/tokens_used": 1016607200, + "objective/train/value_avg": -0.00507354736328125, + "objective/train/value_loss": 6.681554077658802e-05, + "objective/train/value_max": -0.0005054473876953125, + "objective/train/value_min": -0.073974609375, + "objective/train/value_reward_corr": 0.351609927087696, + "objective/train/value_std": 0.004344940185546875, + "objective/train/weight_avg": 0.9961098432540894, + "objective/train/weighted_lm_loss": 2.9670310020446777, + "objective/train/weights_max": 1.8901444673538208, + "objective/train/weights_min": 0.49736276268959045, + "theoretical_loss": 3.6505323968108674, + "tokens_seen": 996147200 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003526379137412237, + "loss": 1.3634, + "theoretical_loss": 3.6505323968108674, + "tokens_seen": 996147200 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035262788365095286, + "loss": 1.3589, + "theoretical_loss": 3.6505095255411497, + "tokens_seen": 996212736 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035261785356068204, + "loss": 1.2939, + "theoretical_loss": 3.650486656197226, + "tokens_seen": 996278272 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003526078234704113, + "loss": 1.3707, + "theoretical_loss": 3.650463788778807, + "tokens_seen": 996343808 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003525977933801404, + "loss": 1.359, + "theoretical_loss": 3.650440923285605, + "tokens_seen": 996409344 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035258776328986964, + "loss": 1.4222, + "theoretical_loss": 3.65041805971733, + "tokens_seen": 996474880 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035257773319959877, + "loss": 1.4593, + "theoretical_loss": 3.650395198073694, + "tokens_seen": 996540416 + }, + { + "epoch": 0.3, + "learning_rate": 0.000352567703109328, + "loss": 1.2842, + "theoretical_loss": 3.6503723383544084, + "tokens_seen": 996605952 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003525576730190572, + "loss": 1.3294, + "theoretical_loss": 3.6503494805591847, + "tokens_seen": 996671488 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035254764292878637, + "loss": 1.4045, + "theoretical_loss": 3.650326624687735, + "tokens_seen": 996737024 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035253761283851555, + "loss": 1.3057, + "theoretical_loss": 3.65030377073977, + "tokens_seen": 996802560 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003525275827482448, + "loss": 1.4237, + "theoretical_loss": 3.6502809187150014, + "tokens_seen": 996868096 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003525175526579739, + "loss": 1.4672, + "theoretical_loss": 3.650258068613142, + "tokens_seen": 996933632 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035250752256770314, + "loss": 1.3609, + "theoretical_loss": 3.650235220433902, + "tokens_seen": 996999168 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035249749247743227, + "loss": 1.3697, + "theoretical_loss": 3.650212374176995, + "tokens_seen": 997064704 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003524874623871615, + "loss": 1.3788, + "theoretical_loss": 3.6501895298421316, + "tokens_seen": 997130240 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003524774322968907, + "loss": 1.2953, + "theoretical_loss": 3.6501666874290244, + "tokens_seen": 997195776 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035246740220661987, + "loss": 1.3408, + "theoretical_loss": 3.6501438469373855, + "tokens_seen": 997261312 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035245737211634905, + "loss": 1.4601, + "theoretical_loss": 3.6501210083669267, + "tokens_seen": 997326848 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035244734202607823, + "loss": 1.4213, + "theoretical_loss": 3.65009817171736, + "tokens_seen": 997392384 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003524373119358074, + "loss": 1.4444, + "theoretical_loss": 3.6500753369883983, + "tokens_seen": 997457920 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035242728184553665, + "loss": 1.4622, + "theoretical_loss": 3.6500525041797536, + "tokens_seen": 997523456 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003524172517552658, + "loss": 1.5121, + "theoretical_loss": 3.6500296732911384, + "tokens_seen": 997588992 + }, + { + "epoch": 0.3, + "learning_rate": 0.000352407221664995, + "loss": 1.4395, + "theoretical_loss": 3.650006844322265, + "tokens_seen": 997654528 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035239719157472414, + "loss": 1.3855, + "theoretical_loss": 3.6499840172728453, + "tokens_seen": 997720064 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": 5.951366802037228e-06, + "objective/train/docs_used": 572250, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1679322719573975, + "objective/train/original_loss": 3.1679325103759766, + "objective/train/theoretical_loss": 3.6499611921425927, + "objective/train/tokens_used": 1018245600, + "objective/train/value_avg": -0.005092620849609375, + "objective/train/value_loss": 5.8849964261753485e-05, + "objective/train/value_max": -0.00043392181396484375, + "objective/train/value_min": -0.10430908203125, + "objective/train/value_reward_corr": 0.40211552472541806, + "objective/train/value_std": 0.004177093505859375, + "objective/train/weight_avg": 1.0028456449508667, + "objective/train/weighted_lm_loss": 3.1842455863952637, + "objective/train/weights_max": 2.5070929527282715, + "objective/train/weights_min": 0.39224252104759216, + "theoretical_loss": 3.6499611921425927, + "tokens_seen": 997785600 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035238716148445337, + "loss": 1.4184, + "theoretical_loss": 3.6499611921425927, + "tokens_seen": 997785600 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035237713139418255, + "loss": 1.35, + "theoretical_loss": 3.6499383689312195, + "tokens_seen": 997851136 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035236710130391173, + "loss": 1.3836, + "theoretical_loss": 3.6499155476384377, + "tokens_seen": 997916672 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003523570712136409, + "loss": 1.4245, + "theoretical_loss": 3.6498927282639615, + "tokens_seen": 997982208 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035234704112337015, + "loss": 1.4339, + "theoretical_loss": 3.649869910807503, + "tokens_seen": 998047744 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003523370110330993, + "loss": 1.4373, + "theoretical_loss": 3.6498470952687745, + "tokens_seen": 998113280 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003523269809428285, + "loss": 1.2725, + "theoretical_loss": 3.649824281647489, + "tokens_seen": 998178816 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035231695085255764, + "loss": 1.3011, + "theoretical_loss": 3.6498014699433603, + "tokens_seen": 998244352 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003523069207622869, + "loss": 1.3991, + "theoretical_loss": 3.649778660156101, + "tokens_seen": 998309888 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035229689067201606, + "loss": 1.3557, + "theoretical_loss": 3.6497558522854243, + "tokens_seen": 998375424 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035228686058174524, + "loss": 1.4099, + "theoretical_loss": 3.6497330463310433, + "tokens_seen": 998440960 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003522768304914744, + "loss": 1.3368, + "theoretical_loss": 3.649710242292671, + "tokens_seen": 998506496 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003522668004012036, + "loss": 1.3784, + "theoretical_loss": 3.6496874401700206, + "tokens_seen": 998572032 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003522567703109328, + "loss": 1.2121, + "theoretical_loss": 3.649664639962806, + "tokens_seen": 998637568 + }, + { + "epoch": 0.3, + "learning_rate": 0.000352246740220662, + "loss": 1.4299, + "theoretical_loss": 3.64964184167074, + "tokens_seen": 998703104 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035223671013039114, + "loss": 1.3112, + "theoretical_loss": 3.6496190452935373, + "tokens_seen": 998768640 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003522266800401204, + "loss": 1.3983, + "theoretical_loss": 3.6495962508309097, + "tokens_seen": 998834176 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035221664994984956, + "loss": 1.3773, + "theoretical_loss": 3.649573458282572, + "tokens_seen": 998899712 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035220661985957874, + "loss": 1.3342, + "theoretical_loss": 3.6495506676482377, + "tokens_seen": 998965248 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003521965897693079, + "loss": 1.4155, + "theoretical_loss": 3.64952787892762, + "tokens_seen": 999030784 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003521865596790371, + "loss": 1.2503, + "theoretical_loss": 3.649505092120433, + "tokens_seen": 999096320 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003521765295887663, + "loss": 1.3925, + "theoretical_loss": 3.649482307226391, + "tokens_seen": 999161856 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003521664994984955, + "loss": 1.3453, + "theoretical_loss": 3.6494595242452066, + "tokens_seen": 999227392 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035215646940822465, + "loss": 1.3253, + "theoretical_loss": 3.6494367431765955, + "tokens_seen": 999292928 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003521464393179539, + "loss": 1.3162, + "theoretical_loss": 3.6494139640202707, + "tokens_seen": 999358464 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -5.4896431720408145e-06, + "objective/train/docs_used": 573370, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0865938663482666, + "objective/train/original_loss": 3.0865938663482666, + "objective/train/theoretical_loss": 3.6493911867759463, + "objective/train/tokens_used": 1019884000, + "objective/train/value_avg": -0.00623321533203125, + "objective/train/value_loss": 0.00010268687037751079, + "objective/train/value_max": -0.0003275871276855469, + "objective/train/value_min": -0.1358642578125, + "objective/train/value_reward_corr": 0.5151425473091755, + "objective/train/value_std": 0.005527496337890625, + "objective/train/weight_avg": 1.0043493509292603, + "objective/train/weighted_lm_loss": 3.112570285797119, + "objective/train/weights_max": 1.7472434043884277, + "objective/train/weights_min": 0.2049255520105362, + "theoretical_loss": 3.6493911867759463, + "tokens_seen": 999424000 + }, + { + "epoch": 0.3, + "learning_rate": 0.000352136409227683, + "loss": 1.3991, + "theoretical_loss": 3.6493911867759463, + "tokens_seen": 999424000 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035212637913741224, + "loss": 1.2522, + "theoretical_loss": 3.6493684114433362, + "tokens_seen": 999489536 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003521163490471414, + "loss": 1.2995, + "theoretical_loss": 3.6493456380221554, + "tokens_seen": 999555072 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003521063189568706, + "loss": 1.4733, + "theoretical_loss": 3.649322866512118, + "tokens_seen": 999620608 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035209628886659984, + "loss": 1.323, + "theoretical_loss": 3.649300096912938, + "tokens_seen": 999686144 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035208625877632897, + "loss": 1.427, + "theoretical_loss": 3.64927732922433, + "tokens_seen": 999751680 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003520762286860582, + "loss": 1.3951, + "theoretical_loss": 3.649254563446008, + "tokens_seen": 999817216 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003520661985957874, + "loss": 1.3696, + "theoretical_loss": 3.649231799577687, + "tokens_seen": 999882752 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035205616850551657, + "loss": 1.4288, + "theoretical_loss": 3.649209037619082, + "tokens_seen": 999948288 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035204613841524575, + "loss": 1.4604, + "theoretical_loss": 3.649186277569907, + "tokens_seen": 1000013824 + }, + { + "epoch": 0.3, + "learning_rate": 0.000352036108324975, + "loss": 1.3727, + "theoretical_loss": 3.6491635194298766, + "tokens_seen": 1000079360 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003520260782347041, + "loss": 1.4089, + "theoretical_loss": 3.6491407631987056, + "tokens_seen": 1000144896 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035201604814443334, + "loss": 1.3615, + "theoretical_loss": 3.6491180088761093, + "tokens_seen": 1000210432 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035200601805416247, + "loss": 1.499, + "theoretical_loss": 3.649095256461802, + "tokens_seen": 1000275968 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003519959879638917, + "loss": 1.3075, + "theoretical_loss": 3.6490725059554996, + "tokens_seen": 1000341504 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003519859578736209, + "loss": 1.3545, + "theoretical_loss": 3.6490497573569165, + "tokens_seen": 1000407040 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035197592778335007, + "loss": 1.4287, + "theoretical_loss": 3.649027010665767, + "tokens_seen": 1000472576 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035196589769307925, + "loss": 1.2896, + "theoretical_loss": 3.6490042658817674, + "tokens_seen": 1000538112 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035195586760280843, + "loss": 1.4552, + "theoretical_loss": 3.6489815230046325, + "tokens_seen": 1000603648 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003519458375125376, + "loss": 1.3951, + "theoretical_loss": 3.6489587820340774, + "tokens_seen": 1000669184 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035193580742226685, + "loss": 1.4694, + "theoretical_loss": 3.648936042969818, + "tokens_seen": 1000734720 + }, + { + "epoch": 0.3, + "learning_rate": 0.000351925777331996, + "loss": 1.5346, + "theoretical_loss": 3.6489133058115684, + "tokens_seen": 1000800256 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003519157472417252, + "loss": 1.1777, + "theoretical_loss": 3.6488905705590455, + "tokens_seen": 1000865792 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035190571715145434, + "loss": 1.4351, + "theoretical_loss": 3.6488678372119643, + "tokens_seen": 1000931328 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035189568706118357, + "loss": 1.4219, + "theoretical_loss": 3.64884510577004, + "tokens_seen": 1000996864 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -6.655182369286194e-05, + "objective/train/docs_used": 574030, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.0209197998046875, + "objective/train/original_loss": 2.0209195613861084, + "objective/train/theoretical_loss": 3.6488223762329888, + "objective/train/tokens_used": 1021522400, + "objective/train/value_avg": -0.004795074462890625, + "objective/train/value_loss": 4.693028313340619e-05, + "objective/train/value_max": -0.0003936290740966797, + "objective/train/value_min": -0.049591064453125, + "objective/train/value_reward_corr": 0.41335945406387237, + "objective/train/value_std": 0.0041656494140625, + "objective/train/weight_avg": 1.0014867782592773, + "objective/train/weighted_lm_loss": 2.0184810161590576, + "objective/train/weights_max": 1.6195976734161377, + "objective/train/weights_min": 0.2548407018184662, + "theoretical_loss": 3.6488223762329888, + "tokens_seen": 1001062400 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035188565697091275, + "loss": 1.359, + "theoretical_loss": 3.6488223762329888, + "tokens_seen": 1001062400 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035187562688064193, + "loss": 1.284, + "theoretical_loss": 3.6487996486005256, + "tokens_seen": 1001127936 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003518655967903711, + "loss": 1.3599, + "theoretical_loss": 3.6487769228723668, + "tokens_seen": 1001193472 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035185556670010035, + "loss": 1.3939, + "theoretical_loss": 3.648754199048228, + "tokens_seen": 1001259008 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003518455366098295, + "loss": 1.4115, + "theoretical_loss": 3.6487314771278245, + "tokens_seen": 1001324544 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003518355065195587, + "loss": 1.4363, + "theoretical_loss": 3.648708757110873, + "tokens_seen": 1001390080 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035182547642928784, + "loss": 1.4501, + "theoretical_loss": 3.6486860389970897, + "tokens_seen": 1001455616 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003518154463390171, + "loss": 1.3339, + "theoretical_loss": 3.6486633227861907, + "tokens_seen": 1001521152 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035180541624874626, + "loss": 1.4901, + "theoretical_loss": 3.6486406084778906, + "tokens_seen": 1001586688 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035179538615847544, + "loss": 1.4995, + "theoretical_loss": 3.648617896071907, + "tokens_seen": 1001652224 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003517853560682046, + "loss": 1.3038, + "theoretical_loss": 3.648595185567956, + "tokens_seen": 1001717760 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003517753259779338, + "loss": 1.3319, + "theoretical_loss": 3.648572476965753, + "tokens_seen": 1001783296 + }, + { + "epoch": 0.3, + "learning_rate": 0.000351765295887663, + "loss": 1.4576, + "theoretical_loss": 3.6485497702650154, + "tokens_seen": 1001848832 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003517552657973922, + "loss": 1.5221, + "theoretical_loss": 3.6485270654654594, + "tokens_seen": 1001914368 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035174523570712134, + "loss": 1.3787, + "theoretical_loss": 3.6485043625668014, + "tokens_seen": 1001979904 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003517352056168506, + "loss": 1.3898, + "theoretical_loss": 3.6484816615687574, + "tokens_seen": 1002045440 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035172517552657976, + "loss": 1.4165, + "theoretical_loss": 3.6484589624710444, + "tokens_seen": 1002110976 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035171514543630894, + "loss": 1.3482, + "theoretical_loss": 3.648436265273379, + "tokens_seen": 1002176512 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003517051153460381, + "loss": 1.377, + "theoretical_loss": 3.6484135699754785, + "tokens_seen": 1002242048 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003516950852557673, + "loss": 1.4843, + "theoretical_loss": 3.648390876577059, + "tokens_seen": 1002307584 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003516850551654965, + "loss": 1.4128, + "theoretical_loss": 3.648368185077837, + "tokens_seen": 1002373120 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003516750250752257, + "loss": 1.3606, + "theoretical_loss": 3.6483454954775305, + "tokens_seen": 1002438656 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035166499498495485, + "loss": 1.4327, + "theoretical_loss": 3.648322807775856, + "tokens_seen": 1002504192 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003516549648946841, + "loss": 1.385, + "theoretical_loss": 3.6483001219725297, + "tokens_seen": 1002569728 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003516449348044132, + "loss": 1.5463, + "theoretical_loss": 3.64827743806727, + "tokens_seen": 1002635264 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": 5.226217399467714e-05, + "objective/train/docs_used": 574442, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.20947265625, + "objective/train/original_loss": 3.209472894668579, + "objective/train/theoretical_loss": 3.648254756059793, + "objective/train/tokens_used": 1023160800, + "objective/train/value_avg": -0.005001068115234375, + "objective/train/value_loss": 5.020542812417261e-05, + "objective/train/value_max": -0.00035691261291503906, + "objective/train/value_min": -0.08135986328125, + "objective/train/value_reward_corr": 0.4301537266657938, + "objective/train/value_std": 0.00421905517578125, + "objective/train/weight_avg": 1.0028067827224731, + "objective/train/weighted_lm_loss": 3.2298080921173096, + "objective/train/weights_max": 2.151827096939087, + "objective/train/weights_min": 0.13764700293540955, + "theoretical_loss": 3.648254756059793, + "tokens_seen": 1002700800 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035163490471414244, + "loss": 1.451, + "theoretical_loss": 3.648254756059793, + "tokens_seen": 1002700800 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003516248746238716, + "loss": 1.3565, + "theoretical_loss": 3.648232075949817, + "tokens_seen": 1002766336 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003516148445336008, + "loss": 1.3233, + "theoretical_loss": 3.648209397737058, + "tokens_seen": 1002831872 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035160481444333, + "loss": 1.4941, + "theoretical_loss": 3.6481867214212342, + "tokens_seen": 1002897408 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035159478435305917, + "loss": 1.3912, + "theoretical_loss": 3.648164047002062, + "tokens_seen": 1002962944 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035158475426278835, + "loss": 1.3812, + "theoretical_loss": 3.648141374479261, + "tokens_seen": 1003028480 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003515747241725176, + "loss": 1.3288, + "theoretical_loss": 3.648118703852546, + "tokens_seen": 1003094016 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003515646940822467, + "loss": 1.4687, + "theoretical_loss": 3.648096035121637, + "tokens_seen": 1003159552 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035155466399197595, + "loss": 1.3992, + "theoretical_loss": 3.6480733682862496, + "tokens_seen": 1003225088 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035154463390170513, + "loss": 1.4834, + "theoretical_loss": 3.6480507033461023, + "tokens_seen": 1003290624 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003515346038114343, + "loss": 1.4489, + "theoretical_loss": 3.648028040300913, + "tokens_seen": 1003356160 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003515245737211635, + "loss": 1.4206, + "theoretical_loss": 3.6480053791504, + "tokens_seen": 1003421696 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035151454363089267, + "loss": 1.3627, + "theoretical_loss": 3.64798271989428, + "tokens_seen": 1003487232 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035150451354062185, + "loss": 1.3577, + "theoretical_loss": 3.6479600625322717, + "tokens_seen": 1003552768 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003514944834503511, + "loss": 1.5247, + "theoretical_loss": 3.647937407064093, + "tokens_seen": 1003618304 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003514844533600802, + "loss": 1.4, + "theoretical_loss": 3.6479147534894616, + "tokens_seen": 1003683840 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035147442326980945, + "loss": 1.4673, + "theoretical_loss": 3.647892101808096, + "tokens_seen": 1003749376 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003514643931795386, + "loss": 1.41, + "theoretical_loss": 3.647869452019714, + "tokens_seen": 1003814912 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003514543630892678, + "loss": 1.4034, + "theoretical_loss": 3.647846804124035, + "tokens_seen": 1003880448 + }, + { + "epoch": 0.3, + "learning_rate": 0.000351444332998997, + "loss": 1.3799, + "theoretical_loss": 3.6478241581207755, + "tokens_seen": 1003945984 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003514343029087262, + "loss": 1.3277, + "theoretical_loss": 3.647801514009655, + "tokens_seen": 1004011520 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035142427281845536, + "loss": 1.2836, + "theoretical_loss": 3.6477788717903916, + "tokens_seen": 1004077056 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035141424272818454, + "loss": 1.3162, + "theoretical_loss": 3.647756231462704, + "tokens_seen": 1004142592 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003514042126379137, + "loss": 1.4218, + "theoretical_loss": 3.6477335930263104, + "tokens_seen": 1004208128 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035139418254764295, + "loss": 1.4296, + "theoretical_loss": 3.6477109564809296, + "tokens_seen": 1004273664 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -0.0009402118739672005, + "objective/train/docs_used": 575683, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.835296154022217, + "objective/train/original_loss": 2.835296154022217, + "objective/train/theoretical_loss": 3.64768832182628, + "objective/train/tokens_used": 1024799200, + "objective/train/value_avg": -0.005931854248046875, + "objective/train/value_loss": 5.932616113568656e-05, + "objective/train/value_max": -0.00046181678771972656, + "objective/train/value_min": -0.06756591796875, + "objective/train/value_reward_corr": 0.4625448531824251, + "objective/train/value_std": 0.00518035888671875, + "objective/train/weight_avg": 0.9933973550796509, + "objective/train/weighted_lm_loss": 2.8253793716430664, + "objective/train/weights_max": 1.6195976734161377, + "objective/train/weights_min": 0.2622670531272888, + "theoretical_loss": 3.64768832182628, + "tokens_seen": 1004339200 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003513841524573721, + "loss": 1.4127, + "theoretical_loss": 3.64768832182628, + "tokens_seen": 1004339200 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003513741223671013, + "loss": 1.3694, + "theoretical_loss": 3.6476656890620807, + "tokens_seen": 1004404736 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003513640922768305, + "loss": 1.3036, + "theoretical_loss": 3.6476430581880503, + "tokens_seen": 1004470272 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003513540621865597, + "loss": 1.296, + "theoretical_loss": 3.647620429203908, + "tokens_seen": 1004535808 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003513440320962889, + "loss": 1.3816, + "theoretical_loss": 3.6475978021093716, + "tokens_seen": 1004601344 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035133400200601804, + "loss": 1.266, + "theoretical_loss": 3.6475751769041613, + "tokens_seen": 1004666880 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003513239719157473, + "loss": 1.2823, + "theoretical_loss": 3.647552553587995, + "tokens_seen": 1004732416 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035131394182547646, + "loss": 1.392, + "theoretical_loss": 3.6475299321605927, + "tokens_seen": 1004797952 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035130391173520564, + "loss": 1.2306, + "theoretical_loss": 3.6475073126216735, + "tokens_seen": 1004863488 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003512938816449348, + "loss": 1.3506, + "theoretical_loss": 3.6474846949709563, + "tokens_seen": 1004929024 + }, + { + "epoch": 0.3, + "learning_rate": 0.000351283851554664, + "loss": 1.3799, + "theoretical_loss": 3.64746207920816, + "tokens_seen": 1004994560 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003512738214643932, + "loss": 1.5005, + "theoretical_loss": 3.6474394653330044, + "tokens_seen": 1005060096 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003512637913741224, + "loss": 1.4801, + "theoretical_loss": 3.647416853345209, + "tokens_seen": 1005125632 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035125376128385154, + "loss": 1.3872, + "theoretical_loss": 3.647394243244493, + "tokens_seen": 1005191168 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003512437311935808, + "loss": 1.3604, + "theoretical_loss": 3.6473716350305754, + "tokens_seen": 1005256704 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035123370110330996, + "loss": 1.3288, + "theoretical_loss": 3.6473490287031765, + "tokens_seen": 1005322240 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035122367101303914, + "loss": 1.4433, + "theoretical_loss": 3.6473264242620154, + "tokens_seen": 1005387776 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003512136409227683, + "loss": 1.392, + "theoretical_loss": 3.6473038217068128, + "tokens_seen": 1005453312 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003512036108324975, + "loss": 1.4446, + "theoretical_loss": 3.647281221037287, + "tokens_seen": 1005518848 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003511935807422267, + "loss": 1.4137, + "theoretical_loss": 3.6472586222531587, + "tokens_seen": 1005584384 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003511835506519559, + "loss": 1.4093, + "theoretical_loss": 3.6472360253541476, + "tokens_seen": 1005649920 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035117352056168505, + "loss": 1.3531, + "theoretical_loss": 3.6472134303399733, + "tokens_seen": 1005715456 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003511634904714143, + "loss": 1.3988, + "theoretical_loss": 3.6471908372103563, + "tokens_seen": 1005780992 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003511534603811434, + "loss": 1.4426, + "theoretical_loss": 3.647168245965016, + "tokens_seen": 1005846528 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035114343029087264, + "loss": 1.4584, + "theoretical_loss": 3.6471456566036737, + "tokens_seen": 1005912064 + }, + { + "epoch": 0.3, + "objective/train/advantage_avg": -0.0010220212861895561, + "objective/train/docs_used": 576356, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9948570728302, + "objective/train/original_loss": 2.994856834411621, + "objective/train/theoretical_loss": 3.6471230691260477, + "objective/train/tokens_used": 1026437600, + "objective/train/value_avg": -0.004535675048828125, + "objective/train/value_loss": 5.783267624792643e-05, + "objective/train/value_max": -0.0002913475036621094, + "objective/train/value_min": -0.05340576171875, + "objective/train/value_reward_corr": 0.37190452777230787, + "objective/train/value_std": 0.0035686492919921875, + "objective/train/weight_avg": 0.9924459457397461, + "objective/train/weighted_lm_loss": 2.9787847995758057, + "objective/train/weights_max": 1.7058398723602295, + "objective/train/weights_min": 0.3365126848220825, + "theoretical_loss": 3.6471230691260477, + "tokens_seen": 1005977600 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003511334002006018, + "loss": 1.5307, + "theoretical_loss": 3.6471230691260477, + "tokens_seen": 1005977600 + }, + { + "epoch": 0.3, + "learning_rate": 0.000351123370110331, + "loss": 1.4304, + "theoretical_loss": 3.64710048353186, + "tokens_seen": 1006043136 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003511133400200602, + "loss": 1.4106, + "theoretical_loss": 3.6470778998208297, + "tokens_seen": 1006108672 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035110330992978937, + "loss": 1.3964, + "theoretical_loss": 3.6470553179926775, + "tokens_seen": 1006174208 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035109327983951855, + "loss": 1.3567, + "theoretical_loss": 3.647032738047124, + "tokens_seen": 1006239744 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003510832497492478, + "loss": 1.316, + "theoretical_loss": 3.647010159983889, + "tokens_seen": 1006305280 + }, + { + "epoch": 0.3, + "learning_rate": 0.0003510732196589769, + "loss": 1.3746, + "theoretical_loss": 3.646987583802694, + "tokens_seen": 1006370816 + }, + { + "epoch": 0.3, + "learning_rate": 0.00035106318956870615, + "loss": 1.3689, + "theoretical_loss": 3.646965009503259, + "tokens_seen": 1006436352 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035105315947843533, + "loss": 1.3282, + "theoretical_loss": 3.6469424370853045, + "tokens_seen": 1006501888 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003510431293881645, + "loss": 1.4633, + "theoretical_loss": 3.6469198665485516, + "tokens_seen": 1006567424 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003510330992978937, + "loss": 1.4766, + "theoretical_loss": 3.6468972978927208, + "tokens_seen": 1006632960 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035102306920762287, + "loss": 1.3792, + "theoretical_loss": 3.646874731117533, + "tokens_seen": 1006698496 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035101303911735205, + "loss": 1.3705, + "theoretical_loss": 3.646852166222709, + "tokens_seen": 1006764032 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003510030090270813, + "loss": 1.435, + "theoretical_loss": 3.6468296032079697, + "tokens_seen": 1006829568 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003509929789368104, + "loss": 1.3809, + "theoretical_loss": 3.646807042073036, + "tokens_seen": 1006895104 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035098294884653965, + "loss": 1.3536, + "theoretical_loss": 3.6467844828176297, + "tokens_seen": 1006960640 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003509729187562688, + "loss": 1.4111, + "theoretical_loss": 3.6467619254414707, + "tokens_seen": 1007026176 + }, + { + "epoch": 0.31, + "learning_rate": 0.000350962888665998, + "loss": 1.418, + "theoretical_loss": 3.6467393699442807, + "tokens_seen": 1007091712 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003509528585757272, + "loss": 1.359, + "theoretical_loss": 3.6467168163257817, + "tokens_seen": 1007157248 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003509428284854564, + "loss": 1.3777, + "theoretical_loss": 3.646694264585694, + "tokens_seen": 1007222784 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035093279839518556, + "loss": 1.3775, + "theoretical_loss": 3.646671714723739, + "tokens_seen": 1007288320 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035092276830491474, + "loss": 1.4034, + "theoretical_loss": 3.6466491667396386, + "tokens_seen": 1007353856 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003509127382146439, + "loss": 1.4185, + "theoretical_loss": 3.6466266206331137, + "tokens_seen": 1007419392 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035090270812437315, + "loss": 1.4421, + "theoretical_loss": 3.6466040764038867, + "tokens_seen": 1007484928 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003508926780341023, + "loss": 1.3119, + "theoretical_loss": 3.6465815340516783, + "tokens_seen": 1007550464 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": 0.0012201805366203189, + "objective/train/docs_used": 577861, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2334036827087402, + "objective/train/original_loss": 3.2334041595458984, + "objective/train/theoretical_loss": 3.6465589935762104, + "objective/train/tokens_used": 1028076000, + "objective/train/value_avg": -0.005405426025390625, + "objective/train/value_loss": 4.9823313020169735e-05, + "objective/train/value_max": -0.0004305839538574219, + "objective/train/value_min": -0.06939697265625, + "objective/train/value_reward_corr": 0.49602924986508634, + "objective/train/value_std": 0.00482177734375, + "objective/train/weight_avg": 1.0146238803863525, + "objective/train/weighted_lm_loss": 3.2606916427612305, + "objective/train/weights_max": 1.4447705745697021, + "objective/train/weights_min": 0.4854995012283325, + "theoretical_loss": 3.6465589935762104, + "tokens_seen": 1007616000 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003508826479438315, + "loss": 1.4381, + "theoretical_loss": 3.6465589935762104, + "tokens_seen": 1007616000 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003508726178535607, + "loss": 1.4204, + "theoretical_loss": 3.646536454977205, + "tokens_seen": 1007681536 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003508625877632899, + "loss": 1.3947, + "theoretical_loss": 3.6465139182543833, + "tokens_seen": 1007747072 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035085255767301906, + "loss": 1.271, + "theoretical_loss": 3.6464913834074677, + "tokens_seen": 1007812608 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035084252758274824, + "loss": 1.457, + "theoretical_loss": 3.64646885043618, + "tokens_seen": 1007878144 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003508324974924774, + "loss": 1.2481, + "theoretical_loss": 3.646446319340242, + "tokens_seen": 1007943680 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035082246740220666, + "loss": 1.373, + "theoretical_loss": 3.6464237901193757, + "tokens_seen": 1008009216 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003508124373119358, + "loss": 1.3206, + "theoretical_loss": 3.6464012627733027, + "tokens_seen": 1008074752 + }, + { + "epoch": 0.31, + "learning_rate": 0.000350802407221665, + "loss": 1.4713, + "theoretical_loss": 3.646378737301746, + "tokens_seen": 1008140288 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035079237713139415, + "loss": 1.2949, + "theoretical_loss": 3.6463562137044274, + "tokens_seen": 1008205824 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003507823470411234, + "loss": 1.3637, + "theoretical_loss": 3.646333691981069, + "tokens_seen": 1008271360 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035077231695085256, + "loss": 1.3775, + "theoretical_loss": 3.6463111721313934, + "tokens_seen": 1008336896 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035076228686058174, + "loss": 1.3754, + "theoretical_loss": 3.6462886541551223, + "tokens_seen": 1008402432 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003507522567703109, + "loss": 1.3977, + "theoretical_loss": 3.646266138051979, + "tokens_seen": 1008467968 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035074222668004016, + "loss": 1.4253, + "theoretical_loss": 3.646243623821685, + "tokens_seen": 1008533504 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003507321965897693, + "loss": 1.4584, + "theoretical_loss": 3.646221111463964, + "tokens_seen": 1008599040 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003507221664994985, + "loss": 1.3256, + "theoretical_loss": 3.646198600978538, + "tokens_seen": 1008664576 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035071213640922765, + "loss": 1.4276, + "theoretical_loss": 3.6461760923651294, + "tokens_seen": 1008730112 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003507021063189569, + "loss": 1.4348, + "theoretical_loss": 3.646153585623461, + "tokens_seen": 1008795648 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035069207622868607, + "loss": 1.5347, + "theoretical_loss": 3.646131080753255, + "tokens_seen": 1008861184 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035068204613841525, + "loss": 1.2859, + "theoretical_loss": 3.6461085777542364, + "tokens_seen": 1008926720 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035067201604814443, + "loss": 1.4279, + "theoretical_loss": 3.646086076626126, + "tokens_seen": 1008992256 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003506619859578736, + "loss": 1.3196, + "theoretical_loss": 3.6460635773686474, + "tokens_seen": 1009057792 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003506519558676028, + "loss": 1.6097, + "theoretical_loss": 3.6460410799815235, + "tokens_seen": 1009123328 + }, + { + "epoch": 0.31, + "learning_rate": 0.000350641925777332, + "loss": 1.2722, + "theoretical_loss": 3.6460185844644775, + "tokens_seen": 1009188864 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.0014629128854721785, + "objective/train/docs_used": 578450, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9684720039367676, + "objective/train/original_loss": 2.968472480773926, + "objective/train/theoretical_loss": 3.645996090817232, + "objective/train/tokens_used": 1029714400, + "objective/train/value_avg": -0.004627227783203125, + "objective/train/value_loss": 0.00010175445640925318, + "objective/train/value_max": -0.00034737586975097656, + "objective/train/value_min": -0.06903076171875, + "objective/train/value_reward_corr": 0.3305466744346759, + "objective/train/value_std": 0.0036296844482421875, + "objective/train/weight_avg": 0.9897840619087219, + "objective/train/weighted_lm_loss": 2.945239543914795, + "objective/train/weights_max": 1.8596299886703491, + "objective/train/weights_min": 0.13791637122631073, + "theoretical_loss": 3.645996090817232, + "tokens_seen": 1009254400 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035063189568706115, + "loss": 1.3686, + "theoretical_loss": 3.645996090817232, + "tokens_seen": 1009254400 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003506218655967904, + "loss": 1.4178, + "theoretical_loss": 3.645973599039511, + "tokens_seen": 1009319936 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003506118355065195, + "loss": 1.3229, + "theoretical_loss": 3.6459511091310377, + "tokens_seen": 1009385472 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035060180541624875, + "loss": 1.3665, + "theoretical_loss": 3.645928621091535, + "tokens_seen": 1009451008 + }, + { + "epoch": 0.31, + "learning_rate": 0.000350591775325978, + "loss": 1.4111, + "theoretical_loss": 3.6459061349207262, + "tokens_seen": 1009516544 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003505817452357071, + "loss": 1.4096, + "theoretical_loss": 3.6458836506183347, + "tokens_seen": 1009582080 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035057171514543635, + "loss": 1.3961, + "theoretical_loss": 3.6458611681840845, + "tokens_seen": 1009647616 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035056168505516553, + "loss": 1.3342, + "theoretical_loss": 3.6458386876176987, + "tokens_seen": 1009713152 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003505516549648947, + "loss": 1.4332, + "theoretical_loss": 3.645816208918901, + "tokens_seen": 1009778688 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003505416248746239, + "loss": 1.3359, + "theoretical_loss": 3.645793732087415, + "tokens_seen": 1009844224 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035053159478435307, + "loss": 1.3773, + "theoretical_loss": 3.6457712571229646, + "tokens_seen": 1009909760 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035052156469408225, + "loss": 1.4366, + "theoretical_loss": 3.6457487840252734, + "tokens_seen": 1009975296 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003505115346038115, + "loss": 1.3579, + "theoretical_loss": 3.645726312794065, + "tokens_seen": 1010040832 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003505015045135406, + "loss": 1.3958, + "theoretical_loss": 3.6457038434290636, + "tokens_seen": 1010106368 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035049147442326985, + "loss": 1.3701, + "theoretical_loss": 3.6456813759299935, + "tokens_seen": 1010171904 + }, + { + "epoch": 0.31, + "learning_rate": 0.000350481444332999, + "loss": 1.3955, + "theoretical_loss": 3.6456589102965777, + "tokens_seen": 1010237440 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003504714142427282, + "loss": 1.2812, + "theoretical_loss": 3.645636446528542, + "tokens_seen": 1010302976 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003504613841524574, + "loss": 1.4411, + "theoretical_loss": 3.6456139846256077, + "tokens_seen": 1010368512 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003504513540621866, + "loss": 1.3249, + "theoretical_loss": 3.645591524587502, + "tokens_seen": 1010434048 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035044132397191576, + "loss": 1.2693, + "theoretical_loss": 3.6455690664139473, + "tokens_seen": 1010499584 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035043129388164494, + "loss": 1.3944, + "theoretical_loss": 3.6455466101046685, + "tokens_seen": 1010565120 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003504212637913741, + "loss": 1.3437, + "theoretical_loss": 3.6455241556593903, + "tokens_seen": 1010630656 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035041123370110335, + "loss": 1.4208, + "theoretical_loss": 3.6455017030778363, + "tokens_seen": 1010696192 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003504012036108325, + "loss": 1.4422, + "theoretical_loss": 3.6454792523597312, + "tokens_seen": 1010761728 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003503911735205617, + "loss": 1.2938, + "theoretical_loss": 3.6454568035048003, + "tokens_seen": 1010827264 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.00020359725749585778, + "objective/train/docs_used": 579792, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.795928478240967, + "objective/train/original_loss": 2.7959280014038086, + "objective/train/theoretical_loss": 3.645434356512767, + "objective/train/tokens_used": 1031352800, + "objective/train/value_avg": -0.005451202392578125, + "objective/train/value_loss": 6.0810125432908535e-05, + "objective/train/value_max": -0.00037550926208496094, + "objective/train/value_min": -0.058441162109375, + "objective/train/value_reward_corr": 0.3799238206831055, + "objective/train/value_std": 0.004703521728515625, + "objective/train/weight_avg": 1.0007236003875732, + "objective/train/weighted_lm_loss": 2.809145927429199, + "objective/train/weights_max": 1.7474656105041504, + "objective/train/weights_min": 0.20259372889995575, + "theoretical_loss": 3.645434356512767, + "tokens_seen": 1010892800 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003503811434302909, + "loss": 1.2868, + "theoretical_loss": 3.645434356512767, + "tokens_seen": 1010892800 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003503711133400201, + "loss": 1.5654, + "theoretical_loss": 3.645411911383357, + "tokens_seen": 1010958336 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035036108324974926, + "loss": 1.2983, + "theoretical_loss": 3.6453894681162944, + "tokens_seen": 1011023872 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035035105315947844, + "loss": 1.3935, + "theoretical_loss": 3.6453670267113045, + "tokens_seen": 1011089408 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003503410230692076, + "loss": 1.3436, + "theoretical_loss": 3.645344587168111, + "tokens_seen": 1011154944 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035033099297893686, + "loss": 1.2926, + "theoretical_loss": 3.6453221494864403, + "tokens_seen": 1011220480 + }, + { + "epoch": 0.31, + "learning_rate": 0.000350320962888666, + "loss": 1.261, + "theoretical_loss": 3.6452997136660166, + "tokens_seen": 1011286016 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003503109327983952, + "loss": 1.3804, + "theoretical_loss": 3.6452772797065642, + "tokens_seen": 1011351552 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035030090270812435, + "loss": 1.4257, + "theoretical_loss": 3.6452548476078093, + "tokens_seen": 1011417088 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003502908726178536, + "loss": 1.3693, + "theoretical_loss": 3.645232417369477, + "tokens_seen": 1011482624 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035028084252758276, + "loss": 1.3501, + "theoretical_loss": 3.6452099889912914, + "tokens_seen": 1011548160 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035027081243731194, + "loss": 1.2493, + "theoretical_loss": 3.645187562472979, + "tokens_seen": 1011613696 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003502607823470411, + "loss": 1.2464, + "theoretical_loss": 3.6451651378142644, + "tokens_seen": 1011679232 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035025075225677036, + "loss": 1.2206, + "theoretical_loss": 3.645142715014873, + "tokens_seen": 1011744768 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003502407221664995, + "loss": 1.2396, + "theoretical_loss": 3.6451202940745295, + "tokens_seen": 1011810304 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003502306920762287, + "loss": 1.298, + "theoretical_loss": 3.645097874992961, + "tokens_seen": 1011875840 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035022066198595785, + "loss": 1.3594, + "theoretical_loss": 3.645075457769892, + "tokens_seen": 1011941376 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003502106318956871, + "loss": 1.4854, + "theoretical_loss": 3.6450530424050482, + "tokens_seen": 1012006912 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035020060180541627, + "loss": 1.4881, + "theoretical_loss": 3.645030628898155, + "tokens_seen": 1012072448 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035019057171514545, + "loss": 1.3053, + "theoretical_loss": 3.6450082172489395, + "tokens_seen": 1012137984 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035018054162487463, + "loss": 1.5237, + "theoretical_loss": 3.6449858074571253, + "tokens_seen": 1012203520 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003501705115346038, + "loss": 1.388, + "theoretical_loss": 3.6449633995224398, + "tokens_seen": 1012269056 + }, + { + "epoch": 0.31, + "learning_rate": 0.000350160481444333, + "loss": 1.3687, + "theoretical_loss": 3.644940993444608, + "tokens_seen": 1012334592 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003501504513540622, + "loss": 1.3722, + "theoretical_loss": 3.6449185892233564, + "tokens_seen": 1012400128 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035014042126379135, + "loss": 1.311, + "theoretical_loss": 3.6448961868584107, + "tokens_seen": 1012465664 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.001665561692789197, + "objective/train/docs_used": 580179, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.40946888923645, + "objective/train/original_loss": 2.40946888923645, + "objective/train/theoretical_loss": 3.644873786349497, + "objective/train/tokens_used": 1032991200, + "objective/train/value_avg": -0.004642486572265625, + "objective/train/value_loss": 8.672828698763624e-05, + "objective/train/value_max": -0.00028014183044433594, + "objective/train/value_min": -0.053497314453125, + "objective/train/value_reward_corr": 0.41631854993446904, + "objective/train/value_std": 0.0035877227783203125, + "objective/train/weight_avg": 0.9872369170188904, + "objective/train/weighted_lm_loss": 2.39235258102417, + "objective/train/weights_max": 1.5542280673980713, + "objective/train/weights_min": 0.37932270765304565, + "theoretical_loss": 3.644873786349497, + "tokens_seen": 1012531200 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003501303911735206, + "loss": 1.2928, + "theoretical_loss": 3.644873786349497, + "tokens_seen": 1012531200 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003501203610832497, + "loss": 1.4547, + "theoretical_loss": 3.6448513876963418, + "tokens_seen": 1012596736 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035011033099297895, + "loss": 1.4461, + "theoretical_loss": 3.644828990898671, + "tokens_seen": 1012662272 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035010030090270813, + "loss": 1.3564, + "theoretical_loss": 3.6448065959562106, + "tokens_seen": 1012727808 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003500902708124373, + "loss": 1.4664, + "theoretical_loss": 3.644784202868687, + "tokens_seen": 1012793344 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003500802407221665, + "loss": 1.377, + "theoretical_loss": 3.644761811635827, + "tokens_seen": 1012858880 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035007021063189573, + "loss": 1.3962, + "theoretical_loss": 3.6447394222573557, + "tokens_seen": 1012924416 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035006018054162486, + "loss": 1.3442, + "theoretical_loss": 3.6447170347330013, + "tokens_seen": 1012989952 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003500501504513541, + "loss": 1.4169, + "theoretical_loss": 3.644694649062489, + "tokens_seen": 1013055488 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003500401203610832, + "loss": 1.3973, + "theoretical_loss": 3.6446722652455463, + "tokens_seen": 1013121024 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035003009027081245, + "loss": 1.3362, + "theoretical_loss": 3.644649883281899, + "tokens_seen": 1013186560 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035002006018054163, + "loss": 1.4695, + "theoretical_loss": 3.6446275031712743, + "tokens_seen": 1013252096 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003500100300902708, + "loss": 1.4277, + "theoretical_loss": 3.6446051249133986, + "tokens_seen": 1013317632 + }, + { + "epoch": 0.31, + "learning_rate": 0.00035, + "loss": 1.4735, + "theoretical_loss": 3.6445827485079993, + "tokens_seen": 1013383168 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003499899699097292, + "loss": 1.213, + "theoretical_loss": 3.6445603739548025, + "tokens_seen": 1013448704 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034997993981945836, + "loss": 1.3235, + "theoretical_loss": 3.6445380012535358, + "tokens_seen": 1013514240 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003499699097291876, + "loss": 1.3447, + "theoretical_loss": 3.644515630403926, + "tokens_seen": 1013579776 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003499598796389167, + "loss": 1.3077, + "theoretical_loss": 3.6444932614057, + "tokens_seen": 1013645312 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034994984954864596, + "loss": 1.4101, + "theoretical_loss": 3.644470894258585, + "tokens_seen": 1013710848 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003499398194583751, + "loss": 1.2428, + "theoretical_loss": 3.644448528962308, + "tokens_seen": 1013776384 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003499297893681043, + "loss": 1.3772, + "theoretical_loss": 3.644426165516596, + "tokens_seen": 1013841920 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003499197592778335, + "loss": 1.3341, + "theoretical_loss": 3.6444038039211772, + "tokens_seen": 1013907456 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003499097291875627, + "loss": 1.3434, + "theoretical_loss": 3.644381444175778, + "tokens_seen": 1013972992 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034989969909729186, + "loss": 1.3136, + "theoretical_loss": 3.644359086280126, + "tokens_seen": 1014038528 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003498896690070211, + "loss": 1.3548, + "theoretical_loss": 3.644336730233949, + "tokens_seen": 1014104064 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.0009648163104429841, + "objective/train/docs_used": 581599, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.100292921066284, + "objective/train/original_loss": 3.100292921066284, + "objective/train/theoretical_loss": 3.6443143760369736, + "objective/train/tokens_used": 1034629600, + "objective/train/value_avg": -0.00537872314453125, + "objective/train/value_loss": 6.84471451677382e-05, + "objective/train/value_max": -0.0005316734313964844, + "objective/train/value_min": -0.044921875, + "objective/train/value_reward_corr": 0.45537894731885276, + "objective/train/value_std": 0.00439453125, + "objective/train/weight_avg": 0.9936003088951111, + "objective/train/weighted_lm_loss": 3.07995867729187, + "objective/train/weights_max": 1.405328392982483, + "objective/train/weights_min": 0.4716514050960541, + "theoretical_loss": 3.6443143760369736, + "tokens_seen": 1014169600 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003498796389167502, + "loss": 1.563, + "theoretical_loss": 3.6443143760369736, + "tokens_seen": 1014169600 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034986960882647946, + "loss": 1.3199, + "theoretical_loss": 3.644292023688929, + "tokens_seen": 1014235136 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003498595787362086, + "loss": 1.3742, + "theoretical_loss": 3.644269673189541, + "tokens_seen": 1014300672 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003498495486459378, + "loss": 1.2936, + "theoretical_loss": 3.6442473245385387, + "tokens_seen": 1014366208 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034983951855566706, + "loss": 1.3657, + "theoretical_loss": 3.6442249777356492, + "tokens_seen": 1014431744 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003498294884653962, + "loss": 1.3193, + "theoretical_loss": 3.6442026327806003, + "tokens_seen": 1014497280 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003498194583751254, + "loss": 1.473, + "theoretical_loss": 3.6441802896731197, + "tokens_seen": 1014562816 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034980942828485455, + "loss": 1.4424, + "theoretical_loss": 3.644157948412935, + "tokens_seen": 1014628352 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003497993981945838, + "loss": 1.3067, + "theoretical_loss": 3.6441356089997754, + "tokens_seen": 1014693888 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034978936810431296, + "loss": 1.3317, + "theoretical_loss": 3.644113271433368, + "tokens_seen": 1014759424 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034977933801404214, + "loss": 1.397, + "theoretical_loss": 3.6440909357134412, + "tokens_seen": 1014824960 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003497693079237713, + "loss": 1.3839, + "theoretical_loss": 3.6440686018397224, + "tokens_seen": 1014890496 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034975927783350056, + "loss": 1.3171, + "theoretical_loss": 3.644046269811941, + "tokens_seen": 1014956032 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003497492477432297, + "loss": 1.378, + "theoretical_loss": 3.6440239396298244, + "tokens_seen": 1015021568 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003497392176529589, + "loss": 1.3633, + "theoretical_loss": 3.644001611293101, + "tokens_seen": 1015087104 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034972918756268805, + "loss": 1.3968, + "theoretical_loss": 3.6439792848014996, + "tokens_seen": 1015152640 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003497191574724173, + "loss": 1.3902, + "theoretical_loss": 3.643956960154748, + "tokens_seen": 1015218176 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034970912738214647, + "loss": 1.3694, + "theoretical_loss": 3.643934637352575, + "tokens_seen": 1015283712 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034969909729187565, + "loss": 1.3061, + "theoretical_loss": 3.643912316394709, + "tokens_seen": 1015349248 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034968906720160483, + "loss": 1.444, + "theoretical_loss": 3.6438899972808794, + "tokens_seen": 1015414784 + }, + { + "epoch": 0.31, + "learning_rate": 0.000349679037111334, + "loss": 1.4243, + "theoretical_loss": 3.6438676800108136, + "tokens_seen": 1015480320 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003496690070210632, + "loss": 1.4177, + "theoretical_loss": 3.6438453645842412, + "tokens_seen": 1015545856 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003496589769307924, + "loss": 1.3871, + "theoretical_loss": 3.643823051000891, + "tokens_seen": 1015611392 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034964894684052155, + "loss": 1.2191, + "theoretical_loss": 3.6438007392604908, + "tokens_seen": 1015676928 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003496389167502508, + "loss": 1.3416, + "theoretical_loss": 3.643778429362771, + "tokens_seen": 1015742464 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.001337905414402485, + "objective/train/docs_used": 582330, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.686654806137085, + "objective/train/original_loss": 2.686655044555664, + "objective/train/theoretical_loss": 3.643756121307459, + "objective/train/tokens_used": 1036268000, + "objective/train/value_avg": -0.006336212158203125, + "objective/train/value_loss": 0.00020728791423607618, + "objective/train/value_max": -0.0004973411560058594, + "objective/train/value_min": -0.10504150390625, + "objective/train/value_reward_corr": 0.5390673555429519, + "objective/train/value_std": 0.006404876708984375, + "objective/train/weight_avg": 0.9946704506874084, + "objective/train/weighted_lm_loss": 2.6803314685821533, + "objective/train/weights_max": 2.705996036529541, + "objective/train/weights_min": 0.18056190013885498, + "theoretical_loss": 3.643756121307459, + "tokens_seen": 1015808000 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003496288866599799, + "loss": 1.2932, + "theoretical_loss": 3.643756121307459, + "tokens_seen": 1015808000 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034961885656970915, + "loss": 1.32, + "theoretical_loss": 3.643733815094285, + "tokens_seen": 1015873536 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034960882647943833, + "loss": 1.3531, + "theoretical_loss": 3.6437115107229774, + "tokens_seen": 1015939072 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003495987963891675, + "loss": 1.3439, + "theoretical_loss": 3.6436892081932655, + "tokens_seen": 1016004608 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003495887662988967, + "loss": 1.3738, + "theoretical_loss": 3.643666907504879, + "tokens_seen": 1016070144 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034957873620862593, + "loss": 1.3646, + "theoretical_loss": 3.643644608657546, + "tokens_seen": 1016135680 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034956870611835506, + "loss": 1.4172, + "theoretical_loss": 3.6436223116509963, + "tokens_seen": 1016201216 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003495586760280843, + "loss": 1.3371, + "theoretical_loss": 3.6436000164849593, + "tokens_seen": 1016266752 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003495486459378134, + "loss": 1.4116, + "theoretical_loss": 3.643577723159165, + "tokens_seen": 1016332288 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034953861584754265, + "loss": 1.403, + "theoretical_loss": 3.643555431673342, + "tokens_seen": 1016397824 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034952858575727183, + "loss": 1.2597, + "theoretical_loss": 3.64353314202722, + "tokens_seen": 1016463360 + }, + { + "epoch": 0.31, + "learning_rate": 0.000349518555667001, + "loss": 1.3746, + "theoretical_loss": 3.643510854220529, + "tokens_seen": 1016528896 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003495085255767302, + "loss": 1.4367, + "theoretical_loss": 3.643488568252998, + "tokens_seen": 1016594432 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003494984954864594, + "loss": 1.2761, + "theoretical_loss": 3.643466284124357, + "tokens_seen": 1016659968 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034948846539618856, + "loss": 1.3614, + "theoretical_loss": 3.643444001834336, + "tokens_seen": 1016725504 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003494784353059178, + "loss": 1.3213, + "theoretical_loss": 3.6434217213826647, + "tokens_seen": 1016791040 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003494684052156469, + "loss": 1.2721, + "theoretical_loss": 3.643399442769072, + "tokens_seen": 1016856576 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034945837512537616, + "loss": 1.2302, + "theoretical_loss": 3.6433771659932894, + "tokens_seen": 1016922112 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003494483450351053, + "loss": 1.3966, + "theoretical_loss": 3.6433548910550457, + "tokens_seen": 1016987648 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003494383149448345, + "loss": 1.3599, + "theoretical_loss": 3.643332617954071, + "tokens_seen": 1017053184 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003494282848545637, + "loss": 1.2698, + "theoretical_loss": 3.6433103466900962, + "tokens_seen": 1017118720 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003494182547642929, + "loss": 1.4327, + "theoretical_loss": 3.643288077262851, + "tokens_seen": 1017184256 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034940822467402206, + "loss": 1.3583, + "theoretical_loss": 3.6432658096720654, + "tokens_seen": 1017249792 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003493981945837513, + "loss": 1.4116, + "theoretical_loss": 3.6432435439174693, + "tokens_seen": 1017315328 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003493881644934804, + "loss": 1.3844, + "theoretical_loss": 3.6432212799987944, + "tokens_seen": 1017380864 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.0018575182184576988, + "objective/train/docs_used": 582938, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7531487941741943, + "objective/train/original_loss": 2.7531485557556152, + "objective/train/theoretical_loss": 3.6431990179157694, + "objective/train/tokens_used": 1037906400, + "objective/train/value_avg": -0.004856109619140625, + "objective/train/value_loss": 8.790977153694257e-05, + "objective/train/value_max": -0.0002199411392211914, + "objective/train/value_min": -0.053009033203125, + "objective/train/value_reward_corr": 0.27200957702125356, + "objective/train/value_std": 0.00341033935546875, + "objective/train/weight_avg": 0.9854031205177307, + "objective/train/weighted_lm_loss": 2.7199509143829346, + "objective/train/weights_max": 1.6990858316421509, + "objective/train/weights_min": 0.19681203365325928, + "theoretical_loss": 3.6431990179157694, + "tokens_seen": 1017446400 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034937813440320966, + "loss": 1.4751, + "theoretical_loss": 3.6431990179157694, + "tokens_seen": 1017446400 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003493681043129388, + "loss": 1.4239, + "theoretical_loss": 3.643176757668126, + "tokens_seen": 1017511936 + }, + { + "epoch": 0.31, + "learning_rate": 0.000349358074222668, + "loss": 1.3646, + "theoretical_loss": 3.643154499255594, + "tokens_seen": 1017577472 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003493480441323972, + "loss": 1.3816, + "theoretical_loss": 3.643132242677904, + "tokens_seen": 1017643008 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003493380140421264, + "loss": 1.4946, + "theoretical_loss": 3.643109987934787, + "tokens_seen": 1017708544 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034932798395185557, + "loss": 1.39, + "theoretical_loss": 3.643087735025973, + "tokens_seen": 1017774080 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034931795386158475, + "loss": 1.481, + "theoretical_loss": 3.6430654839511933, + "tokens_seen": 1017839616 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034930792377131393, + "loss": 1.2837, + "theoretical_loss": 3.6430432347101784, + "tokens_seen": 1017905152 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034929789368104316, + "loss": 1.4483, + "theoretical_loss": 3.6430209873026596, + "tokens_seen": 1017970688 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003492878635907723, + "loss": 1.3464, + "theoretical_loss": 3.642998741728367, + "tokens_seen": 1018036224 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003492778335005015, + "loss": 1.4145, + "theoretical_loss": 3.642976497987032, + "tokens_seen": 1018101760 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034926780341023065, + "loss": 1.3244, + "theoretical_loss": 3.6429542560783856, + "tokens_seen": 1018167296 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003492577733199599, + "loss": 1.3764, + "theoretical_loss": 3.6429320160021588, + "tokens_seen": 1018232832 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034924774322968907, + "loss": 1.3528, + "theoretical_loss": 3.6429097777580823, + "tokens_seen": 1018298368 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034923771313941825, + "loss": 1.3175, + "theoretical_loss": 3.6428875413458877, + "tokens_seen": 1018363904 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034922768304914743, + "loss": 1.3621, + "theoretical_loss": 3.642865306765306, + "tokens_seen": 1018429440 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034921765295887667, + "loss": 1.3676, + "theoretical_loss": 3.6428430740160693, + "tokens_seen": 1018494976 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003492076228686058, + "loss": 1.3081, + "theoretical_loss": 3.642820843097908, + "tokens_seen": 1018560512 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034919759277833503, + "loss": 1.4856, + "theoretical_loss": 3.6427986140105535, + "tokens_seen": 1018626048 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034918756268806416, + "loss": 1.3356, + "theoretical_loss": 3.642776386753738, + "tokens_seen": 1018691584 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003491775325977934, + "loss": 1.4345, + "theoretical_loss": 3.642754161327192, + "tokens_seen": 1018757120 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034916750250752257, + "loss": 1.3743, + "theoretical_loss": 3.642731937730648, + "tokens_seen": 1018822656 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034915747241725175, + "loss": 1.3766, + "theoretical_loss": 3.642709715963837, + "tokens_seen": 1018888192 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034914744232698093, + "loss": 1.3633, + "theoretical_loss": 3.642687496026491, + "tokens_seen": 1018953728 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003491374122367101, + "loss": 1.3054, + "theoretical_loss": 3.6426652779183417, + "tokens_seen": 1019019264 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -3.8479320210171863e-05, + "objective/train/docs_used": 584109, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.683405637741089, + "objective/train/original_loss": 2.683405637741089, + "objective/train/theoretical_loss": 3.642643061639121, + "objective/train/tokens_used": 1039544800, + "objective/train/value_avg": -0.00513458251953125, + "objective/train/value_loss": 4.572399484459311e-05, + "objective/train/value_max": -0.00030303001403808594, + "objective/train/value_min": -0.1485595703125, + "objective/train/value_reward_corr": 0.286891275207097, + "objective/train/value_std": 0.00395965576171875, + "objective/train/weight_avg": 1.0018113851547241, + "objective/train/weighted_lm_loss": 2.7092931270599365, + "objective/train/weights_max": 2.494696617126465, + "objective/train/weights_min": 0.3444810211658478, + "theoretical_loss": 3.642643061639121, + "tokens_seen": 1019084800 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003491273821464393, + "loss": 1.39, + "theoretical_loss": 3.642643061639121, + "tokens_seen": 1019084800 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034911735205616853, + "loss": 1.4, + "theoretical_loss": 3.64262084718856, + "tokens_seen": 1019150336 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034910732196589766, + "loss": 1.3008, + "theoretical_loss": 3.6425986345663914, + "tokens_seen": 1019215872 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003490972918756269, + "loss": 1.2961, + "theoretical_loss": 3.642576423772347, + "tokens_seen": 1019281408 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034908726178535613, + "loss": 1.3202, + "theoretical_loss": 3.642554214806159, + "tokens_seen": 1019346944 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034907723169508526, + "loss": 1.2963, + "theoretical_loss": 3.6425320076675587, + "tokens_seen": 1019412480 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003490672016048145, + "loss": 1.3489, + "theoretical_loss": 3.6425098023562787, + "tokens_seen": 1019478016 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003490571715145436, + "loss": 1.3857, + "theoretical_loss": 3.6424875988720515, + "tokens_seen": 1019543552 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034904714142427285, + "loss": 1.4735, + "theoretical_loss": 3.6424653972146093, + "tokens_seen": 1019609088 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034903711133400203, + "loss": 1.2004, + "theoretical_loss": 3.6424431973836837, + "tokens_seen": 1019674624 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003490270812437312, + "loss": 1.3489, + "theoretical_loss": 3.642420999379008, + "tokens_seen": 1019740160 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003490170511534604, + "loss": 1.4189, + "theoretical_loss": 3.6423988032003143, + "tokens_seen": 1019805696 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003490070210631896, + "loss": 1.2598, + "theoretical_loss": 3.642376608847334, + "tokens_seen": 1019871232 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034899699097291876, + "loss": 1.3746, + "theoretical_loss": 3.642354416319801, + "tokens_seen": 1019936768 + }, + { + "epoch": 0.31, + "learning_rate": 0.000348986960882648, + "loss": 1.469, + "theoretical_loss": 3.6423322256174475, + "tokens_seen": 1020002304 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003489769307923771, + "loss": 1.4417, + "theoretical_loss": 3.642310036740006, + "tokens_seen": 1020067840 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034896690070210636, + "loss": 1.3379, + "theoretical_loss": 3.642287849687209, + "tokens_seen": 1020133376 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003489568706118355, + "loss": 1.366, + "theoretical_loss": 3.6422656644587903, + "tokens_seen": 1020198912 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003489468405215647, + "loss": 1.402, + "theoretical_loss": 3.6422434810544813, + "tokens_seen": 1020264448 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003489368104312939, + "loss": 1.2193, + "theoretical_loss": 3.6422212994740155, + "tokens_seen": 1020329984 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003489267803410231, + "loss": 1.3386, + "theoretical_loss": 3.6421991197171257, + "tokens_seen": 1020395520 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034891675025075226, + "loss": 1.4533, + "theoretical_loss": 3.6421769417835446, + "tokens_seen": 1020461056 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003489067201604815, + "loss": 1.39, + "theoretical_loss": 3.642154765673006, + "tokens_seen": 1020526592 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003488966900702106, + "loss": 1.3782, + "theoretical_loss": 3.6421325913852427, + "tokens_seen": 1020592128 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034888665997993986, + "loss": 1.3909, + "theoretical_loss": 3.6421104189199873, + "tokens_seen": 1020657664 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.00011707434168783948, + "objective/train/docs_used": 584789, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.417205572128296, + "objective/train/original_loss": 3.417205572128296, + "objective/train/theoretical_loss": 3.642088248276974, + "objective/train/tokens_used": 1041183200, + "objective/train/value_avg": -0.006900787353515625, + "objective/train/value_loss": 7.901938806753606e-05, + "objective/train/value_max": -0.0003669261932373047, + "objective/train/value_min": -0.07464599609375, + "objective/train/value_reward_corr": 0.4983213484022935, + "objective/train/value_std": 0.00653076171875, + "objective/train/weight_avg": 1.0024828910827637, + "objective/train/weighted_lm_loss": 3.4254310131073, + "objective/train/weights_max": 2.0071356296539307, + "objective/train/weights_min": 0.2453070431947708, + "theoretical_loss": 3.642088248276974, + "tokens_seen": 1020723200 + }, + { + "epoch": 0.31, + "learning_rate": 0.000348876629889669, + "loss": 1.4991, + "theoretical_loss": 3.642088248276974, + "tokens_seen": 1020723200 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003488665997993982, + "loss": 1.3093, + "theoretical_loss": 3.642066079455935, + "tokens_seen": 1020788736 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003488565697091274, + "loss": 1.4872, + "theoretical_loss": 3.642043912456604, + "tokens_seen": 1020854272 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003488465396188566, + "loss": 1.4625, + "theoretical_loss": 3.642021747278715, + "tokens_seen": 1020919808 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034883650952858577, + "loss": 1.298, + "theoretical_loss": 3.6419995839220007, + "tokens_seen": 1020985344 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034882647943831495, + "loss": 1.3668, + "theoretical_loss": 3.6419774223861943, + "tokens_seen": 1021050880 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034881644934804413, + "loss": 1.3217, + "theoretical_loss": 3.64195526267103, + "tokens_seen": 1021116416 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034880641925777336, + "loss": 1.2893, + "theoretical_loss": 3.6419331047762418, + "tokens_seen": 1021181952 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003487963891675025, + "loss": 1.4083, + "theoretical_loss": 3.641910948701562, + "tokens_seen": 1021247488 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003487863590772317, + "loss": 1.3235, + "theoretical_loss": 3.641888794446725, + "tokens_seen": 1021313024 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034877632898696085, + "loss": 1.314, + "theoretical_loss": 3.6418666420114647, + "tokens_seen": 1021378560 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003487662988966901, + "loss": 1.3795, + "theoretical_loss": 3.6418444913955152, + "tokens_seen": 1021444096 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034875626880641927, + "loss": 1.4544, + "theoretical_loss": 3.6418223425986094, + "tokens_seen": 1021509632 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034874623871614845, + "loss": 1.3746, + "theoretical_loss": 3.6418001956204824, + "tokens_seen": 1021575168 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034873620862587763, + "loss": 1.2792, + "theoretical_loss": 3.6417780504608674, + "tokens_seen": 1021640704 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034872617853560687, + "loss": 1.3875, + "theoretical_loss": 3.6417559071194985, + "tokens_seen": 1021706240 + }, + { + "epoch": 0.31, + "learning_rate": 0.000348716148445336, + "loss": 1.2485, + "theoretical_loss": 3.64173376559611, + "tokens_seen": 1021771776 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034870611835506523, + "loss": 1.2265, + "theoretical_loss": 3.6417116258904363, + "tokens_seen": 1021837312 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034869608826479436, + "loss": 1.3263, + "theoretical_loss": 3.6416894880022106, + "tokens_seen": 1021902848 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003486860581745236, + "loss": 1.4331, + "theoretical_loss": 3.6416673519311686, + "tokens_seen": 1021968384 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034867602808425277, + "loss": 1.3706, + "theoretical_loss": 3.641645217677043, + "tokens_seen": 1022033920 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034866599799398195, + "loss": 1.4127, + "theoretical_loss": 3.64162308523957, + "tokens_seen": 1022099456 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034865596790371113, + "loss": 1.2763, + "theoretical_loss": 3.6416009546184824, + "tokens_seen": 1022164992 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003486459378134403, + "loss": 1.4463, + "theoretical_loss": 3.6415788258135158, + "tokens_seen": 1022230528 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003486359077231695, + "loss": 1.4483, + "theoretical_loss": 3.641556698824404, + "tokens_seen": 1022296064 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.0011900985846295953, + "objective/train/docs_used": 586155, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9647042751312256, + "objective/train/original_loss": 2.9647042751312256, + "objective/train/theoretical_loss": 3.6415345736508824, + "objective/train/tokens_used": 1042821600, + "objective/train/value_avg": -0.0054779052734375, + "objective/train/value_loss": 6.735356146236882e-05, + "objective/train/value_max": -0.0003814697265625, + "objective/train/value_min": -0.06512451171875, + "objective/train/value_reward_corr": 0.35904430587207237, + "objective/train/value_std": 0.00493621826171875, + "objective/train/weight_avg": 0.9913259148597717, + "objective/train/weighted_lm_loss": 2.9421751499176025, + "objective/train/weights_max": 1.525827407836914, + "objective/train/weights_min": 0.5734212398529053, + "theoretical_loss": 3.6415345736508824, + "tokens_seen": 1022361600 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034862587763289873, + "loss": 1.2673, + "theoretical_loss": 3.6415345736508824, + "tokens_seen": 1022361600 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034861584754262786, + "loss": 1.4281, + "theoretical_loss": 3.641512450292685, + "tokens_seen": 1022427136 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003486058174523571, + "loss": 1.3203, + "theoretical_loss": 3.641490328749547, + "tokens_seen": 1022492672 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003485957873620862, + "loss": 1.4276, + "theoretical_loss": 3.641468209021202, + "tokens_seen": 1022558208 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034858575727181546, + "loss": 1.3514, + "theoretical_loss": 3.6414460911073863, + "tokens_seen": 1022623744 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034857572718154464, + "loss": 1.5064, + "theoretical_loss": 3.641423975007834, + "tokens_seen": 1022689280 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003485656970912738, + "loss": 1.3707, + "theoretical_loss": 3.6414018607222802, + "tokens_seen": 1022754816 + }, + { + "epoch": 0.31, + "learning_rate": 0.000348555667001003, + "loss": 1.3536, + "theoretical_loss": 3.64137974825046, + "tokens_seen": 1022820352 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034854563691073223, + "loss": 1.2524, + "theoretical_loss": 3.641357637592108, + "tokens_seen": 1022885888 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034853560682046136, + "loss": 1.2854, + "theoretical_loss": 3.6413355287469598, + "tokens_seen": 1022951424 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003485255767301906, + "loss": 1.3194, + "theoretical_loss": 3.6413134217147505, + "tokens_seen": 1023016960 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003485155466399197, + "loss": 1.3206, + "theoretical_loss": 3.6412913164952148, + "tokens_seen": 1023082496 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034850551654964896, + "loss": 1.405, + "theoretical_loss": 3.641269213088089, + "tokens_seen": 1023148032 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034849548645937814, + "loss": 1.3136, + "theoretical_loss": 3.641247111493107, + "tokens_seen": 1023213568 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003484854563691073, + "loss": 1.4456, + "theoretical_loss": 3.641225011710006, + "tokens_seen": 1023279104 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003484754262788365, + "loss": 1.3904, + "theoretical_loss": 3.64120291373852, + "tokens_seen": 1023344640 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003484653961885657, + "loss": 1.3824, + "theoretical_loss": 3.6411808175783844, + "tokens_seen": 1023410176 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034845536609829486, + "loss": 1.3472, + "theoretical_loss": 3.641158723229336, + "tokens_seen": 1023475712 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003484453360080241, + "loss": 1.3147, + "theoretical_loss": 3.641136630691109, + "tokens_seen": 1023541248 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003484353059177532, + "loss": 1.2926, + "theoretical_loss": 3.64111453996344, + "tokens_seen": 1023606784 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034842527582748246, + "loss": 1.4128, + "theoretical_loss": 3.641092451046064, + "tokens_seen": 1023672320 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034841524573721164, + "loss": 1.4177, + "theoretical_loss": 3.6410703639387174, + "tokens_seen": 1023737856 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003484052156469408, + "loss": 1.4175, + "theoretical_loss": 3.6410482786411364, + "tokens_seen": 1023803392 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034839518555667, + "loss": 1.3228, + "theoretical_loss": 3.6410261951530556, + "tokens_seen": 1023868928 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003483851554663992, + "loss": 1.3002, + "theoretical_loss": 3.641004113474212, + "tokens_seen": 1023934464 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.0007775081321597099, + "objective/train/docs_used": 586672, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4075872898101807, + "objective/train/original_loss": 2.4075872898101807, + "objective/train/theoretical_loss": 3.6409820336043413, + "objective/train/tokens_used": 1044460000, + "objective/train/value_avg": -0.004917144775390625, + "objective/train/value_loss": 6.61528974887915e-05, + "objective/train/value_max": -0.0004222393035888672, + "objective/train/value_min": -0.058990478515625, + "objective/train/value_reward_corr": 0.2976840726950683, + "objective/train/value_std": 0.0038890838623046875, + "objective/train/weight_avg": 0.9953282475471497, + "objective/train/weighted_lm_loss": 2.402186632156372, + "objective/train/weights_max": 1.564395785331726, + "objective/train/weights_min": 0.38984090089797974, + "theoretical_loss": 3.6409820336043413, + "tokens_seen": 1024000000 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034837512537612837, + "loss": 1.3037, + "theoretical_loss": 3.6409820336043413, + "tokens_seen": 1024000000 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003483650952858576, + "loss": 1.3541, + "theoretical_loss": 3.6409599555431793, + "tokens_seen": 1024065536 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034835506519558673, + "loss": 1.4195, + "theoretical_loss": 3.640937879290462, + "tokens_seen": 1024131072 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034834503510531597, + "loss": 1.2517, + "theoretical_loss": 3.640915804845926, + "tokens_seen": 1024196608 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034833500501504515, + "loss": 1.3615, + "theoretical_loss": 3.640893732209307, + "tokens_seen": 1024262144 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034832497492477433, + "loss": 1.4027, + "theoretical_loss": 3.6408716613803422, + "tokens_seen": 1024327680 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034831494483450356, + "loss": 1.2858, + "theoretical_loss": 3.6408495923587667, + "tokens_seen": 1024393216 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003483049147442327, + "loss": 1.296, + "theoretical_loss": 3.640827525144318, + "tokens_seen": 1024458752 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003482948846539619, + "loss": 1.3186, + "theoretical_loss": 3.640805459736732, + "tokens_seen": 1024524288 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034828485456369105, + "loss": 1.3335, + "theoretical_loss": 3.640783396135745, + "tokens_seen": 1024589824 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003482748244734203, + "loss": 1.5856, + "theoretical_loss": 3.640761334341094, + "tokens_seen": 1024655360 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034826479438314947, + "loss": 1.4669, + "theoretical_loss": 3.640739274352515, + "tokens_seen": 1024720896 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034825476429287865, + "loss": 1.3158, + "theoretical_loss": 3.640717216169745, + "tokens_seen": 1024786432 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034824473420260783, + "loss": 1.3737, + "theoretical_loss": 3.640695159792521, + "tokens_seen": 1024851968 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034823470411233707, + "loss": 1.2866, + "theoretical_loss": 3.6406731052205794, + "tokens_seen": 1024917504 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003482246740220662, + "loss": 1.3475, + "theoretical_loss": 3.640651052453657, + "tokens_seen": 1024983040 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034821464393179543, + "loss": 1.2542, + "theoretical_loss": 3.640629001491491, + "tokens_seen": 1025048576 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034820461384152456, + "loss": 1.4436, + "theoretical_loss": 3.640606952333818, + "tokens_seen": 1025114112 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003481945837512538, + "loss": 1.3104, + "theoretical_loss": 3.6405849049803747, + "tokens_seen": 1025179648 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034818455366098297, + "loss": 1.2773, + "theoretical_loss": 3.640562859430899, + "tokens_seen": 1025245184 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034817452357071215, + "loss": 1.3114, + "theoretical_loss": 3.6405408156851276, + "tokens_seen": 1025310720 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034816449348044133, + "loss": 1.4173, + "theoretical_loss": 3.6405187737427975, + "tokens_seen": 1025376256 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003481544633901705, + "loss": 1.2617, + "theoretical_loss": 3.640496733603646, + "tokens_seen": 1025441792 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003481444332998997, + "loss": 1.3525, + "theoretical_loss": 3.64047469526741, + "tokens_seen": 1025507328 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034813440320962893, + "loss": 1.3901, + "theoretical_loss": 3.6404526587338273, + "tokens_seen": 1025572864 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.0001317841379204765, + "objective/train/docs_used": 587795, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8443408012390137, + "objective/train/original_loss": 2.8443408012390137, + "objective/train/theoretical_loss": 3.6404306240026356, + "objective/train/tokens_used": 1046098400, + "objective/train/value_avg": -0.00539398193359375, + "objective/train/value_loss": 6.309487071121112e-05, + "objective/train/value_max": -0.0005955696105957031, + "objective/train/value_min": -0.0521240234375, + "objective/train/value_reward_corr": 0.362393795644516, + "objective/train/value_std": 0.00402069091796875, + "objective/train/weight_avg": 1.0016127824783325, + "objective/train/weighted_lm_loss": 2.852743148803711, + "objective/train/weights_max": 1.4462262392044067, + "objective/train/weights_min": 0.39333635568618774, + "theoretical_loss": 3.6404306240026356, + "tokens_seen": 1025638400 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034812437311935806, + "loss": 1.2926, + "theoretical_loss": 3.6404306240026356, + "tokens_seen": 1025638400 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003481143430290873, + "loss": 1.3816, + "theoretical_loss": 3.6404085910735713, + "tokens_seen": 1025703936 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003481043129388164, + "loss": 1.3536, + "theoretical_loss": 3.6403865599463727, + "tokens_seen": 1025769472 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034809428284854566, + "loss": 1.386, + "theoretical_loss": 3.640364530620777, + "tokens_seen": 1025835008 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034808425275827484, + "loss": 1.3742, + "theoretical_loss": 3.640342503096522, + "tokens_seen": 1025900544 + }, + { + "epoch": 0.31, + "learning_rate": 0.000348074222668004, + "loss": 1.3282, + "theoretical_loss": 3.6403204773733453, + "tokens_seen": 1025966080 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003480641925777332, + "loss": 1.4112, + "theoretical_loss": 3.640298453450984, + "tokens_seen": 1026031616 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034805416248746243, + "loss": 1.319, + "theoretical_loss": 3.640276431329177, + "tokens_seen": 1026097152 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034804413239719156, + "loss": 1.3456, + "theoretical_loss": 3.640254411007661, + "tokens_seen": 1026162688 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003480341023069208, + "loss": 1.4242, + "theoretical_loss": 3.6402323924861744, + "tokens_seen": 1026228224 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003480240722166499, + "loss": 1.2903, + "theoretical_loss": 3.6402103757644553, + "tokens_seen": 1026293760 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034801404212637916, + "loss": 1.3161, + "theoretical_loss": 3.6401883608422416, + "tokens_seen": 1026359296 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034800401203610834, + "loss": 1.2661, + "theoretical_loss": 3.6401663477192705, + "tokens_seen": 1026424832 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003479939819458375, + "loss": 1.3358, + "theoretical_loss": 3.6401443363952812, + "tokens_seen": 1026490368 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003479839518555667, + "loss": 1.5407, + "theoretical_loss": 3.640122326870012, + "tokens_seen": 1026555904 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003479739217652959, + "loss": 1.2767, + "theoretical_loss": 3.6401003191431993, + "tokens_seen": 1026621440 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034796389167502506, + "loss": 1.4126, + "theoretical_loss": 3.640078313214583, + "tokens_seen": 1026686976 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003479538615847543, + "loss": 1.3628, + "theoretical_loss": 3.640056309083901, + "tokens_seen": 1026752512 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034794383149448343, + "loss": 1.4, + "theoretical_loss": 3.640034306750892, + "tokens_seen": 1026818048 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034793380140421266, + "loss": 1.3705, + "theoretical_loss": 3.6400123062152936, + "tokens_seen": 1026883584 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034792377131394184, + "loss": 1.4081, + "theoretical_loss": 3.639990307476845, + "tokens_seen": 1026949120 + }, + { + "epoch": 0.31, + "learning_rate": 0.000347913741223671, + "loss": 1.2652, + "theoretical_loss": 3.6399683105352842, + "tokens_seen": 1027014656 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003479037111334002, + "loss": 1.3522, + "theoretical_loss": 3.6399463153903495, + "tokens_seen": 1027080192 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003478936810431294, + "loss": 1.3375, + "theoretical_loss": 3.6399243220417805, + "tokens_seen": 1027145728 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034788365095285857, + "loss": 1.4278, + "theoretical_loss": 3.6399023304893157, + "tokens_seen": 1027211264 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.0012419959530234337, + "objective/train/docs_used": 588595, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.795722723007202, + "objective/train/original_loss": 2.795722484588623, + "objective/train/theoretical_loss": 3.6398803407326934, + "objective/train/tokens_used": 1047736800, + "objective/train/value_avg": -0.005199432373046875, + "objective/train/value_loss": 5.006892024539411e-05, + "objective/train/value_max": -0.00032258033752441406, + "objective/train/value_min": -0.06927490234375, + "objective/train/value_reward_corr": 0.43439440854205535, + "objective/train/value_std": 0.0043792724609375, + "objective/train/weight_avg": 0.9899711608886719, + "objective/train/weighted_lm_loss": 2.7725000381469727, + "objective/train/weights_max": 1.7315400838851929, + "objective/train/weights_min": 0.4939432740211487, + "theoretical_loss": 3.6398803407326934, + "tokens_seen": 1027276800 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003478736208625878, + "loss": 1.4431, + "theoretical_loss": 3.6398803407326934, + "tokens_seen": 1027276800 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034786359077231693, + "loss": 1.4678, + "theoretical_loss": 3.639858352771652, + "tokens_seen": 1027342336 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034785356068204617, + "loss": 1.3324, + "theoretical_loss": 3.639836366605932, + "tokens_seen": 1027407872 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003478435305917753, + "loss": 1.3242, + "theoretical_loss": 3.6398143822352704, + "tokens_seen": 1027473408 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034783350050150453, + "loss": 1.4699, + "theoretical_loss": 3.639792399659407, + "tokens_seen": 1027538944 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003478234704112337, + "loss": 1.4146, + "theoretical_loss": 3.639770418878081, + "tokens_seen": 1027604480 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003478134403209629, + "loss": 1.4424, + "theoretical_loss": 3.639748439891031, + "tokens_seen": 1027670016 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034780341023069207, + "loss": 1.3181, + "theoretical_loss": 3.639726462697997, + "tokens_seen": 1027735552 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034779338014042125, + "loss": 1.2991, + "theoretical_loss": 3.639704487298717, + "tokens_seen": 1027801088 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034778335005015043, + "loss": 1.2867, + "theoretical_loss": 3.6396825136929314, + "tokens_seen": 1027866624 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034777331995987967, + "loss": 1.2477, + "theoretical_loss": 3.639660541880378, + "tokens_seen": 1027932160 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003477632898696088, + "loss": 1.3244, + "theoretical_loss": 3.6396385718607975, + "tokens_seen": 1027997696 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034775325977933803, + "loss": 1.3333, + "theoretical_loss": 3.639616603633929, + "tokens_seen": 1028063232 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003477432296890672, + "loss": 1.3837, + "theoretical_loss": 3.6395946371995116, + "tokens_seen": 1028128768 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003477331995987964, + "loss": 1.2294, + "theoretical_loss": 3.639572672557285, + "tokens_seen": 1028194304 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003477231695085256, + "loss": 1.3107, + "theoretical_loss": 3.6395507097069886, + "tokens_seen": 1028259840 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034771313941825476, + "loss": 1.427, + "theoretical_loss": 3.6395287486483623, + "tokens_seen": 1028325376 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034770310932798394, + "loss": 1.2936, + "theoretical_loss": 3.6395067893811457, + "tokens_seen": 1028390912 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034769307923771317, + "loss": 1.442, + "theoretical_loss": 3.6394848319050785, + "tokens_seen": 1028456448 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003476830491474423, + "loss": 1.3128, + "theoretical_loss": 3.6394628762199, + "tokens_seen": 1028521984 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034767301905717153, + "loss": 1.493, + "theoretical_loss": 3.6394409223253508, + "tokens_seen": 1028587520 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034766298896690066, + "loss": 1.3212, + "theoretical_loss": 3.6394189702211706, + "tokens_seen": 1028653056 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003476529588766299, + "loss": 1.3514, + "theoretical_loss": 3.639397019907099, + "tokens_seen": 1028718592 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003476429287863591, + "loss": 1.3897, + "theoretical_loss": 3.639375071382876, + "tokens_seen": 1028784128 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034763289869608826, + "loss": 1.4125, + "theoretical_loss": 3.6393531246482422, + "tokens_seen": 1028849664 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": 0.00011982597061432898, + "objective/train/docs_used": 589927, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7058944702148438, + "objective/train/original_loss": 2.7058939933776855, + "objective/train/theoretical_loss": 3.6393311797029373, + "objective/train/tokens_used": 1049375200, + "objective/train/value_avg": -0.0052337646484375, + "objective/train/value_loss": 5.303691068547778e-05, + "objective/train/value_max": -0.00047850608825683594, + "objective/train/value_min": -0.055938720703125, + "objective/train/value_reward_corr": 0.43017755646471273, + "objective/train/value_std": 0.004169464111328125, + "objective/train/weight_avg": 1.0037285089492798, + "objective/train/weighted_lm_loss": 2.7154736518859863, + "objective/train/weights_max": 1.4660135507583618, + "objective/train/weights_min": 0.5737342834472656, + "theoretical_loss": 3.6393311797029373, + "tokens_seen": 1028915200 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034762286860581744, + "loss": 1.3278, + "theoretical_loss": 3.6393311797029373, + "tokens_seen": 1028915200 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003476128385155466, + "loss": 1.3847, + "theoretical_loss": 3.6393092365467012, + "tokens_seen": 1028980736 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003476028084252758, + "loss": 1.4081, + "theoretical_loss": 3.639287295179275, + "tokens_seen": 1029046272 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034759277833500504, + "loss": 1.4043, + "theoretical_loss": 3.639265355600398, + "tokens_seen": 1029111808 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003475827482447342, + "loss": 1.4153, + "theoretical_loss": 3.639243417809811, + "tokens_seen": 1029177344 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003475727181544634, + "loss": 1.4349, + "theoretical_loss": 3.639221481807254, + "tokens_seen": 1029242880 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034756268806419263, + "loss": 1.3229, + "theoretical_loss": 3.639199547592468, + "tokens_seen": 1029308416 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034755265797392176, + "loss": 1.4636, + "theoretical_loss": 3.6391776151651936, + "tokens_seen": 1029373952 + }, + { + "epoch": 0.31, + "learning_rate": 0.000347542627883651, + "loss": 1.447, + "theoretical_loss": 3.6391556845251705, + "tokens_seen": 1029439488 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003475325977933801, + "loss": 1.3954, + "theoretical_loss": 3.6391337556721393, + "tokens_seen": 1029505024 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034752256770310936, + "loss": 1.3518, + "theoretical_loss": 3.6391118286058415, + "tokens_seen": 1029570560 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034751253761283854, + "loss": 1.419, + "theoretical_loss": 3.639089903326018, + "tokens_seen": 1029636096 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003475025075225677, + "loss": 1.3351, + "theoretical_loss": 3.639067979832408, + "tokens_seen": 1029701632 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003474924774322969, + "loss": 1.3568, + "theoretical_loss": 3.6390460581247535, + "tokens_seen": 1029767168 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003474824473420261, + "loss": 1.385, + "theoretical_loss": 3.639024138202795, + "tokens_seen": 1029832704 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034747241725175527, + "loss": 1.3253, + "theoretical_loss": 3.639002220066274, + "tokens_seen": 1029898240 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003474623871614845, + "loss": 1.3989, + "theoretical_loss": 3.6389803037149306, + "tokens_seen": 1029963776 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034745235707121363, + "loss": 1.3729, + "theoretical_loss": 3.638958389148506, + "tokens_seen": 1030029312 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034744232698094286, + "loss": 1.3666, + "theoretical_loss": 3.6389364763667422, + "tokens_seen": 1030094848 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034743229689067204, + "loss": 1.4213, + "theoretical_loss": 3.6389145653693795, + "tokens_seen": 1030160384 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003474222668004012, + "loss": 1.1532, + "theoretical_loss": 3.638892656156159, + "tokens_seen": 1030225920 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003474122367101304, + "loss": 1.2857, + "theoretical_loss": 3.638870748726822, + "tokens_seen": 1030291456 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003474022066198596, + "loss": 1.2867, + "theoretical_loss": 3.63884884308111, + "tokens_seen": 1030356992 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034739217652958877, + "loss": 1.3574, + "theoretical_loss": 3.6388269392187644, + "tokens_seen": 1030422528 + }, + { + "epoch": 0.31, + "learning_rate": 0.000347382146439318, + "loss": 1.4422, + "theoretical_loss": 3.6388050371395266, + "tokens_seen": 1030488064 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.000812353624496609, + "objective/train/docs_used": 590720, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9765918254852295, + "objective/train/original_loss": 2.9765918254852295, + "objective/train/theoretical_loss": 3.638783136843138, + "objective/train/tokens_used": 1051013600, + "objective/train/value_avg": -0.0056610107421875, + "objective/train/value_loss": 6.720771489199251e-05, + "objective/train/value_max": -0.0002913475036621094, + "objective/train/value_min": -0.06744384765625, + "objective/train/value_reward_corr": 0.43146675048970284, + "objective/train/value_std": 0.00479888916015625, + "objective/train/weight_avg": 0.9950158596038818, + "objective/train/weighted_lm_loss": 2.963654041290283, + "objective/train/weights_max": 1.569179892539978, + "objective/train/weights_min": 0.2939666509628296, + "theoretical_loss": 3.638783136843138, + "tokens_seen": 1030553600 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034737211634904713, + "loss": 1.3145, + "theoretical_loss": 3.638783136843138, + "tokens_seen": 1030553600 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034736208625877637, + "loss": 1.2525, + "theoretical_loss": 3.63876123832934, + "tokens_seen": 1030619136 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003473520561685055, + "loss": 1.5213, + "theoretical_loss": 3.6387393415978746, + "tokens_seen": 1030684672 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034734202607823473, + "loss": 1.341, + "theoretical_loss": 3.6387174466484824, + "tokens_seen": 1030750208 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003473319959879639, + "loss": 1.4605, + "theoretical_loss": 3.6386955534809062, + "tokens_seen": 1030815744 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003473219658976931, + "loss": 1.3342, + "theoretical_loss": 3.638673662094887, + "tokens_seen": 1030881280 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034731193580742227, + "loss": 1.3743, + "theoretical_loss": 3.6386517724901672, + "tokens_seen": 1030946816 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034730190571715145, + "loss": 1.3568, + "theoretical_loss": 3.6386298846664884, + "tokens_seen": 1031012352 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034729187562688063, + "loss": 1.4239, + "theoretical_loss": 3.638607998623592, + "tokens_seen": 1031077888 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034728184553660987, + "loss": 1.4016, + "theoretical_loss": 3.63858611436122, + "tokens_seen": 1031143424 + }, + { + "epoch": 0.31, + "learning_rate": 0.000347271815446339, + "loss": 1.3937, + "theoretical_loss": 3.6385642318791156, + "tokens_seen": 1031208960 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034726178535606823, + "loss": 1.3001, + "theoretical_loss": 3.638542351177019, + "tokens_seen": 1031274496 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003472517552657974, + "loss": 1.4095, + "theoretical_loss": 3.6385204722546742, + "tokens_seen": 1031340032 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003472417251755266, + "loss": 1.3764, + "theoretical_loss": 3.638498595111822, + "tokens_seen": 1031405568 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003472316950852558, + "loss": 1.4364, + "theoretical_loss": 3.638476719748205, + "tokens_seen": 1031471104 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034722166499498496, + "loss": 1.2734, + "theoretical_loss": 3.6384548461635653, + "tokens_seen": 1031536640 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034721163490471414, + "loss": 1.342, + "theoretical_loss": 3.6384329743576456, + "tokens_seen": 1031602176 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034720160481444337, + "loss": 1.4835, + "theoretical_loss": 3.638411104330188, + "tokens_seen": 1031667712 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003471915747241725, + "loss": 1.3455, + "theoretical_loss": 3.6383892360809353, + "tokens_seen": 1031733248 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034718154463390173, + "loss": 1.4038, + "theoretical_loss": 3.6383673696096297, + "tokens_seen": 1031798784 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034717151454363086, + "loss": 1.3731, + "theoretical_loss": 3.638345504916013, + "tokens_seen": 1031864320 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003471614844533601, + "loss": 1.2971, + "theoretical_loss": 3.6383236419998295, + "tokens_seen": 1031929856 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003471514543630893, + "loss": 1.4995, + "theoretical_loss": 3.6383017808608207, + "tokens_seen": 1031995392 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034714142427281846, + "loss": 1.3415, + "theoretical_loss": 3.6382799214987287, + "tokens_seen": 1032060928 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034713139418254764, + "loss": 1.3106, + "theoretical_loss": 3.6382580639132978, + "tokens_seen": 1032126464 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": 0.0009231420117430389, + "objective/train/docs_used": 591991, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5441277027130127, + "objective/train/original_loss": 2.5441274642944336, + "objective/train/theoretical_loss": 3.63823620810427, + "objective/train/tokens_used": 1052652000, + "objective/train/value_avg": -0.00615692138671875, + "objective/train/value_loss": 6.679283978883177e-05, + "objective/train/value_max": -0.00036978721618652344, + "objective/train/value_min": -0.06488037109375, + "objective/train/value_reward_corr": 0.283438192498869, + "objective/train/value_std": 0.00470733642578125, + "objective/train/weight_avg": 1.0123306512832642, + "objective/train/weighted_lm_loss": 2.5722103118896484, + "objective/train/weights_max": 1.7285465002059937, + "objective/train/weights_min": 0.3393654525279999, + "theoretical_loss": 3.63823620810427, + "tokens_seen": 1032192000 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003471213640922768, + "loss": 1.3025, + "theoretical_loss": 3.63823620810427, + "tokens_seen": 1032192000 + }, + { + "epoch": 0.31, + "learning_rate": 0.000347111334002006, + "loss": 1.4159, + "theoretical_loss": 3.6382143540713874, + "tokens_seen": 1032257536 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034710130391173524, + "loss": 1.3381, + "theoretical_loss": 3.6381925018143946, + "tokens_seen": 1032323072 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034709127382146436, + "loss": 1.3245, + "theoretical_loss": 3.638170651333033, + "tokens_seen": 1032388608 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003470812437311936, + "loss": 1.3696, + "theoretical_loss": 3.638148802627046, + "tokens_seen": 1032454144 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003470712136409228, + "loss": 1.4148, + "theoretical_loss": 3.6381269556961775, + "tokens_seen": 1032519680 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034706118355065196, + "loss": 1.287, + "theoretical_loss": 3.63810511054017, + "tokens_seen": 1032585216 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034705115346038114, + "loss": 1.4047, + "theoretical_loss": 3.638083267158766, + "tokens_seen": 1032650752 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003470411233701103, + "loss": 1.3671, + "theoretical_loss": 3.6380614255517107, + "tokens_seen": 1032716288 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003470310932798395, + "loss": 1.2997, + "theoretical_loss": 3.638039585718745, + "tokens_seen": 1032781824 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034702106318956874, + "loss": 1.3877, + "theoretical_loss": 3.638017747659614, + "tokens_seen": 1032847360 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034701103309929787, + "loss": 1.329, + "theoretical_loss": 3.6379959113740603, + "tokens_seen": 1032912896 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003470010030090271, + "loss": 1.432, + "theoretical_loss": 3.6379740768618274, + "tokens_seen": 1032978432 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034699097291875623, + "loss": 1.393, + "theoretical_loss": 3.637952244122659, + "tokens_seen": 1033043968 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034698094282848547, + "loss": 1.3901, + "theoretical_loss": 3.6379304131562984, + "tokens_seen": 1033109504 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034697091273821465, + "loss": 1.2318, + "theoretical_loss": 3.6379085839624894, + "tokens_seen": 1033175040 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034696088264794383, + "loss": 1.2979, + "theoretical_loss": 3.6378867565409756, + "tokens_seen": 1033240576 + }, + { + "epoch": 0.31, + "learning_rate": 0.000346950852557673, + "loss": 1.4005, + "theoretical_loss": 3.6378649308915008, + "tokens_seen": 1033306112 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034694082246740224, + "loss": 1.4133, + "theoretical_loss": 3.6378431070138086, + "tokens_seen": 1033371648 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034693079237713137, + "loss": 1.284, + "theoretical_loss": 3.637821284907643, + "tokens_seen": 1033437184 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003469207622868606, + "loss": 1.426, + "theoretical_loss": 3.6377994645727476, + "tokens_seen": 1033502720 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034691073219658973, + "loss": 1.356, + "theoretical_loss": 3.6377776460088667, + "tokens_seen": 1033568256 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034690070210631897, + "loss": 1.3176, + "theoretical_loss": 3.6377558292157435, + "tokens_seen": 1033633792 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034689067201604815, + "loss": 1.2509, + "theoretical_loss": 3.637734014193123, + "tokens_seen": 1033699328 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034688064192577733, + "loss": 1.3054, + "theoretical_loss": 3.637712200940749, + "tokens_seen": 1033764864 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": 0.0001706752082100138, + "objective/train/docs_used": 592462, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.662944793701172, + "objective/train/original_loss": 2.6629445552825928, + "objective/train/theoretical_loss": 3.637690389458365, + "objective/train/tokens_used": 1054290400, + "objective/train/value_avg": -0.004730224609375, + "objective/train/value_loss": 4.843303395318799e-05, + "objective/train/value_max": -0.00019562244415283203, + "objective/train/value_min": -0.062103271484375, + "objective/train/value_reward_corr": 0.30480114930021857, + "objective/train/value_std": 0.004009246826171875, + "objective/train/weight_avg": 1.0040305852890015, + "objective/train/weighted_lm_loss": 2.6857802867889404, + "objective/train/weights_max": 1.7469559907913208, + "objective/train/weights_min": 0.5139598250389099, + "theoretical_loss": 3.637690389458365, + "tokens_seen": 1033830400 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003468706118355065, + "loss": 1.4262, + "theoretical_loss": 3.637690389458365, + "tokens_seen": 1033830400 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003468605817452357, + "loss": 1.4183, + "theoretical_loss": 3.637668579745716, + "tokens_seen": 1033895936 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003468505516549649, + "loss": 1.5345, + "theoretical_loss": 3.637646771802546, + "tokens_seen": 1033961472 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003468405215646941, + "loss": 1.389, + "theoretical_loss": 3.6376249656285995, + "tokens_seen": 1034027008 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003468304914744233, + "loss": 1.3596, + "theoretical_loss": 3.63760316122362, + "tokens_seen": 1034092544 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034682046138415247, + "loss": 1.4556, + "theoretical_loss": 3.637581358587353, + "tokens_seen": 1034158080 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034681043129388165, + "loss": 1.372, + "theoretical_loss": 3.637559557719542, + "tokens_seen": 1034223616 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034680040120361083, + "loss": 1.3139, + "theoretical_loss": 3.6375377586199322, + "tokens_seen": 1034289152 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034679037111334007, + "loss": 1.392, + "theoretical_loss": 3.637515961288268, + "tokens_seen": 1034354688 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003467803410230692, + "loss": 1.3146, + "theoretical_loss": 3.637494165724294, + "tokens_seen": 1034420224 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034677031093279843, + "loss": 1.4248, + "theoretical_loss": 3.637472371927754, + "tokens_seen": 1034485760 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003467602808425276, + "loss": 1.4838, + "theoretical_loss": 3.6374505798983945, + "tokens_seen": 1034551296 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003467502507522568, + "loss": 1.3076, + "theoretical_loss": 3.637428789635959, + "tokens_seen": 1034616832 + }, + { + "epoch": 0.31, + "learning_rate": 0.000346740220661986, + "loss": 1.2722, + "theoretical_loss": 3.6374070011401924, + "tokens_seen": 1034682368 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034673019057171516, + "loss": 1.3067, + "theoretical_loss": 3.6373852144108403, + "tokens_seen": 1034747904 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034672016048144434, + "loss": 1.272, + "theoretical_loss": 3.6373634294476465, + "tokens_seen": 1034813440 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034671013039117357, + "loss": 1.3486, + "theoretical_loss": 3.6373416462503565, + "tokens_seen": 1034878976 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003467001003009027, + "loss": 1.3976, + "theoretical_loss": 3.637319864818716, + "tokens_seen": 1034944512 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034669007021063193, + "loss": 1.2748, + "theoretical_loss": 3.6372980851524694, + "tokens_seen": 1035010048 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034668004012036106, + "loss": 1.3013, + "theoretical_loss": 3.637276307251362, + "tokens_seen": 1035075584 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003466700100300903, + "loss": 1.2806, + "theoretical_loss": 3.637254531115139, + "tokens_seen": 1035141120 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003466599799398195, + "loss": 1.306, + "theoretical_loss": 3.637232756743545, + "tokens_seen": 1035206656 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034664994984954866, + "loss": 1.4062, + "theoretical_loss": 3.6372109841363267, + "tokens_seen": 1035272192 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034663991975927784, + "loss": 1.3664, + "theoretical_loss": 3.6371892132932286, + "tokens_seen": 1035337728 + }, + { + "epoch": 0.31, + "learning_rate": 0.000346629889669007, + "loss": 1.3786, + "theoretical_loss": 3.637167444213996, + "tokens_seen": 1035403264 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.0009097110596485436, + "objective/train/docs_used": 593828, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.171966791152954, + "objective/train/original_loss": 3.171966791152954, + "objective/train/theoretical_loss": 3.637145676898374, + "objective/train/tokens_used": 1055928800, + "objective/train/value_avg": -0.006610870361328125, + "objective/train/value_loss": 9.156298619927838e-05, + "objective/train/value_max": -0.0002779960632324219, + "objective/train/value_min": -0.061767578125, + "objective/train/value_reward_corr": 0.37980381580077655, + "objective/train/value_std": 0.005657196044921875, + "objective/train/weight_avg": 0.9952318668365479, + "objective/train/weighted_lm_loss": 3.14700984954834, + "objective/train/weights_max": 1.6240520477294922, + "objective/train/weights_min": 0.4693826138973236, + "theoretical_loss": 3.637145676898374, + "tokens_seen": 1035468800 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003466198595787362, + "loss": 1.4436, + "theoretical_loss": 3.637145676898374, + "tokens_seen": 1035468800 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034660982948846544, + "loss": 1.4147, + "theoretical_loss": 3.6371239113461096, + "tokens_seen": 1035534336 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034659979939819456, + "loss": 1.504, + "theoretical_loss": 3.637102147556947, + "tokens_seen": 1035599872 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003465897693079238, + "loss": 1.4583, + "theoretical_loss": 3.6370803855306324, + "tokens_seen": 1035665408 + }, + { + "epoch": 0.31, + "learning_rate": 0.000346579739217653, + "loss": 1.3654, + "theoretical_loss": 3.6370586252669117, + "tokens_seen": 1035730944 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034656970912738216, + "loss": 1.4191, + "theoretical_loss": 3.6370368667655297, + "tokens_seen": 1035796480 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034655967903711134, + "loss": 1.4512, + "theoretical_loss": 3.637015110026233, + "tokens_seen": 1035862016 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003465496489468405, + "loss": 1.2989, + "theoretical_loss": 3.6369933550487668, + "tokens_seen": 1035927552 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003465396188565697, + "loss": 1.4335, + "theoretical_loss": 3.6369716018328777, + "tokens_seen": 1035993088 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034652958876629894, + "loss": 1.2885, + "theoretical_loss": 3.6369498503783113, + "tokens_seen": 1036058624 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034651955867602807, + "loss": 1.4248, + "theoretical_loss": 3.6369281006848135, + "tokens_seen": 1036124160 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003465095285857573, + "loss": 1.3686, + "theoretical_loss": 3.6369063527521304, + "tokens_seen": 1036189696 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034649949849548643, + "loss": 1.4549, + "theoretical_loss": 3.6368846065800082, + "tokens_seen": 1036255232 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034648946840521567, + "loss": 1.4008, + "theoretical_loss": 3.6368628621681927, + "tokens_seen": 1036320768 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034647943831494485, + "loss": 1.4049, + "theoretical_loss": 3.636841119516431, + "tokens_seen": 1036386304 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034646940822467403, + "loss": 1.4092, + "theoretical_loss": 3.636819378624468, + "tokens_seen": 1036451840 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003464593781344032, + "loss": 1.3812, + "theoretical_loss": 3.636797639492051, + "tokens_seen": 1036517376 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034644934804413244, + "loss": 1.2875, + "theoretical_loss": 3.6367759021189263, + "tokens_seen": 1036582912 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034643931795386157, + "loss": 1.4318, + "theoretical_loss": 3.63675416650484, + "tokens_seen": 1036648448 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003464292878635908, + "loss": 1.3541, + "theoretical_loss": 3.6367324326495387, + "tokens_seen": 1036713984 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034641925777331993, + "loss": 1.3597, + "theoretical_loss": 3.6367107005527686, + "tokens_seen": 1036779520 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034640922768304917, + "loss": 1.3839, + "theoretical_loss": 3.6366889702142764, + "tokens_seen": 1036845056 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034639919759277835, + "loss": 1.3938, + "theoretical_loss": 3.636667241633809, + "tokens_seen": 1036910592 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034638916750250753, + "loss": 1.381, + "theoretical_loss": 3.6366455148111134, + "tokens_seen": 1036976128 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003463791374122367, + "loss": 1.4141, + "theoretical_loss": 3.6366237897459355, + "tokens_seen": 1037041664 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.0011785542592406273, + "objective/train/docs_used": 594513, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.212005853652954, + "objective/train/original_loss": 3.212005615234375, + "objective/train/theoretical_loss": 3.636602066438022, + "objective/train/tokens_used": 1057567200, + "objective/train/value_avg": -0.004642486572265625, + "objective/train/value_loss": 0.00011092972272308543, + "objective/train/value_max": -0.0003845691680908203, + "objective/train/value_min": -0.06597900390625, + "objective/train/value_reward_corr": 0.3273756650415579, + "objective/train/value_std": 0.004344940185546875, + "objective/train/weight_avg": 0.9930518865585327, + "objective/train/weighted_lm_loss": 3.1974265575408936, + "objective/train/weights_max": 1.6434968709945679, + "objective/train/weights_min": 0.2554636299610138, + "theoretical_loss": 3.636602066438022, + "tokens_seen": 1037107200 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003463691073219659, + "loss": 1.3662, + "theoretical_loss": 3.636602066438022, + "tokens_seen": 1037107200 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003463590772316951, + "loss": 1.2925, + "theoretical_loss": 3.636580344887121, + "tokens_seen": 1037172736 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003463490471414243, + "loss": 1.4296, + "theoretical_loss": 3.6365586250929782, + "tokens_seen": 1037238272 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034633901705115344, + "loss": 1.3872, + "theoretical_loss": 3.6365369070553406, + "tokens_seen": 1037303808 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034632898696088267, + "loss": 1.5017, + "theoretical_loss": 3.636515190773956, + "tokens_seen": 1037369344 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003463189568706118, + "loss": 1.3186, + "theoretical_loss": 3.6364934762485706, + "tokens_seen": 1037434880 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034630892678034103, + "loss": 1.5026, + "theoretical_loss": 3.636471763478932, + "tokens_seen": 1037500416 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003462988966900702, + "loss": 1.3835, + "theoretical_loss": 3.6364500524647867, + "tokens_seen": 1037565952 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003462888665997994, + "loss": 1.4672, + "theoretical_loss": 3.636428343205883, + "tokens_seen": 1037631488 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003462788365095286, + "loss": 1.4753, + "theoretical_loss": 3.636406635701967, + "tokens_seen": 1037697024 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003462688064192578, + "loss": 1.4566, + "theoretical_loss": 3.6363849299527864, + "tokens_seen": 1037762560 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034625877632898694, + "loss": 1.3212, + "theoretical_loss": 3.636363225958089, + "tokens_seen": 1037828096 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003462487462387162, + "loss": 1.3678, + "theoretical_loss": 3.636341523717622, + "tokens_seen": 1037893632 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003462387161484453, + "loss": 1.5276, + "theoretical_loss": 3.6363198232311325, + "tokens_seen": 1037959168 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034622868605817454, + "loss": 1.4124, + "theoretical_loss": 3.636298124498368, + "tokens_seen": 1038024704 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003462186559679037, + "loss": 1.3048, + "theoretical_loss": 3.6362764275190766, + "tokens_seen": 1038090240 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003462086258776329, + "loss": 1.4267, + "theoretical_loss": 3.636254732293006, + "tokens_seen": 1038155776 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003461985957873621, + "loss": 1.407, + "theoretical_loss": 3.6362330388199027, + "tokens_seen": 1038221312 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034618856569709126, + "loss": 1.3878, + "theoretical_loss": 3.6362113470995157, + "tokens_seen": 1038286848 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034617853560682044, + "loss": 1.4369, + "theoretical_loss": 3.6361896571315926, + "tokens_seen": 1038352384 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003461685055165497, + "loss": 1.4212, + "theoretical_loss": 3.6361679689158803, + "tokens_seen": 1038417920 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003461584754262788, + "loss": 1.3423, + "theoretical_loss": 3.6361462824521276, + "tokens_seen": 1038483456 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034614844533600804, + "loss": 1.3319, + "theoretical_loss": 3.636124597740082, + "tokens_seen": 1038548992 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034613841524573717, + "loss": 1.417, + "theoretical_loss": 3.6361029147794914, + "tokens_seen": 1038614528 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003461283851554664, + "loss": 1.3298, + "theoretical_loss": 3.6360812335701045, + "tokens_seen": 1038680064 + }, + { + "epoch": 0.31, + "objective/train/advantage_avg": -0.00048074559890665114, + "objective/train/docs_used": 595196, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.529658555984497, + "objective/train/original_loss": 2.529658794403076, + "objective/train/theoretical_loss": 3.636059554111668, + "objective/train/tokens_used": 1059205600, + "objective/train/value_avg": -0.00598907470703125, + "objective/train/value_loss": 4.973800605512224e-05, + "objective/train/value_max": -0.00031757354736328125, + "objective/train/value_min": -0.04620361328125, + "objective/train/value_reward_corr": 0.44671766896874027, + "objective/train/value_std": 0.0049896240234375, + "objective/train/weight_avg": 0.9976010322570801, + "objective/train/weighted_lm_loss": 2.5200552940368652, + "objective/train/weights_max": 1.4967610836029053, + "objective/train/weights_min": 0.5235633254051208, + "theoretical_loss": 3.636059554111668, + "tokens_seen": 1038745600 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003461183550651956, + "loss": 1.2514, + "theoretical_loss": 3.636059554111668, + "tokens_seen": 1038745600 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034610832497492476, + "loss": 1.4101, + "theoretical_loss": 3.6360378764039316, + "tokens_seen": 1038811136 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034609829488465395, + "loss": 1.3585, + "theoretical_loss": 3.636016200446643, + "tokens_seen": 1038876672 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003460882647943832, + "loss": 1.3291, + "theoretical_loss": 3.63599452623955, + "tokens_seen": 1038942208 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034607823470411236, + "loss": 1.3387, + "theoretical_loss": 3.6359728537824014, + "tokens_seen": 1039007744 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034606820461384154, + "loss": 1.2858, + "theoretical_loss": 3.6359511830749454, + "tokens_seen": 1039073280 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003460581745235707, + "loss": 1.3034, + "theoretical_loss": 3.6359295141169303, + "tokens_seen": 1039138816 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003460481444332999, + "loss": 1.3019, + "theoretical_loss": 3.635907846908104, + "tokens_seen": 1039204352 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034603811434302914, + "loss": 1.3431, + "theoretical_loss": 3.635886181448216, + "tokens_seen": 1039269888 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034602808425275827, + "loss": 1.3594, + "theoretical_loss": 3.635864517737015, + "tokens_seen": 1039335424 + }, + { + "epoch": 0.31, + "learning_rate": 0.0003460180541624875, + "loss": 1.2989, + "theoretical_loss": 3.635842855774248, + "tokens_seen": 1039400960 + }, + { + "epoch": 0.31, + "learning_rate": 0.00034600802407221663, + "loss": 1.2797, + "theoretical_loss": 3.6358211955596653, + "tokens_seen": 1039466496 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034599799398194587, + "loss": 1.4222, + "theoretical_loss": 3.635799537093015, + "tokens_seen": 1039532032 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034598796389167505, + "loss": 1.3292, + "theoretical_loss": 3.6357778803740457, + "tokens_seen": 1039597568 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034597793380140423, + "loss": 1.3448, + "theoretical_loss": 3.635756225402506, + "tokens_seen": 1039663104 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003459679037111334, + "loss": 1.3837, + "theoretical_loss": 3.635734572178146, + "tokens_seen": 1039728640 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034595787362086264, + "loss": 1.2297, + "theoretical_loss": 3.6357129207007137, + "tokens_seen": 1039794176 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034594784353059177, + "loss": 1.3345, + "theoretical_loss": 3.635691270969958, + "tokens_seen": 1039859712 + }, + { + "epoch": 0.32, + "learning_rate": 0.000345937813440321, + "loss": 1.3126, + "theoretical_loss": 3.6356696229856276, + "tokens_seen": 1039925248 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034592778335005013, + "loss": 1.3523, + "theoretical_loss": 3.6356479767474723, + "tokens_seen": 1039990784 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034591775325977937, + "loss": 1.3918, + "theoretical_loss": 3.6356263322552413, + "tokens_seen": 1040056320 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034590772316950855, + "loss": 1.3895, + "theoretical_loss": 3.635604689508683, + "tokens_seen": 1040121856 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034589769307923773, + "loss": 1.3048, + "theoretical_loss": 3.6355830485075473, + "tokens_seen": 1040187392 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003458876629889669, + "loss": 1.3521, + "theoretical_loss": 3.6355614092515833, + "tokens_seen": 1040252928 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003458776328986961, + "loss": 1.2757, + "theoretical_loss": 3.6355397717405404, + "tokens_seen": 1040318464 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": 0.00048197846626862884, + "objective/train/docs_used": 596146, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.992327928543091, + "objective/train/original_loss": 2.992327928543091, + "objective/train/theoretical_loss": 3.6355181359741673, + "objective/train/tokens_used": 1060844000, + "objective/train/value_avg": -0.005008697509765625, + "objective/train/value_loss": 3.890653533744626e-05, + "objective/train/value_max": -0.0002397298812866211, + "objective/train/value_min": -0.045440673828125, + "objective/train/value_reward_corr": 0.31795999717227674, + "objective/train/value_std": 0.0038661956787109375, + "objective/train/weight_avg": 1.0067194700241089, + "objective/train/weighted_lm_loss": 3.010608434677124, + "objective/train/weights_max": 1.479799747467041, + "objective/train/weights_min": 0.5157667398452759, + "theoretical_loss": 3.6355181359741673, + "tokens_seen": 1040384000 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003458676028084253, + "loss": 1.4634, + "theoretical_loss": 3.6355181359741673, + "tokens_seen": 1040384000 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003458575727181545, + "loss": 1.3663, + "theoretical_loss": 3.6354965019522147, + "tokens_seen": 1040449536 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034584754262788364, + "loss": 1.4618, + "theoretical_loss": 3.635474869674431, + "tokens_seen": 1040515072 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034583751253761287, + "loss": 1.4471, + "theoretical_loss": 3.635453239140566, + "tokens_seen": 1040580608 + }, + { + "epoch": 0.32, + "learning_rate": 0.000345827482447342, + "loss": 1.3848, + "theoretical_loss": 3.63543161035037, + "tokens_seen": 1040646144 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034581745235707123, + "loss": 1.433, + "theoretical_loss": 3.6354099833035916, + "tokens_seen": 1040711680 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003458074222668004, + "loss": 1.399, + "theoretical_loss": 3.6353883579999815, + "tokens_seen": 1040777216 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003457973921765296, + "loss": 1.3749, + "theoretical_loss": 3.6353667344392884, + "tokens_seen": 1040842752 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003457873620862588, + "loss": 1.2976, + "theoretical_loss": 3.6353451126212626, + "tokens_seen": 1040908288 + }, + { + "epoch": 0.32, + "learning_rate": 0.000345777331995988, + "loss": 1.3868, + "theoretical_loss": 3.6353234925456546, + "tokens_seen": 1040973824 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034576730190571714, + "loss": 1.4112, + "theoretical_loss": 3.635301874212213, + "tokens_seen": 1041039360 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003457572718154464, + "loss": 1.4108, + "theoretical_loss": 3.635280257620688, + "tokens_seen": 1041104896 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003457472417251755, + "loss": 1.275, + "theoretical_loss": 3.6352586427708315, + "tokens_seen": 1041170432 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034573721163490474, + "loss": 1.3507, + "theoretical_loss": 3.635237029662391, + "tokens_seen": 1041235968 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003457271815446339, + "loss": 1.3949, + "theoretical_loss": 3.635215418295118, + "tokens_seen": 1041301504 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003457171514543631, + "loss": 1.4436, + "theoretical_loss": 3.6351938086687623, + "tokens_seen": 1041367040 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003457071213640923, + "loss": 1.4078, + "theoretical_loss": 3.6351722007830745, + "tokens_seen": 1041432576 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034569709127382146, + "loss": 1.4244, + "theoretical_loss": 3.635150594637804, + "tokens_seen": 1041498112 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034568706118355064, + "loss": 1.3879, + "theoretical_loss": 3.635128990232702, + "tokens_seen": 1041563648 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003456770310932799, + "loss": 1.4346, + "theoretical_loss": 3.635107387567518, + "tokens_seen": 1041629184 + }, + { + "epoch": 0.32, + "learning_rate": 0.000345667001003009, + "loss": 1.3914, + "theoretical_loss": 3.6350857866420037, + "tokens_seen": 1041694720 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034565697091273824, + "loss": 1.4125, + "theoretical_loss": 3.6350641874559084, + "tokens_seen": 1041760256 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034564694082246737, + "loss": 1.3193, + "theoretical_loss": 3.6350425900089833, + "tokens_seen": 1041825792 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003456369107321966, + "loss": 1.3593, + "theoretical_loss": 3.6350209943009784, + "tokens_seen": 1041891328 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003456268806419258, + "loss": 1.3016, + "theoretical_loss": 3.634999400331645, + "tokens_seen": 1041956864 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": 1.4156539691612124e-06, + "objective/train/docs_used": 596810, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.235262393951416, + "objective/train/original_loss": 3.235261917114258, + "objective/train/theoretical_loss": 3.6349778081007327, + "objective/train/tokens_used": 1062482400, + "objective/train/value_avg": -0.0053558349609375, + "objective/train/value_loss": 4.5431635953718796e-05, + "objective/train/value_max": -0.0004878044128417969, + "objective/train/value_min": -0.054412841796875, + "objective/train/value_reward_corr": 0.4488369025420771, + "objective/train/value_std": 0.004749298095703125, + "objective/train/weight_avg": 1.0022358894348145, + "objective/train/weighted_lm_loss": 3.2512824535369873, + "objective/train/weights_max": 1.5807849168777466, + "objective/train/weights_min": 0.5721757411956787, + "theoretical_loss": 3.6349778081007327, + "tokens_seen": 1042022400 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034561685055165496, + "loss": 1.4198, + "theoretical_loss": 3.6349778081007327, + "tokens_seen": 1042022400 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034560682046138415, + "loss": 1.4256, + "theoretical_loss": 3.6349562176079933, + "tokens_seen": 1042087936 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003455967903711134, + "loss": 1.2869, + "theoretical_loss": 3.634934628853177, + "tokens_seen": 1042153472 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003455867602808425, + "loss": 1.3349, + "theoretical_loss": 3.634913041836035, + "tokens_seen": 1042219008 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034557673019057174, + "loss": 1.4313, + "theoretical_loss": 3.6348914565563186, + "tokens_seen": 1042284544 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034556670010030087, + "loss": 1.4244, + "theoretical_loss": 3.6348698730137774, + "tokens_seen": 1042350080 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003455566700100301, + "loss": 1.4447, + "theoretical_loss": 3.634848291208163, + "tokens_seen": 1042415616 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003455466399197593, + "loss": 1.2726, + "theoretical_loss": 3.6348267111392274, + "tokens_seen": 1042481152 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034553660982948847, + "loss": 1.3931, + "theoretical_loss": 3.6348051328067204, + "tokens_seen": 1042546688 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034552657973921765, + "loss": 1.2725, + "theoretical_loss": 3.6347835562103934, + "tokens_seen": 1042612224 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034551654964894683, + "loss": 1.426, + "theoretical_loss": 3.634761981349998, + "tokens_seen": 1042677760 + }, + { + "epoch": 0.32, + "learning_rate": 0.000345506519558676, + "loss": 1.32, + "theoretical_loss": 3.634740408225285, + "tokens_seen": 1042743296 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034549648946840525, + "loss": 1.2692, + "theoretical_loss": 3.634718836836006, + "tokens_seen": 1042808832 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003454864593781344, + "loss": 1.3023, + "theoretical_loss": 3.634697267181912, + "tokens_seen": 1042874368 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003454764292878636, + "loss": 1.3043, + "theoretical_loss": 3.634675699262756, + "tokens_seen": 1042939904 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034546639919759274, + "loss": 1.3596, + "theoretical_loss": 3.6346541330782864, + "tokens_seen": 1043005440 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034545636910732197, + "loss": 1.3582, + "theoretical_loss": 3.6346325686282572, + "tokens_seen": 1043070976 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034544633901705115, + "loss": 1.4281, + "theoretical_loss": 3.634611005912419, + "tokens_seen": 1043136512 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034543630892678033, + "loss": 1.452, + "theoretical_loss": 3.6345894449305236, + "tokens_seen": 1043202048 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003454262788365095, + "loss": 1.4316, + "theoretical_loss": 3.6345678856823227, + "tokens_seen": 1043267584 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034541624874623875, + "loss": 1.5111, + "theoretical_loss": 3.6345463281675676, + "tokens_seen": 1043333120 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003454062186559679, + "loss": 1.3641, + "theoretical_loss": 3.6345247723860106, + "tokens_seen": 1043398656 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003453961885656971, + "loss": 1.4542, + "theoretical_loss": 3.634503218337403, + "tokens_seen": 1043464192 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034538615847542624, + "loss": 1.3341, + "theoretical_loss": 3.6344816660214967, + "tokens_seen": 1043529728 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003453761283851555, + "loss": 1.3817, + "theoretical_loss": 3.6344601154380443, + "tokens_seen": 1043595264 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.0008375108591280878, + "objective/train/docs_used": 598197, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7524189949035645, + "objective/train/original_loss": 2.7524189949035645, + "objective/train/theoretical_loss": 3.6344385665867973, + "objective/train/tokens_used": 1064120800, + "objective/train/value_avg": -0.005382537841796875, + "objective/train/value_loss": 0.00010457867756485939, + "objective/train/value_max": -0.00023233890533447266, + "objective/train/value_min": -0.0999755859375, + "objective/train/value_reward_corr": 0.3617730504458602, + "objective/train/value_std": 0.0049896240234375, + "objective/train/weight_avg": 0.9962438941001892, + "objective/train/weighted_lm_loss": 2.7311360836029053, + "objective/train/weights_max": 2.1418299674987793, + "objective/train/weights_min": 0.2804151475429535, + "theoretical_loss": 3.6344385665867973, + "tokens_seen": 1043660800 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034536609829488466, + "loss": 1.4589, + "theoretical_loss": 3.6344385665867973, + "tokens_seen": 1043660800 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034535606820461384, + "loss": 1.4323, + "theoretical_loss": 3.6344170194675076, + "tokens_seen": 1043726336 + }, + { + "epoch": 0.32, + "learning_rate": 0.000345346038114343, + "loss": 1.2421, + "theoretical_loss": 3.634395474079927, + "tokens_seen": 1043791872 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003453360080240722, + "loss": 1.342, + "theoretical_loss": 3.6343739304238083, + "tokens_seen": 1043857408 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034532597793380143, + "loss": 1.426, + "theoretical_loss": 3.634352388498903, + "tokens_seen": 1043922944 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003453159478435306, + "loss": 1.435, + "theoretical_loss": 3.6343308483049634, + "tokens_seen": 1043988480 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003453059177532598, + "loss": 1.3372, + "theoretical_loss": 3.6343093098417425, + "tokens_seen": 1044054016 + }, + { + "epoch": 0.32, + "learning_rate": 0.000345295887662989, + "loss": 1.3835, + "theoretical_loss": 3.634287773108992, + "tokens_seen": 1044119552 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003452858575727182, + "loss": 1.4473, + "theoretical_loss": 3.6342662381064645, + "tokens_seen": 1044185088 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034527582748244734, + "loss": 1.3635, + "theoretical_loss": 3.634244704833912, + "tokens_seen": 1044250624 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003452657973921766, + "loss": 1.3519, + "theoretical_loss": 3.6342231732910872, + "tokens_seen": 1044316160 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003452557673019057, + "loss": 1.4535, + "theoretical_loss": 3.6342016434777427, + "tokens_seen": 1044381696 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034524573721163494, + "loss": 1.4198, + "theoretical_loss": 3.6341801153936313, + "tokens_seen": 1044447232 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003452357071213641, + "loss": 1.4596, + "theoretical_loss": 3.634158589038505, + "tokens_seen": 1044512768 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003452256770310933, + "loss": 1.397, + "theoretical_loss": 3.634137064412117, + "tokens_seen": 1044578304 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003452156469408225, + "loss": 1.4376, + "theoretical_loss": 3.6341155415142197, + "tokens_seen": 1044643840 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034520561685055166, + "loss": 1.3738, + "theoretical_loss": 3.634094020344566, + "tokens_seen": 1044709376 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034519558676028084, + "loss": 1.2686, + "theoretical_loss": 3.634072500902909, + "tokens_seen": 1044774912 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003451855566700101, + "loss": 1.3805, + "theoretical_loss": 3.6340509831890015, + "tokens_seen": 1044840448 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003451755265797392, + "loss": 1.265, + "theoretical_loss": 3.6340294672025957, + "tokens_seen": 1044905984 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034516549648946844, + "loss": 1.3877, + "theoretical_loss": 3.6340079529434455, + "tokens_seen": 1044971520 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034515546639919757, + "loss": 1.3143, + "theoretical_loss": 3.633986440411303, + "tokens_seen": 1045037056 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003451454363089268, + "loss": 1.2084, + "theoretical_loss": 3.6339649296059227, + "tokens_seen": 1045102592 + }, + { + "epoch": 0.32, + "learning_rate": 0.000345135406218656, + "loss": 1.4108, + "theoretical_loss": 3.633943420527056, + "tokens_seen": 1045168128 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034512537612838517, + "loss": 1.3774, + "theoretical_loss": 3.6339219131744573, + "tokens_seen": 1045233664 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.002577277598902583, + "objective/train/docs_used": 598880, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.005614995956421, + "objective/train/original_loss": 2.005614757537842, + "objective/train/theoretical_loss": 3.6339004075478796, + "objective/train/tokens_used": 1065759200, + "objective/train/value_avg": -0.00452423095703125, + "objective/train/value_loss": 9.155476436717436e-05, + "objective/train/value_max": -0.0004372596740722656, + "objective/train/value_min": -0.07305908203125, + "objective/train/value_reward_corr": 0.26348444262584164, + "objective/train/value_std": 0.0035419464111328125, + "objective/train/weight_avg": 0.9784821271896362, + "objective/train/weighted_lm_loss": 1.979283094406128, + "objective/train/weights_max": 1.855103850364685, + "objective/train/weights_min": 0.34264618158340454, + "theoretical_loss": 3.6339004075478796, + "tokens_seen": 1045299200 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034511534603811435, + "loss": 1.2349, + "theoretical_loss": 3.6339004075478796, + "tokens_seen": 1045299200 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003451053159478436, + "loss": 1.3118, + "theoretical_loss": 3.6338789036470756, + "tokens_seen": 1045364736 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003450952858575727, + "loss": 1.4393, + "theoretical_loss": 3.6338574014717997, + "tokens_seen": 1045430272 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034508525576730194, + "loss": 1.3196, + "theoretical_loss": 3.633835901021804, + "tokens_seen": 1045495808 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034507522567703107, + "loss": 1.3632, + "theoretical_loss": 3.633814402296843, + "tokens_seen": 1045561344 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003450651955867603, + "loss": 1.3249, + "theoretical_loss": 3.6337929052966693, + "tokens_seen": 1045626880 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003450551654964895, + "loss": 1.3915, + "theoretical_loss": 3.633771410021038, + "tokens_seen": 1045692416 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034504513540621867, + "loss": 1.2311, + "theoretical_loss": 3.6337499164697005, + "tokens_seen": 1045757952 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034503510531594785, + "loss": 1.4729, + "theoretical_loss": 3.633728424642412, + "tokens_seen": 1045823488 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034502507522567703, + "loss": 1.3321, + "theoretical_loss": 3.6337069345389255, + "tokens_seen": 1045889024 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003450150451354062, + "loss": 1.4404, + "theoretical_loss": 3.633685446158995, + "tokens_seen": 1045954560 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034500501504513545, + "loss": 1.3193, + "theoretical_loss": 3.6336639595023743, + "tokens_seen": 1046020096 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003449949849548646, + "loss": 1.3173, + "theoretical_loss": 3.633642474568817, + "tokens_seen": 1046085632 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003449849548645938, + "loss": 1.4027, + "theoretical_loss": 3.6336209913580775, + "tokens_seen": 1046151168 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034497492477432294, + "loss": 1.2854, + "theoretical_loss": 3.633599509869909, + "tokens_seen": 1046216704 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034496489468405217, + "loss": 1.4772, + "theoretical_loss": 3.6335780301040663, + "tokens_seen": 1046282240 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034495486459378135, + "loss": 1.329, + "theoretical_loss": 3.633556552060303, + "tokens_seen": 1046347776 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034494483450351053, + "loss": 1.3298, + "theoretical_loss": 3.633535075738373, + "tokens_seen": 1046413312 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003449348044132397, + "loss": 1.3462, + "theoretical_loss": 3.6335136011380307, + "tokens_seen": 1046478848 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034492477432296895, + "loss": 1.4697, + "theoretical_loss": 3.6334921282590305, + "tokens_seen": 1046544384 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003449147442326981, + "loss": 1.3208, + "theoretical_loss": 3.633470657101126, + "tokens_seen": 1046609920 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003449047141424273, + "loss": 1.2913, + "theoretical_loss": 3.6334491876640724, + "tokens_seen": 1046675456 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034489468405215644, + "loss": 1.4608, + "theoretical_loss": 3.633427719947623, + "tokens_seen": 1046740992 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003448846539618857, + "loss": 1.4148, + "theoretical_loss": 3.633406253951533, + "tokens_seen": 1046806528 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034487462387161486, + "loss": 1.3579, + "theoretical_loss": 3.6333847896755564, + "tokens_seen": 1046872064 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": 0.0004699326818808913, + "objective/train/docs_used": 599993, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5044503211975098, + "objective/train/original_loss": 2.5044503211975098, + "objective/train/theoretical_loss": 3.633363327119448, + "objective/train/tokens_used": 1067397600, + "objective/train/value_avg": -0.00499725341796875, + "objective/train/value_loss": 4.8452286137035117e-05, + "objective/train/value_max": -0.0004601478576660156, + "objective/train/value_min": -0.05499267578125, + "objective/train/value_reward_corr": 0.2689462968212912, + "objective/train/value_std": 0.00353240966796875, + "objective/train/weight_avg": 1.006962776184082, + "objective/train/weighted_lm_loss": 2.5258851051330566, + "objective/train/weights_max": 1.3977627754211426, + "objective/train/weights_min": 0.3547496795654297, + "theoretical_loss": 3.633363327119448, + "tokens_seen": 1046937600 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034486459378134404, + "loss": 1.3267, + "theoretical_loss": 3.633363327119448, + "tokens_seen": 1046937600 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003448545636910732, + "loss": 1.3701, + "theoretical_loss": 3.6333418662829624, + "tokens_seen": 1047003136 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003448445336008024, + "loss": 1.3699, + "theoretical_loss": 3.633320407165854, + "tokens_seen": 1047068672 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003448345035105316, + "loss": 1.2666, + "theoretical_loss": 3.633298949767877, + "tokens_seen": 1047134208 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003448244734202608, + "loss": 1.3679, + "theoretical_loss": 3.633277494088787, + "tokens_seen": 1047199744 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034481444332998994, + "loss": 1.3379, + "theoretical_loss": 3.6332560401283382, + "tokens_seen": 1047265280 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003448044132397192, + "loss": 1.404, + "theoretical_loss": 3.6332345878862853, + "tokens_seen": 1047330816 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034479438314944836, + "loss": 1.3048, + "theoretical_loss": 3.6332131373623833, + "tokens_seen": 1047396352 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034478435305917754, + "loss": 1.4463, + "theoretical_loss": 3.633191688556388, + "tokens_seen": 1047461888 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003447743229689067, + "loss": 1.3798, + "theoretical_loss": 3.6331702414680525, + "tokens_seen": 1047527424 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003447642928786359, + "loss": 1.3187, + "theoretical_loss": 3.6331487960971334, + "tokens_seen": 1047592960 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003447542627883651, + "loss": 1.3209, + "theoretical_loss": 3.6331273524433847, + "tokens_seen": 1047658496 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003447442326980943, + "loss": 1.3574, + "theoretical_loss": 3.633105910506562, + "tokens_seen": 1047724032 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034473420260782345, + "loss": 1.3404, + "theoretical_loss": 3.6330844702864202, + "tokens_seen": 1047789568 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003447241725175527, + "loss": 1.4428, + "theoretical_loss": 3.633063031782715, + "tokens_seen": 1047855104 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003447141424272818, + "loss": 1.4221, + "theoretical_loss": 3.633041594995201, + "tokens_seen": 1047920640 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034470411233701104, + "loss": 1.5308, + "theoretical_loss": 3.6330201599236345, + "tokens_seen": 1047986176 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003446940822467402, + "loss": 1.3491, + "theoretical_loss": 3.63299872656777, + "tokens_seen": 1048051712 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003446840521564694, + "loss": 1.3546, + "theoretical_loss": 3.6329772949273624, + "tokens_seen": 1048117248 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003446740220661986, + "loss": 1.3028, + "theoretical_loss": 3.6329558650021685, + "tokens_seen": 1048182784 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034466399197592777, + "loss": 1.4142, + "theoretical_loss": 3.6329344367919427, + "tokens_seen": 1048248320 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034465396188565695, + "loss": 1.2606, + "theoretical_loss": 3.6329130102964413, + "tokens_seen": 1048313856 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003446439317953862, + "loss": 1.3942, + "theoretical_loss": 3.6328915855154196, + "tokens_seen": 1048379392 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003446339017051153, + "loss": 1.4301, + "theoretical_loss": 3.632870162448633, + "tokens_seen": 1048444928 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034462387161484455, + "loss": 1.3708, + "theoretical_loss": 3.632848741095838, + "tokens_seen": 1048510464 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.0009832927025854588, + "objective/train/docs_used": 601088, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8934741020202637, + "objective/train/original_loss": 2.8934738636016846, + "objective/train/theoretical_loss": 3.632827321456789, + "objective/train/tokens_used": 1069036000, + "objective/train/value_avg": -0.005420684814453125, + "objective/train/value_loss": 7.863714563427493e-05, + "objective/train/value_max": -0.00033926963806152344, + "objective/train/value_min": -0.115966796875, + "objective/train/value_reward_corr": 0.44384499135806177, + "objective/train/value_std": 0.005107879638671875, + "objective/train/weight_avg": 0.9937934875488281, + "objective/train/weighted_lm_loss": 2.8844664096832275, + "objective/train/weights_max": 1.922615647315979, + "objective/train/weights_min": 0.242025688290596, + "theoretical_loss": 3.632827321456789, + "tokens_seen": 1048576000 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034461384152457373, + "loss": 1.3978, + "theoretical_loss": 3.632827321456789, + "tokens_seen": 1048576000 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003446038114343029, + "loss": 1.2463, + "theoretical_loss": 3.632805903531243, + "tokens_seen": 1048641536 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034459378134403214, + "loss": 1.3602, + "theoretical_loss": 3.6327844873189554, + "tokens_seen": 1048707072 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034458375125376127, + "loss": 1.3638, + "theoretical_loss": 3.632763072819682, + "tokens_seen": 1048772608 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003445737211634905, + "loss": 1.4006, + "theoretical_loss": 3.632741660033179, + "tokens_seen": 1048838144 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003445636910732197, + "loss": 1.3352, + "theoretical_loss": 3.6327202489592025, + "tokens_seen": 1048903680 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034455366098294887, + "loss": 1.3902, + "theoretical_loss": 3.632698839597508, + "tokens_seen": 1048969216 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034454363089267805, + "loss": 1.3805, + "theoretical_loss": 3.632677431947852, + "tokens_seen": 1049034752 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034453360080240723, + "loss": 1.4114, + "theoretical_loss": 3.6326560260099905, + "tokens_seen": 1049100288 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003445235707121364, + "loss": 1.3337, + "theoretical_loss": 3.63263462178368, + "tokens_seen": 1049165824 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034451354062186565, + "loss": 1.3389, + "theoretical_loss": 3.632613219268676, + "tokens_seen": 1049231360 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003445035105315948, + "loss": 1.4379, + "theoretical_loss": 3.6325918184647366, + "tokens_seen": 1049296896 + }, + { + "epoch": 0.32, + "learning_rate": 0.000344493480441324, + "loss": 1.2568, + "theoretical_loss": 3.632570419371616, + "tokens_seen": 1049362432 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034448345035105314, + "loss": 1.4402, + "theoretical_loss": 3.632549021989071, + "tokens_seen": 1049427968 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034447342026078237, + "loss": 1.3858, + "theoretical_loss": 3.6325276263168593, + "tokens_seen": 1049493504 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034446339017051155, + "loss": 1.3537, + "theoretical_loss": 3.6325062323547366, + "tokens_seen": 1049559040 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034445336008024073, + "loss": 1.3827, + "theoretical_loss": 3.6324848401024594, + "tokens_seen": 1049624576 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003444433299899699, + "loss": 1.3433, + "theoretical_loss": 3.6324634495597845, + "tokens_seen": 1049690112 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034443329989969915, + "loss": 1.2679, + "theoretical_loss": 3.6324420607264685, + "tokens_seen": 1049755648 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003444232698094283, + "loss": 1.3206, + "theoretical_loss": 3.632420673602268, + "tokens_seen": 1049821184 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003444132397191575, + "loss": 1.4593, + "theoretical_loss": 3.6323992881869396, + "tokens_seen": 1049886720 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034440320962888664, + "loss": 1.4035, + "theoretical_loss": 3.6323779044802404, + "tokens_seen": 1049952256 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003443931795386159, + "loss": 1.48, + "theoretical_loss": 3.632356522481927, + "tokens_seen": 1050017792 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034438314944834506, + "loss": 1.4032, + "theoretical_loss": 3.6323351421917565, + "tokens_seen": 1050083328 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034437311935807424, + "loss": 1.5778, + "theoretical_loss": 3.6323137636094858, + "tokens_seen": 1050148864 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.0008290043915621936, + "objective/train/docs_used": 601638, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7351326942443848, + "objective/train/original_loss": 2.7351322174072266, + "objective/train/theoretical_loss": 3.632292386734872, + "objective/train/tokens_used": 1070674400, + "objective/train/value_avg": -0.00490570068359375, + "objective/train/value_loss": 6.606072565773502e-05, + "objective/train/value_max": -0.0002982616424560547, + "objective/train/value_min": -0.052337646484375, + "objective/train/value_reward_corr": 0.3220477356054372, + "objective/train/value_std": 0.003780364990234375, + "objective/train/weight_avg": 0.9946095943450928, + "objective/train/weighted_lm_loss": 2.717972755432129, + "objective/train/weights_max": 1.424458622932434, + "objective/train/weights_min": 0.0846383348107338, + "theoretical_loss": 3.632292386734872, + "tokens_seen": 1050214400 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003443630892678034, + "loss": 1.282, + "theoretical_loss": 3.632292386734872, + "tokens_seen": 1050214400 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003443530591775326, + "loss": 1.2881, + "theoretical_loss": 3.6322710115676724, + "tokens_seen": 1050279936 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003443430290872618, + "loss": 1.4049, + "theoretical_loss": 3.632249638107643, + "tokens_seen": 1050345472 + }, + { + "epoch": 0.32, + "learning_rate": 0.000344332998996991, + "loss": 1.4098, + "theoretical_loss": 3.6322282663545424, + "tokens_seen": 1050411008 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034432296890672014, + "loss": 1.3825, + "theoretical_loss": 3.6322068963081264, + "tokens_seen": 1050476544 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003443129388164494, + "loss": 1.3675, + "theoretical_loss": 3.6321855279681534, + "tokens_seen": 1050542080 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034430290872617856, + "loss": 1.4761, + "theoretical_loss": 3.63216416133438, + "tokens_seen": 1050607616 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034429287863590774, + "loss": 1.379, + "theoretical_loss": 3.632142796406564, + "tokens_seen": 1050673152 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003442828485456369, + "loss": 1.472, + "theoretical_loss": 3.6321214331844622, + "tokens_seen": 1050738688 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003442728184553661, + "loss": 1.3697, + "theoretical_loss": 3.6321000716678324, + "tokens_seen": 1050804224 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003442627883650953, + "loss": 1.3416, + "theoretical_loss": 3.6320787118564324, + "tokens_seen": 1050869760 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003442527582748245, + "loss": 1.407, + "theoretical_loss": 3.6320573537500196, + "tokens_seen": 1050935296 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034424272818455365, + "loss": 1.4521, + "theoretical_loss": 3.6320359973483507, + "tokens_seen": 1051000832 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003442326980942829, + "loss": 1.3303, + "theoretical_loss": 3.632014642651185, + "tokens_seen": 1051066368 + }, + { + "epoch": 0.32, + "learning_rate": 0.000344222668004012, + "loss": 1.3626, + "theoretical_loss": 3.6319932896582787, + "tokens_seen": 1051131904 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034421263791374124, + "loss": 1.2684, + "theoretical_loss": 3.6319719383693903, + "tokens_seen": 1051197440 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003442026078234704, + "loss": 1.4775, + "theoretical_loss": 3.631950588784278, + "tokens_seen": 1051262976 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003441925777331996, + "loss": 1.4173, + "theoretical_loss": 3.631929240902698, + "tokens_seen": 1051328512 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003441825476429288, + "loss": 1.4383, + "theoretical_loss": 3.63190789472441, + "tokens_seen": 1051394048 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034417251755265797, + "loss": 1.3963, + "theoretical_loss": 3.6318865502491713, + "tokens_seen": 1051459584 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034416248746238715, + "loss": 1.3982, + "theoretical_loss": 3.6318652074767392, + "tokens_seen": 1051525120 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003441524573721164, + "loss": 1.3812, + "theoretical_loss": 3.6318438664068724, + "tokens_seen": 1051590656 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003441424272818455, + "loss": 1.454, + "theoretical_loss": 3.631822527039329, + "tokens_seen": 1051656192 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034413239719157475, + "loss": 1.2411, + "theoretical_loss": 3.631801189373867, + "tokens_seen": 1051721728 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034412236710130393, + "loss": 1.3901, + "theoretical_loss": 3.6317798534102455, + "tokens_seen": 1051787264 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": 0.0004239085828885436, + "objective/train/docs_used": 602388, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3964009284973145, + "objective/train/original_loss": 2.3964006900787354, + "objective/train/theoretical_loss": 3.631758519148221, + "objective/train/tokens_used": 1072312800, + "objective/train/value_avg": -0.004184722900390625, + "objective/train/value_loss": 3.3116870326921344e-05, + "objective/train/value_max": -0.0002453327178955078, + "objective/train/value_min": -0.11279296875, + "objective/train/value_reward_corr": 0.43994117873975874, + "objective/train/value_std": 0.004299163818359375, + "objective/train/weight_avg": 1.0058561563491821, + "objective/train/weighted_lm_loss": 2.4167473316192627, + "objective/train/weights_max": 2.3010807037353516, + "objective/train/weights_min": 0.5184604525566101, + "theoretical_loss": 3.631758519148221, + "tokens_seen": 1051852800 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003441123370110331, + "loss": 1.3282, + "theoretical_loss": 3.631758519148221, + "tokens_seen": 1051852800 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003441023069207623, + "loss": 1.168, + "theoretical_loss": 3.6317371865875527, + "tokens_seen": 1051918336 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034409227683049147, + "loss": 1.4785, + "theoretical_loss": 3.631715855727999, + "tokens_seen": 1051983872 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034408224674022065, + "loss": 1.3199, + "theoretical_loss": 3.6316945265693183, + "tokens_seen": 1052049408 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003440722166499499, + "loss": 1.3906, + "theoretical_loss": 3.631673199111269, + "tokens_seen": 1052114944 + }, + { + "epoch": 0.32, + "learning_rate": 0.000344062186559679, + "loss": 1.3893, + "theoretical_loss": 3.6316518733536096, + "tokens_seen": 1052180480 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034405215646940825, + "loss": 1.4862, + "theoretical_loss": 3.631630549296098, + "tokens_seen": 1052246016 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003440421263791374, + "loss": 1.3585, + "theoretical_loss": 3.631609226938494, + "tokens_seen": 1052311552 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003440320962888666, + "loss": 1.3308, + "theoretical_loss": 3.631587906280555, + "tokens_seen": 1052377088 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003440220661985958, + "loss": 1.3492, + "theoretical_loss": 3.6315665873220406, + "tokens_seen": 1052442624 + }, + { + "epoch": 0.32, + "learning_rate": 0.000344012036108325, + "loss": 1.3958, + "theoretical_loss": 3.631545270062709, + "tokens_seen": 1052508160 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034400200601805416, + "loss": 1.3501, + "theoretical_loss": 3.6315239545023195, + "tokens_seen": 1052573696 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034399197592778334, + "loss": 1.3663, + "theoretical_loss": 3.63150264064063, + "tokens_seen": 1052639232 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003439819458375125, + "loss": 1.4731, + "theoretical_loss": 3.6314813284774003, + "tokens_seen": 1052704768 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034397191574724175, + "loss": 1.3877, + "theoretical_loss": 3.631460018012389, + "tokens_seen": 1052770304 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003439618856569709, + "loss": 1.378, + "theoretical_loss": 3.631438709245355, + "tokens_seen": 1052835840 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003439518555667001, + "loss": 1.412, + "theoretical_loss": 3.6314174021760577, + "tokens_seen": 1052901376 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003439418254764293, + "loss": 1.3889, + "theoretical_loss": 3.631396096804256, + "tokens_seen": 1052966912 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003439317953861585, + "loss": 1.3441, + "theoretical_loss": 3.631374793129708, + "tokens_seen": 1053032448 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034392176529588766, + "loss": 1.399, + "theoretical_loss": 3.6313534911521743, + "tokens_seen": 1053097984 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034391173520561684, + "loss": 1.2237, + "theoretical_loss": 3.6313321908714133, + "tokens_seen": 1053163520 + }, + { + "epoch": 0.32, + "learning_rate": 0.000343901705115346, + "loss": 1.4796, + "theoretical_loss": 3.6313108922871855, + "tokens_seen": 1053229056 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034389167502507526, + "loss": 1.3842, + "theoretical_loss": 3.6312895953992483, + "tokens_seen": 1053294592 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003438816449348044, + "loss": 1.4333, + "theoretical_loss": 3.6312683002073625, + "tokens_seen": 1053360128 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003438716148445336, + "loss": 1.4466, + "theoretical_loss": 3.6312470067112868, + "tokens_seen": 1053425664 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.0004100018704775721, + "objective/train/docs_used": 603451, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.055126667022705, + "objective/train/original_loss": 3.055126905441284, + "objective/train/theoretical_loss": 3.6312257149107814, + "objective/train/tokens_used": 1073951200, + "objective/train/value_avg": -0.0059356689453125, + "objective/train/value_loss": 6.389611371560022e-05, + "objective/train/value_max": -0.00033020973205566406, + "objective/train/value_min": -0.05975341796875, + "objective/train/value_reward_corr": 0.6057980007761059, + "objective/train/value_std": 0.005680084228515625, + "objective/train/weight_avg": 0.9989879727363586, + "objective/train/weighted_lm_loss": 3.040876626968384, + "objective/train/weights_max": 1.638988971710205, + "objective/train/weights_min": 0.5770580768585205, + "theoretical_loss": 3.6312257149107814, + "tokens_seen": 1053491200 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034386158475426275, + "loss": 1.3844, + "theoretical_loss": 3.6312257149107814, + "tokens_seen": 1053491200 + }, + { + "epoch": 0.32, + "learning_rate": 0.000343851554663992, + "loss": 1.4307, + "theoretical_loss": 3.6312044248056052, + "tokens_seen": 1053556736 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003438415245737212, + "loss": 1.4512, + "theoretical_loss": 3.631183136395518, + "tokens_seen": 1053622272 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034383149448345034, + "loss": 1.3789, + "theoretical_loss": 3.631161849680279, + "tokens_seen": 1053687808 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003438214643931796, + "loss": 1.4275, + "theoretical_loss": 3.631140564659648, + "tokens_seen": 1053753344 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034381143430290876, + "loss": 1.4788, + "theoretical_loss": 3.631119281333386, + "tokens_seen": 1053818880 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034380140421263794, + "loss": 1.3853, + "theoretical_loss": 3.631097999701251, + "tokens_seen": 1053884416 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003437913741223671, + "loss": 1.3964, + "theoretical_loss": 3.6310767197630036, + "tokens_seen": 1053949952 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003437813440320963, + "loss": 1.3745, + "theoretical_loss": 3.6310554415184035, + "tokens_seen": 1054015488 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003437713139418255, + "loss": 1.3845, + "theoretical_loss": 3.631034164967211, + "tokens_seen": 1054081024 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003437612838515547, + "loss": 1.3793, + "theoretical_loss": 3.6310128901091856, + "tokens_seen": 1054146560 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034375125376128385, + "loss": 1.553, + "theoretical_loss": 3.6309916169440872, + "tokens_seen": 1054212096 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003437412236710131, + "loss": 1.3619, + "theoretical_loss": 3.630970345471676, + "tokens_seen": 1054277632 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003437311935807422, + "loss": 1.5085, + "theoretical_loss": 3.6309490756917127, + "tokens_seen": 1054343168 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034372116349047144, + "loss": 1.4272, + "theoretical_loss": 3.630927807603957, + "tokens_seen": 1054408704 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003437111334002006, + "loss": 1.4003, + "theoretical_loss": 3.6309065412081685, + "tokens_seen": 1054474240 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003437011033099298, + "loss": 1.3511, + "theoretical_loss": 3.630885276504108, + "tokens_seen": 1054539776 + }, + { + "epoch": 0.32, + "learning_rate": 0.000343691073219659, + "loss": 1.2413, + "theoretical_loss": 3.6308640134915366, + "tokens_seen": 1054605312 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034368104312938817, + "loss": 1.3261, + "theoretical_loss": 3.6308427521702136, + "tokens_seen": 1054670848 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034367101303911735, + "loss": 1.3531, + "theoretical_loss": 3.6308214925398996, + "tokens_seen": 1054736384 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003436609829488466, + "loss": 1.4319, + "theoretical_loss": 3.630800234600355, + "tokens_seen": 1054801920 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003436509528585757, + "loss": 1.3496, + "theoretical_loss": 3.6307789783513402, + "tokens_seen": 1054867456 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034364092276830495, + "loss": 1.3966, + "theoretical_loss": 3.630757723792616, + "tokens_seen": 1054932992 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034363089267803413, + "loss": 1.3836, + "theoretical_loss": 3.630736470923943, + "tokens_seen": 1054998528 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003436208625877633, + "loss": 1.4143, + "theoretical_loss": 3.6307152197450825, + "tokens_seen": 1055064064 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.00027385653811506927, + "objective/train/docs_used": 603451, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.002429485321045, + "objective/train/original_loss": 3.002429246902466, + "objective/train/theoretical_loss": 3.630693970255794, + "objective/train/tokens_used": 1075589600, + "objective/train/value_avg": -0.0051116943359375, + "objective/train/value_loss": 4.886027454631403e-05, + "objective/train/value_max": -0.0002913475036621094, + "objective/train/value_min": -0.06646728515625, + "objective/train/value_reward_corr": 0.4210054092703847, + "objective/train/value_std": 0.004184722900390625, + "objective/train/weight_avg": 0.9995908141136169, + "objective/train/weighted_lm_loss": 2.9860098361968994, + "objective/train/weights_max": 1.6928677558898926, + "objective/train/weights_min": 0.34358859062194824, + "theoretical_loss": 3.630693970255794, + "tokens_seen": 1055129600 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003436108324974925, + "loss": 1.4005, + "theoretical_loss": 3.630693970255794, + "tokens_seen": 1055129600 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034360080240722167, + "loss": 1.5009, + "theoretical_loss": 3.630672722455839, + "tokens_seen": 1055195136 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034359077231695085, + "loss": 1.3925, + "theoretical_loss": 3.630651476344977, + "tokens_seen": 1055260672 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003435807422266801, + "loss": 1.4672, + "theoretical_loss": 3.6306302319229706, + "tokens_seen": 1055326208 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003435707121364092, + "loss": 1.367, + "theoretical_loss": 3.6306089891895805, + "tokens_seen": 1055391744 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034356068204613845, + "loss": 1.3966, + "theoretical_loss": 3.630587748144566, + "tokens_seen": 1055457280 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003435506519558676, + "loss": 1.3729, + "theoretical_loss": 3.63056650878769, + "tokens_seen": 1055522816 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003435406218655968, + "loss": 1.4391, + "theoretical_loss": 3.630545271118713, + "tokens_seen": 1055588352 + }, + { + "epoch": 0.32, + "learning_rate": 0.000343530591775326, + "loss": 1.4036, + "theoretical_loss": 3.6305240351373946, + "tokens_seen": 1055653888 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003435205616850552, + "loss": 1.4064, + "theoretical_loss": 3.630502800843498, + "tokens_seen": 1055719424 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034351053159478436, + "loss": 1.473, + "theoretical_loss": 3.6304815682367835, + "tokens_seen": 1055784960 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034350050150451354, + "loss": 1.438, + "theoretical_loss": 3.630460337317012, + "tokens_seen": 1055850496 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003434904714142427, + "loss": 1.3568, + "theoretical_loss": 3.6304391080839453, + "tokens_seen": 1055916032 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034348044132397195, + "loss": 1.5122, + "theoretical_loss": 3.630417880537344, + "tokens_seen": 1055981568 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003434704112337011, + "loss": 1.4607, + "theoretical_loss": 3.6303966546769706, + "tokens_seen": 1056047104 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003434603811434303, + "loss": 1.3524, + "theoretical_loss": 3.6303754305025855, + "tokens_seen": 1056112640 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003434503510531595, + "loss": 1.4156, + "theoretical_loss": 3.630354208013951, + "tokens_seen": 1056178176 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003434403209628887, + "loss": 1.4504, + "theoretical_loss": 3.630332987210828, + "tokens_seen": 1056243712 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034343029087261786, + "loss": 1.3852, + "theoretical_loss": 3.6303117680929784, + "tokens_seen": 1056309248 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034342026078234704, + "loss": 1.4071, + "theoretical_loss": 3.6302905506601633, + "tokens_seen": 1056374784 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003434102306920762, + "loss": 1.5311, + "theoretical_loss": 3.6302693349121453, + "tokens_seen": 1056440320 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034340020060180546, + "loss": 1.4138, + "theoretical_loss": 3.630248120848685, + "tokens_seen": 1056505856 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003433901705115346, + "loss": 1.4314, + "theoretical_loss": 3.6302269084695453, + "tokens_seen": 1056571392 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003433801404212638, + "loss": 1.2934, + "theoretical_loss": 3.630205697774487, + "tokens_seen": 1056636928 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034337011033099295, + "loss": 1.5211, + "theoretical_loss": 3.630184488763273, + "tokens_seen": 1056702464 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.00018848794570658356, + "objective/train/docs_used": 603451, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.651853561401367, + "objective/train/original_loss": 2.651853322982788, + "objective/train/theoretical_loss": 3.6301632814356637, + "objective/train/tokens_used": 1077228000, + "objective/train/value_avg": -0.0043792724609375, + "objective/train/value_loss": 3.874830872518942e-05, + "objective/train/value_max": -0.0002846717834472656, + "objective/train/value_min": -0.048309326171875, + "objective/train/value_reward_corr": 0.33586138178945085, + "objective/train/value_std": 0.0034236907958984375, + "objective/train/weight_avg": 0.9999891519546509, + "objective/train/weighted_lm_loss": 2.6532859802246094, + "objective/train/weights_max": 1.621080994606018, + "objective/train/weights_min": 0.6468944549560547, + "theoretical_loss": 3.6301632814356637, + "tokens_seen": 1056768000 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003433600802407222, + "loss": 1.3864, + "theoretical_loss": 3.6301632814356637, + "tokens_seen": 1056768000 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034335005015045136, + "loss": 1.4715, + "theoretical_loss": 3.6301420757914227, + "tokens_seen": 1056833536 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034334002006018054, + "loss": 1.3903, + "theoretical_loss": 3.6301208718303113, + "tokens_seen": 1056899072 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003433299899699097, + "loss": 1.359, + "theoretical_loss": 3.630099669552091, + "tokens_seen": 1056964608 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003433199598796389, + "loss": 1.3246, + "theoretical_loss": 3.630078468956525, + "tokens_seen": 1057030144 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003433099297893681, + "loss": 1.4766, + "theoretical_loss": 3.630057270043374, + "tokens_seen": 1057095680 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003432998996990973, + "loss": 1.4319, + "theoretical_loss": 3.630036072812402, + "tokens_seen": 1057161216 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034328986960882645, + "loss": 1.5343, + "theoretical_loss": 3.63001487726337, + "tokens_seen": 1057226752 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003432798395185557, + "loss": 1.4408, + "theoretical_loss": 3.629993683396041, + "tokens_seen": 1057292288 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034326980942828487, + "loss": 1.4225, + "theoretical_loss": 3.629972491210176, + "tokens_seen": 1057357824 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034325977933801405, + "loss": 1.4031, + "theoretical_loss": 3.629951300705539, + "tokens_seen": 1057423360 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034324974924774323, + "loss": 1.3746, + "theoretical_loss": 3.6299301118818916, + "tokens_seen": 1057488896 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003432397191574724, + "loss": 1.3432, + "theoretical_loss": 3.629908924738997, + "tokens_seen": 1057554432 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003432296890672016, + "loss": 1.4879, + "theoretical_loss": 3.629887739276616, + "tokens_seen": 1057619968 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003432196589769308, + "loss": 1.4043, + "theoretical_loss": 3.6298665554945133, + "tokens_seen": 1057685504 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034320962888665995, + "loss": 1.4051, + "theoretical_loss": 3.6298453733924507, + "tokens_seen": 1057751040 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003431995987963892, + "loss": 1.2893, + "theoretical_loss": 3.62982419297019, + "tokens_seen": 1057816576 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003431895687061183, + "loss": 1.3794, + "theoretical_loss": 3.6298030142274955, + "tokens_seen": 1057882112 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034317953861584755, + "loss": 1.3593, + "theoretical_loss": 3.6297818371641286, + "tokens_seen": 1057947648 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034316950852557673, + "loss": 1.5216, + "theoretical_loss": 3.6297606617798532, + "tokens_seen": 1058013184 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003431594784353059, + "loss": 1.5336, + "theoretical_loss": 3.629739488074432, + "tokens_seen": 1058078720 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003431494483450351, + "loss": 1.4043, + "theoretical_loss": 3.629718316047627, + "tokens_seen": 1058144256 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034313941825476433, + "loss": 1.502, + "theoretical_loss": 3.629697145699202, + "tokens_seen": 1058209792 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034312938816449345, + "loss": 1.3973, + "theoretical_loss": 3.6296759770289198, + "tokens_seen": 1058275328 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003431193580742227, + "loss": 1.4787, + "theoretical_loss": 3.6296548100365436, + "tokens_seen": 1058340864 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.000621949729975313, + "objective/train/docs_used": 604503, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.980889320373535, + "objective/train/original_loss": 2.9808895587921143, + "objective/train/theoretical_loss": 3.629633644721836, + "objective/train/tokens_used": 1078866400, + "objective/train/value_avg": -0.005340576171875, + "objective/train/value_loss": 5.816238990519196e-05, + "objective/train/value_max": -0.00032258033752441406, + "objective/train/value_min": -0.0792236328125, + "objective/train/value_reward_corr": 0.4213111932589433, + "objective/train/value_std": 0.00470733642578125, + "objective/train/weight_avg": 0.9964987635612488, + "objective/train/weighted_lm_loss": 2.9611713886260986, + "objective/train/weights_max": 1.9525561332702637, + "objective/train/weights_min": 0.27409127354621887, + "theoretical_loss": 3.629633644721836, + "tokens_seen": 1058406400 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003431093279839518, + "loss": 1.4173, + "theoretical_loss": 3.629633644721836, + "tokens_seen": 1058406400 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034309929789368105, + "loss": 1.3817, + "theoretical_loss": 3.6296124810845614, + "tokens_seen": 1058471936 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003430892678034103, + "loss": 1.3849, + "theoretical_loss": 3.629591319124481, + "tokens_seen": 1058537472 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003430792377131394, + "loss": 1.4002, + "theoretical_loss": 3.6295701588413603, + "tokens_seen": 1058603008 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034306920762286865, + "loss": 1.4146, + "theoretical_loss": 3.629549000234961, + "tokens_seen": 1058668544 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003430591775325978, + "loss": 1.3179, + "theoretical_loss": 3.6295278433050466, + "tokens_seen": 1058734080 + }, + { + "epoch": 0.32, + "learning_rate": 0.000343049147442327, + "loss": 1.4377, + "theoretical_loss": 3.6295066880513813, + "tokens_seen": 1058799616 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003430391173520562, + "loss": 1.449, + "theoretical_loss": 3.6294855344737287, + "tokens_seen": 1058865152 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003430290872617854, + "loss": 1.4048, + "theoretical_loss": 3.629464382571851, + "tokens_seen": 1058930688 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034301905717151456, + "loss": 1.44, + "theoretical_loss": 3.629443232345513, + "tokens_seen": 1058996224 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034300902708124374, + "loss": 1.4834, + "theoretical_loss": 3.629422083794477, + "tokens_seen": 1059061760 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003429989969909729, + "loss": 1.4223, + "theoretical_loss": 3.629400936918508, + "tokens_seen": 1059127296 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034298896690070215, + "loss": 1.4675, + "theoretical_loss": 3.629379791717369, + "tokens_seen": 1059192832 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003429789368104313, + "loss": 1.455, + "theoretical_loss": 3.6293586481908235, + "tokens_seen": 1059258368 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003429689067201605, + "loss": 1.4998, + "theoretical_loss": 3.6293375063386364, + "tokens_seen": 1059323904 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003429588766298897, + "loss": 1.4094, + "theoretical_loss": 3.62931636616057, + "tokens_seen": 1059389440 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003429488465396189, + "loss": 1.4252, + "theoretical_loss": 3.6292952276563897, + "tokens_seen": 1059454976 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034293881644934806, + "loss": 1.3974, + "theoretical_loss": 3.6292740908258576, + "tokens_seen": 1059520512 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034292878635907724, + "loss": 1.3511, + "theoretical_loss": 3.62925295566874, + "tokens_seen": 1059586048 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003429187562688064, + "loss": 1.5312, + "theoretical_loss": 3.6292318221847992, + "tokens_seen": 1059651584 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034290872617853566, + "loss": 1.4885, + "theoretical_loss": 3.6292106903737995, + "tokens_seen": 1059717120 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003428986960882648, + "loss": 1.5202, + "theoretical_loss": 3.6291895602355053, + "tokens_seen": 1059782656 + }, + { + "epoch": 0.32, + "learning_rate": 0.000342888665997994, + "loss": 1.5412, + "theoretical_loss": 3.629168431769681, + "tokens_seen": 1059848192 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034287863590772315, + "loss": 1.5149, + "theoretical_loss": 3.6291473049760903, + "tokens_seen": 1059913728 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003428686058174524, + "loss": 1.4454, + "theoretical_loss": 3.629126179854498, + "tokens_seen": 1059979264 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.0009389205370098352, + "objective/train/docs_used": 605021, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.759038209915161, + "objective/train/original_loss": 2.7590384483337402, + "objective/train/theoretical_loss": 3.6291050564046676, + "objective/train/tokens_used": 1080504800, + "objective/train/value_avg": -0.00432586669921875, + "objective/train/value_loss": 6.285901326918975e-05, + "objective/train/value_max": -0.00023233890533447266, + "objective/train/value_min": -0.06097412109375, + "objective/train/value_reward_corr": 0.34574517768662544, + "objective/train/value_std": 0.003936767578125, + "objective/train/weight_avg": 0.9934537410736084, + "objective/train/weighted_lm_loss": 2.7467734813690186, + "objective/train/weights_max": 1.6363024711608887, + "objective/train/weights_min": 0.3218264877796173, + "theoretical_loss": 3.6291050564046676, + "tokens_seen": 1060044800 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034285857572718156, + "loss": 1.4673, + "theoretical_loss": 3.6291050564046676, + "tokens_seen": 1060044800 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034284854563691074, + "loss": 1.3869, + "theoretical_loss": 3.6290839346263644, + "tokens_seen": 1060110336 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003428385155466399, + "loss": 1.417, + "theoretical_loss": 3.6290628145193526, + "tokens_seen": 1060175872 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003428284854563691, + "loss": 1.4249, + "theoretical_loss": 3.629041696083396, + "tokens_seen": 1060241408 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003428184553660983, + "loss": 1.3509, + "theoretical_loss": 3.62902057931826, + "tokens_seen": 1060306944 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003428084252758275, + "loss": 1.4323, + "theoretical_loss": 3.6289994642237087, + "tokens_seen": 1060372480 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034279839518555665, + "loss": 1.4798, + "theoretical_loss": 3.6289783507995064, + "tokens_seen": 1060438016 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003427883650952859, + "loss": 1.4897, + "theoretical_loss": 3.628957239045419, + "tokens_seen": 1060503552 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034277833500501507, + "loss": 1.3576, + "theoretical_loss": 3.628936128961209, + "tokens_seen": 1060569088 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034276830491474425, + "loss": 1.4032, + "theoretical_loss": 3.6289150205466427, + "tokens_seen": 1060634624 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034275827482447343, + "loss": 1.4217, + "theoretical_loss": 3.628893913801485, + "tokens_seen": 1060700160 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003427482447342026, + "loss": 1.4838, + "theoretical_loss": 3.6288728087255, + "tokens_seen": 1060765696 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003427382146439318, + "loss": 1.4427, + "theoretical_loss": 3.6288517053184535, + "tokens_seen": 1060831232 + }, + { + "epoch": 0.32, + "learning_rate": 0.000342728184553661, + "loss": 1.4472, + "theoretical_loss": 3.6288306035801092, + "tokens_seen": 1060896768 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034271815446339015, + "loss": 1.4453, + "theoretical_loss": 3.628809503510233, + "tokens_seen": 1060962304 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003427081243731194, + "loss": 1.4156, + "theoretical_loss": 3.6287884051085895, + "tokens_seen": 1061027840 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003426980942828485, + "loss": 1.4201, + "theoretical_loss": 3.628767308374944, + "tokens_seen": 1061093376 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034268806419257775, + "loss": 1.4647, + "theoretical_loss": 3.6287462133090616, + "tokens_seen": 1061158912 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034267803410230693, + "loss": 1.4295, + "theoretical_loss": 3.6287251199107073, + "tokens_seen": 1061224448 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003426680040120361, + "loss": 1.4545, + "theoretical_loss": 3.628704028179647, + "tokens_seen": 1061289984 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003426579739217653, + "loss": 1.41, + "theoretical_loss": 3.6286829381156447, + "tokens_seen": 1061355520 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034264794383149453, + "loss": 1.4478, + "theoretical_loss": 3.628661849718467, + "tokens_seen": 1061421056 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034263791374122366, + "loss": 1.4018, + "theoretical_loss": 3.628640762987878, + "tokens_seen": 1061486592 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003426278836509529, + "loss": 1.3628, + "theoretical_loss": 3.6286196779236444, + "tokens_seen": 1061552128 + }, + { + "epoch": 0.32, + "learning_rate": 0.000342617853560682, + "loss": 1.4368, + "theoretical_loss": 3.628598594525531, + "tokens_seen": 1061617664 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.0005862255929969251, + "objective/train/docs_used": 606310, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.4363224506378174, + "objective/train/original_loss": 3.4363226890563965, + "objective/train/theoretical_loss": 3.628577512793303, + "objective/train/tokens_used": 1082143200, + "objective/train/value_avg": -0.005306243896484375, + "objective/train/value_loss": 5.054706343798898e-05, + "objective/train/value_max": -0.0002758502960205078, + "objective/train/value_min": -0.07598876953125, + "objective/train/value_reward_corr": 0.5636678866394841, + "objective/train/value_std": 0.00534820556640625, + "objective/train/weight_avg": 0.9965653419494629, + "objective/train/weighted_lm_loss": 3.4230103492736816, + "objective/train/weights_max": 1.584398865699768, + "objective/train/weights_min": 0.45294612646102905, + "theoretical_loss": 3.628577512793303, + "tokens_seen": 1061683200 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034260782347041125, + "loss": 1.4966, + "theoretical_loss": 3.628577512793303, + "tokens_seen": 1061683200 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034259779338014043, + "loss": 1.3737, + "theoretical_loss": 3.628556432726727, + "tokens_seen": 1061748736 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003425877632898696, + "loss": 1.3511, + "theoretical_loss": 3.6285353543255674, + "tokens_seen": 1061814272 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003425777331995988, + "loss": 1.4404, + "theoretical_loss": 3.6285142775895904, + "tokens_seen": 1061879808 + }, + { + "epoch": 0.32, + "learning_rate": 0.000342567703109328, + "loss": 1.3947, + "theoretical_loss": 3.6284932025185617, + "tokens_seen": 1061945344 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034255767301905716, + "loss": 1.4035, + "theoretical_loss": 3.6284721291122466, + "tokens_seen": 1062010880 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003425476429287864, + "loss": 1.3353, + "theoretical_loss": 3.6284510573704116, + "tokens_seen": 1062076416 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003425376128385155, + "loss": 1.5529, + "theoretical_loss": 3.6284299872928223, + "tokens_seen": 1062141952 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034252758274824476, + "loss": 1.4135, + "theoretical_loss": 3.6284089188792445, + "tokens_seen": 1062207488 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003425175526579739, + "loss": 1.4537, + "theoretical_loss": 3.6283878521294444, + "tokens_seen": 1062273024 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003425075225677031, + "loss": 1.3944, + "theoretical_loss": 3.6283667870431873, + "tokens_seen": 1062338560 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003424974924774323, + "loss": 1.3888, + "theoretical_loss": 3.6283457236202397, + "tokens_seen": 1062404096 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003424874623871615, + "loss": 1.4365, + "theoretical_loss": 3.6283246618603675, + "tokens_seen": 1062469632 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034247743229689066, + "loss": 1.509, + "theoretical_loss": 3.628303601763337, + "tokens_seen": 1062535168 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003424674022066199, + "loss": 1.4204, + "theoretical_loss": 3.6282825433289148, + "tokens_seen": 1062600704 + }, + { + "epoch": 0.32, + "learning_rate": 0.000342457372116349, + "loss": 1.4094, + "theoretical_loss": 3.628261486556866, + "tokens_seen": 1062666240 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034244734202607826, + "loss": 1.2665, + "theoretical_loss": 3.6282404314469576, + "tokens_seen": 1062731776 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003424373119358074, + "loss": 1.4778, + "theoretical_loss": 3.6282193779989558, + "tokens_seen": 1062797312 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003424272818455366, + "loss": 1.5023, + "theoretical_loss": 3.628198326212627, + "tokens_seen": 1062862848 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003424172517552658, + "loss": 1.4565, + "theoretical_loss": 3.6281772760877375, + "tokens_seen": 1062928384 + }, + { + "epoch": 0.32, + "learning_rate": 0.000342407221664995, + "loss": 1.4969, + "theoretical_loss": 3.6281562276240535, + "tokens_seen": 1062993920 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034239719157472416, + "loss": 1.3736, + "theoretical_loss": 3.6281351808213422, + "tokens_seen": 1063059456 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034238716148445335, + "loss": 1.4026, + "theoretical_loss": 3.6281141356793696, + "tokens_seen": 1063124992 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003423771313941825, + "loss": 1.389, + "theoretical_loss": 3.6280930921979024, + "tokens_seen": 1063190528 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034236710130391176, + "loss": 1.42, + "theoretical_loss": 3.6280720503767077, + "tokens_seen": 1063256064 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.0009698625653982162, + "objective/train/docs_used": 606853, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8193612098693848, + "objective/train/original_loss": 2.8193604946136475, + "objective/train/theoretical_loss": 3.628051010215551, + "objective/train/tokens_used": 1083781600, + "objective/train/value_avg": -0.00566864013671875, + "objective/train/value_loss": 0.00011047911539208144, + "objective/train/value_max": -0.00033283233642578125, + "objective/train/value_min": -0.09368896484375, + "objective/train/value_reward_corr": 0.5604795231079334, + "objective/train/value_std": 0.005680084228515625, + "objective/train/weight_avg": 0.9953106045722961, + "objective/train/weighted_lm_loss": 2.7995502948760986, + "objective/train/weights_max": 1.7006419897079468, + "objective/train/weights_min": 0.39148014783859253, + "theoretical_loss": 3.628051010215551, + "tokens_seen": 1063321600 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003423570712136409, + "loss": 1.4964, + "theoretical_loss": 3.628051010215551, + "tokens_seen": 1063321600 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003423470411233701, + "loss": 1.4752, + "theoretical_loss": 3.6280299717142004, + "tokens_seen": 1063387136 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003423370110330993, + "loss": 1.3474, + "theoretical_loss": 3.6280089348724216, + "tokens_seen": 1063452672 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003423269809428285, + "loss": 1.279, + "theoretical_loss": 3.627987899689982, + "tokens_seen": 1063518208 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003423169508525577, + "loss": 1.4327, + "theoretical_loss": 3.6279668661666484, + "tokens_seen": 1063583744 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034230692076228685, + "loss": 1.442, + "theoretical_loss": 3.6279458343021878, + "tokens_seen": 1063649280 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003422968906720161, + "loss": 1.3989, + "theoretical_loss": 3.6279248040963674, + "tokens_seen": 1063714816 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034228686058174527, + "loss": 1.4369, + "theoretical_loss": 3.627903775548953, + "tokens_seen": 1063780352 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034227683049147445, + "loss": 1.4867, + "theoretical_loss": 3.6278827486597134, + "tokens_seen": 1063845888 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034226680040120363, + "loss": 1.4997, + "theoretical_loss": 3.6278617234284143, + "tokens_seen": 1063911424 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003422567703109328, + "loss": 1.3893, + "theoretical_loss": 3.627840699854824, + "tokens_seen": 1063976960 + }, + { + "epoch": 0.32, + "learning_rate": 0.000342246740220662, + "loss": 1.4497, + "theoretical_loss": 3.6278196779387084, + "tokens_seen": 1064042496 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003422367101303912, + "loss": 1.3641, + "theoretical_loss": 3.627798657679836, + "tokens_seen": 1064108032 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034222668004012035, + "loss": 1.4685, + "theoretical_loss": 3.6277776390779732, + "tokens_seen": 1064173568 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003422166499498496, + "loss": 1.4283, + "theoretical_loss": 3.627756622132888, + "tokens_seen": 1064239104 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003422066198595787, + "loss": 1.3487, + "theoretical_loss": 3.627735606844347, + "tokens_seen": 1064304640 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034219658976930795, + "loss": 1.4502, + "theoretical_loss": 3.6277145932121186, + "tokens_seen": 1064370176 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034218655967903713, + "loss": 1.507, + "theoretical_loss": 3.6276935812359703, + "tokens_seen": 1064435712 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003421765295887663, + "loss": 1.4216, + "theoretical_loss": 3.627672570915668, + "tokens_seen": 1064501248 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003421664994984955, + "loss": 1.3496, + "theoretical_loss": 3.6276515622509815, + "tokens_seen": 1064566784 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034215646940822473, + "loss": 1.5785, + "theoretical_loss": 3.627630555241677, + "tokens_seen": 1064632320 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034214643931795386, + "loss": 1.5123, + "theoretical_loss": 3.6276095498875227, + "tokens_seen": 1064697856 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003421364092276831, + "loss": 1.4062, + "theoretical_loss": 3.627588546188286, + "tokens_seen": 1064763392 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003421263791374122, + "loss": 1.4028, + "theoretical_loss": 3.6275675441437345, + "tokens_seen": 1064828928 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034211634904714145, + "loss": 1.3557, + "theoretical_loss": 3.627546543753637, + "tokens_seen": 1064894464 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.000695951224770397, + "objective/train/docs_used": 607789, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.504951238632202, + "objective/train/original_loss": 2.504951000213623, + "objective/train/theoretical_loss": 3.62752554501776, + "objective/train/tokens_used": 1085420000, + "objective/train/value_avg": -0.00478363037109375, + "objective/train/value_loss": 7.730251672910526e-05, + "objective/train/value_max": -0.0002415180206298828, + "objective/train/value_min": -0.0704345703125, + "objective/train/value_reward_corr": 0.38078108308667113, + "objective/train/value_std": 0.0039520263671875, + "objective/train/weight_avg": 0.9964646100997925, + "objective/train/weighted_lm_loss": 2.5074775218963623, + "objective/train/weights_max": 1.7810927629470825, + "objective/train/weights_min": 0.19595424830913544, + "theoretical_loss": 3.62752554501776, + "tokens_seen": 1064960000 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034210631895687063, + "loss": 1.3978, + "theoretical_loss": 3.62752554501776, + "tokens_seen": 1064960000 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003420962888665998, + "loss": 1.487, + "theoretical_loss": 3.6275045479358727, + "tokens_seen": 1065025536 + }, + { + "epoch": 0.32, + "learning_rate": 0.000342086258776329, + "loss": 1.4158, + "theoretical_loss": 3.6274835525077425, + "tokens_seen": 1065091072 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003420762286860582, + "loss": 1.3636, + "theoretical_loss": 3.6274625587331375, + "tokens_seen": 1065156608 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034206619859578736, + "loss": 1.4146, + "theoretical_loss": 3.627441566611825, + "tokens_seen": 1065222144 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003420561685055166, + "loss": 1.4973, + "theoretical_loss": 3.6274205761435745, + "tokens_seen": 1065287680 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003420461384152457, + "loss": 1.4795, + "theoretical_loss": 3.627399587328153, + "tokens_seen": 1065353216 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034203610832497496, + "loss": 1.2944, + "theoretical_loss": 3.6273786001653288, + "tokens_seen": 1065418752 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003420260782347041, + "loss": 1.357, + "theoretical_loss": 3.627357614654871, + "tokens_seen": 1065484288 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003420160481444333, + "loss": 1.3503, + "theoretical_loss": 3.6273366307965467, + "tokens_seen": 1065549824 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003420060180541625, + "loss": 1.447, + "theoretical_loss": 3.627315648590125, + "tokens_seen": 1065615360 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003419959879638917, + "loss": 1.4442, + "theoretical_loss": 3.627294668035374, + "tokens_seen": 1065680896 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034198595787362086, + "loss": 1.2435, + "theoretical_loss": 3.6272736891320627, + "tokens_seen": 1065746432 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003419759277833501, + "loss": 1.4848, + "theoretical_loss": 3.627252711879959, + "tokens_seen": 1065811968 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003419658976930792, + "loss": 1.4924, + "theoretical_loss": 3.627231736278831, + "tokens_seen": 1065877504 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034195586760280846, + "loss": 1.3694, + "theoretical_loss": 3.627210762328448, + "tokens_seen": 1065943040 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003419458375125376, + "loss": 1.4251, + "theoretical_loss": 3.6271897900285786, + "tokens_seen": 1066008576 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003419358074222668, + "loss": 1.3766, + "theoretical_loss": 3.6271688193789915, + "tokens_seen": 1066074112 + }, + { + "epoch": 0.32, + "learning_rate": 0.000341925777331996, + "loss": 1.4982, + "theoretical_loss": 3.6271478503794548, + "tokens_seen": 1066139648 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003419157472417252, + "loss": 1.4514, + "theoretical_loss": 3.627126883029737, + "tokens_seen": 1066205184 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034190571715145436, + "loss": 1.4713, + "theoretical_loss": 3.627105917329608, + "tokens_seen": 1066270720 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034189568706118355, + "loss": 1.4278, + "theoretical_loss": 3.6270849532788363, + "tokens_seen": 1066336256 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003418856569709127, + "loss": 1.5448, + "theoretical_loss": 3.6270639908771907, + "tokens_seen": 1066401792 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034187562688064196, + "loss": 1.4833, + "theoretical_loss": 3.62704303012444, + "tokens_seen": 1066467328 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003418655967903711, + "loss": 1.4056, + "theoretical_loss": 3.627022071020353, + "tokens_seen": 1066532864 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.0012295874767005444, + "objective/train/docs_used": 608475, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.373603582382202, + "objective/train/original_loss": 3.3736038208007812, + "objective/train/theoretical_loss": 3.627001113564699, + "objective/train/tokens_used": 1087058400, + "objective/train/value_avg": -0.005584716796875, + "objective/train/value_loss": 7.293685484910384e-05, + "objective/train/value_max": -0.0003528594970703125, + "objective/train/value_min": -0.07464599609375, + "objective/train/value_reward_corr": 0.4211466157394408, + "objective/train/value_std": 0.004482269287109375, + "objective/train/weight_avg": 0.9911050796508789, + "objective/train/weighted_lm_loss": 3.3409743309020996, + "objective/train/weights_max": 1.9556103944778442, + "objective/train/weights_min": 0.31323161721229553, + "theoretical_loss": 3.627001113564699, + "tokens_seen": 1066598400 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003418555667001003, + "loss": 1.4894, + "theoretical_loss": 3.627001113564699, + "tokens_seen": 1066598400 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034184553660982945, + "loss": 1.3499, + "theoretical_loss": 3.626980157757247, + "tokens_seen": 1066663936 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003418355065195587, + "loss": 1.3437, + "theoretical_loss": 3.6269592035977665, + "tokens_seen": 1066729472 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034182547642928787, + "loss": 1.4077, + "theoretical_loss": 3.6269382510860257, + "tokens_seen": 1066795008 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034181544633901705, + "loss": 1.5536, + "theoretical_loss": 3.6269173002217956, + "tokens_seen": 1066860544 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034180541624874623, + "loss": 1.3762, + "theoretical_loss": 3.6268963510048433, + "tokens_seen": 1066926080 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034179538615847547, + "loss": 1.4089, + "theoretical_loss": 3.6268754034349397, + "tokens_seen": 1066991616 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003417853560682046, + "loss": 1.5543, + "theoretical_loss": 3.626854457511853, + "tokens_seen": 1067057152 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034177532597793383, + "loss": 1.5049, + "theoretical_loss": 3.6268335132353533, + "tokens_seen": 1067122688 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034176529588766295, + "loss": 1.3947, + "theoretical_loss": 3.6268125706052103, + "tokens_seen": 1067188224 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003417552657973922, + "loss": 1.481, + "theoretical_loss": 3.6267916296211933, + "tokens_seen": 1067253760 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034174523570712137, + "loss": 1.3041, + "theoretical_loss": 3.626770690283071, + "tokens_seen": 1067319296 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034173520561685055, + "loss": 1.4157, + "theoretical_loss": 3.6267497525906145, + "tokens_seen": 1067384832 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034172517552657973, + "loss": 1.4298, + "theoretical_loss": 3.6267288165435922, + "tokens_seen": 1067450368 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003417151454363089, + "loss": 1.3787, + "theoretical_loss": 3.626707882141774, + "tokens_seen": 1067515904 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003417051153460381, + "loss": 1.3246, + "theoretical_loss": 3.62668694938493, + "tokens_seen": 1067581440 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034169508525576733, + "loss": 1.4839, + "theoretical_loss": 3.6266660182728296, + "tokens_seen": 1067646976 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034168505516549646, + "loss": 1.4561, + "theoretical_loss": 3.626645088805243, + "tokens_seen": 1067712512 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003416750250752257, + "loss": 1.4222, + "theoretical_loss": 3.62662416098194, + "tokens_seen": 1067778048 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003416649949849548, + "loss": 1.4764, + "theoretical_loss": 3.62660323480269, + "tokens_seen": 1067843584 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034165496489468406, + "loss": 1.4097, + "theoretical_loss": 3.6265823102672634, + "tokens_seen": 1067909120 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034164493480441324, + "loss": 1.4053, + "theoretical_loss": 3.62656138737543, + "tokens_seen": 1067974656 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003416349047141424, + "loss": 1.4824, + "theoretical_loss": 3.6265404661269605, + "tokens_seen": 1068040192 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003416248746238716, + "loss": 1.4502, + "theoretical_loss": 3.6265195465216244, + "tokens_seen": 1068105728 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034161484453360083, + "loss": 1.333, + "theoretical_loss": 3.6264986285591916, + "tokens_seen": 1068171264 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.00027989037334918976, + "objective/train/docs_used": 609657, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.784149408340454, + "objective/train/original_loss": 2.784149646759033, + "objective/train/theoretical_loss": 3.6264777122394327, + "objective/train/tokens_used": 1088696800, + "objective/train/value_avg": -0.00424957275390625, + "objective/train/value_loss": 3.747572918655351e-05, + "objective/train/value_max": -0.00028252601623535156, + "objective/train/value_min": -0.06805419921875, + "objective/train/value_reward_corr": 0.30030085562389514, + "objective/train/value_std": 0.003574371337890625, + "objective/train/weight_avg": 0.9990359544754028, + "objective/train/weighted_lm_loss": 2.782238721847534, + "objective/train/weights_max": 1.9749478101730347, + "objective/train/weights_min": 0.6705014705657959, + "theoretical_loss": 3.6264777122394327, + "tokens_seen": 1068236800 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034160481444332996, + "loss": 1.5013, + "theoretical_loss": 3.6264777122394327, + "tokens_seen": 1068236800 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003415947843530592, + "loss": 1.482, + "theoretical_loss": 3.626456797562118, + "tokens_seen": 1068302336 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003415847542627884, + "loss": 1.3847, + "theoretical_loss": 3.6264358845270177, + "tokens_seen": 1068367872 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034157472417251756, + "loss": 1.4429, + "theoretical_loss": 3.626414973133902, + "tokens_seen": 1068433408 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003415646940822468, + "loss": 1.395, + "theoretical_loss": 3.626394063382541, + "tokens_seen": 1068498944 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003415546639919759, + "loss": 1.533, + "theoretical_loss": 3.626373155272706, + "tokens_seen": 1068564480 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034154463390170516, + "loss": 1.3758, + "theoretical_loss": 3.6263522488041664, + "tokens_seen": 1068630016 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003415346038114343, + "loss": 1.51, + "theoretical_loss": 3.6263313439766938, + "tokens_seen": 1068695552 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003415245737211635, + "loss": 1.3687, + "theoretical_loss": 3.626310440790058, + "tokens_seen": 1068761088 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003415145436308927, + "loss": 1.392, + "theoretical_loss": 3.6262895392440297, + "tokens_seen": 1068826624 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003415045135406219, + "loss": 1.4801, + "theoretical_loss": 3.626268639338379, + "tokens_seen": 1068892160 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034149448345035106, + "loss": 1.4582, + "theoretical_loss": 3.6262477410728784, + "tokens_seen": 1068957696 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003414844533600803, + "loss": 1.3585, + "theoretical_loss": 3.626226844447297, + "tokens_seen": 1069023232 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003414744232698094, + "loss": 1.3872, + "theoretical_loss": 3.626205949461406, + "tokens_seen": 1069088768 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034146439317953866, + "loss": 1.5264, + "theoretical_loss": 3.6261850561149767, + "tokens_seen": 1069154304 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003414543630892678, + "loss": 1.3119, + "theoretical_loss": 3.626164164407779, + "tokens_seen": 1069219840 + }, + { + "epoch": 0.32, + "learning_rate": 0.000341444332998997, + "loss": 1.4802, + "theoretical_loss": 3.626143274339585, + "tokens_seen": 1069285376 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003414343029087262, + "loss": 1.3636, + "theoretical_loss": 3.626122385910165, + "tokens_seen": 1069350912 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003414242728184554, + "loss": 1.4292, + "theoretical_loss": 3.62610149911929, + "tokens_seen": 1069416448 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034141424272818456, + "loss": 1.456, + "theoretical_loss": 3.6260806139667316, + "tokens_seen": 1069481984 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034140421263791375, + "loss": 1.3738, + "theoretical_loss": 3.62605973045226, + "tokens_seen": 1069547520 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034139418254764293, + "loss": 1.3819, + "theoretical_loss": 3.6260388485756474, + "tokens_seen": 1069613056 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034138415245737216, + "loss": 1.3498, + "theoretical_loss": 3.626017968336664, + "tokens_seen": 1069678592 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003413741223671013, + "loss": 1.4291, + "theoretical_loss": 3.6259970897350815, + "tokens_seen": 1069744128 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003413640922768305, + "loss": 1.5172, + "theoretical_loss": 3.625976212770672, + "tokens_seen": 1069809664 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": 0.0010531679727137089, + "objective/train/docs_used": 610408, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0552947521209717, + "objective/train/original_loss": 3.0552945137023926, + "objective/train/theoretical_loss": 3.625955337443205, + "objective/train/tokens_used": 1090335200, + "objective/train/value_avg": -0.005565643310546875, + "objective/train/value_loss": 4.4123094994574785e-05, + "objective/train/value_max": -0.000469207763671875, + "objective/train/value_min": -0.06854248046875, + "objective/train/value_reward_corr": 0.3383582984737771, + "objective/train/value_std": 0.004558563232421875, + "objective/train/weight_avg": 1.0127172470092773, + "objective/train/weighted_lm_loss": 3.106431007385254, + "objective/train/weights_max": 1.8444198369979858, + "objective/train/weights_min": 0.4997047483921051, + "theoretical_loss": 3.625955337443205, + "tokens_seen": 1069875200 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034135406218655965, + "loss": 1.4353, + "theoretical_loss": 3.625955337443205, + "tokens_seen": 1069875200 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003413440320962889, + "loss": 1.25, + "theoretical_loss": 3.6259344637524533, + "tokens_seen": 1069940736 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034133400200601807, + "loss": 1.4945, + "theoretical_loss": 3.6259135916981884, + "tokens_seen": 1070006272 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034132397191574725, + "loss": 1.4922, + "theoretical_loss": 3.6258927212801813, + "tokens_seen": 1070071808 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034131394182547643, + "loss": 1.4169, + "theoretical_loss": 3.625871852498203, + "tokens_seen": 1070137344 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034130391173520567, + "loss": 1.3131, + "theoretical_loss": 3.6258509853520264, + "tokens_seen": 1070202880 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003412938816449348, + "loss": 1.4387, + "theoretical_loss": 3.625830119841422, + "tokens_seen": 1070268416 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034128385155466403, + "loss": 1.4116, + "theoretical_loss": 3.625809255966162, + "tokens_seen": 1070333952 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034127382146439315, + "loss": 1.4154, + "theoretical_loss": 3.6257883937260176, + "tokens_seen": 1070399488 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003412637913741224, + "loss": 1.3571, + "theoretical_loss": 3.6257675331207615, + "tokens_seen": 1070465024 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034125376128385157, + "loss": 1.5027, + "theoretical_loss": 3.6257466741501645, + "tokens_seen": 1070530560 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034124373119358075, + "loss": 1.4276, + "theoretical_loss": 3.6257258168139987, + "tokens_seen": 1070596096 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034123370110330993, + "loss": 1.4344, + "theoretical_loss": 3.625704961112037, + "tokens_seen": 1070661632 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003412236710130391, + "loss": 1.511, + "theoretical_loss": 3.6256841070440498, + "tokens_seen": 1070727168 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003412136409227683, + "loss": 1.3671, + "theoretical_loss": 3.62566325460981, + "tokens_seen": 1070792704 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034120361083249753, + "loss": 1.3785, + "theoretical_loss": 3.6256424038090893, + "tokens_seen": 1070858240 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034119358074222666, + "loss": 1.458, + "theoretical_loss": 3.6256215546416595, + "tokens_seen": 1070923776 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003411835506519559, + "loss": 1.5605, + "theoretical_loss": 3.625600707107294, + "tokens_seen": 1070989312 + }, + { + "epoch": 0.32, + "learning_rate": 0.000341173520561685, + "loss": 1.4378, + "theoretical_loss": 3.6255798612057633, + "tokens_seen": 1071054848 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034116349047141426, + "loss": 1.3731, + "theoretical_loss": 3.6255590169368404, + "tokens_seen": 1071120384 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034115346038114344, + "loss": 1.5478, + "theoretical_loss": 3.6255381743002983, + "tokens_seen": 1071185920 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003411434302908726, + "loss": 1.4328, + "theoretical_loss": 3.6255173332959076, + "tokens_seen": 1071251456 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003411334002006018, + "loss": 1.3623, + "theoretical_loss": 3.6254964939234426, + "tokens_seen": 1071316992 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034112337011033103, + "loss": 1.2996, + "theoretical_loss": 3.625475656182674, + "tokens_seen": 1071382528 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034111334002006016, + "loss": 1.3764, + "theoretical_loss": 3.625454820073375, + "tokens_seen": 1071448064 + }, + { + "epoch": 0.32, + "objective/train/advantage_avg": -0.0009361907723359764, + "objective/train/docs_used": 611059, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.78226375579834, + "objective/train/original_loss": 2.78226375579834, + "objective/train/theoretical_loss": 3.6254339855953184, + "objective/train/tokens_used": 1091973600, + "objective/train/value_avg": -0.0049591064453125, + "objective/train/value_loss": 5.551685535465367e-05, + "objective/train/value_max": -0.00031757354736328125, + "objective/train/value_min": -0.0548095703125, + "objective/train/value_reward_corr": 0.329078354049392, + "objective/train/value_std": 0.003997802734375, + "objective/train/weight_avg": 0.9932489991188049, + "objective/train/weighted_lm_loss": 2.7679169178009033, + "objective/train/weights_max": 1.577041745185852, + "objective/train/weights_min": 0.4574260115623474, + "theoretical_loss": 3.6254339855953184, + "tokens_seen": 1071513600 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003411033099297894, + "loss": 1.4272, + "theoretical_loss": 3.6254339855953184, + "tokens_seen": 1071513600 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003410932798395185, + "loss": 1.4246, + "theoretical_loss": 3.625413152748276, + "tokens_seen": 1071579136 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034108324974924776, + "loss": 1.3907, + "theoretical_loss": 3.625392321532021, + "tokens_seen": 1071644672 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034107321965897694, + "loss": 1.4185, + "theoretical_loss": 3.6253714919463254, + "tokens_seen": 1071710208 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003410631895687061, + "loss": 1.3163, + "theoretical_loss": 3.6253506639909627, + "tokens_seen": 1071775744 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003410531594784353, + "loss": 1.5312, + "theoretical_loss": 3.6253298376657046, + "tokens_seen": 1071841280 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003410431293881645, + "loss": 1.4283, + "theoretical_loss": 3.625309012970325, + "tokens_seen": 1071906816 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034103309929789366, + "loss": 1.429, + "theoretical_loss": 3.6252881899045954, + "tokens_seen": 1071972352 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003410230692076229, + "loss": 1.51, + "theoretical_loss": 3.62526736846829, + "tokens_seen": 1072037888 + }, + { + "epoch": 0.32, + "learning_rate": 0.000341013039117352, + "loss": 1.4932, + "theoretical_loss": 3.625246548661181, + "tokens_seen": 1072103424 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034100300902708126, + "loss": 1.4834, + "theoretical_loss": 3.625225730483041, + "tokens_seen": 1072168960 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034099297893681044, + "loss": 1.3712, + "theoretical_loss": 3.625204913933644, + "tokens_seen": 1072234496 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003409829488465396, + "loss": 1.4028, + "theoretical_loss": 3.625184099012762, + "tokens_seen": 1072300032 + }, + { + "epoch": 0.32, + "learning_rate": 0.0003409729187562688, + "loss": 1.46, + "theoretical_loss": 3.6251632857201685, + "tokens_seen": 1072365568 + }, + { + "epoch": 0.32, + "learning_rate": 0.000340962888665998, + "loss": 1.4069, + "theoretical_loss": 3.625142474055637, + "tokens_seen": 1072431104 + }, + { + "epoch": 0.32, + "learning_rate": 0.00034095285857572717, + "loss": 1.4214, + "theoretical_loss": 3.62512166401894, + "tokens_seen": 1072496640 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003409428284854564, + "loss": 1.3973, + "theoretical_loss": 3.625100855609851, + "tokens_seen": 1072562176 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034093279839518553, + "loss": 1.5213, + "theoretical_loss": 3.6250800488281434, + "tokens_seen": 1072627712 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034092276830491477, + "loss": 1.4952, + "theoretical_loss": 3.6250592436735904, + "tokens_seen": 1072693248 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003409127382146439, + "loss": 1.4693, + "theoretical_loss": 3.625038440145965, + "tokens_seen": 1072758784 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034090270812437313, + "loss": 1.44, + "theoretical_loss": 3.6250176382450414, + "tokens_seen": 1072824320 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003408926780341023, + "loss": 1.3298, + "theoretical_loss": 3.624996837970593, + "tokens_seen": 1072889856 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003408826479438315, + "loss": 1.4309, + "theoretical_loss": 3.6249760393223918, + "tokens_seen": 1072955392 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034087261785356067, + "loss": 1.3751, + "theoretical_loss": 3.624955242300213, + "tokens_seen": 1073020928 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034086258776328985, + "loss": 1.4676, + "theoretical_loss": 3.6249344469038296, + "tokens_seen": 1073086464 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0001765193446772173, + "objective/train/docs_used": 612309, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.981520175933838, + "objective/train/original_loss": 2.981520652770996, + "objective/train/theoretical_loss": 3.624913653133015, + "objective/train/tokens_used": 1093612000, + "objective/train/value_avg": -0.00396728515625, + "objective/train/value_loss": 4.27341801696457e-05, + "objective/train/value_max": -0.0002982616424560547, + "objective/train/value_min": -0.034698486328125, + "objective/train/value_reward_corr": 0.28342825932209936, + "objective/train/value_std": 0.00275421142578125, + "objective/train/weight_avg": 1.0001742839813232, + "objective/train/weighted_lm_loss": 2.989581346511841, + "objective/train/weights_max": 1.2884984016418457, + "objective/train/weights_min": 0.2446155548095703, + "theoretical_loss": 3.624913653133015, + "tokens_seen": 1073152000 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034085255767301903, + "loss": 1.4662, + "theoretical_loss": 3.624913653133015, + "tokens_seen": 1073152000 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034084252758274827, + "loss": 1.3912, + "theoretical_loss": 3.6248928609875435, + "tokens_seen": 1073217536 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034083249749247745, + "loss": 1.4802, + "theoretical_loss": 3.6248720704671875, + "tokens_seen": 1073283072 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034082246740220663, + "loss": 1.3919, + "theoretical_loss": 3.624851281571723, + "tokens_seen": 1073348608 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034081243731193587, + "loss": 1.4328, + "theoretical_loss": 3.6248304943009213, + "tokens_seen": 1073414144 + }, + { + "epoch": 0.33, + "learning_rate": 0.000340802407221665, + "loss": 1.556, + "theoretical_loss": 3.624809708654558, + "tokens_seen": 1073479680 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034079237713139423, + "loss": 1.391, + "theoretical_loss": 3.624788924632406, + "tokens_seen": 1073545216 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034078234704112335, + "loss": 1.4009, + "theoretical_loss": 3.62476814223424, + "tokens_seen": 1073610752 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003407723169508526, + "loss": 1.3456, + "theoretical_loss": 3.624747361459834, + "tokens_seen": 1073676288 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034076228686058177, + "loss": 1.4989, + "theoretical_loss": 3.624726582308961, + "tokens_seen": 1073741824 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034075225677031095, + "loss": 1.3942, + "theoretical_loss": 3.624705804781396, + "tokens_seen": 1073807360 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034074222668004013, + "loss": 1.3804, + "theoretical_loss": 3.624685028876913, + "tokens_seen": 1073872896 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003407321965897693, + "loss": 1.4063, + "theoretical_loss": 3.6246642545952854, + "tokens_seen": 1073938432 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003407221664994985, + "loss": 1.3439, + "theoretical_loss": 3.624643481936289, + "tokens_seen": 1074003968 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034071213640922773, + "loss": 1.3513, + "theoretical_loss": 3.6246227108996965, + "tokens_seen": 1074069504 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034070210631895686, + "loss": 1.451, + "theoretical_loss": 3.624601941485283, + "tokens_seen": 1074135040 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003406920762286861, + "loss": 1.5098, + "theoretical_loss": 3.624581173692823, + "tokens_seen": 1074200576 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003406820461384152, + "loss": 1.4698, + "theoretical_loss": 3.6245604075220896, + "tokens_seen": 1074266112 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034067201604814446, + "loss": 1.3448, + "theoretical_loss": 3.6245396429728594, + "tokens_seen": 1074331648 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034066198595787364, + "loss": 1.3328, + "theoretical_loss": 3.6245188800449046, + "tokens_seen": 1074397184 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003406519558676028, + "loss": 1.3721, + "theoretical_loss": 3.6244981187380017, + "tokens_seen": 1074462720 + }, + { + "epoch": 0.33, + "learning_rate": 0.000340641925777332, + "loss": 1.3666, + "theoretical_loss": 3.624477359051924, + "tokens_seen": 1074528256 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034063189568706123, + "loss": 1.4076, + "theoretical_loss": 3.6244566009864463, + "tokens_seen": 1074593792 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034062186559679036, + "loss": 1.4322, + "theoretical_loss": 3.6244358445413436, + "tokens_seen": 1074659328 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003406118355065196, + "loss": 1.5773, + "theoretical_loss": 3.6244150897163907, + "tokens_seen": 1074724864 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0002532843791414052, + "objective/train/docs_used": 613082, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.19303297996521, + "objective/train/original_loss": 3.193033218383789, + "objective/train/theoretical_loss": 3.624394336511362, + "objective/train/tokens_used": 1095250400, + "objective/train/value_avg": -0.00493621826171875, + "objective/train/value_loss": 4.534547770163044e-05, + "objective/train/value_max": -0.00044417381286621094, + "objective/train/value_min": -0.06597900390625, + "objective/train/value_reward_corr": 0.48692581551851677, + "objective/train/value_std": 0.00414276123046875, + "objective/train/weight_avg": 0.999659538269043, + "objective/train/weighted_lm_loss": 3.201528787612915, + "objective/train/weights_max": 1.934386134147644, + "objective/train/weights_min": 0.6694967746734619, + "theoretical_loss": 3.624394336511362, + "tokens_seen": 1074790400 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003406018054162487, + "loss": 1.4513, + "theoretical_loss": 3.624394336511362, + "tokens_seen": 1074790400 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034059177532597796, + "loss": 1.454, + "theoretical_loss": 3.6243735849260323, + "tokens_seen": 1074855936 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034058174523570714, + "loss": 1.4723, + "theoretical_loss": 3.624352834960177, + "tokens_seen": 1074921472 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003405717151454363, + "loss": 1.4201, + "theoretical_loss": 3.6243320866135704, + "tokens_seen": 1074987008 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003405616850551655, + "loss": 1.4541, + "theoretical_loss": 3.6243113398859874, + "tokens_seen": 1075052544 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003405516549648947, + "loss": 1.523, + "theoretical_loss": 3.6242905947772033, + "tokens_seen": 1075118080 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034054162487462386, + "loss": 1.4138, + "theoretical_loss": 3.624269851286993, + "tokens_seen": 1075183616 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003405315947843531, + "loss": 1.3953, + "theoretical_loss": 3.624249109415132, + "tokens_seen": 1075249152 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003405215646940822, + "loss": 1.4148, + "theoretical_loss": 3.6242283691613943, + "tokens_seen": 1075314688 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034051153460381146, + "loss": 1.4963, + "theoretical_loss": 3.624207630525556, + "tokens_seen": 1075380224 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034050150451354064, + "loss": 1.3818, + "theoretical_loss": 3.6241868935073924, + "tokens_seen": 1075445760 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003404914744232698, + "loss": 1.4673, + "theoretical_loss": 3.6241661581066777, + "tokens_seen": 1075511296 + }, + { + "epoch": 0.33, + "learning_rate": 0.000340481444332999, + "loss": 1.46, + "theoretical_loss": 3.6241454243231885, + "tokens_seen": 1075576832 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003404714142427282, + "loss": 1.5311, + "theoretical_loss": 3.624124692156699, + "tokens_seen": 1075642368 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034046138415245737, + "loss": 1.4139, + "theoretical_loss": 3.624103961606986, + "tokens_seen": 1075707904 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003404513540621866, + "loss": 1.5083, + "theoretical_loss": 3.624083232673823, + "tokens_seen": 1075773440 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034044132397191573, + "loss": 1.4241, + "theoretical_loss": 3.6240625053569873, + "tokens_seen": 1075838976 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034043129388164497, + "loss": 1.4033, + "theoretical_loss": 3.6240417796562534, + "tokens_seen": 1075904512 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003404212637913741, + "loss": 1.3864, + "theoretical_loss": 3.624021055571397, + "tokens_seen": 1075970048 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034041123370110333, + "loss": 1.4174, + "theoretical_loss": 3.624000333102194, + "tokens_seen": 1076035584 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003404012036108325, + "loss": 1.3359, + "theoretical_loss": 3.62397961224842, + "tokens_seen": 1076101120 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003403911735205617, + "loss": 1.3023, + "theoretical_loss": 3.62395889300985, + "tokens_seen": 1076166656 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034038114343029087, + "loss": 1.2963, + "theoretical_loss": 3.623938175386261, + "tokens_seen": 1076232192 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034037111334002005, + "loss": 1.493, + "theoretical_loss": 3.623917459377428, + "tokens_seen": 1076297728 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034036108324974923, + "loss": 1.5112, + "theoretical_loss": 3.623896744983127, + "tokens_seen": 1076363264 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0011042814003303647, + "objective/train/docs_used": 614240, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.409572124481201, + "objective/train/original_loss": 2.4095723628997803, + "objective/train/theoretical_loss": 3.6238760322031336, + "objective/train/tokens_used": 1096888800, + "objective/train/value_avg": -0.004253387451171875, + "objective/train/value_loss": 6.504255725303665e-05, + "objective/train/value_max": -0.0003528594970703125, + "objective/train/value_min": -0.060211181640625, + "objective/train/value_reward_corr": 0.2596531148661607, + "objective/train/value_std": 0.0032901763916015625, + "objective/train/weight_avg": 0.9918383359909058, + "objective/train/weighted_lm_loss": 2.3952527046203613, + "objective/train/weights_max": 1.6407606601715088, + "objective/train/weights_min": 0.083999864757061, + "theoretical_loss": 3.6238760322031336, + "tokens_seen": 1076428800 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034035105315947847, + "loss": 1.3342, + "theoretical_loss": 3.6238760322031336, + "tokens_seen": 1076428800 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003403410230692076, + "loss": 1.4359, + "theoretical_loss": 3.6238553210372237, + "tokens_seen": 1076494336 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034033099297893683, + "loss": 1.4014, + "theoretical_loss": 3.6238346114851736, + "tokens_seen": 1076559872 + }, + { + "epoch": 0.33, + "learning_rate": 0.000340320962888666, + "loss": 1.4677, + "theoretical_loss": 3.6238139035467594, + "tokens_seen": 1076625408 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003403109327983952, + "loss": 1.2988, + "theoretical_loss": 3.6237931972217567, + "tokens_seen": 1076690944 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003403009027081244, + "loss": 1.3617, + "theoretical_loss": 3.6237724925099424, + "tokens_seen": 1076756480 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034029087261785356, + "loss": 1.4338, + "theoretical_loss": 3.6237517894110916, + "tokens_seen": 1076822016 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034028084252758274, + "loss": 1.4358, + "theoretical_loss": 3.6237310879249813, + "tokens_seen": 1076887552 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034027081243731197, + "loss": 1.4668, + "theoretical_loss": 3.6237103880513875, + "tokens_seen": 1076953088 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003402607823470411, + "loss": 1.372, + "theoretical_loss": 3.623689689790086, + "tokens_seen": 1077018624 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034025075225677033, + "loss": 1.301, + "theoretical_loss": 3.6236689931408543, + "tokens_seen": 1077084160 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034024072216649946, + "loss": 1.4895, + "theoretical_loss": 3.6236482981034674, + "tokens_seen": 1077149696 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003402306920762287, + "loss": 1.3284, + "theoretical_loss": 3.6236276046777025, + "tokens_seen": 1077215232 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003402206619859579, + "loss": 1.297, + "theoretical_loss": 3.6236069128633357, + "tokens_seen": 1077280768 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034021063189568706, + "loss": 1.404, + "theoretical_loss": 3.623586222660144, + "tokens_seen": 1077346304 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034020060180541624, + "loss": 1.364, + "theoretical_loss": 3.6235655340679034, + "tokens_seen": 1077411840 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003401905717151454, + "loss": 1.4737, + "theoretical_loss": 3.623544847086391, + "tokens_seen": 1077477376 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003401805416248746, + "loss": 1.4936, + "theoretical_loss": 3.623524161715383, + "tokens_seen": 1077542912 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034017051153460384, + "loss": 1.3976, + "theoretical_loss": 3.623503477954656, + "tokens_seen": 1077608448 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034016048144433296, + "loss": 1.4658, + "theoretical_loss": 3.6234827958039872, + "tokens_seen": 1077673984 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003401504513540622, + "loss": 1.4161, + "theoretical_loss": 3.6234621152631536, + "tokens_seen": 1077739520 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003401404212637914, + "loss": 1.3966, + "theoretical_loss": 3.623441436331931, + "tokens_seen": 1077805056 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034013039117352056, + "loss": 1.3949, + "theoretical_loss": 3.623420759010097, + "tokens_seen": 1077870592 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034012036108324974, + "loss": 1.4026, + "theoretical_loss": 3.6234000832974282, + "tokens_seen": 1077936128 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003401103309929789, + "loss": 1.3693, + "theoretical_loss": 3.623379409193702, + "tokens_seen": 1078001664 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0008382908999919891, + "objective/train/docs_used": 615029, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9882259368896484, + "objective/train/original_loss": 2.9882256984710693, + "objective/train/theoretical_loss": 3.6233587366986946, + "objective/train/tokens_used": 1098527200, + "objective/train/value_avg": -0.00400543212890625, + "objective/train/value_loss": 3.774848664761521e-05, + "objective/train/value_max": -0.0003859996795654297, + "objective/train/value_min": -0.045440673828125, + "objective/train/value_reward_corr": 0.42973052657918676, + "objective/train/value_std": 0.00341033935546875, + "objective/train/weight_avg": 0.9934412837028503, + "objective/train/weighted_lm_loss": 2.9731392860412598, + "objective/train/weights_max": 1.4383515119552612, + "objective/train/weights_min": 0.5492543578147888, + "theoretical_loss": 3.6233587366986946, + "tokens_seen": 1078067200 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003401003009027081, + "loss": 1.4466, + "theoretical_loss": 3.6233587366986946, + "tokens_seen": 1078067200 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034009027081243734, + "loss": 1.4429, + "theoretical_loss": 3.6233380658121837, + "tokens_seen": 1078132736 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003400802407221665, + "loss": 1.4896, + "theoretical_loss": 3.6233173965339462, + "tokens_seen": 1078198272 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003400702106318957, + "loss": 1.5676, + "theoretical_loss": 3.6232967288637594, + "tokens_seen": 1078263808 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003400601805416249, + "loss": 1.5246, + "theoretical_loss": 3.6232760628014002, + "tokens_seen": 1078329344 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034005015045135406, + "loss": 1.3766, + "theoretical_loss": 3.623255398346646, + "tokens_seen": 1078394880 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003400401203610833, + "loss": 1.3548, + "theoretical_loss": 3.6232347354992736, + "tokens_seen": 1078460416 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003400300902708124, + "loss": 1.5352, + "theoretical_loss": 3.6232140742590606, + "tokens_seen": 1078525952 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034002006018054166, + "loss": 1.3256, + "theoretical_loss": 3.623193414625785, + "tokens_seen": 1078591488 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034001003009027084, + "loss": 1.4248, + "theoretical_loss": 3.6231727565992227, + "tokens_seen": 1078657024 + }, + { + "epoch": 0.33, + "learning_rate": 0.00034, + "loss": 1.4306, + "theoretical_loss": 3.623152100179153, + "tokens_seen": 1078722560 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003399899699097292, + "loss": 1.4367, + "theoretical_loss": 3.6231314453653516, + "tokens_seen": 1078788096 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003399799398194584, + "loss": 1.3556, + "theoretical_loss": 3.6231107921575973, + "tokens_seen": 1078853632 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033996990972918757, + "loss": 1.3183, + "theoretical_loss": 3.623090140555667, + "tokens_seen": 1078919168 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003399598796389168, + "loss": 1.3621, + "theoretical_loss": 3.623069490559339, + "tokens_seen": 1078984704 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033994984954864593, + "loss": 1.3758, + "theoretical_loss": 3.62304884216839, + "tokens_seen": 1079050240 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033993981945837517, + "loss": 1.3394, + "theoretical_loss": 3.6230281953825987, + "tokens_seen": 1079115776 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003399297893681043, + "loss": 1.3091, + "theoretical_loss": 3.623007550201742, + "tokens_seen": 1079181312 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033991975927783353, + "loss": 1.3927, + "theoretical_loss": 3.6229869066255977, + "tokens_seen": 1079246848 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003399097291875627, + "loss": 1.4723, + "theoretical_loss": 3.6229662646539444, + "tokens_seen": 1079312384 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003398996990972919, + "loss": 1.4096, + "theoretical_loss": 3.6229456242865594, + "tokens_seen": 1079377920 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033988966900702107, + "loss": 1.4016, + "theoretical_loss": 3.6229249855232206, + "tokens_seen": 1079443456 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033987963891675025, + "loss": 1.4542, + "theoretical_loss": 3.6229043483637065, + "tokens_seen": 1079508992 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033986960882647943, + "loss": 1.2824, + "theoretical_loss": 3.622883712807795, + "tokens_seen": 1079574528 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033985957873620867, + "loss": 1.366, + "theoretical_loss": 3.6228630788552634, + "tokens_seen": 1079640064 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0011463694972917438, + "objective/train/docs_used": 616160, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.475850820541382, + "objective/train/original_loss": 2.4758505821228027, + "objective/train/theoretical_loss": 3.6228424465058904, + "objective/train/tokens_used": 1100165600, + "objective/train/value_avg": -0.00479888916015625, + "objective/train/value_loss": 6.876705447211862e-05, + "objective/train/value_max": -0.0004012584686279297, + "objective/train/value_min": -0.07501220703125, + "objective/train/value_reward_corr": 0.346905038876858, + "objective/train/value_std": 0.0036945343017578125, + "objective/train/weight_avg": 0.9916856288909912, + "objective/train/weighted_lm_loss": 2.4681458473205566, + "objective/train/weights_max": 1.7262638807296753, + "objective/train/weights_min": 0.38031521439552307, + "theoretical_loss": 3.6228424465058904, + "tokens_seen": 1079705600 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003398495486459378, + "loss": 1.3852, + "theoretical_loss": 3.6228424465058904, + "tokens_seen": 1079705600 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033983951855566703, + "loss": 1.4388, + "theoretical_loss": 3.622821815759454, + "tokens_seen": 1079771136 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003398294884653962, + "loss": 1.405, + "theoretical_loss": 3.6228011866157326, + "tokens_seen": 1079836672 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003398194583751254, + "loss": 1.5201, + "theoretical_loss": 3.6227805590745037, + "tokens_seen": 1079902208 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003398094282848546, + "loss": 1.3474, + "theoretical_loss": 3.622759933135547, + "tokens_seen": 1079967744 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033979939819458376, + "loss": 1.3818, + "theoretical_loss": 3.6227393087986393, + "tokens_seen": 1080033280 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033978936810431294, + "loss": 1.351, + "theoretical_loss": 3.6227186860635596, + "tokens_seen": 1080098816 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033977933801404217, + "loss": 1.3267, + "theoretical_loss": 3.622698064930087, + "tokens_seen": 1080164352 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003397693079237713, + "loss": 1.3437, + "theoretical_loss": 3.622677445397999, + "tokens_seen": 1080229888 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033975927783350053, + "loss": 1.3099, + "theoretical_loss": 3.6226568274670745, + "tokens_seen": 1080295424 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033974924774322966, + "loss": 1.5244, + "theoretical_loss": 3.6226362111370918, + "tokens_seen": 1080360960 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003397392176529589, + "loss": 1.3637, + "theoretical_loss": 3.6226155964078295, + "tokens_seen": 1080426496 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003397291875626881, + "loss": 1.2983, + "theoretical_loss": 3.6225949832790665, + "tokens_seen": 1080492032 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033971915747241726, + "loss": 1.3606, + "theoretical_loss": 3.6225743717505816, + "tokens_seen": 1080557568 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033970912738214644, + "loss": 1.4482, + "theoretical_loss": 3.622553761822153, + "tokens_seen": 1080623104 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003396990972918756, + "loss": 1.5154, + "theoretical_loss": 3.6225331534935594, + "tokens_seen": 1080688640 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003396890672016048, + "loss": 1.3619, + "theoretical_loss": 3.6225125467645807, + "tokens_seen": 1080754176 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033967903711133404, + "loss": 1.4731, + "theoretical_loss": 3.622491941634994, + "tokens_seen": 1080819712 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033966900702106316, + "loss": 1.4394, + "theoretical_loss": 3.62247133810458, + "tokens_seen": 1080885248 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003396589769307924, + "loss": 1.4109, + "theoretical_loss": 3.622450736173116, + "tokens_seen": 1080950784 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003396489468405216, + "loss": 1.2711, + "theoretical_loss": 3.6224301358403825, + "tokens_seen": 1081016320 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033963891675025076, + "loss": 1.3893, + "theoretical_loss": 3.622409537106158, + "tokens_seen": 1081081856 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033962888665997994, + "loss": 1.4139, + "theoretical_loss": 3.6223889399702207, + "tokens_seen": 1081147392 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003396188565697091, + "loss": 1.2886, + "theoretical_loss": 3.6223683444323505, + "tokens_seen": 1081212928 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003396088264794383, + "loss": 1.436, + "theoretical_loss": 3.6223477504923265, + "tokens_seen": 1081278464 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0008748840191401541, + "objective/train/docs_used": 616786, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7892367839813232, + "objective/train/original_loss": 2.7892372608184814, + "objective/train/theoretical_loss": 3.622327158149928, + "objective/train/tokens_used": 1101804000, + "objective/train/value_avg": -0.007190704345703125, + "objective/train/value_loss": 6.207483966136351e-05, + "objective/train/value_max": -0.0003654956817626953, + "objective/train/value_min": -0.0753173828125, + "objective/train/value_reward_corr": 0.47003837150322775, + "objective/train/value_std": 0.006267547607421875, + "objective/train/weight_avg": 0.9942536354064941, + "objective/train/weighted_lm_loss": 2.773573875427246, + "objective/train/weights_max": 1.9629303216934204, + "objective/train/weights_min": 0.2489553540945053, + "theoretical_loss": 3.622327158149928, + "tokens_seen": 1081344000 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033959879638916754, + "loss": 1.3378, + "theoretical_loss": 3.622327158149928, + "tokens_seen": 1081344000 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033958876629889667, + "loss": 1.4733, + "theoretical_loss": 3.622306567404934, + "tokens_seen": 1081409536 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003395787362086259, + "loss": 1.3918, + "theoretical_loss": 3.622285978257124, + "tokens_seen": 1081475072 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033956870611835503, + "loss": 1.3158, + "theoretical_loss": 3.6222653907062776, + "tokens_seen": 1081540608 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033955867602808426, + "loss": 1.3934, + "theoretical_loss": 3.622244804752173, + "tokens_seen": 1081606144 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033954864593781345, + "loss": 1.3705, + "theoretical_loss": 3.6222242203945907, + "tokens_seen": 1081671680 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003395386158475426, + "loss": 1.3999, + "theoretical_loss": 3.6222036376333104, + "tokens_seen": 1081737216 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003395285857572718, + "loss": 1.3802, + "theoretical_loss": 3.622183056468111, + "tokens_seen": 1081802752 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033951855566700104, + "loss": 1.3096, + "theoretical_loss": 3.6221624768987715, + "tokens_seen": 1081868288 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033950852557673017, + "loss": 1.4066, + "theoretical_loss": 3.622141898925072, + "tokens_seen": 1081933824 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003394984954864594, + "loss": 1.3776, + "theoretical_loss": 3.622121322546793, + "tokens_seen": 1081999360 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033948846539618853, + "loss": 1.3733, + "theoretical_loss": 3.6221007477637137, + "tokens_seen": 1082064896 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033947843530591777, + "loss": 1.4316, + "theoretical_loss": 3.622080174575613, + "tokens_seen": 1082130432 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033946840521564695, + "loss": 1.4099, + "theoretical_loss": 3.6220596029822714, + "tokens_seen": 1082195968 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033945837512537613, + "loss": 1.3849, + "theoretical_loss": 3.6220390329834684, + "tokens_seen": 1082261504 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003394483450351053, + "loss": 1.4622, + "theoretical_loss": 3.622018464578985, + "tokens_seen": 1082327040 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003394383149448345, + "loss": 1.5197, + "theoretical_loss": 3.621997897768599, + "tokens_seen": 1082392576 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003394282848545637, + "loss": 1.2903, + "theoretical_loss": 3.621977332552092, + "tokens_seen": 1082458112 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003394182547642929, + "loss": 1.3421, + "theoretical_loss": 3.621956768929243, + "tokens_seen": 1082523648 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033940822467402204, + "loss": 1.4462, + "theoretical_loss": 3.621936206899833, + "tokens_seen": 1082589184 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033939819458375127, + "loss": 1.3985, + "theoretical_loss": 3.6219156464636413, + "tokens_seen": 1082654720 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003393881644934804, + "loss": 1.4458, + "theoretical_loss": 3.621895087620449, + "tokens_seen": 1082720256 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033937813440320963, + "loss": 1.4824, + "theoretical_loss": 3.621874530370035, + "tokens_seen": 1082785792 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003393681043129388, + "loss": 1.4425, + "theoretical_loss": 3.62185397471218, + "tokens_seen": 1082851328 + }, + { + "epoch": 0.33, + "learning_rate": 0.000339358074222668, + "loss": 1.4447, + "theoretical_loss": 3.6218334206466647, + "tokens_seen": 1082916864 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.00046636036131531, + "objective/train/docs_used": 617451, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8882522583007812, + "objective/train/original_loss": 2.888252019882202, + "objective/train/theoretical_loss": 3.6218128681732686, + "objective/train/tokens_used": 1103442400, + "objective/train/value_avg": -0.00481414794921875, + "objective/train/value_loss": 5.8720146626001224e-05, + "objective/train/value_max": -0.0004425048828125, + "objective/train/value_min": -0.0509033203125, + "objective/train/value_reward_corr": 0.3015762085443408, + "objective/train/value_std": 0.003299713134765625, + "objective/train/weight_avg": 0.9980894923210144, + "objective/train/weighted_lm_loss": 2.9013543128967285, + "objective/train/weights_max": 1.4867126941680908, + "objective/train/weights_min": 0.3418758809566498, + "theoretical_loss": 3.6218128681732686, + "tokens_seen": 1082982400 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003393480441323972, + "loss": 1.4401, + "theoretical_loss": 3.6218128681732686, + "tokens_seen": 1082982400 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003393380140421264, + "loss": 1.4605, + "theoretical_loss": 3.621792317291773, + "tokens_seen": 1083047936 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003393279839518556, + "loss": 1.3946, + "theoretical_loss": 3.621771768001958, + "tokens_seen": 1083113472 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003393179538615848, + "loss": 1.4207, + "theoretical_loss": 3.6217512203036026, + "tokens_seen": 1083179008 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033930792377131396, + "loss": 1.4628, + "theoretical_loss": 3.621730674196489, + "tokens_seen": 1083244544 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033929789368104314, + "loss": 1.3878, + "theoretical_loss": 3.6217101296803977, + "tokens_seen": 1083310080 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033928786359077237, + "loss": 1.4667, + "theoretical_loss": 3.6216895867551084, + "tokens_seen": 1083375616 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003392778335005015, + "loss": 1.4914, + "theoretical_loss": 3.621669045420402, + "tokens_seen": 1083441152 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033926780341023073, + "loss": 1.3495, + "theoretical_loss": 3.6216485056760592, + "tokens_seen": 1083506688 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033925777331995986, + "loss": 1.4531, + "theoretical_loss": 3.621627967521861, + "tokens_seen": 1083572224 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003392477432296891, + "loss": 1.3718, + "theoretical_loss": 3.621607430957587, + "tokens_seen": 1083637760 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003392377131394183, + "loss": 1.3339, + "theoretical_loss": 3.621586895983019, + "tokens_seen": 1083703296 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033922768304914746, + "loss": 1.453, + "theoretical_loss": 3.621566362597938, + "tokens_seen": 1083768832 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033921765295887664, + "loss": 1.4888, + "theoretical_loss": 3.621545830802124, + "tokens_seen": 1083834368 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003392076228686058, + "loss": 1.4383, + "theoretical_loss": 3.621525300595359, + "tokens_seen": 1083899904 + }, + { + "epoch": 0.33, + "learning_rate": 0.000339197592778335, + "loss": 1.4499, + "theoretical_loss": 3.6215047719774227, + "tokens_seen": 1083965440 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033918756268806424, + "loss": 1.3787, + "theoretical_loss": 3.621484244948097, + "tokens_seen": 1084030976 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033917753259779336, + "loss": 1.4756, + "theoretical_loss": 3.6214637195071626, + "tokens_seen": 1084096512 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003391675025075226, + "loss": 1.3343, + "theoretical_loss": 3.6214431956544004, + "tokens_seen": 1084162048 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003391574724172518, + "loss": 1.3361, + "theoretical_loss": 3.621422673389592, + "tokens_seen": 1084227584 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033914744232698096, + "loss": 1.4413, + "theoretical_loss": 3.621402152712518, + "tokens_seen": 1084293120 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033913741223671014, + "loss": 1.3356, + "theoretical_loss": 3.62138163362296, + "tokens_seen": 1084358656 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003391273821464393, + "loss": 1.3583, + "theoretical_loss": 3.621361116120699, + "tokens_seen": 1084424192 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003391173520561685, + "loss": 1.465, + "theoretical_loss": 3.621340600205517, + "tokens_seen": 1084489728 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033910732196589774, + "loss": 1.5221, + "theoretical_loss": 3.621320085877194, + "tokens_seen": 1084555264 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0004909891868010163, + "objective/train/docs_used": 618647, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3827664852142334, + "objective/train/original_loss": 3.3827669620513916, + "objective/train/theoretical_loss": 3.621299573135513, + "objective/train/tokens_used": 1105080800, + "objective/train/value_avg": -0.0063934326171875, + "objective/train/value_loss": 0.00010292082151863724, + "objective/train/value_max": -0.0003528594970703125, + "objective/train/value_min": -0.0816650390625, + "objective/train/value_reward_corr": 0.47406156874703265, + "objective/train/value_std": 0.006534576416015625, + "objective/train/weight_avg": 0.9998319149017334, + "objective/train/weighted_lm_loss": 3.3798792362213135, + "objective/train/weights_max": 1.6959775686264038, + "objective/train/weights_min": 0.2518928647041321, + "theoretical_loss": 3.621299573135513, + "tokens_seen": 1084620800 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033909729187562687, + "loss": 1.5, + "theoretical_loss": 3.621299573135513, + "tokens_seen": 1084620800 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003390872617853561, + "loss": 1.402, + "theoretical_loss": 3.621279061980254, + "tokens_seen": 1084686336 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033907723169508523, + "loss": 1.4181, + "theoretical_loss": 3.6212585524111995, + "tokens_seen": 1084751872 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033906720160481446, + "loss": 1.5339, + "theoretical_loss": 3.6212380444281305, + "tokens_seen": 1084817408 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033905717151454365, + "loss": 1.5061, + "theoretical_loss": 3.6212175380308285, + "tokens_seen": 1084882944 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033904714142427283, + "loss": 1.4231, + "theoretical_loss": 3.621197033219075, + "tokens_seen": 1084948480 + }, + { + "epoch": 0.33, + "learning_rate": 0.000339037111334002, + "loss": 1.2803, + "theoretical_loss": 3.621176529992652, + "tokens_seen": 1085014016 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033902708124373124, + "loss": 1.3521, + "theoretical_loss": 3.6211560283513418, + "tokens_seen": 1085079552 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033901705115346037, + "loss": 1.2626, + "theoretical_loss": 3.6211355282949245, + "tokens_seen": 1085145088 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003390070210631896, + "loss": 1.4333, + "theoretical_loss": 3.6211150298231827, + "tokens_seen": 1085210624 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033899699097291873, + "loss": 1.4528, + "theoretical_loss": 3.6210945329358992, + "tokens_seen": 1085276160 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033898696088264797, + "loss": 1.4585, + "theoretical_loss": 3.6210740376328543, + "tokens_seen": 1085341696 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033897693079237715, + "loss": 1.2844, + "theoretical_loss": 3.6210535439138303, + "tokens_seen": 1085407232 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033896690070210633, + "loss": 1.2381, + "theoretical_loss": 3.6210330517786096, + "tokens_seen": 1085472768 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003389568706118355, + "loss": 1.4452, + "theoretical_loss": 3.6210125612269746, + "tokens_seen": 1085538304 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003389468405215647, + "loss": 1.4263, + "theoretical_loss": 3.620992072258706, + "tokens_seen": 1085603840 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003389368104312939, + "loss": 1.396, + "theoretical_loss": 3.620971584873587, + "tokens_seen": 1085669376 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003389267803410231, + "loss": 1.4814, + "theoretical_loss": 3.620951099071399, + "tokens_seen": 1085734912 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033891675025075224, + "loss": 1.4584, + "theoretical_loss": 3.6209306148519245, + "tokens_seen": 1085800448 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033890672016048147, + "loss": 1.3446, + "theoretical_loss": 3.620910132214946, + "tokens_seen": 1085865984 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003388966900702106, + "loss": 1.3263, + "theoretical_loss": 3.620889651160245, + "tokens_seen": 1085931520 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033888665997993983, + "loss": 1.3983, + "theoretical_loss": 3.620869171687604, + "tokens_seen": 1085997056 + }, + { + "epoch": 0.33, + "learning_rate": 0.000338876629889669, + "loss": 1.3062, + "theoretical_loss": 3.620848693796806, + "tokens_seen": 1086062592 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003388665997993982, + "loss": 1.4224, + "theoretical_loss": 3.6208282174876327, + "tokens_seen": 1086128128 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003388565697091274, + "loss": 1.4466, + "theoretical_loss": 3.620807742759867, + "tokens_seen": 1086193664 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": 0.00015551225806120783, + "objective/train/docs_used": 619458, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.40102219581604, + "objective/train/original_loss": 2.401021957397461, + "objective/train/theoretical_loss": 3.620787269613291, + "objective/train/tokens_used": 1106719200, + "objective/train/value_avg": -0.00395965576171875, + "objective/train/value_loss": 3.316716538392939e-05, + "objective/train/value_max": -0.00022876262664794922, + "objective/train/value_min": -0.07696533203125, + "objective/train/value_reward_corr": 0.23011537227960643, + "objective/train/value_std": 0.0032558441162109375, + "objective/train/weight_avg": 1.0030509233474731, + "objective/train/weighted_lm_loss": 2.4169044494628906, + "objective/train/weights_max": 1.993024468421936, + "objective/train/weights_min": 0.14019061625003815, + "theoretical_loss": 3.620787269613291, + "tokens_seen": 1086259200 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003388465396188566, + "loss": 1.3889, + "theoretical_loss": 3.620787269613291, + "tokens_seen": 1086259200 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033883650952858574, + "loss": 1.3505, + "theoretical_loss": 3.6207667980476868, + "tokens_seen": 1086324736 + }, + { + "epoch": 0.33, + "learning_rate": 0.000338826479438315, + "loss": 1.3323, + "theoretical_loss": 3.620746328062838, + "tokens_seen": 1086390272 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003388164493480441, + "loss": 1.4073, + "theoretical_loss": 3.6207258596585263, + "tokens_seen": 1086455808 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033880641925777334, + "loss": 1.3913, + "theoretical_loss": 3.6207053928345347, + "tokens_seen": 1086521344 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003387963891675025, + "loss": 1.4914, + "theoretical_loss": 3.620684927590646, + "tokens_seen": 1086586880 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003387863590772317, + "loss": 1.3646, + "theoretical_loss": 3.6206644639266425, + "tokens_seen": 1086652416 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003387763289869609, + "loss": 1.3169, + "theoretical_loss": 3.620644001842308, + "tokens_seen": 1086717952 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033876629889669006, + "loss": 1.4866, + "theoretical_loss": 3.620623541337424, + "tokens_seen": 1086783488 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033875626880641924, + "loss": 1.4934, + "theoretical_loss": 3.6206030824117734, + "tokens_seen": 1086849024 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003387462387161485, + "loss": 1.4062, + "theoretical_loss": 3.6205826250651403, + "tokens_seen": 1086914560 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003387362086258776, + "loss": 1.5062, + "theoretical_loss": 3.6205621692973065, + "tokens_seen": 1086980096 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033872617853560684, + "loss": 1.3088, + "theoretical_loss": 3.6205417151080557, + "tokens_seen": 1087045632 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033871614844533597, + "loss": 1.3853, + "theoretical_loss": 3.6205212624971708, + "tokens_seen": 1087111168 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003387061183550652, + "loss": 1.4628, + "theoretical_loss": 3.620500811464434, + "tokens_seen": 1087176704 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003386960882647944, + "loss": 1.4083, + "theoretical_loss": 3.62048036200963, + "tokens_seen": 1087242240 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033868605817452356, + "loss": 1.4503, + "theoretical_loss": 3.62045991413254, + "tokens_seen": 1087307776 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033867602808425275, + "loss": 1.4586, + "theoretical_loss": 3.620439467832949, + "tokens_seen": 1087373312 + }, + { + "epoch": 0.33, + "learning_rate": 0.000338665997993982, + "loss": 1.347, + "theoretical_loss": 3.620419023110639, + "tokens_seen": 1087438848 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003386559679037111, + "loss": 1.5395, + "theoretical_loss": 3.620398579965394, + "tokens_seen": 1087504384 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033864593781344034, + "loss": 1.3584, + "theoretical_loss": 3.620378138396997, + "tokens_seen": 1087569920 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033863590772316947, + "loss": 1.3741, + "theoretical_loss": 3.6203576984052317, + "tokens_seen": 1087635456 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003386258776328987, + "loss": 1.3805, + "theoretical_loss": 3.6203372599898804, + "tokens_seen": 1087700992 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003386158475426279, + "loss": 1.3949, + "theoretical_loss": 3.620316823150728, + "tokens_seen": 1087766528 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033860581745235707, + "loss": 1.3616, + "theoretical_loss": 3.6202963878875574, + "tokens_seen": 1087832064 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0002860055828932673, + "objective/train/docs_used": 620460, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.099874496459961, + "objective/train/original_loss": 3.099874496459961, + "objective/train/theoretical_loss": 3.620275954200152, + "objective/train/tokens_used": 1108357600, + "objective/train/value_avg": -0.005390167236328125, + "objective/train/value_loss": 5.068505197414197e-05, + "objective/train/value_max": -0.00042057037353515625, + "objective/train/value_min": -0.058441162109375, + "objective/train/value_reward_corr": 0.4838414323750607, + "objective/train/value_std": 0.00463104248046875, + "objective/train/weight_avg": 0.9995958209037781, + "objective/train/weighted_lm_loss": 3.088527202606201, + "objective/train/weights_max": 1.7939351797103882, + "objective/train/weights_min": 0.5474972128868103, + "theoretical_loss": 3.620275954200152, + "tokens_seen": 1087897600 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033859578736208625, + "loss": 1.5081, + "theoretical_loss": 3.620275954200152, + "tokens_seen": 1087897600 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033858575727181543, + "loss": 1.3654, + "theoretical_loss": 3.6202555220882955, + "tokens_seen": 1087963136 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033857572718154467, + "loss": 1.3482, + "theoretical_loss": 3.6202350915517716, + "tokens_seen": 1088028672 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033856569709127385, + "loss": 1.4403, + "theoretical_loss": 3.6202146625903637, + "tokens_seen": 1088094208 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033855566700100303, + "loss": 1.4109, + "theoretical_loss": 3.620194235203856, + "tokens_seen": 1088159744 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003385456369107322, + "loss": 1.4909, + "theoretical_loss": 3.620173809392032, + "tokens_seen": 1088225280 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033853560682046144, + "loss": 1.408, + "theoretical_loss": 3.6201533851546746, + "tokens_seen": 1088290816 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033852557673019057, + "loss": 1.4074, + "theoretical_loss": 3.620132962491569, + "tokens_seen": 1088356352 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003385155466399198, + "loss": 1.4556, + "theoretical_loss": 3.6201125414024986, + "tokens_seen": 1088421888 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033850551654964893, + "loss": 1.399, + "theoretical_loss": 3.620092121887247, + "tokens_seen": 1088487424 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033849548645937817, + "loss": 1.4404, + "theoretical_loss": 3.6200717039455985, + "tokens_seen": 1088552960 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033848545636910735, + "loss": 1.3503, + "theoretical_loss": 3.620051287577337, + "tokens_seen": 1088618496 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033847542627883653, + "loss": 1.4566, + "theoretical_loss": 3.620030872782247, + "tokens_seen": 1088684032 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003384653961885657, + "loss": 1.3637, + "theoretical_loss": 3.6200104595601115, + "tokens_seen": 1088749568 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003384553660982949, + "loss": 1.4174, + "theoretical_loss": 3.619990047910716, + "tokens_seen": 1088815104 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003384453360080241, + "loss": 1.3727, + "theoretical_loss": 3.6199696378338433, + "tokens_seen": 1088880640 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003384353059177533, + "loss": 1.3843, + "theoretical_loss": 3.6199492293292783, + "tokens_seen": 1088946176 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033842527582748244, + "loss": 1.4135, + "theoretical_loss": 3.6199288223968056, + "tokens_seen": 1089011712 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033841524573721167, + "loss": 1.5047, + "theoretical_loss": 3.619908417036209, + "tokens_seen": 1089077248 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003384052156469408, + "loss": 1.3401, + "theoretical_loss": 3.619888013247273, + "tokens_seen": 1089142784 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033839518555667003, + "loss": 1.3911, + "theoretical_loss": 3.619867611029782, + "tokens_seen": 1089208320 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003383851554663992, + "loss": 1.4285, + "theoretical_loss": 3.6198472103835204, + "tokens_seen": 1089273856 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003383751253761284, + "loss": 1.4168, + "theoretical_loss": 3.6198268113082728, + "tokens_seen": 1089339392 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003383650952858576, + "loss": 1.3246, + "theoretical_loss": 3.6198064138038237, + "tokens_seen": 1089404928 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003383550651955868, + "loss": 1.4428, + "theoretical_loss": 3.619786017869957, + "tokens_seen": 1089470464 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.00012866416363976896, + "objective/train/docs_used": 620976, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1461832523345947, + "objective/train/original_loss": 3.1461832523345947, + "objective/train/theoretical_loss": 3.619765623506458, + "objective/train/tokens_used": 1109996000, + "objective/train/value_avg": -0.005367279052734375, + "objective/train/value_loss": 5.996710024191998e-05, + "objective/train/value_max": -0.0004239082336425781, + "objective/train/value_min": -0.052825927734375, + "objective/train/value_reward_corr": 0.32615937428993375, + "objective/train/value_std": 0.00397491455078125, + "objective/train/weight_avg": 1.0015010833740234, + "objective/train/weighted_lm_loss": 3.143662452697754, + "objective/train/weights_max": 1.5932234525680542, + "objective/train/weights_min": 0.33617910742759705, + "theoretical_loss": 3.619765623506458, + "tokens_seen": 1089536000 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033834503510531594, + "loss": 1.3806, + "theoretical_loss": 3.619765623506458, + "tokens_seen": 1089536000 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003383350050150452, + "loss": 1.4087, + "theoretical_loss": 3.6197452307131117, + "tokens_seen": 1089601536 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003383249749247743, + "loss": 1.4202, + "theoretical_loss": 3.619724839489702, + "tokens_seen": 1089667072 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033831494483450354, + "loss": 1.4933, + "theoretical_loss": 3.6197044498360142, + "tokens_seen": 1089732608 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003383049147442327, + "loss": 1.3312, + "theoretical_loss": 3.6196840617518324, + "tokens_seen": 1089798144 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003382948846539619, + "loss": 1.3767, + "theoretical_loss": 3.6196636752369415, + "tokens_seen": 1089863680 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003382848545636911, + "loss": 1.338, + "theoretical_loss": 3.619643290291127, + "tokens_seen": 1089929216 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033827482447342026, + "loss": 1.3529, + "theoretical_loss": 3.6196229069141737, + "tokens_seen": 1089994752 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033826479438314944, + "loss": 1.3393, + "theoretical_loss": 3.6196025251058663, + "tokens_seen": 1090060288 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003382547642928787, + "loss": 1.3707, + "theoretical_loss": 3.6195821448659897, + "tokens_seen": 1090125824 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003382447342026078, + "loss": 1.404, + "theoretical_loss": 3.6195617661943285, + "tokens_seen": 1090191360 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033823470411233704, + "loss": 1.3242, + "theoretical_loss": 3.619541389090669, + "tokens_seen": 1090256896 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033822467402206617, + "loss": 1.3963, + "theoretical_loss": 3.6195210135547953, + "tokens_seen": 1090322432 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003382146439317954, + "loss": 1.472, + "theoretical_loss": 3.6195006395864926, + "tokens_seen": 1090387968 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003382046138415246, + "loss": 1.5565, + "theoretical_loss": 3.619480267185547, + "tokens_seen": 1090453504 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033819458375125376, + "loss": 1.4201, + "theoretical_loss": 3.619459896351742, + "tokens_seen": 1090519040 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033818455366098295, + "loss": 1.4395, + "theoretical_loss": 3.6194395270848645, + "tokens_seen": 1090584576 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003381745235707122, + "loss": 1.3902, + "theoretical_loss": 3.6194191593847, + "tokens_seen": 1090650112 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003381644934804413, + "loss": 1.4726, + "theoretical_loss": 3.6193987932510323, + "tokens_seen": 1090715648 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033815446339017054, + "loss": 1.3451, + "theoretical_loss": 3.619378428683648, + "tokens_seen": 1090781184 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033814443329989967, + "loss": 1.3699, + "theoretical_loss": 3.6193580656823316, + "tokens_seen": 1090846720 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003381344032096289, + "loss": 1.2788, + "theoretical_loss": 3.6193377042468695, + "tokens_seen": 1090912256 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003381243731193581, + "loss": 1.3035, + "theoretical_loss": 3.6193173443770474, + "tokens_seen": 1090977792 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033811434302908727, + "loss": 1.4743, + "theoretical_loss": 3.6192969860726496, + "tokens_seen": 1091043328 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033810431293881645, + "loss": 1.4842, + "theoretical_loss": 3.619276629333463, + "tokens_seen": 1091108864 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.00028466928051784635, + "objective/train/docs_used": 621707, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.898002862930298, + "objective/train/original_loss": 2.8980023860931396, + "objective/train/theoretical_loss": 3.6192562741592726, + "objective/train/tokens_used": 1111634400, + "objective/train/value_avg": -0.004795074462890625, + "objective/train/value_loss": 5.96865065745078e-05, + "objective/train/value_max": -0.00033020973205566406, + "objective/train/value_min": -0.057281494140625, + "objective/train/value_reward_corr": 0.3406529279693875, + "objective/train/value_std": 0.00368499755859375, + "objective/train/weight_avg": 0.9999261498451233, + "objective/train/weighted_lm_loss": 2.903517246246338, + "objective/train/weights_max": 1.5580272674560547, + "objective/train/weights_min": 0.3174784183502197, + "theoretical_loss": 3.6192562741592726, + "tokens_seen": 1091174400 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033809428284854563, + "loss": 1.336, + "theoretical_loss": 3.6192562741592726, + "tokens_seen": 1091174400 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003380842527582748, + "loss": 1.4384, + "theoretical_loss": 3.6192359205498645, + "tokens_seen": 1091239936 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033807422266800405, + "loss": 1.4129, + "theoretical_loss": 3.619215568505024, + "tokens_seen": 1091305472 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003380641925777332, + "loss": 1.4195, + "theoretical_loss": 3.6191952180245375, + "tokens_seen": 1091371008 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003380541624874624, + "loss": 1.4973, + "theoretical_loss": 3.61917486910819, + "tokens_seen": 1091436544 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033804413239719154, + "loss": 1.3787, + "theoretical_loss": 3.6191545217557675, + "tokens_seen": 1091502080 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033803410230692077, + "loss": 1.3716, + "theoretical_loss": 3.6191341759670568, + "tokens_seen": 1091567616 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033802407221664995, + "loss": 1.4902, + "theoretical_loss": 3.619113831741843, + "tokens_seen": 1091633152 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033801404212637913, + "loss": 1.3688, + "theoretical_loss": 3.619093489079913, + "tokens_seen": 1091698688 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003380040120361083, + "loss": 1.4907, + "theoretical_loss": 3.6190731479810516, + "tokens_seen": 1091764224 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033799398194583755, + "loss": 1.4177, + "theoretical_loss": 3.619052808445045, + "tokens_seen": 1091829760 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003379839518555667, + "loss": 1.4144, + "theoretical_loss": 3.619032470471681, + "tokens_seen": 1091895296 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003379739217652959, + "loss": 1.4666, + "theoretical_loss": 3.6190121340607435, + "tokens_seen": 1091960832 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033796389167502504, + "loss": 1.4712, + "theoretical_loss": 3.6189917992120204, + "tokens_seen": 1092026368 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003379538615847543, + "loss": 1.3673, + "theoretical_loss": 3.618971465925297, + "tokens_seen": 1092091904 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033794383149448346, + "loss": 1.3549, + "theoretical_loss": 3.61895113420036, + "tokens_seen": 1092157440 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033793380140421264, + "loss": 1.3354, + "theoretical_loss": 3.618930804036996, + "tokens_seen": 1092222976 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003379237713139418, + "loss": 1.3821, + "theoretical_loss": 3.618910475434991, + "tokens_seen": 1092288512 + }, + { + "epoch": 0.33, + "learning_rate": 0.000337913741223671, + "loss": 1.4032, + "theoretical_loss": 3.618890148394131, + "tokens_seen": 1092354048 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003379037111334002, + "loss": 1.4893, + "theoretical_loss": 3.6188698229142036, + "tokens_seen": 1092419584 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003378936810431294, + "loss": 1.3504, + "theoretical_loss": 3.618849498994994, + "tokens_seen": 1092485120 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033788365095285854, + "loss": 1.4163, + "theoretical_loss": 3.61882917663629, + "tokens_seen": 1092550656 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003378736208625878, + "loss": 1.4253, + "theoretical_loss": 3.618808855837877, + "tokens_seen": 1092616192 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003378635907723169, + "loss": 1.4673, + "theoretical_loss": 3.618788536599543, + "tokens_seen": 1092681728 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033785356068204614, + "loss": 1.4823, + "theoretical_loss": 3.6187682189210726, + "tokens_seen": 1092747264 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.00018335029017180204, + "objective/train/docs_used": 622698, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0862293243408203, + "objective/train/original_loss": 3.0862293243408203, + "objective/train/theoretical_loss": 3.6187479028022547, + "objective/train/tokens_used": 1113272800, + "objective/train/value_avg": -0.004901885986328125, + "objective/train/value_loss": 5.0172151532024145e-05, + "objective/train/value_max": -0.0002512931823730469, + "objective/train/value_min": -0.10430908203125, + "objective/train/value_reward_corr": 0.4263549803592916, + "objective/train/value_std": 0.00440216064453125, + "objective/train/weight_avg": 1.0005918741226196, + "objective/train/weighted_lm_loss": 3.090240478515625, + "objective/train/weights_max": 2.196094512939453, + "objective/train/weights_min": 0.5114759206771851, + "theoretical_loss": 3.6187479028022547, + "tokens_seen": 1092812800 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003378435305917753, + "loss": 1.4448, + "theoretical_loss": 3.6187479028022547, + "tokens_seen": 1092812800 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003378335005015045, + "loss": 1.3026, + "theoretical_loss": 3.6187275882428747, + "tokens_seen": 1092878336 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033782347041123374, + "loss": 1.5027, + "theoretical_loss": 3.61870727524272, + "tokens_seen": 1092943872 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003378134403209629, + "loss": 1.3234, + "theoretical_loss": 3.6186869638015775, + "tokens_seen": 1093009408 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003378034102306921, + "loss": 1.3614, + "theoretical_loss": 3.6186666539192336, + "tokens_seen": 1093074944 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003377933801404213, + "loss": 1.3578, + "theoretical_loss": 3.618646345595475, + "tokens_seen": 1093140480 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033778335005015046, + "loss": 1.3807, + "theoretical_loss": 3.6186260388300897, + "tokens_seen": 1093206016 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033777331995987964, + "loss": 1.4818, + "theoretical_loss": 3.6186057336228643, + "tokens_seen": 1093271552 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003377632898696089, + "loss": 1.3845, + "theoretical_loss": 3.618585429973585, + "tokens_seen": 1093337088 + }, + { + "epoch": 0.33, + "learning_rate": 0.000337753259779338, + "loss": 1.5489, + "theoretical_loss": 3.6185651278820403, + "tokens_seen": 1093402624 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033774322968906724, + "loss": 1.4993, + "theoretical_loss": 3.618544827348016, + "tokens_seen": 1093468160 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033773319959879637, + "loss": 1.2866, + "theoretical_loss": 3.6185245283713003, + "tokens_seen": 1093533696 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003377231695085256, + "loss": 1.4422, + "theoretical_loss": 3.6185042309516797, + "tokens_seen": 1093599232 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003377131394182548, + "loss": 1.551, + "theoretical_loss": 3.6184839350889417, + "tokens_seen": 1093664768 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033770310932798396, + "loss": 1.4785, + "theoretical_loss": 3.618463640782874, + "tokens_seen": 1093730304 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033769307923771315, + "loss": 1.4007, + "theoretical_loss": 3.618443348033263, + "tokens_seen": 1093795840 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003376830491474424, + "loss": 1.4134, + "theoretical_loss": 3.618423056839897, + "tokens_seen": 1093861376 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003376730190571715, + "loss": 1.2969, + "theoretical_loss": 3.6184027672025634, + "tokens_seen": 1093926912 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033766298896690074, + "loss": 1.311, + "theoretical_loss": 3.618382479121049, + "tokens_seen": 1093992448 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033765295887662987, + "loss": 1.4004, + "theoretical_loss": 3.618362192595142, + "tokens_seen": 1094057984 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003376429287863591, + "loss": 1.3808, + "theoretical_loss": 3.6183419076246293, + "tokens_seen": 1094123520 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003376328986960883, + "loss": 1.29, + "theoretical_loss": 3.618321624209299, + "tokens_seen": 1094189056 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033762286860581747, + "loss": 1.4454, + "theoretical_loss": 3.6183013423489383, + "tokens_seen": 1094254592 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033761283851554665, + "loss": 1.4351, + "theoretical_loss": 3.618281062043335, + "tokens_seen": 1094320128 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033760280842527583, + "loss": 1.5802, + "theoretical_loss": 3.6182607832922775, + "tokens_seen": 1094385664 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": 0.0007825143402442336, + "objective/train/docs_used": 623416, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3587119579315186, + "objective/train/original_loss": 3.3587114810943604, + "objective/train/theoretical_loss": 3.6182405060955523, + "objective/train/tokens_used": 1114911200, + "objective/train/value_avg": -0.005100250244140625, + "objective/train/value_loss": 4.659877231460996e-05, + "objective/train/value_max": -0.0005092620849609375, + "objective/train/value_min": -0.048309326171875, + "objective/train/value_reward_corr": 0.34555603631235, + "objective/train/value_std": 0.0038280487060546875, + "objective/train/weight_avg": 1.0099551677703857, + "objective/train/weighted_lm_loss": 3.393878221511841, + "objective/train/weights_max": 1.5185164213180542, + "objective/train/weights_min": 0.1386059820652008, + "theoretical_loss": 3.6182405060955523, + "tokens_seen": 1094451200 + }, + { + "epoch": 0.33, + "learning_rate": 0.000337592778335005, + "loss": 1.5106, + "theoretical_loss": 3.6182405060955523, + "tokens_seen": 1094451200 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033758274824473425, + "loss": 1.5115, + "theoretical_loss": 3.6182202304529483, + "tokens_seen": 1094516736 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003375727181544634, + "loss": 1.4585, + "theoretical_loss": 3.6181999563642533, + "tokens_seen": 1094582272 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003375626880641926, + "loss": 1.4172, + "theoretical_loss": 3.618179683829254, + "tokens_seen": 1094647808 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033755265797392174, + "loss": 1.5173, + "theoretical_loss": 3.6181594128477395, + "tokens_seen": 1094713344 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033754262788365097, + "loss": 1.2538, + "theoretical_loss": 3.618139143419498, + "tokens_seen": 1094778880 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033753259779338015, + "loss": 1.4827, + "theoretical_loss": 3.618118875544316, + "tokens_seen": 1094844416 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033752256770310933, + "loss": 1.4526, + "theoretical_loss": 3.618098609221983, + "tokens_seen": 1094909952 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003375125376128385, + "loss": 1.4233, + "theoretical_loss": 3.618078344452286, + "tokens_seen": 1094975488 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033750250752256775, + "loss": 1.1944, + "theoretical_loss": 3.6180580812350143, + "tokens_seen": 1095041024 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003374924774322969, + "loss": 1.467, + "theoretical_loss": 3.618037819569955, + "tokens_seen": 1095106560 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003374824473420261, + "loss": 1.3269, + "theoretical_loss": 3.6180175594568973, + "tokens_seen": 1095172096 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033747241725175524, + "loss": 1.4054, + "theoretical_loss": 3.6179973008956283, + "tokens_seen": 1095237632 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003374623871614845, + "loss": 1.2463, + "theoretical_loss": 3.6179770438859373, + "tokens_seen": 1095303168 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033745235707121366, + "loss": 1.3688, + "theoretical_loss": 3.6179567884276125, + "tokens_seen": 1095368704 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033744232698094284, + "loss": 1.3541, + "theoretical_loss": 3.6179365345204415, + "tokens_seen": 1095434240 + }, + { + "epoch": 0.33, + "learning_rate": 0.000337432296890672, + "loss": 1.4308, + "theoretical_loss": 3.617916282164214, + "tokens_seen": 1095499776 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003374222668004012, + "loss": 1.3709, + "theoretical_loss": 3.6178960313587174, + "tokens_seen": 1095565312 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003374122367101304, + "loss": 1.3466, + "theoretical_loss": 3.61787578210374, + "tokens_seen": 1095630848 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003374022066198596, + "loss": 1.4753, + "theoretical_loss": 3.6178555343990713, + "tokens_seen": 1095696384 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033739217652958874, + "loss": 1.4926, + "theoretical_loss": 3.6178352882444997, + "tokens_seen": 1095761920 + }, + { + "epoch": 0.33, + "learning_rate": 0.000337382146439318, + "loss": 1.4557, + "theoretical_loss": 3.6178150436398138, + "tokens_seen": 1095827456 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003373721163490471, + "loss": 1.4332, + "theoretical_loss": 3.6177948005848015, + "tokens_seen": 1095892992 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033736208625877634, + "loss": 1.4696, + "theoretical_loss": 3.617774559079253, + "tokens_seen": 1095958528 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003373520561685055, + "loss": 1.3884, + "theoretical_loss": 3.6177543191229553, + "tokens_seen": 1096024064 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0006048284121789038, + "objective/train/docs_used": 624708, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.2736330032348633, + "objective/train/original_loss": 2.2736334800720215, + "objective/train/theoretical_loss": 3.617734080715698, + "objective/train/tokens_used": 1116549600, + "objective/train/value_avg": -0.0047607421875, + "objective/train/value_loss": 4.434125003172085e-05, + "objective/train/value_max": -0.0004012584686279297, + "objective/train/value_min": -0.045013427734375, + "objective/train/value_reward_corr": 0.2806781289891227, + "objective/train/value_std": 0.0033721923828125, + "objective/train/weight_avg": 0.9960418939590454, + "objective/train/weighted_lm_loss": 2.2636265754699707, + "objective/train/weights_max": 1.3982083797454834, + "objective/train/weights_min": 0.43956783413887024, + "theoretical_loss": 3.617734080715698, + "tokens_seen": 1096089600 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003373420260782347, + "loss": 1.2654, + "theoretical_loss": 3.617734080715698, + "tokens_seen": 1096089600 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003373319959879639, + "loss": 1.442, + "theoretical_loss": 3.6177138438572705, + "tokens_seen": 1096155136 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003373219658976931, + "loss": 1.333, + "theoretical_loss": 3.617693608547461, + "tokens_seen": 1096220672 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033731193580742225, + "loss": 1.3568, + "theoretical_loss": 3.6176733747860585, + "tokens_seen": 1096286208 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003373019057171515, + "loss": 1.4413, + "theoretical_loss": 3.617653142572852, + "tokens_seen": 1096351744 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003372918756268806, + "loss": 1.4805, + "theoretical_loss": 3.6176329119076307, + "tokens_seen": 1096417280 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033728184553660984, + "loss": 1.383, + "theoretical_loss": 3.6176126827901838, + "tokens_seen": 1096482816 + }, + { + "epoch": 0.33, + "learning_rate": 0.000337271815446339, + "loss": 1.4603, + "theoretical_loss": 3.6175924552203, + "tokens_seen": 1096548352 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003372617853560682, + "loss": 1.403, + "theoretical_loss": 3.6175722291977683, + "tokens_seen": 1096613888 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003372517552657974, + "loss": 1.4327, + "theoretical_loss": 3.617552004722378, + "tokens_seen": 1096679424 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033724172517552657, + "loss": 1.3746, + "theoretical_loss": 3.617531781793919, + "tokens_seen": 1096744960 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033723169508525575, + "loss": 1.43, + "theoretical_loss": 3.6175115604121793, + "tokens_seen": 1096810496 + }, + { + "epoch": 0.33, + "learning_rate": 0.000337221664994985, + "loss": 1.5234, + "theoretical_loss": 3.617491340576949, + "tokens_seen": 1096876032 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003372116349047141, + "loss": 1.3788, + "theoretical_loss": 3.6174711222880176, + "tokens_seen": 1096941568 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033720160481444335, + "loss": 1.4708, + "theoretical_loss": 3.617450905545174, + "tokens_seen": 1097007104 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003371915747241725, + "loss": 1.3951, + "theoretical_loss": 3.6174306903482076, + "tokens_seen": 1097072640 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003371815446339017, + "loss": 1.4865, + "theoretical_loss": 3.617410476696908, + "tokens_seen": 1097138176 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003371715145436309, + "loss": 1.3857, + "theoretical_loss": 3.617390264591065, + "tokens_seen": 1097203712 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033716148445336007, + "loss": 1.3629, + "theoretical_loss": 3.617370054030468, + "tokens_seen": 1097269248 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033715145436308925, + "loss": 1.4526, + "theoretical_loss": 3.617349845014906, + "tokens_seen": 1097334784 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003371414242728185, + "loss": 1.3208, + "theoretical_loss": 3.6173296375441697, + "tokens_seen": 1097400320 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003371313941825476, + "loss": 1.3678, + "theoretical_loss": 3.6173094316180476, + "tokens_seen": 1097465856 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033712136409227685, + "loss": 1.4164, + "theoretical_loss": 3.6172892272363306, + "tokens_seen": 1097531392 + }, + { + "epoch": 0.33, + "learning_rate": 0.000337111334002006, + "loss": 1.3852, + "theoretical_loss": 3.617269024398807, + "tokens_seen": 1097596928 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003371013039117352, + "loss": 1.3855, + "theoretical_loss": 3.6172488231052675, + "tokens_seen": 1097662464 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0014485574793070555, + "objective/train/docs_used": 625328, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.457493782043457, + "objective/train/original_loss": 2.457493782043457, + "objective/train/theoretical_loss": 3.617228623355502, + "objective/train/tokens_used": 1118188000, + "objective/train/value_avg": -0.00432586669921875, + "objective/train/value_loss": 6.182686774991453e-05, + "objective/train/value_max": -0.00033283233642578125, + "objective/train/value_min": -0.05419921875, + "objective/train/value_reward_corr": 0.32770543851880035, + "objective/train/value_std": 0.0031108856201171875, + "objective/train/weight_avg": 0.9882275462150574, + "objective/train/weighted_lm_loss": 2.445622205734253, + "objective/train/weights_max": 1.5033785104751587, + "objective/train/weights_min": 0.13959559798240662, + "theoretical_loss": 3.617228623355502, + "tokens_seen": 1097728000 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003370912738214644, + "loss": 1.3504, + "theoretical_loss": 3.617228623355502, + "tokens_seen": 1097728000 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003370812437311936, + "loss": 1.3663, + "theoretical_loss": 3.6172084251493004, + "tokens_seen": 1097793536 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003370712136409228, + "loss": 1.5011, + "theoretical_loss": 3.6171882284864525, + "tokens_seen": 1097859072 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033706118355065194, + "loss": 1.4441, + "theoretical_loss": 3.617168033366748, + "tokens_seen": 1097924608 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033705115346038117, + "loss": 1.2843, + "theoretical_loss": 3.617147839789977, + "tokens_seen": 1097990144 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033704112337011035, + "loss": 1.5252, + "theoretical_loss": 3.6171276477559293, + "tokens_seen": 1098055680 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033703109327983953, + "loss": 1.324, + "theoretical_loss": 3.6171074572643955, + "tokens_seen": 1098121216 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003370210631895687, + "loss": 1.463, + "theoretical_loss": 3.6170872683151654, + "tokens_seen": 1098186752 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033701103309929795, + "loss": 1.3065, + "theoretical_loss": 3.6170670809080296, + "tokens_seen": 1098252288 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003370010030090271, + "loss": 1.4505, + "theoretical_loss": 3.617046895042778, + "tokens_seen": 1098317824 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003369909729187563, + "loss": 1.4592, + "theoretical_loss": 3.6170267107192, + "tokens_seen": 1098383360 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033698094282848544, + "loss": 1.4044, + "theoretical_loss": 3.617006527937087, + "tokens_seen": 1098448896 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003369709127382147, + "loss": 1.4469, + "theoretical_loss": 3.6169863466962293, + "tokens_seen": 1098514432 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033696088264794386, + "loss": 1.4931, + "theoretical_loss": 3.6169661669964164, + "tokens_seen": 1098579968 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033695085255767304, + "loss": 1.3556, + "theoretical_loss": 3.61694598883744, + "tokens_seen": 1098645504 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003369408224674022, + "loss": 1.359, + "theoretical_loss": 3.6169258122190895, + "tokens_seen": 1098711040 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003369307923771314, + "loss": 1.4663, + "theoretical_loss": 3.6169056371411554, + "tokens_seen": 1098776576 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003369207622868606, + "loss": 1.4116, + "theoretical_loss": 3.6168854636034284, + "tokens_seen": 1098842112 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003369107321965898, + "loss": 1.2316, + "theoretical_loss": 3.6168652916056994, + "tokens_seen": 1098907648 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033690070210631894, + "loss": 1.3312, + "theoretical_loss": 3.616845121147759, + "tokens_seen": 1098973184 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003368906720160482, + "loss": 1.3294, + "theoretical_loss": 3.6168249522293974, + "tokens_seen": 1099038720 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003368806419257773, + "loss": 1.3676, + "theoretical_loss": 3.6168047848504052, + "tokens_seen": 1099104256 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033687061183550654, + "loss": 1.3881, + "theoretical_loss": 3.6167846190105744, + "tokens_seen": 1099169792 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003368605817452357, + "loss": 1.5701, + "theoretical_loss": 3.616764454709694, + "tokens_seen": 1099235328 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003368505516549649, + "loss": 1.4197, + "theoretical_loss": 3.6167442919475556, + "tokens_seen": 1099300864 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -5.529194095288403e-05, + "objective/train/docs_used": 626712, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7953920364379883, + "objective/train/original_loss": 2.7953922748565674, + "objective/train/theoretical_loss": 3.616724130723951, + "objective/train/tokens_used": 1119826400, + "objective/train/value_avg": -0.004169464111328125, + "objective/train/value_loss": 4.4256266846787184e-05, + "objective/train/value_max": -0.0003407001495361328, + "objective/train/value_min": -0.05340576171875, + "objective/train/value_reward_corr": 0.39607033531237745, + "objective/train/value_std": 0.003387451171875, + "objective/train/weight_avg": 1.0014897584915161, + "objective/train/weighted_lm_loss": 2.8118813037872314, + "objective/train/weights_max": 1.4906461238861084, + "objective/train/weights_min": 0.29577764868736267, + "theoretical_loss": 3.616724130723951, + "tokens_seen": 1099366400 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003368405215646941, + "loss": 1.3707, + "theoretical_loss": 3.616724130723951, + "tokens_seen": 1099366400 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003368304914744233, + "loss": 1.4027, + "theoretical_loss": 3.6167039710386693, + "tokens_seen": 1099431936 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033682046138415245, + "loss": 1.3445, + "theoretical_loss": 3.6166838128915026, + "tokens_seen": 1099497472 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003368104312938817, + "loss": 1.4609, + "theoretical_loss": 3.616663656282242, + "tokens_seen": 1099563008 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003368004012036108, + "loss": 1.4263, + "theoretical_loss": 3.6166435012106777, + "tokens_seen": 1099628544 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033679037111334004, + "loss": 1.4947, + "theoretical_loss": 3.616623347676601, + "tokens_seen": 1099694080 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003367803410230692, + "loss": 1.4854, + "theoretical_loss": 3.6166031956798035, + "tokens_seen": 1099759616 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003367703109327984, + "loss": 1.418, + "theoretical_loss": 3.616583045220076, + "tokens_seen": 1099825152 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003367602808425276, + "loss": 1.4552, + "theoretical_loss": 3.61656289629721, + "tokens_seen": 1099890688 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033675025075225677, + "loss": 1.5106, + "theoretical_loss": 3.6165427489109963, + "tokens_seen": 1099956224 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033674022066198595, + "loss": 1.4691, + "theoretical_loss": 3.6165226030612256, + "tokens_seen": 1100021760 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003367301905717152, + "loss": 1.3915, + "theoretical_loss": 3.6165024587476906, + "tokens_seen": 1100087296 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003367201604814443, + "loss": 1.4674, + "theoretical_loss": 3.616482315970182, + "tokens_seen": 1100152832 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033671013039117355, + "loss": 1.3304, + "theoretical_loss": 3.6164621747284906, + "tokens_seen": 1100218368 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033670010030090273, + "loss": 1.5363, + "theoretical_loss": 3.616442035022409, + "tokens_seen": 1100283904 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003366900702106319, + "loss": 1.4317, + "theoretical_loss": 3.616421896851728, + "tokens_seen": 1100349440 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003366800401203611, + "loss": 1.4024, + "theoretical_loss": 3.616401760216238, + "tokens_seen": 1100414976 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033667001003009027, + "loss": 1.4441, + "theoretical_loss": 3.616381625115733, + "tokens_seen": 1100480512 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033665997993981945, + "loss": 1.3745, + "theoretical_loss": 3.616361491550003, + "tokens_seen": 1100546048 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003366499498495487, + "loss": 1.4576, + "theoretical_loss": 3.61634135951884, + "tokens_seen": 1100611584 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003366399197592778, + "loss": 1.4807, + "theoretical_loss": 3.616321229022035, + "tokens_seen": 1100677120 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033662988966900705, + "loss": 1.3457, + "theoretical_loss": 3.6163011000593803, + "tokens_seen": 1100742656 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003366198595787362, + "loss": 1.3172, + "theoretical_loss": 3.616280972630668, + "tokens_seen": 1100808192 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003366098294884654, + "loss": 1.4185, + "theoretical_loss": 3.6162608467356896, + "tokens_seen": 1100873728 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003365997993981946, + "loss": 1.4433, + "theoretical_loss": 3.6162407223742368, + "tokens_seen": 1100939264 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0012295544147491455, + "objective/train/docs_used": 627263, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7531626224517822, + "objective/train/original_loss": 2.7531626224517822, + "objective/train/theoretical_loss": 3.616220599546101, + "objective/train/tokens_used": 1121464800, + "objective/train/value_avg": -0.005840301513671875, + "objective/train/value_loss": 8.050406904658303e-05, + "objective/train/value_max": -0.0002397298812866211, + "objective/train/value_min": -0.07330322265625, + "objective/train/value_reward_corr": 0.3943140198945385, + "objective/train/value_std": 0.00485992431640625, + "objective/train/weight_avg": 0.9914242625236511, + "objective/train/weighted_lm_loss": 2.739084005355835, + "objective/train/weights_max": 1.5480741262435913, + "objective/train/weights_min": 0.2510007917881012, + "theoretical_loss": 3.616220599546101, + "tokens_seen": 1101004800 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003365897693079238, + "loss": 1.3884, + "theoretical_loss": 3.616220599546101, + "tokens_seen": 1101004800 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033657973921765295, + "loss": 1.4543, + "theoretical_loss": 3.6162004782510753, + "tokens_seen": 1101070336 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033656970912738214, + "loss": 1.3955, + "theoretical_loss": 3.616180358488951, + "tokens_seen": 1101135872 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003365596790371113, + "loss": 1.5544, + "theoretical_loss": 3.6161602402595197, + "tokens_seen": 1101201408 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033654964894684055, + "loss": 1.387, + "theoretical_loss": 3.616140123562574, + "tokens_seen": 1101266944 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003365396188565697, + "loss": 1.3052, + "theoretical_loss": 3.616120008397906, + "tokens_seen": 1101332480 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003365295887662989, + "loss": 1.4216, + "theoretical_loss": 3.6160998947653074, + "tokens_seen": 1101398016 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003365195586760281, + "loss": 1.4596, + "theoretical_loss": 3.616079782664571, + "tokens_seen": 1101463552 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003365095285857573, + "loss": 1.4856, + "theoretical_loss": 3.6160596720954885, + "tokens_seen": 1101529088 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033649949849548646, + "loss": 1.4776, + "theoretical_loss": 3.6160395630578517, + "tokens_seen": 1101594624 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033648946840521564, + "loss": 1.3961, + "theoretical_loss": 3.616019455551454, + "tokens_seen": 1101660160 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003364794383149448, + "loss": 1.4039, + "theoretical_loss": 3.6159993495760867, + "tokens_seen": 1101725696 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033646940822467406, + "loss": 1.306, + "theoretical_loss": 3.615979245131543, + "tokens_seen": 1101791232 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003364593781344032, + "loss": 1.3649, + "theoretical_loss": 3.6159591422176147, + "tokens_seen": 1101856768 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003364493480441324, + "loss": 1.4557, + "theoretical_loss": 3.6159390408340943, + "tokens_seen": 1101922304 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033643931795386154, + "loss": 1.3351, + "theoretical_loss": 3.615918940980775, + "tokens_seen": 1101987840 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003364292878635908, + "loss": 1.4586, + "theoretical_loss": 3.615898842657448, + "tokens_seen": 1102053376 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033641925777331996, + "loss": 1.4143, + "theoretical_loss": 3.615878745863907, + "tokens_seen": 1102118912 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033640922768304914, + "loss": 1.3982, + "theoretical_loss": 3.615858650599944, + "tokens_seen": 1102184448 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003363991975927783, + "loss": 1.4166, + "theoretical_loss": 3.615838556865352, + "tokens_seen": 1102249984 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003363891675025075, + "loss": 1.2802, + "theoretical_loss": 3.615818464659924, + "tokens_seen": 1102315520 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003363791374122367, + "loss": 1.4503, + "theoretical_loss": 3.615798373983451, + "tokens_seen": 1102381056 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003363691073219659, + "loss": 1.518, + "theoretical_loss": 3.615778284835728, + "tokens_seen": 1102446592 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033635907723169505, + "loss": 1.4329, + "theoretical_loss": 3.6157581972165467, + "tokens_seen": 1102512128 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003363490471414243, + "loss": 1.5931, + "theoretical_loss": 3.6157381111257, + "tokens_seen": 1102577664 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": 0.0005165822221897542, + "objective/train/docs_used": 628470, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8273627758026123, + "objective/train/original_loss": 2.8273627758026123, + "objective/train/theoretical_loss": 3.6157180265629814, + "objective/train/tokens_used": 1123103200, + "objective/train/value_avg": -0.005229949951171875, + "objective/train/value_loss": 4.206439552945085e-05, + "objective/train/value_max": -0.00030541419982910156, + "objective/train/value_min": -0.1500244140625, + "objective/train/value_reward_corr": 0.5170397034133503, + "objective/train/value_std": 0.004467010498046875, + "objective/train/weight_avg": 1.0072805881500244, + "objective/train/weighted_lm_loss": 2.8448190689086914, + "objective/train/weights_max": 4.482783317565918, + "objective/train/weights_min": 0.6264002323150635, + "theoretical_loss": 3.6157180265629814, + "tokens_seen": 1102643200 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033633901705115346, + "loss": 1.4449, + "theoretical_loss": 3.6157180265629814, + "tokens_seen": 1102643200 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033632898696088265, + "loss": 1.4112, + "theoretical_loss": 3.6156979435281826, + "tokens_seen": 1102708736 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003363189568706119, + "loss": 1.4301, + "theoretical_loss": 3.6156778620210974, + "tokens_seen": 1102774272 + }, + { + "epoch": 0.33, + "learning_rate": 0.000336308926780341, + "loss": 1.398, + "theoretical_loss": 3.6156577820415188, + "tokens_seen": 1102839808 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033629889669007024, + "loss": 1.4554, + "theoretical_loss": 3.6156377035892397, + "tokens_seen": 1102905344 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003362888665997994, + "loss": 1.3691, + "theoretical_loss": 3.6156176266640534, + "tokens_seen": 1102970880 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003362788365095286, + "loss": 1.4285, + "theoretical_loss": 3.6155975512657523, + "tokens_seen": 1103036416 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003362688064192578, + "loss": 1.4219, + "theoretical_loss": 3.6155774773941305, + "tokens_seen": 1103101952 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033625877632898697, + "loss": 1.4199, + "theoretical_loss": 3.615557405048981, + "tokens_seen": 1103167488 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033624874623871615, + "loss": 1.47, + "theoretical_loss": 3.6155373342300967, + "tokens_seen": 1103233024 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003362387161484454, + "loss": 1.4764, + "theoretical_loss": 3.615517264937271, + "tokens_seen": 1103298560 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003362286860581745, + "loss": 1.4899, + "theoretical_loss": 3.6154971971702974, + "tokens_seen": 1103364096 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033621865596790375, + "loss": 1.2185, + "theoretical_loss": 3.615477130928969, + "tokens_seen": 1103429632 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033620862587763293, + "loss": 1.436, + "theoretical_loss": 3.6154570662130796, + "tokens_seen": 1103495168 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003361985957873621, + "loss": 1.4174, + "theoretical_loss": 3.6154370030224223, + "tokens_seen": 1103560704 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003361885656970913, + "loss": 1.3764, + "theoretical_loss": 3.6154169413567905, + "tokens_seen": 1103626240 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033617853560682047, + "loss": 1.3565, + "theoretical_loss": 3.615396881215978, + "tokens_seen": 1103691776 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033616850551654965, + "loss": 1.4051, + "theoretical_loss": 3.615376822599779, + "tokens_seen": 1103757312 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003361584754262789, + "loss": 1.3797, + "theoretical_loss": 3.6153567655079852, + "tokens_seen": 1103822848 + }, + { + "epoch": 0.33, + "learning_rate": 0.000336148445336008, + "loss": 1.4418, + "theoretical_loss": 3.6153367099403924, + "tokens_seen": 1103888384 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033613841524573725, + "loss": 1.4438, + "theoretical_loss": 3.615316655896793, + "tokens_seen": 1103953920 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003361283851554664, + "loss": 1.3591, + "theoretical_loss": 3.615296603376981, + "tokens_seen": 1104019456 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003361183550651956, + "loss": 1.3811, + "theoretical_loss": 3.6152765523807506, + "tokens_seen": 1104084992 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003361083249749248, + "loss": 1.4347, + "theoretical_loss": 3.615256502907896, + "tokens_seen": 1104150528 + }, + { + "epoch": 0.33, + "learning_rate": 0.000336098294884654, + "loss": 1.3866, + "theoretical_loss": 3.6152364549582092, + "tokens_seen": 1104216064 + }, + { + "epoch": 0.33, + "objective/train/advantage_avg": -0.0013012639246881008, + "objective/train/docs_used": 629051, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.894463539123535, + "objective/train/original_loss": 2.8944637775421143, + "objective/train/theoretical_loss": 3.6152164085314853, + "objective/train/tokens_used": 1124741600, + "objective/train/value_avg": -0.007144927978515625, + "objective/train/value_loss": 0.00010914557060459629, + "objective/train/value_max": -0.0004322528839111328, + "objective/train/value_min": -0.1688232421875, + "objective/train/value_reward_corr": 0.5153391055980154, + "objective/train/value_std": 0.00872039794921875, + "objective/train/weight_avg": 0.9922108054161072, + "objective/train/weighted_lm_loss": 2.8863327503204346, + "objective/train/weights_max": 3.9100770950317383, + "objective/train/weights_min": 0.20148390531539917, + "theoretical_loss": 3.6152164085314853, + "tokens_seen": 1104281600 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033608826479438316, + "loss": 1.4604, + "theoretical_loss": 3.6152164085314853, + "tokens_seen": 1104281600 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033607823470411234, + "loss": 1.4409, + "theoretical_loss": 3.615196363627519, + "tokens_seen": 1104347136 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003360682046138415, + "loss": 1.4477, + "theoretical_loss": 3.615176320246103, + "tokens_seen": 1104412672 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033605817452357075, + "loss": 1.4773, + "theoretical_loss": 3.6151562783870315, + "tokens_seen": 1104478208 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003360481444332999, + "loss": 1.4903, + "theoretical_loss": 3.615136238050099, + "tokens_seen": 1104543744 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003360381143430291, + "loss": 1.3873, + "theoretical_loss": 3.6151161992350995, + "tokens_seen": 1104609280 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003360280842527583, + "loss": 1.4641, + "theoretical_loss": 3.615096161941827, + "tokens_seen": 1104674816 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003360180541624875, + "loss": 1.4121, + "theoretical_loss": 3.6150761261700755, + "tokens_seen": 1104740352 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033600802407221666, + "loss": 1.4171, + "theoretical_loss": 3.61505609191964, + "tokens_seen": 1104805888 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033599799398194584, + "loss": 1.4106, + "theoretical_loss": 3.6150360591903135, + "tokens_seen": 1104871424 + }, + { + "epoch": 0.33, + "learning_rate": 0.000335987963891675, + "loss": 1.4507, + "theoretical_loss": 3.6150160279818913, + "tokens_seen": 1104936960 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033597793380140426, + "loss": 1.4104, + "theoretical_loss": 3.6149959982941677, + "tokens_seen": 1105002496 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003359679037111334, + "loss": 1.3473, + "theoretical_loss": 3.6149759701269364, + "tokens_seen": 1105068032 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003359578736208626, + "loss": 1.3826, + "theoretical_loss": 3.614955943479992, + "tokens_seen": 1105133568 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033594784353059174, + "loss": 1.279, + "theoretical_loss": 3.6149359183531296, + "tokens_seen": 1105199104 + }, + { + "epoch": 0.33, + "learning_rate": 0.000335937813440321, + "loss": 1.4069, + "theoretical_loss": 3.614915894746143, + "tokens_seen": 1105264640 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033592778335005016, + "loss": 1.328, + "theoretical_loss": 3.614895872658827, + "tokens_seen": 1105330176 + }, + { + "epoch": 0.33, + "learning_rate": 0.00033591775325977934, + "loss": 1.246, + "theoretical_loss": 3.6148758520909765, + "tokens_seen": 1105395712 + }, + { + "epoch": 0.33, + "learning_rate": 0.0003359077231695085, + "loss": 1.4124, + "theoretical_loss": 3.6148558330423852, + "tokens_seen": 1105461248 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003358976930792377, + "loss": 1.4938, + "theoretical_loss": 3.6148358155128486, + "tokens_seen": 1105526784 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003358876629889669, + "loss": 1.5361, + "theoretical_loss": 3.6148157995021615, + "tokens_seen": 1105592320 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003358776328986961, + "loss": 1.4156, + "theoretical_loss": 3.614795785010118, + "tokens_seen": 1105657856 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033586760280842525, + "loss": 1.3282, + "theoretical_loss": 3.614775772036513, + "tokens_seen": 1105723392 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003358575727181545, + "loss": 1.3546, + "theoretical_loss": 3.6147557605811413, + "tokens_seen": 1105788928 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033584754262788366, + "loss": 1.4396, + "theoretical_loss": 3.6147357506437983, + "tokens_seen": 1105854464 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0008455727365799248, + "objective/train/docs_used": 630354, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7494518756866455, + "objective/train/original_loss": 2.7494518756866455, + "objective/train/theoretical_loss": 3.614715742224278, + "objective/train/tokens_used": 1126380000, + "objective/train/value_avg": -0.005523681640625, + "objective/train/value_loss": 6.834603118477389e-05, + "objective/train/value_max": -0.0004322528839111328, + "objective/train/value_min": -0.0439453125, + "objective/train/value_reward_corr": 0.3198547364100062, + "objective/train/value_std": 0.003932952880859375, + "objective/train/weight_avg": 0.9947651624679565, + "objective/train/weighted_lm_loss": 2.7245290279388428, + "objective/train/weights_max": 1.4702568054199219, + "objective/train/weights_min": 0.3677152097225189, + "theoretical_loss": 3.614715742224278, + "tokens_seen": 1105920000 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033583751253761285, + "loss": 1.4189, + "theoretical_loss": 3.614715742224278, + "tokens_seen": 1105920000 + }, + { + "epoch": 0.34, + "learning_rate": 0.000335827482447342, + "loss": 1.3809, + "theoretical_loss": 3.614695735322376, + "tokens_seen": 1105985536 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003358174523570712, + "loss": 1.406, + "theoretical_loss": 3.614675729937887, + "tokens_seen": 1106051072 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003358074222668004, + "loss": 1.4414, + "theoretical_loss": 3.6146557260706063, + "tokens_seen": 1106116608 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003357973921765296, + "loss": 1.3791, + "theoretical_loss": 3.614635723720329, + "tokens_seen": 1106182144 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033578736208625875, + "loss": 1.3471, + "theoretical_loss": 3.614615722886849, + "tokens_seen": 1106247680 + }, + { + "epoch": 0.34, + "learning_rate": 0.000335777331995988, + "loss": 1.3482, + "theoretical_loss": 3.6145957235699635, + "tokens_seen": 1106313216 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003357673019057171, + "loss": 1.3818, + "theoretical_loss": 3.614575725769466, + "tokens_seen": 1106378752 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033575727181544635, + "loss": 1.3852, + "theoretical_loss": 3.6145557294851525, + "tokens_seen": 1106444288 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033574724172517553, + "loss": 1.5089, + "theoretical_loss": 3.6145357347168177, + "tokens_seen": 1106509824 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003357372116349047, + "loss": 1.5466, + "theoretical_loss": 3.614515741464257, + "tokens_seen": 1106575360 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003357271815446339, + "loss": 1.4452, + "theoretical_loss": 3.6144957497272667, + "tokens_seen": 1106640896 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033571715145436313, + "loss": 1.4311, + "theoretical_loss": 3.614475759505641, + "tokens_seen": 1106706432 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033570712136409225, + "loss": 1.3862, + "theoretical_loss": 3.6144557707991756, + "tokens_seen": 1106771968 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003356970912738215, + "loss": 1.4464, + "theoretical_loss": 3.614435783607666, + "tokens_seen": 1106837504 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003356870611835506, + "loss": 1.4246, + "theoretical_loss": 3.614415797930908, + "tokens_seen": 1106903040 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033567703109327985, + "loss": 1.341, + "theoretical_loss": 3.614395813768697, + "tokens_seen": 1106968576 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033566700100300903, + "loss": 1.4683, + "theoretical_loss": 3.614375831120828, + "tokens_seen": 1107034112 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003356569709127382, + "loss": 1.4145, + "theoretical_loss": 3.6143558499870974, + "tokens_seen": 1107099648 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003356469408224674, + "loss": 1.4346, + "theoretical_loss": 3.6143358703673005, + "tokens_seen": 1107165184 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003356369107321966, + "loss": 1.2736, + "theoretical_loss": 3.6143158922612333, + "tokens_seen": 1107230720 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033562688064192576, + "loss": 1.4315, + "theoretical_loss": 3.614295915668691, + "tokens_seen": 1107296256 + }, + { + "epoch": 0.34, + "learning_rate": 0.000335616850551655, + "loss": 1.3346, + "theoretical_loss": 3.6142759405894695, + "tokens_seen": 1107361792 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003356068204613841, + "loss": 1.4733, + "theoretical_loss": 3.6142559670233645, + "tokens_seen": 1107427328 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033559679037111336, + "loss": 1.3829, + "theoretical_loss": 3.6142359949701723, + "tokens_seen": 1107492864 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0009850459173321724, + "objective/train/docs_used": 630918, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.881864070892334, + "objective/train/original_loss": 2.881863832473755, + "objective/train/theoretical_loss": 3.6142160244296884, + "objective/train/tokens_used": 1128018400, + "objective/train/value_avg": -0.00531005859375, + "objective/train/value_loss": 6.732402835041285e-05, + "objective/train/value_max": -0.00031757354736328125, + "objective/train/value_min": -0.0726318359375, + "objective/train/value_reward_corr": 0.43703522879312223, + "objective/train/value_std": 0.004871368408203125, + "objective/train/weight_avg": 0.993353009223938, + "objective/train/weighted_lm_loss": 2.877373456954956, + "objective/train/weights_max": 1.8286151885986328, + "objective/train/weights_min": 0.4725986421108246, + "theoretical_loss": 3.6142160244296884, + "tokens_seen": 1107558400 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003355867602808425, + "loss": 1.3784, + "theoretical_loss": 3.6142160244296884, + "tokens_seen": 1107558400 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003355767301905717, + "loss": 1.4798, + "theoretical_loss": 3.614196055401709, + "tokens_seen": 1107623936 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033556670010030095, + "loss": 1.3605, + "theoretical_loss": 3.61417608788603, + "tokens_seen": 1107689472 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003355566700100301, + "loss": 1.3366, + "theoretical_loss": 3.6141561218824476, + "tokens_seen": 1107755008 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003355466399197593, + "loss": 1.3886, + "theoretical_loss": 3.6141361573907567, + "tokens_seen": 1107820544 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003355366098294885, + "loss": 1.4091, + "theoretical_loss": 3.614116194410755, + "tokens_seen": 1107886080 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003355265797392177, + "loss": 1.3863, + "theoretical_loss": 3.614096232942238, + "tokens_seen": 1107951616 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033551654964894686, + "loss": 1.4366, + "theoretical_loss": 3.6140762729850016, + "tokens_seen": 1108017152 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033550651955867604, + "loss": 1.309, + "theoretical_loss": 3.614056314538842, + "tokens_seen": 1108082688 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003354964894684052, + "loss": 1.4262, + "theoretical_loss": 3.6140363576035552, + "tokens_seen": 1108148224 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033548645937813446, + "loss": 1.387, + "theoretical_loss": 3.6140164021789385, + "tokens_seen": 1108213760 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003354764292878636, + "loss": 1.4155, + "theoretical_loss": 3.6139964482647873, + "tokens_seen": 1108279296 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003354663991975928, + "loss": 1.4749, + "theoretical_loss": 3.613976495860898, + "tokens_seen": 1108344832 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033545636910732195, + "loss": 1.3872, + "theoretical_loss": 3.6139565449670683, + "tokens_seen": 1108410368 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003354463390170512, + "loss": 1.4191, + "theoretical_loss": 3.613936595583093, + "tokens_seen": 1108475904 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033543630892678036, + "loss": 1.4184, + "theoretical_loss": 3.613916647708769, + "tokens_seen": 1108541440 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033542627883650954, + "loss": 1.3156, + "theoretical_loss": 3.613896701343893, + "tokens_seen": 1108606976 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003354162487462387, + "loss": 1.3942, + "theoretical_loss": 3.6138767564882617, + "tokens_seen": 1108672512 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003354062186559679, + "loss": 1.3634, + "theoretical_loss": 3.613856813141672, + "tokens_seen": 1108738048 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003353961885656971, + "loss": 1.2962, + "theoretical_loss": 3.6138368713039197, + "tokens_seen": 1108803584 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003353861584754263, + "loss": 1.417, + "theoretical_loss": 3.613816930974802, + "tokens_seen": 1108869120 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033537612838515545, + "loss": 1.3836, + "theoretical_loss": 3.613796992154115, + "tokens_seen": 1108934656 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003353660982948847, + "loss": 1.4182, + "theoretical_loss": 3.6137770548416563, + "tokens_seen": 1109000192 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033535606820461386, + "loss": 1.383, + "theoretical_loss": 3.6137571190372224, + "tokens_seen": 1109065728 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033534603811434305, + "loss": 1.3963, + "theoretical_loss": 3.613737184740611, + "tokens_seen": 1109131264 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0002302096545463428, + "objective/train/docs_used": 632203, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.777298927307129, + "objective/train/original_loss": 2.777298927307129, + "objective/train/theoretical_loss": 3.6137172519516163, + "objective/train/tokens_used": 1129656800, + "objective/train/value_avg": -0.005615234375, + "objective/train/value_loss": 6.531155668199062e-05, + "objective/train/value_max": -0.000293731689453125, + "objective/train/value_min": -0.09686279296875, + "objective/train/value_reward_corr": 0.46275669666567587, + "objective/train/value_std": 0.005100250244140625, + "objective/train/weight_avg": 1.000796914100647, + "objective/train/weighted_lm_loss": 2.7743148803710938, + "objective/train/weights_max": 2.1524388790130615, + "objective/train/weights_min": 0.4965475797653198, + "theoretical_loss": 3.6137172519516163, + "tokens_seen": 1109196800 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003353360080240722, + "loss": 1.3893, + "theoretical_loss": 3.6137172519516163, + "tokens_seen": 1109196800 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003353259779338014, + "loss": 1.3871, + "theoretical_loss": 3.6136973206700382, + "tokens_seen": 1109262336 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003353159478435306, + "loss": 1.5066, + "theoretical_loss": 3.6136773908956723, + "tokens_seen": 1109327872 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003353059177532598, + "loss": 1.5005, + "theoretical_loss": 3.613657462628315, + "tokens_seen": 1109393408 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033529588766298895, + "loss": 1.283, + "theoretical_loss": 3.613637535867765, + "tokens_seen": 1109458944 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003352858575727182, + "loss": 1.4999, + "theoretical_loss": 3.613617610613818, + "tokens_seen": 1109524480 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003352758274824473, + "loss": 1.4157, + "theoretical_loss": 3.613597686866271, + "tokens_seen": 1109590016 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033526579739217655, + "loss": 1.4371, + "theoretical_loss": 3.6135777646249223, + "tokens_seen": 1109655552 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033525576730190573, + "loss": 1.3864, + "theoretical_loss": 3.613557843889568, + "tokens_seen": 1109721088 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003352457372116349, + "loss": 1.3612, + "theoretical_loss": 3.6135379246600063, + "tokens_seen": 1109786624 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003352357071213641, + "loss": 1.4746, + "theoretical_loss": 3.613518006936034, + "tokens_seen": 1109852160 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033522567703109333, + "loss": 1.343, + "theoretical_loss": 3.613498090717448, + "tokens_seen": 1109917696 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033521564694082245, + "loss": 1.45, + "theoretical_loss": 3.613478176004046, + "tokens_seen": 1109983232 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003352056168505517, + "loss": 1.3809, + "theoretical_loss": 3.6134582627956258, + "tokens_seen": 1110048768 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003351955867602808, + "loss": 1.3614, + "theoretical_loss": 3.6134383510919843, + "tokens_seen": 1110114304 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033518555667001005, + "loss": 1.4984, + "theoretical_loss": 3.613418440892919, + "tokens_seen": 1110179840 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033517552657973923, + "loss": 1.4556, + "theoretical_loss": 3.6133985321982274, + "tokens_seen": 1110245376 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003351654964894684, + "loss": 1.4009, + "theoretical_loss": 3.613378625007707, + "tokens_seen": 1110310912 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003351554663991976, + "loss": 1.3289, + "theoretical_loss": 3.613358719321156, + "tokens_seen": 1110376448 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003351454363089268, + "loss": 1.3973, + "theoretical_loss": 3.613338815138371, + "tokens_seen": 1110441984 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033513540621865596, + "loss": 1.3371, + "theoretical_loss": 3.6133189124591505, + "tokens_seen": 1110507520 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003351253761283852, + "loss": 1.4401, + "theoretical_loss": 3.613299011283292, + "tokens_seen": 1110573056 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003351153460381143, + "loss": 1.3996, + "theoretical_loss": 3.613279111610593, + "tokens_seen": 1110638592 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033510531594784356, + "loss": 1.334, + "theoretical_loss": 3.613259213440851, + "tokens_seen": 1110704128 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003350952858575727, + "loss": 1.3965, + "theoretical_loss": 3.613239316773865, + "tokens_seen": 1110769664 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0025044255889952183, + "objective/train/docs_used": 632721, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.688045024871826, + "objective/train/original_loss": 2.6880452632904053, + "objective/train/theoretical_loss": 3.6132194216094313, + "objective/train/tokens_used": 1131295200, + "objective/train/value_avg": -0.005584716796875, + "objective/train/value_loss": 7.710432691965252e-05, + "objective/train/value_max": -0.00023055076599121094, + "objective/train/value_min": -0.06646728515625, + "objective/train/value_reward_corr": 0.4520828169531191, + "objective/train/value_std": 0.00440216064453125, + "objective/train/weight_avg": 0.9785670638084412, + "objective/train/weighted_lm_loss": 2.6373238563537598, + "objective/train/weights_max": 1.5265053510665894, + "objective/train/weights_min": 0.411236435174942, + "theoretical_loss": 3.6132194216094313, + "tokens_seen": 1110835200 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003350852557673019, + "loss": 1.4223, + "theoretical_loss": 3.6132194216094313, + "tokens_seen": 1110835200 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003350752256770311, + "loss": 1.4061, + "theoretical_loss": 3.6131995279473488, + "tokens_seen": 1110900736 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003350651955867603, + "loss": 1.4014, + "theoretical_loss": 3.6131796357874157, + "tokens_seen": 1110966272 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033505516549648946, + "loss": 1.3599, + "theoretical_loss": 3.613159745129429, + "tokens_seen": 1111031808 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003350451354062187, + "loss": 1.4139, + "theoretical_loss": 3.6131398559731873, + "tokens_seen": 1111097344 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003350351053159478, + "loss": 1.4208, + "theoretical_loss": 3.6131199683184887, + "tokens_seen": 1111162880 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033502507522567706, + "loss": 1.503, + "theoretical_loss": 3.613100082165131, + "tokens_seen": 1111228416 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003350150451354062, + "loss": 1.5111, + "theoretical_loss": 3.6130801975129123, + "tokens_seen": 1111293952 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003350050150451354, + "loss": 1.4261, + "theoretical_loss": 3.613060314361631, + "tokens_seen": 1111359488 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003349949849548646, + "loss": 1.5099, + "theoretical_loss": 3.6130404327110854, + "tokens_seen": 1111425024 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003349849548645938, + "loss": 1.3556, + "theoretical_loss": 3.613020552561074, + "tokens_seen": 1111490560 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033497492477432296, + "loss": 1.4027, + "theoretical_loss": 3.613000673911394, + "tokens_seen": 1111556096 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033496489468405215, + "loss": 1.4283, + "theoretical_loss": 3.6129807967618444, + "tokens_seen": 1111621632 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003349548645937813, + "loss": 1.3819, + "theoretical_loss": 3.6129609211122236, + "tokens_seen": 1111687168 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033494483450351056, + "loss": 1.4608, + "theoretical_loss": 3.61294104696233, + "tokens_seen": 1111752704 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003349348044132397, + "loss": 1.4309, + "theoretical_loss": 3.6129211743119622, + "tokens_seen": 1111818240 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003349247743229689, + "loss": 1.3256, + "theoretical_loss": 3.6129013031609185, + "tokens_seen": 1111883776 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033491474423269805, + "loss": 1.4779, + "theoretical_loss": 3.612881433508997, + "tokens_seen": 1111949312 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003349047141424273, + "loss": 1.3462, + "theoretical_loss": 3.612861565355997, + "tokens_seen": 1112014848 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033489468405215647, + "loss": 1.4283, + "theoretical_loss": 3.6128416987017165, + "tokens_seen": 1112080384 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033488465396188565, + "loss": 1.4263, + "theoretical_loss": 3.6128218335459543, + "tokens_seen": 1112145920 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033487462387161483, + "loss": 1.5425, + "theoretical_loss": 3.612801969888509, + "tokens_seen": 1112211456 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033486459378134406, + "loss": 1.2645, + "theoretical_loss": 3.6127821077291804, + "tokens_seen": 1112276992 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003348545636910732, + "loss": 1.4189, + "theoretical_loss": 3.6127622470677654, + "tokens_seen": 1112342528 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003348445336008024, + "loss": 1.3089, + "theoretical_loss": 3.6127423879040643, + "tokens_seen": 1112408064 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0015665647806599736, + "objective/train/docs_used": 634025, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3762714862823486, + "objective/train/original_loss": 2.3762712478637695, + "objective/train/theoretical_loss": 3.612722530237875, + "objective/train/tokens_used": 1132933600, + "objective/train/value_avg": -0.00508880615234375, + "objective/train/value_loss": 0.00010529123392188922, + "objective/train/value_max": -0.0002378225326538086, + "objective/train/value_min": -0.06719970703125, + "objective/train/value_reward_corr": 0.20974028903584296, + "objective/train/value_std": 0.0037174224853515625, + "objective/train/weight_avg": 0.9886195659637451, + "objective/train/weighted_lm_loss": 2.3639016151428223, + "objective/train/weights_max": 1.7333325147628784, + "objective/train/weights_min": 0.14493237435817719, + "theoretical_loss": 3.612722530237875, + "tokens_seen": 1112473600 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003348345035105316, + "loss": 1.3084, + "theoretical_loss": 3.612722530237875, + "tokens_seen": 1112473600 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003348244734202608, + "loss": 1.4706, + "theoretical_loss": 3.6127026740689967, + "tokens_seen": 1112539136 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033481444332999, + "loss": 1.3914, + "theoretical_loss": 3.6126828193972287, + "tokens_seen": 1112604672 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033480441323971915, + "loss": 1.4527, + "theoretical_loss": 3.612662966222369, + "tokens_seen": 1112670208 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003347943831494484, + "loss": 1.4684, + "theoretical_loss": 3.612643114544218, + "tokens_seen": 1112735744 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003347843530591775, + "loss": 1.2858, + "theoretical_loss": 3.612623264362573, + "tokens_seen": 1112801280 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033477432296890675, + "loss": 1.376, + "theoretical_loss": 3.612603415677234, + "tokens_seen": 1112866816 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033476429287863593, + "loss": 1.3369, + "theoretical_loss": 3.6125835684880006, + "tokens_seen": 1112932352 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003347542627883651, + "loss": 1.4195, + "theoretical_loss": 3.612563722794671, + "tokens_seen": 1112997888 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003347442326980943, + "loss": 1.5433, + "theoretical_loss": 3.612543878597045, + "tokens_seen": 1113063424 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033473420260782353, + "loss": 1.4781, + "theoretical_loss": 3.6125240358949213, + "tokens_seen": 1113128960 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033472417251755265, + "loss": 1.3134, + "theoretical_loss": 3.6125041946881, + "tokens_seen": 1113194496 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003347141424272819, + "loss": 1.3989, + "theoretical_loss": 3.612484354976379, + "tokens_seen": 1113260032 + }, + { + "epoch": 0.34, + "learning_rate": 0.000334704112337011, + "loss": 1.4551, + "theoretical_loss": 3.612464516759559, + "tokens_seen": 1113325568 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033469408224674025, + "loss": 1.3589, + "theoretical_loss": 3.612444680037439, + "tokens_seen": 1113391104 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033468405215646943, + "loss": 1.3098, + "theoretical_loss": 3.6124248448098175, + "tokens_seen": 1113456640 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003346740220661986, + "loss": 1.4086, + "theoretical_loss": 3.612405011076495, + "tokens_seen": 1113522176 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003346639919759278, + "loss": 1.4036, + "theoretical_loss": 3.612385178837271, + "tokens_seen": 1113587712 + }, + { + "epoch": 0.34, + "learning_rate": 0.000334653961885657, + "loss": 1.4406, + "theoretical_loss": 3.612365348091945, + "tokens_seen": 1113653248 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033464393179538616, + "loss": 1.3958, + "theoretical_loss": 3.612345518840316, + "tokens_seen": 1113718784 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003346339017051154, + "loss": 1.3913, + "theoretical_loss": 3.6123256910821837, + "tokens_seen": 1113784320 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003346238716148445, + "loss": 1.4238, + "theoretical_loss": 3.612305864817348, + "tokens_seen": 1113849856 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033461384152457376, + "loss": 1.4268, + "theoretical_loss": 3.6122860400456087, + "tokens_seen": 1113915392 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003346038114343029, + "loss": 1.3861, + "theoretical_loss": 3.612266216766765, + "tokens_seen": 1113980928 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003345937813440321, + "loss": 1.3918, + "theoretical_loss": 3.612246394980617, + "tokens_seen": 1114046464 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": 5.0778737204382196e-05, + "objective/train/docs_used": 634689, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.626103162765503, + "objective/train/original_loss": 2.626103401184082, + "objective/train/theoretical_loss": 3.6122265746869653, + "objective/train/tokens_used": 1134572000, + "objective/train/value_avg": -0.005157470703125, + "objective/train/value_loss": 7.463545625796542e-05, + "objective/train/value_max": -0.000263214111328125, + "objective/train/value_min": -0.06707763671875, + "objective/train/value_reward_corr": 0.3936882670243347, + "objective/train/value_std": 0.004970550537109375, + "objective/train/weight_avg": 1.0037449598312378, + "objective/train/weighted_lm_loss": 2.6352250576019287, + "objective/train/weights_max": 1.5199147462844849, + "objective/train/weights_min": 0.08458830416202545, + "theoretical_loss": 3.6122265746869653, + "tokens_seen": 1114112000 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003345837512537613, + "loss": 1.4651, + "theoretical_loss": 3.6122265746869653, + "tokens_seen": 1114112000 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003345737211634905, + "loss": 1.4529, + "theoretical_loss": 3.612206755885608, + "tokens_seen": 1114177536 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033456369107321966, + "loss": 1.3974, + "theoretical_loss": 3.6121869385763463, + "tokens_seen": 1114243072 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003345536609829489, + "loss": 1.4122, + "theoretical_loss": 3.6121671227589798, + "tokens_seen": 1114308608 + }, + { + "epoch": 0.34, + "learning_rate": 0.000334543630892678, + "loss": 1.42, + "theoretical_loss": 3.6121473084333084, + "tokens_seen": 1114374144 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033453360080240726, + "loss": 1.4159, + "theoretical_loss": 3.6121274955991316, + "tokens_seen": 1114439680 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003345235707121364, + "loss": 1.3912, + "theoretical_loss": 3.6121076842562507, + "tokens_seen": 1114505216 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003345135406218656, + "loss": 1.3891, + "theoretical_loss": 3.612087874404464, + "tokens_seen": 1114570752 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003345035105315948, + "loss": 1.3327, + "theoretical_loss": 3.6120680660435736, + "tokens_seen": 1114636288 + }, + { + "epoch": 0.34, + "learning_rate": 0.000334493480441324, + "loss": 1.3284, + "theoretical_loss": 3.612048259173378, + "tokens_seen": 1114701824 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033448345035105316, + "loss": 1.4697, + "theoretical_loss": 3.612028453793678, + "tokens_seen": 1114767360 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033447342026078235, + "loss": 1.4658, + "theoretical_loss": 3.6120086499042747, + "tokens_seen": 1114832896 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003344633901705115, + "loss": 1.4735, + "theoretical_loss": 3.6119888475049664, + "tokens_seen": 1114898432 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033445336008024076, + "loss": 1.5124, + "theoretical_loss": 3.611969046595555, + "tokens_seen": 1114963968 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003344433299899699, + "loss": 1.4722, + "theoretical_loss": 3.6119492471758403, + "tokens_seen": 1115029504 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003344332998996991, + "loss": 1.4975, + "theoretical_loss": 3.611929449245623, + "tokens_seen": 1115095040 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033442326980942825, + "loss": 1.3787, + "theoretical_loss": 3.6119096528047026, + "tokens_seen": 1115160576 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003344132397191575, + "loss": 1.3812, + "theoretical_loss": 3.611889857852881, + "tokens_seen": 1115226112 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033440320962888667, + "loss": 1.3678, + "theoretical_loss": 3.611870064389957, + "tokens_seen": 1115291648 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033439317953861585, + "loss": 1.4719, + "theoretical_loss": 3.611850272415732, + "tokens_seen": 1115357184 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033438314944834503, + "loss": 1.4619, + "theoretical_loss": 3.6118304819300073, + "tokens_seen": 1115422720 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033437311935807427, + "loss": 1.3878, + "theoretical_loss": 3.6118106929325826, + "tokens_seen": 1115488256 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003343630892678034, + "loss": 1.3204, + "theoretical_loss": 3.611790905423259, + "tokens_seen": 1115553792 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033435305917753263, + "loss": 1.5745, + "theoretical_loss": 3.611771119401836, + "tokens_seen": 1115619328 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033434302908726175, + "loss": 1.4676, + "theoretical_loss": 3.6117513348681163, + "tokens_seen": 1115684864 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0003077627916354686, + "objective/train/docs_used": 636068, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0692555904388428, + "objective/train/original_loss": 3.0692553520202637, + "objective/train/theoretical_loss": 3.611731551821899, + "objective/train/tokens_used": 1136210400, + "objective/train/value_avg": -0.005176544189453125, + "objective/train/value_loss": 7.960841321619228e-05, + "objective/train/value_max": -0.0003612041473388672, + "objective/train/value_min": -0.1253662109375, + "objective/train/value_reward_corr": 0.29442097804716416, + "objective/train/value_std": 0.00469970703125, + "objective/train/weight_avg": 1.0006694793701172, + "objective/train/weighted_lm_loss": 3.0775575637817383, + "objective/train/weights_max": 3.1739532947540283, + "objective/train/weights_min": 0.4956987500190735, + "theoretical_loss": 3.611731551821899, + "tokens_seen": 1115750400 + }, + { + "epoch": 0.34, + "learning_rate": 0.000334332998996991, + "loss": 1.4318, + "theoretical_loss": 3.611731551821899, + "tokens_seen": 1115750400 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033432296890672017, + "loss": 1.3679, + "theoretical_loss": 3.6117117702629855, + "tokens_seen": 1115815936 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033431293881644935, + "loss": 1.3508, + "theoretical_loss": 3.6116919901911766, + "tokens_seen": 1115881472 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033430290872617853, + "loss": 1.3589, + "theoretical_loss": 3.6116722116062734, + "tokens_seen": 1115947008 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003342928786359077, + "loss": 1.4539, + "theoretical_loss": 3.611652434508077, + "tokens_seen": 1116012544 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003342828485456369, + "loss": 1.3135, + "theoretical_loss": 3.6116326588963874, + "tokens_seen": 1116078080 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033427281845536613, + "loss": 1.4227, + "theoretical_loss": 3.6116128847710063, + "tokens_seen": 1116143616 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033426278836509526, + "loss": 1.3832, + "theoretical_loss": 3.6115931121317346, + "tokens_seen": 1116209152 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003342527582748245, + "loss": 1.381, + "theoretical_loss": 3.6115733409783735, + "tokens_seen": 1116274688 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003342427281845536, + "loss": 1.4483, + "theoretical_loss": 3.611553571310724, + "tokens_seen": 1116340224 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033423269809428285, + "loss": 1.4347, + "theoretical_loss": 3.6115338031285873, + "tokens_seen": 1116405760 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033422266800401204, + "loss": 1.3568, + "theoretical_loss": 3.611514036431764, + "tokens_seen": 1116471296 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003342126379137412, + "loss": 1.3738, + "theoretical_loss": 3.6114942712200566, + "tokens_seen": 1116536832 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003342026078234704, + "loss": 1.3153, + "theoretical_loss": 3.6114745074932655, + "tokens_seen": 1116602368 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033419257773319963, + "loss": 1.312, + "theoretical_loss": 3.6114547452511916, + "tokens_seen": 1116667904 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033418254764292876, + "loss": 1.3315, + "theoretical_loss": 3.611434984493637, + "tokens_seen": 1116733440 + }, + { + "epoch": 0.34, + "learning_rate": 0.000334172517552658, + "loss": 1.3925, + "theoretical_loss": 3.611415225220403, + "tokens_seen": 1116798976 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003341624874623871, + "loss": 1.4374, + "theoretical_loss": 3.61139546743129, + "tokens_seen": 1116864512 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033415245737211636, + "loss": 1.3989, + "theoretical_loss": 3.611375711126101, + "tokens_seen": 1116930048 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033414242728184554, + "loss": 1.3979, + "theoretical_loss": 3.611355956304637, + "tokens_seen": 1116995584 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003341323971915747, + "loss": 1.5781, + "theoretical_loss": 3.6113362029666987, + "tokens_seen": 1117061120 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003341223671013039, + "loss": 1.4591, + "theoretical_loss": 3.6113164511120885, + "tokens_seen": 1117126656 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003341123370110331, + "loss": 1.4267, + "theoretical_loss": 3.6112967007406076, + "tokens_seen": 1117192192 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033410230692076226, + "loss": 1.4477, + "theoretical_loss": 3.6112769518520578, + "tokens_seen": 1117257728 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003340922768304915, + "loss": 1.4065, + "theoretical_loss": 3.611257204446241, + "tokens_seen": 1117323264 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": 6.133950228104368e-05, + "objective/train/docs_used": 636687, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.658710479736328, + "objective/train/original_loss": 2.65871000289917, + "objective/train/theoretical_loss": 3.6112374585229583, + "objective/train/tokens_used": 1137848800, + "objective/train/value_avg": -0.004337310791015625, + "objective/train/value_loss": 4.590085518429987e-05, + "objective/train/value_max": -0.0003380775451660156, + "objective/train/value_min": -0.037811279296875, + "objective/train/value_reward_corr": 0.22692271203371409, + "objective/train/value_std": 0.0033130645751953125, + "objective/train/weight_avg": 1.0025264024734497, + "objective/train/weighted_lm_loss": 2.684818983078003, + "objective/train/weights_max": 1.4286794662475586, + "objective/train/weights_min": 0.08366327732801437, + "theoretical_loss": 3.6112374585229583, + "tokens_seen": 1117388800 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003340822467402207, + "loss": 1.3695, + "theoretical_loss": 3.6112374585229583, + "tokens_seen": 1117388800 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033407221664994986, + "loss": 1.4767, + "theoretical_loss": 3.611217714082012, + "tokens_seen": 1117454336 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003340621865596791, + "loss": 1.3523, + "theoretical_loss": 3.6111979711232043, + "tokens_seen": 1117519872 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003340521564694082, + "loss": 1.465, + "theoretical_loss": 3.611178229646336, + "tokens_seen": 1117585408 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033404212637913746, + "loss": 1.393, + "theoretical_loss": 3.61115848965121, + "tokens_seen": 1117650944 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003340320962888666, + "loss": 1.4563, + "theoretical_loss": 3.611138751137627, + "tokens_seen": 1117716480 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003340220661985958, + "loss": 1.3542, + "theoretical_loss": 3.6111190141053893, + "tokens_seen": 1117782016 + }, + { + "epoch": 0.34, + "learning_rate": 0.000334012036108325, + "loss": 1.4387, + "theoretical_loss": 3.6110992785543, + "tokens_seen": 1117847552 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003340020060180542, + "loss": 1.3986, + "theoretical_loss": 3.61107954448416, + "tokens_seen": 1117913088 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033399197592778336, + "loss": 1.4063, + "theoretical_loss": 3.611059811894772, + "tokens_seen": 1117978624 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033398194583751255, + "loss": 1.5031, + "theoretical_loss": 3.611040080785937, + "tokens_seen": 1118044160 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003339719157472417, + "loss": 1.4253, + "theoretical_loss": 3.6110203511574586, + "tokens_seen": 1118109696 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033396188565697096, + "loss": 1.3875, + "theoretical_loss": 3.611000623009138, + "tokens_seen": 1118175232 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003339518555667001, + "loss": 1.5298, + "theoretical_loss": 3.6109808963407777, + "tokens_seen": 1118240768 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003339418254764293, + "loss": 1.4666, + "theoretical_loss": 3.61096117115218, + "tokens_seen": 1118306304 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033393179538615845, + "loss": 1.4046, + "theoretical_loss": 3.610941447443147, + "tokens_seen": 1118371840 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003339217652958877, + "loss": 1.3624, + "theoretical_loss": 3.6109217252134815, + "tokens_seen": 1118437376 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033391173520561687, + "loss": 1.3454, + "theoretical_loss": 3.6109020044629854, + "tokens_seen": 1118502912 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033390170511534605, + "loss": 1.3884, + "theoretical_loss": 3.610882285191461, + "tokens_seen": 1118568448 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033389167502507523, + "loss": 1.4772, + "theoretical_loss": 3.610862567398711, + "tokens_seen": 1118633984 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033388164493480447, + "loss": 1.4559, + "theoretical_loss": 3.6108428510845383, + "tokens_seen": 1118699520 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003338716148445336, + "loss": 1.4299, + "theoretical_loss": 3.6108231362487446, + "tokens_seen": 1118765056 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033386158475426283, + "loss": 1.3513, + "theoretical_loss": 3.6108034228911334, + "tokens_seen": 1118830592 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033385155466399195, + "loss": 1.4033, + "theoretical_loss": 3.610783711011506, + "tokens_seen": 1118896128 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003338415245737212, + "loss": 1.3506, + "theoretical_loss": 3.6107640006096657, + "tokens_seen": 1118961664 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.00039539847057312727, + "objective/train/docs_used": 638066, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.2377843856811523, + "objective/train/original_loss": 2.2377841472625732, + "objective/train/theoretical_loss": 3.6107442916854158, + "objective/train/tokens_used": 1139487200, + "objective/train/value_avg": -0.00408172607421875, + "objective/train/value_loss": 7.67499950597994e-05, + "objective/train/value_max": -0.0002758502960205078, + "objective/train/value_min": -0.059967041015625, + "objective/train/value_reward_corr": 0.32688565837633415, + "objective/train/value_std": 0.0030841827392578125, + "objective/train/weight_avg": 0.9992902874946594, + "objective/train/weighted_lm_loss": 2.2379150390625, + "objective/train/weights_max": 1.4230226278305054, + "objective/train/weights_min": 0.20084697008132935, + "theoretical_loss": 3.6107442916854158, + "tokens_seen": 1119027200 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033383149448345037, + "loss": 1.3612, + "theoretical_loss": 3.6107442916854158, + "tokens_seen": 1119027200 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033382146439317955, + "loss": 1.3006, + "theoretical_loss": 3.6107245842385582, + "tokens_seen": 1119092736 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033381143430290873, + "loss": 1.3393, + "theoretical_loss": 3.610704878268896, + "tokens_seen": 1119158272 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003338014042126379, + "loss": 1.4672, + "theoretical_loss": 3.6106851737762318, + "tokens_seen": 1119223808 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003337913741223671, + "loss": 1.3982, + "theoretical_loss": 3.6106654707603685, + "tokens_seen": 1119289344 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033378134403209633, + "loss": 1.345, + "theoretical_loss": 3.610645769221109, + "tokens_seen": 1119354880 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033377131394182546, + "loss": 1.4161, + "theoretical_loss": 3.610626069158256, + "tokens_seen": 1119420416 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003337612838515547, + "loss": 1.3816, + "theoretical_loss": 3.6106063705716123, + "tokens_seen": 1119485952 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003337512537612838, + "loss": 1.3507, + "theoretical_loss": 3.6105866734609817, + "tokens_seen": 1119551488 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033374122367101306, + "loss": 1.3174, + "theoretical_loss": 3.610566977826166, + "tokens_seen": 1119617024 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033373119358074224, + "loss": 1.4469, + "theoretical_loss": 3.6105472836669694, + "tokens_seen": 1119682560 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003337211634904714, + "loss": 1.2998, + "theoretical_loss": 3.610527590983194, + "tokens_seen": 1119748096 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003337111334002006, + "loss": 1.4393, + "theoretical_loss": 3.6105078997746443, + "tokens_seen": 1119813632 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033370110330992983, + "loss": 1.2853, + "theoretical_loss": 3.6104882100411215, + "tokens_seen": 1119879168 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033369107321965896, + "loss": 1.2995, + "theoretical_loss": 3.6104685217824306, + "tokens_seen": 1119944704 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003336810431293882, + "loss": 1.4688, + "theoretical_loss": 3.6104488349983743, + "tokens_seen": 1120010240 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003336710130391173, + "loss": 1.3159, + "theoretical_loss": 3.610429149688755, + "tokens_seen": 1120075776 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033366098294884656, + "loss": 1.3842, + "theoretical_loss": 3.610409465853377, + "tokens_seen": 1120141312 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033365095285857574, + "loss": 1.1966, + "theoretical_loss": 3.6103897834920433, + "tokens_seen": 1120206848 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003336409227683049, + "loss": 1.357, + "theoretical_loss": 3.6103701026045574, + "tokens_seen": 1120272384 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003336308926780341, + "loss": 1.4686, + "theoretical_loss": 3.610350423190723, + "tokens_seen": 1120337920 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003336208625877633, + "loss": 1.4958, + "theoretical_loss": 3.6103307452503426, + "tokens_seen": 1120403456 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033361083249749246, + "loss": 1.3285, + "theoretical_loss": 3.6103110687832207, + "tokens_seen": 1120468992 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003336008024072217, + "loss": 1.3152, + "theoretical_loss": 3.6102913937891605, + "tokens_seen": 1120534528 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003335907723169508, + "loss": 1.2554, + "theoretical_loss": 3.6102717202679653, + "tokens_seen": 1120600064 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.000967455271165818, + "objective/train/docs_used": 638756, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5682477951049805, + "objective/train/original_loss": 2.5682477951049805, + "objective/train/theoretical_loss": 3.6102520482194387, + "objective/train/tokens_used": 1141125600, + "objective/train/value_avg": -0.00478363037109375, + "objective/train/value_loss": 8.176998380804434e-05, + "objective/train/value_max": -0.00030303001403808594, + "objective/train/value_min": -0.08648681640625, + "objective/train/value_reward_corr": 0.5019176466076778, + "objective/train/value_std": 0.004962921142578125, + "objective/train/weight_avg": 0.9939894676208496, + "objective/train/weighted_lm_loss": 2.550982713699341, + "objective/train/weights_max": 1.8559362888336182, + "objective/train/weights_min": 0.26272767782211304, + "theoretical_loss": 3.6102520482194387, + "tokens_seen": 1120665600 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033358074222668006, + "loss": 1.3191, + "theoretical_loss": 3.6102520482194387, + "tokens_seen": 1120665600 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033357071213640924, + "loss": 1.3638, + "theoretical_loss": 3.610232377643385, + "tokens_seen": 1120731136 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003335606820461384, + "loss": 1.4745, + "theoretical_loss": 3.610212708539607, + "tokens_seen": 1120796672 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003335506519558676, + "loss": 1.3114, + "theoretical_loss": 3.6101930409079097, + "tokens_seen": 1120862208 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003335406218655968, + "loss": 1.3388, + "theoretical_loss": 3.6101733747480957, + "tokens_seen": 1120927744 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033353059177532597, + "loss": 1.3957, + "theoretical_loss": 3.6101537100599694, + "tokens_seen": 1120993280 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003335205616850552, + "loss": 1.3452, + "theoretical_loss": 3.610134046843334, + "tokens_seen": 1121058816 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033351053159478433, + "loss": 1.2512, + "theoretical_loss": 3.6101143850979938, + "tokens_seen": 1121124352 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033350050150451356, + "loss": 1.3617, + "theoretical_loss": 3.610094724823753, + "tokens_seen": 1121189888 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003334904714142427, + "loss": 1.5948, + "theoretical_loss": 3.610075066020415, + "tokens_seen": 1121255424 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003334804413239719, + "loss": 1.3848, + "theoretical_loss": 3.610055408687784, + "tokens_seen": 1121320960 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003334704112337011, + "loss": 1.2938, + "theoretical_loss": 3.610035752825665, + "tokens_seen": 1121386496 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003334603811434303, + "loss": 1.3852, + "theoretical_loss": 3.61001609843386, + "tokens_seen": 1121452032 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033345035105315947, + "loss": 1.3858, + "theoretical_loss": 3.609996445512175, + "tokens_seen": 1121517568 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033344032096288865, + "loss": 1.3939, + "theoretical_loss": 3.609976794060413, + "tokens_seen": 1121583104 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033343029087261783, + "loss": 1.3599, + "theoretical_loss": 3.6099571440783786, + "tokens_seen": 1121648640 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033342026078234707, + "loss": 1.4522, + "theoretical_loss": 3.609937495565876, + "tokens_seen": 1121714176 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003334102306920762, + "loss": 1.2677, + "theoretical_loss": 3.6099178485227092, + "tokens_seen": 1121779712 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033340020060180543, + "loss": 1.3952, + "theoretical_loss": 3.6098982029486835, + "tokens_seen": 1121845248 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003333901705115346, + "loss": 1.3431, + "theoretical_loss": 3.6098785588436018, + "tokens_seen": 1121910784 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003333801404212638, + "loss": 1.3896, + "theoretical_loss": 3.609858916207269, + "tokens_seen": 1121976320 + }, + { + "epoch": 0.34, + "learning_rate": 0.000333370110330993, + "loss": 1.4209, + "theoretical_loss": 3.6098392750394903, + "tokens_seen": 1122041856 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033336008024072215, + "loss": 1.4178, + "theoretical_loss": 3.6098196353400693, + "tokens_seen": 1122107392 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033335005015045134, + "loss": 1.3419, + "theoretical_loss": 3.6097999971088104, + "tokens_seen": 1122172928 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033334002006018057, + "loss": 1.367, + "theoretical_loss": 3.6097803603455185, + "tokens_seen": 1122238464 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0015238056657835841, + "objective/train/docs_used": 639983, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9945976734161377, + "objective/train/original_loss": 2.9945976734161377, + "objective/train/theoretical_loss": 3.6097607250499975, + "objective/train/tokens_used": 1142764000, + "objective/train/value_avg": -0.005157470703125, + "objective/train/value_loss": 0.00010746825864771381, + "objective/train/value_max": -0.00018966197967529297, + "objective/train/value_min": -0.09234619140625, + "objective/train/value_reward_corr": 0.47319576823130305, + "objective/train/value_std": 0.005229949951171875, + "objective/train/weight_avg": 0.9893922805786133, + "objective/train/weighted_lm_loss": 2.9595179557800293, + "objective/train/weights_max": 2.1914710998535156, + "objective/train/weights_min": 0.08943478763103485, + "theoretical_loss": 3.6097607250499975, + "tokens_seen": 1122304000 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033332998996990975, + "loss": 1.4941, + "theoretical_loss": 3.6097607250499975, + "tokens_seen": 1122304000 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033331995987963893, + "loss": 1.3572, + "theoretical_loss": 3.609741091222053, + "tokens_seen": 1122369536 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003333099297893681, + "loss": 1.4535, + "theoretical_loss": 3.609721458861489, + "tokens_seen": 1122435072 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003332998996990973, + "loss": 1.4727, + "theoretical_loss": 3.6097018279681103, + "tokens_seen": 1122500608 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033328986960882653, + "loss": 1.3491, + "theoretical_loss": 3.609682198541722, + "tokens_seen": 1122566144 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033327983951855566, + "loss": 1.4684, + "theoretical_loss": 3.609662570582128, + "tokens_seen": 1122631680 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003332698094282849, + "loss": 1.4646, + "theoretical_loss": 3.6096429440891336, + "tokens_seen": 1122697216 + }, + { + "epoch": 0.34, + "learning_rate": 0.000333259779338014, + "loss": 1.4642, + "theoretical_loss": 3.6096233190625435, + "tokens_seen": 1122762752 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033324974924774326, + "loss": 1.3166, + "theoretical_loss": 3.6096036955021624, + "tokens_seen": 1122828288 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033323971915747244, + "loss": 1.3654, + "theoretical_loss": 3.6095840734077957, + "tokens_seen": 1122893824 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003332296890672016, + "loss": 1.3612, + "theoretical_loss": 3.6095644527792476, + "tokens_seen": 1122959360 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003332196589769308, + "loss": 1.401, + "theoretical_loss": 3.609544833616324, + "tokens_seen": 1123024896 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033320962888666003, + "loss": 1.5081, + "theoretical_loss": 3.609525215918829, + "tokens_seen": 1123090432 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033319959879638916, + "loss": 1.4105, + "theoretical_loss": 3.6095055996865684, + "tokens_seen": 1123155968 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003331895687061184, + "loss": 1.3229, + "theoretical_loss": 3.6094859849193464, + "tokens_seen": 1123221504 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003331795386158475, + "loss": 1.3232, + "theoretical_loss": 3.6094663716169686, + "tokens_seen": 1123287040 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033316950852557676, + "loss": 1.3626, + "theoretical_loss": 3.6094467597792406, + "tokens_seen": 1123352576 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033315947843530594, + "loss": 1.4258, + "theoretical_loss": 3.6094271494059664, + "tokens_seen": 1123418112 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003331494483450351, + "loss": 1.3778, + "theoretical_loss": 3.6094075404969526, + "tokens_seen": 1123483648 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003331394182547643, + "loss": 1.3947, + "theoretical_loss": 3.6093879330520036, + "tokens_seen": 1123549184 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003331293881644935, + "loss": 1.3205, + "theoretical_loss": 3.609368327070925, + "tokens_seen": 1123614720 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033311935807422266, + "loss": 1.3786, + "theoretical_loss": 3.6093487225535217, + "tokens_seen": 1123680256 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003331093279839519, + "loss": 1.3366, + "theoretical_loss": 3.6093291194995993, + "tokens_seen": 1123745792 + }, + { + "epoch": 0.34, + "learning_rate": 0.000333099297893681, + "loss": 1.3042, + "theoretical_loss": 3.6093095179089643, + "tokens_seen": 1123811328 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033308926780341026, + "loss": 1.2906, + "theoretical_loss": 3.60928991778142, + "tokens_seen": 1123876864 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0018263397505506873, + "objective/train/docs_used": 640598, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.749943733215332, + "objective/train/original_loss": 2.7499442100524902, + "objective/train/theoretical_loss": 3.6092703191167743, + "objective/train/tokens_used": 1144402400, + "objective/train/value_avg": -0.00438690185546875, + "objective/train/value_loss": 6.626264075748622e-05, + "objective/train/value_max": -0.00023233890533447266, + "objective/train/value_min": -0.062225341796875, + "objective/train/value_reward_corr": 0.367034781444356, + "objective/train/value_std": 0.0034923553466796875, + "objective/train/weight_avg": 0.984835147857666, + "objective/train/weighted_lm_loss": 2.727637767791748, + "objective/train/weights_max": 1.5680606365203857, + "objective/train/weights_min": 0.5293207168579102, + "theoretical_loss": 3.6092703191167743, + "tokens_seen": 1123942400 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033307923771313944, + "loss": 1.3886, + "theoretical_loss": 3.6092703191167743, + "tokens_seen": 1123942400 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003330692076228686, + "loss": 1.3456, + "theoretical_loss": 3.6092507219148304, + "tokens_seen": 1124007936 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003330591775325978, + "loss": 1.4345, + "theoretical_loss": 3.6092311261753958, + "tokens_seen": 1124073472 + }, + { + "epoch": 0.34, + "learning_rate": 0.000333049147442327, + "loss": 1.231, + "theoretical_loss": 3.609211531898275, + "tokens_seen": 1124139008 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033303911735205617, + "loss": 1.3465, + "theoretical_loss": 3.6091919390832734, + "tokens_seen": 1124204544 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003330290872617854, + "loss": 1.3672, + "theoretical_loss": 3.6091723477301976, + "tokens_seen": 1124270080 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033301905717151453, + "loss": 1.4322, + "theoretical_loss": 3.609152757838853, + "tokens_seen": 1124335616 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033300902708124376, + "loss": 1.3352, + "theoretical_loss": 3.6091331694090445, + "tokens_seen": 1124401152 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003329989969909729, + "loss": 1.3561, + "theoretical_loss": 3.609113582440579, + "tokens_seen": 1124466688 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003329889669007021, + "loss": 1.3965, + "theoretical_loss": 3.6090939969332623, + "tokens_seen": 1124532224 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003329789368104313, + "loss": 1.3225, + "theoretical_loss": 3.6090744128868995, + "tokens_seen": 1124597760 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003329689067201605, + "loss": 1.4499, + "theoretical_loss": 3.6090548303012966, + "tokens_seen": 1124663296 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033295887662988967, + "loss": 1.3319, + "theoretical_loss": 3.6090352491762605, + "tokens_seen": 1124728832 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033294884653961885, + "loss": 1.4538, + "theoretical_loss": 3.609015669511596, + "tokens_seen": 1124794368 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033293881644934803, + "loss": 1.3121, + "theoretical_loss": 3.6089960913071097, + "tokens_seen": 1124859904 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033292878635907727, + "loss": 1.4645, + "theoretical_loss": 3.6089765145626074, + "tokens_seen": 1124925440 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003329187562688064, + "loss": 1.4117, + "theoretical_loss": 3.6089569392778955, + "tokens_seen": 1124990976 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033290872617853563, + "loss": 1.3616, + "theoretical_loss": 3.6089373654527797, + "tokens_seen": 1125056512 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003328986960882648, + "loss": 1.3792, + "theoretical_loss": 3.608917793087066, + "tokens_seen": 1125122048 + }, + { + "epoch": 0.34, + "learning_rate": 0.000332888665997994, + "loss": 1.3865, + "theoretical_loss": 3.608898222180562, + "tokens_seen": 1125187584 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003328786359077232, + "loss": 1.3857, + "theoretical_loss": 3.6088786527330723, + "tokens_seen": 1125253120 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033286860581745235, + "loss": 1.3837, + "theoretical_loss": 3.6088590847444038, + "tokens_seen": 1125318656 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033285857572718154, + "loss": 1.2904, + "theoretical_loss": 3.6088395182143627, + "tokens_seen": 1125384192 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033284854563691077, + "loss": 1.2969, + "theoretical_loss": 3.6088199531427554, + "tokens_seen": 1125449728 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003328385155466399, + "loss": 1.3203, + "theoretical_loss": 3.608800389529388, + "tokens_seen": 1125515264 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0008719915058463812, + "objective/train/docs_used": 641290, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.660965919494629, + "objective/train/original_loss": 2.660965919494629, + "objective/train/theoretical_loss": 3.6087808273740674, + "objective/train/tokens_used": 1146040800, + "objective/train/value_avg": -0.004756927490234375, + "objective/train/value_loss": 5.1175422413507476e-05, + "objective/train/value_max": -0.00033020973205566406, + "objective/train/value_min": -0.0635986328125, + "objective/train/value_reward_corr": 0.30508279824497103, + "objective/train/value_std": 0.00400543212890625, + "objective/train/weight_avg": 0.9937275052070618, + "objective/train/weighted_lm_loss": 2.652829170227051, + "objective/train/weights_max": 1.88888418674469, + "objective/train/weights_min": 0.501441240310669, + "theoretical_loss": 3.6087808273740674, + "tokens_seen": 1125580800 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033282848545636913, + "loss": 1.3931, + "theoretical_loss": 3.6087808273740674, + "tokens_seen": 1125580800 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033281845536609826, + "loss": 1.3597, + "theoretical_loss": 3.6087612666766002, + "tokens_seen": 1125646336 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003328084252758275, + "loss": 1.4162, + "theoretical_loss": 3.608741707436792, + "tokens_seen": 1125711872 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003327983951855567, + "loss": 1.3744, + "theoretical_loss": 3.60872214965445, + "tokens_seen": 1125777408 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033278836509528586, + "loss": 1.4386, + "theoretical_loss": 3.608702593329381, + "tokens_seen": 1125842944 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033277833500501504, + "loss": 1.2912, + "theoretical_loss": 3.6086830384613906, + "tokens_seen": 1125908480 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003327683049147442, + "loss": 1.3305, + "theoretical_loss": 3.608663485050286, + "tokens_seen": 1125974016 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003327582748244734, + "loss": 1.3272, + "theoretical_loss": 3.608643933095874, + "tokens_seen": 1126039552 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033274824473420264, + "loss": 1.406, + "theoretical_loss": 3.608624382597962, + "tokens_seen": 1126105088 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033273821464393176, + "loss": 1.4799, + "theoretical_loss": 3.608604833556355, + "tokens_seen": 1126170624 + }, + { + "epoch": 0.34, + "learning_rate": 0.000332728184553661, + "loss": 1.4036, + "theoretical_loss": 3.6085852859708614, + "tokens_seen": 1126236160 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003327181544633902, + "loss": 1.4267, + "theoretical_loss": 3.608565739841287, + "tokens_seen": 1126301696 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033270812437311936, + "loss": 1.299, + "theoretical_loss": 3.608546195167439, + "tokens_seen": 1126367232 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033269809428284854, + "loss": 1.404, + "theoretical_loss": 3.6085266519491244, + "tokens_seen": 1126432768 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003326880641925777, + "loss": 1.3682, + "theoretical_loss": 3.60850711018615, + "tokens_seen": 1126498304 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003326780341023069, + "loss": 1.3435, + "theoretical_loss": 3.6084875698783225, + "tokens_seen": 1126563840 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033266800401203614, + "loss": 1.3458, + "theoretical_loss": 3.60846803102545, + "tokens_seen": 1126629376 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033265797392176527, + "loss": 1.3809, + "theoretical_loss": 3.608448493627338, + "tokens_seen": 1126694912 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003326479438314945, + "loss": 1.5014, + "theoretical_loss": 3.608428957683795, + "tokens_seen": 1126760448 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033263791374122363, + "loss": 1.336, + "theoretical_loss": 3.608409423194627, + "tokens_seen": 1126825984 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033262788365095286, + "loss": 1.3539, + "theoretical_loss": 3.6083898901596414, + "tokens_seen": 1126891520 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033261785356068205, + "loss": 1.4204, + "theoretical_loss": 3.6083703585786457, + "tokens_seen": 1126957056 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003326078234704112, + "loss": 1.3272, + "theoretical_loss": 3.608350828451447, + "tokens_seen": 1127022592 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003325977933801404, + "loss": 1.4174, + "theoretical_loss": 3.608331299777853, + "tokens_seen": 1127088128 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033258776328986964, + "loss": 1.3389, + "theoretical_loss": 3.60831177255767, + "tokens_seen": 1127153664 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.000671414309181273, + "objective/train/docs_used": 642585, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7155323028564453, + "objective/train/original_loss": 2.715532064437866, + "objective/train/theoretical_loss": 3.6082922467907066, + "objective/train/tokens_used": 1147679200, + "objective/train/value_avg": -0.004241943359375, + "objective/train/value_loss": 3.346193989273161e-05, + "objective/train/value_max": -0.0004372596740722656, + "objective/train/value_min": -0.050994873046875, + "objective/train/value_reward_corr": 0.319827769723345, + "objective/train/value_std": 0.003383636474609375, + "objective/train/weight_avg": 0.994918167591095, + "objective/train/weighted_lm_loss": 2.7165019512176514, + "objective/train/weights_max": 1.5646979808807373, + "objective/train/weights_min": 0.5510802865028381, + "theoretical_loss": 3.6082922467907066, + "tokens_seen": 1127219200 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003325777331995988, + "loss": 1.3331, + "theoretical_loss": 3.6082922467907066, + "tokens_seen": 1127219200 + }, + { + "epoch": 0.34, + "learning_rate": 0.000332567703109328, + "loss": 1.3839, + "theoretical_loss": 3.6082727224767686, + "tokens_seen": 1127284736 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003325576730190572, + "loss": 1.2997, + "theoretical_loss": 3.608253199615665, + "tokens_seen": 1127350272 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033254764292878637, + "loss": 1.3027, + "theoretical_loss": 3.608233678207202, + "tokens_seen": 1127415808 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003325376128385156, + "loss": 1.4009, + "theoretical_loss": 3.608214158251188, + "tokens_seen": 1127481344 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033252758274824473, + "loss": 1.3907, + "theoretical_loss": 3.60819463974743, + "tokens_seen": 1127546880 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033251755265797396, + "loss": 1.4218, + "theoretical_loss": 3.6081751226957355, + "tokens_seen": 1127612416 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003325075225677031, + "loss": 1.3313, + "theoretical_loss": 3.608155607095912, + "tokens_seen": 1127677952 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033249749247743233, + "loss": 1.5184, + "theoretical_loss": 3.6081360929477677, + "tokens_seen": 1127743488 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003324874623871615, + "loss": 1.4331, + "theoretical_loss": 3.60811658025111, + "tokens_seen": 1127809024 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003324774322968907, + "loss": 1.4938, + "theoretical_loss": 3.608097069005747, + "tokens_seen": 1127874560 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033246740220661987, + "loss": 1.3947, + "theoretical_loss": 3.6080775592114858, + "tokens_seen": 1127940096 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033245737211634905, + "loss": 1.3723, + "theoretical_loss": 3.608058050868134, + "tokens_seen": 1128005632 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033244734202607823, + "loss": 1.3281, + "theoretical_loss": 3.6080385439755, + "tokens_seen": 1128071168 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033243731193580747, + "loss": 1.3389, + "theoretical_loss": 3.6080190385333912, + "tokens_seen": 1128136704 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003324272818455366, + "loss": 1.4204, + "theoretical_loss": 3.607999534541616, + "tokens_seen": 1128202240 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033241725175526583, + "loss": 1.3025, + "theoretical_loss": 3.6079800319999817, + "tokens_seen": 1128267776 + }, + { + "epoch": 0.34, + "learning_rate": 0.000332407221664995, + "loss": 1.3047, + "theoretical_loss": 3.607960530908297, + "tokens_seen": 1128333312 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003323971915747242, + "loss": 1.4302, + "theoretical_loss": 3.6079410312663693, + "tokens_seen": 1128398848 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003323871614844534, + "loss": 1.4395, + "theoretical_loss": 3.6079215330740064, + "tokens_seen": 1128464384 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033237713139418255, + "loss": 1.3683, + "theoretical_loss": 3.6079020363310175, + "tokens_seen": 1128529920 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033236710130391174, + "loss": 1.3196, + "theoretical_loss": 3.6078825410372097, + "tokens_seen": 1128595456 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033235707121364097, + "loss": 1.4676, + "theoretical_loss": 3.6078630471923914, + "tokens_seen": 1128660992 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003323470411233701, + "loss": 1.4831, + "theoretical_loss": 3.6078435547963705, + "tokens_seen": 1128726528 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033233701103309933, + "loss": 1.4494, + "theoretical_loss": 3.6078240638489563, + "tokens_seen": 1128792064 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0023275017738342285, + "objective/train/docs_used": 643131, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.628387212753296, + "objective/train/original_loss": 2.628387212753296, + "objective/train/theoretical_loss": 3.607804574349956, + "objective/train/tokens_used": 1149317600, + "objective/train/value_avg": -0.005767822265625, + "objective/train/value_loss": 0.000157145710545592, + "objective/train/value_max": -0.00035691261291503906, + "objective/train/value_min": -0.06915283203125, + "objective/train/value_reward_corr": 0.5553396165860627, + "objective/train/value_std": 0.005641937255859375, + "objective/train/weight_avg": 0.9832247495651245, + "objective/train/weighted_lm_loss": 2.5998733043670654, + "objective/train/weights_max": 1.6315034627914429, + "objective/train/weights_min": 0.21387065947055817, + "theoretical_loss": 3.607804574349956, + "tokens_seen": 1128857600 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033232698094282846, + "loss": 1.3699, + "theoretical_loss": 3.607804574349956, + "tokens_seen": 1128857600 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003323169508525577, + "loss": 1.4073, + "theoretical_loss": 3.6077850862991774, + "tokens_seen": 1128923136 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003323069207622869, + "loss": 1.3939, + "theoretical_loss": 3.607765599696431, + "tokens_seen": 1128988672 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033229689067201606, + "loss": 1.3852, + "theoretical_loss": 3.6077461145415226, + "tokens_seen": 1129054208 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033228686058174524, + "loss": 1.2928, + "theoretical_loss": 3.6077266308342617, + "tokens_seen": 1129119744 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003322768304914744, + "loss": 1.4822, + "theoretical_loss": 3.607707148574457, + "tokens_seen": 1129185280 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003322668004012036, + "loss": 1.4393, + "theoretical_loss": 3.607687667761917, + "tokens_seen": 1129250816 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033225677031093284, + "loss": 1.3164, + "theoretical_loss": 3.60766818839645, + "tokens_seen": 1129316352 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033224674022066196, + "loss": 1.3731, + "theoretical_loss": 3.6076487104778643, + "tokens_seen": 1129381888 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003322367101303912, + "loss": 1.3467, + "theoretical_loss": 3.6076292340059686, + "tokens_seen": 1129447424 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003322266800401204, + "loss": 1.4672, + "theoretical_loss": 3.607609758980572, + "tokens_seen": 1129512960 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033221664994984956, + "loss": 1.2827, + "theoretical_loss": 3.607590285401482, + "tokens_seen": 1129578496 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033220661985957874, + "loss": 1.2935, + "theoretical_loss": 3.6075708132685085, + "tokens_seen": 1129644032 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003321965897693079, + "loss": 1.489, + "theoretical_loss": 3.6075513425814596, + "tokens_seen": 1129709568 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003321865596790371, + "loss": 1.3392, + "theoretical_loss": 3.607531873340144, + "tokens_seen": 1129775104 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033217652958876634, + "loss": 1.2823, + "theoretical_loss": 3.6075124055443712, + "tokens_seen": 1129840640 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033216649949849547, + "loss": 1.2206, + "theoretical_loss": 3.607492939193949, + "tokens_seen": 1129906176 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003321564694082247, + "loss": 1.3801, + "theoretical_loss": 3.607473474288687, + "tokens_seen": 1129971712 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033214643931795383, + "loss": 1.3642, + "theoretical_loss": 3.6074540108283943, + "tokens_seen": 1130037248 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033213640922768306, + "loss": 1.3665, + "theoretical_loss": 3.607434548812879, + "tokens_seen": 1130102784 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033212637913741225, + "loss": 1.4065, + "theoretical_loss": 3.607415088241951, + "tokens_seen": 1130168320 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003321163490471414, + "loss": 1.3295, + "theoretical_loss": 3.607395629115419, + "tokens_seen": 1130233856 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003321063189568706, + "loss": 1.4549, + "theoretical_loss": 3.607376171433091, + "tokens_seen": 1130299392 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033209628886659984, + "loss": 1.3903, + "theoretical_loss": 3.6073567151947774, + "tokens_seen": 1130364928 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033208625877632897, + "loss": 1.2528, + "theoretical_loss": 3.6073372604002865, + "tokens_seen": 1130430464 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": 0.00036172501859255135, + "objective/train/docs_used": 644333, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.14505934715271, + "objective/train/original_loss": 3.145059108734131, + "objective/train/theoretical_loss": 3.6073178070494287, + "objective/train/tokens_used": 1150956000, + "objective/train/value_avg": -0.004055023193359375, + "objective/train/value_loss": 3.2188348995987326e-05, + "objective/train/value_max": -0.000293731689453125, + "objective/train/value_min": -0.051177978515625, + "objective/train/value_reward_corr": 0.4077022969463631, + "objective/train/value_std": 0.0031032562255859375, + "objective/train/weight_avg": 1.0051724910736084, + "objective/train/weighted_lm_loss": 3.158599376678467, + "objective/train/weights_max": 1.4116204977035522, + "objective/train/weights_min": 0.6297261118888855, + "theoretical_loss": 3.6073178070494287, + "tokens_seen": 1130496000 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003320762286860582, + "loss": 1.3577, + "theoretical_loss": 3.6073178070494287, + "tokens_seen": 1130496000 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033206619859578733, + "loss": 1.3638, + "theoretical_loss": 3.607298355142012, + "tokens_seen": 1130561536 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033205616850551657, + "loss": 1.4131, + "theoretical_loss": 3.607278904677846, + "tokens_seen": 1130627072 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033204613841524575, + "loss": 1.3502, + "theoretical_loss": 3.60725945565674, + "tokens_seen": 1130692608 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033203610832497493, + "loss": 1.3117, + "theoretical_loss": 3.607240008078503, + "tokens_seen": 1130758144 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003320260782347041, + "loss": 1.5445, + "theoretical_loss": 3.607220561942945, + "tokens_seen": 1130823680 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003320160481444333, + "loss": 1.3407, + "theoretical_loss": 3.6072011172498746, + "tokens_seen": 1130889216 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003320060180541625, + "loss": 1.3178, + "theoretical_loss": 3.607181673999102, + "tokens_seen": 1130954752 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003319959879638917, + "loss": 1.377, + "theoretical_loss": 3.607162232190436, + "tokens_seen": 1131020288 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033198595787362084, + "loss": 1.3793, + "theoretical_loss": 3.6071427918236862, + "tokens_seen": 1131085824 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033197592778335007, + "loss": 1.4779, + "theoretical_loss": 3.607123352898663, + "tokens_seen": 1131151360 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003319658976930792, + "loss": 1.3897, + "theoretical_loss": 3.6071039154151743, + "tokens_seen": 1131216896 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033195586760280843, + "loss": 1.3222, + "theoretical_loss": 3.6070844793730314, + "tokens_seen": 1131282432 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003319458375125376, + "loss": 1.387, + "theoretical_loss": 3.6070650447720425, + "tokens_seen": 1131347968 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003319358074222668, + "loss": 1.3875, + "theoretical_loss": 3.607045611612018, + "tokens_seen": 1131413504 + }, + { + "epoch": 0.34, + "learning_rate": 0.000331925777331996, + "loss": 1.3737, + "theoretical_loss": 3.607026179892768, + "tokens_seen": 1131479040 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003319157472417252, + "loss": 1.378, + "theoretical_loss": 3.6070067496141016, + "tokens_seen": 1131544576 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033190571715145434, + "loss": 1.3797, + "theoretical_loss": 3.6069873207758287, + "tokens_seen": 1131610112 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003318956870611836, + "loss": 1.3169, + "theoretical_loss": 3.6069678933777594, + "tokens_seen": 1131675648 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003318856569709127, + "loss": 1.4731, + "theoretical_loss": 3.606948467419703, + "tokens_seen": 1131741184 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033187562688064194, + "loss": 1.4038, + "theoretical_loss": 3.606929042901469, + "tokens_seen": 1131806720 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003318655967903711, + "loss": 1.3785, + "theoretical_loss": 3.6069096198228694, + "tokens_seen": 1131872256 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003318555667001003, + "loss": 1.4206, + "theoretical_loss": 3.606890198183712, + "tokens_seen": 1131937792 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003318455366098295, + "loss": 1.3629, + "theoretical_loss": 3.606870777983807, + "tokens_seen": 1132003328 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033183550651955866, + "loss": 1.3632, + "theoretical_loss": 3.606851359222966, + "tokens_seen": 1132068864 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0011878589866682887, + "objective/train/docs_used": 644979, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0017101764678955, + "objective/train/original_loss": 3.0017104148864746, + "objective/train/theoretical_loss": 3.6068319419009978, + "objective/train/tokens_used": 1152594400, + "objective/train/value_avg": -0.0057373046875, + "objective/train/value_loss": 6.299770757323131e-05, + "objective/train/value_max": -0.0003905296325683594, + "objective/train/value_min": -0.089111328125, + "objective/train/value_reward_corr": 0.536025794272099, + "objective/train/value_std": 0.0054168701171875, + "objective/train/weight_avg": 0.9911304116249084, + "objective/train/weighted_lm_loss": 2.994743585586548, + "objective/train/weights_max": 2.1918156147003174, + "objective/train/weights_min": 0.501181423664093, + "theoretical_loss": 3.6068319419009978, + "tokens_seen": 1132134400 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003318254764292879, + "loss": 1.3762, + "theoretical_loss": 3.6068319419009978, + "tokens_seen": 1132134400 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003318154463390171, + "loss": 1.4218, + "theoretical_loss": 3.606812526017712, + "tokens_seen": 1132199936 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033180541624874626, + "loss": 1.36, + "theoretical_loss": 3.6067931115729204, + "tokens_seen": 1132265472 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033179538615847544, + "loss": 1.3355, + "theoretical_loss": 3.606773698566432, + "tokens_seen": 1132331008 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003317853560682046, + "loss": 1.4581, + "theoretical_loss": 3.606754286998057, + "tokens_seen": 1132396544 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003317753259779338, + "loss": 1.4149, + "theoretical_loss": 3.6067348768676064, + "tokens_seen": 1132462080 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033176529588766304, + "loss": 1.3405, + "theoretical_loss": 3.6067154681748894, + "tokens_seen": 1132527616 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033175526579739216, + "loss": 1.3829, + "theoretical_loss": 3.6066960609197176, + "tokens_seen": 1132593152 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003317452357071214, + "loss": 1.3854, + "theoretical_loss": 3.6066766551019, + "tokens_seen": 1132658688 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003317352056168506, + "loss": 1.399, + "theoretical_loss": 3.6066572507212475, + "tokens_seen": 1132724224 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033172517552657976, + "loss": 1.398, + "theoretical_loss": 3.6066378477775713, + "tokens_seen": 1132789760 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033171514543630894, + "loss": 1.2767, + "theoretical_loss": 3.606618446270681, + "tokens_seen": 1132855296 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003317051153460381, + "loss": 1.3875, + "theoretical_loss": 3.6065990462003876, + "tokens_seen": 1132920832 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003316950852557673, + "loss": 1.4713, + "theoretical_loss": 3.606579647566501, + "tokens_seen": 1132986368 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033168505516549654, + "loss": 1.2882, + "theoretical_loss": 3.6065602503688323, + "tokens_seen": 1133051904 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033167502507522567, + "loss": 1.4392, + "theoretical_loss": 3.606540854607192, + "tokens_seen": 1133117440 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003316649949849549, + "loss": 1.4687, + "theoretical_loss": 3.6065214602813906, + "tokens_seen": 1133182976 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033165496489468403, + "loss": 1.4831, + "theoretical_loss": 3.606502067391239, + "tokens_seen": 1133248512 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033164493480441326, + "loss": 1.3781, + "theoretical_loss": 3.6064826759365474, + "tokens_seen": 1133314048 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033163490471414245, + "loss": 1.4508, + "theoretical_loss": 3.606463285917127, + "tokens_seen": 1133379584 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003316248746238716, + "loss": 1.3238, + "theoretical_loss": 3.6064438973327886, + "tokens_seen": 1133445120 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003316148445336008, + "loss": 1.425, + "theoretical_loss": 3.606424510183343, + "tokens_seen": 1133510656 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033160481444333004, + "loss": 1.4183, + "theoretical_loss": 3.6064051244686013, + "tokens_seen": 1133576192 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033159478435305917, + "loss": 1.4126, + "theoretical_loss": 3.606385740188373, + "tokens_seen": 1133641728 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003315847542627884, + "loss": 1.2845, + "theoretical_loss": 3.606366357342471, + "tokens_seen": 1133707264 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0010179876117035747, + "objective/train/docs_used": 646384, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.459886074066162, + "objective/train/original_loss": 2.459885835647583, + "objective/train/theoretical_loss": 3.6063469759307054, + "objective/train/tokens_used": 1154232800, + "objective/train/value_avg": -0.006610870361328125, + "objective/train/value_loss": 7.762388850096613e-05, + "objective/train/value_max": -0.0003275871276855469, + "objective/train/value_min": -0.07366943359375, + "objective/train/value_reward_corr": 0.34502264065043897, + "objective/train/value_std": 0.006282806396484375, + "objective/train/weight_avg": 0.9936025142669678, + "objective/train/weighted_lm_loss": 2.4481539726257324, + "objective/train/weights_max": 1.906256914138794, + "objective/train/weights_min": 0.47768434882164, + "theoretical_loss": 3.6063469759307054, + "tokens_seen": 1133772800 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033157472417251753, + "loss": 1.3416, + "theoretical_loss": 3.6063469759307054, + "tokens_seen": 1133772800 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033156469408224677, + "loss": 1.4653, + "theoretical_loss": 3.606327595952887, + "tokens_seen": 1133838336 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033155466399197595, + "loss": 1.2401, + "theoretical_loss": 3.606308217408827, + "tokens_seen": 1133903872 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033154463390170513, + "loss": 1.4625, + "theoretical_loss": 3.606288840298336, + "tokens_seen": 1133969408 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003315346038114343, + "loss": 1.3584, + "theoretical_loss": 3.606269464621226, + "tokens_seen": 1134034944 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003315245737211635, + "loss": 1.4203, + "theoretical_loss": 3.6062500903773076, + "tokens_seen": 1134100480 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003315145436308927, + "loss": 1.3788, + "theoretical_loss": 3.606230717566392, + "tokens_seen": 1134166016 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003315045135406219, + "loss": 1.415, + "theoretical_loss": 3.6062113461882914, + "tokens_seen": 1134231552 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033149448345035104, + "loss": 1.4771, + "theoretical_loss": 3.606191976242815, + "tokens_seen": 1134297088 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033148445336008027, + "loss": 1.3923, + "theoretical_loss": 3.6061726077297758, + "tokens_seen": 1134362624 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003314744232698094, + "loss": 1.4718, + "theoretical_loss": 3.6061532406489842, + "tokens_seen": 1134428160 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033146439317953863, + "loss": 1.3189, + "theoretical_loss": 3.6061338750002525, + "tokens_seen": 1134493696 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003314543630892678, + "loss": 1.4495, + "theoretical_loss": 3.606114510783391, + "tokens_seen": 1134559232 + }, + { + "epoch": 0.34, + "learning_rate": 0.000331444332998997, + "loss": 1.4012, + "theoretical_loss": 3.6060951479982117, + "tokens_seen": 1134624768 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003314343029087262, + "loss": 1.2504, + "theoretical_loss": 3.6060757866445265, + "tokens_seen": 1134690304 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003314242728184554, + "loss": 1.4594, + "theoretical_loss": 3.606056426722146, + "tokens_seen": 1134755840 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033141424272818454, + "loss": 1.352, + "theoretical_loss": 3.606037068230882, + "tokens_seen": 1134821376 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003314042126379138, + "loss": 1.4362, + "theoretical_loss": 3.606017711170547, + "tokens_seen": 1134886912 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003313941825476429, + "loss": 1.4199, + "theoretical_loss": 3.605998355540951, + "tokens_seen": 1134952448 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033138415245737214, + "loss": 1.3495, + "theoretical_loss": 3.6059790013419066, + "tokens_seen": 1135017984 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003313741223671013, + "loss": 1.3411, + "theoretical_loss": 3.6059596485732257, + "tokens_seen": 1135083520 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003313640922768305, + "loss": 1.3126, + "theoretical_loss": 3.605940297234719, + "tokens_seen": 1135149056 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003313540621865597, + "loss": 1.3826, + "theoretical_loss": 3.6059209473261995, + "tokens_seen": 1135214592 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033134403209628886, + "loss": 1.2959, + "theoretical_loss": 3.6059015988474785, + "tokens_seen": 1135280128 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033133400200601804, + "loss": 1.3996, + "theoretical_loss": 3.6058822517983673, + "tokens_seen": 1135345664 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": -0.0011682829353958368, + "objective/train/docs_used": 647116, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7796382904052734, + "objective/train/original_loss": 2.7796385288238525, + "objective/train/theoretical_loss": 3.6058629061786784, + "objective/train/tokens_used": 1155871200, + "objective/train/value_avg": -0.004886627197265625, + "objective/train/value_loss": 6.029922587913461e-05, + "objective/train/value_max": -0.0002269744873046875, + "objective/train/value_min": -0.06927490234375, + "objective/train/value_reward_corr": 0.4361698696913612, + "objective/train/value_std": 0.005390167236328125, + "objective/train/weight_avg": 0.9912258386611938, + "objective/train/weighted_lm_loss": 2.7513153553009033, + "objective/train/weights_max": 1.9992038011550903, + "objective/train/weights_min": 0.4883972704410553, + "theoretical_loss": 3.6058629061786784, + "tokens_seen": 1135411200 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003313239719157473, + "loss": 1.3848, + "theoretical_loss": 3.6058629061786784, + "tokens_seen": 1135411200 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003313139418254764, + "loss": 1.4727, + "theoretical_loss": 3.605843561988223, + "tokens_seen": 1135476736 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033130391173520564, + "loss": 1.3911, + "theoretical_loss": 3.605824219226814, + "tokens_seen": 1135542272 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033129388164493477, + "loss": 1.3779, + "theoretical_loss": 3.605804877894263, + "tokens_seen": 1135607808 + }, + { + "epoch": 0.34, + "learning_rate": 0.000331283851554664, + "loss": 1.4462, + "theoretical_loss": 3.6057855379903816, + "tokens_seen": 1135673344 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003312738214643932, + "loss": 1.2547, + "theoretical_loss": 3.6057661995149823, + "tokens_seen": 1135738880 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033126379137412236, + "loss": 1.3732, + "theoretical_loss": 3.6057468624678766, + "tokens_seen": 1135804416 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033125376128385155, + "loss": 1.5033, + "theoretical_loss": 3.6057275268488773, + "tokens_seen": 1135869952 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003312437311935808, + "loss": 1.3816, + "theoretical_loss": 3.605708192657796, + "tokens_seen": 1135935488 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003312337011033099, + "loss": 1.452, + "theoretical_loss": 3.605688859894445, + "tokens_seen": 1136001024 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033122367101303914, + "loss": 1.36, + "theoretical_loss": 3.6056695285586366, + "tokens_seen": 1136066560 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033121364092276827, + "loss": 1.3563, + "theoretical_loss": 3.605650198650183, + "tokens_seen": 1136132096 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003312036108324975, + "loss": 1.2842, + "theoretical_loss": 3.6056308701688975, + "tokens_seen": 1136197632 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003311935807422267, + "loss": 1.3706, + "theoretical_loss": 3.6056115431145903, + "tokens_seen": 1136263168 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033118355065195587, + "loss": 1.2964, + "theoretical_loss": 3.605592217487075, + "tokens_seen": 1136328704 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033117352056168505, + "loss": 1.36, + "theoretical_loss": 3.6055728932861646, + "tokens_seen": 1136394240 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033116349047141423, + "loss": 1.4621, + "theoretical_loss": 3.6055535705116704, + "tokens_seen": 1136459776 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003311534603811434, + "loss": 1.2828, + "theoretical_loss": 3.605534249163405, + "tokens_seen": 1136525312 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033114343029087265, + "loss": 1.2723, + "theoretical_loss": 3.605514929241182, + "tokens_seen": 1136590848 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033113340020060177, + "loss": 1.3205, + "theoretical_loss": 3.6054956107448124, + "tokens_seen": 1136656384 + }, + { + "epoch": 0.34, + "learning_rate": 0.000331123370110331, + "loss": 1.2247, + "theoretical_loss": 3.60547629367411, + "tokens_seen": 1136721920 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033111334002006013, + "loss": 1.377, + "theoretical_loss": 3.6054569780288865, + "tokens_seen": 1136787456 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033110330992978937, + "loss": 1.3979, + "theoretical_loss": 3.6054376638089547, + "tokens_seen": 1136852992 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033109327983951855, + "loss": 1.306, + "theoretical_loss": 3.605418351014128, + "tokens_seen": 1136918528 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033108324974924773, + "loss": 1.3457, + "theoretical_loss": 3.6053990396442184, + "tokens_seen": 1136984064 + }, + { + "epoch": 0.34, + "objective/train/advantage_avg": 0.0006589660770259798, + "objective/train/docs_used": 648555, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.454420566558838, + "objective/train/original_loss": 2.454420328140259, + "objective/train/theoretical_loss": 3.605379729699039, + "objective/train/tokens_used": 1157509600, + "objective/train/value_avg": -0.0048828125, + "objective/train/value_loss": 4.622160122380592e-05, + "objective/train/value_max": -0.00020825862884521484, + "objective/train/value_min": -0.06427001953125, + "objective/train/value_reward_corr": 0.30740937978806465, + "objective/train/value_std": 0.004093170166015625, + "objective/train/weight_avg": 1.0088136196136475, + "objective/train/weighted_lm_loss": 2.4817590713500977, + "objective/train/weights_max": 1.806019902229309, + "objective/train/weights_min": 0.48269256949424744, + "theoretical_loss": 3.605379729699039, + "tokens_seen": 1137049600 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033107321965897697, + "loss": 1.337, + "theoretical_loss": 3.605379729699039, + "tokens_seen": 1137049600 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033106318956870615, + "loss": 1.3904, + "theoretical_loss": 3.6053604211784025, + "tokens_seen": 1137115136 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033105315947843533, + "loss": 1.4831, + "theoretical_loss": 3.6053411140821208, + "tokens_seen": 1137180672 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003310431293881645, + "loss": 1.296, + "theoretical_loss": 3.6053218084100083, + "tokens_seen": 1137246208 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003310330992978937, + "loss": 1.3731, + "theoretical_loss": 3.605302504161877, + "tokens_seen": 1137311744 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003310230692076229, + "loss": 1.2957, + "theoretical_loss": 3.6052832013375404, + "tokens_seen": 1137377280 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003310130391173521, + "loss": 1.3987, + "theoretical_loss": 3.605263899936811, + "tokens_seen": 1137442816 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033100300902708124, + "loss": 1.3705, + "theoretical_loss": 3.6052445999595015, + "tokens_seen": 1137508352 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033099297893681047, + "loss": 1.3866, + "theoretical_loss": 3.6052253014054254, + "tokens_seen": 1137573888 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003309829488465396, + "loss": 1.3418, + "theoretical_loss": 3.605206004274396, + "tokens_seen": 1137639424 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033097291875626883, + "loss": 1.4584, + "theoretical_loss": 3.605186708566225, + "tokens_seen": 1137704960 + }, + { + "epoch": 0.34, + "learning_rate": 0.000330962888665998, + "loss": 1.304, + "theoretical_loss": 3.6051674142807277, + "tokens_seen": 1137770496 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003309528585757272, + "loss": 1.3466, + "theoretical_loss": 3.6051481214177157, + "tokens_seen": 1137836032 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003309428284854564, + "loss": 1.3495, + "theoretical_loss": 3.605128829977003, + "tokens_seen": 1137901568 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003309327983951856, + "loss": 1.2888, + "theoretical_loss": 3.605109539958402, + "tokens_seen": 1137967104 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033092276830491474, + "loss": 1.3958, + "theoretical_loss": 3.6050902513617267, + "tokens_seen": 1138032640 + }, + { + "epoch": 0.34, + "learning_rate": 0.000330912738214644, + "loss": 1.3459, + "theoretical_loss": 3.60507096418679, + "tokens_seen": 1138098176 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003309027081243731, + "loss": 1.5288, + "theoretical_loss": 3.605051678433406, + "tokens_seen": 1138163712 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033089267803410234, + "loss": 1.3036, + "theoretical_loss": 3.6050323941013867, + "tokens_seen": 1138229248 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003308826479438315, + "loss": 1.4172, + "theoretical_loss": 3.605013111190547, + "tokens_seen": 1138294784 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003308726178535607, + "loss": 1.365, + "theoretical_loss": 3.6049938297006987, + "tokens_seen": 1138360320 + }, + { + "epoch": 0.34, + "learning_rate": 0.0003308625877632899, + "loss": 1.3904, + "theoretical_loss": 3.604974549631657, + "tokens_seen": 1138425856 + }, + { + "epoch": 0.34, + "learning_rate": 0.00033085255767301906, + "loss": 1.3369, + "theoretical_loss": 3.6049552709832344, + "tokens_seen": 1138491392 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033084252758274824, + "loss": 1.3887, + "theoretical_loss": 3.6049359937552454, + "tokens_seen": 1138556928 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003308324974924775, + "loss": 1.4021, + "theoretical_loss": 3.6049167179475026, + "tokens_seen": 1138622464 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.0013039084151387215, + "objective/train/docs_used": 649005, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.542691230773926, + "objective/train/original_loss": 2.542691230773926, + "objective/train/theoretical_loss": 3.6048974435598193, + "objective/train/tokens_used": 1159148000, + "objective/train/value_avg": -0.0053863525390625, + "objective/train/value_loss": 5.808168862131424e-05, + "objective/train/value_max": -0.0003459453582763672, + "objective/train/value_min": -0.0472412109375, + "objective/train/value_reward_corr": 0.47101798843394743, + "objective/train/value_std": 0.003826141357421875, + "objective/train/weight_avg": 0.9897502064704895, + "objective/train/weighted_lm_loss": 2.5262720584869385, + "objective/train/weights_max": 1.4053038358688354, + "objective/train/weights_min": 0.6965574026107788, + "theoretical_loss": 3.6048974435598193, + "tokens_seen": 1138688000 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003308224674022066, + "loss": 1.4243, + "theoretical_loss": 3.6048974435598193, + "tokens_seen": 1138688000 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033081243731193584, + "loss": 1.327, + "theoretical_loss": 3.6048781705920105, + "tokens_seen": 1138753536 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033080240722166497, + "loss": 1.4062, + "theoretical_loss": 3.6048588990438892, + "tokens_seen": 1138819072 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003307923771313942, + "loss": 1.395, + "theoretical_loss": 3.6048396289152684, + "tokens_seen": 1138884608 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003307823470411234, + "loss": 1.4225, + "theoretical_loss": 3.604820360205964, + "tokens_seen": 1138950144 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033077231695085256, + "loss": 1.4157, + "theoretical_loss": 3.604801092915787, + "tokens_seen": 1139015680 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033076228686058175, + "loss": 1.4247, + "theoretical_loss": 3.6047818270445533, + "tokens_seen": 1139081216 + }, + { + "epoch": 0.35, + "learning_rate": 0.000330752256770311, + "loss": 1.4098, + "theoretical_loss": 3.604762562592076, + "tokens_seen": 1139146752 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003307422266800401, + "loss": 1.4009, + "theoretical_loss": 3.6047432995581694, + "tokens_seen": 1139212288 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033073219658976934, + "loss": 1.3577, + "theoretical_loss": 3.604724037942647, + "tokens_seen": 1139277824 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033072216649949847, + "loss": 1.3809, + "theoretical_loss": 3.604704777745323, + "tokens_seen": 1139343360 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003307121364092277, + "loss": 1.3728, + "theoretical_loss": 3.604685518966011, + "tokens_seen": 1139408896 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003307021063189569, + "loss": 1.4064, + "theoretical_loss": 3.604666261604526, + "tokens_seen": 1139474432 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033069207622868607, + "loss": 1.3086, + "theoretical_loss": 3.604647005660681, + "tokens_seen": 1139539968 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033068204613841525, + "loss": 1.3316, + "theoretical_loss": 3.604627751134291, + "tokens_seen": 1139605504 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033067201604814443, + "loss": 1.337, + "theoretical_loss": 3.6046084980251694, + "tokens_seen": 1139671040 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003306619859578736, + "loss": 1.3954, + "theoretical_loss": 3.604589246333131, + "tokens_seen": 1139736576 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033065195586760285, + "loss": 1.3465, + "theoretical_loss": 3.6045699960579896, + "tokens_seen": 1139802112 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033064192577733197, + "loss": 1.3796, + "theoretical_loss": 3.60455074719956, + "tokens_seen": 1139867648 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003306318956870612, + "loss": 1.3343, + "theoretical_loss": 3.6045314997576563, + "tokens_seen": 1139933184 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033062186559679034, + "loss": 1.3972, + "theoretical_loss": 3.604512253732092, + "tokens_seen": 1139998720 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033061183550651957, + "loss": 1.3415, + "theoretical_loss": 3.604493009122683, + "tokens_seen": 1140064256 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033060180541624875, + "loss": 1.3945, + "theoretical_loss": 3.604473765929242, + "tokens_seen": 1140129792 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033059177532597793, + "loss": 1.511, + "theoretical_loss": 3.6044545241515844, + "tokens_seen": 1140195328 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003305817452357071, + "loss": 1.4469, + "theoretical_loss": 3.604435283789525, + "tokens_seen": 1140260864 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": 0.00043071838445030153, + "objective/train/docs_used": 649993, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.362761974334717, + "objective/train/original_loss": 2.362761974334717, + "objective/train/theoretical_loss": 3.6044160448428775, + "objective/train/tokens_used": 1160786400, + "objective/train/value_avg": -0.006072998046875, + "objective/train/value_loss": 6.09866801823955e-05, + "objective/train/value_max": -0.00034332275390625, + "objective/train/value_min": -0.06793212890625, + "objective/train/value_reward_corr": 0.40833404929624173, + "objective/train/value_std": 0.005352020263671875, + "objective/train/weight_avg": 1.0072382688522339, + "objective/train/weighted_lm_loss": 2.374267816543579, + "objective/train/weights_max": 1.5876960754394531, + "objective/train/weights_min": 0.22472696006298065, + "theoretical_loss": 3.6044160448428775, + "tokens_seen": 1140326400 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033057171514543635, + "loss": 1.3392, + "theoretical_loss": 3.6044160448428775, + "tokens_seen": 1140326400 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003305616850551655, + "loss": 1.39, + "theoretical_loss": 3.6043968073114563, + "tokens_seen": 1140391936 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003305516549648947, + "loss": 1.4262, + "theoretical_loss": 3.6043775711950774, + "tokens_seen": 1140457472 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033054162487462384, + "loss": 1.3999, + "theoretical_loss": 3.6043583364935543, + "tokens_seen": 1140523008 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003305315947843531, + "loss": 1.404, + "theoretical_loss": 3.6043391032067014, + "tokens_seen": 1140588544 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033052156469408225, + "loss": 1.301, + "theoretical_loss": 3.604319871334334, + "tokens_seen": 1140654080 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033051153460381144, + "loss": 1.4413, + "theoretical_loss": 3.604300640876266, + "tokens_seen": 1140719616 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003305015045135406, + "loss": 1.4356, + "theoretical_loss": 3.6042814118323134, + "tokens_seen": 1140785152 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003304914744232698, + "loss": 1.441, + "theoretical_loss": 3.60426218420229, + "tokens_seen": 1140850688 + }, + { + "epoch": 0.35, + "learning_rate": 0.000330481444332999, + "loss": 1.4189, + "theoretical_loss": 3.6042429579860116, + "tokens_seen": 1140916224 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003304714142427282, + "loss": 1.3628, + "theoretical_loss": 3.604223733183292, + "tokens_seen": 1140981760 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033046138415245734, + "loss": 1.4811, + "theoretical_loss": 3.6042045097939464, + "tokens_seen": 1141047296 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003304513540621866, + "loss": 1.4121, + "theoretical_loss": 3.6041852878177902, + "tokens_seen": 1141112832 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003304413239719157, + "loss": 1.3444, + "theoretical_loss": 3.604166067254637, + "tokens_seen": 1141178368 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033043129388164494, + "loss": 1.3962, + "theoretical_loss": 3.6041468481043037, + "tokens_seen": 1141243904 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003304212637913741, + "loss": 1.2978, + "theoretical_loss": 3.6041276303666043, + "tokens_seen": 1141309440 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003304112337011033, + "loss": 1.4441, + "theoretical_loss": 3.6041084140413533, + "tokens_seen": 1141374976 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003304012036108325, + "loss": 1.4075, + "theoretical_loss": 3.6040891991283672, + "tokens_seen": 1141440512 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003303911735205617, + "loss": 1.3965, + "theoretical_loss": 3.60406998562746, + "tokens_seen": 1141506048 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033038114343029084, + "loss": 1.4373, + "theoretical_loss": 3.6040507735384475, + "tokens_seen": 1141571584 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003303711133400201, + "loss": 1.4088, + "theoretical_loss": 3.6040315628611443, + "tokens_seen": 1141637120 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003303610832497492, + "loss": 1.5167, + "theoretical_loss": 3.6040123535953663, + "tokens_seen": 1141702656 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033035105315947844, + "loss": 1.426, + "theoretical_loss": 3.6039931457409287, + "tokens_seen": 1141768192 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003303410230692076, + "loss": 1.3121, + "theoretical_loss": 3.6039739392976458, + "tokens_seen": 1141833728 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003303309929789368, + "loss": 1.3931, + "theoretical_loss": 3.603954734265334, + "tokens_seen": 1141899264 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.0002084509324049577, + "objective/train/docs_used": 650682, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3563456535339355, + "objective/train/original_loss": 2.3563451766967773, + "objective/train/theoretical_loss": 3.603935530643809, + "objective/train/tokens_used": 1162424800, + "objective/train/value_avg": -0.005657196044921875, + "objective/train/value_loss": 7.504377572331578e-05, + "objective/train/value_max": -0.0001926422119140625, + "objective/train/value_min": -0.0755615234375, + "objective/train/value_reward_corr": 0.3175268516902235, + "objective/train/value_std": 0.00511932373046875, + "objective/train/weight_avg": 1.0014132261276245, + "objective/train/weighted_lm_loss": 2.369443655014038, + "objective/train/weights_max": 1.8608487844467163, + "objective/train/weights_min": 0.4285251498222351, + "theoretical_loss": 3.603935530643809, + "tokens_seen": 1141964800 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033032096288866604, + "loss": 1.2983, + "theoretical_loss": 3.603935530643809, + "tokens_seen": 1141964800 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033031093279839517, + "loss": 1.4331, + "theoretical_loss": 3.603916328432885, + "tokens_seen": 1142030336 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003303009027081244, + "loss": 1.3171, + "theoretical_loss": 3.603897127632378, + "tokens_seen": 1142095872 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003302908726178536, + "loss": 1.473, + "theoretical_loss": 3.603877928242104, + "tokens_seen": 1142161408 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033028084252758276, + "loss": 1.3251, + "theoretical_loss": 3.6038587302618774, + "tokens_seen": 1142226944 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033027081243731195, + "loss": 1.4282, + "theoretical_loss": 3.6038395336915148, + "tokens_seen": 1142292480 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003302607823470412, + "loss": 1.3127, + "theoretical_loss": 3.603820338530831, + "tokens_seen": 1142358016 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003302507522567703, + "loss": 1.4374, + "theoretical_loss": 3.6038011447796423, + "tokens_seen": 1142423552 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033024072216649954, + "loss": 1.4089, + "theoretical_loss": 3.603781952437764, + "tokens_seen": 1142489088 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033023069207622867, + "loss": 1.398, + "theoretical_loss": 3.603762761505012, + "tokens_seen": 1142554624 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003302206619859579, + "loss": 1.3436, + "theoretical_loss": 3.6037435719812017, + "tokens_seen": 1142620160 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003302106318956871, + "loss": 1.3936, + "theoretical_loss": 3.603724383866149, + "tokens_seen": 1142685696 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033020060180541627, + "loss": 1.2913, + "theoretical_loss": 3.6037051971596696, + "tokens_seen": 1142751232 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033019057171514545, + "loss": 1.4758, + "theoretical_loss": 3.6036860118615794, + "tokens_seen": 1142816768 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033018054162487463, + "loss": 1.3974, + "theoretical_loss": 3.6036668279716944, + "tokens_seen": 1142882304 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003301705115346038, + "loss": 1.3463, + "theoretical_loss": 3.60364764548983, + "tokens_seen": 1142947840 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033016048144433305, + "loss": 1.3129, + "theoretical_loss": 3.603628464415803, + "tokens_seen": 1143013376 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003301504513540622, + "loss": 1.4613, + "theoretical_loss": 3.603609284749429, + "tokens_seen": 1143078912 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003301404212637914, + "loss": 1.4348, + "theoretical_loss": 3.6035901064905236, + "tokens_seen": 1143144448 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033013039117352054, + "loss": 1.4308, + "theoretical_loss": 3.603570929638903, + "tokens_seen": 1143209984 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033012036108324977, + "loss": 1.4345, + "theoretical_loss": 3.603551754194384, + "tokens_seen": 1143275520 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033011033099297895, + "loss": 1.3632, + "theoretical_loss": 3.6035325801567812, + "tokens_seen": 1143341056 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033010030090270813, + "loss": 1.3457, + "theoretical_loss": 3.603513407525912, + "tokens_seen": 1143406592 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003300902708124373, + "loss": 1.2707, + "theoretical_loss": 3.603494236301592, + "tokens_seen": 1143472128 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033008024072216655, + "loss": 1.392, + "theoretical_loss": 3.603475066483638, + "tokens_seen": 1143537664 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.0003034608671441674, + "objective/train/docs_used": 652177, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6995813846588135, + "objective/train/original_loss": 2.6995811462402344, + "objective/train/theoretical_loss": 3.603455898071866, + "objective/train/tokens_used": 1164063200, + "objective/train/value_avg": -0.004360198974609375, + "objective/train/value_loss": 5.163419700693339e-05, + "objective/train/value_max": -0.00029587745666503906, + "objective/train/value_min": -0.042633056640625, + "objective/train/value_reward_corr": 0.33501131539812884, + "objective/train/value_std": 0.0033321380615234375, + "objective/train/weight_avg": 0.9992786049842834, + "objective/train/weighted_lm_loss": 2.6998579502105713, + "objective/train/weights_max": 1.4564129114151, + "objective/train/weights_min": 0.13780726492404938, + "theoretical_loss": 3.603455898071866, + "tokens_seen": 1143603200 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003300702106318957, + "loss": 1.3778, + "theoretical_loss": 3.603455898071866, + "tokens_seen": 1143603200 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003300601805416249, + "loss": 1.4028, + "theoretical_loss": 3.6034367310660915, + "tokens_seen": 1143668736 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033005015045135404, + "loss": 1.3966, + "theoretical_loss": 3.603417565466132, + "tokens_seen": 1143734272 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003300401203610833, + "loss": 1.4405, + "theoretical_loss": 3.6033984012718028, + "tokens_seen": 1143799808 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033003009027081245, + "loss": 1.4236, + "theoretical_loss": 3.603379238482921, + "tokens_seen": 1143865344 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033002006018054164, + "loss": 1.3561, + "theoretical_loss": 3.603360077099303, + "tokens_seen": 1143930880 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003300100300902708, + "loss": 1.2525, + "theoretical_loss": 3.6033409171207644, + "tokens_seen": 1143996416 + }, + { + "epoch": 0.35, + "learning_rate": 0.00033, + "loss": 1.4402, + "theoretical_loss": 3.603321758547123, + "tokens_seen": 1144061952 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003299899699097292, + "loss": 1.3908, + "theoretical_loss": 3.603302601378194, + "tokens_seen": 1144127488 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003299799398194584, + "loss": 1.2805, + "theoretical_loss": 3.603283445613795, + "tokens_seen": 1144193024 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032996990972918754, + "loss": 1.3564, + "theoretical_loss": 3.6032642912537423, + "tokens_seen": 1144258560 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003299598796389168, + "loss": 1.4468, + "theoretical_loss": 3.6032451382978516, + "tokens_seen": 1144324096 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003299498495486459, + "loss": 1.3814, + "theoretical_loss": 3.603225986745941, + "tokens_seen": 1144389632 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032993981945837514, + "loss": 1.3812, + "theoretical_loss": 3.6032068365978267, + "tokens_seen": 1144455168 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003299297893681043, + "loss": 1.3037, + "theoretical_loss": 3.6031876878533247, + "tokens_seen": 1144520704 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003299197592778335, + "loss": 1.4085, + "theoretical_loss": 3.603168540512253, + "tokens_seen": 1144586240 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003299097291875627, + "loss": 1.4972, + "theoretical_loss": 3.603149394574427, + "tokens_seen": 1144651776 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003298996990972919, + "loss": 1.386, + "theoretical_loss": 3.6031302500396647, + "tokens_seen": 1144717312 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032988966900702104, + "loss": 1.3558, + "theoretical_loss": 3.6031111069077824, + "tokens_seen": 1144782848 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003298796389167503, + "loss": 1.2801, + "theoretical_loss": 3.6030919651785966, + "tokens_seen": 1144848384 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003298696088264794, + "loss": 1.4609, + "theoretical_loss": 3.6030728248519255, + "tokens_seen": 1144913920 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032985957873620864, + "loss": 1.3185, + "theoretical_loss": 3.603053685927585, + "tokens_seen": 1144979456 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003298495486459378, + "loss": 1.5147, + "theoretical_loss": 3.6030345484053923, + "tokens_seen": 1145044992 + }, + { + "epoch": 0.35, + "learning_rate": 0.000329839518555667, + "loss": 1.3806, + "theoretical_loss": 3.6030154122851643, + "tokens_seen": 1145110528 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003298294884653962, + "loss": 1.3631, + "theoretical_loss": 3.6029962775667186, + "tokens_seen": 1145176064 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.0002790552971418947, + "objective/train/docs_used": 652765, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6933295726776123, + "objective/train/original_loss": 2.6933295726776123, + "objective/train/theoretical_loss": 3.6029771442498717, + "objective/train/tokens_used": 1165701600, + "objective/train/value_avg": -0.00583648681640625, + "objective/train/value_loss": 6.465191836468875e-05, + "objective/train/value_max": -0.0002892017364501953, + "objective/train/value_min": -0.053192138671875, + "objective/train/value_reward_corr": 0.35009916009393927, + "objective/train/value_std": 0.004474639892578125, + "objective/train/weight_avg": 1.000278353691101, + "objective/train/weighted_lm_loss": 2.6970455646514893, + "objective/train/weights_max": 1.5897618532180786, + "objective/train/weights_min": 0.48261889815330505, + "theoretical_loss": 3.6029771442498717, + "tokens_seen": 1145241600 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032981945837512537, + "loss": 1.2712, + "theoretical_loss": 3.6029771442498717, + "tokens_seen": 1145241600 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032980942828485455, + "loss": 1.3042, + "theoretical_loss": 3.602958012334441, + "tokens_seen": 1145307136 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003297993981945838, + "loss": 1.3857, + "theoretical_loss": 3.6029388818202444, + "tokens_seen": 1145372672 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003297893681043129, + "loss": 1.3087, + "theoretical_loss": 3.6029197527070975, + "tokens_seen": 1145438208 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032977933801404215, + "loss": 1.3956, + "theoretical_loss": 3.6029006249948194, + "tokens_seen": 1145503744 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003297693079237713, + "loss": 1.4464, + "theoretical_loss": 3.602881498683226, + "tokens_seen": 1145569280 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003297592778335005, + "loss": 1.3487, + "theoretical_loss": 3.602862373772135, + "tokens_seen": 1145634816 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003297492477432297, + "loss": 1.2772, + "theoretical_loss": 3.6028432502613637, + "tokens_seen": 1145700352 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032973921765295887, + "loss": 1.4547, + "theoretical_loss": 3.602824128150729, + "tokens_seen": 1145765888 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032972918756268805, + "loss": 1.3723, + "theoretical_loss": 3.60280500744005, + "tokens_seen": 1145831424 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003297191574724173, + "loss": 1.3294, + "theoretical_loss": 3.6027858881291426, + "tokens_seen": 1145896960 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003297091273821464, + "loss": 1.3293, + "theoretical_loss": 3.6027667702178245, + "tokens_seen": 1145962496 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032969909729187565, + "loss": 1.324, + "theoretical_loss": 3.6027476537059133, + "tokens_seen": 1146028032 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003296890672016048, + "loss": 1.333, + "theoretical_loss": 3.602728538593227, + "tokens_seen": 1146093568 + }, + { + "epoch": 0.35, + "learning_rate": 0.000329679037111334, + "loss": 1.2931, + "theoretical_loss": 3.6027094248795826, + "tokens_seen": 1146159104 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003296690070210632, + "loss": 1.3474, + "theoretical_loss": 3.602690312564798, + "tokens_seen": 1146224640 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003296589769307924, + "loss": 1.314, + "theoretical_loss": 3.6026712016486906, + "tokens_seen": 1146290176 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032964894684052155, + "loss": 1.3079, + "theoretical_loss": 3.6026520921310787, + "tokens_seen": 1146355712 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032963891675025074, + "loss": 1.4329, + "theoretical_loss": 3.6026329840117794, + "tokens_seen": 1146421248 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003296288866599799, + "loss": 1.3533, + "theoretical_loss": 3.6026138772906107, + "tokens_seen": 1146486784 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032961885656970915, + "loss": 1.4761, + "theoretical_loss": 3.6025947719673903, + "tokens_seen": 1146552320 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003296088264794383, + "loss": 1.3019, + "theoretical_loss": 3.6025756680419363, + "tokens_seen": 1146617856 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003295987963891675, + "loss": 1.4697, + "theoretical_loss": 3.6025565655140657, + "tokens_seen": 1146683392 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003295887662988967, + "loss": 1.4074, + "theoretical_loss": 3.6025374643835977, + "tokens_seen": 1146748928 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003295787362086259, + "loss": 1.4125, + "theoretical_loss": 3.602518364650349, + "tokens_seen": 1146814464 + }, + { + "debugging/Self-BLEU-5": 0.362408135719388, + "debugging/distinct-1-grams": 0.7792640637334625, + "debugging/distinct-2-grams": 0.9744957819884318, + "debugging/entropy-1-grams": 5.554469750334817, + "debugging/entropy-2-grams": 6.28719269992704, + "debugging/length": 523.8571428571429, + "debugging/num_segments": 7, + "debugging/raw_token_scores_avg": 0.005528885871171951, + "debugging/raw_token_scores_std": 0.006955595687031746, + "debugging/score": 0.005264410443902509, + "debugging/score_std": 0.0034124891309377393, + "epoch": 0.35, + "objective/train/advantage_avg": -0.0002535614767111838, + "objective/train/docs_used": 652965, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1386666297912598, + "objective/train/original_loss": 3.1386663913726807, + "objective/train/theoretical_loss": 3.6024992663141386, + "objective/train/tokens_used": 1167340000, + "objective/train/value_avg": -0.005275726318359375, + "objective/train/value_loss": 4.791158062289469e-05, + "objective/train/value_max": -0.00031256675720214844, + "objective/train/value_min": -0.05499267578125, + "objective/train/value_reward_corr": 0.3192681159730586, + "objective/train/value_std": 0.0043182373046875, + "objective/train/weight_avg": 0.9997957348823547, + "objective/train/weighted_lm_loss": 3.14437198638916, + "objective/train/weights_max": 1.6357858180999756, + "objective/train/weights_min": 0.47669801115989685, + "theoretical_loss": 3.6024992663141386, + "tokens_seen": 1146880000 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003295687061183551, + "loss": 1.5345, + "theoretical_loss": 3.6024992663141386, + "tokens_seen": 1146880000 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032955867602808424, + "loss": 1.3169, + "theoretical_loss": 3.602480169374784, + "tokens_seen": 1146945536 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003295486459378135, + "loss": 1.4203, + "theoretical_loss": 3.602461073832103, + "tokens_seen": 1147011072 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032953861584754265, + "loss": 1.429, + "theoretical_loss": 3.6024419796859135, + "tokens_seen": 1147076608 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032952858575727184, + "loss": 1.4054, + "theoretical_loss": 3.6024228869360346, + "tokens_seen": 1147142144 + }, + { + "epoch": 0.35, + "learning_rate": 0.000329518555667001, + "loss": 1.3754, + "theoretical_loss": 3.6024037955822834, + "tokens_seen": 1147207680 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003295085255767302, + "loss": 1.3654, + "theoretical_loss": 3.6023847056244787, + "tokens_seen": 1147273216 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003294984954864594, + "loss": 1.3958, + "theoretical_loss": 3.6023656170624383, + "tokens_seen": 1147338752 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003294884653961886, + "loss": 1.4749, + "theoretical_loss": 3.6023465298959807, + "tokens_seen": 1147404288 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032947843530591774, + "loss": 1.3281, + "theoretical_loss": 3.6023274441249242, + "tokens_seen": 1147469824 + }, + { + "epoch": 0.35, + "learning_rate": 0.000329468405215647, + "loss": 1.3396, + "theoretical_loss": 3.6023083597490864, + "tokens_seen": 1147535360 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003294583751253761, + "loss": 1.3604, + "theoretical_loss": 3.6022892767682873, + "tokens_seen": 1147600896 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032944834503510534, + "loss": 1.4448, + "theoretical_loss": 3.602270195182343, + "tokens_seen": 1147666432 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003294383149448345, + "loss": 1.371, + "theoretical_loss": 3.6022511149910734, + "tokens_seen": 1147731968 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003294282848545637, + "loss": 1.4518, + "theoretical_loss": 3.602232036194297, + "tokens_seen": 1147797504 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003294182547642929, + "loss": 1.4744, + "theoretical_loss": 3.6022129587918315, + "tokens_seen": 1147863040 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003294082246740221, + "loss": 1.4459, + "theoretical_loss": 3.602193882783496, + "tokens_seen": 1147928576 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032939819458375124, + "loss": 1.4224, + "theoretical_loss": 3.6021748081691083, + "tokens_seen": 1147994112 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003293881644934805, + "loss": 1.4746, + "theoretical_loss": 3.602155734948488, + "tokens_seen": 1148059648 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003293781344032096, + "loss": 1.2876, + "theoretical_loss": 3.602136663121453, + "tokens_seen": 1148125184 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032936810431293884, + "loss": 1.4758, + "theoretical_loss": 3.602117592687822, + "tokens_seen": 1148190720 + }, + { + "epoch": 0.35, + "learning_rate": 0.000329358074222668, + "loss": 1.2448, + "theoretical_loss": 3.602098523647414, + "tokens_seen": 1148256256 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003293480441323972, + "loss": 1.3583, + "theoretical_loss": 3.6020794560000473, + "tokens_seen": 1148321792 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003293380140421264, + "loss": 1.3843, + "theoretical_loss": 3.6020603897455414, + "tokens_seen": 1148387328 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032932798395185557, + "loss": 1.4138, + "theoretical_loss": 3.602041324883714, + "tokens_seen": 1148452864 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.00027805176796391606, + "objective/train/docs_used": 654364, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.755610704421997, + "objective/train/original_loss": 2.755610466003418, + "objective/train/theoretical_loss": 3.6020222614143846, + "objective/train/tokens_used": 1168978400, + "objective/train/value_avg": -0.005859375, + "objective/train/value_loss": 7.38558082957752e-05, + "objective/train/value_max": -0.00047469139099121094, + "objective/train/value_min": -0.1729736328125, + "objective/train/value_reward_corr": 0.2975147236473064, + "objective/train/value_std": 0.004425048828125, + "objective/train/weight_avg": 1.0006316900253296, + "objective/train/weighted_lm_loss": 2.75364351272583, + "objective/train/weights_max": 4.462986469268799, + "objective/train/weights_min": 0.3000735342502594, + "theoretical_loss": 3.6020222614143846, + "tokens_seen": 1148518400 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032931795386158475, + "loss": 1.5132, + "theoretical_loss": 3.6020222614143846, + "tokens_seen": 1148518400 + }, + { + "epoch": 0.35, + "learning_rate": 0.000329307923771314, + "loss": 1.4121, + "theoretical_loss": 3.602003199337372, + "tokens_seen": 1148583936 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003292978936810431, + "loss": 1.3038, + "theoretical_loss": 3.6019841386524947, + "tokens_seen": 1148649472 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032928786359077235, + "loss": 1.2788, + "theoretical_loss": 3.601965079359572, + "tokens_seen": 1148715008 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003292778335005015, + "loss": 1.2857, + "theoretical_loss": 3.6019460214584225, + "tokens_seen": 1148780544 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003292678034102307, + "loss": 1.447, + "theoretical_loss": 3.6019269649488654, + "tokens_seen": 1148846080 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003292577733199599, + "loss": 1.4843, + "theoretical_loss": 3.60190790983072, + "tokens_seen": 1148911616 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032924774322968907, + "loss": 1.4267, + "theoretical_loss": 3.6018888561038045, + "tokens_seen": 1148977152 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032923771313941825, + "loss": 1.3448, + "theoretical_loss": 3.601869803767939, + "tokens_seen": 1149042688 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003292276830491475, + "loss": 1.3887, + "theoretical_loss": 3.601850752822942, + "tokens_seen": 1149108224 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003292176529588766, + "loss": 1.2928, + "theoretical_loss": 3.601831703268633, + "tokens_seen": 1149173760 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032920762286860585, + "loss": 1.2883, + "theoretical_loss": 3.6018126551048306, + "tokens_seen": 1149239296 + }, + { + "epoch": 0.35, + "learning_rate": 0.000329197592778335, + "loss": 1.4822, + "theoretical_loss": 3.601793608331354, + "tokens_seen": 1149304832 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003291875626880642, + "loss": 1.34, + "theoretical_loss": 3.601774562948024, + "tokens_seen": 1149370368 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003291775325977934, + "loss": 1.2739, + "theoretical_loss": 3.6017555189546577, + "tokens_seen": 1149435904 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003291675025075226, + "loss": 1.3634, + "theoretical_loss": 3.6017364763510757, + "tokens_seen": 1149501440 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032915747241725175, + "loss": 1.5328, + "theoretical_loss": 3.6017174351370973, + "tokens_seen": 1149566976 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032914744232698094, + "loss": 1.3129, + "theoretical_loss": 3.6016983953125417, + "tokens_seen": 1149632512 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003291374122367101, + "loss": 1.3542, + "theoretical_loss": 3.601679356877228, + "tokens_seen": 1149698048 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032912738214643935, + "loss": 1.2234, + "theoretical_loss": 3.601660319830976, + "tokens_seen": 1149763584 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003291173520561685, + "loss": 1.5629, + "theoretical_loss": 3.6016412841736054, + "tokens_seen": 1149829120 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003291073219658977, + "loss": 1.1947, + "theoretical_loss": 3.601622249904935, + "tokens_seen": 1149894656 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003290972918756269, + "loss": 1.4872, + "theoretical_loss": 3.6016032170247847, + "tokens_seen": 1149960192 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003290872617853561, + "loss": 1.4353, + "theoretical_loss": 3.6015841855329747, + "tokens_seen": 1150025728 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032907723169508526, + "loss": 1.2912, + "theoretical_loss": 3.6015651554293235, + "tokens_seen": 1150091264 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.00022776929836254567, + "objective/train/docs_used": 654951, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3279616832733154, + "objective/train/original_loss": 2.3279616832733154, + "objective/train/theoretical_loss": 3.601546126713652, + "objective/train/tokens_used": 1170616800, + "objective/train/value_avg": -0.00508880615234375, + "objective/train/value_loss": 4.722634912468493e-05, + "objective/train/value_max": -0.0004172325134277344, + "objective/train/value_min": -0.0809326171875, + "objective/train/value_reward_corr": 0.3072128537547628, + "objective/train/value_std": 0.00392913818359375, + "objective/train/weight_avg": 0.9999482035636902, + "objective/train/weighted_lm_loss": 2.342810869216919, + "objective/train/weights_max": 1.9260668754577637, + "objective/train/weights_min": 0.3773455023765564, + "theoretical_loss": 3.601546126713652, + "tokens_seen": 1150156800 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032906720160481444, + "loss": 1.4321, + "theoretical_loss": 3.601546126713652, + "tokens_seen": 1150156800 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003290571715145436, + "loss": 1.4235, + "theoretical_loss": 3.601527099385779, + "tokens_seen": 1150222336 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032904714142427286, + "loss": 1.4268, + "theoretical_loss": 3.6015080734455243, + "tokens_seen": 1150287872 + }, + { + "epoch": 0.35, + "learning_rate": 0.000329037111334002, + "loss": 1.4644, + "theoretical_loss": 3.601489048892708, + "tokens_seen": 1150353408 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003290270812437312, + "loss": 1.4035, + "theoretical_loss": 3.6014700257271492, + "tokens_seen": 1150418944 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032901705115346034, + "loss": 1.3944, + "theoretical_loss": 3.601451003948669, + "tokens_seen": 1150484480 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003290070210631896, + "loss": 1.3067, + "theoretical_loss": 3.6014319835570863, + "tokens_seen": 1150550016 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032899699097291876, + "loss": 1.4117, + "theoretical_loss": 3.601412964552221, + "tokens_seen": 1150615552 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032898696088264794, + "loss": 1.3035, + "theoretical_loss": 3.601393946933893, + "tokens_seen": 1150681088 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003289769307923771, + "loss": 1.3579, + "theoretical_loss": 3.601374930701923, + "tokens_seen": 1150746624 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003289669007021063, + "loss": 1.317, + "theoretical_loss": 3.60135591585613, + "tokens_seen": 1150812160 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003289568706118355, + "loss": 1.4046, + "theoretical_loss": 3.6013369023963344, + "tokens_seen": 1150877696 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003289468405215647, + "loss": 1.267, + "theoretical_loss": 3.6013178903223566, + "tokens_seen": 1150943232 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032893681043129385, + "loss": 1.4097, + "theoretical_loss": 3.601298879634016, + "tokens_seen": 1151008768 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003289267803410231, + "loss": 1.393, + "theoretical_loss": 3.6012798703311337, + "tokens_seen": 1151074304 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032891675025075226, + "loss": 1.2542, + "theoretical_loss": 3.6012608624135294, + "tokens_seen": 1151139840 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032890672016048145, + "loss": 1.3622, + "theoretical_loss": 3.601241855881023, + "tokens_seen": 1151205376 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003288966900702106, + "loss": 1.3947, + "theoretical_loss": 3.6012228507334347, + "tokens_seen": 1151270912 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003288866599799398, + "loss": 1.3683, + "theoretical_loss": 3.601203846970585, + "tokens_seen": 1151336448 + }, + { + "epoch": 0.35, + "learning_rate": 0.000328876629889669, + "loss": 1.2855, + "theoretical_loss": 3.6011848445922943, + "tokens_seen": 1151401984 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003288665997993982, + "loss": 1.4262, + "theoretical_loss": 3.6011658435983827, + "tokens_seen": 1151467520 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032885656970912735, + "loss": 1.4135, + "theoretical_loss": 3.601146843988671, + "tokens_seen": 1151533056 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003288465396188566, + "loss": 1.4439, + "theoretical_loss": 3.6011278457629787, + "tokens_seen": 1151598592 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003288365095285857, + "loss": 1.3711, + "theoretical_loss": 3.601108848921127, + "tokens_seen": 1151664128 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032882647943831495, + "loss": 1.5088, + "theoretical_loss": 3.6010898534629363, + "tokens_seen": 1151729664 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": 5.601625525741838e-05, + "objective/train/docs_used": 656079, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9972198009490967, + "objective/train/original_loss": 2.9972198009490967, + "objective/train/theoretical_loss": 3.601070859388227, + "objective/train/tokens_used": 1172255200, + "objective/train/value_avg": -0.00506591796875, + "objective/train/value_loss": 5.7510824262863025e-05, + "objective/train/value_max": -0.00026535987854003906, + "objective/train/value_min": -0.06451416015625, + "objective/train/value_reward_corr": 0.4047687267916431, + "objective/train/value_std": 0.00395965576171875, + "objective/train/weight_avg": 1.0032886266708374, + "objective/train/weighted_lm_loss": 2.9933550357818604, + "objective/train/weights_max": 1.7945950031280518, + "objective/train/weights_min": 0.5003408789634705, + "theoretical_loss": 3.601070859388227, + "tokens_seen": 1151795200 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003288164493480442, + "loss": 1.4253, + "theoretical_loss": 3.601070859388227, + "tokens_seen": 1151795200 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003288064192577733, + "loss": 1.5096, + "theoretical_loss": 3.601051866696819, + "tokens_seen": 1151860736 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032879638916750255, + "loss": 1.2617, + "theoretical_loss": 3.601032875388534, + "tokens_seen": 1151926272 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003287863590772317, + "loss": 1.4589, + "theoretical_loss": 3.6010138854631917, + "tokens_seen": 1151991808 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003287763289869609, + "loss": 1.3642, + "theoretical_loss": 3.6009948969206134, + "tokens_seen": 1152057344 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003287662988966901, + "loss": 1.3454, + "theoretical_loss": 3.6009759097606193, + "tokens_seen": 1152122880 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032875626880641927, + "loss": 1.377, + "theoretical_loss": 3.6009569239830306, + "tokens_seen": 1152188416 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032874623871614845, + "loss": 1.4136, + "theoretical_loss": 3.600937939587667, + "tokens_seen": 1152253952 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003287362086258777, + "loss": 1.3791, + "theoretical_loss": 3.60091895657435, + "tokens_seen": 1152319488 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003287261785356068, + "loss": 1.3679, + "theoretical_loss": 3.6008999749429007, + "tokens_seen": 1152385024 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032871614844533605, + "loss": 1.4407, + "theoretical_loss": 3.600880994693139, + "tokens_seen": 1152450560 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003287061183550652, + "loss": 1.3095, + "theoretical_loss": 3.6008620158248865, + "tokens_seen": 1152516096 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003286960882647944, + "loss": 1.4116, + "theoretical_loss": 3.6008430383379637, + "tokens_seen": 1152581632 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003286860581745236, + "loss": 1.4194, + "theoretical_loss": 3.600824062232192, + "tokens_seen": 1152647168 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003286760280842528, + "loss": 1.3848, + "theoretical_loss": 3.600805087507392, + "tokens_seen": 1152712704 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032866599799398195, + "loss": 1.324, + "theoretical_loss": 3.600786114163385, + "tokens_seen": 1152778240 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032865596790371114, + "loss": 1.4395, + "theoretical_loss": 3.600767142199991, + "tokens_seen": 1152843776 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003286459378134403, + "loss": 1.3629, + "theoretical_loss": 3.600748171617033, + "tokens_seen": 1152909312 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032863590772316955, + "loss": 1.4286, + "theoretical_loss": 3.60072920241433, + "tokens_seen": 1152974848 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003286258776328987, + "loss": 1.3999, + "theoretical_loss": 3.6007102345917046, + "tokens_seen": 1153040384 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003286158475426279, + "loss": 1.3587, + "theoretical_loss": 3.600691268148977, + "tokens_seen": 1153105920 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003286058174523571, + "loss": 1.3292, + "theoretical_loss": 3.600672303085969, + "tokens_seen": 1153171456 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003285957873620863, + "loss": 1.4252, + "theoretical_loss": 3.600653339402502, + "tokens_seen": 1153236992 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032858575727181546, + "loss": 1.4206, + "theoretical_loss": 3.600634377098397, + "tokens_seen": 1153302528 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032857572718154464, + "loss": 1.3896, + "theoretical_loss": 3.6006154161734747, + "tokens_seen": 1153368064 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.0007619676180183887, + "objective/train/docs_used": 656588, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3862321376800537, + "objective/train/original_loss": 3.3862316608428955, + "objective/train/theoretical_loss": 3.6005964566275575, + "objective/train/tokens_used": 1173893600, + "objective/train/value_avg": -0.006290435791015625, + "objective/train/value_loss": 7.092250598361716e-05, + "objective/train/value_max": -0.0002512931823730469, + "objective/train/value_min": -0.06097412109375, + "objective/train/value_reward_corr": 0.3802749079967257, + "objective/train/value_std": 0.004703521728515625, + "objective/train/weight_avg": 0.9957212805747986, + "objective/train/weighted_lm_loss": 3.3786723613739014, + "objective/train/weights_max": 1.7205208539962769, + "objective/train/weights_min": 0.40454092621803284, + "theoretical_loss": 3.6005964566275575, + "tokens_seen": 1153433600 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003285656970912738, + "loss": 1.4621, + "theoretical_loss": 3.6005964566275575, + "tokens_seen": 1153433600 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032855566700100306, + "loss": 1.2781, + "theoretical_loss": 3.600577498460466, + "tokens_seen": 1153499136 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003285456369107322, + "loss": 1.3135, + "theoretical_loss": 3.600558541672022, + "tokens_seen": 1153564672 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003285356068204614, + "loss": 1.4725, + "theoretical_loss": 3.6005395862620464, + "tokens_seen": 1153630208 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032852557673019054, + "loss": 1.3766, + "theoretical_loss": 3.600520632230361, + "tokens_seen": 1153695744 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003285155466399198, + "loss": 1.3016, + "theoretical_loss": 3.6005016795767877, + "tokens_seen": 1153761280 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032850551654964896, + "loss": 1.3207, + "theoretical_loss": 3.600482728301148, + "tokens_seen": 1153826816 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032849548645937814, + "loss": 1.4395, + "theoretical_loss": 3.6004637784032623, + "tokens_seen": 1153892352 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003284854563691073, + "loss": 1.4503, + "theoretical_loss": 3.600444829882953, + "tokens_seen": 1153957888 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003284754262788365, + "loss": 1.2678, + "theoretical_loss": 3.600425882740042, + "tokens_seen": 1154023424 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003284653961885657, + "loss": 1.3752, + "theoretical_loss": 3.6004069369743505, + "tokens_seen": 1154088960 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003284553660982949, + "loss": 1.3674, + "theoretical_loss": 3.6003879925857003, + "tokens_seen": 1154154496 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032844533600802405, + "loss": 1.3912, + "theoretical_loss": 3.6003690495739136, + "tokens_seen": 1154220032 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003284353059177533, + "loss": 1.3746, + "theoretical_loss": 3.600350107938812, + "tokens_seen": 1154285568 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032842527582748246, + "loss": 1.3782, + "theoretical_loss": 3.6003311676802165, + "tokens_seen": 1154351104 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032841524573721165, + "loss": 1.4886, + "theoretical_loss": 3.6003122287979497, + "tokens_seen": 1154416640 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003284052156469408, + "loss": 1.3755, + "theoretical_loss": 3.600293291291833, + "tokens_seen": 1154482176 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032839518555667, + "loss": 1.4556, + "theoretical_loss": 3.600274355161689, + "tokens_seen": 1154547712 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003283851554663992, + "loss": 1.3461, + "theoretical_loss": 3.600255420407339, + "tokens_seen": 1154613248 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003283751253761284, + "loss": 1.3879, + "theoretical_loss": 3.6002364870286057, + "tokens_seen": 1154678784 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032836509528585755, + "loss": 1.3431, + "theoretical_loss": 3.60021755502531, + "tokens_seen": 1154744320 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003283550651955868, + "loss": 1.4104, + "theoretical_loss": 3.600198624397274, + "tokens_seen": 1154809856 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003283450351053159, + "loss": 1.3509, + "theoretical_loss": 3.6001796951443206, + "tokens_seen": 1154875392 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032833500501504515, + "loss": 1.3837, + "theoretical_loss": 3.6001607672662717, + "tokens_seen": 1154940928 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032832497492477433, + "loss": 1.3325, + "theoretical_loss": 3.600141840762949, + "tokens_seen": 1155006464 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.0008738139877095819, + "objective/train/docs_used": 657761, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.783278703689575, + "objective/train/original_loss": 2.783278465270996, + "objective/train/theoretical_loss": 3.600122915634175, + "objective/train/tokens_used": 1175532000, + "objective/train/value_avg": -0.004940032958984375, + "objective/train/value_loss": 6.278205546550453e-05, + "objective/train/value_max": -0.0002613067626953125, + "objective/train/value_min": -0.06915283203125, + "objective/train/value_reward_corr": 0.33588443658431033, + "objective/train/value_std": 0.0038356781005859375, + "objective/train/weight_avg": 0.994208812713623, + "objective/train/weighted_lm_loss": 2.7736127376556396, + "objective/train/weights_max": 1.6775728464126587, + "objective/train/weights_min": 0.5178130865097046, + "theoretical_loss": 3.600122915634175, + "tokens_seen": 1155072000 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003283149448345035, + "loss": 1.3893, + "theoretical_loss": 3.600122915634175, + "tokens_seen": 1155072000 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003283049147442327, + "loss": 1.365, + "theoretical_loss": 3.600103991879772, + "tokens_seen": 1155137536 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003282948846539619, + "loss": 1.3095, + "theoretical_loss": 3.6000850694995616, + "tokens_seen": 1155203072 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032828485456369105, + "loss": 1.3709, + "theoretical_loss": 3.600066148493367, + "tokens_seen": 1155268608 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003282748244734203, + "loss": 1.3646, + "theoretical_loss": 3.6000472288610097, + "tokens_seen": 1155334144 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003282647943831494, + "loss": 1.4655, + "theoretical_loss": 3.600028310602312, + "tokens_seen": 1155399680 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032825476429287865, + "loss": 1.3479, + "theoretical_loss": 3.600009393717097, + "tokens_seen": 1155465216 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032824473420260783, + "loss": 1.3576, + "theoretical_loss": 3.5999904782051866, + "tokens_seen": 1155530752 + }, + { + "epoch": 0.35, + "learning_rate": 0.000328234704112337, + "loss": 1.4235, + "theoretical_loss": 3.599971564066403, + "tokens_seen": 1155596288 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003282246740220662, + "loss": 1.4392, + "theoretical_loss": 3.599952651300569, + "tokens_seen": 1155661824 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003282146439317954, + "loss": 1.3721, + "theoretical_loss": 3.599933739907507, + "tokens_seen": 1155727360 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032820461384152456, + "loss": 1.3563, + "theoretical_loss": 3.59991482988704, + "tokens_seen": 1155792896 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003281945837512538, + "loss": 1.4583, + "theoretical_loss": 3.5998959212389896, + "tokens_seen": 1155858432 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003281845536609829, + "loss": 1.3399, + "theoretical_loss": 3.5998770139631793, + "tokens_seen": 1155923968 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032817452357071215, + "loss": 1.358, + "theoretical_loss": 3.5998581080594305, + "tokens_seen": 1155989504 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003281644934804413, + "loss": 1.3683, + "theoretical_loss": 3.5998392035275675, + "tokens_seen": 1156055040 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003281544633901705, + "loss": 1.3662, + "theoretical_loss": 3.5998203003674116, + "tokens_seen": 1156120576 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003281444332998997, + "loss": 1.4077, + "theoretical_loss": 3.5998013985787862, + "tokens_seen": 1156186112 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003281344032096289, + "loss": 1.3239, + "theoretical_loss": 3.599782498161514, + "tokens_seen": 1156251648 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032812437311935806, + "loss": 1.336, + "theoretical_loss": 3.599763599115418, + "tokens_seen": 1156317184 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003281143430290873, + "loss": 1.3548, + "theoretical_loss": 3.5997447014403203, + "tokens_seen": 1156382720 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003281043129388164, + "loss": 1.4686, + "theoretical_loss": 3.5997258051360443, + "tokens_seen": 1156448256 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032809428284854566, + "loss": 1.2706, + "theoretical_loss": 3.5997069102024124, + "tokens_seen": 1156513792 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003280842527582748, + "loss": 1.3751, + "theoretical_loss": 3.5996880166392486, + "tokens_seen": 1156579328 + }, + { + "epoch": 0.35, + "learning_rate": 0.000328074222668004, + "loss": 1.2811, + "theoretical_loss": 3.5996691244463745, + "tokens_seen": 1156644864 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.00037445290945470333, + "objective/train/docs_used": 658495, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4074206352233887, + "objective/train/original_loss": 2.4074206352233887, + "objective/train/theoretical_loss": 3.5996502336236142, + "objective/train/tokens_used": 1177170400, + "objective/train/value_avg": -0.00547027587890625, + "objective/train/value_loss": 4.132292451686226e-05, + "objective/train/value_max": -0.00048041343688964844, + "objective/train/value_min": -0.073974609375, + "objective/train/value_reward_corr": 0.45810818792108804, + "objective/train/value_std": 0.00476837158203125, + "objective/train/weight_avg": 0.9982514977455139, + "objective/train/weighted_lm_loss": 2.3989474773406982, + "objective/train/weights_max": 1.6352770328521729, + "objective/train/weights_min": 0.4854995012283325, + "theoretical_loss": 3.5996502336236142, + "tokens_seen": 1156710400 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032806419257773326, + "loss": 1.3752, + "theoretical_loss": 3.5996502336236142, + "tokens_seen": 1156710400 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003280541624874624, + "loss": 1.3277, + "theoretical_loss": 3.59963134417079, + "tokens_seen": 1156775936 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003280441323971916, + "loss": 1.4036, + "theoretical_loss": 3.5996124560877254, + "tokens_seen": 1156841472 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032803410230692074, + "loss": 1.4484, + "theoretical_loss": 3.5995935693742434, + "tokens_seen": 1156907008 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032802407221665, + "loss": 1.3968, + "theoretical_loss": 3.5995746840301663, + "tokens_seen": 1156972544 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032801404212637916, + "loss": 1.2908, + "theoretical_loss": 3.5995558000553185, + "tokens_seen": 1157038080 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032800401203610834, + "loss": 1.274, + "theoretical_loss": 3.5995369174495226, + "tokens_seen": 1157103616 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003279939819458375, + "loss": 1.4769, + "theoretical_loss": 3.5995180362126016, + "tokens_seen": 1157169152 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003279839518555667, + "loss": 1.4106, + "theoretical_loss": 3.599499156344379, + "tokens_seen": 1157234688 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003279739217652959, + "loss": 1.3343, + "theoretical_loss": 3.5994802778446786, + "tokens_seen": 1157300224 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003279638916750251, + "loss": 1.427, + "theoretical_loss": 3.5994614007133228, + "tokens_seen": 1157365760 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032795386158475425, + "loss": 1.4199, + "theoretical_loss": 3.5994425249501347, + "tokens_seen": 1157431296 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003279438314944835, + "loss": 1.3865, + "theoretical_loss": 3.599423650554939, + "tokens_seen": 1157496832 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032793380140421266, + "loss": 1.4171, + "theoretical_loss": 3.599404777527558, + "tokens_seen": 1157562368 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032792377131394185, + "loss": 1.3832, + "theoretical_loss": 3.599385905867816, + "tokens_seen": 1157627904 + }, + { + "epoch": 0.35, + "learning_rate": 0.000327913741223671, + "loss": 1.3993, + "theoretical_loss": 3.5993670355755363, + "tokens_seen": 1157693440 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003279037111334002, + "loss": 1.4477, + "theoretical_loss": 3.5993481666505414, + "tokens_seen": 1157758976 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003278936810431294, + "loss": 1.3459, + "theoretical_loss": 3.5993292990926555, + "tokens_seen": 1157824512 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003278836509528586, + "loss": 1.304, + "theoretical_loss": 3.599310432901703, + "tokens_seen": 1157890048 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032787362086258775, + "loss": 1.4989, + "theoretical_loss": 3.599291568077506, + "tokens_seen": 1157955584 + }, + { + "epoch": 0.35, + "learning_rate": 0.000327863590772317, + "loss": 1.3164, + "theoretical_loss": 3.5992727046198896, + "tokens_seen": 1158021120 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003278535606820461, + "loss": 1.3453, + "theoretical_loss": 3.599253842528676, + "tokens_seen": 1158086656 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032784353059177535, + "loss": 1.3761, + "theoretical_loss": 3.599234981803691, + "tokens_seen": 1158152192 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032783350050150453, + "loss": 1.3742, + "theoretical_loss": 3.599216122444756, + "tokens_seen": 1158217728 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003278234704112337, + "loss": 1.3853, + "theoretical_loss": 3.5991972644516954, + "tokens_seen": 1158283264 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -1.9274410078651272e-05, + "objective/train/docs_used": 659118, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.098017930984497, + "objective/train/original_loss": 3.098017692565918, + "objective/train/theoretical_loss": 3.5991784078243345, + "objective/train/tokens_used": 1178808800, + "objective/train/value_avg": -0.00518035888671875, + "objective/train/value_loss": 5.786024121334776e-05, + "objective/train/value_max": -0.00014770030975341797, + "objective/train/value_min": -0.090087890625, + "objective/train/value_reward_corr": 0.4080637740888387, + "objective/train/value_std": 0.0045166015625, + "objective/train/weight_avg": 1.0024504661560059, + "objective/train/weighted_lm_loss": 3.1195900440216064, + "objective/train/weights_max": 2.150031328201294, + "objective/train/weights_min": 0.19796407222747803, + "theoretical_loss": 3.5991784078243345, + "tokens_seen": 1158348800 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003278134403209629, + "loss": 1.4724, + "theoretical_loss": 3.5991784078243345, + "tokens_seen": 1158348800 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032780341023069213, + "loss": 1.3416, + "theoretical_loss": 3.5991595525624955, + "tokens_seen": 1158414336 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032779338014042125, + "loss": 1.3784, + "theoretical_loss": 3.5991406986660026, + "tokens_seen": 1158479872 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003277833500501505, + "loss": 1.4719, + "theoretical_loss": 3.5991218461346803, + "tokens_seen": 1158545408 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003277733199598796, + "loss": 1.4751, + "theoretical_loss": 3.5991029949683515, + "tokens_seen": 1158610944 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032776328986960885, + "loss": 1.3313, + "theoretical_loss": 3.5990841451668416, + "tokens_seen": 1158676480 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032775325977933803, + "loss": 1.3104, + "theoretical_loss": 3.5990652967299734, + "tokens_seen": 1158742016 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003277432296890672, + "loss": 1.4127, + "theoretical_loss": 3.5990464496575716, + "tokens_seen": 1158807552 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003277331995987964, + "loss": 1.4652, + "theoretical_loss": 3.5990276039494598, + "tokens_seen": 1158873088 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003277231695085256, + "loss": 1.4315, + "theoretical_loss": 3.5990087596054625, + "tokens_seen": 1158938624 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032771313941825476, + "loss": 1.385, + "theoretical_loss": 3.5989899166254036, + "tokens_seen": 1159004160 + }, + { + "epoch": 0.35, + "learning_rate": 0.000327703109327984, + "loss": 1.3672, + "theoretical_loss": 3.598971075009107, + "tokens_seen": 1159069696 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003276930792377131, + "loss": 1.2266, + "theoretical_loss": 3.5989522347563976, + "tokens_seen": 1159135232 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032768304914744235, + "loss": 1.5678, + "theoretical_loss": 3.598933395867099, + "tokens_seen": 1159200768 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003276730190571715, + "loss": 1.3902, + "theoretical_loss": 3.598914558341036, + "tokens_seen": 1159266304 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003276629889669007, + "loss": 1.3812, + "theoretical_loss": 3.5988957221780318, + "tokens_seen": 1159331840 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003276529588766299, + "loss": 1.3314, + "theoretical_loss": 3.5988768873779122, + "tokens_seen": 1159397376 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003276429287863591, + "loss": 1.4338, + "theoretical_loss": 3.598858053940501, + "tokens_seen": 1159462912 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032763289869608826, + "loss": 1.2994, + "theoretical_loss": 3.598839221865622, + "tokens_seen": 1159528448 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003276228686058175, + "loss": 1.4752, + "theoretical_loss": 3.5988203911531005, + "tokens_seen": 1159593984 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003276128385155466, + "loss": 1.3603, + "theoretical_loss": 3.5988015618027602, + "tokens_seen": 1159659520 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032760280842527586, + "loss": 1.3006, + "theoretical_loss": 3.598782733814426, + "tokens_seen": 1159725056 + }, + { + "epoch": 0.35, + "learning_rate": 0.000327592778335005, + "loss": 1.3065, + "theoretical_loss": 3.5987639071879225, + "tokens_seen": 1159790592 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003275827482447342, + "loss": 1.4743, + "theoretical_loss": 3.5987450819230737, + "tokens_seen": 1159856128 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003275727181544634, + "loss": 1.4928, + "theoretical_loss": 3.5987262580197052, + "tokens_seen": 1159921664 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.0018644656520336866, + "objective/train/docs_used": 660623, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5480599403381348, + "objective/train/original_loss": 2.5480594635009766, + "objective/train/theoretical_loss": 3.5987074354776407, + "objective/train/tokens_used": 1180447200, + "objective/train/value_avg": -0.005096435546875, + "objective/train/value_loss": 6.45980762783438e-05, + "objective/train/value_max": -0.0002434253692626953, + "objective/train/value_min": -0.044586181640625, + "objective/train/value_reward_corr": 0.35708294771635535, + "objective/train/value_std": 0.003643035888671875, + "objective/train/weight_avg": 0.9844179153442383, + "objective/train/weighted_lm_loss": 2.5239710807800293, + "objective/train/weights_max": 1.5020027160644531, + "objective/train/weights_min": 0.503894031047821, + "theoretical_loss": 3.5987074354776407, + "tokens_seen": 1159987200 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003275626880641926, + "loss": 1.4182, + "theoretical_loss": 3.5987074354776407, + "tokens_seen": 1159987200 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032755265797392176, + "loss": 1.3114, + "theoretical_loss": 3.598688614296705, + "tokens_seen": 1160052736 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032754262788365094, + "loss": 1.2691, + "theoretical_loss": 3.598669794476723, + "tokens_seen": 1160118272 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003275325977933801, + "loss": 1.3664, + "theoretical_loss": 3.59865097601752, + "tokens_seen": 1160183808 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032752256770310936, + "loss": 1.3523, + "theoretical_loss": 3.598632158918919, + "tokens_seen": 1160249344 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003275125376128385, + "loss": 1.4652, + "theoretical_loss": 3.598613343180747, + "tokens_seen": 1160314880 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003275025075225677, + "loss": 1.3852, + "theoretical_loss": 3.598594528802827, + "tokens_seen": 1160380416 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032749247743229685, + "loss": 1.3647, + "theoretical_loss": 3.5985757157849854, + "tokens_seen": 1160445952 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003274824473420261, + "loss": 1.3713, + "theoretical_loss": 3.5985569041270455, + "tokens_seen": 1160511488 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032747241725175527, + "loss": 1.3637, + "theoretical_loss": 3.598538093828833, + "tokens_seen": 1160577024 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032746238716148445, + "loss": 1.2744, + "theoretical_loss": 3.598519284890173, + "tokens_seen": 1160642560 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032745235707121363, + "loss": 1.3468, + "theoretical_loss": 3.59850047731089, + "tokens_seen": 1160708096 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032744232698094286, + "loss": 1.3107, + "theoretical_loss": 3.598481671090809, + "tokens_seen": 1160773632 + }, + { + "epoch": 0.35, + "learning_rate": 0.000327432296890672, + "loss": 1.2939, + "theoretical_loss": 3.598462866229756, + "tokens_seen": 1160839168 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003274222668004012, + "loss": 1.361, + "theoretical_loss": 3.5984440627275553, + "tokens_seen": 1160904704 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032741223671013035, + "loss": 1.3413, + "theoretical_loss": 3.5984252605840323, + "tokens_seen": 1160970240 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003274022066198596, + "loss": 1.3411, + "theoretical_loss": 3.598406459799011, + "tokens_seen": 1161035776 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032739217652958877, + "loss": 1.392, + "theoretical_loss": 3.5983876603723184, + "tokens_seen": 1161101312 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032738214643931795, + "loss": 1.3908, + "theoretical_loss": 3.5983688623037784, + "tokens_seen": 1161166848 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032737211634904713, + "loss": 1.4185, + "theoretical_loss": 3.598350065593217, + "tokens_seen": 1161232384 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003273620862587763, + "loss": 1.2988, + "theoretical_loss": 3.598331270240459, + "tokens_seen": 1161297920 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003273520561685055, + "loss": 1.3512, + "theoretical_loss": 3.5983124762453293, + "tokens_seen": 1161363456 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032734202607823473, + "loss": 1.3646, + "theoretical_loss": 3.5982936836076544, + "tokens_seen": 1161428992 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032733199598796386, + "loss": 1.4086, + "theoretical_loss": 3.598274892327259, + "tokens_seen": 1161494528 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003273219658976931, + "loss": 1.3844, + "theoretical_loss": 3.598256102403968, + "tokens_seen": 1161560064 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.0004881534550804645, + "objective/train/docs_used": 661405, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0849344730377197, + "objective/train/original_loss": 3.084934711456299, + "objective/train/theoretical_loss": 3.5982373138376076, + "objective/train/tokens_used": 1182085600, + "objective/train/value_avg": -0.005809783935546875, + "objective/train/value_loss": 7.364273915300146e-05, + "objective/train/value_max": -0.0003199577331542969, + "objective/train/value_min": -0.056976318359375, + "objective/train/value_reward_corr": 0.39367965151069395, + "objective/train/value_std": 0.00479888916015625, + "objective/train/weight_avg": 0.9982894659042358, + "objective/train/weighted_lm_loss": 3.078730344772339, + "objective/train/weights_max": 1.5866732597351074, + "objective/train/weights_min": 0.03728518262505531, + "theoretical_loss": 3.5982373138376076, + "tokens_seen": 1161625600 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032731193580742233, + "loss": 1.3985, + "theoretical_loss": 3.5982373138376076, + "tokens_seen": 1161625600 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032730190571715145, + "loss": 1.2933, + "theoretical_loss": 3.5982185266280036, + "tokens_seen": 1161691136 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003272918756268807, + "loss": 1.4906, + "theoretical_loss": 3.59819974077498, + "tokens_seen": 1161756672 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003272818455366098, + "loss": 1.3421, + "theoretical_loss": 3.5981809562783633, + "tokens_seen": 1161822208 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032727181544633905, + "loss": 1.3723, + "theoretical_loss": 3.5981621731379794, + "tokens_seen": 1161887744 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032726178535606823, + "loss": 1.2678, + "theoretical_loss": 3.5981433913536534, + "tokens_seen": 1161953280 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003272517552657974, + "loss": 1.3629, + "theoretical_loss": 3.598124610925211, + "tokens_seen": 1162018816 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003272417251755266, + "loss": 1.3405, + "theoretical_loss": 3.598105831852478, + "tokens_seen": 1162084352 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003272316950852558, + "loss": 1.3653, + "theoretical_loss": 3.59808705413528, + "tokens_seen": 1162149888 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032722166499498496, + "loss": 1.2015, + "theoretical_loss": 3.598068277773442, + "tokens_seen": 1162215424 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003272116349047142, + "loss": 1.4009, + "theoretical_loss": 3.5980495027667905, + "tokens_seen": 1162280960 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003272016048144433, + "loss": 1.4003, + "theoretical_loss": 3.598030729115152, + "tokens_seen": 1162346496 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032719157472417256, + "loss": 1.4534, + "theoretical_loss": 3.5980119568183504, + "tokens_seen": 1162412032 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003271815446339017, + "loss": 1.3432, + "theoretical_loss": 3.5979931858762138, + "tokens_seen": 1162477568 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003271715145436309, + "loss": 1.3452, + "theoretical_loss": 3.597974416288566, + "tokens_seen": 1162543104 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003271614844533601, + "loss": 1.3511, + "theoretical_loss": 3.597955648055234, + "tokens_seen": 1162608640 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003271514543630893, + "loss": 1.3552, + "theoretical_loss": 3.5979368811760444, + "tokens_seen": 1162674176 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032714142427281846, + "loss": 1.3318, + "theoretical_loss": 3.5979181156508218, + "tokens_seen": 1162739712 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003271313941825477, + "loss": 1.3218, + "theoretical_loss": 3.5978993514793927, + "tokens_seen": 1162805248 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003271213640922768, + "loss": 1.3009, + "theoretical_loss": 3.5978805886615834, + "tokens_seen": 1162870784 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032711133400200606, + "loss": 1.4779, + "theoretical_loss": 3.59786182719722, + "tokens_seen": 1162936320 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003271013039117352, + "loss": 1.448, + "theoretical_loss": 3.597843067086128, + "tokens_seen": 1163001856 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003270912738214644, + "loss": 1.2938, + "theoretical_loss": 3.597824308328134, + "tokens_seen": 1163067392 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003270812437311936, + "loss": 1.3664, + "theoretical_loss": 3.5978055509230646, + "tokens_seen": 1163132928 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003270712136409228, + "loss": 1.3983, + "theoretical_loss": 3.597786794870745, + "tokens_seen": 1163198464 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.00017359502089675516, + "objective/train/docs_used": 662532, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6463186740875244, + "objective/train/original_loss": 2.6463184356689453, + "objective/train/theoretical_loss": 3.597768040171002, + "objective/train/tokens_used": 1183724000, + "objective/train/value_avg": -0.00640106201171875, + "objective/train/value_loss": 5.926802987232804e-05, + "objective/train/value_max": -0.00033926963806152344, + "objective/train/value_min": -0.060211181640625, + "objective/train/value_reward_corr": 0.4513778102648909, + "objective/train/value_std": 0.005352020263671875, + "objective/train/weight_avg": 1.0010753870010376, + "objective/train/weighted_lm_loss": 2.646998167037964, + "objective/train/weights_max": 1.504722237586975, + "objective/train/weights_min": 0.38897931575775146, + "theoretical_loss": 3.597768040171002, + "tokens_seen": 1163264000 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032706118355065196, + "loss": 1.3525, + "theoretical_loss": 3.597768040171002, + "tokens_seen": 1163264000 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032705115346038114, + "loss": 1.3122, + "theoretical_loss": 3.597749286823662, + "tokens_seen": 1163329536 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003270411233701103, + "loss": 1.3206, + "theoretical_loss": 3.5977305348285515, + "tokens_seen": 1163395072 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032703109327983956, + "loss": 1.3634, + "theoretical_loss": 3.5977117841854955, + "tokens_seen": 1163460608 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003270210631895687, + "loss": 1.3167, + "theoretical_loss": 3.597693034894322, + "tokens_seen": 1163526144 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003270110330992979, + "loss": 1.351, + "theoretical_loss": 3.597674286954857, + "tokens_seen": 1163591680 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032700100300902705, + "loss": 1.2367, + "theoretical_loss": 3.597655540366926, + "tokens_seen": 1163657216 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003269909729187563, + "loss": 1.3359, + "theoretical_loss": 3.5976367951303563, + "tokens_seen": 1163722752 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032698094282848547, + "loss": 1.3476, + "theoretical_loss": 3.5976180512449742, + "tokens_seen": 1163788288 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032697091273821465, + "loss": 1.3571, + "theoretical_loss": 3.597599308710606, + "tokens_seen": 1163853824 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032696088264794383, + "loss": 1.4729, + "theoretical_loss": 3.5975805675270784, + "tokens_seen": 1163919360 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032695085255767306, + "loss": 1.3815, + "theoretical_loss": 3.5975618276942187, + "tokens_seen": 1163984896 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003269408224674022, + "loss": 1.4045, + "theoretical_loss": 3.597543089211853, + "tokens_seen": 1164050432 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003269307923771314, + "loss": 1.3983, + "theoretical_loss": 3.597524352079807, + "tokens_seen": 1164115968 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032692076228686055, + "loss": 1.3798, + "theoretical_loss": 3.5975056162979087, + "tokens_seen": 1164181504 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003269107321965898, + "loss": 1.3242, + "theoretical_loss": 3.5974868818659838, + "tokens_seen": 1164247040 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032690070210631897, + "loss": 1.2521, + "theoretical_loss": 3.59746814878386, + "tokens_seen": 1164312576 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032689067201604815, + "loss": 1.4498, + "theoretical_loss": 3.5974494170513642, + "tokens_seen": 1164378112 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032688064192577733, + "loss": 1.3315, + "theoretical_loss": 3.597430686668322, + "tokens_seen": 1164443648 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003268706118355065, + "loss": 1.2722, + "theoretical_loss": 3.5974119576345607, + "tokens_seen": 1164509184 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003268605817452357, + "loss": 1.2956, + "theoretical_loss": 3.597393229949908, + "tokens_seen": 1164574720 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032685055165496493, + "loss": 1.4263, + "theoretical_loss": 3.5973745036141898, + "tokens_seen": 1164640256 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032684052156469406, + "loss": 1.4195, + "theoretical_loss": 3.597355778627233, + "tokens_seen": 1164705792 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003268304914744233, + "loss": 1.2959, + "theoretical_loss": 3.597337054988866, + "tokens_seen": 1164771328 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003268204613841524, + "loss": 1.4051, + "theoretical_loss": 3.597318332698914, + "tokens_seen": 1164836864 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.0011905499268323183, + "objective/train/docs_used": 663185, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7065377235412598, + "objective/train/original_loss": 2.7065370082855225, + "objective/train/theoretical_loss": 3.5972996117572054, + "objective/train/tokens_used": 1185362400, + "objective/train/value_avg": -0.00542449951171875, + "objective/train/value_loss": 8.23230147943832e-05, + "objective/train/value_max": -0.00043892860412597656, + "objective/train/value_min": -0.07666015625, + "objective/train/value_reward_corr": 0.410607915195585, + "objective/train/value_std": 0.004619598388671875, + "objective/train/weight_avg": 0.9919357895851135, + "objective/train/weighted_lm_loss": 2.699019432067871, + "objective/train/weights_max": 1.8422025442123413, + "objective/train/weights_min": 0.3559424579143524, + "theoretical_loss": 3.5972996117572054, + "tokens_seen": 1164902400 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032681043129388165, + "loss": 1.3609, + "theoretical_loss": 3.5972996117572054, + "tokens_seen": 1164902400 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032680040120361084, + "loss": 1.4106, + "theoretical_loss": 3.5972808921635666, + "tokens_seen": 1164967936 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032679037111334, + "loss": 1.372, + "theoretical_loss": 3.5972621739178248, + "tokens_seen": 1165033472 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003267803410230692, + "loss": 1.3683, + "theoretical_loss": 3.5972434570198066, + "tokens_seen": 1165099008 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032677031093279843, + "loss": 1.3786, + "theoretical_loss": 3.5972247414693403, + "tokens_seen": 1165164544 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032676028084252756, + "loss": 1.3307, + "theoretical_loss": 3.5972060272662523, + "tokens_seen": 1165230080 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003267502507522568, + "loss": 1.2882, + "theoretical_loss": 3.59718731441037, + "tokens_seen": 1165295616 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003267402206619859, + "loss": 1.4292, + "theoretical_loss": 3.5971686029015206, + "tokens_seen": 1165361152 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032673019057171516, + "loss": 1.299, + "theoretical_loss": 3.597149892739532, + "tokens_seen": 1165426688 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032672016048144434, + "loss": 1.4622, + "theoretical_loss": 3.5971311839242306, + "tokens_seen": 1165492224 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003267101303911735, + "loss": 1.3953, + "theoretical_loss": 3.597112476455444, + "tokens_seen": 1165557760 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003267001003009027, + "loss": 1.2974, + "theoretical_loss": 3.5970937703329997, + "tokens_seen": 1165623296 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003266900702106319, + "loss": 1.3306, + "theoretical_loss": 3.5970750655567256, + "tokens_seen": 1165688832 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032668004012036106, + "loss": 1.3843, + "theoretical_loss": 3.597056362126448, + "tokens_seen": 1165754368 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003266700100300903, + "loss": 1.3252, + "theoretical_loss": 3.5970376600419955, + "tokens_seen": 1165819904 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003266599799398194, + "loss": 1.3696, + "theoretical_loss": 3.5970189593031954, + "tokens_seen": 1165885440 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032664994984954866, + "loss": 1.3688, + "theoretical_loss": 3.597000259909875, + "tokens_seen": 1165950976 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032663991975927784, + "loss": 1.4595, + "theoretical_loss": 3.5969815618618615, + "tokens_seen": 1166016512 + }, + { + "epoch": 0.35, + "learning_rate": 0.000326629889669007, + "loss": 1.3274, + "theoretical_loss": 3.596962865158983, + "tokens_seen": 1166082048 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003266198595787362, + "loss": 1.3378, + "theoretical_loss": 3.5969441698010676, + "tokens_seen": 1166147584 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003266098294884654, + "loss": 1.3971, + "theoretical_loss": 3.596925475787942, + "tokens_seen": 1166213120 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032659979939819457, + "loss": 1.3051, + "theoretical_loss": 3.596906783119435, + "tokens_seen": 1166278656 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003265897693079238, + "loss": 1.3666, + "theoretical_loss": 3.5968880917953725, + "tokens_seen": 1166344192 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032657973921765293, + "loss": 1.4222, + "theoretical_loss": 3.5968694018155842, + "tokens_seen": 1166409728 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032656970912738216, + "loss": 1.3153, + "theoretical_loss": 3.5968507131798972, + "tokens_seen": 1166475264 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.001287700142711401, + "objective/train/docs_used": 664585, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8250439167022705, + "objective/train/original_loss": 2.8250441551208496, + "objective/train/theoretical_loss": 3.5968320258881388, + "objective/train/tokens_used": 1187000800, + "objective/train/value_avg": -0.0055389404296875, + "objective/train/value_loss": 0.00012194130977150053, + "objective/train/value_max": -0.0005526542663574219, + "objective/train/value_min": -0.1229248046875, + "objective/train/value_reward_corr": 0.5775641110314857, + "objective/train/value_std": 0.0057220458984375, + "objective/train/weight_avg": 0.9925577640533447, + "objective/train/weighted_lm_loss": 2.8049376010894775, + "objective/train/weights_max": 2.7757325172424316, + "objective/train/weights_min": 0.32143282890319824, + "theoretical_loss": 3.5968320258881388, + "tokens_seen": 1166540800 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032655967903711135, + "loss": 1.4839, + "theoretical_loss": 3.5968320258881388, + "tokens_seen": 1166540800 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003265496489468405, + "loss": 1.2247, + "theoretical_loss": 3.5968133399401374, + "tokens_seen": 1166606336 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032653961885656976, + "loss": 1.3396, + "theoretical_loss": 3.5967946553357213, + "tokens_seen": 1166671872 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003265295887662989, + "loss": 1.3958, + "theoretical_loss": 3.5967759720747177, + "tokens_seen": 1166737408 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003265195586760281, + "loss": 1.4094, + "theoretical_loss": 3.5967572901569547, + "tokens_seen": 1166802944 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032650952858575725, + "loss": 1.3147, + "theoretical_loss": 3.5967386095822604, + "tokens_seen": 1166868480 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003264994984954865, + "loss": 1.2232, + "theoretical_loss": 3.5967199303504627, + "tokens_seen": 1166934016 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032648946840521567, + "loss": 1.3512, + "theoretical_loss": 3.59670125246139, + "tokens_seen": 1166999552 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032647943831494485, + "loss": 1.3308, + "theoretical_loss": 3.5966825759148704, + "tokens_seen": 1167065088 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032646940822467403, + "loss": 1.2709, + "theoretical_loss": 3.5966639007107313, + "tokens_seen": 1167130624 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032645937813440326, + "loss": 1.4373, + "theoretical_loss": 3.5966452268488016, + "tokens_seen": 1167196160 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003264493480441324, + "loss": 1.413, + "theoretical_loss": 3.5966265543289095, + "tokens_seen": 1167261696 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003264393179538616, + "loss": 1.3148, + "theoretical_loss": 3.5966078831508828, + "tokens_seen": 1167327232 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032642928786359075, + "loss": 1.4125, + "theoretical_loss": 3.5965892133145494, + "tokens_seen": 1167392768 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032641925777332, + "loss": 1.4512, + "theoretical_loss": 3.5965705448197385, + "tokens_seen": 1167458304 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032640922768304917, + "loss": 1.3636, + "theoretical_loss": 3.5965518776662773, + "tokens_seen": 1167523840 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032639919759277835, + "loss": 1.3464, + "theoretical_loss": 3.596533211853995, + "tokens_seen": 1167589376 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032638916750250753, + "loss": 1.3477, + "theoretical_loss": 3.5965145473827205, + "tokens_seen": 1167654912 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003263791374122367, + "loss": 1.2472, + "theoretical_loss": 3.5964958842522803, + "tokens_seen": 1167720448 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003263691073219659, + "loss": 1.3909, + "theoretical_loss": 3.5964772224625046, + "tokens_seen": 1167785984 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032635907723169513, + "loss": 1.369, + "theoretical_loss": 3.5964585620132214, + "tokens_seen": 1167851520 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032634904714142426, + "loss": 1.379, + "theoretical_loss": 3.596439902904259, + "tokens_seen": 1167917056 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003263390170511535, + "loss": 1.4297, + "theoretical_loss": 3.5964212451354456, + "tokens_seen": 1167982592 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003263289869608826, + "loss": 1.3171, + "theoretical_loss": 3.59640258870661, + "tokens_seen": 1168048128 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032631895687061185, + "loss": 1.405, + "theoretical_loss": 3.5963839336175814, + "tokens_seen": 1168113664 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.001782007864676416, + "objective/train/docs_used": 665171, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.468254804611206, + "objective/train/original_loss": 2.468254804611206, + "objective/train/theoretical_loss": 3.5963652798681878, + "objective/train/tokens_used": 1188639200, + "objective/train/value_avg": -0.00530242919921875, + "objective/train/value_loss": 0.00010541149822529405, + "objective/train/value_max": -0.0004992485046386719, + "objective/train/value_min": -0.03656005859375, + "objective/train/value_reward_corr": 0.19929565618017556, + "objective/train/value_std": 0.0035228729248046875, + "objective/train/weight_avg": 0.986802875995636, + "objective/train/weighted_lm_loss": 2.449761152267456, + "objective/train/weights_max": 1.441379427909851, + "objective/train/weights_min": 0.08398383855819702, + "theoretical_loss": 3.5963652798681878, + "tokens_seen": 1168179200 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032630892678034104, + "loss": 1.3801, + "theoretical_loss": 3.5963652798681878, + "tokens_seen": 1168179200 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003262988966900702, + "loss": 1.2586, + "theoretical_loss": 3.596346627458258, + "tokens_seen": 1168244736 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003262888665997994, + "loss": 1.3741, + "theoretical_loss": 3.5963279763876206, + "tokens_seen": 1168310272 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032627883650952863, + "loss": 1.4191, + "theoretical_loss": 3.596309326656104, + "tokens_seen": 1168375808 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032626880641925776, + "loss": 1.516, + "theoretical_loss": 3.596290678263538, + "tokens_seen": 1168441344 + }, + { + "epoch": 0.35, + "learning_rate": 0.000326258776328987, + "loss": 1.4245, + "theoretical_loss": 3.5962720312097503, + "tokens_seen": 1168506880 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003262487462387161, + "loss": 1.3634, + "theoretical_loss": 3.5962533854945704, + "tokens_seen": 1168572416 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032623871614844536, + "loss": 1.4176, + "theoretical_loss": 3.5962347411178266, + "tokens_seen": 1168637952 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032622868605817454, + "loss": 1.421, + "theoretical_loss": 3.5962160980793483, + "tokens_seen": 1168703488 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003262186559679037, + "loss": 1.379, + "theoretical_loss": 3.596197456378964, + "tokens_seen": 1168769024 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003262086258776329, + "loss": 1.33, + "theoretical_loss": 3.5961788160165025, + "tokens_seen": 1168834560 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003261985957873621, + "loss": 1.443, + "theoretical_loss": 3.596160176991794, + "tokens_seen": 1168900096 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032618856569709126, + "loss": 1.4624, + "theoretical_loss": 3.596141539304665, + "tokens_seen": 1168965632 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003261785356068205, + "loss": 1.3557, + "theoretical_loss": 3.5961229029549475, + "tokens_seen": 1169031168 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003261685055165496, + "loss": 1.4086, + "theoretical_loss": 3.5961042679424686, + "tokens_seen": 1169096704 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032615847542627886, + "loss": 1.2995, + "theoretical_loss": 3.596085634267058, + "tokens_seen": 1169162240 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032614844533600804, + "loss": 1.3123, + "theoretical_loss": 3.596067001928545, + "tokens_seen": 1169227776 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003261384152457372, + "loss": 1.2305, + "theoretical_loss": 3.596048370926758, + "tokens_seen": 1169293312 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003261283851554664, + "loss": 1.3616, + "theoretical_loss": 3.5960297412615265, + "tokens_seen": 1169358848 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003261183550651956, + "loss": 1.4353, + "theoretical_loss": 3.596011112932681, + "tokens_seen": 1169424384 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032610832497492477, + "loss": 1.3424, + "theoretical_loss": 3.5959924859400485, + "tokens_seen": 1169489920 + }, + { + "epoch": 0.35, + "learning_rate": 0.000326098294884654, + "loss": 1.3207, + "theoretical_loss": 3.5959738602834603, + "tokens_seen": 1169555456 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032608826479438313, + "loss": 1.4489, + "theoretical_loss": 3.5959552359627445, + "tokens_seen": 1169620992 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032607823470411236, + "loss": 1.3748, + "theoretical_loss": 3.595936612977731, + "tokens_seen": 1169686528 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003260682046138415, + "loss": 1.4058, + "theoretical_loss": 3.595917991328249, + "tokens_seen": 1169752064 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": 0.0001723209279589355, + "objective/train/docs_used": 666471, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.284627914428711, + "objective/train/original_loss": 2.284627914428711, + "objective/train/theoretical_loss": 3.595899371014127, + "objective/train/tokens_used": 1190277600, + "objective/train/value_avg": -0.00638580322265625, + "objective/train/value_loss": 7.126013224478811e-05, + "objective/train/value_max": -0.0003101825714111328, + "objective/train/value_min": -0.1341552734375, + "objective/train/value_reward_corr": 0.45947650678337504, + "objective/train/value_std": 0.007656097412109375, + "objective/train/weight_avg": 1.005240559577942, + "objective/train/weighted_lm_loss": 2.306692123413086, + "objective/train/weights_max": 3.1631550788879395, + "objective/train/weights_min": 0.4250081181526184, + "theoretical_loss": 3.595899371014127, + "tokens_seen": 1169817600 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003260581745235707, + "loss": 1.3693, + "theoretical_loss": 3.595899371014127, + "tokens_seen": 1169817600 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003260481444332999, + "loss": 1.4714, + "theoretical_loss": 3.5958807520351965, + "tokens_seen": 1169883136 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003260381143430291, + "loss": 1.2699, + "theoretical_loss": 3.595862134391285, + "tokens_seen": 1169948672 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032602808425275827, + "loss": 1.4369, + "theoretical_loss": 3.5958435180822237, + "tokens_seen": 1170014208 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032601805416248745, + "loss": 1.4825, + "theoretical_loss": 3.595824903107841, + "tokens_seen": 1170079744 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032600802407221663, + "loss": 1.4104, + "theoretical_loss": 3.5958062894679665, + "tokens_seen": 1170145280 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032599799398194587, + "loss": 1.4099, + "theoretical_loss": 3.5957876771624298, + "tokens_seen": 1170210816 + }, + { + "epoch": 0.35, + "learning_rate": 0.000325987963891675, + "loss": 1.3241, + "theoretical_loss": 3.5957690661910613, + "tokens_seen": 1170276352 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032597793380140423, + "loss": 1.4327, + "theoretical_loss": 3.5957504565536893, + "tokens_seen": 1170341888 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003259679037111334, + "loss": 1.3534, + "theoretical_loss": 3.5957318482501446, + "tokens_seen": 1170407424 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003259578736208626, + "loss": 1.372, + "theoretical_loss": 3.595713241280257, + "tokens_seen": 1170472960 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003259478435305918, + "loss": 1.33, + "theoretical_loss": 3.5956946356438553, + "tokens_seen": 1170538496 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032593781344032095, + "loss": 1.3443, + "theoretical_loss": 3.59567603134077, + "tokens_seen": 1170604032 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032592778335005014, + "loss": 1.3543, + "theoretical_loss": 3.595657428370831, + "tokens_seen": 1170669568 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032591775325977937, + "loss": 1.3661, + "theoretical_loss": 3.5956388267338673, + "tokens_seen": 1170735104 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003259077231695085, + "loss": 1.3528, + "theoretical_loss": 3.59562022642971, + "tokens_seen": 1170800640 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032589769307923773, + "loss": 1.3708, + "theoretical_loss": 3.5956016274581883, + "tokens_seen": 1170866176 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032588766298896686, + "loss": 1.3925, + "theoretical_loss": 3.595583029819132, + "tokens_seen": 1170931712 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003258776328986961, + "loss": 1.4194, + "theoretical_loss": 3.5955644335123713, + "tokens_seen": 1170997248 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003258676028084253, + "loss": 1.3877, + "theoretical_loss": 3.5955458385377366, + "tokens_seen": 1171062784 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032585757271815446, + "loss": 1.3646, + "theoretical_loss": 3.595527244895057, + "tokens_seen": 1171128320 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032584754262788364, + "loss": 1.3427, + "theoretical_loss": 3.5955086525841633, + "tokens_seen": 1171193856 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003258375125376128, + "loss": 1.39, + "theoretical_loss": 3.5954900616048855, + "tokens_seen": 1171259392 + }, + { + "epoch": 0.35, + "learning_rate": 0.000325827482447342, + "loss": 1.2172, + "theoretical_loss": 3.5954714719570537, + "tokens_seen": 1171324928 + }, + { + "epoch": 0.35, + "learning_rate": 0.00032581745235707124, + "loss": 1.2949, + "theoretical_loss": 3.5954528836404975, + "tokens_seen": 1171390464 + }, + { + "epoch": 0.35, + "objective/train/advantage_avg": -0.0010374359553679824, + "objective/train/docs_used": 666886, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5635147094726562, + "objective/train/original_loss": 2.5635147094726562, + "objective/train/theoretical_loss": 3.595434296655048, + "objective/train/tokens_used": 1191916000, + "objective/train/value_avg": -0.005214691162109375, + "objective/train/value_loss": 7.216428639367223e-05, + "objective/train/value_max": -0.0003418922424316406, + "objective/train/value_min": -0.09979248046875, + "objective/train/value_reward_corr": 0.46361782980116273, + "objective/train/value_std": 0.00449371337890625, + "objective/train/weight_avg": 0.9929002523422241, + "objective/train/weighted_lm_loss": 2.5464835166931152, + "objective/train/weights_max": 1.7921932935714722, + "objective/train/weights_min": 0.3003483712673187, + "theoretical_loss": 3.595434296655048, + "tokens_seen": 1171456000 + }, + { + "epoch": 0.35, + "learning_rate": 0.0003258074222668004, + "loss": 1.4627, + "theoretical_loss": 3.595434296655048, + "tokens_seen": 1171456000 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003257973921765296, + "loss": 1.4314, + "theoretical_loss": 3.595415711000535, + "tokens_seen": 1171521536 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032578736208625883, + "loss": 1.3992, + "theoretical_loss": 3.5953971266767883, + "tokens_seen": 1171587072 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032577733199598796, + "loss": 1.4502, + "theoretical_loss": 3.5953785436836387, + "tokens_seen": 1171652608 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003257673019057172, + "loss": 1.3946, + "theoretical_loss": 3.5953599620209165, + "tokens_seen": 1171718144 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003257572718154463, + "loss": 1.3616, + "theoretical_loss": 3.5953413816884523, + "tokens_seen": 1171783680 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032574724172517556, + "loss": 1.4142, + "theoretical_loss": 3.5953228026860757, + "tokens_seen": 1171849216 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032573721163490474, + "loss": 1.4086, + "theoretical_loss": 3.595304225013618, + "tokens_seen": 1171914752 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003257271815446339, + "loss": 1.4065, + "theoretical_loss": 3.5952856486709086, + "tokens_seen": 1171980288 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003257171514543631, + "loss": 1.4133, + "theoretical_loss": 3.595267073657779, + "tokens_seen": 1172045824 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003257071213640923, + "loss": 1.3848, + "theoretical_loss": 3.5952484999740593, + "tokens_seen": 1172111360 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032569709127382146, + "loss": 1.3743, + "theoretical_loss": 3.59522992761958, + "tokens_seen": 1172176896 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003256870611835507, + "loss": 1.4117, + "theoretical_loss": 3.5952113565941715, + "tokens_seen": 1172242432 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003256770310932798, + "loss": 1.3196, + "theoretical_loss": 3.5951927868976643, + "tokens_seen": 1172307968 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032566700100300906, + "loss": 1.357, + "theoretical_loss": 3.59517421852989, + "tokens_seen": 1172373504 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032565697091273824, + "loss": 1.4147, + "theoretical_loss": 3.595155651490678, + "tokens_seen": 1172439040 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003256469408224674, + "loss": 1.4513, + "theoretical_loss": 3.5951370857798595, + "tokens_seen": 1172504576 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003256369107321966, + "loss": 1.2848, + "theoretical_loss": 3.595118521397265, + "tokens_seen": 1172570112 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003256268806419258, + "loss": 1.368, + "theoretical_loss": 3.5950999583427263, + "tokens_seen": 1172635648 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032561685055165497, + "loss": 1.4795, + "theoretical_loss": 3.595081396616073, + "tokens_seen": 1172701184 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003256068204613842, + "loss": 1.38, + "theoretical_loss": 3.595062836217136, + "tokens_seen": 1172766720 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032559679037111333, + "loss": 1.3872, + "theoretical_loss": 3.5950442771457465, + "tokens_seen": 1172832256 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032558676028084256, + "loss": 1.3517, + "theoretical_loss": 3.595025719401735, + "tokens_seen": 1172897792 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003255767301905717, + "loss": 1.4698, + "theoretical_loss": 3.595007162984934, + "tokens_seen": 1172963328 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003255667001003009, + "loss": 1.2571, + "theoretical_loss": 3.5949886078951714, + "tokens_seen": 1173028864 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": 0.00037965402589179575, + "objective/train/docs_used": 667905, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9378128051757812, + "objective/train/original_loss": 2.937812566757202, + "objective/train/theoretical_loss": 3.594970054132281, + "objective/train/tokens_used": 1193554400, + "objective/train/value_avg": -0.005298614501953125, + "objective/train/value_loss": 4.50032894150354e-05, + "objective/train/value_max": -0.0003829002380371094, + "objective/train/value_min": -0.0567626953125, + "objective/train/value_reward_corr": 0.38725249157870406, + "objective/train/value_std": 0.0038738250732421875, + "objective/train/weight_avg": 1.0059894323349, + "objective/train/weighted_lm_loss": 2.950241804122925, + "objective/train/weights_max": 1.526960015296936, + "objective/train/weights_min": 0.593850314617157, + "theoretical_loss": 3.594970054132281, + "tokens_seen": 1173094400 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003255566700100301, + "loss": 1.4912, + "theoretical_loss": 3.594970054132281, + "tokens_seen": 1173094400 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003255466399197593, + "loss": 1.4012, + "theoretical_loss": 3.5949515016960922, + "tokens_seen": 1173159936 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032553660982948847, + "loss": 1.4728, + "theoretical_loss": 3.594932950586437, + "tokens_seen": 1173225472 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032552657973921765, + "loss": 1.4289, + "theoretical_loss": 3.5949144008031455, + "tokens_seen": 1173291008 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032551654964894683, + "loss": 1.3988, + "theoretical_loss": 3.5948958523460495, + "tokens_seen": 1173356544 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032550651955867607, + "loss": 1.4853, + "theoretical_loss": 3.59487730521498, + "tokens_seen": 1173422080 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003254964894684052, + "loss": 1.2072, + "theoretical_loss": 3.5948587594097674, + "tokens_seen": 1173487616 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032548645937813443, + "loss": 1.451, + "theoretical_loss": 3.594840214930244, + "tokens_seen": 1173553152 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003254764292878636, + "loss": 1.3756, + "theoretical_loss": 3.5948216717762405, + "tokens_seen": 1173618688 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003254663991975928, + "loss": 1.3414, + "theoretical_loss": 3.5948031299475884, + "tokens_seen": 1173684224 + }, + { + "epoch": 0.36, + "learning_rate": 0.000325456369107322, + "loss": 1.3926, + "theoretical_loss": 3.5947845894441186, + "tokens_seen": 1173749760 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032544633901705115, + "loss": 1.263, + "theoretical_loss": 3.5947660502656626, + "tokens_seen": 1173815296 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032543630892678034, + "loss": 1.401, + "theoretical_loss": 3.5947475124120514, + "tokens_seen": 1173880832 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032542627883650957, + "loss": 1.4144, + "theoretical_loss": 3.594728975883117, + "tokens_seen": 1173946368 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003254162487462387, + "loss": 1.2598, + "theoretical_loss": 3.5947104406786905, + "tokens_seen": 1174011904 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032540621865596793, + "loss": 1.3591, + "theoretical_loss": 3.594691906798603, + "tokens_seen": 1174077440 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032539618856569706, + "loss": 1.4199, + "theoretical_loss": 3.5946733742426864, + "tokens_seen": 1174142976 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003253861584754263, + "loss": 1.3596, + "theoretical_loss": 3.5946548430107716, + "tokens_seen": 1174208512 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003253761283851555, + "loss": 1.3659, + "theoretical_loss": 3.5946363131026917, + "tokens_seen": 1174274048 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032536609829488466, + "loss": 1.4367, + "theoretical_loss": 3.594617784518276, + "tokens_seen": 1174339584 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032535606820461384, + "loss": 1.4489, + "theoretical_loss": 3.5945992572573577, + "tokens_seen": 1174405120 + }, + { + "epoch": 0.36, + "learning_rate": 0.000325346038114343, + "loss": 1.4008, + "theoretical_loss": 3.594580731319768, + "tokens_seen": 1174470656 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003253360080240722, + "loss": 1.3709, + "theoretical_loss": 3.594562206705338, + "tokens_seen": 1174536192 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032532597793380144, + "loss": 1.2847, + "theoretical_loss": 3.5945436834139004, + "tokens_seen": 1174601728 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032531594784353056, + "loss": 1.3788, + "theoretical_loss": 3.594525161445286, + "tokens_seen": 1174667264 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": 0.0002698350581340492, + "objective/train/docs_used": 668590, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0860893726348877, + "objective/train/original_loss": 3.086089611053467, + "objective/train/theoretical_loss": 3.594506640799327, + "objective/train/tokens_used": 1195192800, + "objective/train/value_avg": -0.005336761474609375, + "objective/train/value_loss": 6.888589996378869e-05, + "objective/train/value_max": -0.00031495094299316406, + "objective/train/value_min": -0.06085205078125, + "objective/train/value_reward_corr": 0.33798480746091264, + "objective/train/value_std": 0.003997802734375, + "objective/train/weight_avg": 1.0059144496917725, + "objective/train/weighted_lm_loss": 3.1092820167541504, + "objective/train/weights_max": 1.6892623901367188, + "objective/train/weights_min": 0.38774967193603516, + "theoretical_loss": 3.594506640799327, + "tokens_seen": 1174732800 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003253059177532598, + "loss": 1.4546, + "theoretical_loss": 3.594506640799327, + "tokens_seen": 1174732800 + }, + { + "epoch": 0.36, + "learning_rate": 0.000325295887662989, + "loss": 1.3853, + "theoretical_loss": 3.5944881214758553, + "tokens_seen": 1174798336 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032528585757271816, + "loss": 1.2925, + "theoretical_loss": 3.594469603474702, + "tokens_seen": 1174863872 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032527582748244734, + "loss": 1.3031, + "theoretical_loss": 3.5944510867957, + "tokens_seen": 1174929408 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003252657973921765, + "loss": 1.3845, + "theoretical_loss": 3.59443257143868, + "tokens_seen": 1174994944 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003252557673019057, + "loss": 1.3663, + "theoretical_loss": 3.5944140574034744, + "tokens_seen": 1175060480 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032524573721163494, + "loss": 1.3123, + "theoretical_loss": 3.594395544689915, + "tokens_seen": 1175126016 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032523570712136407, + "loss": 1.1818, + "theoretical_loss": 3.5943770332978344, + "tokens_seen": 1175191552 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003252256770310933, + "loss": 1.4174, + "theoretical_loss": 3.594358523227064, + "tokens_seen": 1175257088 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032521564694082243, + "loss": 1.5679, + "theoretical_loss": 3.5943400144774356, + "tokens_seen": 1175322624 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032520561685055166, + "loss": 1.355, + "theoretical_loss": 3.5943215070487815, + "tokens_seen": 1175388160 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032519558676028084, + "loss": 1.3956, + "theoretical_loss": 3.5943030009409345, + "tokens_seen": 1175453696 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032518555667001, + "loss": 1.333, + "theoretical_loss": 3.5942844961537257, + "tokens_seen": 1175519232 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003251755265797392, + "loss": 1.3503, + "theoretical_loss": 3.5942659926869873, + "tokens_seen": 1175584768 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032516549648946844, + "loss": 1.4436, + "theoretical_loss": 3.5942474905405524, + "tokens_seen": 1175650304 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032515546639919757, + "loss": 1.3356, + "theoretical_loss": 3.594228989714252, + "tokens_seen": 1175715840 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003251454363089268, + "loss": 1.4211, + "theoretical_loss": 3.594210490207919, + "tokens_seen": 1175781376 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032513540621865593, + "loss": 1.3153, + "theoretical_loss": 3.5941919920213854, + "tokens_seen": 1175846912 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032512537612838517, + "loss": 1.3832, + "theoretical_loss": 3.5941734951544837, + "tokens_seen": 1175912448 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032511534603811435, + "loss": 1.2563, + "theoretical_loss": 3.5941549996070465, + "tokens_seen": 1175977984 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032510531594784353, + "loss": 1.321, + "theoretical_loss": 3.594136505378905, + "tokens_seen": 1176043520 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003250952858575727, + "loss": 1.275, + "theoretical_loss": 3.594118012469893, + "tokens_seen": 1176109056 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003250852557673019, + "loss": 1.4545, + "theoretical_loss": 3.5940995208798423, + "tokens_seen": 1176174592 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032507522567703107, + "loss": 1.3298, + "theoretical_loss": 3.594081030608585, + "tokens_seen": 1176240128 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003250651955867603, + "loss": 1.4378, + "theoretical_loss": 3.5940625416559544, + "tokens_seen": 1176305664 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -2.4973545805551112e-05, + "objective/train/docs_used": 669752, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.66440749168396, + "objective/train/original_loss": 2.66440749168396, + "objective/train/theoretical_loss": 3.594044054021782, + "objective/train/tokens_used": 1196831200, + "objective/train/value_avg": -0.00534820556640625, + "objective/train/value_loss": 4.636498124455102e-05, + "objective/train/value_max": -0.0004513263702392578, + "objective/train/value_min": -0.0775146484375, + "objective/train/value_reward_corr": 0.2836789434699826, + "objective/train/value_std": 0.004180908203125, + "objective/train/weight_avg": 1.0020581483840942, + "objective/train/weighted_lm_loss": 2.6815690994262695, + "objective/train/weights_max": 2.170910120010376, + "objective/train/weights_min": 0.6616348028182983, + "theoretical_loss": 3.594044054021782, + "tokens_seen": 1176371200 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003250551654964895, + "loss": 1.2761, + "theoretical_loss": 3.594044054021782, + "tokens_seen": 1176371200 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032504513540621867, + "loss": 1.5141, + "theoretical_loss": 3.594025567705901, + "tokens_seen": 1176436736 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032503510531594785, + "loss": 1.3317, + "theoretical_loss": 3.5940070827081443, + "tokens_seen": 1176502272 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032502507522567703, + "loss": 1.4506, + "theoretical_loss": 3.5939885990283438, + "tokens_seen": 1176567808 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032501504513540627, + "loss": 1.4654, + "theoretical_loss": 3.5939701166663323, + "tokens_seen": 1176633344 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003250050150451354, + "loss": 1.4885, + "theoretical_loss": 3.5939516356219423, + "tokens_seen": 1176698880 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032499498495486463, + "loss": 1.3675, + "theoretical_loss": 3.593933155895007, + "tokens_seen": 1176764416 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003249849548645938, + "loss": 1.3692, + "theoretical_loss": 3.5939146774853583, + "tokens_seen": 1176829952 + }, + { + "epoch": 0.36, + "learning_rate": 0.000324974924774323, + "loss": 1.3667, + "theoretical_loss": 3.59389620039283, + "tokens_seen": 1176895488 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003249648946840522, + "loss": 1.2712, + "theoretical_loss": 3.593877724617254, + "tokens_seen": 1176961024 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032495486459378135, + "loss": 1.4818, + "theoretical_loss": 3.5938592501584634, + "tokens_seen": 1177026560 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032494483450351054, + "loss": 1.3417, + "theoretical_loss": 3.5938407770162915, + "tokens_seen": 1177092096 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032493480441323977, + "loss": 1.4024, + "theoretical_loss": 3.5938223051905704, + "tokens_seen": 1177157632 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003249247743229689, + "loss": 1.4426, + "theoretical_loss": 3.5938038346811334, + "tokens_seen": 1177223168 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032491474423269813, + "loss": 1.4059, + "theoretical_loss": 3.593785365487814, + "tokens_seen": 1177288704 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032490471414242726, + "loss": 1.3827, + "theoretical_loss": 3.5937668976104438, + "tokens_seen": 1177354240 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003248946840521565, + "loss": 1.4318, + "theoretical_loss": 3.5937484310488568, + "tokens_seen": 1177419776 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003248846539618857, + "loss": 1.4557, + "theoretical_loss": 3.5937299658028863, + "tokens_seen": 1177485312 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032487462387161486, + "loss": 1.4435, + "theoretical_loss": 3.593711501872364, + "tokens_seen": 1177550848 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032486459378134404, + "loss": 1.4328, + "theoretical_loss": 3.5936930392571242, + "tokens_seen": 1177616384 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003248545636910732, + "loss": 1.3982, + "theoretical_loss": 3.5936745779569996, + "tokens_seen": 1177681920 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003248445336008024, + "loss": 1.3687, + "theoretical_loss": 3.5936561179718236, + "tokens_seen": 1177747456 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032483450351053164, + "loss": 1.2337, + "theoretical_loss": 3.5936376593014288, + "tokens_seen": 1177812992 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032482447342026076, + "loss": 1.285, + "theoretical_loss": 3.593619201945649, + "tokens_seen": 1177878528 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032481444332999, + "loss": 1.2998, + "theoretical_loss": 3.5936007459043173, + "tokens_seen": 1177944064 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -0.0003846451290883124, + "objective/train/docs_used": 670392, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7903778553009033, + "objective/train/original_loss": 2.7903778553009033, + "objective/train/theoretical_loss": 3.593582291177267, + "objective/train/tokens_used": 1198469600, + "objective/train/value_avg": -0.00478363037109375, + "objective/train/value_loss": 5.5318832892226055e-05, + "objective/train/value_max": -0.0001780986785888672, + "objective/train/value_min": -0.0526123046875, + "objective/train/value_reward_corr": 0.37497548498043476, + "objective/train/value_std": 0.00386810302734375, + "objective/train/weight_avg": 0.9987320303916931, + "objective/train/weighted_lm_loss": 2.788949966430664, + "objective/train/weights_max": 1.692358374595642, + "objective/train/weights_min": 0.3116583228111267, + "theoretical_loss": 3.593582291177267, + "tokens_seen": 1178009600 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003248044132397192, + "loss": 1.4373, + "theoretical_loss": 3.593582291177267, + "tokens_seen": 1178009600 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032479438314944836, + "loss": 1.4884, + "theoretical_loss": 3.5935638377643313, + "tokens_seen": 1178075136 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032478435305917754, + "loss": 1.384, + "theoretical_loss": 3.593545385665343, + "tokens_seen": 1178140672 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003247743229689067, + "loss": 1.3801, + "theoretical_loss": 3.593526934880136, + "tokens_seen": 1178206208 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003247642928786359, + "loss": 1.3771, + "theoretical_loss": 3.5935084854085444, + "tokens_seen": 1178271744 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032475426278836514, + "loss": 1.3608, + "theoretical_loss": 3.5934900372504, + "tokens_seen": 1178337280 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032474423269809427, + "loss": 1.3788, + "theoretical_loss": 3.593471590405538, + "tokens_seen": 1178402816 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003247342026078235, + "loss": 1.2717, + "theoretical_loss": 3.5934531448737905, + "tokens_seen": 1178468352 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032472417251755263, + "loss": 1.3838, + "theoretical_loss": 3.593434700654991, + "tokens_seen": 1178533888 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032471414242728186, + "loss": 1.4094, + "theoretical_loss": 3.5934162577489746, + "tokens_seen": 1178599424 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032470411233701104, + "loss": 1.3967, + "theoretical_loss": 3.5933978161555733, + "tokens_seen": 1178664960 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003246940822467402, + "loss": 1.3541, + "theoretical_loss": 3.5933793758746213, + "tokens_seen": 1178730496 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003246840521564694, + "loss": 1.3618, + "theoretical_loss": 3.5933609369059525, + "tokens_seen": 1178796032 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032467402206619864, + "loss": 1.3578, + "theoretical_loss": 3.5933424992494, + "tokens_seen": 1178861568 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032466399197592777, + "loss": 1.3997, + "theoretical_loss": 3.5933240629047973, + "tokens_seen": 1178927104 + }, + { + "epoch": 0.36, + "learning_rate": 0.000324653961885657, + "loss": 1.4361, + "theoretical_loss": 3.5933056278719793, + "tokens_seen": 1178992640 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032464393179538613, + "loss": 1.26, + "theoretical_loss": 3.5932871941507782, + "tokens_seen": 1179058176 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032463390170511537, + "loss": 1.4192, + "theoretical_loss": 3.593268761741029, + "tokens_seen": 1179123712 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032462387161484455, + "loss": 1.2375, + "theoretical_loss": 3.593250330642565, + "tokens_seen": 1179189248 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032461384152457373, + "loss": 1.4109, + "theoretical_loss": 3.5932319008552205, + "tokens_seen": 1179254784 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003246038114343029, + "loss": 1.3403, + "theoretical_loss": 3.593213472378829, + "tokens_seen": 1179320320 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003245937813440321, + "loss": 1.4227, + "theoretical_loss": 3.593195045213224, + "tokens_seen": 1179385856 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032458375125376127, + "loss": 1.4241, + "theoretical_loss": 3.59317661935824, + "tokens_seen": 1179451392 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003245737211634905, + "loss": 1.4446, + "theoretical_loss": 3.593158194813711, + "tokens_seen": 1179516928 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032456369107321963, + "loss": 1.3731, + "theoretical_loss": 3.593139771579471, + "tokens_seen": 1179582464 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -0.0001004342339001596, + "objective/train/docs_used": 671618, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9555225372314453, + "objective/train/original_loss": 2.9555225372314453, + "objective/train/theoretical_loss": 3.5931213496553536, + "objective/train/tokens_used": 1200108000, + "objective/train/value_avg": -0.00490570068359375, + "objective/train/value_loss": 6.358182145049796e-05, + "objective/train/value_max": -0.00042724609375, + "objective/train/value_min": -0.042816162109375, + "objective/train/value_reward_corr": 0.34805768055117914, + "objective/train/value_std": 0.003475189208984375, + "objective/train/weight_avg": 1.0018205642700195, + "objective/train/weighted_lm_loss": 2.9654858112335205, + "objective/train/weights_max": 1.352529764175415, + "objective/train/weights_min": 0.1422213315963745, + "theoretical_loss": 3.5931213496553536, + "tokens_seen": 1179648000 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032455366098294887, + "loss": 1.4728, + "theoretical_loss": 3.5931213496553536, + "tokens_seen": 1179648000 + }, + { + "epoch": 0.36, + "learning_rate": 0.000324543630892678, + "loss": 1.4422, + "theoretical_loss": 3.5931029290411933, + "tokens_seen": 1179713536 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032453360080240723, + "loss": 1.3729, + "theoretical_loss": 3.5930845097368236, + "tokens_seen": 1179779072 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003245235707121364, + "loss": 1.3755, + "theoretical_loss": 3.593066091742079, + "tokens_seen": 1179844608 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003245135406218656, + "loss": 1.289, + "theoretical_loss": 3.593047675056794, + "tokens_seen": 1179910144 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003245035105315948, + "loss": 1.4707, + "theoretical_loss": 3.5930292596808027, + "tokens_seen": 1179975680 + }, + { + "epoch": 0.36, + "learning_rate": 0.000324493480441324, + "loss": 1.3363, + "theoretical_loss": 3.5930108456139385, + "tokens_seen": 1180041216 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032448345035105314, + "loss": 1.3895, + "theoretical_loss": 3.592992432856037, + "tokens_seen": 1180106752 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003244734202607824, + "loss": 1.4024, + "theoretical_loss": 3.5929740214069312, + "tokens_seen": 1180172288 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003244633901705115, + "loss": 1.2989, + "theoretical_loss": 3.5929556112664556, + "tokens_seen": 1180237824 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032445336008024074, + "loss": 1.3331, + "theoretical_loss": 3.592937202434445, + "tokens_seen": 1180303360 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003244433299899699, + "loss": 1.4414, + "theoretical_loss": 3.5929187949107333, + "tokens_seen": 1180368896 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003244332998996991, + "loss": 1.3584, + "theoretical_loss": 3.5929003886951554, + "tokens_seen": 1180434432 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003244232698094283, + "loss": 1.4699, + "theoretical_loss": 3.5928819837875454, + "tokens_seen": 1180499968 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032441323971915746, + "loss": 1.5493, + "theoretical_loss": 3.5928635801877378, + "tokens_seen": 1180565504 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032440320962888664, + "loss": 1.3076, + "theoretical_loss": 3.5928451778955672, + "tokens_seen": 1180631040 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003243931795386159, + "loss": 1.311, + "theoretical_loss": 3.5928267769108677, + "tokens_seen": 1180696576 + }, + { + "epoch": 0.36, + "learning_rate": 0.000324383149448345, + "loss": 1.3391, + "theoretical_loss": 3.5928083772334745, + "tokens_seen": 1180762112 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032437311935807424, + "loss": 1.4211, + "theoretical_loss": 3.5927899788632214, + "tokens_seen": 1180827648 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032436308926780337, + "loss": 1.2952, + "theoretical_loss": 3.592771581799944, + "tokens_seen": 1180893184 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003243530591775326, + "loss": 1.3681, + "theoretical_loss": 3.592753186043476, + "tokens_seen": 1180958720 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003243430290872618, + "loss": 1.1977, + "theoretical_loss": 3.592734791593652, + "tokens_seen": 1181024256 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032433299899699096, + "loss": 1.3123, + "theoretical_loss": 3.5927163984503077, + "tokens_seen": 1181089792 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003243229689067202, + "loss": 1.3815, + "theoretical_loss": 3.592698006613277, + "tokens_seen": 1181155328 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003243129388164494, + "loss": 1.2547, + "theoretical_loss": 3.592679616082395, + "tokens_seen": 1181220864 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": 0.0011996534885838628, + "objective/train/docs_used": 672229, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.863023519515991, + "objective/train/original_loss": 2.863023519515991, + "objective/train/theoretical_loss": 3.592661226857496, + "objective/train/tokens_used": 1201746400, + "objective/train/value_avg": -0.0056610107421875, + "objective/train/value_loss": 5.508285539690405e-05, + "objective/train/value_max": -0.0002868175506591797, + "objective/train/value_min": -0.0753173828125, + "objective/train/value_reward_corr": 0.3006876750896302, + "objective/train/value_std": 0.00518035888671875, + "objective/train/weight_avg": 1.0147347450256348, + "objective/train/weighted_lm_loss": 2.8963208198547363, + "objective/train/weights_max": 1.934386134147644, + "objective/train/weights_min": 0.29483702778816223, + "theoretical_loss": 3.592661226857496, + "tokens_seen": 1181286400 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032430290872617856, + "loss": 1.3782, + "theoretical_loss": 3.592661226857496, + "tokens_seen": 1181286400 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032429287863590774, + "loss": 1.4006, + "theoretical_loss": 3.5926428389384153, + "tokens_seen": 1181351936 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003242828485456369, + "loss": 1.3329, + "theoretical_loss": 3.5926244523249875, + "tokens_seen": 1181417472 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003242728184553661, + "loss": 1.3271, + "theoretical_loss": 3.592606067017048, + "tokens_seen": 1181483008 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032426278836509534, + "loss": 1.3586, + "theoretical_loss": 3.5925876830144308, + "tokens_seen": 1181548544 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032425275827482447, + "loss": 1.3834, + "theoretical_loss": 3.5925693003169714, + "tokens_seen": 1181614080 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003242427281845537, + "loss": 1.4634, + "theoretical_loss": 3.5925509189245046, + "tokens_seen": 1181679616 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032423269809428283, + "loss": 1.5089, + "theoretical_loss": 3.5925325388368656, + "tokens_seen": 1181745152 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032422266800401206, + "loss": 1.3279, + "theoretical_loss": 3.5925141600538892, + "tokens_seen": 1181810688 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032421263791374125, + "loss": 1.4139, + "theoretical_loss": 3.5924957825754102, + "tokens_seen": 1181876224 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003242026078234704, + "loss": 1.265, + "theoretical_loss": 3.592477406401265, + "tokens_seen": 1181941760 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003241925777331996, + "loss": 1.4294, + "theoretical_loss": 3.5924590315312868, + "tokens_seen": 1182007296 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032418254764292884, + "loss": 1.4106, + "theoretical_loss": 3.5924406579653114, + "tokens_seen": 1182072832 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032417251755265797, + "loss": 1.2446, + "theoretical_loss": 3.592422285703175, + "tokens_seen": 1182138368 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003241624874623872, + "loss": 1.3837, + "theoretical_loss": 3.592403914744712, + "tokens_seen": 1182203904 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032415245737211633, + "loss": 1.3454, + "theoretical_loss": 3.592385545089757, + "tokens_seen": 1182269440 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032414242728184557, + "loss": 1.3023, + "theoretical_loss": 3.592367176738146, + "tokens_seen": 1182334976 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032413239719157475, + "loss": 1.4415, + "theoretical_loss": 3.592348809689715, + "tokens_seen": 1182400512 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032412236710130393, + "loss": 1.3218, + "theoretical_loss": 3.592330443944298, + "tokens_seen": 1182466048 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003241123370110331, + "loss": 1.4388, + "theoretical_loss": 3.5923120795017307, + "tokens_seen": 1182531584 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003241023069207623, + "loss": 1.415, + "theoretical_loss": 3.592293716361849, + "tokens_seen": 1182597120 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032409227683049147, + "loss": 1.337, + "theoretical_loss": 3.592275354524488, + "tokens_seen": 1182662656 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003240822467402207, + "loss": 1.376, + "theoretical_loss": 3.5922569939894826, + "tokens_seen": 1182728192 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032407221664994984, + "loss": 1.3999, + "theoretical_loss": 3.5922386347566695, + "tokens_seen": 1182793728 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032406218655967907, + "loss": 1.3836, + "theoretical_loss": 3.592220276825883, + "tokens_seen": 1182859264 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -0.000576832506339997, + "objective/train/docs_used": 673021, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6801369190216064, + "objective/train/original_loss": 2.6801366806030273, + "objective/train/theoretical_loss": 3.592201920196959, + "objective/train/tokens_used": 1203384800, + "objective/train/value_avg": -0.0065155029296875, + "objective/train/value_loss": 9.334627247881144e-05, + "objective/train/value_max": -0.0003669261932373047, + "objective/train/value_min": -0.04840087890625, + "objective/train/value_reward_corr": 0.2030456170358733, + "objective/train/value_std": 0.00478363037109375, + "objective/train/weight_avg": 0.9984043836593628, + "objective/train/weighted_lm_loss": 2.686634063720703, + "objective/train/weights_max": 1.5776441097259521, + "objective/train/weights_min": 0.15672121942043304, + "theoretical_loss": 3.592201920196959, + "tokens_seen": 1182924800 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003240521564694082, + "loss": 1.3592, + "theoretical_loss": 3.592201920196959, + "tokens_seen": 1182924800 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032404212637913743, + "loss": 1.4157, + "theoretical_loss": 3.5921835648697336, + "tokens_seen": 1182990336 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003240320962888666, + "loss": 1.4581, + "theoretical_loss": 3.5921652108440414, + "tokens_seen": 1183055872 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003240220661985958, + "loss": 1.3468, + "theoretical_loss": 3.592146858119719, + "tokens_seen": 1183121408 + }, + { + "epoch": 0.36, + "learning_rate": 0.000324012036108325, + "loss": 1.5419, + "theoretical_loss": 3.5921285066966018, + "tokens_seen": 1183186944 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003240020060180542, + "loss": 1.403, + "theoretical_loss": 3.592110156574525, + "tokens_seen": 1183252480 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032399197592778334, + "loss": 1.3623, + "theoretical_loss": 3.5920918077533246, + "tokens_seen": 1183318016 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003239819458375126, + "loss": 1.3581, + "theoretical_loss": 3.5920734602328364, + "tokens_seen": 1183383552 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003239719157472417, + "loss": 1.4651, + "theoretical_loss": 3.5920551140128962, + "tokens_seen": 1183449088 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032396188565697094, + "loss": 1.4213, + "theoretical_loss": 3.5920367690933395, + "tokens_seen": 1183514624 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003239518555667001, + "loss": 1.3513, + "theoretical_loss": 3.5920184254740026, + "tokens_seen": 1183580160 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003239418254764293, + "loss": 1.3104, + "theoretical_loss": 3.592000083154721, + "tokens_seen": 1183645696 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003239317953861585, + "loss": 1.3945, + "theoretical_loss": 3.5919817421353306, + "tokens_seen": 1183711232 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032392176529588766, + "loss": 1.3613, + "theoretical_loss": 3.5919634024156677, + "tokens_seen": 1183776768 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032391173520561684, + "loss": 1.3005, + "theoretical_loss": 3.591945063995568, + "tokens_seen": 1183842304 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003239017051153461, + "loss": 1.3374, + "theoretical_loss": 3.591926726874867, + "tokens_seen": 1183907840 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003238916750250752, + "loss": 1.3292, + "theoretical_loss": 3.5919083910534013, + "tokens_seen": 1183973376 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032388164493480444, + "loss": 1.3646, + "theoretical_loss": 3.591890056531007, + "tokens_seen": 1184038912 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032387161484453357, + "loss": 1.3498, + "theoretical_loss": 3.59187172330752, + "tokens_seen": 1184104448 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003238615847542628, + "loss": 1.3812, + "theoretical_loss": 3.5918533913827764, + "tokens_seen": 1184169984 + }, + { + "epoch": 0.36, + "learning_rate": 0.000323851554663992, + "loss": 1.4027, + "theoretical_loss": 3.591835060756612, + "tokens_seen": 1184235520 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032384152457372116, + "loss": 1.2486, + "theoretical_loss": 3.591816731428863, + "tokens_seen": 1184301056 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032383149448345034, + "loss": 1.4758, + "theoretical_loss": 3.591798403399366, + "tokens_seen": 1184366592 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003238214643931796, + "loss": 1.3449, + "theoretical_loss": 3.5917800766679573, + "tokens_seen": 1184432128 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003238114343029087, + "loss": 1.2845, + "theoretical_loss": 3.591761751234473, + "tokens_seen": 1184497664 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -0.00019509036792442203, + "objective/train/docs_used": 674157, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.512732982635498, + "objective/train/original_loss": 2.512733221054077, + "objective/train/theoretical_loss": 3.5917434270987485, + "objective/train/tokens_used": 1205023200, + "objective/train/value_avg": -0.005702972412109375, + "objective/train/value_loss": 7.331513916142285e-05, + "objective/train/value_max": -0.0002512931823730469, + "objective/train/value_min": -0.1807861328125, + "objective/train/value_reward_corr": 0.48347049395117264, + "objective/train/value_std": 0.00598907470703125, + "objective/train/weight_avg": 1.0016297101974487, + "objective/train/weighted_lm_loss": 2.51010799407959, + "objective/train/weights_max": 5.397360801696777, + "objective/train/weights_min": 0.27381956577301025, + "theoretical_loss": 3.5917434270987485, + "tokens_seen": 1184563200 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032380140421263794, + "loss": 1.401, + "theoretical_loss": 3.5917434270987485, + "tokens_seen": 1184563200 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032379137412236707, + "loss": 1.3655, + "theoretical_loss": 3.5917251042606217, + "tokens_seen": 1184628736 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003237813440320963, + "loss": 1.4143, + "theoretical_loss": 3.5917067827199274, + "tokens_seen": 1184694272 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003237713139418255, + "loss": 1.3768, + "theoretical_loss": 3.5916884624765033, + "tokens_seen": 1184759808 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032376128385155467, + "loss": 1.3923, + "theoretical_loss": 3.5916701435301848, + "tokens_seen": 1184825344 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032375125376128385, + "loss": 1.3425, + "theoretical_loss": 3.591651825880809, + "tokens_seen": 1184890880 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032374122367101303, + "loss": 1.2288, + "theoretical_loss": 3.5916335095282115, + "tokens_seen": 1184956416 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003237311935807422, + "loss": 1.3153, + "theoretical_loss": 3.5916151944722294, + "tokens_seen": 1185021952 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032372116349047145, + "loss": 1.3635, + "theoretical_loss": 3.5915968807126997, + "tokens_seen": 1185087488 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032371113340020057, + "loss": 1.241, + "theoretical_loss": 3.591578568249458, + "tokens_seen": 1185153024 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003237011033099298, + "loss": 1.419, + "theoretical_loss": 3.5915602570823415, + "tokens_seen": 1185218560 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032369107321965893, + "loss": 1.3615, + "theoretical_loss": 3.5915419472111862, + "tokens_seen": 1185284096 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032368104312938817, + "loss": 1.3018, + "theoretical_loss": 3.591523638635829, + "tokens_seen": 1185349632 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032367101303911735, + "loss": 1.3709, + "theoretical_loss": 3.5915053313561067, + "tokens_seen": 1185415168 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032366098294884653, + "loss": 1.3656, + "theoretical_loss": 3.591487025371856, + "tokens_seen": 1185480704 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003236509528585757, + "loss": 1.3777, + "theoretical_loss": 3.5914687206829137, + "tokens_seen": 1185546240 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032364092276830495, + "loss": 1.3818, + "theoretical_loss": 3.591450417289116, + "tokens_seen": 1185611776 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003236308926780341, + "loss": 1.3032, + "theoretical_loss": 3.5914321151903006, + "tokens_seen": 1185677312 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003236208625877633, + "loss": 1.4319, + "theoretical_loss": 3.5914138143863035, + "tokens_seen": 1185742848 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032361083249749244, + "loss": 1.372, + "theoretical_loss": 3.591395514876962, + "tokens_seen": 1185808384 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003236008024072217, + "loss": 1.3032, + "theoretical_loss": 3.5913772166621123, + "tokens_seen": 1185873920 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032359077231695085, + "loss": 1.4321, + "theoretical_loss": 3.591358919741592, + "tokens_seen": 1185939456 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032358074222668004, + "loss": 1.3965, + "theoretical_loss": 3.5913406241152384, + "tokens_seen": 1186004992 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032357071213640927, + "loss": 1.274, + "theoretical_loss": 3.591322329782887, + "tokens_seen": 1186070528 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003235606820461384, + "loss": 1.3987, + "theoretical_loss": 3.5913040367443765, + "tokens_seen": 1186136064 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -0.00020640002912841737, + "objective/train/docs_used": 674881, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.500990867614746, + "objective/train/original_loss": 2.500990629196167, + "objective/train/theoretical_loss": 3.591285744999542, + "objective/train/tokens_used": 1206661600, + "objective/train/value_avg": -0.00485992431640625, + "objective/train/value_loss": 4.4719912693835795e-05, + "objective/train/value_max": -0.000415802001953125, + "objective/train/value_min": -0.055511474609375, + "objective/train/value_reward_corr": 0.384577642192397, + "objective/train/value_std": 0.0037364959716796875, + "objective/train/weight_avg": 1.0000710487365723, + "objective/train/weighted_lm_loss": 2.5116584300994873, + "objective/train/weights_max": 1.551596760749817, + "objective/train/weights_min": 0.41689106822013855, + "theoretical_loss": 3.591285744999542, + "tokens_seen": 1186201600 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032355065195586763, + "loss": 1.3097, + "theoretical_loss": 3.591285744999542, + "tokens_seen": 1186201600 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003235406218655968, + "loss": 1.4994, + "theoretical_loss": 3.5912674545482224, + "tokens_seen": 1186267136 + }, + { + "epoch": 0.36, + "learning_rate": 0.000323530591775326, + "loss": 1.3778, + "theoretical_loss": 3.591249165390254, + "tokens_seen": 1186332672 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003235205616850552, + "loss": 1.3861, + "theoretical_loss": 3.591230877525474, + "tokens_seen": 1186398208 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003235105315947844, + "loss": 1.3929, + "theoretical_loss": 3.591212590953719, + "tokens_seen": 1186463744 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032350050150451354, + "loss": 1.4029, + "theoretical_loss": 3.5911943056748266, + "tokens_seen": 1186529280 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003234904714142428, + "loss": 1.4914, + "theoretical_loss": 3.5911760216886344, + "tokens_seen": 1186594816 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003234804413239719, + "loss": 1.2393, + "theoretical_loss": 3.591157738994979, + "tokens_seen": 1186660352 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032347041123370114, + "loss": 1.3343, + "theoretical_loss": 3.5911394575936977, + "tokens_seen": 1186725888 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003234603811434303, + "loss": 1.3278, + "theoretical_loss": 3.591121177484628, + "tokens_seen": 1186791424 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003234503510531595, + "loss": 1.4434, + "theoretical_loss": 3.5911028986676072, + "tokens_seen": 1186856960 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003234403209628887, + "loss": 1.3859, + "theoretical_loss": 3.591084621142473, + "tokens_seen": 1186922496 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032343029087261786, + "loss": 1.4784, + "theoretical_loss": 3.591066344909062, + "tokens_seen": 1186988032 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032342026078234704, + "loss": 1.4365, + "theoretical_loss": 3.5910480699672114, + "tokens_seen": 1187053568 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003234102306920763, + "loss": 1.2189, + "theoretical_loss": 3.59102979631676, + "tokens_seen": 1187119104 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003234002006018054, + "loss": 1.3441, + "theoretical_loss": 3.591011523957544, + "tokens_seen": 1187184640 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032339017051153464, + "loss": 1.3027, + "theoretical_loss": 3.5909932528894015, + "tokens_seen": 1187250176 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032338014042126377, + "loss": 1.297, + "theoretical_loss": 3.59097498311217, + "tokens_seen": 1187315712 + }, + { + "epoch": 0.36, + "learning_rate": 0.000323370110330993, + "loss": 1.3435, + "theoretical_loss": 3.590956714625687, + "tokens_seen": 1187381248 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003233600802407222, + "loss": 1.403, + "theoretical_loss": 3.5909384474297896, + "tokens_seen": 1187446784 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032335005015045136, + "loss": 1.4374, + "theoretical_loss": 3.590920181524316, + "tokens_seen": 1187512320 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032334002006018054, + "loss": 1.3433, + "theoretical_loss": 3.5909019169091034, + "tokens_seen": 1187577856 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003233299899699098, + "loss": 1.4415, + "theoretical_loss": 3.59088365358399, + "tokens_seen": 1187643392 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003233199598796389, + "loss": 1.2304, + "theoretical_loss": 3.590865391548813, + "tokens_seen": 1187708928 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032330992978936814, + "loss": 1.4645, + "theoretical_loss": 3.59084713080341, + "tokens_seen": 1187774464 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -0.0007370962994173169, + "objective/train/docs_used": 676005, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3279409408569336, + "objective/train/original_loss": 2.3279404640197754, + "objective/train/theoretical_loss": 3.5908288713476195, + "objective/train/tokens_used": 1208300000, + "objective/train/value_avg": -0.004993438720703125, + "objective/train/value_loss": 4.5361925003817305e-05, + "objective/train/value_max": -0.0003669261932373047, + "objective/train/value_min": -0.05072021484375, + "objective/train/value_reward_corr": 0.268579676912337, + "objective/train/value_std": 0.0034656524658203125, + "objective/train/weight_avg": 0.9947816133499146, + "objective/train/weighted_lm_loss": 2.3199543952941895, + "objective/train/weights_max": 1.5430728197097778, + "objective/train/weights_min": 0.4847222566604614, + "theoretical_loss": 3.5908288713476195, + "tokens_seen": 1187840000 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032329989969909727, + "loss": 1.2894, + "theoretical_loss": 3.5908288713476195, + "tokens_seen": 1187840000 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003232898696088265, + "loss": 1.3411, + "theoretical_loss": 3.5908106131812785, + "tokens_seen": 1187905536 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003232798395185557, + "loss": 1.343, + "theoretical_loss": 3.5907923563042257, + "tokens_seen": 1187971072 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032326980942828487, + "loss": 1.4391, + "theoretical_loss": 3.590774100716298, + "tokens_seen": 1188036608 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032325977933801405, + "loss": 1.3868, + "theoretical_loss": 3.5907558464173337, + "tokens_seen": 1188102144 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032324974924774323, + "loss": 1.2777, + "theoretical_loss": 3.5907375934071704, + "tokens_seen": 1188167680 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003232397191574724, + "loss": 1.2856, + "theoretical_loss": 3.590719341685647, + "tokens_seen": 1188233216 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032322968906720165, + "loss": 1.4122, + "theoretical_loss": 3.5907010912526, + "tokens_seen": 1188298752 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032321965897693077, + "loss": 1.4274, + "theoretical_loss": 3.590682842107868, + "tokens_seen": 1188364288 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032320962888666, + "loss": 1.3708, + "theoretical_loss": 3.5906645942512894, + "tokens_seen": 1188429824 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032319959879638913, + "loss": 1.3902, + "theoretical_loss": 3.5906463476827026, + "tokens_seen": 1188495360 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032318956870611837, + "loss": 1.1663, + "theoretical_loss": 3.5906281024019444, + "tokens_seen": 1188560896 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032317953861584755, + "loss": 1.3894, + "theoretical_loss": 3.590609858408854, + "tokens_seen": 1188626432 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032316950852557673, + "loss": 1.4012, + "theoretical_loss": 3.5905916157032682, + "tokens_seen": 1188691968 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003231594784353059, + "loss": 1.3043, + "theoretical_loss": 3.590573374285027, + "tokens_seen": 1188757504 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032314944834503515, + "loss": 1.415, + "theoretical_loss": 3.590555134153967, + "tokens_seen": 1188823040 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003231394182547643, + "loss": 1.4831, + "theoretical_loss": 3.5905368953099277, + "tokens_seen": 1188888576 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003231293881644935, + "loss": 1.3269, + "theoretical_loss": 3.590518657752746, + "tokens_seen": 1188954112 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032311935807422264, + "loss": 1.2853, + "theoretical_loss": 3.590500421482261, + "tokens_seen": 1189019648 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003231093279839519, + "loss": 1.3953, + "theoretical_loss": 3.5904821864983116, + "tokens_seen": 1189085184 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032309929789368105, + "loss": 1.3659, + "theoretical_loss": 3.590463952800735, + "tokens_seen": 1189150720 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032308926780341024, + "loss": 1.295, + "theoretical_loss": 3.5904457203893703, + "tokens_seen": 1189216256 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003230792377131394, + "loss": 1.3285, + "theoretical_loss": 3.590427489264055, + "tokens_seen": 1189281792 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003230692076228686, + "loss": 1.4086, + "theoretical_loss": 3.5904092594246286, + "tokens_seen": 1189347328 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003230591775325978, + "loss": 1.3121, + "theoretical_loss": 3.5903910308709293, + "tokens_seen": 1189412864 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -0.00039260636549443007, + "objective/train/docs_used": 676669, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9086954593658447, + "objective/train/original_loss": 2.9086952209472656, + "objective/train/theoretical_loss": 3.590372803602795, + "objective/train/tokens_used": 1209938400, + "objective/train/value_avg": -0.00530242919921875, + "objective/train/value_loss": 5.79508014197927e-05, + "objective/train/value_max": -0.00026726722717285156, + "objective/train/value_min": -0.064208984375, + "objective/train/value_reward_corr": 0.37247999577868907, + "objective/train/value_std": 0.004486083984375, + "objective/train/weight_avg": 0.9987197518348694, + "objective/train/weighted_lm_loss": 2.907992124557495, + "objective/train/weights_max": 1.7597413063049316, + "objective/train/weights_min": 0.08460766822099686, + "theoretical_loss": 3.590372803602795, + "tokens_seen": 1189478400 + }, + { + "epoch": 0.36, + "learning_rate": 0.000323049147442327, + "loss": 1.3535, + "theoretical_loss": 3.590372803602795, + "tokens_seen": 1189478400 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032303911735205614, + "loss": 1.4019, + "theoretical_loss": 3.5903545776200643, + "tokens_seen": 1189543936 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003230290872617854, + "loss": 1.4168, + "theoretical_loss": 3.5903363529225762, + "tokens_seen": 1189609472 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003230190571715145, + "loss": 1.3012, + "theoretical_loss": 3.5903181295101696, + "tokens_seen": 1189675008 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032300902708124374, + "loss": 1.4816, + "theoretical_loss": 3.5902999073826827, + "tokens_seen": 1189740544 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003229989969909729, + "loss": 1.4791, + "theoretical_loss": 3.590281686539953, + "tokens_seen": 1189806080 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003229889669007021, + "loss": 1.4246, + "theoretical_loss": 3.5902634669818214, + "tokens_seen": 1189871616 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003229789368104313, + "loss": 1.4681, + "theoretical_loss": 3.5902452487081247, + "tokens_seen": 1189937152 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003229689067201605, + "loss": 1.4309, + "theoretical_loss": 3.590227031718702, + "tokens_seen": 1190002688 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032295887662988964, + "loss": 1.3113, + "theoretical_loss": 3.590208816013393, + "tokens_seen": 1190068224 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003229488465396189, + "loss": 1.3734, + "theoretical_loss": 3.5901906015920355, + "tokens_seen": 1190133760 + }, + { + "epoch": 0.36, + "learning_rate": 0.000322938816449348, + "loss": 1.3397, + "theoretical_loss": 3.5901723884544685, + "tokens_seen": 1190199296 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032292878635907724, + "loss": 1.3199, + "theoretical_loss": 3.5901541766005316, + "tokens_seen": 1190264832 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003229187562688064, + "loss": 1.3316, + "theoretical_loss": 3.5901359660300622, + "tokens_seen": 1190330368 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003229087261785356, + "loss": 1.4442, + "theoretical_loss": 3.5901177567429006, + "tokens_seen": 1190395904 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003228986960882648, + "loss": 1.4583, + "theoretical_loss": 3.5900995487388845, + "tokens_seen": 1190461440 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032288866599799397, + "loss": 1.375, + "theoretical_loss": 3.590081342017854, + "tokens_seen": 1190526976 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032287863590772315, + "loss": 1.3273, + "theoretical_loss": 3.5900631365796474, + "tokens_seen": 1190592512 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003228686058174524, + "loss": 1.4076, + "theoretical_loss": 3.590044932424104, + "tokens_seen": 1190658048 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003228585757271815, + "loss": 1.4389, + "theoretical_loss": 3.5900267295510626, + "tokens_seen": 1190723584 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032284854563691074, + "loss": 1.3707, + "theoretical_loss": 3.5900085279603626, + "tokens_seen": 1190789120 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003228385155466399, + "loss": 1.3717, + "theoretical_loss": 3.5899903276518423, + "tokens_seen": 1190854656 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003228284854563691, + "loss": 1.3324, + "theoretical_loss": 3.589972128625342, + "tokens_seen": 1190920192 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032281845536609834, + "loss": 1.4285, + "theoretical_loss": 3.5899539308807, + "tokens_seen": 1190985728 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032280842527582747, + "loss": 1.4117, + "theoretical_loss": 3.5899357344177556, + "tokens_seen": 1191051264 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": 0.0003646821132861078, + "objective/train/docs_used": 677774, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.66536021232605, + "objective/train/original_loss": 2.665360689163208, + "objective/train/theoretical_loss": 3.5899175392363483, + "objective/train/tokens_used": 1211576800, + "objective/train/value_avg": -0.0064697265625, + "objective/train/value_loss": 0.00010640378604875877, + "objective/train/value_max": -0.0003275871276855469, + "objective/train/value_min": -0.0889892578125, + "objective/train/value_reward_corr": 0.3384817621411099, + "objective/train/value_std": 0.00551605224609375, + "objective/train/weight_avg": 1.0082467794418335, + "objective/train/weighted_lm_loss": 2.68105411529541, + "objective/train/weights_max": 2.2998833656311035, + "objective/train/weights_min": 0.2958115041255951, + "theoretical_loss": 3.5899175392363483, + "tokens_seen": 1191116800 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003227983951855567, + "loss": 1.3567, + "theoretical_loss": 3.5899175392363483, + "tokens_seen": 1191116800 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003227883650952859, + "loss": 1.3323, + "theoretical_loss": 3.5898993453363173, + "tokens_seen": 1191182336 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032277833500501507, + "loss": 1.3576, + "theoretical_loss": 3.5898811527175014, + "tokens_seen": 1191247872 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032276830491474425, + "loss": 1.466, + "theoretical_loss": 3.5898629613797404, + "tokens_seen": 1191313408 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032275827482447343, + "loss": 1.3645, + "theoretical_loss": 3.589844771322874, + "tokens_seen": 1191378944 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003227482447342026, + "loss": 1.3035, + "theoretical_loss": 3.58982658254674, + "tokens_seen": 1191444480 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032273821464393185, + "loss": 1.3936, + "theoretical_loss": 3.58980839505118, + "tokens_seen": 1191510016 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032272818455366097, + "loss": 1.4973, + "theoretical_loss": 3.5897902088360314, + "tokens_seen": 1191575552 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003227181544633902, + "loss": 1.3489, + "theoretical_loss": 3.589772023901135, + "tokens_seen": 1191641088 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032270812437311933, + "loss": 1.3812, + "theoretical_loss": 3.5897538402463294, + "tokens_seen": 1191706624 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032269809428284857, + "loss": 1.2947, + "theoretical_loss": 3.5897356578714543, + "tokens_seen": 1191772160 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032268806419257775, + "loss": 1.3882, + "theoretical_loss": 3.5897174767763493, + "tokens_seen": 1191837696 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032267803410230693, + "loss": 1.3854, + "theoretical_loss": 3.5896992969608545, + "tokens_seen": 1191903232 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003226680040120361, + "loss": 1.2899, + "theoretical_loss": 3.589681118424809, + "tokens_seen": 1191968768 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032265797392176535, + "loss": 1.3164, + "theoretical_loss": 3.5896629411680525, + "tokens_seen": 1192034304 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003226479438314945, + "loss": 1.3371, + "theoretical_loss": 3.589644765190424, + "tokens_seen": 1192099840 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003226379137412237, + "loss": 1.4797, + "theoretical_loss": 3.5896265904917644, + "tokens_seen": 1192165376 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032262788365095284, + "loss": 1.35, + "theoretical_loss": 3.5896084170719127, + "tokens_seen": 1192230912 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003226178535606821, + "loss": 1.4248, + "theoretical_loss": 3.589590244930709, + "tokens_seen": 1192296448 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032260782347041125, + "loss": 1.4511, + "theoretical_loss": 3.589572074067992, + "tokens_seen": 1192361984 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032259779338014044, + "loss": 1.4791, + "theoretical_loss": 3.5895539044836027, + "tokens_seen": 1192427520 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003225877632898696, + "loss": 1.3384, + "theoretical_loss": 3.58953573617738, + "tokens_seen": 1192493056 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003225777331995988, + "loss": 1.4123, + "theoretical_loss": 3.589517569149165, + "tokens_seen": 1192558592 + }, + { + "epoch": 0.36, + "learning_rate": 0.000322567703109328, + "loss": 1.3265, + "theoretical_loss": 3.589499403398796, + "tokens_seen": 1192624128 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003225576730190572, + "loss": 1.3442, + "theoretical_loss": 3.5894812389261146, + "tokens_seen": 1192689664 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": 0.00027569744270294905, + "objective/train/docs_used": 678420, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7522456645965576, + "objective/train/original_loss": 2.7522456645965576, + "objective/train/theoretical_loss": 3.589463075730959, + "objective/train/tokens_used": 1213215200, + "objective/train/value_avg": -0.006023406982421875, + "objective/train/value_loss": 5.7324679801240563e-05, + "objective/train/value_max": -0.0004601478576660156, + "objective/train/value_min": -0.072509765625, + "objective/train/value_reward_corr": 0.44525716200830645, + "objective/train/value_std": 0.005596160888671875, + "objective/train/weight_avg": 1.0055209398269653, + "objective/train/weighted_lm_loss": 2.781036615371704, + "objective/train/weights_max": 1.8251111507415771, + "objective/train/weights_min": 0.46444156765937805, + "theoretical_loss": 3.589463075730959, + "tokens_seen": 1192755200 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032254764292878634, + "loss": 1.4747, + "theoretical_loss": 3.589463075730959, + "tokens_seen": 1192755200 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003225376128385156, + "loss": 1.5198, + "theoretical_loss": 3.5894449138131703, + "tokens_seen": 1192820736 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003225275827482447, + "loss": 1.3164, + "theoretical_loss": 3.5894267531725887, + "tokens_seen": 1192886272 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032251755265797394, + "loss": 1.2855, + "theoretical_loss": 3.589408593809053, + "tokens_seen": 1192951808 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003225075225677031, + "loss": 1.476, + "theoretical_loss": 3.5893904357224047, + "tokens_seen": 1193017344 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003224974924774323, + "loss": 1.3374, + "theoretical_loss": 3.5893722789124825, + "tokens_seen": 1193082880 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003224874623871615, + "loss": 1.4724, + "theoretical_loss": 3.5893541233791275, + "tokens_seen": 1193148416 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003224774322968907, + "loss": 1.3462, + "theoretical_loss": 3.589335969122179, + "tokens_seen": 1193213952 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032246740220661984, + "loss": 1.4519, + "theoretical_loss": 3.5893178161414783, + "tokens_seen": 1193279488 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003224573721163491, + "loss": 1.4128, + "theoretical_loss": 3.5892996644368647, + "tokens_seen": 1193345024 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003224473420260782, + "loss": 1.3879, + "theoretical_loss": 3.589281514008179, + "tokens_seen": 1193410560 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032243731193580744, + "loss": 1.419, + "theoretical_loss": 3.5892633648552605, + "tokens_seen": 1193476096 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003224272818455366, + "loss": 1.4107, + "theoretical_loss": 3.5892452169779503, + "tokens_seen": 1193541632 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003224172517552658, + "loss": 1.3048, + "theoretical_loss": 3.589227070376089, + "tokens_seen": 1193607168 + }, + { + "epoch": 0.36, + "learning_rate": 0.000322407221664995, + "loss": 1.4581, + "theoretical_loss": 3.5892089250495163, + "tokens_seen": 1193672704 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032239719157472417, + "loss": 1.3727, + "theoretical_loss": 3.5891907809980723, + "tokens_seen": 1193738240 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032238716148445335, + "loss": 1.4606, + "theoretical_loss": 3.589172638221598, + "tokens_seen": 1193803776 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003223771313941826, + "loss": 1.2813, + "theoretical_loss": 3.589154496719934, + "tokens_seen": 1193869312 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003223671013039117, + "loss": 1.3111, + "theoretical_loss": 3.58913635649292, + "tokens_seen": 1193934848 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032235707121364095, + "loss": 1.3338, + "theoretical_loss": 3.5891182175403973, + "tokens_seen": 1194000384 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003223470411233701, + "loss": 1.3323, + "theoretical_loss": 3.5891000798622055, + "tokens_seen": 1194065920 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003223370110330993, + "loss": 1.4077, + "theoretical_loss": 3.5890819434581855, + "tokens_seen": 1194131456 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003223269809428285, + "loss": 1.3665, + "theoretical_loss": 3.589063808328179, + "tokens_seen": 1194196992 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032231695085255767, + "loss": 1.4325, + "theoretical_loss": 3.5890456744720245, + "tokens_seen": 1194262528 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032230692076228685, + "loss": 1.4085, + "theoretical_loss": 3.589027541889564, + "tokens_seen": 1194328064 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -0.003473278135061264, + "objective/train/docs_used": 679445, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.559112548828125, + "objective/train/original_loss": 2.559112071990967, + "objective/train/theoretical_loss": 3.5890094105806383, + "objective/train/tokens_used": 1214853600, + "objective/train/value_avg": -0.0063629150390625, + "objective/train/value_loss": 0.00010857381130335853, + "objective/train/value_max": -0.00040459632873535156, + "objective/train/value_min": -0.08209228515625, + "objective/train/value_reward_corr": 0.42596821052889217, + "objective/train/value_std": 0.00537109375, + "objective/train/weight_avg": 0.9702932238578796, + "objective/train/weighted_lm_loss": 2.5229432582855225, + "objective/train/weights_max": 2.2725961208343506, + "objective/train/weights_min": 0.13916492462158203, + "theoretical_loss": 3.5890094105806383, + "tokens_seen": 1194393600 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003222968906720161, + "loss": 1.3259, + "theoretical_loss": 3.5890094105806383, + "tokens_seen": 1194393600 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003222868605817452, + "loss": 1.3455, + "theoretical_loss": 3.5889912805450876, + "tokens_seen": 1194459136 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032227683049147445, + "loss": 1.3804, + "theoretical_loss": 3.588973151782752, + "tokens_seen": 1194524672 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003222668004012036, + "loss": 1.3039, + "theoretical_loss": 3.5889550242934734, + "tokens_seen": 1194590208 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003222567703109328, + "loss": 1.4696, + "theoretical_loss": 3.588936898077092, + "tokens_seen": 1194655744 + }, + { + "epoch": 0.36, + "learning_rate": 0.000322246740220662, + "loss": 1.3103, + "theoretical_loss": 3.5889187731334493, + "tokens_seen": 1194721280 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032223671013039117, + "loss": 1.4042, + "theoretical_loss": 3.588900649462385, + "tokens_seen": 1194786816 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032222668004012035, + "loss": 1.3208, + "theoretical_loss": 3.58888252706374, + "tokens_seen": 1194852352 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032221664994984953, + "loss": 1.3652, + "theoretical_loss": 3.588864405937356, + "tokens_seen": 1194917888 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003222066198595787, + "loss": 1.2459, + "theoretical_loss": 3.588846286083074, + "tokens_seen": 1194983424 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032219658976930795, + "loss": 1.375, + "theoretical_loss": 3.5888281675007336, + "tokens_seen": 1195048960 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003221865596790371, + "loss": 1.3568, + "theoretical_loss": 3.5888100501901774, + "tokens_seen": 1195114496 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003221765295887663, + "loss": 1.4388, + "theoretical_loss": 3.5887919341512453, + "tokens_seen": 1195180032 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003221664994984955, + "loss": 1.4276, + "theoretical_loss": 3.5887738193837793, + "tokens_seen": 1195245568 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003221564694082247, + "loss": 1.3715, + "theoretical_loss": 3.588755705887619, + "tokens_seen": 1195311104 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032214643931795386, + "loss": 1.3923, + "theoretical_loss": 3.5887375936626067, + "tokens_seen": 1195376640 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032213640922768304, + "loss": 1.3759, + "theoretical_loss": 3.588719482708583, + "tokens_seen": 1195442176 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003221263791374122, + "loss": 1.3681, + "theoretical_loss": 3.5887013730253896, + "tokens_seen": 1195507712 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032211634904714145, + "loss": 1.4322, + "theoretical_loss": 3.588683264612867, + "tokens_seen": 1195573248 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003221063189568706, + "loss": 1.3622, + "theoretical_loss": 3.5886651574708566, + "tokens_seen": 1195638784 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003220962888665998, + "loss": 1.3422, + "theoretical_loss": 3.5886470515991995, + "tokens_seen": 1195704320 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032208625877632894, + "loss": 1.4651, + "theoretical_loss": 3.588628946997737, + "tokens_seen": 1195769856 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003220762286860582, + "loss": 1.3231, + "theoretical_loss": 3.588610843666311, + "tokens_seen": 1195835392 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003220661985957874, + "loss": 1.3134, + "theoretical_loss": 3.588592741604762, + "tokens_seen": 1195900928 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032205616850551654, + "loss": 1.29, + "theoretical_loss": 3.588574640812932, + "tokens_seen": 1195966464 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -0.0004735648399218917, + "objective/train/docs_used": 680084, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5525765419006348, + "objective/train/original_loss": 2.5525760650634766, + "objective/train/theoretical_loss": 3.5885565412906617, + "objective/train/tokens_used": 1216492000, + "objective/train/value_avg": -0.004390716552734375, + "objective/train/value_loss": 5.17114567628596e-05, + "objective/train/value_max": -0.0001881122589111328, + "objective/train/value_min": -0.08038330078125, + "objective/train/value_reward_corr": 0.3328775229247493, + "objective/train/value_std": 0.0032939910888671875, + "objective/train/weight_avg": 0.9977243542671204, + "objective/train/weighted_lm_loss": 2.5476832389831543, + "objective/train/weights_max": 2.099139928817749, + "objective/train/weights_min": 0.5350926518440247, + "theoretical_loss": 3.5885565412906617, + "tokens_seen": 1196032000 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003220461384152458, + "loss": 1.2548, + "theoretical_loss": 3.5885565412906617, + "tokens_seen": 1196032000 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003220361083249749, + "loss": 1.3266, + "theoretical_loss": 3.588538443037793, + "tokens_seen": 1196097536 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032202607823470414, + "loss": 1.3772, + "theoretical_loss": 3.588520346054167, + "tokens_seen": 1196163072 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003220160481444333, + "loss": 1.4244, + "theoretical_loss": 3.5885022503396256, + "tokens_seen": 1196228608 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003220060180541625, + "loss": 1.3099, + "theoretical_loss": 3.58848415589401, + "tokens_seen": 1196294144 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003219959879638917, + "loss": 1.2937, + "theoretical_loss": 3.588466062717161, + "tokens_seen": 1196359680 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003219859578736209, + "loss": 1.3099, + "theoretical_loss": 3.5884479708089216, + "tokens_seen": 1196425216 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032197592778335004, + "loss": 1.3193, + "theoretical_loss": 3.5884298801691323, + "tokens_seen": 1196490752 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003219658976930793, + "loss": 1.3838, + "theoretical_loss": 3.588411790797635, + "tokens_seen": 1196556288 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003219558676028084, + "loss": 1.3553, + "theoretical_loss": 3.5883937026942716, + "tokens_seen": 1196621824 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032194583751253764, + "loss": 1.3693, + "theoretical_loss": 3.588375615858883, + "tokens_seen": 1196687360 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003219358074222668, + "loss": 1.3901, + "theoretical_loss": 3.588357530291312, + "tokens_seen": 1196752896 + }, + { + "epoch": 0.36, + "learning_rate": 0.000321925777331996, + "loss": 1.3934, + "theoretical_loss": 3.5883394459913993, + "tokens_seen": 1196818432 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003219157472417252, + "loss": 1.3636, + "theoretical_loss": 3.588321362958987, + "tokens_seen": 1196883968 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032190571715145437, + "loss": 1.4042, + "theoretical_loss": 3.5883032811939173, + "tokens_seen": 1196949504 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032189568706118355, + "loss": 1.3758, + "theoretical_loss": 3.5882852006960313, + "tokens_seen": 1197015040 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003218856569709128, + "loss": 1.3562, + "theoretical_loss": 3.588267121465171, + "tokens_seen": 1197080576 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003218756268806419, + "loss": 1.3759, + "theoretical_loss": 3.588249043501178, + "tokens_seen": 1197146112 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032186559679037115, + "loss": 1.3605, + "theoretical_loss": 3.588230966803895, + "tokens_seen": 1197211648 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003218555667001003, + "loss": 1.3239, + "theoretical_loss": 3.5882128913731637, + "tokens_seen": 1197277184 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003218455366098295, + "loss": 1.4218, + "theoretical_loss": 3.5881948172088256, + "tokens_seen": 1197342720 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003218355065195587, + "loss": 1.235, + "theoretical_loss": 3.5881767443107226, + "tokens_seen": 1197408256 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032182547642928787, + "loss": 1.4074, + "theoretical_loss": 3.5881586726786976, + "tokens_seen": 1197473792 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032181544633901705, + "loss": 1.3848, + "theoretical_loss": 3.588140602312591, + "tokens_seen": 1197539328 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003218054162487463, + "loss": 1.4441, + "theoretical_loss": 3.5881225332122466, + "tokens_seen": 1197604864 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": 0.0003491348179522902, + "objective/train/docs_used": 681502, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.816324472427368, + "objective/train/original_loss": 2.816324234008789, + "objective/train/theoretical_loss": 3.5881044653775054, + "objective/train/tokens_used": 1218130400, + "objective/train/value_avg": -0.006839752197265625, + "objective/train/value_loss": 8.715782314538956e-05, + "objective/train/value_max": -0.0003769397735595703, + "objective/train/value_min": -0.08038330078125, + "objective/train/value_reward_corr": 0.322275066776073, + "objective/train/value_std": 0.006023406982421875, + "objective/train/weight_avg": 1.0075523853302002, + "objective/train/weighted_lm_loss": 2.84322452545166, + "objective/train/weights_max": 2.2340877056121826, + "objective/train/weights_min": 0.08613007515668869, + "theoretical_loss": 3.5881044653775054, + "tokens_seen": 1197670400 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003217953861584754, + "loss": 1.3204, + "theoretical_loss": 3.5881044653775054, + "tokens_seen": 1197670400 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032178535606820465, + "loss": 1.3936, + "theoretical_loss": 3.58808639880821, + "tokens_seen": 1197735936 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003217753259779338, + "loss": 1.3004, + "theoretical_loss": 3.588068333504202, + "tokens_seen": 1197801472 + }, + { + "epoch": 0.36, + "learning_rate": 0.000321765295887663, + "loss": 1.3078, + "theoretical_loss": 3.588050269465324, + "tokens_seen": 1197867008 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003217552657973922, + "loss": 1.3689, + "theoretical_loss": 3.5880322066914183, + "tokens_seen": 1197932544 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032174523570712137, + "loss": 1.425, + "theoretical_loss": 3.588014145182327, + "tokens_seen": 1197998080 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032173520561685055, + "loss": 1.2322, + "theoretical_loss": 3.587996084937892, + "tokens_seen": 1198063616 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032172517552657974, + "loss": 1.3711, + "theoretical_loss": 3.587978025957956, + "tokens_seen": 1198129152 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003217151454363089, + "loss": 1.4212, + "theoretical_loss": 3.587959968242361, + "tokens_seen": 1198194688 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032170511534603815, + "loss": 1.3824, + "theoretical_loss": 3.5879419117909497, + "tokens_seen": 1198260224 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003216950852557673, + "loss": 1.3567, + "theoretical_loss": 3.587923856603564, + "tokens_seen": 1198325760 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003216950852557673, + "loss": 1.3714, + "theoretical_loss": 3.587905802680047, + "tokens_seen": 1198391296 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003216850551654965, + "loss": 1.3049, + "theoretical_loss": 3.587887750020241, + "tokens_seen": 1198456832 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003216750250752257, + "loss": 1.3434, + "theoretical_loss": 3.587869698623987, + "tokens_seen": 1198522368 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003216649949849549, + "loss": 1.3841, + "theoretical_loss": 3.587851648491129, + "tokens_seen": 1198587904 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032165496489468406, + "loss": 1.3003, + "theoretical_loss": 3.5878335996215096, + "tokens_seen": 1198653440 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032164493480441324, + "loss": 1.3244, + "theoretical_loss": 3.5878155520149706, + "tokens_seen": 1198718976 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003216349047141424, + "loss": 1.3339, + "theoretical_loss": 3.5877975056713547, + "tokens_seen": 1198784512 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032162487462387165, + "loss": 1.2822, + "theoretical_loss": 3.5877794605905042, + "tokens_seen": 1198850048 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003216148445336008, + "loss": 1.3944, + "theoretical_loss": 3.5877614167722625, + "tokens_seen": 1198915584 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032160481444333, + "loss": 1.4684, + "theoretical_loss": 3.5877433742164717, + "tokens_seen": 1198981120 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032159478435305914, + "loss": 1.3664, + "theoretical_loss": 3.587725332922975, + "tokens_seen": 1199046656 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003215847542627884, + "loss": 1.2883, + "theoretical_loss": 3.587707292891614, + "tokens_seen": 1199112192 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032157472417251756, + "loss": 1.3107, + "theoretical_loss": 3.5876892541222327, + "tokens_seen": 1199177728 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032156469408224674, + "loss": 1.3602, + "theoretical_loss": 3.587671216614673, + "tokens_seen": 1199243264 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -0.0008391356095671654, + "objective/train/docs_used": 682215, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6231067180633545, + "objective/train/original_loss": 2.6231067180633545, + "objective/train/theoretical_loss": 3.5876531803687786, + "objective/train/tokens_used": 1219768800, + "objective/train/value_avg": -0.006069183349609375, + "objective/train/value_loss": 5.529732152353972e-05, + "objective/train/value_max": -0.0003006458282470703, + "objective/train/value_min": -0.060089111328125, + "objective/train/value_reward_corr": 0.44002474204757464, + "objective/train/value_std": 0.005069732666015625, + "objective/train/weight_avg": 0.9942431449890137, + "objective/train/weighted_lm_loss": 2.621985912322998, + "objective/train/weights_max": 1.7437677383422852, + "objective/train/weights_min": 0.4297851026058197, + "theoretical_loss": 3.5876531803687786, + "tokens_seen": 1199308800 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003215546639919759, + "loss": 1.4138, + "theoretical_loss": 3.5876531803687786, + "tokens_seen": 1199308800 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003215446339017051, + "loss": 1.3159, + "theoretical_loss": 3.5876351453843913, + "tokens_seen": 1199374336 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003215346038114343, + "loss": 1.3645, + "theoretical_loss": 3.5876171116613547, + "tokens_seen": 1199439872 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003215245737211635, + "loss": 1.3768, + "theoretical_loss": 3.5875990791995105, + "tokens_seen": 1199505408 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032151454363089265, + "loss": 1.3711, + "theoretical_loss": 3.587581047998703, + "tokens_seen": 1199570944 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003215045135406219, + "loss": 1.3687, + "theoretical_loss": 3.5875630180587743, + "tokens_seen": 1199636480 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032149448345035106, + "loss": 1.2894, + "theoretical_loss": 3.587544989379568, + "tokens_seen": 1199702016 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032148445336008024, + "loss": 1.3846, + "theoretical_loss": 3.5875269619609265, + "tokens_seen": 1199767552 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003214744232698094, + "loss": 1.4756, + "theoretical_loss": 3.587508935802693, + "tokens_seen": 1199833088 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003214643931795386, + "loss": 1.4522, + "theoretical_loss": 3.58749091090471, + "tokens_seen": 1199898624 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003214543630892678, + "loss": 1.4459, + "theoretical_loss": 3.587472887266822, + "tokens_seen": 1199964160 + }, + { + "epoch": 0.36, + "learning_rate": 0.000321444332998997, + "loss": 1.423, + "theoretical_loss": 3.5874548648888704, + "tokens_seen": 1200029696 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032143430290872615, + "loss": 1.4503, + "theoretical_loss": 3.5874368437706994, + "tokens_seen": 1200095232 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003214242728184554, + "loss": 1.5017, + "theoretical_loss": 3.5874188239121523, + "tokens_seen": 1200160768 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003214142427281845, + "loss": 1.3728, + "theoretical_loss": 3.587400805313071, + "tokens_seen": 1200226304 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032140421263791375, + "loss": 1.3362, + "theoretical_loss": 3.5873827879733007, + "tokens_seen": 1200291840 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032139418254764293, + "loss": 1.4115, + "theoretical_loss": 3.587364771892682, + "tokens_seen": 1200357376 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003213841524573721, + "loss": 1.4487, + "theoretical_loss": 3.587346757071061, + "tokens_seen": 1200422912 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003213741223671013, + "loss": 1.3092, + "theoretical_loss": 3.5873287435082792, + "tokens_seen": 1200488448 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003213640922768305, + "loss": 1.3803, + "theoretical_loss": 3.5873107312041803, + "tokens_seen": 1200553984 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032135406218655965, + "loss": 1.3952, + "theoretical_loss": 3.587292720158608, + "tokens_seen": 1200619520 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003213440320962889, + "loss": 1.317, + "theoretical_loss": 3.587274710371405, + "tokens_seen": 1200685056 + }, + { + "epoch": 0.36, + "learning_rate": 0.000321334002006018, + "loss": 1.4744, + "theoretical_loss": 3.5872567018424153, + "tokens_seen": 1200750592 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032132397191574725, + "loss": 1.4795, + "theoretical_loss": 3.5872386945714823, + "tokens_seen": 1200816128 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003213139418254765, + "loss": 1.4156, + "theoretical_loss": 3.587220688558449, + "tokens_seen": 1200881664 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": 0.0006731298635713756, + "objective/train/docs_used": 683386, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.843524217605591, + "objective/train/original_loss": 2.843524694442749, + "objective/train/theoretical_loss": 3.5872026838031594, + "objective/train/tokens_used": 1221407200, + "objective/train/value_avg": -0.005611419677734375, + "objective/train/value_loss": 5.9895115555264056e-05, + "objective/train/value_max": -0.0004601478576660156, + "objective/train/value_min": -0.07305908203125, + "objective/train/value_reward_corr": 0.42050667202529235, + "objective/train/value_std": 0.004322052001953125, + "objective/train/weight_avg": 1.0095324516296387, + "objective/train/weighted_lm_loss": 2.874478578567505, + "objective/train/weights_max": 1.9179273843765259, + "objective/train/weights_min": 0.3149089217185974, + "theoretical_loss": 3.5872026838031594, + "tokens_seen": 1200947200 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003213039117352056, + "loss": 1.3847, + "theoretical_loss": 3.5872026838031594, + "tokens_seen": 1200947200 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032129388164493485, + "loss": 1.4188, + "theoretical_loss": 3.587184680305457, + "tokens_seen": 1201012736 + }, + { + "epoch": 0.36, + "learning_rate": 0.000321283851554664, + "loss": 1.4386, + "theoretical_loss": 3.5871666780651847, + "tokens_seen": 1201078272 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003212738214643932, + "loss": 1.3765, + "theoretical_loss": 3.5871486770821868, + "tokens_seen": 1201143808 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003212637913741224, + "loss": 1.3966, + "theoretical_loss": 3.5871306773563068, + "tokens_seen": 1201209344 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003212537612838516, + "loss": 1.3333, + "theoretical_loss": 3.587112678887388, + "tokens_seen": 1201274880 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032124373119358075, + "loss": 1.3891, + "theoretical_loss": 3.587094681675274, + "tokens_seen": 1201340416 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032123370110330994, + "loss": 1.3898, + "theoretical_loss": 3.5870766857198086, + "tokens_seen": 1201405952 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003212236710130391, + "loss": 1.5038, + "theoretical_loss": 3.587058691020836, + "tokens_seen": 1201471488 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032121364092276835, + "loss": 1.3433, + "theoretical_loss": 3.5870406975781997, + "tokens_seen": 1201537024 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003212036108324975, + "loss": 1.3767, + "theoretical_loss": 3.5870227053917425, + "tokens_seen": 1201602560 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003211935807422267, + "loss": 1.3698, + "theoretical_loss": 3.58700471446131, + "tokens_seen": 1201668096 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003211835506519559, + "loss": 1.3067, + "theoretical_loss": 3.5869867247867444, + "tokens_seen": 1201733632 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003211735205616851, + "loss": 1.365, + "theoretical_loss": 3.5869687363678904, + "tokens_seen": 1201799168 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032116349047141426, + "loss": 1.3234, + "theoretical_loss": 3.5869507492045916, + "tokens_seen": 1201864704 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032115346038114344, + "loss": 1.3693, + "theoretical_loss": 3.586932763296692, + "tokens_seen": 1201930240 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003211434302908726, + "loss": 1.3748, + "theoretical_loss": 3.5869147786440356, + "tokens_seen": 1201995776 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032113340020060185, + "loss": 1.3733, + "theoretical_loss": 3.5868967952464663, + "tokens_seen": 1202061312 + }, + { + "epoch": 0.36, + "learning_rate": 0.000321123370110331, + "loss": 1.3976, + "theoretical_loss": 3.5868788131038283, + "tokens_seen": 1202126848 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003211133400200602, + "loss": 1.4966, + "theoretical_loss": 3.5868608322159647, + "tokens_seen": 1202192384 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032110330992978934, + "loss": 1.3584, + "theoretical_loss": 3.5868428525827207, + "tokens_seen": 1202257920 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003210932798395186, + "loss": 1.4201, + "theoretical_loss": 3.58682487420394, + "tokens_seen": 1202323456 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032108324974924776, + "loss": 1.425, + "theoretical_loss": 3.586806897079466, + "tokens_seen": 1202388992 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032107321965897694, + "loss": 1.4425, + "theoretical_loss": 3.586788921209144, + "tokens_seen": 1202454528 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003210631895687061, + "loss": 1.4309, + "theoretical_loss": 3.586770946592818, + "tokens_seen": 1202520064 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": -0.0028696232475340366, + "objective/train/docs_used": 683937, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7175748348236084, + "objective/train/original_loss": 2.7175745964050293, + "objective/train/theoretical_loss": 3.5867529732303307, + "objective/train/tokens_used": 1223045600, + "objective/train/value_avg": -0.00710296630859375, + "objective/train/value_loss": 0.0001434431760571897, + "objective/train/value_max": -0.0004076957702636719, + "objective/train/value_min": -0.10430908203125, + "objective/train/value_reward_corr": 0.5526201574636218, + "objective/train/value_std": 0.00693511962890625, + "objective/train/weight_avg": 0.977752685546875, + "objective/train/weighted_lm_loss": 2.661174774169922, + "objective/train/weights_max": 2.402294158935547, + "objective/train/weights_min": 0.15354931354522705, + "theoretical_loss": 3.5867529732303307, + "tokens_seen": 1202585600 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003210531594784353, + "loss": 1.3299, + "theoretical_loss": 3.5867529732303307, + "tokens_seen": 1202585600 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003210431293881645, + "loss": 1.3984, + "theoretical_loss": 3.586735001121528, + "tokens_seen": 1202651136 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003210330992978937, + "loss": 1.27, + "theoretical_loss": 3.5867170302662537, + "tokens_seen": 1202716672 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032102306920762285, + "loss": 1.3321, + "theoretical_loss": 3.586699060664352, + "tokens_seen": 1202782208 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003210130391173521, + "loss": 1.5196, + "theoretical_loss": 3.5866810923156667, + "tokens_seen": 1202847744 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032100300902708126, + "loss": 1.4825, + "theoretical_loss": 3.586663125220043, + "tokens_seen": 1202913280 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032099297893681044, + "loss": 1.3421, + "theoretical_loss": 3.5866451593773245, + "tokens_seen": 1202978816 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003209829488465396, + "loss": 1.5276, + "theoretical_loss": 3.586627194787356, + "tokens_seen": 1203044352 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003209729187562688, + "loss": 1.3657, + "theoretical_loss": 3.586609231449982, + "tokens_seen": 1203109888 + }, + { + "epoch": 0.36, + "learning_rate": 0.000320962888665998, + "loss": 1.4, + "theoretical_loss": 3.586591269365046, + "tokens_seen": 1203175424 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003209528585757272, + "loss": 1.4534, + "theoretical_loss": 3.586573308532394, + "tokens_seen": 1203240960 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032094282848545635, + "loss": 1.4214, + "theoretical_loss": 3.5865553489518693, + "tokens_seen": 1203306496 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003209327983951856, + "loss": 1.3889, + "theoretical_loss": 3.586537390623317, + "tokens_seen": 1203372032 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003209227683049147, + "loss": 1.3888, + "theoretical_loss": 3.5865194335465818, + "tokens_seen": 1203437568 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032091273821464395, + "loss": 1.3406, + "theoretical_loss": 3.5865014777215074, + "tokens_seen": 1203503104 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032090270812437313, + "loss": 1.4172, + "theoretical_loss": 3.586483523147939, + "tokens_seen": 1203568640 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003208926780341023, + "loss": 1.4295, + "theoretical_loss": 3.5864655698257213, + "tokens_seen": 1203634176 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003208826479438315, + "loss": 1.3239, + "theoretical_loss": 3.5864476177546987, + "tokens_seen": 1203699712 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003208726178535607, + "loss": 1.3499, + "theoretical_loss": 3.586429666934716, + "tokens_seen": 1203765248 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032086258776328985, + "loss": 1.3677, + "theoretical_loss": 3.586411717365618, + "tokens_seen": 1203830784 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003208525576730191, + "loss": 1.3867, + "theoretical_loss": 3.5863937690472496, + "tokens_seen": 1203896320 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003208425275827482, + "loss": 1.3085, + "theoretical_loss": 3.586375821979455, + "tokens_seen": 1203961856 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032083249749247745, + "loss": 1.4847, + "theoretical_loss": 3.5863578761620794, + "tokens_seen": 1204027392 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032082246740220663, + "loss": 1.3885, + "theoretical_loss": 3.586339931594968, + "tokens_seen": 1204092928 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003208124373119358, + "loss": 1.5267, + "theoretical_loss": 3.5863219882779647, + "tokens_seen": 1204158464 + }, + { + "epoch": 0.36, + "objective/train/advantage_avg": 0.0001190394614241086, + "objective/train/docs_used": 684175, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6329638957977295, + "objective/train/original_loss": 2.6329636573791504, + "objective/train/theoretical_loss": 3.5863040462109153, + "objective/train/tokens_used": 1224684000, + "objective/train/value_avg": -0.00494384765625, + "objective/train/value_loss": 4.591863034875132e-05, + "objective/train/value_max": -0.0004601478576660156, + "objective/train/value_min": -0.07489013671875, + "objective/train/value_reward_corr": 0.32737178459927674, + "objective/train/value_std": 0.0033245086669921875, + "objective/train/weight_avg": 1.0033938884735107, + "objective/train/weighted_lm_loss": 2.6411309242248535, + "objective/train/weights_max": 1.9108535051345825, + "objective/train/weights_min": 0.5527223944664001, + "theoretical_loss": 3.5863040462109153, + "tokens_seen": 1204224000 + }, + { + "epoch": 0.36, + "learning_rate": 0.000320802407221665, + "loss": 1.3118, + "theoretical_loss": 3.5863040462109153, + "tokens_seen": 1204224000 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003207923771313942, + "loss": 1.3463, + "theoretical_loss": 3.586286105393664, + "tokens_seen": 1204289536 + }, + { + "epoch": 0.36, + "learning_rate": 0.00032078234704112336, + "loss": 1.3569, + "theoretical_loss": 3.5862681658260565, + "tokens_seen": 1204355072 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003207723169508526, + "loss": 1.4741, + "theoretical_loss": 3.5862502275079366, + "tokens_seen": 1204420608 + }, + { + "epoch": 0.36, + "learning_rate": 0.0003207622868605817, + "loss": 1.3632, + "theoretical_loss": 3.586232290439151, + "tokens_seen": 1204486144 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032075225677031095, + "loss": 1.3619, + "theoretical_loss": 3.5862143546195426, + "tokens_seen": 1204551680 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003207422266800401, + "loss": 1.4803, + "theoretical_loss": 3.5861964200489584, + "tokens_seen": 1204617216 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003207321965897693, + "loss": 1.2703, + "theoretical_loss": 3.5861784867272424, + "tokens_seen": 1204682752 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003207221664994985, + "loss": 1.3384, + "theoretical_loss": 3.5861605546542403, + "tokens_seen": 1204748288 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003207121364092277, + "loss": 1.4296, + "theoretical_loss": 3.5861426238297964, + "tokens_seen": 1204813824 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032070210631895686, + "loss": 1.3952, + "theoretical_loss": 3.5861246942537566, + "tokens_seen": 1204879360 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003206920762286861, + "loss": 1.4147, + "theoretical_loss": 3.5861067659259653, + "tokens_seen": 1204944896 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003206820461384152, + "loss": 1.5725, + "theoretical_loss": 3.586088838846269, + "tokens_seen": 1205010432 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032067201604814446, + "loss": 1.3494, + "theoretical_loss": 3.5860709130145114, + "tokens_seen": 1205075968 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003206619859578736, + "loss": 1.4083, + "theoretical_loss": 3.586052988430539, + "tokens_seen": 1205141504 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003206519558676028, + "loss": 1.421, + "theoretical_loss": 3.586035065094197, + "tokens_seen": 1205207040 + }, + { + "epoch": 0.37, + "learning_rate": 0.000320641925777332, + "loss": 1.4483, + "theoretical_loss": 3.5860171430053303, + "tokens_seen": 1205272576 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003206318956870612, + "loss": 1.3549, + "theoretical_loss": 3.5859992221637835, + "tokens_seen": 1205338112 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032062186559679036, + "loss": 1.3456, + "theoretical_loss": 3.5859813025694036, + "tokens_seen": 1205403648 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032061183550651954, + "loss": 1.4039, + "theoretical_loss": 3.585963384222035, + "tokens_seen": 1205469184 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003206018054162487, + "loss": 1.36, + "theoretical_loss": 3.5859454671215234, + "tokens_seen": 1205534720 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032059177532597796, + "loss": 1.383, + "theoretical_loss": 3.585927551267714, + "tokens_seen": 1205600256 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003205817452357071, + "loss": 1.4946, + "theoretical_loss": 3.585909636660453, + "tokens_seen": 1205665792 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003205717151454363, + "loss": 1.3406, + "theoretical_loss": 3.585891723299585, + "tokens_seen": 1205731328 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003205616850551655, + "loss": 1.4746, + "theoretical_loss": 3.585873811184956, + "tokens_seen": 1205796864 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.0003243250248488039, + "objective/train/docs_used": 684684, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4674696922302246, + "objective/train/original_loss": 2.4674696922302246, + "objective/train/theoretical_loss": 3.585855900316411, + "objective/train/tokens_used": 1226322400, + "objective/train/value_avg": -0.00536346435546875, + "objective/train/value_loss": 4.3743522837758064e-05, + "objective/train/value_max": -0.00047850608825683594, + "objective/train/value_min": -0.064208984375, + "objective/train/value_reward_corr": 0.28284740720757884, + "objective/train/value_std": 0.003398895263671875, + "objective/train/weight_avg": 0.9988846778869629, + "objective/train/weighted_lm_loss": 2.463629722595215, + "objective/train/weights_max": 1.6369521617889404, + "objective/train/weights_min": 0.4521520435810089, + "theoretical_loss": 3.585855900316411, + "tokens_seen": 1205862400 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003205516549648947, + "loss": 1.3481, + "theoretical_loss": 3.585855900316411, + "tokens_seen": 1205862400 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003205416248746239, + "loss": 1.3664, + "theoretical_loss": 3.585837990693797, + "tokens_seen": 1205927936 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032053159478435305, + "loss": 1.5461, + "theoretical_loss": 3.5858200823169586, + "tokens_seen": 1205993472 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003205215646940823, + "loss": 1.3435, + "theoretical_loss": 3.585802175185741, + "tokens_seen": 1206059008 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032051153460381146, + "loss": 1.3116, + "theoretical_loss": 3.5857842692999915, + "tokens_seen": 1206124544 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032050150451354064, + "loss": 1.536, + "theoretical_loss": 3.5857663646595537, + "tokens_seen": 1206190080 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003204914744232698, + "loss": 1.4853, + "theoretical_loss": 3.5857484612642754, + "tokens_seen": 1206255616 + }, + { + "epoch": 0.37, + "learning_rate": 0.000320481444332999, + "loss": 1.5106, + "theoretical_loss": 3.5857305591140007, + "tokens_seen": 1206321152 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003204714142427282, + "loss": 1.4233, + "theoretical_loss": 3.5857126582085765, + "tokens_seen": 1206386688 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003204613841524574, + "loss": 1.4784, + "theoretical_loss": 3.585694758547848, + "tokens_seen": 1206452224 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032045135406218655, + "loss": 1.4049, + "theoretical_loss": 3.5856768601316613, + "tokens_seen": 1206517760 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003204413239719158, + "loss": 1.3405, + "theoretical_loss": 3.585658962959862, + "tokens_seen": 1206583296 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003204312938816449, + "loss": 1.3871, + "theoretical_loss": 3.585641067032297, + "tokens_seen": 1206648832 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032042126379137415, + "loss": 1.4374, + "theoretical_loss": 3.585623172348811, + "tokens_seen": 1206714368 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032041123370110333, + "loss": 1.4044, + "theoretical_loss": 3.58560527890925, + "tokens_seen": 1206779904 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003204012036108325, + "loss": 1.424, + "theoretical_loss": 3.5855873867134607, + "tokens_seen": 1206845440 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003203911735205617, + "loss": 1.4097, + "theoretical_loss": 3.5855694957612894, + "tokens_seen": 1206910976 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003203811434302909, + "loss": 1.3374, + "theoretical_loss": 3.5855516060525807, + "tokens_seen": 1206976512 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032037111334002005, + "loss": 1.4786, + "theoretical_loss": 3.585533717587182, + "tokens_seen": 1207042048 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003203610832497493, + "loss": 1.5011, + "theoretical_loss": 3.5855158303649386, + "tokens_seen": 1207107584 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003203510531594784, + "loss": 1.4605, + "theoretical_loss": 3.585497944385697, + "tokens_seen": 1207173120 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032034102306920765, + "loss": 1.3849, + "theoretical_loss": 3.5854800596493037, + "tokens_seen": 1207238656 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032033099297893683, + "loss": 1.3986, + "theoretical_loss": 3.585462176155604, + "tokens_seen": 1207304192 + }, + { + "epoch": 0.37, + "learning_rate": 0.000320320962888666, + "loss": 1.4804, + "theoretical_loss": 3.5854442939044446, + "tokens_seen": 1207369728 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003203109327983952, + "loss": 1.3896, + "theoretical_loss": 3.5854264128956723, + "tokens_seen": 1207435264 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.0002570803335402161, + "objective/train/docs_used": 686035, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9326772689819336, + "objective/train/original_loss": 2.9326770305633545, + "objective/train/theoretical_loss": 3.585408533129132, + "objective/train/tokens_used": 1227960800, + "objective/train/value_avg": -0.005290985107421875, + "objective/train/value_loss": 5.463486013468355e-05, + "objective/train/value_max": -0.00047850608825683594, + "objective/train/value_min": -0.040985107421875, + "objective/train/value_reward_corr": 0.41946863176803295, + "objective/train/value_std": 0.004314422607421875, + "objective/train/weight_avg": 0.9999415278434753, + "objective/train/weighted_lm_loss": 2.9357712268829346, + "objective/train/weights_max": 1.3794091939926147, + "objective/train/weights_min": 0.19491425156593323, + "theoretical_loss": 3.585408533129132, + "tokens_seen": 1207500800 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003203009027081244, + "loss": 1.4618, + "theoretical_loss": 3.585408533129132, + "tokens_seen": 1207500800 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032029087261785356, + "loss": 1.3743, + "theoretical_loss": 3.585390654604671, + "tokens_seen": 1207566336 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003202808425275828, + "loss": 1.5033, + "theoretical_loss": 3.585372777322135, + "tokens_seen": 1207631872 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003202708124373119, + "loss": 1.3044, + "theoretical_loss": 3.5853549012813715, + "tokens_seen": 1207697408 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032026078234704115, + "loss": 1.3743, + "theoretical_loss": 3.5853370264822253, + "tokens_seen": 1207762944 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003202507522567703, + "loss": 1.342, + "theoretical_loss": 3.5853191529245443, + "tokens_seen": 1207828480 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003202407221664995, + "loss": 1.4783, + "theoretical_loss": 3.5853012806081734, + "tokens_seen": 1207894016 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003202306920762287, + "loss": 1.3691, + "theoretical_loss": 3.58528340953296, + "tokens_seen": 1207959552 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003202206619859579, + "loss": 1.3388, + "theoretical_loss": 3.5852655396987507, + "tokens_seen": 1208025088 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032021063189568706, + "loss": 1.4696, + "theoretical_loss": 3.5852476711053916, + "tokens_seen": 1208090624 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003202006018054163, + "loss": 1.47, + "theoretical_loss": 3.5852298037527293, + "tokens_seen": 1208156160 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003201905717151454, + "loss": 1.3558, + "theoretical_loss": 3.5852119376406106, + "tokens_seen": 1208221696 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032018054162487466, + "loss": 1.3846, + "theoretical_loss": 3.585194072768881, + "tokens_seen": 1208287232 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003201705115346038, + "loss": 1.3584, + "theoretical_loss": 3.585176209137389, + "tokens_seen": 1208352768 + }, + { + "epoch": 0.37, + "learning_rate": 0.000320160481444333, + "loss": 1.4508, + "theoretical_loss": 3.58515834674598, + "tokens_seen": 1208418304 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003201504513540622, + "loss": 1.3491, + "theoretical_loss": 3.5851404855945006, + "tokens_seen": 1208483840 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003201404212637914, + "loss": 1.3979, + "theoretical_loss": 3.585122625682798, + "tokens_seen": 1208549376 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032013039117352056, + "loss": 1.5207, + "theoretical_loss": 3.585104767010719, + "tokens_seen": 1208614912 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032012036108324974, + "loss": 1.3903, + "theoretical_loss": 3.5850869095781093, + "tokens_seen": 1208680448 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003201103309929789, + "loss": 1.2596, + "theoretical_loss": 3.5850690533848173, + "tokens_seen": 1208745984 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032010030090270816, + "loss": 1.4372, + "theoretical_loss": 3.5850511984306888, + "tokens_seen": 1208811520 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003200902708124373, + "loss": 1.3839, + "theoretical_loss": 3.58503334471557, + "tokens_seen": 1208877056 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003200802407221665, + "loss": 1.2744, + "theoretical_loss": 3.585015492239309, + "tokens_seen": 1208942592 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032007021063189565, + "loss": 1.3904, + "theoretical_loss": 3.5849976410017526, + "tokens_seen": 1209008128 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003200601805416249, + "loss": 1.3998, + "theoretical_loss": 3.584979791002747, + "tokens_seen": 1209073664 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.0005930964834988117, + "objective/train/docs_used": 686549, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0138299465179443, + "objective/train/original_loss": 3.0138301849365234, + "objective/train/theoretical_loss": 3.5849619422421393, + "objective/train/tokens_used": 1229599200, + "objective/train/value_avg": -0.00467681884765625, + "objective/train/value_loss": 5.155201506568119e-05, + "objective/train/value_max": -0.0002378225326538086, + "objective/train/value_min": -0.04779052734375, + "objective/train/value_reward_corr": 0.4384439238501189, + "objective/train/value_std": 0.003582000732421875, + "objective/train/weight_avg": 0.9965124726295471, + "objective/train/weighted_lm_loss": 3.014849901199341, + "objective/train/weights_max": 1.4554990530014038, + "objective/train/weights_min": 0.5366871953010559, + "theoretical_loss": 3.5849619422421393, + "tokens_seen": 1209139200 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032005015045135407, + "loss": 1.3518, + "theoretical_loss": 3.5849619422421393, + "tokens_seen": 1209139200 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032004012036108325, + "loss": 1.4425, + "theoretical_loss": 3.584944094719777, + "tokens_seen": 1209204736 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032003009027081243, + "loss": 1.4325, + "theoretical_loss": 3.5849262484355067, + "tokens_seen": 1209270272 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032002006018054166, + "loss": 1.3902, + "theoretical_loss": 3.584908403389176, + "tokens_seen": 1209335808 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003200100300902708, + "loss": 1.2827, + "theoretical_loss": 3.5848905595806304, + "tokens_seen": 1209401344 + }, + { + "epoch": 0.37, + "learning_rate": 0.00032, + "loss": 1.311, + "theoretical_loss": 3.584872717009718, + "tokens_seen": 1209466880 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031998996990972915, + "loss": 1.4041, + "theoretical_loss": 3.5848548756762866, + "tokens_seen": 1209532416 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003199799398194584, + "loss": 1.3568, + "theoretical_loss": 3.584837035580182, + "tokens_seen": 1209597952 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031996990972918757, + "loss": 1.4056, + "theoretical_loss": 3.5848191967212526, + "tokens_seen": 1209663488 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031995987963891675, + "loss": 1.5528, + "theoretical_loss": 3.5848013590993446, + "tokens_seen": 1209729024 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031994984954864593, + "loss": 1.3281, + "theoretical_loss": 3.5847835227143054, + "tokens_seen": 1209794560 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003199398194583751, + "loss": 1.2643, + "theoretical_loss": 3.584765687565983, + "tokens_seen": 1209860096 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003199297893681043, + "loss": 1.4055, + "theoretical_loss": 3.5847478536542234, + "tokens_seen": 1209925632 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031991975927783353, + "loss": 1.4226, + "theoretical_loss": 3.584730020978875, + "tokens_seen": 1209991168 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031990972918756266, + "loss": 1.4349, + "theoretical_loss": 3.5847121895397844, + "tokens_seen": 1210056704 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003198996990972919, + "loss": 1.3772, + "theoretical_loss": 3.5846943593367997, + "tokens_seen": 1210122240 + }, + { + "epoch": 0.37, + "learning_rate": 0.000319889669007021, + "loss": 1.3997, + "theoretical_loss": 3.5846765303697676, + "tokens_seen": 1210187776 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031987963891675025, + "loss": 1.3849, + "theoretical_loss": 3.5846587026385355, + "tokens_seen": 1210253312 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031986960882647943, + "loss": 1.3093, + "theoretical_loss": 3.5846408761429513, + "tokens_seen": 1210318848 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003198595787362086, + "loss": 1.2439, + "theoretical_loss": 3.584623050882862, + "tokens_seen": 1210384384 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003198495486459378, + "loss": 1.3685, + "theoretical_loss": 3.5846052268581157, + "tokens_seen": 1210449920 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031983951855566703, + "loss": 1.3525, + "theoretical_loss": 3.584587404068559, + "tokens_seen": 1210515456 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031982948846539616, + "loss": 1.4465, + "theoretical_loss": 3.5845695825140407, + "tokens_seen": 1210580992 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003198194583751254, + "loss": 1.3399, + "theoretical_loss": 3.5845517621944065, + "tokens_seen": 1210646528 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003198094282848546, + "loss": 1.2975, + "theoretical_loss": 3.584533943109506, + "tokens_seen": 1210712064 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.0003979795437771827, + "objective/train/docs_used": 687794, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.233442783355713, + "objective/train/original_loss": 3.233442783355713, + "objective/train/theoretical_loss": 3.5845161252591855, + "objective/train/tokens_used": 1231237600, + "objective/train/value_avg": -0.004962921142578125, + "objective/train/value_loss": 5.998182678013109e-05, + "objective/train/value_max": -0.00034737586975097656, + "objective/train/value_min": -0.07171630859375, + "objective/train/value_reward_corr": 0.4434056309545491, + "objective/train/value_std": 0.0039215087890625, + "objective/train/weight_avg": 0.998746931552887, + "objective/train/weighted_lm_loss": 3.228851318359375, + "objective/train/weights_max": 1.7189687490463257, + "objective/train/weights_min": 0.14896823465824127, + "theoretical_loss": 3.5845161252591855, + "tokens_seen": 1210777600 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031979939819458376, + "loss": 1.4666, + "theoretical_loss": 3.5845161252591855, + "tokens_seen": 1210777600 + }, + { + "epoch": 0.37, + "learning_rate": 0.000319789368104313, + "loss": 1.4836, + "theoretical_loss": 3.584498308643293, + "tokens_seen": 1210843136 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003197793380140421, + "loss": 1.3453, + "theoretical_loss": 3.5844804932616765, + "tokens_seen": 1210908672 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031976930792377135, + "loss": 1.3072, + "theoretical_loss": 3.5844626791141834, + "tokens_seen": 1210974208 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003197592778335005, + "loss": 1.4887, + "theoretical_loss": 3.5844448662006614, + "tokens_seen": 1211039744 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003197492477432297, + "loss": 1.3834, + "theoretical_loss": 3.5844270545209582, + "tokens_seen": 1211105280 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003197392176529589, + "loss": 1.2881, + "theoretical_loss": 3.584409244074922, + "tokens_seen": 1211170816 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003197291875626881, + "loss": 1.3908, + "theoretical_loss": 3.5843914348623995, + "tokens_seen": 1211236352 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031971915747241726, + "loss": 1.4625, + "theoretical_loss": 3.5843736268832402, + "tokens_seen": 1211301888 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003197091273821465, + "loss": 1.4055, + "theoretical_loss": 3.584355820137291, + "tokens_seen": 1211367424 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003196990972918756, + "loss": 1.4294, + "theoretical_loss": 3.5843380146243993, + "tokens_seen": 1211432960 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031968906720160486, + "loss": 1.3496, + "theoretical_loss": 3.584320210344414, + "tokens_seen": 1211498496 + }, + { + "epoch": 0.37, + "learning_rate": 0.000319679037111334, + "loss": 1.4158, + "theoretical_loss": 3.5843024072971827, + "tokens_seen": 1211564032 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003196690070210632, + "loss": 1.3914, + "theoretical_loss": 3.5842846054825532, + "tokens_seen": 1211629568 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003196589769307924, + "loss": 1.4811, + "theoretical_loss": 3.584266804900374, + "tokens_seen": 1211695104 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003196489468405216, + "loss": 1.3759, + "theoretical_loss": 3.5842490055504923, + "tokens_seen": 1211760640 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031963891675025076, + "loss": 1.3153, + "theoretical_loss": 3.5842312074327567, + "tokens_seen": 1211826176 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031962888665997994, + "loss": 1.3586, + "theoretical_loss": 3.584213410547015, + "tokens_seen": 1211891712 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003196188565697091, + "loss": 1.382, + "theoretical_loss": 3.5841956148931153, + "tokens_seen": 1211957248 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031960882647943836, + "loss": 1.4285, + "theoretical_loss": 3.5841778204709067, + "tokens_seen": 1212022784 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003195987963891675, + "loss": 1.3001, + "theoretical_loss": 3.5841600272802356, + "tokens_seen": 1212088320 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003195887662988967, + "loss": 1.4535, + "theoretical_loss": 3.584142235320952, + "tokens_seen": 1212153856 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031957873620862585, + "loss": 1.3822, + "theoretical_loss": 3.584124444592903, + "tokens_seen": 1212219392 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003195687061183551, + "loss": 1.3245, + "theoretical_loss": 3.584106655095937, + "tokens_seen": 1212284928 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031955867602808427, + "loss": 1.4571, + "theoretical_loss": 3.584088866829902, + "tokens_seen": 1212350464 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": 0.000763782998546958, + "objective/train/docs_used": 688453, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.676137685775757, + "objective/train/original_loss": 2.676138162612915, + "objective/train/theoretical_loss": 3.584071079794647, + "objective/train/tokens_used": 1232876000, + "objective/train/value_avg": -0.00504302978515625, + "objective/train/value_loss": 7.062086660880595e-05, + "objective/train/value_max": -0.00015115737915039062, + "objective/train/value_min": -0.062225341796875, + "objective/train/value_reward_corr": 0.40625735870293816, + "objective/train/value_std": 0.004100799560546875, + "objective/train/weight_avg": 1.0105899572372437, + "objective/train/weighted_lm_loss": 2.706721067428589, + "objective/train/weights_max": 1.863121747970581, + "objective/train/weights_min": 0.1348153054714203, + "theoretical_loss": 3.584071079794647, + "tokens_seen": 1212416000 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031954864593781345, + "loss": 1.3286, + "theoretical_loss": 3.584071079794647, + "tokens_seen": 1212416000 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031953861584754263, + "loss": 1.4599, + "theoretical_loss": 3.58405329399002, + "tokens_seen": 1212481536 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031952858575727186, + "loss": 1.4976, + "theoretical_loss": 3.5840355094158696, + "tokens_seen": 1212547072 + }, + { + "epoch": 0.37, + "learning_rate": 0.000319518555667001, + "loss": 1.3879, + "theoretical_loss": 3.5840177260720436, + "tokens_seen": 1212612608 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003195085255767302, + "loss": 1.2957, + "theoretical_loss": 3.5839999439583905, + "tokens_seen": 1212678144 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031949849548645935, + "loss": 1.34, + "theoretical_loss": 3.583982163074759, + "tokens_seen": 1212743680 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003194884653961886, + "loss": 1.3948, + "theoretical_loss": 3.5839643834209975, + "tokens_seen": 1212809216 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031947843530591777, + "loss": 1.3081, + "theoretical_loss": 3.583946604996955, + "tokens_seen": 1212874752 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031946840521564695, + "loss": 1.4379, + "theoretical_loss": 3.5839288278024783, + "tokens_seen": 1212940288 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031945837512537613, + "loss": 1.3776, + "theoretical_loss": 3.583911051837418, + "tokens_seen": 1213005824 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003194483450351053, + "loss": 1.3761, + "theoretical_loss": 3.5838932771016214, + "tokens_seen": 1213071360 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003194383149448345, + "loss": 1.3864, + "theoretical_loss": 3.583875503594938, + "tokens_seen": 1213136896 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031942828485456373, + "loss": 1.4326, + "theoretical_loss": 3.5838577313172157, + "tokens_seen": 1213202432 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031941825476429286, + "loss": 1.3558, + "theoretical_loss": 3.583839960268303, + "tokens_seen": 1213267968 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003194082246740221, + "loss": 1.3329, + "theoretical_loss": 3.583822190448049, + "tokens_seen": 1213333504 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003193981945837512, + "loss": 1.3681, + "theoretical_loss": 3.583804421856302, + "tokens_seen": 1213399040 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031938816449348045, + "loss": 1.4004, + "theoretical_loss": 3.5837866544929113, + "tokens_seen": 1213464576 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031937813440320964, + "loss": 1.4823, + "theoretical_loss": 3.5837688883577252, + "tokens_seen": 1213530112 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003193681043129388, + "loss": 1.2478, + "theoretical_loss": 3.5837511234505928, + "tokens_seen": 1213595648 + }, + { + "epoch": 0.37, + "learning_rate": 0.000319358074222668, + "loss": 1.5278, + "theoretical_loss": 3.5837333597713625, + "tokens_seen": 1213661184 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031934804413239723, + "loss": 1.4585, + "theoretical_loss": 3.5837155973198835, + "tokens_seen": 1213726720 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031933801404212636, + "loss": 1.5295, + "theoretical_loss": 3.583697836096005, + "tokens_seen": 1213792256 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003193279839518556, + "loss": 1.4641, + "theoretical_loss": 3.5836800760995744, + "tokens_seen": 1213857792 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003193179538615847, + "loss": 1.4656, + "theoretical_loss": 3.5836623173304423, + "tokens_seen": 1213923328 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031930792377131396, + "loss": 1.3828, + "theoretical_loss": 3.583644559788457, + "tokens_seen": 1213988864 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.0013951535802334547, + "objective/train/docs_used": 689723, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6151533126831055, + "objective/train/original_loss": 2.6151528358459473, + "objective/train/theoretical_loss": 3.583626803473467, + "objective/train/tokens_used": 1234514400, + "objective/train/value_avg": -0.00478363037109375, + "objective/train/value_loss": 6.026513074175455e-05, + "objective/train/value_max": -0.0003006458282470703, + "objective/train/value_min": -0.050323486328125, + "objective/train/value_reward_corr": 0.4449814047553388, + "objective/train/value_std": 0.0035858154296875, + "objective/train/weight_avg": 0.988889753818512, + "objective/train/weighted_lm_loss": 2.592240333557129, + "objective/train/weights_max": 1.3452321290969849, + "objective/train/weights_min": 0.506985068321228, + "theoretical_loss": 3.583626803473467, + "tokens_seen": 1214054400 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031929789368104314, + "loss": 1.3519, + "theoretical_loss": 3.583626803473467, + "tokens_seen": 1214054400 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003192878635907723, + "loss": 1.4136, + "theoretical_loss": 3.5836090483853216, + "tokens_seen": 1214119936 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003192778335005015, + "loss": 1.2803, + "theoretical_loss": 3.5835912945238704, + "tokens_seen": 1214185472 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003192678034102307, + "loss": 1.3046, + "theoretical_loss": 3.5835735418889616, + "tokens_seen": 1214251008 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031925777331995986, + "loss": 1.4383, + "theoretical_loss": 3.583555790480445, + "tokens_seen": 1214316544 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003192477432296891, + "loss": 1.2713, + "theoretical_loss": 3.583538040298169, + "tokens_seen": 1214382080 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003192377131394182, + "loss": 1.4259, + "theoretical_loss": 3.583520291341983, + "tokens_seen": 1214447616 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031922768304914746, + "loss": 1.3766, + "theoretical_loss": 3.5835025436117363, + "tokens_seen": 1214513152 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003192176529588766, + "loss": 1.4192, + "theoretical_loss": 3.5834847971072783, + "tokens_seen": 1214578688 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003192076228686058, + "loss": 1.3603, + "theoretical_loss": 3.5834670518284577, + "tokens_seen": 1214644224 + }, + { + "epoch": 0.37, + "learning_rate": 0.000319197592778335, + "loss": 1.3578, + "theoretical_loss": 3.5834493077751235, + "tokens_seen": 1214709760 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003191875626880642, + "loss": 1.4891, + "theoretical_loss": 3.5834315649471256, + "tokens_seen": 1214775296 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031917753259779337, + "loss": 1.413, + "theoretical_loss": 3.583413823344313, + "tokens_seen": 1214840832 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003191675025075226, + "loss": 1.4396, + "theoretical_loss": 3.5833960829665354, + "tokens_seen": 1214906368 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031915747241725173, + "loss": 1.4828, + "theoretical_loss": 3.5833783438136417, + "tokens_seen": 1214971904 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031914744232698096, + "loss": 1.3571, + "theoretical_loss": 3.583360605885481, + "tokens_seen": 1215037440 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003191374122367101, + "loss": 1.3978, + "theoretical_loss": 3.5833428691819034, + "tokens_seen": 1215102976 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003191273821464393, + "loss": 1.4528, + "theoretical_loss": 3.583325133702758, + "tokens_seen": 1215168512 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003191173520561685, + "loss": 1.428, + "theoretical_loss": 3.583307399447894, + "tokens_seen": 1215234048 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003191073219658977, + "loss": 1.529, + "theoretical_loss": 3.583289666417161, + "tokens_seen": 1215299584 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031909729187562687, + "loss": 1.3942, + "theoretical_loss": 3.5832719346104094, + "tokens_seen": 1215365120 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031908726178535605, + "loss": 1.3857, + "theoretical_loss": 3.5832542040274875, + "tokens_seen": 1215430656 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031907723169508523, + "loss": 1.4062, + "theoretical_loss": 3.5832364746682446, + "tokens_seen": 1215496192 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031906720160481447, + "loss": 1.3853, + "theoretical_loss": 3.5832187465325314, + "tokens_seen": 1215561728 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031905717151454365, + "loss": 1.2944, + "theoretical_loss": 3.5832010196201973, + "tokens_seen": 1215627264 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": 8.197131683118641e-05, + "objective/train/docs_used": 690230, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.165236234664917, + "objective/train/original_loss": 2.165236234664917, + "objective/train/theoretical_loss": 3.583183293931091, + "objective/train/tokens_used": 1236152800, + "objective/train/value_avg": -0.005218505859375, + "objective/train/value_loss": 4.111710950382985e-05, + "objective/train/value_max": -0.00019872188568115234, + "objective/train/value_min": -0.048309326171875, + "objective/train/value_reward_corr": 0.4138505524201088, + "objective/train/value_std": 0.00421142578125, + "objective/train/weight_avg": 1.0028103590011597, + "objective/train/weighted_lm_loss": 2.1628174781799316, + "objective/train/weights_max": 1.4639885425567627, + "objective/train/weights_min": 0.5445587635040283, + "theoretical_loss": 3.583183293931091, + "tokens_seen": 1215692800 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031904714142427283, + "loss": 1.2136, + "theoretical_loss": 3.583183293931091, + "tokens_seen": 1215692800 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031903711133400206, + "loss": 1.4391, + "theoretical_loss": 3.5831655694650637, + "tokens_seen": 1215758336 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003190270812437312, + "loss": 1.423, + "theoretical_loss": 3.583147846221964, + "tokens_seen": 1215823872 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003190170511534604, + "loss": 1.4019, + "theoretical_loss": 3.583130124201641, + "tokens_seen": 1215889408 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031900702106318955, + "loss": 1.3864, + "theoretical_loss": 3.583112403403946, + "tokens_seen": 1215954944 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003189969909729188, + "loss": 1.3985, + "theoretical_loss": 3.583094683828728, + "tokens_seen": 1216020480 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031898696088264797, + "loss": 1.4155, + "theoretical_loss": 3.5830769654758368, + "tokens_seen": 1216086016 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031897693079237715, + "loss": 1.4673, + "theoretical_loss": 3.583059248345122, + "tokens_seen": 1216151552 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031896690070210633, + "loss": 1.4498, + "theoretical_loss": 3.5830415324364338, + "tokens_seen": 1216217088 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003189568706118355, + "loss": 1.2987, + "theoretical_loss": 3.5830238177496216, + "tokens_seen": 1216282624 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003189468405215647, + "loss": 1.452, + "theoretical_loss": 3.5830061042845363, + "tokens_seen": 1216348160 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031893681043129393, + "loss": 1.3751, + "theoretical_loss": 3.5829883920410266, + "tokens_seen": 1216413696 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031892678034102306, + "loss": 1.3578, + "theoretical_loss": 3.582970681018943, + "tokens_seen": 1216479232 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003189167502507523, + "loss": 1.3928, + "theoretical_loss": 3.582952971218136, + "tokens_seen": 1216544768 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003189067201604814, + "loss": 1.4699, + "theoretical_loss": 3.582935262638455, + "tokens_seen": 1216610304 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031889669007021065, + "loss": 1.3772, + "theoretical_loss": 3.58291755527975, + "tokens_seen": 1216675840 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031888665997993984, + "loss": 1.3797, + "theoretical_loss": 3.582899849141871, + "tokens_seen": 1216741376 + }, + { + "epoch": 0.37, + "learning_rate": 0.000318876629889669, + "loss": 1.5126, + "theoretical_loss": 3.5828821442246683, + "tokens_seen": 1216806912 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003188665997993982, + "loss": 1.2983, + "theoretical_loss": 3.5828644405279917, + "tokens_seen": 1216872448 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031885656970912743, + "loss": 1.423, + "theoretical_loss": 3.5828467380516917, + "tokens_seen": 1216937984 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031884653961885656, + "loss": 1.447, + "theoretical_loss": 3.582829036795619, + "tokens_seen": 1217003520 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003188365095285858, + "loss": 1.2976, + "theoretical_loss": 3.5828113367596224, + "tokens_seen": 1217069056 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003188264794383149, + "loss": 1.3592, + "theoretical_loss": 3.582793637943553, + "tokens_seen": 1217134592 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031881644934804416, + "loss": 1.4327, + "theoretical_loss": 3.582775940347261, + "tokens_seen": 1217200128 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031880641925777334, + "loss": 1.4082, + "theoretical_loss": 3.5827582439705967, + "tokens_seen": 1217265664 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.0018671659054234624, + "objective/train/docs_used": 691368, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8914215564727783, + "objective/train/original_loss": 2.8914217948913574, + "objective/train/theoretical_loss": 3.58274054881341, + "objective/train/tokens_used": 1237791200, + "objective/train/value_avg": -0.006206512451171875, + "objective/train/value_loss": 0.00012628188414964825, + "objective/train/value_max": -0.0005130767822265625, + "objective/train/value_min": -0.06396484375, + "objective/train/value_reward_corr": 0.6809863726371388, + "objective/train/value_std": 0.0058135986328125, + "objective/train/weight_avg": 0.9869283437728882, + "objective/train/weighted_lm_loss": 2.8523504734039307, + "objective/train/weights_max": 1.5172054767608643, + "objective/train/weights_min": 0.3666175901889801, + "theoretical_loss": 3.58274054881341, + "tokens_seen": 1217331200 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003187963891675025, + "loss": 1.3877, + "theoretical_loss": 3.58274054881341, + "tokens_seen": 1217331200 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003187863590772317, + "loss": 1.3764, + "theoretical_loss": 3.582722854875552, + "tokens_seen": 1217396736 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003187763289869609, + "loss": 1.2708, + "theoretical_loss": 3.5827051621568717, + "tokens_seen": 1217462272 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031876629889669006, + "loss": 1.3535, + "theoretical_loss": 3.582687470657221, + "tokens_seen": 1217527808 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003187562688064193, + "loss": 1.4326, + "theoretical_loss": 3.5826697803764493, + "tokens_seen": 1217593344 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003187462387161484, + "loss": 1.4682, + "theoretical_loss": 3.5826520913144075, + "tokens_seen": 1217658880 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031873620862587766, + "loss": 1.3263, + "theoretical_loss": 3.5826344034709456, + "tokens_seen": 1217724416 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003187261785356068, + "loss": 1.3243, + "theoretical_loss": 3.5826167168459144, + "tokens_seen": 1217789952 + }, + { + "epoch": 0.37, + "learning_rate": 0.000318716148445336, + "loss": 1.2849, + "theoretical_loss": 3.5825990314391642, + "tokens_seen": 1217855488 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003187061183550652, + "loss": 1.4371, + "theoretical_loss": 3.5825813472505463, + "tokens_seen": 1217921024 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003186960882647944, + "loss": 1.2154, + "theoretical_loss": 3.58256366427991, + "tokens_seen": 1217986560 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031868605817452357, + "loss": 1.3569, + "theoretical_loss": 3.582545982527107, + "tokens_seen": 1218052096 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003186760280842528, + "loss": 1.4424, + "theoretical_loss": 3.582528301991987, + "tokens_seen": 1218117632 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031866599799398193, + "loss": 1.3186, + "theoretical_loss": 3.582510622674401, + "tokens_seen": 1218183168 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031865596790371116, + "loss": 1.3954, + "theoretical_loss": 3.5824929445742004, + "tokens_seen": 1218248704 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003186459378134403, + "loss": 1.3805, + "theoretical_loss": 3.5824752676912346, + "tokens_seen": 1218314240 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003186359077231695, + "loss": 1.2844, + "theoretical_loss": 3.582457592025355, + "tokens_seen": 1218379776 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003186258776328987, + "loss": 1.4198, + "theoretical_loss": 3.5824399175764126, + "tokens_seen": 1218445312 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003186158475426279, + "loss": 1.3241, + "theoretical_loss": 3.582422244344258, + "tokens_seen": 1218510848 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031860581745235707, + "loss": 1.3226, + "theoretical_loss": 3.582404572328741, + "tokens_seen": 1218576384 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031859578736208625, + "loss": 1.3995, + "theoretical_loss": 3.582386901529714, + "tokens_seen": 1218641920 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031858575727181543, + "loss": 1.4294, + "theoretical_loss": 3.582369231947027, + "tokens_seen": 1218707456 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031857572718154467, + "loss": 1.3922, + "theoretical_loss": 3.5823515635805308, + "tokens_seen": 1218772992 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003185656970912738, + "loss": 1.3738, + "theoretical_loss": 3.582333896430076, + "tokens_seen": 1218838528 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031855566700100303, + "loss": 1.3837, + "theoretical_loss": 3.5823162304955147, + "tokens_seen": 1218904064 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -8.4747611253988e-05, + "objective/train/docs_used": 692060, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8726673126220703, + "objective/train/original_loss": 2.872667074203491, + "objective/train/theoretical_loss": 3.5822985657766973, + "objective/train/tokens_used": 1239429600, + "objective/train/value_avg": -0.00518798828125, + "objective/train/value_loss": 6.380466220434755e-05, + "objective/train/value_max": -0.00033545494079589844, + "objective/train/value_min": -0.051544189453125, + "objective/train/value_reward_corr": 0.31449631469133993, + "objective/train/value_std": 0.003787994384765625, + "objective/train/weight_avg": 1.0020451545715332, + "objective/train/weighted_lm_loss": 2.863325595855713, + "objective/train/weights_max": 1.6126002073287964, + "objective/train/weights_min": 0.1392100602388382, + "theoretical_loss": 3.5822985657766973, + "tokens_seen": 1218969600 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003185456369107322, + "loss": 1.3693, + "theoretical_loss": 3.5822985657766973, + "tokens_seen": 1218969600 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003185356068204614, + "loss": 1.2991, + "theoretical_loss": 3.5822809022734745, + "tokens_seen": 1219035136 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031852557673019057, + "loss": 1.2008, + "theoretical_loss": 3.582263239985697, + "tokens_seen": 1219100672 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031851554663991975, + "loss": 1.3441, + "theoretical_loss": 3.5822455789132164, + "tokens_seen": 1219166208 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031850551654964893, + "loss": 1.3962, + "theoretical_loss": 3.582227919055884, + "tokens_seen": 1219231744 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031849548645937817, + "loss": 1.3348, + "theoretical_loss": 3.58221026041355, + "tokens_seen": 1219297280 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003184854563691073, + "loss": 1.3253, + "theoretical_loss": 3.5821926029860665, + "tokens_seen": 1219362816 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031847542627883653, + "loss": 1.3982, + "theoretical_loss": 3.582174946773284, + "tokens_seen": 1219428352 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031846539618856566, + "loss": 1.3497, + "theoretical_loss": 3.5821572917750535, + "tokens_seen": 1219493888 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003184553660982949, + "loss": 1.3134, + "theoretical_loss": 3.582139637991227, + "tokens_seen": 1219559424 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003184453360080241, + "loss": 1.3873, + "theoretical_loss": 3.5821219854216553, + "tokens_seen": 1219624960 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031843530591775326, + "loss": 1.3488, + "theoretical_loss": 3.582104334066189, + "tokens_seen": 1219690496 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031842527582748244, + "loss": 1.416, + "theoretical_loss": 3.5820866839246803, + "tokens_seen": 1219756032 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003184152457372116, + "loss": 1.3971, + "theoretical_loss": 3.58206903499698, + "tokens_seen": 1219821568 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003184052156469408, + "loss": 1.2769, + "theoretical_loss": 3.58205138728294, + "tokens_seen": 1219887104 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031839518555667004, + "loss": 1.2407, + "theoretical_loss": 3.5820337407824105, + "tokens_seen": 1219952640 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031838515546639916, + "loss": 1.3723, + "theoretical_loss": 3.582016095495244, + "tokens_seen": 1220018176 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003183751253761284, + "loss": 1.4449, + "theoretical_loss": 3.581998451421291, + "tokens_seen": 1220083712 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003183650952858576, + "loss": 1.4487, + "theoretical_loss": 3.581980808560404, + "tokens_seen": 1220149248 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031835506519558676, + "loss": 1.3016, + "theoretical_loss": 3.581963166912433, + "tokens_seen": 1220214784 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031834503510531594, + "loss": 1.3757, + "theoretical_loss": 3.5819455264772317, + "tokens_seen": 1220280320 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003183350050150451, + "loss": 1.3251, + "theoretical_loss": 3.581927887254649, + "tokens_seen": 1220345856 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003183249749247743, + "loss": 1.3521, + "theoretical_loss": 3.581910249244538, + "tokens_seen": 1220411392 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031831494483450354, + "loss": 1.3922, + "theoretical_loss": 3.58189261244675, + "tokens_seen": 1220476928 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003183049147442327, + "loss": 1.4098, + "theoretical_loss": 3.5818749768611364, + "tokens_seen": 1220542464 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": 0.000741469906643033, + "objective/train/docs_used": 693613, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.711808443069458, + "objective/train/original_loss": 2.711808204650879, + "objective/train/theoretical_loss": 3.581857342487549, + "objective/train/tokens_used": 1241068000, + "objective/train/value_avg": -0.004756927490234375, + "objective/train/value_loss": 3.8503585528815165e-05, + "objective/train/value_max": -0.00047278404235839844, + "objective/train/value_min": -0.047149658203125, + "objective/train/value_reward_corr": 0.2688895161438349, + "objective/train/value_std": 0.00341033935546875, + "objective/train/weight_avg": 1.0092328786849976, + "objective/train/weighted_lm_loss": 2.7423975467681885, + "objective/train/weights_max": 1.4512982368469238, + "objective/train/weights_min": 0.3706868290901184, + "theoretical_loss": 3.581857342487549, + "tokens_seen": 1220608000 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003182948846539619, + "loss": 1.2811, + "theoretical_loss": 3.581857342487549, + "tokens_seen": 1220608000 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003182848545636911, + "loss": 1.4098, + "theoretical_loss": 3.5818397093258394, + "tokens_seen": 1220673536 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031827482447342026, + "loss": 1.3866, + "theoretical_loss": 3.5818220773758584, + "tokens_seen": 1220739072 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003182647943831495, + "loss": 1.3328, + "theoretical_loss": 3.5818044466374594, + "tokens_seen": 1220804608 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003182547642928786, + "loss": 1.4364, + "theoretical_loss": 3.581786817110493, + "tokens_seen": 1220870144 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031824473420260786, + "loss": 1.301, + "theoretical_loss": 3.581769188794811, + "tokens_seen": 1220935680 + }, + { + "epoch": 0.37, + "learning_rate": 0.000318234704112337, + "loss": 1.3735, + "theoretical_loss": 3.581751561690265, + "tokens_seen": 1221001216 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003182246740220662, + "loss": 1.3168, + "theoretical_loss": 3.5817339357967075, + "tokens_seen": 1221066752 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003182146439317954, + "loss": 1.4286, + "theoretical_loss": 3.58171631111399, + "tokens_seen": 1221132288 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003182046138415246, + "loss": 1.2236, + "theoretical_loss": 3.5816986876419636, + "tokens_seen": 1221197824 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031819458375125377, + "loss": 1.2893, + "theoretical_loss": 3.581681065380481, + "tokens_seen": 1221263360 + }, + { + "epoch": 0.37, + "learning_rate": 0.000318184553660983, + "loss": 1.377, + "theoretical_loss": 3.5816634443293944, + "tokens_seen": 1221328896 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031817452357071213, + "loss": 1.4552, + "theoretical_loss": 3.5816458244885547, + "tokens_seen": 1221394432 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031816449348044136, + "loss": 1.3556, + "theoretical_loss": 3.5816282058578146, + "tokens_seen": 1221459968 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003181544633901705, + "loss": 1.4587, + "theoretical_loss": 3.5816105884370257, + "tokens_seen": 1221525504 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003181444332998997, + "loss": 1.4022, + "theoretical_loss": 3.5815929722260402, + "tokens_seen": 1221591040 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003181344032096289, + "loss": 1.4332, + "theoretical_loss": 3.58157535722471, + "tokens_seen": 1221656576 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003181243731193581, + "loss": 1.3187, + "theoretical_loss": 3.5815577434328874, + "tokens_seen": 1221722112 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031811434302908727, + "loss": 1.3497, + "theoretical_loss": 3.581540130850424, + "tokens_seen": 1221787648 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031810431293881645, + "loss": 1.3501, + "theoretical_loss": 3.5815225194771725, + "tokens_seen": 1221853184 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031809428284854563, + "loss": 1.4958, + "theoretical_loss": 3.5815049093129847, + "tokens_seen": 1221918720 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031808425275827487, + "loss": 1.4351, + "theoretical_loss": 3.581487300357712, + "tokens_seen": 1221984256 + }, + { + "epoch": 0.37, + "learning_rate": 0.000318074222668004, + "loss": 1.4443, + "theoretical_loss": 3.5814696926112086, + "tokens_seen": 1222049792 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031806419257773323, + "loss": 1.3537, + "theoretical_loss": 3.5814520860733245, + "tokens_seen": 1222115328 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003180541624874624, + "loss": 1.3937, + "theoretical_loss": 3.5814344807439134, + "tokens_seen": 1222180864 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.0010129016591235995, + "objective/train/docs_used": 694027, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.624931812286377, + "objective/train/original_loss": 2.624931812286377, + "objective/train/theoretical_loss": 3.5814168766228267, + "objective/train/tokens_used": 1242706400, + "objective/train/value_avg": -0.005615234375, + "objective/train/value_loss": 7.41843759897165e-05, + "objective/train/value_max": -0.000469207763671875, + "objective/train/value_min": -0.11614990234375, + "objective/train/value_reward_corr": 0.41243929722318556, + "objective/train/value_std": 0.00460052490234375, + "objective/train/weight_avg": 0.993318498134613, + "objective/train/weighted_lm_loss": 2.6156935691833496, + "objective/train/weights_max": 2.3469789028167725, + "objective/train/weights_min": 0.41002246737480164, + "theoretical_loss": 3.5814168766228267, + "tokens_seen": 1222246400 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003180441323971916, + "loss": 1.3425, + "theoretical_loss": 3.5814168766228267, + "tokens_seen": 1222246400 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031803410230692077, + "loss": 1.3208, + "theoretical_loss": 3.581399273709917, + "tokens_seen": 1222311936 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031802407221664995, + "loss": 1.2826, + "theoretical_loss": 3.5813816720050373, + "tokens_seen": 1222377472 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031801404212637913, + "loss": 1.3568, + "theoretical_loss": 3.581364071508039, + "tokens_seen": 1222443008 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031800401203610837, + "loss": 1.3502, + "theoretical_loss": 3.5813464722187742, + "tokens_seen": 1222508544 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003179939819458375, + "loss": 1.3131, + "theoretical_loss": 3.5813288741370966, + "tokens_seen": 1222574080 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031798395185556673, + "loss": 1.292, + "theoretical_loss": 3.5813112772628575, + "tokens_seen": 1222639616 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031797392176529586, + "loss": 1.2884, + "theoretical_loss": 3.5812936815959096, + "tokens_seen": 1222705152 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003179638916750251, + "loss": 1.2853, + "theoretical_loss": 3.5812760871361053, + "tokens_seen": 1222770688 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003179538615847543, + "loss": 1.3253, + "theoretical_loss": 3.581258493883298, + "tokens_seen": 1222836224 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031794383149448346, + "loss": 1.3866, + "theoretical_loss": 3.581240901837339, + "tokens_seen": 1222901760 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031793380140421264, + "loss": 1.3422, + "theoretical_loss": 3.581223310998081, + "tokens_seen": 1222967296 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003179237713139418, + "loss": 1.4113, + "theoretical_loss": 3.5812057213653774, + "tokens_seen": 1223032832 + }, + { + "epoch": 0.37, + "learning_rate": 0.000317913741223671, + "loss": 1.4161, + "theoretical_loss": 3.5811881329390802, + "tokens_seen": 1223098368 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031790371113340024, + "loss": 1.2735, + "theoretical_loss": 3.5811705457190417, + "tokens_seen": 1223163904 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031789368104312936, + "loss": 1.4545, + "theoretical_loss": 3.5811529597051153, + "tokens_seen": 1223229440 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003178836509528586, + "loss": 1.3539, + "theoretical_loss": 3.581135374897153, + "tokens_seen": 1223294976 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003178736208625878, + "loss": 1.3585, + "theoretical_loss": 3.5811177912950085, + "tokens_seen": 1223360512 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031786359077231696, + "loss": 1.4634, + "theoretical_loss": 3.5811002088985333, + "tokens_seen": 1223426048 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031785356068204614, + "loss": 1.2685, + "theoretical_loss": 3.5810826277075805, + "tokens_seen": 1223491584 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003178435305917753, + "loss": 1.4659, + "theoretical_loss": 3.581065047722004, + "tokens_seen": 1223557120 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003178335005015045, + "loss": 1.3347, + "theoretical_loss": 3.581047468941655, + "tokens_seen": 1223622656 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031782347041123374, + "loss": 1.447, + "theoretical_loss": 3.581029891366387, + "tokens_seen": 1223688192 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031781344032096287, + "loss": 1.4274, + "theoretical_loss": 3.581012314996053, + "tokens_seen": 1223753728 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003178034102306921, + "loss": 1.3703, + "theoretical_loss": 3.5809947398305058, + "tokens_seen": 1223819264 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": 0.0010754793183878064, + "objective/train/docs_used": 695222, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5445151329040527, + "objective/train/original_loss": 2.5445146560668945, + "objective/train/theoretical_loss": 3.580977165869598, + "objective/train/tokens_used": 1244344800, + "objective/train/value_avg": -0.004791259765625, + "objective/train/value_loss": 3.175079109496437e-05, + "objective/train/value_max": -0.0005593299865722656, + "objective/train/value_min": -0.0384521484375, + "objective/train/value_reward_corr": 0.37910044408530075, + "objective/train/value_std": 0.0034503936767578125, + "objective/train/weight_avg": 1.0122967958450317, + "objective/train/weighted_lm_loss": 2.5695741176605225, + "objective/train/weights_max": 1.3156791925430298, + "objective/train/weights_min": 0.5835326313972473, + "theoretical_loss": 3.580977165869598, + "tokens_seen": 1223884800 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031779338014042123, + "loss": 1.4463, + "theoretical_loss": 3.580977165869598, + "tokens_seen": 1223884800 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031778335005015046, + "loss": 1.37, + "theoretical_loss": 3.580959593113183, + "tokens_seen": 1223950336 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031777331995987964, + "loss": 1.438, + "theoretical_loss": 3.580942021561113, + "tokens_seen": 1224015872 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003177632898696088, + "loss": 1.3555, + "theoretical_loss": 3.5809244512132423, + "tokens_seen": 1224081408 + }, + { + "epoch": 0.37, + "learning_rate": 0.000317753259779338, + "loss": 1.4237, + "theoretical_loss": 3.5809068820694234, + "tokens_seen": 1224146944 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003177432296890672, + "loss": 1.4524, + "theoretical_loss": 3.580889314129508, + "tokens_seen": 1224212480 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031773319959879637, + "loss": 1.4675, + "theoretical_loss": 3.580871747393351, + "tokens_seen": 1224278016 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003177231695085256, + "loss": 1.2935, + "theoretical_loss": 3.5808541818608046, + "tokens_seen": 1224343552 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031771313941825473, + "loss": 1.371, + "theoretical_loss": 3.5808366175317223, + "tokens_seen": 1224409088 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031770310932798397, + "loss": 1.4785, + "theoretical_loss": 3.580819054405956, + "tokens_seen": 1224474624 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031769307923771315, + "loss": 1.4708, + "theoretical_loss": 3.580801492483361, + "tokens_seen": 1224540160 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031768304914744233, + "loss": 1.3791, + "theoretical_loss": 3.580783931763789, + "tokens_seen": 1224605696 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003176730190571715, + "loss": 1.3025, + "theoretical_loss": 3.5807663722470933, + "tokens_seen": 1224671232 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003176629889669007, + "loss": 1.3591, + "theoretical_loss": 3.5807488139331274, + "tokens_seen": 1224736768 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031765295887662987, + "loss": 1.2894, + "theoretical_loss": 3.5807312568217444, + "tokens_seen": 1224802304 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003176429287863591, + "loss": 1.3355, + "theoretical_loss": 3.5807137009127983, + "tokens_seen": 1224867840 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031763289869608823, + "loss": 1.2904, + "theoretical_loss": 3.5806961462061415, + "tokens_seen": 1224933376 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031762286860581747, + "loss": 1.4133, + "theoretical_loss": 3.580678592701628, + "tokens_seen": 1224998912 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003176128385155466, + "loss": 1.3377, + "theoretical_loss": 3.5806610403991104, + "tokens_seen": 1225064448 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031760280842527583, + "loss": 1.4338, + "theoretical_loss": 3.580643489298443, + "tokens_seen": 1225129984 + }, + { + "epoch": 0.37, + "learning_rate": 0.000317592778335005, + "loss": 1.2301, + "theoretical_loss": 3.5806259393994786, + "tokens_seen": 1225195520 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003175827482447342, + "loss": 1.3918, + "theoretical_loss": 3.580608390702071, + "tokens_seen": 1225261056 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003175727181544634, + "loss": 1.4053, + "theoretical_loss": 3.5805908432060733, + "tokens_seen": 1225326592 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003175626880641926, + "loss": 1.3004, + "theoretical_loss": 3.5805732969113393, + "tokens_seen": 1225392128 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003175526579739218, + "loss": 1.3706, + "theoretical_loss": 3.5805557518177227, + "tokens_seen": 1225457664 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.0004101573722437024, + "objective/train/docs_used": 695595, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5151705741882324, + "objective/train/original_loss": 2.5151705741882324, + "objective/train/theoretical_loss": 3.580538207925077, + "objective/train/tokens_used": 1245983200, + "objective/train/value_avg": -0.004566192626953125, + "objective/train/value_loss": 6.143997597973794e-05, + "objective/train/value_max": -0.0001150369644165039, + "objective/train/value_min": -0.061431884765625, + "objective/train/value_reward_corr": 0.41489735917437875, + "objective/train/value_std": 0.003932952880859375, + "objective/train/weight_avg": 0.998752236366272, + "objective/train/weighted_lm_loss": 2.5123932361602783, + "objective/train/weights_max": 1.7536288499832153, + "objective/train/weights_min": 0.2929016351699829, + "theoretical_loss": 3.580538207925077, + "tokens_seen": 1225523200 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031754262788365097, + "loss": 1.3061, + "theoretical_loss": 3.580538207925077, + "tokens_seen": 1225523200 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031753259779338015, + "loss": 1.2974, + "theoretical_loss": 3.580520665233255, + "tokens_seen": 1225588736 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031752256770310933, + "loss": 1.3988, + "theoretical_loss": 3.580503123742111, + "tokens_seen": 1225654272 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031751253761283857, + "loss": 1.3602, + "theoretical_loss": 3.5804855834514986, + "tokens_seen": 1225719808 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003175025075225677, + "loss": 1.4199, + "theoretical_loss": 3.5804680443612718, + "tokens_seen": 1225785344 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031749247743229693, + "loss": 1.4056, + "theoretical_loss": 3.580450506471283, + "tokens_seen": 1225850880 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031748244734202606, + "loss": 1.3709, + "theoretical_loss": 3.5804329697813873, + "tokens_seen": 1225916416 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003174724172517553, + "loss": 1.304, + "theoretical_loss": 3.580415434291438, + "tokens_seen": 1225981952 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003174623871614845, + "loss": 1.4001, + "theoretical_loss": 3.580397900001288, + "tokens_seen": 1226047488 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031745235707121366, + "loss": 1.4246, + "theoretical_loss": 3.5803803669107923, + "tokens_seen": 1226113024 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031744232698094284, + "loss": 1.3665, + "theoretical_loss": 3.5803628350198045, + "tokens_seen": 1226178560 + }, + { + "epoch": 0.37, + "learning_rate": 0.000317432296890672, + "loss": 1.3705, + "theoretical_loss": 3.580345304328178, + "tokens_seen": 1226244096 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003174222668004012, + "loss": 1.3116, + "theoretical_loss": 3.5803277748357667, + "tokens_seen": 1226309632 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031741223671013044, + "loss": 1.3952, + "theoretical_loss": 3.5803102465424246, + "tokens_seen": 1226375168 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031740220661985956, + "loss": 1.3012, + "theoretical_loss": 3.580292719448005, + "tokens_seen": 1226440704 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003173921765295888, + "loss": 1.3966, + "theoretical_loss": 3.580275193552364, + "tokens_seen": 1226506240 + }, + { + "epoch": 0.37, + "learning_rate": 0.000317382146439318, + "loss": 1.4098, + "theoretical_loss": 3.5802576688553525, + "tokens_seen": 1226571776 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031737211634904716, + "loss": 1.3641, + "theoretical_loss": 3.580240145356827, + "tokens_seen": 1226637312 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031736208625877634, + "loss": 1.4955, + "theoretical_loss": 3.58022262305664, + "tokens_seen": 1226702848 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003173520561685055, + "loss": 1.3692, + "theoretical_loss": 3.580205101954646, + "tokens_seen": 1226768384 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003173420260782347, + "loss": 1.3645, + "theoretical_loss": 3.5801875820506988, + "tokens_seen": 1226833920 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031733199598796394, + "loss": 1.3375, + "theoretical_loss": 3.580170063344654, + "tokens_seen": 1226899456 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031732196589769307, + "loss": 1.4053, + "theoretical_loss": 3.5801525458363637, + "tokens_seen": 1226964992 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003173119358074223, + "loss": 1.4142, + "theoretical_loss": 3.5801350295256826, + "tokens_seen": 1227030528 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031730190571715143, + "loss": 1.4161, + "theoretical_loss": 3.580117514412465, + "tokens_seen": 1227096064 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.001018648617900908, + "objective/train/docs_used": 696474, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.745933771133423, + "objective/train/original_loss": 2.7459330558776855, + "objective/train/theoretical_loss": 3.580100000496566, + "objective/train/tokens_used": 1247621600, + "objective/train/value_avg": -0.006130218505859375, + "objective/train/value_loss": 8.760880882618949e-05, + "objective/train/value_max": -0.00037860870361328125, + "objective/train/value_min": -0.11737060546875, + "objective/train/value_reward_corr": 0.4163831762552819, + "objective/train/value_std": 0.004825592041015625, + "objective/train/weight_avg": 0.993909478187561, + "objective/train/weighted_lm_loss": 2.7344775199890137, + "objective/train/weights_max": 2.670151472091675, + "objective/train/weights_min": 0.4332253634929657, + "theoretical_loss": 3.580100000496566, + "tokens_seen": 1227161600 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031729187562688066, + "loss": 1.3909, + "theoretical_loss": 3.580100000496566, + "tokens_seen": 1227161600 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031728184553660984, + "loss": 1.3909, + "theoretical_loss": 3.5800824877778386, + "tokens_seen": 1227227136 + }, + { + "epoch": 0.37, + "learning_rate": 0.000317271815446339, + "loss": 1.3954, + "theoretical_loss": 3.5800649762561374, + "tokens_seen": 1227292672 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003172617853560682, + "loss": 1.475, + "theoretical_loss": 3.580047465931317, + "tokens_seen": 1227358208 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003172517552657974, + "loss": 1.4925, + "theoretical_loss": 3.580029956803231, + "tokens_seen": 1227423744 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031724172517552657, + "loss": 1.3909, + "theoretical_loss": 3.5800124488717344, + "tokens_seen": 1227489280 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003172316950852558, + "loss": 1.3855, + "theoretical_loss": 3.5799949421366817, + "tokens_seen": 1227554816 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031722166499498493, + "loss": 1.373, + "theoretical_loss": 3.579977436597926, + "tokens_seen": 1227620352 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031721163490471417, + "loss": 1.3845, + "theoretical_loss": 3.5799599322553233, + "tokens_seen": 1227685888 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031720160481444335, + "loss": 1.2003, + "theoretical_loss": 3.579942429108727, + "tokens_seen": 1227751424 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031719157472417253, + "loss": 1.4912, + "theoretical_loss": 3.5799249271579923, + "tokens_seen": 1227816960 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003171815446339017, + "loss": 1.3158, + "theoretical_loss": 3.579907426402972, + "tokens_seen": 1227882496 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003171715145436309, + "loss": 1.3989, + "theoretical_loss": 3.579889926843523, + "tokens_seen": 1227948032 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031716148445336007, + "loss": 1.2959, + "theoretical_loss": 3.579872428479498, + "tokens_seen": 1228013568 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003171514543630893, + "loss": 1.4587, + "theoretical_loss": 3.579854931310752, + "tokens_seen": 1228079104 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031714142427281843, + "loss": 1.3256, + "theoretical_loss": 3.57983743533714, + "tokens_seen": 1228144640 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031713139418254767, + "loss": 1.3046, + "theoretical_loss": 3.579819940558517, + "tokens_seen": 1228210176 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003171213640922768, + "loss": 1.4185, + "theoretical_loss": 3.5798024469747363, + "tokens_seen": 1228275712 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031711133400200603, + "loss": 1.4176, + "theoretical_loss": 3.5797849545856533, + "tokens_seen": 1228341248 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003171013039117352, + "loss": 1.501, + "theoretical_loss": 3.5797674633911223, + "tokens_seen": 1228406784 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003170912738214644, + "loss": 1.359, + "theoretical_loss": 3.5797499733909985, + "tokens_seen": 1228472320 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003170812437311936, + "loss": 1.3559, + "theoretical_loss": 3.5797324845851364, + "tokens_seen": 1228537856 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003170712136409228, + "loss": 1.4532, + "theoretical_loss": 3.579714996973391, + "tokens_seen": 1228603392 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031706118355065194, + "loss": 1.4402, + "theoretical_loss": 3.579697510555617, + "tokens_seen": 1228668928 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003170511534603812, + "loss": 1.2754, + "theoretical_loss": 3.579680025331668, + "tokens_seen": 1228734464 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": 0.0001489095011493191, + "objective/train/docs_used": 696846, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.010443925857544, + "objective/train/original_loss": 3.010443925857544, + "objective/train/theoretical_loss": 3.579662541301401, + "objective/train/tokens_used": 1249260000, + "objective/train/value_avg": -0.004802703857421875, + "objective/train/value_loss": 5.160841828910634e-05, + "objective/train/value_max": -0.00017404556274414062, + "objective/train/value_min": -0.0810546875, + "objective/train/value_reward_corr": 0.35504073953751725, + "objective/train/value_std": 0.004360198974609375, + "objective/train/weight_avg": 1.0039401054382324, + "objective/train/weighted_lm_loss": 3.0245659351348877, + "objective/train/weights_max": 2.2491376399993896, + "objective/train/weights_min": 0.38409870862960815, + "theoretical_loss": 3.579662541301401, + "tokens_seen": 1228800000 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003170411233701103, + "loss": 1.508, + "theoretical_loss": 3.579662541301401, + "tokens_seen": 1228800000 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031703109327983954, + "loss": 1.2806, + "theoretical_loss": 3.579645058464669, + "tokens_seen": 1228865536 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003170210631895687, + "loss": 1.387, + "theoretical_loss": 3.579627576821328, + "tokens_seen": 1228931072 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003170110330992979, + "loss": 1.3745, + "theoretical_loss": 3.5796100963712325, + "tokens_seen": 1228996608 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003170010030090271, + "loss": 1.2673, + "theoretical_loss": 3.579592617114237, + "tokens_seen": 1229062144 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031699097291875626, + "loss": 1.3664, + "theoretical_loss": 3.5795751390501973, + "tokens_seen": 1229127680 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031698094282848544, + "loss": 1.2902, + "theoretical_loss": 3.579557662178968, + "tokens_seen": 1229193216 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003169709127382147, + "loss": 1.2245, + "theoretical_loss": 3.579540186500404, + "tokens_seen": 1229258752 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003169608826479438, + "loss": 1.4666, + "theoretical_loss": 3.5795227120143607, + "tokens_seen": 1229324288 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031695085255767304, + "loss": 1.3804, + "theoretical_loss": 3.5795052387206923, + "tokens_seen": 1229389824 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031694082246740217, + "loss": 1.2557, + "theoretical_loss": 3.5794877666192546, + "tokens_seen": 1229455360 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003169307923771314, + "loss": 1.2399, + "theoretical_loss": 3.579470295709903, + "tokens_seen": 1229520896 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003169207622868606, + "loss": 1.3812, + "theoretical_loss": 3.579452825992492, + "tokens_seen": 1229586432 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031691073219658976, + "loss": 1.3948, + "theoretical_loss": 3.5794353574668767, + "tokens_seen": 1229651968 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031690070210631894, + "loss": 1.3759, + "theoretical_loss": 3.5794178901329126, + "tokens_seen": 1229717504 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003168906720160482, + "loss": 1.3785, + "theoretical_loss": 3.579400423990455, + "tokens_seen": 1229783040 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003168806419257773, + "loss": 1.4316, + "theoretical_loss": 3.579382959039359, + "tokens_seen": 1229848576 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031687061183550654, + "loss": 1.3626, + "theoretical_loss": 3.57936549527948, + "tokens_seen": 1229914112 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031686058174523567, + "loss": 1.3666, + "theoretical_loss": 3.579348032710672, + "tokens_seen": 1229979648 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003168505516549649, + "loss": 1.4512, + "theoretical_loss": 3.5793305713327923, + "tokens_seen": 1230045184 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003168405215646941, + "loss": 1.3412, + "theoretical_loss": 3.5793131111456953, + "tokens_seen": 1230110720 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031683049147442327, + "loss": 1.3721, + "theoretical_loss": 3.579295652149237, + "tokens_seen": 1230176256 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031682046138415245, + "loss": 1.4702, + "theoretical_loss": 3.579278194343271, + "tokens_seen": 1230241792 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031681043129388163, + "loss": 1.3519, + "theoretical_loss": 3.5792607377276546, + "tokens_seen": 1230307328 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031680040120361086, + "loss": 1.3898, + "theoretical_loss": 3.579243282302243, + "tokens_seen": 1230372864 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.00042161348392255604, + "objective/train/docs_used": 698095, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8575563430786133, + "objective/train/original_loss": 2.857556104660034, + "objective/train/theoretical_loss": 3.5792258280668903, + "objective/train/tokens_used": 1250898400, + "objective/train/value_avg": -0.00467681884765625, + "objective/train/value_loss": 5.129772034706548e-05, + "objective/train/value_max": -0.0002868175506591797, + "objective/train/value_min": -0.06549072265625, + "objective/train/value_reward_corr": 0.45060839324361224, + "objective/train/value_std": 0.004055023193359375, + "objective/train/weight_avg": 0.9982052445411682, + "objective/train/weighted_lm_loss": 2.854168653488159, + "objective/train/weights_max": 1.4260451793670654, + "objective/train/weights_min": 0.4534648060798645, + "theoretical_loss": 3.5792258280668903, + "tokens_seen": 1230438400 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031679037111334004, + "loss": 1.3422, + "theoretical_loss": 3.5792258280668903, + "tokens_seen": 1230438400 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003167803410230692, + "loss": 1.4648, + "theoretical_loss": 3.5792083750214534, + "tokens_seen": 1230503936 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003167703109327984, + "loss": 1.3274, + "theoretical_loss": 3.579190923165787, + "tokens_seen": 1230569472 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003167602808425276, + "loss": 1.3303, + "theoretical_loss": 3.5791734724997473, + "tokens_seen": 1230635008 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031675025075225677, + "loss": 1.3954, + "theoretical_loss": 3.5791560230231894, + "tokens_seen": 1230700544 + }, + { + "epoch": 0.37, + "learning_rate": 0.000316740220661986, + "loss": 1.4795, + "theoretical_loss": 3.5791385747359685, + "tokens_seen": 1230766080 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031673019057171513, + "loss": 1.4871, + "theoretical_loss": 3.5791211276379418, + "tokens_seen": 1230831616 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031672016048144437, + "loss": 1.3545, + "theoretical_loss": 3.579103681728963, + "tokens_seen": 1230897152 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031671013039117355, + "loss": 1.2662, + "theoretical_loss": 3.5790862370088887, + "tokens_seen": 1230962688 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031670010030090273, + "loss": 1.359, + "theoretical_loss": 3.5790687934775747, + "tokens_seen": 1231028224 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003166900702106319, + "loss": 1.4622, + "theoretical_loss": 3.5790513511348765, + "tokens_seen": 1231093760 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003166800401203611, + "loss": 1.1567, + "theoretical_loss": 3.57903390998065, + "tokens_seen": 1231159296 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031667001003009027, + "loss": 1.3895, + "theoretical_loss": 3.57901647001475, + "tokens_seen": 1231224832 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003166599799398195, + "loss": 1.4459, + "theoretical_loss": 3.578999031237034, + "tokens_seen": 1231290368 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031664994984954863, + "loss": 1.2638, + "theoretical_loss": 3.578981593647357, + "tokens_seen": 1231355904 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031663991975927787, + "loss": 1.3382, + "theoretical_loss": 3.578964157245574, + "tokens_seen": 1231421440 + }, + { + "epoch": 0.37, + "learning_rate": 0.000316629889669007, + "loss": 1.4272, + "theoretical_loss": 3.578946722031542, + "tokens_seen": 1231486976 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031661985957873623, + "loss": 1.3195, + "theoretical_loss": 3.5789292880051162, + "tokens_seen": 1231552512 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003166098294884654, + "loss": 1.2773, + "theoretical_loss": 3.578911855166153, + "tokens_seen": 1231618048 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003165997993981946, + "loss": 1.3762, + "theoretical_loss": 3.5788944235145084, + "tokens_seen": 1231683584 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003165897693079238, + "loss": 1.3982, + "theoretical_loss": 3.578876993050038, + "tokens_seen": 1231749120 + }, + { + "epoch": 0.37, + "learning_rate": 0.000316579739217653, + "loss": 1.4415, + "theoretical_loss": 3.578859563772598, + "tokens_seen": 1231814656 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031656970912738214, + "loss": 1.4184, + "theoretical_loss": 3.578842135682044, + "tokens_seen": 1231880192 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003165596790371114, + "loss": 1.4436, + "theoretical_loss": 3.5788247087782326, + "tokens_seen": 1231945728 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003165496489468405, + "loss": 1.4164, + "theoretical_loss": 3.57880728306102, + "tokens_seen": 1232011264 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": 6.347503222059458e-05, + "objective/train/docs_used": 698724, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7014851570129395, + "objective/train/original_loss": 2.7014851570129395, + "objective/train/theoretical_loss": 3.5787898585302615, + "objective/train/tokens_used": 1252536800, + "objective/train/value_avg": -0.005489349365234375, + "objective/train/value_loss": 8.477332448819652e-05, + "objective/train/value_max": -0.00017535686492919922, + "objective/train/value_min": -0.08062744140625, + "objective/train/value_reward_corr": 0.19778647423405407, + "objective/train/value_std": 0.0038509368896484375, + "objective/train/weight_avg": 1.0044833421707153, + "objective/train/weighted_lm_loss": 2.717369556427002, + "objective/train/weights_max": 1.8019747734069824, + "objective/train/weights_min": 0.35179829597473145, + "theoretical_loss": 3.5787898585302615, + "tokens_seen": 1232076800 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031653961885656974, + "loss": 1.2968, + "theoretical_loss": 3.5787898585302615, + "tokens_seen": 1232076800 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003165295887662989, + "loss": 1.2304, + "theoretical_loss": 3.5787724351858134, + "tokens_seen": 1232142336 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003165195586760281, + "loss": 1.4251, + "theoretical_loss": 3.578755013027533, + "tokens_seen": 1232207872 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003165095285857573, + "loss": 1.4112, + "theoretical_loss": 3.5787375920552744, + "tokens_seen": 1232273408 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031649949849548646, + "loss": 1.3018, + "theoretical_loss": 3.5787201722688957, + "tokens_seen": 1232338944 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031648946840521564, + "loss": 1.417, + "theoretical_loss": 3.578702753668252, + "tokens_seen": 1232404480 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003164794383149449, + "loss": 1.3618, + "theoretical_loss": 3.5786853362532005, + "tokens_seen": 1232470016 + }, + { + "epoch": 0.37, + "learning_rate": 0.000316469408224674, + "loss": 1.4461, + "theoretical_loss": 3.5786679200235962, + "tokens_seen": 1232535552 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031645937813440324, + "loss": 1.3389, + "theoretical_loss": 3.5786505049792963, + "tokens_seen": 1232601088 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031644934804413237, + "loss": 1.3227, + "theoretical_loss": 3.5786330911201567, + "tokens_seen": 1232666624 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003164393179538616, + "loss": 1.2758, + "theoretical_loss": 3.5786156784460346, + "tokens_seen": 1232732160 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003164292878635908, + "loss": 1.3151, + "theoretical_loss": 3.578598266956785, + "tokens_seen": 1232797696 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031641925777331996, + "loss": 1.4806, + "theoretical_loss": 3.578580856652265, + "tokens_seen": 1232863232 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031640922768304914, + "loss": 1.3912, + "theoretical_loss": 3.578563447532331, + "tokens_seen": 1232928768 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003163991975927784, + "loss": 1.3734, + "theoretical_loss": 3.57854603959684, + "tokens_seen": 1232994304 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003163891675025075, + "loss": 1.3393, + "theoretical_loss": 3.5785286328456474, + "tokens_seen": 1233059840 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031637913741223674, + "loss": 1.3697, + "theoretical_loss": 3.57851122727861, + "tokens_seen": 1233125376 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031636910732196587, + "loss": 1.3476, + "theoretical_loss": 3.5784938228955845, + "tokens_seen": 1233190912 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003163590772316951, + "loss": 1.3548, + "theoretical_loss": 3.578476419696428, + "tokens_seen": 1233256448 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003163490471414243, + "loss": 1.4109, + "theoretical_loss": 3.5784590176809963, + "tokens_seen": 1233321984 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031633901705115347, + "loss": 1.3782, + "theoretical_loss": 3.578441616849146, + "tokens_seen": 1233387520 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031632898696088265, + "loss": 1.324, + "theoretical_loss": 3.5784242172007343, + "tokens_seen": 1233453056 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031631895687061183, + "loss": 1.3504, + "theoretical_loss": 3.5784068187356173, + "tokens_seen": 1233518592 + }, + { + "epoch": 0.37, + "learning_rate": 0.000316308926780341, + "loss": 1.3631, + "theoretical_loss": 3.5783894214536516, + "tokens_seen": 1233584128 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031629889669007024, + "loss": 1.3871, + "theoretical_loss": 3.5783720253546942, + "tokens_seen": 1233649664 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": 0.00048428704030811787, + "objective/train/docs_used": 699861, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.379434108734131, + "objective/train/original_loss": 3.379434108734131, + "objective/train/theoretical_loss": 3.578354630438602, + "objective/train/tokens_used": 1254175200, + "objective/train/value_avg": -0.005352020263671875, + "objective/train/value_loss": 4.3909680243814364e-05, + "objective/train/value_max": -0.00039505958557128906, + "objective/train/value_min": -0.045257568359375, + "objective/train/value_reward_corr": 0.3703117674840526, + "objective/train/value_std": 0.0038127899169921875, + "objective/train/weight_avg": 1.006994605064392, + "objective/train/weighted_lm_loss": 3.4017093181610107, + "objective/train/weights_max": 1.5480741262435913, + "objective/train/weights_min": 0.5779279470443726, + "theoretical_loss": 3.578354630438602, + "tokens_seen": 1233715200 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031628886659979937, + "loss": 1.5263, + "theoretical_loss": 3.578354630438602, + "tokens_seen": 1233715200 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003162788365095286, + "loss": 1.4214, + "theoretical_loss": 3.578337236705231, + "tokens_seen": 1233780736 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031626880641925773, + "loss": 1.3698, + "theoretical_loss": 3.5783198441544393, + "tokens_seen": 1233846272 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031625877632898697, + "loss": 1.355, + "theoretical_loss": 3.578302452786082, + "tokens_seen": 1233911808 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031624874623871615, + "loss": 1.2991, + "theoretical_loss": 3.578285062600017, + "tokens_seen": 1233977344 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031623871614844533, + "loss": 1.4675, + "theoretical_loss": 3.578267673596101, + "tokens_seen": 1234042880 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003162286860581745, + "loss": 1.4934, + "theoretical_loss": 3.5782502857741907, + "tokens_seen": 1234108416 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031621865596790375, + "loss": 1.2976, + "theoretical_loss": 3.578232899134143, + "tokens_seen": 1234173952 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003162086258776329, + "loss": 1.3368, + "theoretical_loss": 3.578215513675815, + "tokens_seen": 1234239488 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003161985957873621, + "loss": 1.3699, + "theoretical_loss": 3.5781981293990635, + "tokens_seen": 1234305024 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031618856569709124, + "loss": 1.3643, + "theoretical_loss": 3.5781807463037456, + "tokens_seen": 1234370560 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031617853560682047, + "loss": 1.3117, + "theoretical_loss": 3.5781633643897175, + "tokens_seen": 1234436096 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031616850551654965, + "loss": 1.4475, + "theoretical_loss": 3.578145983656838, + "tokens_seen": 1234501632 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031615847542627883, + "loss": 1.3979, + "theoretical_loss": 3.578128604104962, + "tokens_seen": 1234567168 + }, + { + "epoch": 0.37, + "learning_rate": 0.000316148445336008, + "loss": 1.4089, + "theoretical_loss": 3.578111225733948, + "tokens_seen": 1234632704 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003161384152457372, + "loss": 1.3753, + "theoretical_loss": 3.5780938485436526, + "tokens_seen": 1234698240 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003161283851554664, + "loss": 1.3846, + "theoretical_loss": 3.578076472533933, + "tokens_seen": 1234763776 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003161183550651956, + "loss": 1.3824, + "theoretical_loss": 3.5780590977046467, + "tokens_seen": 1234829312 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031610832497492474, + "loss": 1.423, + "theoretical_loss": 3.57804172405565, + "tokens_seen": 1234894848 + }, + { + "epoch": 0.37, + "learning_rate": 0.000316098294884654, + "loss": 1.4001, + "theoretical_loss": 3.5780243515868007, + "tokens_seen": 1234960384 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003160882647943831, + "loss": 1.4137, + "theoretical_loss": 3.578006980297956, + "tokens_seen": 1235025920 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031607823470411234, + "loss": 1.4385, + "theoretical_loss": 3.5779896101889737, + "tokens_seen": 1235091456 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003160682046138415, + "loss": 1.304, + "theoretical_loss": 3.5779722412597095, + "tokens_seen": 1235156992 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003160581745235707, + "loss": 1.334, + "theoretical_loss": 3.5779548735100217, + "tokens_seen": 1235222528 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031604814443329994, + "loss": 1.3675, + "theoretical_loss": 3.577937506939768, + "tokens_seen": 1235288064 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.0001220099875354208, + "objective/train/docs_used": 700550, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.050438642501831, + "objective/train/original_loss": 3.050438642501831, + "objective/train/theoretical_loss": 3.577920141548805, + "objective/train/tokens_used": 1255813600, + "objective/train/value_avg": -0.00555419921875, + "objective/train/value_loss": 5.150799916009419e-05, + "objective/train/value_max": -0.0004878044128417969, + "objective/train/value_min": -0.06427001953125, + "objective/train/value_reward_corr": 0.50866845673992, + "objective/train/value_std": 0.00510406494140625, + "objective/train/weight_avg": 1.001135230064392, + "objective/train/weighted_lm_loss": 3.07585072517395, + "objective/train/weights_max": 1.5680609941482544, + "objective/train/weights_min": 0.15017783641815186, + "theoretical_loss": 3.577920141548805, + "tokens_seen": 1235353600 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003160381143430291, + "loss": 1.3646, + "theoretical_loss": 3.577920141548805, + "tokens_seen": 1235353600 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003160280842527583, + "loss": 1.3994, + "theoretical_loss": 3.5779027773369902, + "tokens_seen": 1235419136 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003160180541624875, + "loss": 1.3253, + "theoretical_loss": 3.577885414304181, + "tokens_seen": 1235484672 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031600802407221666, + "loss": 1.4473, + "theoretical_loss": 3.5778680524502353, + "tokens_seen": 1235550208 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031599799398194584, + "loss": 1.4322, + "theoretical_loss": 3.5778506917750095, + "tokens_seen": 1235615744 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003159879638916751, + "loss": 1.569, + "theoretical_loss": 3.577833332278362, + "tokens_seen": 1235681280 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003159779338014042, + "loss": 1.4123, + "theoretical_loss": 3.5778159739601505, + "tokens_seen": 1235746816 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031596790371113344, + "loss": 1.4783, + "theoretical_loss": 3.577798616820232, + "tokens_seen": 1235812352 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031595787362086257, + "loss": 1.3743, + "theoretical_loss": 3.5777812608584636, + "tokens_seen": 1235877888 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003159478435305918, + "loss": 1.4267, + "theoretical_loss": 3.5777639060747033, + "tokens_seen": 1235943424 + }, + { + "epoch": 0.37, + "learning_rate": 0.000315937813440321, + "loss": 1.2921, + "theoretical_loss": 3.577746552468809, + "tokens_seen": 1236008960 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031592778335005016, + "loss": 1.3997, + "theoretical_loss": 3.577729200040638, + "tokens_seen": 1236074496 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031591775325977934, + "loss": 1.334, + "theoretical_loss": 3.577711848790048, + "tokens_seen": 1236140032 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003159077231695086, + "loss": 1.3351, + "theoretical_loss": 3.5776944987168964, + "tokens_seen": 1236205568 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003158976930792377, + "loss": 1.35, + "theoretical_loss": 3.5776771498210413, + "tokens_seen": 1236271104 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031588766298896694, + "loss": 1.5021, + "theoretical_loss": 3.57765980210234, + "tokens_seen": 1236336640 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031587763289869607, + "loss": 1.4323, + "theoretical_loss": 3.5776424555606505, + "tokens_seen": 1236402176 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003158676028084253, + "loss": 1.4184, + "theoretical_loss": 3.5776251101958305, + "tokens_seen": 1236467712 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003158575727181545, + "loss": 1.5072, + "theoretical_loss": 3.5776077660077377, + "tokens_seen": 1236533248 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031584754262788367, + "loss": 1.3865, + "theoretical_loss": 3.5775904229962303, + "tokens_seen": 1236598784 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031583751253761285, + "loss": 1.3919, + "theoretical_loss": 3.577573081161166, + "tokens_seen": 1236664320 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031582748244734203, + "loss": 1.4384, + "theoretical_loss": 3.577555740502402, + "tokens_seen": 1236729856 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003158174523570712, + "loss": 1.3236, + "theoretical_loss": 3.5775384010197966, + "tokens_seen": 1236795392 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031580742226680044, + "loss": 1.4111, + "theoretical_loss": 3.5775210627132084, + "tokens_seen": 1236860928 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031579739217652957, + "loss": 1.3278, + "theoretical_loss": 3.5775037255824946, + "tokens_seen": 1236926464 + }, + { + "epoch": 0.37, + "objective/train/advantage_avg": -0.0005261441110633314, + "objective/train/docs_used": 701918, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.916473388671875, + "objective/train/original_loss": 2.9164726734161377, + "objective/train/theoretical_loss": 3.577486389627513, + "objective/train/tokens_used": 1257452000, + "objective/train/value_avg": -0.00539398193359375, + "objective/train/value_loss": 5.6326909543713555e-05, + "objective/train/value_max": -0.00028252601623535156, + "objective/train/value_min": -0.046630859375, + "objective/train/value_reward_corr": 0.3696607317145875, + "objective/train/value_std": 0.004238128662109375, + "objective/train/weight_avg": 0.9974172711372375, + "objective/train/weighted_lm_loss": 2.9149181842803955, + "objective/train/weights_max": 1.449613094329834, + "objective/train/weights_min": 0.42182615399360657, + "theoretical_loss": 3.577486389627513, + "tokens_seen": 1236992000 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003157873620862588, + "loss": 1.3949, + "theoretical_loss": 3.577486389627513, + "tokens_seen": 1236992000 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031577733199598793, + "loss": 1.5724, + "theoretical_loss": 3.577469054848122, + "tokens_seen": 1237057536 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031576730190571717, + "loss": 1.4861, + "theoretical_loss": 3.577451721244179, + "tokens_seen": 1237123072 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031575727181544635, + "loss": 1.3746, + "theoretical_loss": 3.577434388815543, + "tokens_seen": 1237188608 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031574724172517553, + "loss": 1.4449, + "theoretical_loss": 3.577417057562071, + "tokens_seen": 1237254144 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003157372116349047, + "loss": 1.3555, + "theoretical_loss": 3.5773997274836224, + "tokens_seen": 1237319680 + }, + { + "epoch": 0.37, + "learning_rate": 0.00031572718154463395, + "loss": 1.3611, + "theoretical_loss": 3.5773823985800544, + "tokens_seen": 1237385216 + }, + { + "epoch": 0.37, + "learning_rate": 0.0003157171514543631, + "loss": 1.2637, + "theoretical_loss": 3.5773650708512252, + "tokens_seen": 1237450752 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003157071213640923, + "loss": 1.3001, + "theoretical_loss": 3.577347744296993, + "tokens_seen": 1237516288 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031569709127382144, + "loss": 1.2833, + "theoretical_loss": 3.577330418917216, + "tokens_seen": 1237581824 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031568706118355067, + "loss": 1.4953, + "theoretical_loss": 3.5773130947117524, + "tokens_seen": 1237647360 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031567703109327985, + "loss": 1.3487, + "theoretical_loss": 3.5772957716804603, + "tokens_seen": 1237712896 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031566700100300903, + "loss": 1.4126, + "theoretical_loss": 3.5772784498231984, + "tokens_seen": 1237778432 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003156569709127382, + "loss": 1.4155, + "theoretical_loss": 3.577261129139824, + "tokens_seen": 1237843968 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003156469408224674, + "loss": 1.3397, + "theoretical_loss": 3.577243809630197, + "tokens_seen": 1237909504 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003156369107321966, + "loss": 1.3105, + "theoretical_loss": 3.5772264912941747, + "tokens_seen": 1237975040 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003156268806419258, + "loss": 1.2898, + "theoretical_loss": 3.577209174131615, + "tokens_seen": 1238040576 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031561685055165494, + "loss": 1.5196, + "theoretical_loss": 3.577191858142377, + "tokens_seen": 1238106112 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003156068204613842, + "loss": 1.4273, + "theoretical_loss": 3.5771745433263193, + "tokens_seen": 1238171648 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003155967903711133, + "loss": 1.3656, + "theoretical_loss": 3.5771572296832996, + "tokens_seen": 1238237184 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031558676028084254, + "loss": 1.3107, + "theoretical_loss": 3.577139917213177, + "tokens_seen": 1238302720 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003155767301905717, + "loss": 1.3388, + "theoretical_loss": 3.577122605915809, + "tokens_seen": 1238368256 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003155667001003009, + "loss": 1.4266, + "theoretical_loss": 3.577105295791056, + "tokens_seen": 1238433792 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003155566700100301, + "loss": 1.3737, + "theoretical_loss": 3.5770879868387744, + "tokens_seen": 1238499328 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003155466399197593, + "loss": 1.3779, + "theoretical_loss": 3.577070679058824, + "tokens_seen": 1238564864 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.0005748810363002121, + "objective/train/docs_used": 702757, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3754262924194336, + "objective/train/original_loss": 2.3754260540008545, + "objective/train/theoretical_loss": 3.5770533724510627, + "objective/train/tokens_used": 1259090400, + "objective/train/value_avg": -0.004703521728515625, + "objective/train/value_loss": 5.6601718824822456e-05, + "objective/train/value_max": -0.0002613067626953125, + "objective/train/value_min": -0.0823974609375, + "objective/train/value_reward_corr": 0.32237077662237396, + "objective/train/value_std": 0.004261016845703125, + "objective/train/weight_avg": 0.9969199895858765, + "objective/train/weighted_lm_loss": 2.381747245788574, + "objective/train/weights_max": 2.110140562057495, + "objective/train/weights_min": 0.4125877916812897, + "theoretical_loss": 3.5770533724510627, + "tokens_seen": 1238630400 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031553660982948844, + "loss": 1.2993, + "theoretical_loss": 3.5770533724510627, + "tokens_seen": 1238630400 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003155265797392177, + "loss": 1.364, + "theoretical_loss": 3.5770360670153494, + "tokens_seen": 1238695936 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003155165496489468, + "loss": 1.4333, + "theoretical_loss": 3.5770187627515435, + "tokens_seen": 1238761472 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031550651955867604, + "loss": 1.3339, + "theoretical_loss": 3.577001459659502, + "tokens_seen": 1238827008 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003154964894684052, + "loss": 1.3831, + "theoretical_loss": 3.5769841577390853, + "tokens_seen": 1238892544 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003154864593781344, + "loss": 1.397, + "theoretical_loss": 3.5769668569901505, + "tokens_seen": 1238958080 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003154764292878636, + "loss": 1.3419, + "theoretical_loss": 3.576949557412558, + "tokens_seen": 1239023616 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031546639919759277, + "loss": 1.5017, + "theoretical_loss": 3.5769322590061647, + "tokens_seen": 1239089152 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031545636910732195, + "loss": 1.4038, + "theoretical_loss": 3.5769149617708305, + "tokens_seen": 1239154688 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003154463390170512, + "loss": 1.3477, + "theoretical_loss": 3.576897665706414, + "tokens_seen": 1239220224 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003154363089267803, + "loss": 1.4765, + "theoretical_loss": 3.5768803708127743, + "tokens_seen": 1239285760 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031542627883650954, + "loss": 1.3785, + "theoretical_loss": 3.57686307708977, + "tokens_seen": 1239351296 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003154162487462387, + "loss": 1.2809, + "theoretical_loss": 3.5768457845372597, + "tokens_seen": 1239416832 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003154062186559679, + "loss": 1.3387, + "theoretical_loss": 3.576828493155103, + "tokens_seen": 1239482368 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003153961885656971, + "loss": 1.3433, + "theoretical_loss": 3.5768112029431576, + "tokens_seen": 1239547904 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031538615847542627, + "loss": 1.3534, + "theoretical_loss": 3.576793913901284, + "tokens_seen": 1239613440 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031537612838515545, + "loss": 1.2856, + "theoretical_loss": 3.5767766260293397, + "tokens_seen": 1239678976 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003153660982948847, + "loss": 1.3176, + "theoretical_loss": 3.576759339327184, + "tokens_seen": 1239744512 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003153560682046138, + "loss": 1.409, + "theoretical_loss": 3.576742053794677, + "tokens_seen": 1239810048 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031534603811434305, + "loss": 1.3308, + "theoretical_loss": 3.576724769431677, + "tokens_seen": 1239875584 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003153360080240722, + "loss": 1.3896, + "theoretical_loss": 3.5767074862380426, + "tokens_seen": 1239941120 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003153259779338014, + "loss": 1.4233, + "theoretical_loss": 3.5766902042136337, + "tokens_seen": 1240006656 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003153159478435306, + "loss": 1.2936, + "theoretical_loss": 3.5766729233583083, + "tokens_seen": 1240072192 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031530591775325977, + "loss": 1.3298, + "theoretical_loss": 3.576655643671927, + "tokens_seen": 1240137728 + }, + { + "epoch": 0.38, + "learning_rate": 0.000315295887662989, + "loss": 1.4422, + "theoretical_loss": 3.5766383651543485, + "tokens_seen": 1240203264 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.0001383753406116739, + "objective/train/docs_used": 703958, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.622417688369751, + "objective/train/original_loss": 2.6224172115325928, + "objective/train/theoretical_loss": 3.5766210878054308, + "objective/train/tokens_used": 1260728800, + "objective/train/value_avg": -0.005054473876953125, + "objective/train/value_loss": 5.137634798302315e-05, + "objective/train/value_max": -0.0002758502960205078, + "objective/train/value_min": -0.06951904296875, + "objective/train/value_reward_corr": 0.60741194889498, + "objective/train/value_std": 0.005252838134765625, + "objective/train/weight_avg": 1.0010215044021606, + "objective/train/weighted_lm_loss": 2.6216092109680176, + "objective/train/weights_max": 1.5199191570281982, + "objective/train/weights_min": 0.2462211400270462, + "theoretical_loss": 3.5766210878054308, + "tokens_seen": 1240268800 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031528585757271813, + "loss": 1.2446, + "theoretical_loss": 3.5766210878054308, + "tokens_seen": 1240268800 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031527582748244737, + "loss": 1.4144, + "theoretical_loss": 3.5766038116250347, + "tokens_seen": 1240334336 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031526579739217655, + "loss": 1.3353, + "theoretical_loss": 3.5765865366130187, + "tokens_seen": 1240399872 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031525576730190573, + "loss": 1.3743, + "theoretical_loss": 3.576569262769242, + "tokens_seen": 1240465408 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003152457372116349, + "loss": 1.3587, + "theoretical_loss": 3.5765519900935643, + "tokens_seen": 1240530944 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031523570712136415, + "loss": 1.4027, + "theoretical_loss": 3.5765347185858443, + "tokens_seen": 1240596480 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003152256770310933, + "loss": 1.3159, + "theoretical_loss": 3.5765174482459416, + "tokens_seen": 1240662016 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003152156469408225, + "loss": 1.414, + "theoretical_loss": 3.576500179073716, + "tokens_seen": 1240727552 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031520561685055164, + "loss": 1.4002, + "theoretical_loss": 3.576482911069026, + "tokens_seen": 1240793088 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031519558676028087, + "loss": 1.3803, + "theoretical_loss": 3.5764656442317317, + "tokens_seen": 1240858624 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031518555667001005, + "loss": 1.4102, + "theoretical_loss": 3.5764483785616923, + "tokens_seen": 1240924160 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031517552657973924, + "loss": 1.2303, + "theoretical_loss": 3.5764311140587672, + "tokens_seen": 1240989696 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003151654964894684, + "loss": 1.2668, + "theoretical_loss": 3.5764138507228163, + "tokens_seen": 1241055232 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003151554663991976, + "loss": 1.3508, + "theoretical_loss": 3.5763965885536986, + "tokens_seen": 1241120768 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003151454363089268, + "loss": 1.3039, + "theoretical_loss": 3.5763793275512734, + "tokens_seen": 1241186304 + }, + { + "epoch": 0.38, + "learning_rate": 0.000315135406218656, + "loss": 1.4041, + "theoretical_loss": 3.576362067715401, + "tokens_seen": 1241251840 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031512537612838514, + "loss": 1.3833, + "theoretical_loss": 3.576344809045941, + "tokens_seen": 1241317376 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003151153460381144, + "loss": 1.3664, + "theoretical_loss": 3.576327551542752, + "tokens_seen": 1241382912 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003151053159478435, + "loss": 1.357, + "theoretical_loss": 3.5763102952056944, + "tokens_seen": 1241448448 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031509528585757274, + "loss": 1.2893, + "theoretical_loss": 3.576293040034628, + "tokens_seen": 1241513984 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003150852557673019, + "loss": 1.3883, + "theoretical_loss": 3.576275786029411, + "tokens_seen": 1241579520 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003150752256770311, + "loss": 1.4063, + "theoretical_loss": 3.5762585331899057, + "tokens_seen": 1241645056 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003150651955867603, + "loss": 1.3838, + "theoretical_loss": 3.5762412815159696, + "tokens_seen": 1241710592 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003150551654964895, + "loss": 1.3538, + "theoretical_loss": 3.576224031007463, + "tokens_seen": 1241776128 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031504513540621864, + "loss": 1.3564, + "theoretical_loss": 3.576206781664246, + "tokens_seen": 1241841664 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.0022417395375669003, + "objective/train/docs_used": 704662, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9508445262908936, + "objective/train/original_loss": 2.9508447647094727, + "objective/train/theoretical_loss": 3.576189533486179, + "objective/train/tokens_used": 1262367200, + "objective/train/value_avg": -0.00508880615234375, + "objective/train/value_loss": 0.00010208124876953661, + "objective/train/value_max": -0.0003542900085449219, + "objective/train/value_min": -0.06671142578125, + "objective/train/value_reward_corr": 0.3915864057715393, + "objective/train/value_std": 0.00431060791015625, + "objective/train/weight_avg": 0.9821457266807556, + "objective/train/weighted_lm_loss": 2.896406412124634, + "objective/train/weights_max": 1.948606014251709, + "objective/train/weights_min": 0.14493237435817719, + "theoretical_loss": 3.576189533486179, + "tokens_seen": 1241907200 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003150351053159479, + "loss": 1.3436, + "theoretical_loss": 3.576189533486179, + "tokens_seen": 1241907200 + }, + { + "epoch": 0.38, + "learning_rate": 0.000315025075225677, + "loss": 1.3923, + "theoretical_loss": 3.5761722864731205, + "tokens_seen": 1241972736 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031501504513540624, + "loss": 1.3443, + "theoretical_loss": 3.576155040624931, + "tokens_seen": 1242038272 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003150050150451354, + "loss": 1.4505, + "theoretical_loss": 3.5761377959414706, + "tokens_seen": 1242103808 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003149949849548646, + "loss": 1.4189, + "theoretical_loss": 3.5761205524225987, + "tokens_seen": 1242169344 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003149849548645938, + "loss": 1.5016, + "theoretical_loss": 3.5761033100681754, + "tokens_seen": 1242234880 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031497492477432297, + "loss": 1.3375, + "theoretical_loss": 3.5760860688780607, + "tokens_seen": 1242300416 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031496489468405215, + "loss": 1.3294, + "theoretical_loss": 3.5760688288521143, + "tokens_seen": 1242365952 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003149548645937814, + "loss": 1.3478, + "theoretical_loss": 3.5760515899901963, + "tokens_seen": 1242431488 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003149448345035105, + "loss": 1.4655, + "theoretical_loss": 3.576034352292167, + "tokens_seen": 1242497024 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031493480441323974, + "loss": 1.4427, + "theoretical_loss": 3.576017115757886, + "tokens_seen": 1242562560 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003149247743229689, + "loss": 1.284, + "theoretical_loss": 3.575999880387214, + "tokens_seen": 1242628096 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003149147442326981, + "loss": 1.3379, + "theoretical_loss": 3.5759826461800106, + "tokens_seen": 1242693632 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003149047141424273, + "loss": 1.4037, + "theoretical_loss": 3.575965413136136, + "tokens_seen": 1242759168 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031489468405215647, + "loss": 1.3751, + "theoretical_loss": 3.5759481812554506, + "tokens_seen": 1242824704 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031488465396188565, + "loss": 1.3944, + "theoretical_loss": 3.5759309505378134, + "tokens_seen": 1242890240 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003148746238716149, + "loss": 1.2406, + "theoretical_loss": 3.5759137209830865, + "tokens_seen": 1242955776 + }, + { + "epoch": 0.38, + "learning_rate": 0.000314864593781344, + "loss": 1.4107, + "theoretical_loss": 3.5758964925911285, + "tokens_seen": 1243021312 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031485456369107325, + "loss": 1.3994, + "theoretical_loss": 3.5758792653618006, + "tokens_seen": 1243086848 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003148445336008024, + "loss": 1.3331, + "theoretical_loss": 3.5758620392949623, + "tokens_seen": 1243152384 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003148345035105316, + "loss": 1.3547, + "theoretical_loss": 3.575844814390474, + "tokens_seen": 1243217920 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003148244734202608, + "loss": 1.3908, + "theoretical_loss": 3.5758275906481964, + "tokens_seen": 1243283456 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031481444332998997, + "loss": 1.3858, + "theoretical_loss": 3.57581036806799, + "tokens_seen": 1243348992 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031480441323971915, + "loss": 1.4338, + "theoretical_loss": 3.5757931466497146, + "tokens_seen": 1243414528 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031479438314944833, + "loss": 1.3804, + "theoretical_loss": 3.575775926393231, + "tokens_seen": 1243480064 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": 0.00019262715068180114, + "objective/train/docs_used": 705833, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8013999462127686, + "objective/train/original_loss": 2.8014001846313477, + "objective/train/theoretical_loss": 3.5757587072983994, + "objective/train/tokens_used": 1264005600, + "objective/train/value_avg": -0.00446319580078125, + "objective/train/value_loss": 4.454014197108336e-05, + "objective/train/value_max": -0.0001366138458251953, + "objective/train/value_min": -0.1783447265625, + "objective/train/value_reward_corr": 0.28758245861532916, + "objective/train/value_std": 0.004085540771484375, + "objective/train/weight_avg": 1.0042240619659424, + "objective/train/weighted_lm_loss": 2.8213000297546387, + "objective/train/weights_max": 5.703467845916748, + "objective/train/weights_min": 0.6641278266906738, + "theoretical_loss": 3.5757587072983994, + "tokens_seen": 1243545600 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003147843530591775, + "loss": 1.4421, + "theoretical_loss": 3.5757587072983994, + "tokens_seen": 1243545600 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031477432296890675, + "loss": 1.3963, + "theoretical_loss": 3.57574148936508, + "tokens_seen": 1243611136 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003147642928786359, + "loss": 1.4816, + "theoretical_loss": 3.5757242725931335, + "tokens_seen": 1243676672 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003147542627883651, + "loss": 1.3696, + "theoretical_loss": 3.5757070569824205, + "tokens_seen": 1243742208 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003147442326980943, + "loss": 1.4186, + "theoretical_loss": 3.575689842532801, + "tokens_seen": 1243807744 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003147342026078235, + "loss": 1.4176, + "theoretical_loss": 3.575672629244136, + "tokens_seen": 1243873280 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031472417251755266, + "loss": 1.4955, + "theoretical_loss": 3.5756554171162858, + "tokens_seen": 1243938816 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031471414242728184, + "loss": 1.4611, + "theoretical_loss": 3.575638206149111, + "tokens_seen": 1244004352 + }, + { + "epoch": 0.38, + "learning_rate": 0.000314704112337011, + "loss": 1.3882, + "theoretical_loss": 3.5756209963424728, + "tokens_seen": 1244069888 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031469408224674025, + "loss": 1.3133, + "theoretical_loss": 3.575603787696231, + "tokens_seen": 1244135424 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003146840521564694, + "loss": 1.2949, + "theoretical_loss": 3.575586580210246, + "tokens_seen": 1244200960 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003146740220661986, + "loss": 1.3796, + "theoretical_loss": 3.5755693738843792, + "tokens_seen": 1244266496 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031466399197592774, + "loss": 1.3067, + "theoretical_loss": 3.5755521687184917, + "tokens_seen": 1244332032 + }, + { + "epoch": 0.38, + "learning_rate": 0.000314653961885657, + "loss": 1.3535, + "theoretical_loss": 3.5755349647124426, + "tokens_seen": 1244397568 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031464393179538616, + "loss": 1.3555, + "theoretical_loss": 3.5755177618660943, + "tokens_seen": 1244463104 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031463390170511534, + "loss": 1.3872, + "theoretical_loss": 3.5755005601793064, + "tokens_seen": 1244528640 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003146238716148445, + "loss": 1.4424, + "theoretical_loss": 3.5754833596519404, + "tokens_seen": 1244594176 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003146138415245737, + "loss": 1.43, + "theoretical_loss": 3.575466160283857, + "tokens_seen": 1244659712 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003146038114343029, + "loss": 1.4224, + "theoretical_loss": 3.5754489620749164, + "tokens_seen": 1244725248 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003145937813440321, + "loss": 1.3041, + "theoretical_loss": 3.5754317650249803, + "tokens_seen": 1244790784 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031458375125376125, + "loss": 1.3873, + "theoretical_loss": 3.575414569133909, + "tokens_seen": 1244856320 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003145737211634905, + "loss": 1.4543, + "theoretical_loss": 3.5753973744015637, + "tokens_seen": 1244921856 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003145636910732197, + "loss": 1.2733, + "theoretical_loss": 3.5753801808278056, + "tokens_seen": 1244987392 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031455366098294884, + "loss": 1.2758, + "theoretical_loss": 3.575362988412495, + "tokens_seen": 1245052928 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003145436308926781, + "loss": 1.396, + "theoretical_loss": 3.575345797155493, + "tokens_seen": 1245118464 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.0011316646123304963, + "objective/train/docs_used": 706471, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8180954456329346, + "objective/train/original_loss": 2.8180949687957764, + "objective/train/theoretical_loss": 3.5753286070566617, + "objective/train/tokens_used": 1265644000, + "objective/train/value_avg": -0.00591278076171875, + "objective/train/value_loss": 7.692143844906241e-05, + "objective/train/value_max": -0.00039505958557128906, + "objective/train/value_min": -0.0966796875, + "objective/train/value_reward_corr": 0.2984389051122653, + "objective/train/value_std": 0.00437164306640625, + "objective/train/weight_avg": 0.9923002123832703, + "objective/train/weighted_lm_loss": 2.796119451522827, + "objective/train/weights_max": 2.427341938018799, + "objective/train/weights_min": 0.38046029210090637, + "theoretical_loss": 3.5753286070566617, + "tokens_seen": 1245184000 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003145336008024072, + "loss": 1.4013, + "theoretical_loss": 3.5753286070566617, + "tokens_seen": 1245184000 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031452357071213644, + "loss": 1.4233, + "theoretical_loss": 3.5753114181158603, + "tokens_seen": 1245249536 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003145135406218656, + "loss": 1.2574, + "theoretical_loss": 3.575294230332951, + "tokens_seen": 1245315072 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003145035105315948, + "loss": 1.3183, + "theoretical_loss": 3.5752770437077945, + "tokens_seen": 1245380608 + }, + { + "epoch": 0.38, + "learning_rate": 0.000314493480441324, + "loss": 1.3438, + "theoretical_loss": 3.575259858240252, + "tokens_seen": 1245446144 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031448345035105317, + "loss": 1.3871, + "theoretical_loss": 3.575242673930185, + "tokens_seen": 1245511680 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031447342026078235, + "loss": 1.3932, + "theoretical_loss": 3.575225490777454, + "tokens_seen": 1245577216 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003144633901705116, + "loss": 1.351, + "theoretical_loss": 3.575208308781921, + "tokens_seen": 1245642752 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003144533600802407, + "loss": 1.4015, + "theoretical_loss": 3.575191127943446, + "tokens_seen": 1245708288 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031444332998996994, + "loss": 1.3237, + "theoretical_loss": 3.5751739482618907, + "tokens_seen": 1245773824 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003144332998996991, + "loss": 1.3684, + "theoretical_loss": 3.5751567697371174, + "tokens_seen": 1245839360 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003144232698094283, + "loss": 1.4498, + "theoretical_loss": 3.575139592368986, + "tokens_seen": 1245904896 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003144132397191575, + "loss": 1.398, + "theoretical_loss": 3.575122416157358, + "tokens_seen": 1245970432 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031440320962888667, + "loss": 1.3617, + "theoretical_loss": 3.5751052411020954, + "tokens_seen": 1246035968 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031439317953861585, + "loss": 1.396, + "theoretical_loss": 3.5750880672030583, + "tokens_seen": 1246101504 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003143831494483451, + "loss": 1.433, + "theoretical_loss": 3.5750708944601097, + "tokens_seen": 1246167040 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003143731193580742, + "loss": 1.5373, + "theoretical_loss": 3.57505372287311, + "tokens_seen": 1246232576 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031436308926780345, + "loss": 1.3717, + "theoretical_loss": 3.5750365524419205, + "tokens_seen": 1246298112 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003143530591775326, + "loss": 1.4255, + "theoretical_loss": 3.5750193831664028, + "tokens_seen": 1246363648 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003143430290872618, + "loss": 1.2319, + "theoretical_loss": 3.575002215046419, + "tokens_seen": 1246429184 + }, + { + "epoch": 0.38, + "learning_rate": 0.000314332998996991, + "loss": 1.3996, + "theoretical_loss": 3.5749850480818295, + "tokens_seen": 1246494720 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031432296890672017, + "loss": 1.3899, + "theoretical_loss": 3.574967882272497, + "tokens_seen": 1246560256 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031431293881644935, + "loss": 1.3932, + "theoretical_loss": 3.5749507176182815, + "tokens_seen": 1246625792 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031430290872617853, + "loss": 1.4878, + "theoretical_loss": 3.5749335541190455, + "tokens_seen": 1246691328 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003142928786359077, + "loss": 1.3053, + "theoretical_loss": 3.574916391774651, + "tokens_seen": 1246756864 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.001129014533944428, + "objective/train/docs_used": 707914, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8537893295288086, + "objective/train/original_loss": 2.8537890911102295, + "objective/train/theoretical_loss": 3.574899230584959, + "objective/train/tokens_used": 1267282400, + "objective/train/value_avg": -0.00658416748046875, + "objective/train/value_loss": 5.373041858547367e-05, + "objective/train/value_max": -0.00033545494079589844, + "objective/train/value_min": -0.06756591796875, + "objective/train/value_reward_corr": 0.47554660901822743, + "objective/train/value_std": 0.0052490234375, + "objective/train/weight_avg": 0.9913144111633301, + "objective/train/weighted_lm_loss": 2.8288614749908447, + "objective/train/weights_max": 1.7451709508895874, + "objective/train/weights_min": 0.578942060470581, + "theoretical_loss": 3.574899230584959, + "tokens_seen": 1246822400 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031428284854563695, + "loss": 1.3956, + "theoretical_loss": 3.574899230584959, + "tokens_seen": 1246822400 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003142728184553661, + "loss": 1.327, + "theoretical_loss": 3.574882070549831, + "tokens_seen": 1246887936 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003142627883650953, + "loss": 1.4869, + "theoretical_loss": 3.574864911669129, + "tokens_seen": 1246953472 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003142527582748245, + "loss": 1.3942, + "theoretical_loss": 3.574847753942714, + "tokens_seen": 1247019008 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003142427281845537, + "loss": 1.3576, + "theoretical_loss": 3.5748305973704486, + "tokens_seen": 1247084544 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031423269809428286, + "loss": 1.3445, + "theoretical_loss": 3.5748134419521946, + "tokens_seen": 1247150080 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031422266800401204, + "loss": 1.386, + "theoretical_loss": 3.5747962876878123, + "tokens_seen": 1247215616 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003142126379137412, + "loss": 1.3656, + "theoretical_loss": 3.5747791345771653, + "tokens_seen": 1247281152 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031420260782347045, + "loss": 1.3012, + "theoretical_loss": 3.5747619826201142, + "tokens_seen": 1247346688 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003141925777331996, + "loss": 1.5074, + "theoretical_loss": 3.574744831816521, + "tokens_seen": 1247412224 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003141825476429288, + "loss": 1.3576, + "theoretical_loss": 3.574727682166248, + "tokens_seen": 1247477760 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031417251755265794, + "loss": 1.457, + "theoretical_loss": 3.5747105336691565, + "tokens_seen": 1247543296 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003141624874623872, + "loss": 1.399, + "theoretical_loss": 3.5746933863251087, + "tokens_seen": 1247608832 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031415245737211636, + "loss": 1.424, + "theoretical_loss": 3.5746762401339662, + "tokens_seen": 1247674368 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031414242728184554, + "loss": 1.3741, + "theoretical_loss": 3.5746590950955914, + "tokens_seen": 1247739904 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003141323971915747, + "loss": 1.3289, + "theoretical_loss": 3.5746419512098457, + "tokens_seen": 1247805440 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003141223671013039, + "loss": 1.3477, + "theoretical_loss": 3.5746248084765915, + "tokens_seen": 1247870976 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003141123370110331, + "loss": 1.4191, + "theoretical_loss": 3.574607666895691, + "tokens_seen": 1247936512 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003141023069207623, + "loss": 1.4614, + "theoretical_loss": 3.5745905264670057, + "tokens_seen": 1248002048 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031409227683049145, + "loss": 1.4665, + "theoretical_loss": 3.574573387190398, + "tokens_seen": 1248067584 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003140822467402207, + "loss": 1.3168, + "theoretical_loss": 3.57455624906573, + "tokens_seen": 1248133120 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031407221664994986, + "loss": 1.2431, + "theoretical_loss": 3.574539112092863, + "tokens_seen": 1248198656 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031406218655967904, + "loss": 1.3906, + "theoretical_loss": 3.5745219762716607, + "tokens_seen": 1248264192 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003140521564694082, + "loss": 1.4892, + "theoretical_loss": 3.574504841601984, + "tokens_seen": 1248329728 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003140421263791374, + "loss": 1.4758, + "theoretical_loss": 3.5744877080836948, + "tokens_seen": 1248395264 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.0004256096144672483, + "objective/train/docs_used": 708590, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.572789430618286, + "objective/train/original_loss": 2.572789192199707, + "objective/train/theoretical_loss": 3.5744705757166564, + "objective/train/tokens_used": 1268920800, + "objective/train/value_avg": -0.005107879638671875, + "objective/train/value_loss": 9.558727469993755e-05, + "objective/train/value_max": -0.00017535686492919922, + "objective/train/value_min": -0.1134033203125, + "objective/train/value_reward_corr": 0.4509471855419601, + "objective/train/value_std": 0.0045928955078125, + "objective/train/weight_avg": 0.9999969005584717, + "objective/train/weighted_lm_loss": 2.5664491653442383, + "objective/train/weights_max": 3.1081671714782715, + "objective/train/weights_min": 0.2108088582754135, + "theoretical_loss": 3.5744705757166564, + "tokens_seen": 1248460800 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003140320962888666, + "loss": 1.3882, + "theoretical_loss": 3.5744705757166564, + "tokens_seen": 1248460800 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003140220661985958, + "loss": 1.2826, + "theoretical_loss": 3.5744534445007305, + "tokens_seen": 1248526336 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031401203610832495, + "loss": 1.3903, + "theoretical_loss": 3.574436314435779, + "tokens_seen": 1248591872 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003140020060180542, + "loss": 1.4643, + "theoretical_loss": 3.5744191855216645, + "tokens_seen": 1248657408 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003139919759277833, + "loss": 1.3212, + "theoretical_loss": 3.57440205775825, + "tokens_seen": 1248722944 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031398194583751255, + "loss": 1.363, + "theoretical_loss": 3.5743849311453966, + "tokens_seen": 1248788480 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031397191574724173, + "loss": 1.4052, + "theoretical_loss": 3.574367805682967, + "tokens_seen": 1248854016 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003139618856569709, + "loss": 1.4743, + "theoretical_loss": 3.5743506813708237, + "tokens_seen": 1248919552 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003139518555667001, + "loss": 1.3405, + "theoretical_loss": 3.5743335582088296, + "tokens_seen": 1248985088 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003139418254764293, + "loss": 1.4104, + "theoretical_loss": 3.5743164361968462, + "tokens_seen": 1249050624 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031393179538615845, + "loss": 1.3949, + "theoretical_loss": 3.574299315334736, + "tokens_seen": 1249116160 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003139217652958877, + "loss": 1.469, + "theoretical_loss": 3.5742821956223625, + "tokens_seen": 1249181696 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003139117352056168, + "loss": 1.3825, + "theoretical_loss": 3.5742650770595867, + "tokens_seen": 1249247232 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031390170511534605, + "loss": 1.3205, + "theoretical_loss": 3.5742479596462724, + "tokens_seen": 1249312768 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031389167502507523, + "loss": 1.2789, + "theoretical_loss": 3.5742308433822814, + "tokens_seen": 1249378304 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003138816449348044, + "loss": 1.4582, + "theoretical_loss": 3.574213728267477, + "tokens_seen": 1249443840 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003138716148445336, + "loss": 1.2787, + "theoretical_loss": 3.5741966143017203, + "tokens_seen": 1249509376 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003138615847542628, + "loss": 1.1986, + "theoretical_loss": 3.574179501484875, + "tokens_seen": 1249574912 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031385155466399196, + "loss": 1.4175, + "theoretical_loss": 3.574162389816804, + "tokens_seen": 1249640448 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003138415245737212, + "loss": 1.4295, + "theoretical_loss": 3.574145279297369, + "tokens_seen": 1249705984 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003138314944834503, + "loss": 1.342, + "theoretical_loss": 3.574128169926433, + "tokens_seen": 1249771520 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031382146439317955, + "loss": 1.2183, + "theoretical_loss": 3.574111061703859, + "tokens_seen": 1249837056 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031381143430290873, + "loss": 1.263, + "theoretical_loss": 3.57409395462951, + "tokens_seen": 1249902592 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003138014042126379, + "loss": 1.3541, + "theoretical_loss": 3.574076848703248, + "tokens_seen": 1249968128 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031379137412236715, + "loss": 1.2504, + "theoretical_loss": 3.574059743924936, + "tokens_seen": 1250033664 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": 0.0002512949286028743, + "objective/train/docs_used": 709757, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5189924240112305, + "objective/train/original_loss": 2.5189921855926514, + "objective/train/theoretical_loss": 3.5740426402944365, + "objective/train/tokens_used": 1270559200, + "objective/train/value_avg": -0.004474639892578125, + "objective/train/value_loss": 3.0292547307908535e-05, + "objective/train/value_max": -0.0003101825714111328, + "objective/train/value_min": -0.058349609375, + "objective/train/value_reward_corr": 0.4070884432839937, + "objective/train/value_std": 0.003528594970703125, + "objective/train/weight_avg": 1.0040010213851929, + "objective/train/weighted_lm_loss": 2.5356481075286865, + "objective/train/weights_max": 1.6910617351531982, + "objective/train/weights_min": 0.7020418047904968, + "theoretical_loss": 3.5740426402944365, + "tokens_seen": 1250099200 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003137813440320963, + "loss": 1.3532, + "theoretical_loss": 3.5740426402944365, + "tokens_seen": 1250099200 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003137713139418255, + "loss": 1.4077, + "theoretical_loss": 3.574025537811613, + "tokens_seen": 1250164736 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003137612838515547, + "loss": 1.4077, + "theoretical_loss": 3.5740084364763276, + "tokens_seen": 1250230272 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003137512537612839, + "loss": 1.3987, + "theoretical_loss": 3.5739913362884437, + "tokens_seen": 1250295808 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031374122367101306, + "loss": 1.3564, + "theoretical_loss": 3.573974237247824, + "tokens_seen": 1250361344 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031373119358074224, + "loss": 1.3372, + "theoretical_loss": 3.5739571393543317, + "tokens_seen": 1250426880 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003137211634904714, + "loss": 1.3308, + "theoretical_loss": 3.5739400426078287, + "tokens_seen": 1250492416 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031371113340020065, + "loss": 1.3136, + "theoretical_loss": 3.573922947008179, + "tokens_seen": 1250557952 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003137011033099298, + "loss": 1.315, + "theoretical_loss": 3.5739058525552454, + "tokens_seen": 1250623488 + }, + { + "epoch": 0.38, + "learning_rate": 0.000313691073219659, + "loss": 1.4207, + "theoretical_loss": 3.5738887592488906, + "tokens_seen": 1250689024 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031368104312938814, + "loss": 1.407, + "theoretical_loss": 3.5738716670889783, + "tokens_seen": 1250754560 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003136710130391174, + "loss": 1.2534, + "theoretical_loss": 3.5738545760753704, + "tokens_seen": 1250820096 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031366098294884656, + "loss": 1.374, + "theoretical_loss": 3.573837486207931, + "tokens_seen": 1250885632 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031365095285857574, + "loss": 1.3911, + "theoretical_loss": 3.5738203974865224, + "tokens_seen": 1250951168 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003136409227683049, + "loss": 1.387, + "theoretical_loss": 3.573803309911008, + "tokens_seen": 1251016704 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003136308926780341, + "loss": 1.4027, + "theoretical_loss": 3.5737862234812514, + "tokens_seen": 1251082240 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003136208625877633, + "loss": 1.4285, + "theoretical_loss": 3.5737691381971146, + "tokens_seen": 1251147776 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003136108324974925, + "loss": 1.3654, + "theoretical_loss": 3.5737520540584624, + "tokens_seen": 1251213312 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031360080240722165, + "loss": 1.3394, + "theoretical_loss": 3.573734971065157, + "tokens_seen": 1251278848 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003135907723169509, + "loss": 1.3431, + "theoretical_loss": 3.573717889217062, + "tokens_seen": 1251344384 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031358074222668006, + "loss": 1.3367, + "theoretical_loss": 3.57370080851404, + "tokens_seen": 1251409920 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031357071213640924, + "loss": 1.3634, + "theoretical_loss": 3.573683728955955, + "tokens_seen": 1251475456 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003135606820461384, + "loss": 1.3487, + "theoretical_loss": 3.5736666505426697, + "tokens_seen": 1251540992 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003135506519558676, + "loss": 1.3058, + "theoretical_loss": 3.5736495732740483, + "tokens_seen": 1251606528 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003135406218655968, + "loss": 1.4388, + "theoretical_loss": 3.573632497149953, + "tokens_seen": 1251672064 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.00030948620405979455, + "objective/train/docs_used": 710221, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7208492755889893, + "objective/train/original_loss": 2.720848560333252, + "objective/train/theoretical_loss": 3.5736154221702483, + "objective/train/tokens_used": 1272197600, + "objective/train/value_avg": -0.004886627197265625, + "objective/train/value_loss": 4.755033296532929e-05, + "objective/train/value_max": -0.0003199577331542969, + "objective/train/value_min": -0.044189453125, + "objective/train/value_reward_corr": 0.42166770941741977, + "objective/train/value_std": 0.00445556640625, + "objective/train/weight_avg": 0.9991965889930725, + "objective/train/weighted_lm_loss": 2.728527545928955, + "objective/train/weights_max": 1.440885066986084, + "objective/train/weights_min": 0.5642259120941162, + "theoretical_loss": 3.5736154221702483, + "tokens_seen": 1251737600 + }, + { + "epoch": 0.38, + "learning_rate": 0.000313530591775326, + "loss": 1.3716, + "theoretical_loss": 3.5736154221702483, + "tokens_seen": 1251737600 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031352056168505515, + "loss": 1.4197, + "theoretical_loss": 3.573598348334797, + "tokens_seen": 1251803136 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003135105315947844, + "loss": 1.3501, + "theoretical_loss": 3.5735812756434626, + "tokens_seen": 1251868672 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003135005015045135, + "loss": 1.3683, + "theoretical_loss": 3.5735642040961086, + "tokens_seen": 1251934208 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031349047141424275, + "loss": 1.3785, + "theoretical_loss": 3.5735471336925984, + "tokens_seen": 1251999744 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031348044132397193, + "loss": 1.4518, + "theoretical_loss": 3.5735300644327954, + "tokens_seen": 1252065280 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003134704112337011, + "loss": 1.3442, + "theoretical_loss": 3.5735129963165635, + "tokens_seen": 1252130816 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003134603811434303, + "loss": 1.3031, + "theoretical_loss": 3.573495929343766, + "tokens_seen": 1252196352 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003134503510531595, + "loss": 1.3332, + "theoretical_loss": 3.573478863514266, + "tokens_seen": 1252261888 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031344032096288865, + "loss": 1.3286, + "theoretical_loss": 3.5734617988279274, + "tokens_seen": 1252327424 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003134302908726179, + "loss": 1.2614, + "theoretical_loss": 3.5734447352846144, + "tokens_seen": 1252392960 + }, + { + "epoch": 0.38, + "learning_rate": 0.000313420260782347, + "loss": 1.3805, + "theoretical_loss": 3.57342767288419, + "tokens_seen": 1252458496 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031341023069207625, + "loss": 1.3956, + "theoretical_loss": 3.573410611626518, + "tokens_seen": 1252524032 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031340020060180543, + "loss": 1.4177, + "theoretical_loss": 3.5733935515114617, + "tokens_seen": 1252589568 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003133901705115346, + "loss": 1.4635, + "theoretical_loss": 3.573376492538886, + "tokens_seen": 1252655104 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003133801404212638, + "loss": 1.3446, + "theoretical_loss": 3.573359434708653, + "tokens_seen": 1252720640 + }, + { + "epoch": 0.38, + "learning_rate": 0.000313370110330993, + "loss": 1.4082, + "theoretical_loss": 3.5733423780206275, + "tokens_seen": 1252786176 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031336008024072216, + "loss": 1.312, + "theoretical_loss": 3.5733253224746733, + "tokens_seen": 1252851712 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003133500501504514, + "loss": 1.3394, + "theoretical_loss": 3.5733082680706536, + "tokens_seen": 1252917248 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003133400200601805, + "loss": 1.3986, + "theoretical_loss": 3.5732912148084326, + "tokens_seen": 1252982784 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031332998996990975, + "loss": 1.3065, + "theoretical_loss": 3.5732741626878743, + "tokens_seen": 1253048320 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003133199598796389, + "loss": 1.3451, + "theoretical_loss": 3.573257111708842, + "tokens_seen": 1253113856 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003133099297893681, + "loss": 1.287, + "theoretical_loss": 3.5732400618712004, + "tokens_seen": 1253179392 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003132998996990973, + "loss": 1.4375, + "theoretical_loss": 3.573223013174813, + "tokens_seen": 1253244928 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003132898696088265, + "loss": 1.3918, + "theoretical_loss": 3.573205965619543, + "tokens_seen": 1253310464 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.0007298932760022581, + "objective/train/docs_used": 710893, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.641024351119995, + "objective/train/original_loss": 2.641024351119995, + "objective/train/theoretical_loss": 3.573188919205256, + "objective/train/tokens_used": 1273836000, + "objective/train/value_avg": -0.005512237548828125, + "objective/train/value_loss": 7.080622162902728e-05, + "objective/train/value_max": -0.00043892860412597656, + "objective/train/value_min": -0.05572509765625, + "objective/train/value_reward_corr": 0.45605658234503343, + "objective/train/value_std": 0.0048065185546875, + "objective/train/weight_avg": 0.9960278868675232, + "objective/train/weighted_lm_loss": 2.6223952770233154, + "objective/train/weights_max": 1.552805781364441, + "objective/train/weights_min": 0.4847222566604614, + "theoretical_loss": 3.573188919205256, + "tokens_seen": 1253376000 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031327983951855566, + "loss": 1.2541, + "theoretical_loss": 3.573188919205256, + "tokens_seen": 1253376000 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003132698094282849, + "loss": 1.4669, + "theoretical_loss": 3.5731718739318143, + "tokens_seen": 1253441536 + }, + { + "epoch": 0.38, + "learning_rate": 0.000313259779338014, + "loss": 1.3568, + "theoretical_loss": 3.5731548297990825, + "tokens_seen": 1253507072 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031324974924774326, + "loss": 1.3299, + "theoretical_loss": 3.573137786806925, + "tokens_seen": 1253572608 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003132397191574724, + "loss": 1.3673, + "theoretical_loss": 3.573120744955206, + "tokens_seen": 1253638144 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003132296890672016, + "loss": 1.3321, + "theoretical_loss": 3.5731037042437883, + "tokens_seen": 1253703680 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003132196589769308, + "loss": 1.404, + "theoretical_loss": 3.5730866646725374, + "tokens_seen": 1253769216 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031320962888666, + "loss": 1.4526, + "theoretical_loss": 3.573069626241317, + "tokens_seen": 1253834752 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031319959879638916, + "loss": 1.3845, + "theoretical_loss": 3.573052588949991, + "tokens_seen": 1253900288 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031318956870611834, + "loss": 1.325, + "theoretical_loss": 3.573035552798424, + "tokens_seen": 1253965824 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003131795386158475, + "loss": 1.2995, + "theoretical_loss": 3.5730185177864797, + "tokens_seen": 1254031360 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031316950852557676, + "loss": 1.3879, + "theoretical_loss": 3.5730014839140223, + "tokens_seen": 1254096896 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003131594784353059, + "loss": 1.3324, + "theoretical_loss": 3.5729844511809166, + "tokens_seen": 1254162432 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003131494483450351, + "loss": 1.3626, + "theoretical_loss": 3.5729674195870262, + "tokens_seen": 1254227968 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031313941825476425, + "loss": 1.4159, + "theoretical_loss": 3.572950389132216, + "tokens_seen": 1254293504 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003131293881644935, + "loss": 1.3879, + "theoretical_loss": 3.5729333598163495, + "tokens_seen": 1254359040 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031311935807422267, + "loss": 1.3005, + "theoretical_loss": 3.5729163316392922, + "tokens_seen": 1254424576 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031310932798395185, + "loss": 1.4137, + "theoretical_loss": 3.5728993046009077, + "tokens_seen": 1254490112 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031309929789368103, + "loss": 1.3857, + "theoretical_loss": 3.57288227870106, + "tokens_seen": 1254555648 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031308926780341026, + "loss": 1.303, + "theoretical_loss": 3.5728652539396144, + "tokens_seen": 1254621184 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003130792377131394, + "loss": 1.3886, + "theoretical_loss": 3.572848230316435, + "tokens_seen": 1254686720 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003130692076228686, + "loss": 1.3395, + "theoretical_loss": 3.5728312078313857, + "tokens_seen": 1254752256 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003130591775325978, + "loss": 1.4113, + "theoretical_loss": 3.5728141864843312, + "tokens_seen": 1254817792 + }, + { + "epoch": 0.38, + "learning_rate": 0.000313049147442327, + "loss": 1.2902, + "theoretical_loss": 3.5727971662751363, + "tokens_seen": 1254883328 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003130391173520562, + "loss": 1.3753, + "theoretical_loss": 3.572780147203666, + "tokens_seen": 1254948864 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -4.46679478045553e-05, + "objective/train/docs_used": 712142, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7817816734313965, + "objective/train/original_loss": 2.7817819118499756, + "objective/train/theoretical_loss": 3.5727631292697843, + "objective/train/tokens_used": 1275474400, + "objective/train/value_avg": -0.005176544189453125, + "objective/train/value_loss": 5.409595905803144e-05, + "objective/train/value_max": -0.00018966197967529297, + "objective/train/value_min": -0.0782470703125, + "objective/train/value_reward_corr": 0.3235156696676195, + "objective/train/value_std": 0.004085540771484375, + "objective/train/weight_avg": 1.002092957496643, + "objective/train/weighted_lm_loss": 2.802234172821045, + "objective/train/weights_max": 1.972650170326233, + "objective/train/weights_min": 0.2935352325439453, + "theoretical_loss": 3.5727631292697843, + "tokens_seen": 1255014400 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031302908726178535, + "loss": 1.3329, + "theoretical_loss": 3.5727631292697843, + "tokens_seen": 1255014400 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003130190571715146, + "loss": 1.3488, + "theoretical_loss": 3.5727461124733555, + "tokens_seen": 1255079936 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003130090270812437, + "loss": 1.4345, + "theoretical_loss": 3.5727290968142444, + "tokens_seen": 1255145472 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031299899699097295, + "loss": 1.3408, + "theoretical_loss": 3.5727120822923153, + "tokens_seen": 1255211008 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031298896690070213, + "loss": 1.2985, + "theoretical_loss": 3.5726950689074335, + "tokens_seen": 1255276544 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003129789368104313, + "loss": 1.2776, + "theoretical_loss": 3.572678056659463, + "tokens_seen": 1255342080 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003129689067201605, + "loss": 1.3444, + "theoretical_loss": 3.5726610455482692, + "tokens_seen": 1255407616 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003129588766298897, + "loss": 1.2937, + "theoretical_loss": 3.572644035573716, + "tokens_seen": 1255473152 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031294884653961885, + "loss": 1.2653, + "theoretical_loss": 3.5726270267356686, + "tokens_seen": 1255538688 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003129388164493481, + "loss": 1.3262, + "theoretical_loss": 3.5726100190339922, + "tokens_seen": 1255604224 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003129287863590772, + "loss": 1.3774, + "theoretical_loss": 3.5725930124685505, + "tokens_seen": 1255669760 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031291875626880645, + "loss": 1.3406, + "theoretical_loss": 3.572576007039209, + "tokens_seen": 1255735296 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031290872617853563, + "loss": 1.3545, + "theoretical_loss": 3.5725590027458325, + "tokens_seen": 1255800832 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003128986960882648, + "loss": 1.3802, + "theoretical_loss": 3.572541999588285, + "tokens_seen": 1255866368 + }, + { + "epoch": 0.38, + "learning_rate": 0.000312888665997994, + "loss": 1.3802, + "theoretical_loss": 3.5725249975664326, + "tokens_seen": 1255931904 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003128786359077232, + "loss": 1.319, + "theoretical_loss": 3.57250799668014, + "tokens_seen": 1255997440 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031286860581745236, + "loss": 1.5132, + "theoretical_loss": 3.5724909969292713, + "tokens_seen": 1256062976 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003128585757271816, + "loss": 1.5149, + "theoretical_loss": 3.572473998313692, + "tokens_seen": 1256128512 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003128485456369107, + "loss": 1.4481, + "theoretical_loss": 3.572457000833267, + "tokens_seen": 1256194048 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031283851554663995, + "loss": 1.3235, + "theoretical_loss": 3.572440004487861, + "tokens_seen": 1256259584 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003128284854563691, + "loss": 1.2911, + "theoretical_loss": 3.5724230092773395, + "tokens_seen": 1256325120 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003128184553660983, + "loss": 1.3084, + "theoretical_loss": 3.572406015201567, + "tokens_seen": 1256390656 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003128084252758275, + "loss": 1.3037, + "theoretical_loss": 3.5723890222604093, + "tokens_seen": 1256456192 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003127983951855567, + "loss": 1.3804, + "theoretical_loss": 3.5723720304537308, + "tokens_seen": 1256521728 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031278836509528586, + "loss": 1.3284, + "theoretical_loss": 3.5723550397813963, + "tokens_seen": 1256587264 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.0015633301809430122, + "objective/train/docs_used": 712613, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9949357509613037, + "objective/train/original_loss": 2.9949357509613037, + "objective/train/theoretical_loss": 3.572338050243272, + "objective/train/tokens_used": 1277112800, + "objective/train/value_avg": -0.00655364990234375, + "objective/train/value_loss": 0.00010649304749676958, + "objective/train/value_max": -0.0004355907440185547, + "objective/train/value_min": -0.09173583984375, + "objective/train/value_reward_corr": 0.3306044679258135, + "objective/train/value_std": 0.006458282470703125, + "objective/train/weight_avg": 0.9894128441810608, + "objective/train/weighted_lm_loss": 2.994668483734131, + "objective/train/weights_max": 1.791911005973816, + "objective/train/weights_min": 0.5677884817123413, + "theoretical_loss": 3.572338050243272, + "tokens_seen": 1256652800 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003127783350050151, + "loss": 1.4814, + "theoretical_loss": 3.572338050243272, + "tokens_seen": 1256652800 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003127683049147442, + "loss": 1.3912, + "theoretical_loss": 3.572321061839222, + "tokens_seen": 1256718336 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031275827482447346, + "loss": 1.4521, + "theoretical_loss": 3.5723040745691117, + "tokens_seen": 1256783872 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003127482447342026, + "loss": 1.3025, + "theoretical_loss": 3.572287088432807, + "tokens_seen": 1256849408 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003127382146439318, + "loss": 1.4481, + "theoretical_loss": 3.5722701034301725, + "tokens_seen": 1256914944 + }, + { + "epoch": 0.38, + "learning_rate": 0.000312728184553661, + "loss": 1.379, + "theoretical_loss": 3.572253119561073, + "tokens_seen": 1256980480 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003127181544633902, + "loss": 1.2901, + "theoretical_loss": 3.572236136825375, + "tokens_seen": 1257046016 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031270812437311936, + "loss": 1.3898, + "theoretical_loss": 3.5722191552229425, + "tokens_seen": 1257111552 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031269809428284854, + "loss": 1.3839, + "theoretical_loss": 3.5722021747536417, + "tokens_seen": 1257177088 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003126880641925777, + "loss": 1.3508, + "theoretical_loss": 3.5721851954173376, + "tokens_seen": 1257242624 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031267803410230696, + "loss": 1.3844, + "theoretical_loss": 3.5721682172138953, + "tokens_seen": 1257308160 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003126680040120361, + "loss": 1.4069, + "theoretical_loss": 3.5721512401431808, + "tokens_seen": 1257373696 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003126579739217653, + "loss": 1.377, + "theoretical_loss": 3.572134264205059, + "tokens_seen": 1257439232 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031264794383149445, + "loss": 1.3872, + "theoretical_loss": 3.5721172893993955, + "tokens_seen": 1257504768 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003126379137412237, + "loss": 1.3637, + "theoretical_loss": 3.5721003157260554, + "tokens_seen": 1257570304 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031262788365095287, + "loss": 1.3419, + "theoretical_loss": 3.5720833431849046, + "tokens_seen": 1257635840 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031261785356068205, + "loss": 1.4267, + "theoretical_loss": 3.5720663717758083, + "tokens_seen": 1257701376 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031260782347041123, + "loss": 1.3955, + "theoretical_loss": 3.5720494014986324, + "tokens_seen": 1257766912 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031259779338014046, + "loss": 1.404, + "theoretical_loss": 3.5720324323532417, + "tokens_seen": 1257832448 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003125877632898696, + "loss": 1.4074, + "theoretical_loss": 3.5720154643395023, + "tokens_seen": 1257897984 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003125777331995988, + "loss": 1.3809, + "theoretical_loss": 3.57199849745728, + "tokens_seen": 1257963520 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031256770310932795, + "loss": 1.4148, + "theoretical_loss": 3.57198153170644, + "tokens_seen": 1258029056 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003125576730190572, + "loss": 1.2648, + "theoretical_loss": 3.571964567086848, + "tokens_seen": 1258094592 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031254764292878637, + "loss": 1.3476, + "theoretical_loss": 3.57194760359837, + "tokens_seen": 1258160128 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031253761283851555, + "loss": 1.382, + "theoretical_loss": 3.5719306412408707, + "tokens_seen": 1258225664 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.0006191346328705549, + "objective/train/docs_used": 713934, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.570558547973633, + "objective/train/original_loss": 2.5705583095550537, + "objective/train/theoretical_loss": 3.571913680014217, + "objective/train/tokens_used": 1278751200, + "objective/train/value_avg": -0.004711151123046875, + "objective/train/value_loss": 8.369031274924055e-05, + "objective/train/value_max": -0.00020825862884521484, + "objective/train/value_min": -0.037384033203125, + "objective/train/value_reward_corr": 0.3324871729614143, + "objective/train/value_std": 0.00347900390625, + "objective/train/weight_avg": 0.9974455833435059, + "objective/train/weighted_lm_loss": 2.5589137077331543, + "objective/train/weights_max": 1.3929460048675537, + "objective/train/weights_min": 0.1412535011768341, + "theoretical_loss": 3.571913680014217, + "tokens_seen": 1258291200 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031252758274824473, + "loss": 1.3513, + "theoretical_loss": 3.571913680014217, + "tokens_seen": 1258291200 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003125175526579739, + "loss": 1.3766, + "theoretical_loss": 3.5718967199182736, + "tokens_seen": 1258356736 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003125075225677031, + "loss": 1.3785, + "theoretical_loss": 3.571879760952907, + "tokens_seen": 1258422272 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031249749247743233, + "loss": 1.3713, + "theoretical_loss": 3.5718628031179827, + "tokens_seen": 1258487808 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031248746238716146, + "loss": 1.2775, + "theoretical_loss": 3.571845846413366, + "tokens_seen": 1258553344 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003124774322968907, + "loss": 1.4274, + "theoretical_loss": 3.5718288908389235, + "tokens_seen": 1258618880 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003124674022066198, + "loss": 1.2876, + "theoretical_loss": 3.571811936394521, + "tokens_seen": 1258684416 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031245737211634905, + "loss": 1.3173, + "theoretical_loss": 3.5717949830800233, + "tokens_seen": 1258749952 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031244734202607823, + "loss": 1.3816, + "theoretical_loss": 3.5717780308952976, + "tokens_seen": 1258815488 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003124373119358074, + "loss": 1.3729, + "theoretical_loss": 3.5717610798402086, + "tokens_seen": 1258881024 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003124272818455366, + "loss": 1.3974, + "theoretical_loss": 3.5717441299146238, + "tokens_seen": 1258946560 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031241725175526583, + "loss": 1.417, + "theoretical_loss": 3.5717271811184075, + "tokens_seen": 1259012096 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031240722166499496, + "loss": 1.3562, + "theoretical_loss": 3.571710233451426, + "tokens_seen": 1259077632 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003123971915747242, + "loss": 1.416, + "theoretical_loss": 3.571693286913546, + "tokens_seen": 1259143168 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003123871614844533, + "loss": 1.3268, + "theoretical_loss": 3.5716763415046335, + "tokens_seen": 1259208704 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031237713139418256, + "loss": 1.3676, + "theoretical_loss": 3.571659397224554, + "tokens_seen": 1259274240 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031236710130391174, + "loss": 1.3097, + "theoretical_loss": 3.5716424540731735, + "tokens_seen": 1259339776 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003123570712136409, + "loss": 1.4839, + "theoretical_loss": 3.5716255120503586, + "tokens_seen": 1259405312 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003123470411233701, + "loss": 1.3499, + "theoretical_loss": 3.5716085711559753, + "tokens_seen": 1259470848 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003123370110330993, + "loss": 1.3508, + "theoretical_loss": 3.5715916313898894, + "tokens_seen": 1259536384 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031232698094282846, + "loss": 1.3499, + "theoretical_loss": 3.571574692751967, + "tokens_seen": 1259601920 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003123169508525577, + "loss": 1.4263, + "theoretical_loss": 3.5715577552420745, + "tokens_seen": 1259667456 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003123069207622869, + "loss": 1.5353, + "theoretical_loss": 3.571540818860078, + "tokens_seen": 1259732992 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031229689067201606, + "loss": 1.3061, + "theoretical_loss": 3.5715238836058445, + "tokens_seen": 1259798528 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003122868605817453, + "loss": 1.4891, + "theoretical_loss": 3.5715069494792395, + "tokens_seen": 1259864064 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.0017165289027616382, + "objective/train/docs_used": 714588, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9801383018493652, + "objective/train/original_loss": 2.9801385402679443, + "objective/train/theoretical_loss": 3.5714900164801286, + "objective/train/tokens_used": 1280389600, + "objective/train/value_avg": -0.006549835205078125, + "objective/train/value_loss": 8.31943325465545e-05, + "objective/train/value_max": -0.00034737586975097656, + "objective/train/value_min": -0.07696533203125, + "objective/train/value_reward_corr": 0.5097918939590722, + "objective/train/value_std": 0.006500244140625, + "objective/train/weight_avg": 0.9867137670516968, + "objective/train/weighted_lm_loss": 2.9435842037200928, + "objective/train/weights_max": 1.7273359298706055, + "objective/train/weights_min": 0.40801751613616943, + "theoretical_loss": 3.5714900164801286, + "tokens_seen": 1259929600 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003122768304914744, + "loss": 1.3447, + "theoretical_loss": 3.5714900164801286, + "tokens_seen": 1259929600 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031226680040120366, + "loss": 1.3883, + "theoretical_loss": 3.571473084608379, + "tokens_seen": 1259995136 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003122567703109328, + "loss": 1.2976, + "theoretical_loss": 3.571456153863857, + "tokens_seen": 1260060672 + }, + { + "epoch": 0.38, + "learning_rate": 0.000312246740220662, + "loss": 1.4023, + "theoretical_loss": 3.571439224246429, + "tokens_seen": 1260126208 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003122367101303912, + "loss": 1.3775, + "theoretical_loss": 3.5714222957559607, + "tokens_seen": 1260191744 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003122266800401204, + "loss": 1.3522, + "theoretical_loss": 3.571405368392319, + "tokens_seen": 1260257280 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031221664994984956, + "loss": 1.2044, + "theoretical_loss": 3.5713884421553703, + "tokens_seen": 1260322816 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031220661985957874, + "loss": 1.4196, + "theoretical_loss": 3.571371517044981, + "tokens_seen": 1260388352 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003121965897693079, + "loss": 1.313, + "theoretical_loss": 3.571354593061017, + "tokens_seen": 1260453888 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031218655967903716, + "loss": 1.3571, + "theoretical_loss": 3.571337670203346, + "tokens_seen": 1260519424 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003121765295887663, + "loss": 1.5045, + "theoretical_loss": 3.571320748471833, + "tokens_seen": 1260584960 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003121664994984955, + "loss": 1.4666, + "theoretical_loss": 3.5713038278663456, + "tokens_seen": 1260650496 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031215646940822465, + "loss": 1.304, + "theoretical_loss": 3.5712869083867496, + "tokens_seen": 1260716032 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003121464393179539, + "loss": 1.4658, + "theoretical_loss": 3.5712699900329126, + "tokens_seen": 1260781568 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031213640922768307, + "loss": 1.318, + "theoretical_loss": 3.5712530728047005, + "tokens_seen": 1260847104 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031212637913741225, + "loss": 1.3565, + "theoretical_loss": 3.5712361567019792, + "tokens_seen": 1260912640 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031211634904714143, + "loss": 1.3973, + "theoretical_loss": 3.571219241724617, + "tokens_seen": 1260978176 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031210631895687066, + "loss": 1.3743, + "theoretical_loss": 3.571202327872479, + "tokens_seen": 1261043712 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003120962888665998, + "loss": 1.4101, + "theoretical_loss": 3.5711854151454325, + "tokens_seen": 1261109248 + }, + { + "epoch": 0.38, + "learning_rate": 0.000312086258776329, + "loss": 1.3909, + "theoretical_loss": 3.5711685035433445, + "tokens_seen": 1261174784 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031207622868605815, + "loss": 1.4029, + "theoretical_loss": 3.571151593066081, + "tokens_seen": 1261240320 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003120661985957874, + "loss": 1.4237, + "theoretical_loss": 3.5711346837135096, + "tokens_seen": 1261305856 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031205616850551657, + "loss": 1.3849, + "theoretical_loss": 3.5711177754854964, + "tokens_seen": 1261371392 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031204613841524575, + "loss": 1.4806, + "theoretical_loss": 3.571100868381909, + "tokens_seen": 1261436928 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031203610832497493, + "loss": 1.3122, + "theoretical_loss": 3.571083962402613, + "tokens_seen": 1261502464 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": 0.00012204131053294986, + "objective/train/docs_used": 715954, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7935187816619873, + "objective/train/original_loss": 2.793518543243408, + "objective/train/theoretical_loss": 3.5710670575474763, + "objective/train/tokens_used": 1282028000, + "objective/train/value_avg": -0.005786895751953125, + "objective/train/value_loss": 8.566679753130302e-05, + "objective/train/value_max": -0.00044083595275878906, + "objective/train/value_min": -0.0638427734375, + "objective/train/value_reward_corr": 0.3695813050454375, + "objective/train/value_std": 0.00499725341796875, + "objective/train/weight_avg": 1.005096197128296, + "objective/train/weighted_lm_loss": 2.8111822605133057, + "objective/train/weights_max": 1.715844750404358, + "objective/train/weights_min": 0.2460145801305771, + "theoretical_loss": 3.5710670575474763, + "tokens_seen": 1261568000 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003120260782347041, + "loss": 1.3228, + "theoretical_loss": 3.5710670575474763, + "tokens_seen": 1261568000 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003120160481444333, + "loss": 1.3486, + "theoretical_loss": 3.5710501538163646, + "tokens_seen": 1261633536 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031200601805416253, + "loss": 1.4538, + "theoretical_loss": 3.571033251209146, + "tokens_seen": 1261699072 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031199598796389166, + "loss": 1.3306, + "theoretical_loss": 3.571016349725687, + "tokens_seen": 1261764608 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003119859578736209, + "loss": 1.343, + "theoretical_loss": 3.5709994493658543, + "tokens_seen": 1261830144 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031197592778335, + "loss": 1.5249, + "theoretical_loss": 3.570982550129515, + "tokens_seen": 1261895680 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031196589769307925, + "loss": 1.3178, + "theoretical_loss": 3.570965652016536, + "tokens_seen": 1261961216 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031195586760280843, + "loss": 1.4296, + "theoretical_loss": 3.5709487550267847, + "tokens_seen": 1262026752 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003119458375125376, + "loss": 1.3041, + "theoretical_loss": 3.570931859160127, + "tokens_seen": 1262092288 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003119358074222668, + "loss": 1.3637, + "theoretical_loss": 3.5709149644164313, + "tokens_seen": 1262157824 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031192577733199603, + "loss": 1.2831, + "theoretical_loss": 3.5708980707955638, + "tokens_seen": 1262223360 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031191574724172516, + "loss": 1.4206, + "theoretical_loss": 3.570881178297392, + "tokens_seen": 1262288896 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003119057171514544, + "loss": 1.3596, + "theoretical_loss": 3.5708642869217826, + "tokens_seen": 1262354432 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003118956870611835, + "loss": 1.4133, + "theoretical_loss": 3.5708473966686034, + "tokens_seen": 1262419968 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031188565697091276, + "loss": 1.3464, + "theoretical_loss": 3.5708305075377207, + "tokens_seen": 1262485504 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031187562688064194, + "loss": 1.3293, + "theoretical_loss": 3.5708136195290026, + "tokens_seen": 1262551040 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003118655967903711, + "loss": 1.3754, + "theoretical_loss": 3.5707967326423153, + "tokens_seen": 1262616576 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003118555667001003, + "loss": 1.3439, + "theoretical_loss": 3.5707798468775267, + "tokens_seen": 1262682112 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003118455366098295, + "loss": 1.3901, + "theoretical_loss": 3.570762962234504, + "tokens_seen": 1262747648 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031183550651955866, + "loss": 1.2872, + "theoretical_loss": 3.570746078713114, + "tokens_seen": 1262813184 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003118254764292879, + "loss": 1.3482, + "theoretical_loss": 3.5707291963132244, + "tokens_seen": 1262878720 + }, + { + "epoch": 0.38, + "learning_rate": 0.000311815446339017, + "loss": 1.3557, + "theoretical_loss": 3.5707123150347027, + "tokens_seen": 1262944256 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031180541624874626, + "loss": 1.4583, + "theoretical_loss": 3.5706954348774156, + "tokens_seen": 1263009792 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003117953861584754, + "loss": 1.3551, + "theoretical_loss": 3.5706785558412313, + "tokens_seen": 1263075328 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003117853560682046, + "loss": 1.4204, + "theoretical_loss": 3.5706616779260165, + "tokens_seen": 1263140864 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.000302395987091586, + "objective/train/docs_used": 716702, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0051965713500977, + "objective/train/original_loss": 3.0051963329315186, + "objective/train/theoretical_loss": 3.570644801131639, + "objective/train/tokens_used": 1283666400, + "objective/train/value_avg": -0.00531768798828125, + "objective/train/value_loss": 5.3385065257316455e-05, + "objective/train/value_max": -0.00038909912109375, + "objective/train/value_min": -0.044097900390625, + "objective/train/value_reward_corr": 0.3810308816011231, + "objective/train/value_std": 0.003925323486328125, + "objective/train/weight_avg": 0.9994899034500122, + "objective/train/weighted_lm_loss": 3.0138025283813477, + "objective/train/weights_max": 1.4082056283950806, + "objective/train/weights_min": 0.38615548610687256, + "theoretical_loss": 3.570644801131639, + "tokens_seen": 1263206400 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003117753259779338, + "loss": 1.4254, + "theoretical_loss": 3.570644801131639, + "tokens_seen": 1263206400 + }, + { + "epoch": 0.38, + "learning_rate": 0.000311765295887663, + "loss": 1.2538, + "theoretical_loss": 3.5706279254579654, + "tokens_seen": 1263271936 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031175526579739217, + "loss": 1.3378, + "theoretical_loss": 3.570611050904864, + "tokens_seen": 1263337472 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003117452357071214, + "loss": 1.3573, + "theoretical_loss": 3.570594177472202, + "tokens_seen": 1263403008 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031173520561685053, + "loss": 1.2696, + "theoretical_loss": 3.5705773051598477, + "tokens_seen": 1263468544 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031172517552657976, + "loss": 1.2554, + "theoretical_loss": 3.5705604339676666, + "tokens_seen": 1263534080 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003117151454363089, + "loss": 1.4122, + "theoretical_loss": 3.5705435638955283, + "tokens_seen": 1263599616 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003117051153460381, + "loss": 1.4515, + "theoretical_loss": 3.5705266949432994, + "tokens_seen": 1263665152 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003116950852557673, + "loss": 1.3666, + "theoretical_loss": 3.5705098271108477, + "tokens_seen": 1263730688 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003116850551654965, + "loss": 1.462, + "theoretical_loss": 3.57049296039804, + "tokens_seen": 1263796224 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031167502507522567, + "loss": 1.4139, + "theoretical_loss": 3.5704760948047456, + "tokens_seen": 1263861760 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031166499498495485, + "loss": 1.5104, + "theoretical_loss": 3.5704592303308305, + "tokens_seen": 1263927296 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031165496489468403, + "loss": 1.4338, + "theoretical_loss": 3.570442366976163, + "tokens_seen": 1263992832 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031164493480441327, + "loss": 1.3315, + "theoretical_loss": 3.5704255047406113, + "tokens_seen": 1264058368 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003116349047141424, + "loss": 1.3074, + "theoretical_loss": 3.5704086436240425, + "tokens_seen": 1264123904 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031162487462387163, + "loss": 1.44, + "theoretical_loss": 3.5703917836263246, + "tokens_seen": 1264189440 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003116148445336008, + "loss": 1.3616, + "theoretical_loss": 3.570374924747325, + "tokens_seen": 1264254976 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031160481444333, + "loss": 1.4357, + "theoretical_loss": 3.5703580669869126, + "tokens_seen": 1264320512 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031159478435305917, + "loss": 1.3387, + "theoretical_loss": 3.5703412103449534, + "tokens_seen": 1264386048 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031158475426278835, + "loss": 1.3662, + "theoretical_loss": 3.5703243548213166, + "tokens_seen": 1264451584 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031157472417251753, + "loss": 1.2926, + "theoretical_loss": 3.5703075004158693, + "tokens_seen": 1264517120 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031156469408224677, + "loss": 1.408, + "theoretical_loss": 3.57029064712848, + "tokens_seen": 1264582656 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031155466399197595, + "loss": 1.317, + "theoretical_loss": 3.570273794959016, + "tokens_seen": 1264648192 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031154463390170513, + "loss": 1.3156, + "theoretical_loss": 3.5702569439073457, + "tokens_seen": 1264713728 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003115346038114343, + "loss": 1.2927, + "theoretical_loss": 3.5702400939733367, + "tokens_seen": 1264779264 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.001210279413498938, + "objective/train/docs_used": 717941, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.969787836074829, + "objective/train/original_loss": 2.969787359237671, + "objective/train/theoretical_loss": 3.570223245156858, + "objective/train/tokens_used": 1285304800, + "objective/train/value_avg": -0.005130767822265625, + "objective/train/value_loss": 0.0001128536750911735, + "objective/train/value_max": -0.0004601478576660156, + "objective/train/value_min": -0.0850830078125, + "objective/train/value_reward_corr": 0.22326804906128886, + "objective/train/value_std": 0.004241943359375, + "objective/train/weight_avg": 0.9926717877388, + "objective/train/weighted_lm_loss": 2.951857805252075, + "objective/train/weights_max": 2.1206698417663574, + "objective/train/weights_min": 0.14367729425430298, + "theoretical_loss": 3.570223245156858, + "tokens_seen": 1264844800 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003115245737211635, + "loss": 1.3064, + "theoretical_loss": 3.570223245156858, + "tokens_seen": 1264844800 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031151454363089273, + "loss": 1.4043, + "theoretical_loss": 3.570206397457776, + "tokens_seen": 1264910336 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031150451354062186, + "loss": 1.4057, + "theoretical_loss": 3.570189550875959, + "tokens_seen": 1264975872 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003114944834503511, + "loss": 1.3066, + "theoretical_loss": 3.570172705411276, + "tokens_seen": 1265041408 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003114844533600802, + "loss": 1.2846, + "theoretical_loss": 3.5701558610635944, + "tokens_seen": 1265106944 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031147442326980945, + "loss": 1.32, + "theoretical_loss": 3.5701390178327825, + "tokens_seen": 1265172480 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031146439317953863, + "loss": 1.4515, + "theoretical_loss": 3.5701221757187076, + "tokens_seen": 1265238016 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003114543630892678, + "loss": 1.4655, + "theoretical_loss": 3.570105334721239, + "tokens_seen": 1265303552 + }, + { + "epoch": 0.38, + "learning_rate": 0.000311444332998997, + "loss": 1.2918, + "theoretical_loss": 3.5700884948402445, + "tokens_seen": 1265369088 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031143430290872623, + "loss": 1.4372, + "theoretical_loss": 3.570071656075592, + "tokens_seen": 1265434624 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031142427281845536, + "loss": 1.4463, + "theoretical_loss": 3.57005481842715, + "tokens_seen": 1265500160 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003114142427281846, + "loss": 1.4, + "theoretical_loss": 3.570037981894786, + "tokens_seen": 1265565696 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003114042126379137, + "loss": 1.3577, + "theoretical_loss": 3.5700211464783687, + "tokens_seen": 1265631232 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031139418254764296, + "loss": 1.4609, + "theoretical_loss": 3.570004312177767, + "tokens_seen": 1265696768 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031138415245737214, + "loss": 1.3157, + "theoretical_loss": 3.5699874789928483, + "tokens_seen": 1265762304 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003113741223671013, + "loss": 1.3514, + "theoretical_loss": 3.5699706469234807, + "tokens_seen": 1265827840 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003113640922768305, + "loss": 1.3195, + "theoretical_loss": 3.5699538159695337, + "tokens_seen": 1265893376 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003113540621865597, + "loss": 1.3804, + "theoretical_loss": 3.569936986130874, + "tokens_seen": 1265958912 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031134403209628886, + "loss": 1.4166, + "theoretical_loss": 3.5699201574073713, + "tokens_seen": 1266024448 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003113340020060181, + "loss": 1.3909, + "theoretical_loss": 3.569903329798894, + "tokens_seen": 1266089984 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003113239719157472, + "loss": 1.2769, + "theoretical_loss": 3.5698865033053098, + "tokens_seen": 1266155520 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031131394182547646, + "loss": 1.3611, + "theoretical_loss": 3.569869677926487, + "tokens_seen": 1266221056 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003113039117352056, + "loss": 1.4385, + "theoretical_loss": 3.569852853662295, + "tokens_seen": 1266286592 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003112938816449348, + "loss": 1.3233, + "theoretical_loss": 3.569836030512602, + "tokens_seen": 1266352128 + }, + { + "epoch": 0.38, + "learning_rate": 0.000311283851554664, + "loss": 1.35, + "theoretical_loss": 3.5698192084772757, + "tokens_seen": 1266417664 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": 7.109035504981875e-05, + "objective/train/docs_used": 718481, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6956632137298584, + "objective/train/original_loss": 2.6956632137298584, + "objective/train/theoretical_loss": 3.5698023875561855, + "objective/train/tokens_used": 1286943200, + "objective/train/value_avg": -0.005962371826171875, + "objective/train/value_loss": 8.36200633784756e-05, + "objective/train/value_max": -0.00047850608825683594, + "objective/train/value_min": -0.06256103515625, + "objective/train/value_reward_corr": 0.3567856978718531, + "objective/train/value_std": 0.00547027587890625, + "objective/train/weight_avg": 1.0044746398925781, + "objective/train/weighted_lm_loss": 2.7104716300964355, + "objective/train/weights_max": 1.6180254220962524, + "objective/train/weights_min": 0.24627280235290527, + "theoretical_loss": 3.5698023875561855, + "tokens_seen": 1266483200 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003112738214643932, + "loss": 1.3698, + "theoretical_loss": 3.5698023875561855, + "tokens_seen": 1266483200 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031126379137412237, + "loss": 1.3624, + "theoretical_loss": 3.569785567749199, + "tokens_seen": 1266548736 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003112537612838516, + "loss": 1.4049, + "theoretical_loss": 3.5697687490561862, + "tokens_seen": 1266614272 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031124373119358073, + "loss": 1.3531, + "theoretical_loss": 3.5697519314770148, + "tokens_seen": 1266679808 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031123370110330996, + "loss": 1.4713, + "theoretical_loss": 3.569735115011553, + "tokens_seen": 1266745344 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003112236710130391, + "loss": 1.3718, + "theoretical_loss": 3.5697182996596704, + "tokens_seen": 1266810880 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003112136409227683, + "loss": 1.3358, + "theoretical_loss": 3.5697014854212346, + "tokens_seen": 1266876416 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003112036108324975, + "loss": 1.3569, + "theoretical_loss": 3.5696846722961153, + "tokens_seen": 1266941952 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003111935807422267, + "loss": 1.4292, + "theoretical_loss": 3.5696678602841807, + "tokens_seen": 1267007488 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031118355065195587, + "loss": 1.2976, + "theoretical_loss": 3.5696510493852998, + "tokens_seen": 1267073024 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031117352056168505, + "loss": 1.4409, + "theoretical_loss": 3.5696342395993406, + "tokens_seen": 1267138560 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031116349047141423, + "loss": 1.427, + "theoretical_loss": 3.5696174309261726, + "tokens_seen": 1267204096 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031115346038114347, + "loss": 1.4676, + "theoretical_loss": 3.5696006233656643, + "tokens_seen": 1267269632 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003111434302908726, + "loss": 1.38, + "theoretical_loss": 3.569583816917685, + "tokens_seen": 1267335168 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031113340020060183, + "loss": 1.4157, + "theoretical_loss": 3.5695670115821025, + "tokens_seen": 1267400704 + }, + { + "epoch": 0.38, + "learning_rate": 0.000311123370110331, + "loss": 1.3407, + "theoretical_loss": 3.5695502073587866, + "tokens_seen": 1267466240 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003111133400200602, + "loss": 1.3824, + "theoretical_loss": 3.569533404247606, + "tokens_seen": 1267531776 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031110330992978937, + "loss": 1.3674, + "theoretical_loss": 3.569516602248429, + "tokens_seen": 1267597312 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031109327983951855, + "loss": 1.4557, + "theoretical_loss": 3.5694998013611254, + "tokens_seen": 1267662848 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031108324974924773, + "loss": 1.3684, + "theoretical_loss": 3.5694830015855636, + "tokens_seen": 1267728384 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031107321965897697, + "loss": 1.3611, + "theoretical_loss": 3.569466202921613, + "tokens_seen": 1267793920 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003110631895687061, + "loss": 1.2182, + "theoretical_loss": 3.5694494053691423, + "tokens_seen": 1267859456 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031105315947843533, + "loss": 1.4771, + "theoretical_loss": 3.56943260892802, + "tokens_seen": 1267924992 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031104312938816446, + "loss": 1.3599, + "theoretical_loss": 3.569415813598116, + "tokens_seen": 1267990528 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003110330992978937, + "loss": 1.3668, + "theoretical_loss": 3.569399019379299, + "tokens_seen": 1268056064 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.0011708532692864537, + "objective/train/docs_used": 719716, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9512906074523926, + "objective/train/original_loss": 2.9512908458709717, + "objective/train/theoretical_loss": 3.569382226271438, + "objective/train/tokens_used": 1288581600, + "objective/train/value_avg": -0.00481414794921875, + "objective/train/value_loss": 7.20950192771852e-05, + "objective/train/value_max": -0.0003981590270996094, + "objective/train/value_min": -0.0482177734375, + "objective/train/value_reward_corr": 0.32533811856875733, + "objective/train/value_std": 0.0036220550537109375, + "objective/train/weight_avg": 0.9915037751197815, + "objective/train/weighted_lm_loss": 2.9294583797454834, + "objective/train/weights_max": 1.5887565612792969, + "objective/train/weights_min": 0.21956071257591248, + "theoretical_loss": 3.569382226271438, + "tokens_seen": 1268121600 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003110230692076229, + "loss": 1.4295, + "theoretical_loss": 3.569382226271438, + "tokens_seen": 1268121600 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031101303911735206, + "loss": 1.3351, + "theoretical_loss": 3.569365434274402, + "tokens_seen": 1268187136 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031100300902708124, + "loss": 1.4439, + "theoretical_loss": 3.569348643388061, + "tokens_seen": 1268252672 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003109929789368104, + "loss": 1.4819, + "theoretical_loss": 3.5693318536122822, + "tokens_seen": 1268318208 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003109829488465396, + "loss": 1.3726, + "theoretical_loss": 3.569315064946937, + "tokens_seen": 1268383744 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031097291875626883, + "loss": 1.3202, + "theoretical_loss": 3.5692982773918933, + "tokens_seen": 1268449280 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031096288866599796, + "loss": 1.3352, + "theoretical_loss": 3.569281490947021, + "tokens_seen": 1268514816 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003109528585757272, + "loss": 1.4241, + "theoretical_loss": 3.5692647056121882, + "tokens_seen": 1268580352 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003109428284854564, + "loss": 1.3021, + "theoretical_loss": 3.5692479213872654, + "tokens_seen": 1268645888 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031093279839518556, + "loss": 1.3356, + "theoretical_loss": 3.569231138272121, + "tokens_seen": 1268711424 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031092276830491474, + "loss": 1.4255, + "theoretical_loss": 3.569214356266625, + "tokens_seen": 1268776960 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003109127382146439, + "loss": 1.3856, + "theoretical_loss": 3.569197575370646, + "tokens_seen": 1268842496 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003109027081243731, + "loss": 1.4292, + "theoretical_loss": 3.569180795584055, + "tokens_seen": 1268908032 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031089267803410234, + "loss": 1.3424, + "theoretical_loss": 3.569164016906719, + "tokens_seen": 1268973568 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031088264794383147, + "loss": 1.4678, + "theoretical_loss": 3.5691472393385086, + "tokens_seen": 1269039104 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003108726178535607, + "loss": 1.4311, + "theoretical_loss": 3.5691304628792935, + "tokens_seen": 1269104640 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031086258776328983, + "loss": 1.1848, + "theoretical_loss": 3.569113687528942, + "tokens_seen": 1269170176 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031085255767301906, + "loss": 1.2709, + "theoretical_loss": 3.569096913287325, + "tokens_seen": 1269235712 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031084252758274824, + "loss": 1.4124, + "theoretical_loss": 3.569080140154311, + "tokens_seen": 1269301248 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003108324974924774, + "loss": 1.3806, + "theoretical_loss": 3.56906336812977, + "tokens_seen": 1269366784 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003108224674022066, + "loss": 1.4273, + "theoretical_loss": 3.569046597213571, + "tokens_seen": 1269432320 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003108124373119358, + "loss": 1.3682, + "theoretical_loss": 3.569029827405584, + "tokens_seen": 1269497856 + }, + { + "epoch": 0.38, + "learning_rate": 0.000310802407221665, + "loss": 1.3668, + "theoretical_loss": 3.5690130587056785, + "tokens_seen": 1269563392 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003107923771313942, + "loss": 1.3975, + "theoretical_loss": 3.5689962911137236, + "tokens_seen": 1269628928 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003107823470411234, + "loss": 1.2449, + "theoretical_loss": 3.56897952462959, + "tokens_seen": 1269694464 + }, + { + "epoch": 0.38, + "objective/train/advantage_avg": -0.001950297737494111, + "objective/train/docs_used": 720440, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8277056217193604, + "objective/train/original_loss": 2.8277060985565186, + "objective/train/theoretical_loss": 3.5689627592531465, + "objective/train/tokens_used": 1290220000, + "objective/train/value_avg": -0.005115509033203125, + "objective/train/value_loss": 8.565014286432415e-05, + "objective/train/value_max": -0.0003998279571533203, + "objective/train/value_min": -0.060089111328125, + "objective/train/value_reward_corr": 0.42293730022542775, + "objective/train/value_std": 0.0038890838623046875, + "objective/train/weight_avg": 0.9845120310783386, + "objective/train/weighted_lm_loss": 2.795412540435791, + "objective/train/weights_max": 1.4515321254730225, + "objective/train/weights_min": 0.5362036228179932, + "theoretical_loss": 3.5689627592531465, + "tokens_seen": 1269760000 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031077231695085257, + "loss": 1.4475, + "theoretical_loss": 3.5689627592531465, + "tokens_seen": 1269760000 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003107622868605818, + "loss": 1.3771, + "theoretical_loss": 3.5689459949842623, + "tokens_seen": 1269825536 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031075225677031093, + "loss": 1.3583, + "theoretical_loss": 3.568929231822808, + "tokens_seen": 1269891072 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031074222668004016, + "loss": 1.3575, + "theoretical_loss": 3.5689124697686534, + "tokens_seen": 1269956608 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003107321965897693, + "loss": 1.314, + "theoretical_loss": 3.568895708821667, + "tokens_seen": 1270022144 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003107221664994985, + "loss": 1.3415, + "theoretical_loss": 3.5688789489817196, + "tokens_seen": 1270087680 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003107121364092277, + "loss": 1.3582, + "theoretical_loss": 3.568862190248681, + "tokens_seen": 1270153216 + }, + { + "epoch": 0.38, + "learning_rate": 0.0003107021063189569, + "loss": 1.2518, + "theoretical_loss": 3.568845432622421, + "tokens_seen": 1270218752 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031069207622868607, + "loss": 1.2969, + "theoretical_loss": 3.5688286761028083, + "tokens_seen": 1270284288 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031068204613841525, + "loss": 1.3784, + "theoretical_loss": 3.568811920689714, + "tokens_seen": 1270349824 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031067201604814443, + "loss": 1.4439, + "theoretical_loss": 3.5687951663830075, + "tokens_seen": 1270415360 + }, + { + "epoch": 0.38, + "learning_rate": 0.00031066198595787367, + "loss": 1.4327, + "theoretical_loss": 3.5687784131825584, + "tokens_seen": 1270480896 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003106519558676028, + "loss": 1.428, + "theoretical_loss": 3.568761661088237, + "tokens_seen": 1270546432 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031064192577733203, + "loss": 1.4346, + "theoretical_loss": 3.568744910099913, + "tokens_seen": 1270611968 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003106318956870612, + "loss": 1.4183, + "theoretical_loss": 3.5687281602174563, + "tokens_seen": 1270677504 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003106218655967904, + "loss": 1.4187, + "theoretical_loss": 3.568711411440737, + "tokens_seen": 1270743040 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031061183550651957, + "loss": 1.3788, + "theoretical_loss": 3.568694663769625, + "tokens_seen": 1270808576 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031060180541624875, + "loss": 1.3068, + "theoretical_loss": 3.5686779172039906, + "tokens_seen": 1270874112 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031059177532597793, + "loss": 1.4016, + "theoretical_loss": 3.5686611717437033, + "tokens_seen": 1270939648 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031058174523570717, + "loss": 1.3855, + "theoretical_loss": 3.5686444273886337, + "tokens_seen": 1271005184 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003105717151454363, + "loss": 1.2832, + "theoretical_loss": 3.5686276841386517, + "tokens_seen": 1271070720 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031056168505516553, + "loss": 1.3667, + "theoretical_loss": 3.5686109419936267, + "tokens_seen": 1271136256 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031055165496489466, + "loss": 1.3077, + "theoretical_loss": 3.56859420095343, + "tokens_seen": 1271201792 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003105416248746239, + "loss": 1.2534, + "theoretical_loss": 3.5685774610179313, + "tokens_seen": 1271267328 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003105315947843531, + "loss": 1.3762, + "theoretical_loss": 3.5685607221870006, + "tokens_seen": 1271332864 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": 0.00018899721908383071, + "objective/train/docs_used": 721295, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3176016807556152, + "objective/train/original_loss": 3.317601442337036, + "objective/train/theoretical_loss": 3.568543984460508, + "objective/train/tokens_used": 1291858400, + "objective/train/value_avg": -0.00543975830078125, + "objective/train/value_loss": 4.9984140787273645e-05, + "objective/train/value_max": -0.0003006458282470703, + "objective/train/value_min": -0.049224853515625, + "objective/train/value_reward_corr": 0.43411122886797077, + "objective/train/value_std": 0.004913330078125, + "objective/train/weight_avg": 1.004335880279541, + "objective/train/weighted_lm_loss": 3.3293137550354004, + "objective/train/weights_max": 1.6359906196594238, + "objective/train/weights_min": 0.5063507556915283, + "theoretical_loss": 3.568543984460508, + "tokens_seen": 1271398400 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031052156469408226, + "loss": 1.4568, + "theoretical_loss": 3.568543984460508, + "tokens_seen": 1271398400 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031051153460381144, + "loss": 1.3211, + "theoretical_loss": 3.568527247838323, + "tokens_seen": 1271463936 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003105015045135406, + "loss": 1.3028, + "theoretical_loss": 3.5685105123203176, + "tokens_seen": 1271529472 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003104914744232698, + "loss": 1.2944, + "theoretical_loss": 3.568493777906361, + "tokens_seen": 1271595008 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031048144433299904, + "loss": 1.4404, + "theoretical_loss": 3.568477044596323, + "tokens_seen": 1271660544 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031047141424272816, + "loss": 1.3812, + "theoretical_loss": 3.568460312390075, + "tokens_seen": 1271726080 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003104613841524574, + "loss": 1.3105, + "theoretical_loss": 3.568443581287487, + "tokens_seen": 1271791616 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003104513540621866, + "loss": 1.4183, + "theoretical_loss": 3.5684268512884283, + "tokens_seen": 1271857152 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031044132397191576, + "loss": 1.3729, + "theoretical_loss": 3.5684101223927702, + "tokens_seen": 1271922688 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031043129388164494, + "loss": 1.3868, + "theoretical_loss": 3.568393394600383, + "tokens_seen": 1271988224 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003104212637913741, + "loss": 1.4039, + "theoretical_loss": 3.5683766679111377, + "tokens_seen": 1272053760 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003104112337011033, + "loss": 1.475, + "theoretical_loss": 3.5683599423249035, + "tokens_seen": 1272119296 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031040120361083254, + "loss": 1.4199, + "theoretical_loss": 3.5683432178415515, + "tokens_seen": 1272184832 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031039117352056167, + "loss": 1.3067, + "theoretical_loss": 3.5683264944609516, + "tokens_seen": 1272250368 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003103811434302909, + "loss": 1.392, + "theoretical_loss": 3.5683097721829755, + "tokens_seen": 1272315904 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031037111334002003, + "loss": 1.2617, + "theoretical_loss": 3.5682930510074926, + "tokens_seen": 1272381440 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031036108324974926, + "loss": 1.3287, + "theoretical_loss": 3.5682763309343737, + "tokens_seen": 1272446976 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031035105315947844, + "loss": 1.2778, + "theoretical_loss": 3.56825961196349, + "tokens_seen": 1272512512 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003103410230692076, + "loss": 1.3989, + "theoretical_loss": 3.568242894094711, + "tokens_seen": 1272578048 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003103309929789368, + "loss": 1.2201, + "theoretical_loss": 3.568226177327908, + "tokens_seen": 1272643584 + }, + { + "epoch": 0.39, + "learning_rate": 0.000310320962888666, + "loss": 1.4041, + "theoretical_loss": 3.568209461662951, + "tokens_seen": 1272709120 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031031093279839517, + "loss": 1.3234, + "theoretical_loss": 3.5681927470997117, + "tokens_seen": 1272774656 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003103009027081244, + "loss": 1.3513, + "theoretical_loss": 3.5681760336380606, + "tokens_seen": 1272840192 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031029087261785353, + "loss": 1.4433, + "theoretical_loss": 3.568159321277867, + "tokens_seen": 1272905728 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031028084252758277, + "loss": 1.3543, + "theoretical_loss": 3.568142610019003, + "tokens_seen": 1272971264 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": 6.598741310881451e-05, + "objective/train/docs_used": 722001, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.765514373779297, + "objective/train/original_loss": 2.7655141353607178, + "objective/train/theoretical_loss": 3.5681258998613385, + "objective/train/tokens_used": 1293496800, + "objective/train/value_avg": -0.00565338134765625, + "objective/train/value_loss": 5.690971011063084e-05, + "objective/train/value_max": -0.00042557716369628906, + "objective/train/value_min": -0.05706787109375, + "objective/train/value_reward_corr": 0.382621689673505, + "objective/train/value_std": 0.00418853759765625, + "objective/train/weight_avg": 1.0034106969833374, + "objective/train/weighted_lm_loss": 2.778099775314331, + "objective/train/weights_max": 1.67131507396698, + "objective/train/weights_min": 0.5329205393791199, + "theoretical_loss": 3.5681258998613385, + "tokens_seen": 1273036800 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031027081243731195, + "loss": 1.3707, + "theoretical_loss": 3.5681258998613385, + "tokens_seen": 1273036800 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031026078234704113, + "loss": 1.3164, + "theoretical_loss": 3.568109190804745, + "tokens_seen": 1273102336 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003102507522567703, + "loss": 1.3561, + "theoretical_loss": 3.568092482849093, + "tokens_seen": 1273167872 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003102407221664995, + "loss": 1.2913, + "theoretical_loss": 3.568075775994253, + "tokens_seen": 1273233408 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031023069207622867, + "loss": 1.4779, + "theoretical_loss": 3.5680590702400963, + "tokens_seen": 1273298944 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003102206619859579, + "loss": 1.398, + "theoretical_loss": 3.568042365586493, + "tokens_seen": 1273364480 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031021063189568703, + "loss": 1.392, + "theoretical_loss": 3.568025662033315, + "tokens_seen": 1273430016 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031020060180541627, + "loss": 1.3956, + "theoretical_loss": 3.568008959580432, + "tokens_seen": 1273495552 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003101905717151454, + "loss": 1.3992, + "theoretical_loss": 3.567992258227716, + "tokens_seen": 1273561088 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031018054162487463, + "loss": 1.3674, + "theoretical_loss": 3.567975557975037, + "tokens_seen": 1273626624 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003101705115346038, + "loss": 1.462, + "theoretical_loss": 3.5679588588222666, + "tokens_seen": 1273692160 + }, + { + "epoch": 0.39, + "learning_rate": 0.000310160481444333, + "loss": 1.3751, + "theoretical_loss": 3.5679421607692756, + "tokens_seen": 1273757696 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003101504513540622, + "loss": 1.5198, + "theoretical_loss": 3.567925463815935, + "tokens_seen": 1273823232 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003101404212637914, + "loss": 1.3922, + "theoretical_loss": 3.5679087679621153, + "tokens_seen": 1273888768 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031013039117352054, + "loss": 1.4284, + "theoretical_loss": 3.5678920732076884, + "tokens_seen": 1273954304 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031012036108324977, + "loss": 1.4254, + "theoretical_loss": 3.567875379552525, + "tokens_seen": 1274019840 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003101103309929789, + "loss": 1.5217, + "theoretical_loss": 3.567858686996496, + "tokens_seen": 1274085376 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031010030090270813, + "loss": 1.4324, + "theoretical_loss": 3.5678419955394727, + "tokens_seen": 1274150912 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003100902708124373, + "loss": 1.5214, + "theoretical_loss": 3.567825305181326, + "tokens_seen": 1274216448 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003100802407221665, + "loss": 1.4928, + "theoretical_loss": 3.567808615921927, + "tokens_seen": 1274281984 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003100702106318957, + "loss": 1.4, + "theoretical_loss": 3.5677919277611476, + "tokens_seen": 1274347520 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031006018054162486, + "loss": 1.4335, + "theoretical_loss": 3.5677752406988583, + "tokens_seen": 1274413056 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003100501504513541, + "loss": 1.3292, + "theoretical_loss": 3.5677585547349304, + "tokens_seen": 1274478592 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003100401203610833, + "loss": 1.3811, + "theoretical_loss": 3.5677418698692347, + "tokens_seen": 1274544128 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031003009027081246, + "loss": 1.2022, + "theoretical_loss": 3.5677251861016437, + "tokens_seen": 1274609664 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": 0.0005898136296309531, + "objective/train/docs_used": 723183, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.474947690963745, + "objective/train/original_loss": 2.474947452545166, + "objective/train/theoretical_loss": 3.5677085034320273, + "objective/train/tokens_used": 1295135200, + "objective/train/value_avg": -0.00557708740234375, + "objective/train/value_loss": 4.7522367822239175e-05, + "objective/train/value_max": -0.0002199411392211914, + "objective/train/value_min": -0.05242919921875, + "objective/train/value_reward_corr": 0.3535106180480819, + "objective/train/value_std": 0.0040435791015625, + "objective/train/weight_avg": 1.0082029104232788, + "objective/train/weighted_lm_loss": 2.492288112640381, + "objective/train/weights_max": 1.4422595500946045, + "objective/train/weights_min": 0.5008660554885864, + "theoretical_loss": 3.5677085034320273, + "tokens_seen": 1274675200 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031002006018054164, + "loss": 1.3328, + "theoretical_loss": 3.5677085034320273, + "tokens_seen": 1274675200 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003100100300902708, + "loss": 1.3142, + "theoretical_loss": 3.5676918218602576, + "tokens_seen": 1274740736 + }, + { + "epoch": 0.39, + "learning_rate": 0.00031, + "loss": 1.3744, + "theoretical_loss": 3.567675141386206, + "tokens_seen": 1274806272 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030998996990972924, + "loss": 1.3946, + "theoretical_loss": 3.567658462009743, + "tokens_seen": 1274871808 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030997993981945836, + "loss": 1.3924, + "theoretical_loss": 3.567641783730741, + "tokens_seen": 1274937344 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003099699097291876, + "loss": 1.3665, + "theoretical_loss": 3.5676251065490705, + "tokens_seen": 1275002880 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003099598796389168, + "loss": 1.4008, + "theoretical_loss": 3.567608430464604, + "tokens_seen": 1275068416 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030994984954864596, + "loss": 1.4847, + "theoretical_loss": 3.5675917554772116, + "tokens_seen": 1275133952 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030993981945837514, + "loss": 1.4134, + "theoretical_loss": 3.5675750815867655, + "tokens_seen": 1275199488 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003099297893681043, + "loss": 1.3568, + "theoretical_loss": 3.567558408793137, + "tokens_seen": 1275265024 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003099197592778335, + "loss": 1.437, + "theoretical_loss": 3.5675417370961977, + "tokens_seen": 1275330560 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030990972918756274, + "loss": 1.4549, + "theoretical_loss": 3.5675250664958194, + "tokens_seen": 1275396096 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030989969909729187, + "loss": 1.3593, + "theoretical_loss": 3.567508396991873, + "tokens_seen": 1275461632 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003098896690070211, + "loss": 1.491, + "theoretical_loss": 3.5674917285842302, + "tokens_seen": 1275527168 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030987963891675023, + "loss": 1.3943, + "theoretical_loss": 3.567475061272763, + "tokens_seen": 1275592704 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030986960882647946, + "loss": 1.3937, + "theoretical_loss": 3.5674583950573426, + "tokens_seen": 1275658240 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030985957873620864, + "loss": 1.4737, + "theoretical_loss": 3.5674417299378405, + "tokens_seen": 1275723776 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003098495486459378, + "loss": 1.2826, + "theoretical_loss": 3.567425065914129, + "tokens_seen": 1275789312 + }, + { + "epoch": 0.39, + "learning_rate": 0.000309839518555667, + "loss": 1.3818, + "theoretical_loss": 3.567408402986079, + "tokens_seen": 1275854848 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003098294884653962, + "loss": 1.3272, + "theoretical_loss": 3.5673917411535627, + "tokens_seen": 1275920384 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030981945837512537, + "loss": 1.3707, + "theoretical_loss": 3.5673750804164515, + "tokens_seen": 1275985920 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003098094282848546, + "loss": 1.3844, + "theoretical_loss": 3.5673584207746174, + "tokens_seen": 1276051456 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030979939819458373, + "loss": 1.4696, + "theoretical_loss": 3.567341762227932, + "tokens_seen": 1276116992 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030978936810431297, + "loss": 1.4511, + "theoretical_loss": 3.567325104776267, + "tokens_seen": 1276182528 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030977933801404215, + "loss": 1.4277, + "theoretical_loss": 3.5673084484194946, + "tokens_seen": 1276248064 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -6.545869109686464e-05, + "objective/train/docs_used": 723876, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8593974113464355, + "objective/train/original_loss": 2.8593974113464355, + "objective/train/theoretical_loss": 3.567291793157486, + "objective/train/tokens_used": 1296773600, + "objective/train/value_avg": -0.005462646484375, + "objective/train/value_loss": 5.087005774839781e-05, + "objective/train/value_max": -0.00030303001403808594, + "objective/train/value_min": -0.1407470703125, + "objective/train/value_reward_corr": 0.3245437054274668, + "objective/train/value_std": 0.004180908203125, + "objective/train/weight_avg": 1.0018208026885986, + "objective/train/weighted_lm_loss": 2.878936767578125, + "objective/train/weights_max": 3.7161340713500977, + "objective/train/weights_min": 0.3358522355556488, + "theoretical_loss": 3.567291793157486, + "tokens_seen": 1276313600 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030976930792377133, + "loss": 1.4821, + "theoretical_loss": 3.567291793157486, + "tokens_seen": 1276313600 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003097592778335005, + "loss": 1.4486, + "theoretical_loss": 3.567275138990113, + "tokens_seen": 1276379136 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003097492477432297, + "loss": 1.4991, + "theoretical_loss": 3.5672584859172485, + "tokens_seen": 1276444672 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030973921765295887, + "loss": 1.3326, + "theoretical_loss": 3.5672418339387635, + "tokens_seen": 1276510208 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003097291875626881, + "loss": 1.4378, + "theoretical_loss": 3.56722518305453, + "tokens_seen": 1276575744 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030971915747241723, + "loss": 1.3405, + "theoretical_loss": 3.5672085332644197, + "tokens_seen": 1276641280 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030970912738214647, + "loss": 1.3393, + "theoretical_loss": 3.567191884568305, + "tokens_seen": 1276706816 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003096990972918756, + "loss": 1.3861, + "theoretical_loss": 3.567175236966058, + "tokens_seen": 1276772352 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030968906720160483, + "loss": 1.2324, + "theoretical_loss": 3.5671585904575505, + "tokens_seen": 1276837888 + }, + { + "epoch": 0.39, + "learning_rate": 0.000309679037111334, + "loss": 1.4611, + "theoretical_loss": 3.5671419450426534, + "tokens_seen": 1276903424 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003096690070210632, + "loss": 1.382, + "theoretical_loss": 3.5671253007212407, + "tokens_seen": 1276968960 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003096589769307924, + "loss": 1.3856, + "theoretical_loss": 3.567108657493183, + "tokens_seen": 1277034496 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003096489468405216, + "loss": 1.3518, + "theoretical_loss": 3.5670920153583534, + "tokens_seen": 1277100032 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030963891675025074, + "loss": 1.4955, + "theoretical_loss": 3.567075374316623, + "tokens_seen": 1277165568 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030962888665997997, + "loss": 1.4296, + "theoretical_loss": 3.5670587343678646, + "tokens_seen": 1277231104 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003096188565697091, + "loss": 1.4346, + "theoretical_loss": 3.56704209551195, + "tokens_seen": 1277296640 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030960882647943833, + "loss": 1.4054, + "theoretical_loss": 3.567025457748751, + "tokens_seen": 1277362176 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003095987963891675, + "loss": 1.3198, + "theoretical_loss": 3.567008821078141, + "tokens_seen": 1277427712 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003095887662988967, + "loss": 1.3242, + "theoretical_loss": 3.5669921854999913, + "tokens_seen": 1277493248 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003095787362086259, + "loss": 1.3022, + "theoretical_loss": 3.566975551014174, + "tokens_seen": 1277558784 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030956870611835506, + "loss": 1.3409, + "theoretical_loss": 3.5669589176205623, + "tokens_seen": 1277624320 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030955867602808424, + "loss": 1.4321, + "theoretical_loss": 3.566942285319027, + "tokens_seen": 1277689856 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003095486459378135, + "loss": 1.3618, + "theoretical_loss": 3.566925654109441, + "tokens_seen": 1277755392 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003095386158475426, + "loss": 1.3961, + "theoretical_loss": 3.566909023991678, + "tokens_seen": 1277820928 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030952858575727184, + "loss": 1.3497, + "theoretical_loss": 3.566892394965608, + "tokens_seen": 1277886464 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -8.34118909551762e-05, + "objective/train/docs_used": 725220, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0593972206115723, + "objective/train/original_loss": 3.059396982192993, + "objective/train/theoretical_loss": 3.566875767031105, + "objective/train/tokens_used": 1298412000, + "objective/train/value_avg": -0.005100250244140625, + "objective/train/value_loss": 4.435091614141129e-05, + "objective/train/value_max": -0.0003077983856201172, + "objective/train/value_min": -0.076416015625, + "objective/train/value_reward_corr": 0.4083793407797433, + "objective/train/value_std": 0.00411224365234375, + "objective/train/weight_avg": 1.001224398612976, + "objective/train/weighted_lm_loss": 3.06562876701355, + "objective/train/weights_max": 2.0478014945983887, + "objective/train/weights_min": 0.25813794136047363, + "theoretical_loss": 3.566875767031105, + "tokens_seen": 1277952000 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030951855566700096, + "loss": 1.4016, + "theoretical_loss": 3.566875767031105, + "tokens_seen": 1277952000 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003095085255767302, + "loss": 1.3581, + "theoretical_loss": 3.5668591401880407, + "tokens_seen": 1278017536 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003094984954864594, + "loss": 1.359, + "theoretical_loss": 3.5668425144362876, + "tokens_seen": 1278083072 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030948846539618856, + "loss": 1.3863, + "theoretical_loss": 3.5668258897757186, + "tokens_seen": 1278148608 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030947843530591774, + "loss": 1.3969, + "theoretical_loss": 3.5668092662062048, + "tokens_seen": 1278214144 + }, + { + "epoch": 0.39, + "learning_rate": 0.000309468405215647, + "loss": 1.4177, + "theoretical_loss": 3.56679264372762, + "tokens_seen": 1278279680 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003094583751253761, + "loss": 1.4025, + "theoretical_loss": 3.5667760223398366, + "tokens_seen": 1278345216 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030944834503510534, + "loss": 1.3338, + "theoretical_loss": 3.5667594020427265, + "tokens_seen": 1278410752 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030943831494483447, + "loss": 1.3866, + "theoretical_loss": 3.5667427828361626, + "tokens_seen": 1278476288 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003094282848545637, + "loss": 1.3357, + "theoretical_loss": 3.5667261647200172, + "tokens_seen": 1278541824 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003094182547642929, + "loss": 1.3779, + "theoretical_loss": 3.5667095476941633, + "tokens_seen": 1278607360 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030940822467402207, + "loss": 1.376, + "theoretical_loss": 3.566692931758473, + "tokens_seen": 1278672896 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030939819458375125, + "loss": 1.3824, + "theoretical_loss": 3.566676316912819, + "tokens_seen": 1278738432 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030938816449348043, + "loss": 1.2523, + "theoretical_loss": 3.5666597031570735, + "tokens_seen": 1278803968 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003093781344032096, + "loss": 1.448, + "theoretical_loss": 3.56664309049111, + "tokens_seen": 1278869504 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030936810431293884, + "loss": 1.4206, + "theoretical_loss": 3.5666264789148014, + "tokens_seen": 1278935040 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030935807422266797, + "loss": 1.3579, + "theoretical_loss": 3.5666098684280194, + "tokens_seen": 1279000576 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003093480441323972, + "loss": 1.4167, + "theoretical_loss": 3.566593259030637, + "tokens_seen": 1279066112 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030933801404212633, + "loss": 1.4151, + "theoretical_loss": 3.566576650722528, + "tokens_seen": 1279131648 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030932798395185557, + "loss": 1.3356, + "theoretical_loss": 3.5665600435035634, + "tokens_seen": 1279197184 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030931795386158475, + "loss": 1.5805, + "theoretical_loss": 3.566543437373617, + "tokens_seen": 1279262720 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030930792377131393, + "loss": 1.4176, + "theoretical_loss": 3.5665268323325616, + "tokens_seen": 1279328256 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030929789368104317, + "loss": 1.3276, + "theoretical_loss": 3.5665102283802694, + "tokens_seen": 1279393792 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030928786359077235, + "loss": 1.3556, + "theoretical_loss": 3.5664936255166144, + "tokens_seen": 1279459328 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030927783350050153, + "loss": 1.4259, + "theoretical_loss": 3.5664770237414687, + "tokens_seen": 1279524864 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -0.00027564921765588224, + "objective/train/docs_used": 725568, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7841877937316895, + "objective/train/original_loss": 2.7841880321502686, + "objective/train/theoretical_loss": 3.5664604230547052, + "objective/train/tokens_used": 1300050400, + "objective/train/value_avg": -0.006328582763671875, + "objective/train/value_loss": 7.400899630738422e-05, + "objective/train/value_max": -0.00038909912109375, + "objective/train/value_min": -0.07781982421875, + "objective/train/value_reward_corr": 0.44826560156755113, + "objective/train/value_std": 0.005279541015625, + "objective/train/weight_avg": 1.0007708072662354, + "objective/train/weighted_lm_loss": 2.7888429164886475, + "objective/train/weights_max": 2.1775453090667725, + "objective/train/weights_min": 0.38544905185699463, + "theoretical_loss": 3.5664604230547052, + "tokens_seen": 1279590400 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003092678034102307, + "loss": 1.3764, + "theoretical_loss": 3.5664604230547052, + "tokens_seen": 1279590400 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003092577733199599, + "loss": 1.4884, + "theoretical_loss": 3.5664438234561966, + "tokens_seen": 1279655936 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030924774322968907, + "loss": 1.462, + "theoretical_loss": 3.566427224945816, + "tokens_seen": 1279721472 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003092377131394183, + "loss": 1.3738, + "theoretical_loss": 3.566410627523437, + "tokens_seen": 1279787008 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030922768304914743, + "loss": 1.4294, + "theoretical_loss": 3.566394031188932, + "tokens_seen": 1279852544 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030921765295887667, + "loss": 1.4827, + "theoretical_loss": 3.566377435942174, + "tokens_seen": 1279918080 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003092076228686058, + "loss": 1.4296, + "theoretical_loss": 3.5663608417830357, + "tokens_seen": 1279983616 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030919759277833503, + "loss": 1.3864, + "theoretical_loss": 3.566344248711391, + "tokens_seen": 1280049152 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003091875626880642, + "loss": 1.3433, + "theoretical_loss": 3.566327656727112, + "tokens_seen": 1280114688 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003091775325977934, + "loss": 1.4025, + "theoretical_loss": 3.5663110658300727, + "tokens_seen": 1280180224 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003091675025075226, + "loss": 1.3902, + "theoretical_loss": 3.566294476020145, + "tokens_seen": 1280245760 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003091574724172518, + "loss": 1.285, + "theoretical_loss": 3.5662778872972036, + "tokens_seen": 1280311296 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030914744232698094, + "loss": 1.3936, + "theoretical_loss": 3.5662612996611207, + "tokens_seen": 1280376832 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030913741223671017, + "loss": 1.4673, + "theoretical_loss": 3.566244713111769, + "tokens_seen": 1280442368 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003091273821464393, + "loss": 1.4054, + "theoretical_loss": 3.5662281276490226, + "tokens_seen": 1280507904 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030911735205616853, + "loss": 1.3957, + "theoretical_loss": 3.5662115432727544, + "tokens_seen": 1280573440 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003091073219658977, + "loss": 1.4663, + "theoretical_loss": 3.5661949599828375, + "tokens_seen": 1280638976 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003090972918756269, + "loss": 1.3842, + "theoretical_loss": 3.5661783777791456, + "tokens_seen": 1280704512 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003090872617853561, + "loss": 1.2967, + "theoretical_loss": 3.566161796661551, + "tokens_seen": 1280770048 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030907723169508526, + "loss": 1.3651, + "theoretical_loss": 3.566145216629928, + "tokens_seen": 1280835584 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030906720160481444, + "loss": 1.4151, + "theoretical_loss": 3.56612863768415, + "tokens_seen": 1280901120 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003090571715145437, + "loss": 1.3104, + "theoretical_loss": 3.566112059824089, + "tokens_seen": 1280966656 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003090471414242728, + "loss": 1.3197, + "theoretical_loss": 3.5660954830496197, + "tokens_seen": 1281032192 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030903711133400204, + "loss": 1.3687, + "theoretical_loss": 3.5660789073606147, + "tokens_seen": 1281097728 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030902708124373116, + "loss": 1.3237, + "theoretical_loss": 3.566062332756948, + "tokens_seen": 1281163264 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -0.00043188343988731503, + "objective/train/docs_used": 726917, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6179966926574707, + "objective/train/original_loss": 2.6179966926574707, + "objective/train/theoretical_loss": 3.5660457592384924, + "objective/train/tokens_used": 1301688800, + "objective/train/value_avg": -0.004791259765625, + "objective/train/value_loss": 6.821785063948482e-05, + "objective/train/value_max": -0.00023233890533447266, + "objective/train/value_min": -0.057281494140625, + "objective/train/value_reward_corr": 0.4163634382698739, + "objective/train/value_std": 0.0041656494140625, + "objective/train/weight_avg": 0.9987714290618896, + "objective/train/weighted_lm_loss": 2.6251626014709473, + "objective/train/weights_max": 1.7732515335083008, + "objective/train/weights_min": 0.2687038481235504, + "theoretical_loss": 3.5660457592384924, + "tokens_seen": 1281228800 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003090170511534604, + "loss": 1.3386, + "theoretical_loss": 3.5660457592384924, + "tokens_seen": 1281228800 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003090070210631896, + "loss": 1.2561, + "theoretical_loss": 3.566029186805122, + "tokens_seen": 1281294336 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030899699097291876, + "loss": 1.299, + "theoretical_loss": 3.56601261545671, + "tokens_seen": 1281359872 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030898696088264794, + "loss": 1.4416, + "theoretical_loss": 3.5659960451931294, + "tokens_seen": 1281425408 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003089769307923772, + "loss": 1.3918, + "theoretical_loss": 3.5659794760142542, + "tokens_seen": 1281490944 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003089669007021063, + "loss": 1.4373, + "theoretical_loss": 3.5659629079199586, + "tokens_seen": 1281556480 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030895687061183554, + "loss": 1.5473, + "theoretical_loss": 3.5659463409101146, + "tokens_seen": 1281622016 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030894684052156467, + "loss": 1.3835, + "theoretical_loss": 3.5659297749845966, + "tokens_seen": 1281687552 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003089368104312939, + "loss": 1.3567, + "theoretical_loss": 3.5659132101432784, + "tokens_seen": 1281753088 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003089267803410231, + "loss": 1.4092, + "theoretical_loss": 3.5658966463860335, + "tokens_seen": 1281818624 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030891675025075227, + "loss": 1.401, + "theoretical_loss": 3.5658800837127353, + "tokens_seen": 1281884160 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030890672016048145, + "loss": 1.3432, + "theoretical_loss": 3.5658635221232573, + "tokens_seen": 1281949696 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030889669007021063, + "loss": 1.3552, + "theoretical_loss": 3.5658469616174737, + "tokens_seen": 1282015232 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003088866599799398, + "loss": 1.4378, + "theoretical_loss": 3.5658304021952576, + "tokens_seen": 1282080768 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030887662988966904, + "loss": 1.4548, + "theoretical_loss": 3.5658138438564837, + "tokens_seen": 1282146304 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030886659979939817, + "loss": 1.2594, + "theoretical_loss": 3.5657972866010246, + "tokens_seen": 1282211840 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003088565697091274, + "loss": 1.4218, + "theoretical_loss": 3.5657807304287545, + "tokens_seen": 1282277376 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030884653961885653, + "loss": 1.2501, + "theoretical_loss": 3.565764175339547, + "tokens_seen": 1282342912 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030883650952858577, + "loss": 1.4263, + "theoretical_loss": 3.565747621333277, + "tokens_seen": 1282408448 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030882647943831495, + "loss": 1.3979, + "theoretical_loss": 3.565731068409817, + "tokens_seen": 1282473984 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030881644934804413, + "loss": 1.3476, + "theoretical_loss": 3.5657145165690407, + "tokens_seen": 1282539520 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003088064192577733, + "loss": 1.4299, + "theoretical_loss": 3.5656979658108234, + "tokens_seen": 1282605056 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030879638916750255, + "loss": 1.4093, + "theoretical_loss": 3.565681416135038, + "tokens_seen": 1282670592 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003087863590772317, + "loss": 1.335, + "theoretical_loss": 3.5656648675415585, + "tokens_seen": 1282736128 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003087763289869609, + "loss": 1.4086, + "theoretical_loss": 3.5656483200302587, + "tokens_seen": 1282801664 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": 0.0006778365932404995, + "objective/train/docs_used": 727506, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1078860759735107, + "objective/train/original_loss": 3.1078860759735107, + "objective/train/theoretical_loss": 3.5656317736010132, + "objective/train/tokens_used": 1303327200, + "objective/train/value_avg": -0.006168365478515625, + "objective/train/value_loss": 5.5387907195836306e-05, + "objective/train/value_max": -0.0003275871276855469, + "objective/train/value_min": -0.1070556640625, + "objective/train/value_reward_corr": 0.4079346611512755, + "objective/train/value_std": 0.005157470703125, + "objective/train/weight_avg": 1.0095109939575195, + "objective/train/weighted_lm_loss": 3.1329455375671387, + "objective/train/weights_max": 2.4156248569488525, + "objective/train/weights_min": 0.5112028121948242, + "theoretical_loss": 3.5656317736010132, + "tokens_seen": 1282867200 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030876629889669004, + "loss": 1.3585, + "theoretical_loss": 3.5656317736010132, + "tokens_seen": 1282867200 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030875626880641927, + "loss": 1.3507, + "theoretical_loss": 3.5656152282536953, + "tokens_seen": 1282932736 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030874623871614845, + "loss": 1.4765, + "theoretical_loss": 3.5655986839881786, + "tokens_seen": 1282998272 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030873620862587763, + "loss": 1.4095, + "theoretical_loss": 3.5655821408043384, + "tokens_seen": 1283063808 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003087261785356068, + "loss": 1.4245, + "theoretical_loss": 3.565565598702048, + "tokens_seen": 1283129344 + }, + { + "epoch": 0.39, + "learning_rate": 0.000308716148445336, + "loss": 1.4728, + "theoretical_loss": 3.5655490576811815, + "tokens_seen": 1283194880 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003087061183550652, + "loss": 1.3185, + "theoretical_loss": 3.5655325177416124, + "tokens_seen": 1283260416 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003086960882647944, + "loss": 1.4946, + "theoretical_loss": 3.5655159788832163, + "tokens_seen": 1283325952 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030868605817452354, + "loss": 1.2959, + "theoretical_loss": 3.565499441105866, + "tokens_seen": 1283391488 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003086760280842528, + "loss": 1.3627, + "theoretical_loss": 3.565482904409436, + "tokens_seen": 1283457024 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003086659979939819, + "loss": 1.3647, + "theoretical_loss": 3.5654663687938006, + "tokens_seen": 1283522560 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030865596790371114, + "loss": 1.4588, + "theoretical_loss": 3.5654498342588337, + "tokens_seen": 1283588096 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003086459378134403, + "loss": 1.3926, + "theoretical_loss": 3.56543330080441, + "tokens_seen": 1283653632 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003086359077231695, + "loss": 1.4304, + "theoretical_loss": 3.5654167684304032, + "tokens_seen": 1283719168 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003086258776328987, + "loss": 1.3579, + "theoretical_loss": 3.5654002371366875, + "tokens_seen": 1283784704 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003086158475426279, + "loss": 1.3919, + "theoretical_loss": 3.5653837069231384, + "tokens_seen": 1283850240 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030860581745235704, + "loss": 1.4358, + "theoretical_loss": 3.5653671777896285, + "tokens_seen": 1283915776 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003085957873620863, + "loss": 1.3362, + "theoretical_loss": 3.5653506497360326, + "tokens_seen": 1283981312 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003085857572718154, + "loss": 1.3581, + "theoretical_loss": 3.565334122762226, + "tokens_seen": 1284046848 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030857572718154464, + "loss": 1.4183, + "theoretical_loss": 3.5653175968680815, + "tokens_seen": 1284112384 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003085656970912738, + "loss": 1.3497, + "theoretical_loss": 3.565301072053475, + "tokens_seen": 1284177920 + }, + { + "epoch": 0.39, + "learning_rate": 0.000308555667001003, + "loss": 1.4181, + "theoretical_loss": 3.56528454831828, + "tokens_seen": 1284243456 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030854563691073224, + "loss": 1.2937, + "theoretical_loss": 3.565268025662371, + "tokens_seen": 1284308992 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030853560682046137, + "loss": 1.3979, + "theoretical_loss": 3.5652515040856225, + "tokens_seen": 1284374528 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003085255767301906, + "loss": 1.3481, + "theoretical_loss": 3.565234983587909, + "tokens_seen": 1284440064 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -0.00015464212629012764, + "objective/train/docs_used": 728764, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3829963207244873, + "objective/train/original_loss": 2.3829963207244873, + "objective/train/theoretical_loss": 3.5652184641691047, + "objective/train/tokens_used": 1304965600, + "objective/train/value_avg": -0.003997802734375, + "objective/train/value_loss": 3.670323712867685e-05, + "objective/train/value_max": -0.0003712177276611328, + "objective/train/value_min": -0.06866455078125, + "objective/train/value_reward_corr": 0.3273284771776016, + "objective/train/value_std": 0.00311279296875, + "objective/train/weight_avg": 1.0002014636993408, + "objective/train/weighted_lm_loss": 2.3850038051605225, + "objective/train/weights_max": 1.876982569694519, + "objective/train/weights_min": 0.4179898202419281, + "theoretical_loss": 3.5652184641691047, + "tokens_seen": 1284505600 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003085155466399198, + "loss": 1.327, + "theoretical_loss": 3.5652184641691047, + "tokens_seen": 1284505600 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030850551654964896, + "loss": 1.4305, + "theoretical_loss": 3.5652019458290845, + "tokens_seen": 1284571136 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030849548645937814, + "loss": 1.3242, + "theoretical_loss": 3.565185428567723, + "tokens_seen": 1284636672 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003084854563691074, + "loss": 1.2932, + "theoretical_loss": 3.5651689123848946, + "tokens_seen": 1284702208 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003084754262788365, + "loss": 1.4312, + "theoretical_loss": 3.5651523972804737, + "tokens_seen": 1284767744 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030846539618856574, + "loss": 1.3743, + "theoretical_loss": 3.5651358832543347, + "tokens_seen": 1284833280 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030845536609829487, + "loss": 1.239, + "theoretical_loss": 3.565119370306353, + "tokens_seen": 1284898816 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003084453360080241, + "loss": 1.3995, + "theoretical_loss": 3.5651028584364024, + "tokens_seen": 1284964352 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003084353059177533, + "loss": 1.4186, + "theoretical_loss": 3.5650863476443577, + "tokens_seen": 1285029888 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030842527582748247, + "loss": 1.4133, + "theoretical_loss": 3.5650698379300936, + "tokens_seen": 1285095424 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030841524573721165, + "loss": 1.3611, + "theoretical_loss": 3.565053329293485, + "tokens_seen": 1285160960 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030840521564694083, + "loss": 1.3096, + "theoretical_loss": 3.565036821734407, + "tokens_seen": 1285226496 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030839518555667, + "loss": 1.3504, + "theoretical_loss": 3.565020315252733, + "tokens_seen": 1285292032 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030838515546639924, + "loss": 1.4213, + "theoretical_loss": 3.565003809848339, + "tokens_seen": 1285357568 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030837512537612837, + "loss": 1.3781, + "theoretical_loss": 3.5649873055210994, + "tokens_seen": 1285423104 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003083650952858576, + "loss": 1.5002, + "theoretical_loss": 3.564970802270889, + "tokens_seen": 1285488640 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030835506519558673, + "loss": 1.4602, + "theoretical_loss": 3.5649543000975825, + "tokens_seen": 1285554176 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030834503510531597, + "loss": 1.4163, + "theoretical_loss": 3.5649377990010542, + "tokens_seen": 1285619712 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030833500501504515, + "loss": 1.3766, + "theoretical_loss": 3.5649212989811803, + "tokens_seen": 1285685248 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030832497492477433, + "loss": 1.3232, + "theoretical_loss": 3.5649048000378345, + "tokens_seen": 1285750784 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003083149448345035, + "loss": 1.459, + "theoretical_loss": 3.5648883021708926, + "tokens_seen": 1285816320 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030830491474423275, + "loss": 1.3718, + "theoretical_loss": 3.5648718053802284, + "tokens_seen": 1285881856 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003082948846539619, + "loss": 1.4734, + "theoretical_loss": 3.564855309665718, + "tokens_seen": 1285947392 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003082848545636911, + "loss": 1.4164, + "theoretical_loss": 3.564838815027236, + "tokens_seen": 1286012928 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030827482447342024, + "loss": 1.468, + "theoretical_loss": 3.5648223214646566, + "tokens_seen": 1286078464 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -0.0012599341571331024, + "objective/train/docs_used": 729512, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.996227502822876, + "objective/train/original_loss": 2.996227741241455, + "objective/train/theoretical_loss": 3.564805828977856, + "objective/train/tokens_used": 1306604000, + "objective/train/value_avg": -0.006259918212890625, + "objective/train/value_loss": 6.967399531276897e-05, + "objective/train/value_max": -0.00036263465881347656, + "objective/train/value_min": -0.057281494140625, + "objective/train/value_reward_corr": 0.3760654544734951, + "objective/train/value_std": 0.00531768798828125, + "objective/train/weight_avg": 0.9907569289207458, + "objective/train/weighted_lm_loss": 2.9937875270843506, + "objective/train/weights_max": 1.668045997619629, + "objective/train/weights_min": 0.504653811454773, + "theoretical_loss": 3.564805828977856, + "tokens_seen": 1286144000 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030826479438314947, + "loss": 1.451, + "theoretical_loss": 3.564805828977856, + "tokens_seen": 1286144000 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030825476429287865, + "loss": 1.3311, + "theoretical_loss": 3.5647893375667077, + "tokens_seen": 1286209536 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030824473420260783, + "loss": 1.3892, + "theoretical_loss": 3.5647728472310884, + "tokens_seen": 1286275072 + }, + { + "epoch": 0.39, + "learning_rate": 0.000308234704112337, + "loss": 1.3937, + "theoretical_loss": 3.5647563579708725, + "tokens_seen": 1286340608 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003082246740220662, + "loss": 1.3432, + "theoretical_loss": 3.5647398697859343, + "tokens_seen": 1286406144 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003082146439317954, + "loss": 1.3193, + "theoretical_loss": 3.56472338267615, + "tokens_seen": 1286471680 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003082046138415246, + "loss": 1.2805, + "theoretical_loss": 3.564706896641395, + "tokens_seen": 1286537216 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030819458375125374, + "loss": 1.4211, + "theoretical_loss": 3.564690411681543, + "tokens_seen": 1286602752 + }, + { + "epoch": 0.39, + "learning_rate": 0.000308184553660983, + "loss": 1.4523, + "theoretical_loss": 3.5646739277964707, + "tokens_seen": 1286668288 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003081745235707121, + "loss": 1.41, + "theoretical_loss": 3.564657444986052, + "tokens_seen": 1286733824 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030816449348044134, + "loss": 1.4479, + "theoretical_loss": 3.564640963250163, + "tokens_seen": 1286799360 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003081544633901705, + "loss": 1.4248, + "theoretical_loss": 3.564624482588678, + "tokens_seen": 1286864896 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003081444332998997, + "loss": 1.3144, + "theoretical_loss": 3.5646080030014735, + "tokens_seen": 1286930432 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003081344032096289, + "loss": 1.3333, + "theoretical_loss": 3.564591524488424, + "tokens_seen": 1286995968 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003081243731193581, + "loss": 1.4973, + "theoretical_loss": 3.564575047049405, + "tokens_seen": 1287061504 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030811434302908724, + "loss": 1.4446, + "theoretical_loss": 3.564558570684291, + "tokens_seen": 1287127040 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003081043129388165, + "loss": 1.4389, + "theoretical_loss": 3.5645420953929587, + "tokens_seen": 1287192576 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003080942828485456, + "loss": 1.373, + "theoretical_loss": 3.564525621175283, + "tokens_seen": 1287258112 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030808425275827484, + "loss": 1.4351, + "theoretical_loss": 3.5645091480311386, + "tokens_seen": 1287323648 + }, + { + "epoch": 0.39, + "learning_rate": 0.000308074222668004, + "loss": 1.4324, + "theoretical_loss": 3.5644926759604014, + "tokens_seen": 1287389184 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003080641925777332, + "loss": 1.2992, + "theoretical_loss": 3.564476204962947, + "tokens_seen": 1287454720 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003080541624874624, + "loss": 1.3186, + "theoretical_loss": 3.5644597350386507, + "tokens_seen": 1287520256 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030804413239719157, + "loss": 1.4092, + "theoretical_loss": 3.564443266187388, + "tokens_seen": 1287585792 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030803410230692075, + "loss": 1.1953, + "theoretical_loss": 3.564426798409034, + "tokens_seen": 1287651328 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030802407221665, + "loss": 1.3712, + "theoretical_loss": 3.564410331703465, + "tokens_seen": 1287716864 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -0.0007150119636207819, + "objective/train/docs_used": 730476, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.2846310138702393, + "objective/train/original_loss": 2.28463077545166, + "objective/train/theoretical_loss": 3.5643938660705556, + "objective/train/tokens_used": 1308242400, + "objective/train/value_avg": -0.005191802978515625, + "objective/train/value_loss": 8.187557978089899e-05, + "objective/train/value_max": -0.0003905296325683594, + "objective/train/value_min": -0.05792236328125, + "objective/train/value_reward_corr": 0.6132559115553577, + "objective/train/value_std": 0.004825592041015625, + "objective/train/weight_avg": 0.9965876340866089, + "objective/train/weighted_lm_loss": 2.2693116664886475, + "objective/train/weights_max": 1.418717861175537, + "objective/train/weights_min": 0.31992170214653015, + "theoretical_loss": 3.5643938660705556, + "tokens_seen": 1287782400 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003080140421263791, + "loss": 1.2625, + "theoretical_loss": 3.5643938660705556, + "tokens_seen": 1287782400 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030800401203610834, + "loss": 1.2821, + "theoretical_loss": 3.564377401510182, + "tokens_seen": 1287847936 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003079939819458375, + "loss": 1.3522, + "theoretical_loss": 3.564360938022219, + "tokens_seen": 1287913472 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003079839518555667, + "loss": 1.3741, + "theoretical_loss": 3.564344475606543, + "tokens_seen": 1287979008 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003079739217652959, + "loss": 1.4079, + "theoretical_loss": 3.5643280142630296, + "tokens_seen": 1288044544 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030796389167502507, + "loss": 1.343, + "theoretical_loss": 3.564311553991554, + "tokens_seen": 1288110080 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030795386158475425, + "loss": 1.4171, + "theoretical_loss": 3.5642950947919916, + "tokens_seen": 1288175616 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003079438314944835, + "loss": 1.3726, + "theoretical_loss": 3.564278636664219, + "tokens_seen": 1288241152 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003079338014042126, + "loss": 1.3865, + "theoretical_loss": 3.564262179608111, + "tokens_seen": 1288306688 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030792377131394185, + "loss": 1.3946, + "theoretical_loss": 3.564245723623544, + "tokens_seen": 1288372224 + }, + { + "epoch": 0.39, + "learning_rate": 0.000307913741223671, + "loss": 1.4126, + "theoretical_loss": 3.5642292687103927, + "tokens_seen": 1288437760 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003079037111334002, + "loss": 1.3318, + "theoretical_loss": 3.564212814868534, + "tokens_seen": 1288503296 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003078936810431294, + "loss": 1.2497, + "theoretical_loss": 3.564196362097843, + "tokens_seen": 1288568832 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030788365095285857, + "loss": 1.3236, + "theoretical_loss": 3.564179910398196, + "tokens_seen": 1288634368 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030787362086258775, + "loss": 1.4335, + "theoretical_loss": 3.5641634597694685, + "tokens_seen": 1288699904 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030786359077231693, + "loss": 1.3558, + "theoretical_loss": 3.5641470102115362, + "tokens_seen": 1288765440 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003078535606820461, + "loss": 1.4385, + "theoretical_loss": 3.564130561724275, + "tokens_seen": 1288830976 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030784353059177535, + "loss": 1.3134, + "theoretical_loss": 3.5641141143075608, + "tokens_seen": 1288896512 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003078335005015045, + "loss": 1.4029, + "theoretical_loss": 3.5640976679612697, + "tokens_seen": 1288962048 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003078234704112337, + "loss": 1.2546, + "theoretical_loss": 3.5640812226852776, + "tokens_seen": 1289027584 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003078134403209629, + "loss": 1.3459, + "theoretical_loss": 3.56406477847946, + "tokens_seen": 1289093120 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003078034102306921, + "loss": 1.4721, + "theoretical_loss": 3.5640483353436934, + "tokens_seen": 1289158656 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003077933801404213, + "loss": 1.3786, + "theoretical_loss": 3.5640318932778534, + "tokens_seen": 1289224192 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030778335005015044, + "loss": 1.3786, + "theoretical_loss": 3.5640154522818164, + "tokens_seen": 1289289728 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030777331995987967, + "loss": 1.4263, + "theoretical_loss": 3.563999012355458, + "tokens_seen": 1289355264 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -0.0008584392489865422, + "objective/train/docs_used": 731120, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5723156929016113, + "objective/train/original_loss": 2.5723156929016113, + "objective/train/theoretical_loss": 3.563982573498654, + "objective/train/tokens_used": 1309880800, + "objective/train/value_avg": -0.00461578369140625, + "objective/train/value_loss": 5.7186265621567145e-05, + "objective/train/value_max": -0.00034737586975097656, + "objective/train/value_min": -0.061309814453125, + "objective/train/value_reward_corr": 0.3170115458246725, + "objective/train/value_std": 0.0037288665771484375, + "objective/train/weight_avg": 0.9940857291221619, + "objective/train/weighted_lm_loss": 2.5674500465393066, + "objective/train/weights_max": 1.6616897583007812, + "objective/train/weights_min": 0.3901979923248291, + "theoretical_loss": 3.563982573498654, + "tokens_seen": 1289420800 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030776328986960885, + "loss": 1.3334, + "theoretical_loss": 3.563982573498654, + "tokens_seen": 1289420800 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030775325977933803, + "loss": 1.4181, + "theoretical_loss": 3.5639661357112815, + "tokens_seen": 1289486336 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003077432296890672, + "loss": 1.3864, + "theoretical_loss": 3.563949698993216, + "tokens_seen": 1289551872 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003077331995987964, + "loss": 1.4059, + "theoretical_loss": 3.563933263344333, + "tokens_seen": 1289617408 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003077231695085256, + "loss": 1.3458, + "theoretical_loss": 3.5639168287645093, + "tokens_seen": 1289682944 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003077131394182548, + "loss": 1.3498, + "theoretical_loss": 3.5639003952536212, + "tokens_seen": 1289748480 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030770310932798394, + "loss": 1.4235, + "theoretical_loss": 3.5638839628115444, + "tokens_seen": 1289814016 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003076930792377132, + "loss": 1.4058, + "theoretical_loss": 3.5638675314381554, + "tokens_seen": 1289879552 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003076830491474423, + "loss": 1.2892, + "theoretical_loss": 3.56385110113333, + "tokens_seen": 1289945088 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030767301905717154, + "loss": 1.3199, + "theoretical_loss": 3.563834671896945, + "tokens_seen": 1290010624 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003076629889669007, + "loss": 1.378, + "theoretical_loss": 3.563818243728876, + "tokens_seen": 1290076160 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003076529588766299, + "loss": 1.2889, + "theoretical_loss": 3.5638018166289998, + "tokens_seen": 1290141696 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003076429287863591, + "loss": 1.3597, + "theoretical_loss": 3.5637853905971926, + "tokens_seen": 1290207232 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003076328986960883, + "loss": 1.3904, + "theoretical_loss": 3.5637689656333302, + "tokens_seen": 1290272768 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030762286860581744, + "loss": 1.3162, + "theoretical_loss": 3.5637525417372897, + "tokens_seen": 1290338304 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003076128385155467, + "loss": 1.3638, + "theoretical_loss": 3.563736118908947, + "tokens_seen": 1290403840 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003076028084252758, + "loss": 1.3817, + "theoretical_loss": 3.563719697148179, + "tokens_seen": 1290469376 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030759277833500504, + "loss": 1.4431, + "theoretical_loss": 3.563703276454861, + "tokens_seen": 1290534912 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003075827482447342, + "loss": 1.3012, + "theoretical_loss": 3.56368685682887, + "tokens_seen": 1290600448 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003075727181544634, + "loss": 1.3609, + "theoretical_loss": 3.563670438270083, + "tokens_seen": 1290665984 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003075626880641926, + "loss": 1.3618, + "theoretical_loss": 3.563654020778376, + "tokens_seen": 1290731520 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030755265797392177, + "loss": 1.284, + "theoretical_loss": 3.563637604353625, + "tokens_seen": 1290797056 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030754262788365095, + "loss": 1.3965, + "theoretical_loss": 3.5636211889957066, + "tokens_seen": 1290862592 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003075325977933802, + "loss": 1.3572, + "theoretical_loss": 3.5636047747044985, + "tokens_seen": 1290928128 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003075225677031093, + "loss": 1.427, + "theoretical_loss": 3.5635883614798756, + "tokens_seen": 1290993664 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -0.0012723995605483651, + "objective/train/docs_used": 732356, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.271653413772583, + "objective/train/original_loss": 3.271653175354004, + "objective/train/theoretical_loss": 3.5635719493217155, + "objective/train/tokens_used": 1311519200, + "objective/train/value_avg": -0.006134033203125, + "objective/train/value_loss": 7.081093644956127e-05, + "objective/train/value_max": -0.0002715587615966797, + "objective/train/value_min": -0.0634765625, + "objective/train/value_reward_corr": 0.6604745652753466, + "objective/train/value_std": 0.00621795654296875, + "objective/train/weight_avg": 0.9906281232833862, + "objective/train/weighted_lm_loss": 3.2389533519744873, + "objective/train/weights_max": 1.5225297212600708, + "objective/train/weights_min": 0.4670711159706116, + "theoretical_loss": 3.5635719493217155, + "tokens_seen": 1291059200 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030751253761283854, + "loss": 1.397, + "theoretical_loss": 3.5635719493217155, + "tokens_seen": 1291059200 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003075025075225677, + "loss": 1.3777, + "theoretical_loss": 3.5635555382298945, + "tokens_seen": 1291124736 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003074924774322969, + "loss": 1.3913, + "theoretical_loss": 3.563539128204289, + "tokens_seen": 1291190272 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003074824473420261, + "loss": 1.3788, + "theoretical_loss": 3.563522719244776, + "tokens_seen": 1291255808 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030747241725175527, + "loss": 1.296, + "theoretical_loss": 3.563506311351232, + "tokens_seen": 1291321344 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030746238716148445, + "loss": 1.3918, + "theoretical_loss": 3.563489904523533, + "tokens_seen": 1291386880 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003074523570712137, + "loss": 1.3765, + "theoretical_loss": 3.563473498761557, + "tokens_seen": 1291452416 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003074423269809428, + "loss": 1.4162, + "theoretical_loss": 3.56345709406518, + "tokens_seen": 1291517952 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030743229689067205, + "loss": 1.3622, + "theoretical_loss": 3.5634406904342786, + "tokens_seen": 1291583488 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003074222668004012, + "loss": 1.3012, + "theoretical_loss": 3.5634242878687292, + "tokens_seen": 1291649024 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003074122367101304, + "loss": 1.3774, + "theoretical_loss": 3.5634078863684095, + "tokens_seen": 1291714560 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003074022066198596, + "loss": 1.435, + "theoretical_loss": 3.563391485933195, + "tokens_seen": 1291780096 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030739217652958877, + "loss": 1.3839, + "theoretical_loss": 3.563375086562964, + "tokens_seen": 1291845632 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030738214643931795, + "loss": 1.4316, + "theoretical_loss": 3.5633586882575923, + "tokens_seen": 1291911168 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030737211634904713, + "loss": 1.5013, + "theoretical_loss": 3.563342291016957, + "tokens_seen": 1291976704 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003073620862587763, + "loss": 1.2651, + "theoretical_loss": 3.5633258948409354, + "tokens_seen": 1292042240 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030735205616850555, + "loss": 1.2813, + "theoretical_loss": 3.563309499729404, + "tokens_seen": 1292107776 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003073420260782347, + "loss": 1.4148, + "theoretical_loss": 3.5632931056822392, + "tokens_seen": 1292173312 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003073319959879639, + "loss": 1.4406, + "theoretical_loss": 3.563276712699319, + "tokens_seen": 1292238848 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003073219658976931, + "loss": 1.5581, + "theoretical_loss": 3.563260320780519, + "tokens_seen": 1292304384 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003073119358074223, + "loss": 1.4906, + "theoretical_loss": 3.5632439299257177, + "tokens_seen": 1292369920 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030730190571715146, + "loss": 1.3622, + "theoretical_loss": 3.5632275401347906, + "tokens_seen": 1292435456 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030729187562688064, + "loss": 1.343, + "theoretical_loss": 3.563211151407616, + "tokens_seen": 1292500992 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003072818455366098, + "loss": 1.4304, + "theoretical_loss": 3.56319476374407, + "tokens_seen": 1292566528 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030727181544633905, + "loss": 1.3472, + "theoretical_loss": 3.56317837714403, + "tokens_seen": 1292632064 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": 9.43584309425205e-05, + "objective/train/docs_used": 733109, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.821479558944702, + "objective/train/original_loss": 2.821479320526123, + "objective/train/theoretical_loss": 3.563161991607373, + "objective/train/tokens_used": 1313157600, + "objective/train/value_avg": -0.004791259765625, + "objective/train/value_loss": 5.259335375740193e-05, + "objective/train/value_max": -0.00023233890533447266, + "objective/train/value_min": -0.0703125, + "objective/train/value_reward_corr": 0.38877817561092176, + "objective/train/value_std": 0.004425048828125, + "objective/train/weight_avg": 1.0033382177352905, + "objective/train/weighted_lm_loss": 2.836993932723999, + "objective/train/weights_max": 1.5501253604888916, + "objective/train/weights_min": 0.13719095289707184, + "theoretical_loss": 3.563161991607373, + "tokens_seen": 1292697600 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003072617853560682, + "loss": 1.3635, + "theoretical_loss": 3.563161991607373, + "tokens_seen": 1292697600 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003072517552657974, + "loss": 1.3374, + "theoretical_loss": 3.5631456071339764, + "tokens_seen": 1292763136 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030724172517552654, + "loss": 1.3646, + "theoretical_loss": 3.563129223723717, + "tokens_seen": 1292828672 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003072316950852558, + "loss": 1.3758, + "theoretical_loss": 3.563112841376472, + "tokens_seen": 1292894208 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030722166499498496, + "loss": 1.2611, + "theoretical_loss": 3.5630964600921184, + "tokens_seen": 1292959744 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030721163490471414, + "loss": 1.4153, + "theoretical_loss": 3.5630800798705335, + "tokens_seen": 1293025280 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003072016048144433, + "loss": 1.3626, + "theoretical_loss": 3.5630637007115946, + "tokens_seen": 1293090816 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003071915747241725, + "loss": 1.402, + "theoretical_loss": 3.563047322615179, + "tokens_seen": 1293156352 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003071815446339017, + "loss": 1.3308, + "theoretical_loss": 3.5630309455811635, + "tokens_seen": 1293221888 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003071715145436309, + "loss": 1.4531, + "theoretical_loss": 3.5630145696094258, + "tokens_seen": 1293287424 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030716148445336005, + "loss": 1.4635, + "theoretical_loss": 3.5629981946998424, + "tokens_seen": 1293352960 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003071514543630893, + "loss": 1.4939, + "theoretical_loss": 3.5629818208522916, + "tokens_seen": 1293418496 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030714142427281846, + "loss": 1.4133, + "theoretical_loss": 3.5629654480666506, + "tokens_seen": 1293484032 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030713139418254764, + "loss": 1.4945, + "theoretical_loss": 3.562949076342796, + "tokens_seen": 1293549568 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003071213640922768, + "loss": 1.3982, + "theoretical_loss": 3.5629327056806055, + "tokens_seen": 1293615104 + }, + { + "epoch": 0.39, + "learning_rate": 0.000307111334002006, + "loss": 1.3099, + "theoretical_loss": 3.5629163360799563, + "tokens_seen": 1293680640 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003071013039117352, + "loss": 1.3552, + "theoretical_loss": 3.5628999675407265, + "tokens_seen": 1293746176 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003070912738214644, + "loss": 1.4426, + "theoretical_loss": 3.5628836000627926, + "tokens_seen": 1293811712 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030708124373119355, + "loss": 1.2968, + "theoretical_loss": 3.562867233646033, + "tokens_seen": 1293877248 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003070712136409228, + "loss": 1.4087, + "theoretical_loss": 3.562850868290324, + "tokens_seen": 1293942784 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003070611835506519, + "loss": 1.3369, + "theoretical_loss": 3.5628345039955445, + "tokens_seen": 1294008320 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030705115346038115, + "loss": 1.3137, + "theoretical_loss": 3.5628181407615704, + "tokens_seen": 1294073856 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003070411233701104, + "loss": 1.2891, + "theoretical_loss": 3.5628017785882804, + "tokens_seen": 1294139392 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003070310932798395, + "loss": 1.3533, + "theoretical_loss": 3.5627854174755518, + "tokens_seen": 1294204928 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030702106318956874, + "loss": 1.44, + "theoretical_loss": 3.562769057423262, + "tokens_seen": 1294270464 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": 0.000246052019065246, + "objective/train/docs_used": 734261, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.840787172317505, + "objective/train/original_loss": 2.840787410736084, + "objective/train/theoretical_loss": 3.5627526984312885, + "objective/train/tokens_used": 1314796000, + "objective/train/value_avg": -0.00476837158203125, + "objective/train/value_loss": 3.711276804096997e-05, + "objective/train/value_max": -0.0004973411560058594, + "objective/train/value_min": -0.05877685546875, + "objective/train/value_reward_corr": 0.23025937979833755, + "objective/train/value_std": 0.0033721923828125, + "objective/train/weight_avg": 1.004258632659912, + "objective/train/weighted_lm_loss": 2.8562376499176025, + "objective/train/weights_max": 1.7999674081802368, + "objective/train/weights_min": 0.47744590044021606, + "theoretical_loss": 3.5627526984312885, + "tokens_seen": 1294336000 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003070110330992979, + "loss": 1.402, + "theoretical_loss": 3.5627526984312885, + "tokens_seen": 1294336000 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003070010030090271, + "loss": 1.3944, + "theoretical_loss": 3.5627363404995087, + "tokens_seen": 1294401536 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003069909729187563, + "loss": 1.3813, + "theoretical_loss": 3.562719983627801, + "tokens_seen": 1294467072 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030698094282848547, + "loss": 1.3529, + "theoretical_loss": 3.5627036278160418, + "tokens_seen": 1294532608 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030697091273821465, + "loss": 1.3312, + "theoretical_loss": 3.5626872730641104, + "tokens_seen": 1294598144 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003069608826479439, + "loss": 1.3718, + "theoretical_loss": 3.562670919371883, + "tokens_seen": 1294663680 + }, + { + "epoch": 0.39, + "learning_rate": 0.000306950852557673, + "loss": 1.4003, + "theoretical_loss": 3.562654566739238, + "tokens_seen": 1294729216 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030694082246740225, + "loss": 1.3736, + "theoretical_loss": 3.5626382151660527, + "tokens_seen": 1294794752 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003069307923771314, + "loss": 1.5105, + "theoretical_loss": 3.562621864652206, + "tokens_seen": 1294860288 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003069207622868606, + "loss": 1.4475, + "theoretical_loss": 3.562605515197574, + "tokens_seen": 1294925824 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003069107321965898, + "loss": 1.4029, + "theoretical_loss": 3.5625891668020353, + "tokens_seen": 1294991360 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030690070210631897, + "loss": 1.341, + "theoretical_loss": 3.562572819465468, + "tokens_seen": 1295056896 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030689067201604815, + "loss": 1.4649, + "theoretical_loss": 3.5625564731877493, + "tokens_seen": 1295122432 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030688064192577733, + "loss": 1.399, + "theoretical_loss": 3.5625401279687576, + "tokens_seen": 1295187968 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003068706118355065, + "loss": 1.3967, + "theoretical_loss": 3.5625237838083708, + "tokens_seen": 1295253504 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030686058174523575, + "loss": 1.3931, + "theoretical_loss": 3.562507440706466, + "tokens_seen": 1295319040 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003068505516549649, + "loss": 1.3027, + "theoretical_loss": 3.562491098662922, + "tokens_seen": 1295384576 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003068405215646941, + "loss": 1.3981, + "theoretical_loss": 3.5624747576776157, + "tokens_seen": 1295450112 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003068304914744233, + "loss": 1.3937, + "theoretical_loss": 3.562458417750426, + "tokens_seen": 1295515648 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003068204613841525, + "loss": 1.3846, + "theoretical_loss": 3.5624420788812308, + "tokens_seen": 1295581184 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030681043129388166, + "loss": 1.4161, + "theoretical_loss": 3.5624257410699074, + "tokens_seen": 1295646720 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030680040120361084, + "loss": 1.3851, + "theoretical_loss": 3.5624094043163343, + "tokens_seen": 1295712256 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030679037111334, + "loss": 1.3406, + "theoretical_loss": 3.5623930686203895, + "tokens_seen": 1295777792 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030678034102306925, + "loss": 1.3488, + "theoretical_loss": 3.562376733981951, + "tokens_seen": 1295843328 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003067703109327984, + "loss": 1.4881, + "theoretical_loss": 3.5623604004008964, + "tokens_seen": 1295908864 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -0.0005633692489936948, + "objective/train/docs_used": 734748, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8928816318511963, + "objective/train/original_loss": 2.8928816318511963, + "objective/train/theoretical_loss": 3.562344067877105, + "objective/train/tokens_used": 1316434400, + "objective/train/value_avg": -0.004730224609375, + "objective/train/value_loss": 6.09232738497667e-05, + "objective/train/value_max": -0.0004372596740722656, + "objective/train/value_min": -0.0631103515625, + "objective/train/value_reward_corr": 0.29570180790286016, + "objective/train/value_std": 0.00362396240234375, + "objective/train/weight_avg": 0.9971159100532532, + "objective/train/weighted_lm_loss": 2.892956256866455, + "objective/train/weights_max": 1.6928128004074097, + "objective/train/weights_min": 0.20525413751602173, + "theoretical_loss": 3.562344067877105, + "tokens_seen": 1295974400 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003067602808425276, + "loss": 1.3637, + "theoretical_loss": 3.562344067877105, + "tokens_seen": 1295974400 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030675025075225674, + "loss": 1.4015, + "theoretical_loss": 3.5623277364104537, + "tokens_seen": 1296039936 + }, + { + "epoch": 0.39, + "learning_rate": 0.000306740220661986, + "loss": 1.2742, + "theoretical_loss": 3.5623114060008207, + "tokens_seen": 1296105472 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030673019057171516, + "loss": 1.3703, + "theoretical_loss": 3.562295076648085, + "tokens_seen": 1296171008 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030672016048144434, + "loss": 1.3499, + "theoretical_loss": 3.5622787483521243, + "tokens_seen": 1296236544 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003067101303911735, + "loss": 1.4468, + "theoretical_loss": 3.562262421112816, + "tokens_seen": 1296302080 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003067001003009027, + "loss": 1.3693, + "theoretical_loss": 3.56224609493004, + "tokens_seen": 1296367616 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003066900702106319, + "loss": 1.3438, + "theoretical_loss": 3.5622297698036736, + "tokens_seen": 1296433152 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003066800401203611, + "loss": 1.4606, + "theoretical_loss": 3.5622134457335943, + "tokens_seen": 1296498688 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030667001003009025, + "loss": 1.4231, + "theoretical_loss": 3.5621971227196814, + "tokens_seen": 1296564224 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003066599799398195, + "loss": 1.3537, + "theoretical_loss": 3.562180800761813, + "tokens_seen": 1296629760 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030664994984954866, + "loss": 1.453, + "theoretical_loss": 3.562164479859867, + "tokens_seen": 1296695296 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030663991975927784, + "loss": 1.4876, + "theoretical_loss": 3.5621481600137224, + "tokens_seen": 1296760832 + }, + { + "epoch": 0.39, + "learning_rate": 0.000306629889669007, + "loss": 1.4009, + "theoretical_loss": 3.5621318412232568, + "tokens_seen": 1296826368 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003066198595787362, + "loss": 1.2493, + "theoretical_loss": 3.562115523488349, + "tokens_seen": 1296891904 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003066098294884654, + "loss": 1.3385, + "theoretical_loss": 3.5620992068088775, + "tokens_seen": 1296957440 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003065997993981946, + "loss": 1.3232, + "theoretical_loss": 3.56208289118472, + "tokens_seen": 1297022976 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030658976930792375, + "loss": 1.354, + "theoretical_loss": 3.562066576615756, + "tokens_seen": 1297088512 + }, + { + "epoch": 0.39, + "learning_rate": 0.000306579739217653, + "loss": 1.3786, + "theoretical_loss": 3.5620502631018636, + "tokens_seen": 1297154048 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003065697091273821, + "loss": 1.4211, + "theoretical_loss": 3.5620339506429204, + "tokens_seen": 1297219584 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030655967903711135, + "loss": 1.311, + "theoretical_loss": 3.562017639238806, + "tokens_seen": 1297285120 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030654964894684053, + "loss": 1.3378, + "theoretical_loss": 3.5620013288893984, + "tokens_seen": 1297350656 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003065396188565697, + "loss": 1.2516, + "theoretical_loss": 3.5619850195945757, + "tokens_seen": 1297416192 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003065295887662989, + "loss": 1.4777, + "theoretical_loss": 3.5619687113542176, + "tokens_seen": 1297481728 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003065195586760281, + "loss": 1.2999, + "theoretical_loss": 3.561952404168202, + "tokens_seen": 1297547264 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": 0.0005342800286598504, + "objective/train/docs_used": 735602, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.698647975921631, + "objective/train/original_loss": 2.6986477375030518, + "objective/train/theoretical_loss": 3.5619360980364068, + "objective/train/tokens_used": 1318072800, + "objective/train/value_avg": -0.004665374755859375, + "objective/train/value_loss": 6.484634650405496e-05, + "objective/train/value_max": -0.0004012584686279297, + "objective/train/value_min": -0.047515869140625, + "objective/train/value_reward_corr": 0.26852781156106054, + "objective/train/value_std": 0.0032405853271484375, + "objective/train/weight_avg": 1.0082155466079712, + "objective/train/weighted_lm_loss": 2.7321298122406006, + "objective/train/weights_max": 1.4481253623962402, + "objective/train/weights_min": 0.2601744532585144, + "theoretical_loss": 3.5619360980364068, + "tokens_seen": 1297612800 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030650952858575725, + "loss": 1.3441, + "theoretical_loss": 3.5619360980364068, + "tokens_seen": 1297612800 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003064994984954865, + "loss": 1.4605, + "theoretical_loss": 3.5619197929587116, + "tokens_seen": 1297678336 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003064894684052156, + "loss": 1.4075, + "theoretical_loss": 3.561903488934995, + "tokens_seen": 1297743872 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030647943831494485, + "loss": 1.4348, + "theoretical_loss": 3.5618871859651353, + "tokens_seen": 1297809408 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030646940822467403, + "loss": 1.3519, + "theoretical_loss": 3.561870884049011, + "tokens_seen": 1297874944 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003064593781344032, + "loss": 1.2676, + "theoretical_loss": 3.561854583186501, + "tokens_seen": 1297940480 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003064493480441324, + "loss": 1.3495, + "theoretical_loss": 3.561838283377485, + "tokens_seen": 1298006016 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003064393179538616, + "loss": 1.3108, + "theoretical_loss": 3.5618219846218393, + "tokens_seen": 1298071552 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030642928786359076, + "loss": 1.4458, + "theoretical_loss": 3.5618056869194454, + "tokens_seen": 1298137088 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030641925777332, + "loss": 1.378, + "theoretical_loss": 3.56178939027018, + "tokens_seen": 1298202624 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003064092276830491, + "loss": 1.363, + "theoretical_loss": 3.561773094673923, + "tokens_seen": 1298268160 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030639919759277835, + "loss": 1.4437, + "theoretical_loss": 3.561756800130553, + "tokens_seen": 1298333696 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003063891675025075, + "loss": 1.2607, + "theoretical_loss": 3.5617405066399486, + "tokens_seen": 1298399232 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003063791374122367, + "loss": 1.3514, + "theoretical_loss": 3.5617242142019885, + "tokens_seen": 1298464768 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003063691073219659, + "loss": 1.3944, + "theoretical_loss": 3.5617079228165522, + "tokens_seen": 1298530304 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003063590772316951, + "loss": 1.3116, + "theoretical_loss": 3.5616916324835177, + "tokens_seen": 1298595840 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030634904714142426, + "loss": 1.4337, + "theoretical_loss": 3.561675343202765, + "tokens_seen": 1298661376 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003063390170511535, + "loss": 1.4799, + "theoretical_loss": 3.5616590549741725, + "tokens_seen": 1298726912 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003063289869608826, + "loss": 1.4018, + "theoretical_loss": 3.5616427677976192, + "tokens_seen": 1298792448 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030631895687061186, + "loss": 1.3536, + "theoretical_loss": 3.5616264816729837, + "tokens_seen": 1298857984 + }, + { + "epoch": 0.39, + "learning_rate": 0.000306308926780341, + "loss": 1.5552, + "theoretical_loss": 3.561610196600145, + "tokens_seen": 1298923520 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003062988966900702, + "loss": 1.4576, + "theoretical_loss": 3.561593912578983, + "tokens_seen": 1298989056 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030628886659979945, + "loss": 1.3217, + "theoretical_loss": 3.561577629609376, + "tokens_seen": 1299054592 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003062788365095286, + "loss": 1.4525, + "theoretical_loss": 3.5615613476912027, + "tokens_seen": 1299120128 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003062688064192578, + "loss": 1.3452, + "theoretical_loss": 3.561545066824343, + "tokens_seen": 1299185664 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -0.0009743698174133897, + "objective/train/docs_used": 736089, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.943082571029663, + "objective/train/original_loss": 2.9430830478668213, + "objective/train/theoretical_loss": 3.5615287870086756, + "objective/train/tokens_used": 1319711200, + "objective/train/value_avg": -0.005313873291015625, + "objective/train/value_loss": 5.838671131641604e-05, + "objective/train/value_max": -0.0002378225326538086, + "objective/train/value_min": -0.091552734375, + "objective/train/value_reward_corr": 0.4041537525293448, + "objective/train/value_std": 0.00458526611328125, + "objective/train/weight_avg": 0.9930224418640137, + "objective/train/weighted_lm_loss": 2.925159454345703, + "objective/train/weights_max": 2.4980924129486084, + "objective/train/weights_min": 0.3115394413471222, + "theoretical_loss": 3.5615287870086756, + "tokens_seen": 1299251200 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030625877632898694, + "loss": 1.3682, + "theoretical_loss": 3.5615287870086756, + "tokens_seen": 1299251200 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003062487462387162, + "loss": 1.4211, + "theoretical_loss": 3.5615125082440793, + "tokens_seen": 1299316736 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030623871614844536, + "loss": 1.4591, + "theoretical_loss": 3.561496230530434, + "tokens_seen": 1299382272 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030622868605817454, + "loss": 1.4812, + "theoretical_loss": 3.5614799538676185, + "tokens_seen": 1299447808 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003062186559679037, + "loss": 1.364, + "theoretical_loss": 3.561463678255512, + "tokens_seen": 1299513344 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003062086258776329, + "loss": 1.3415, + "theoretical_loss": 3.561447403693993, + "tokens_seen": 1299578880 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003061985957873621, + "loss": 1.3663, + "theoretical_loss": 3.5614311301829416, + "tokens_seen": 1299644416 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003061885656970913, + "loss": 1.3827, + "theoretical_loss": 3.561414857722237, + "tokens_seen": 1299709952 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030617853560682045, + "loss": 1.3657, + "theoretical_loss": 3.561398586311758, + "tokens_seen": 1299775488 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003061685055165497, + "loss": 1.4839, + "theoretical_loss": 3.5613823159513838, + "tokens_seen": 1299841024 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030615847542627886, + "loss": 1.3583, + "theoretical_loss": 3.5613660466409938, + "tokens_seen": 1299906560 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030614844533600804, + "loss": 1.3493, + "theoretical_loss": 3.561349778380468, + "tokens_seen": 1299972096 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003061384152457372, + "loss": 1.2881, + "theoretical_loss": 3.5613335111696847, + "tokens_seen": 1300037632 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003061283851554664, + "loss": 1.491, + "theoretical_loss": 3.561317245008524, + "tokens_seen": 1300103168 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003061183550651956, + "loss": 1.4294, + "theoretical_loss": 3.561300979896865, + "tokens_seen": 1300168704 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003061083249749248, + "loss": 1.4933, + "theoretical_loss": 3.561284715834587, + "tokens_seen": 1300234240 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030609829488465395, + "loss": 1.3386, + "theoretical_loss": 3.5612684528215697, + "tokens_seen": 1300299776 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003060882647943832, + "loss": 1.3939, + "theoretical_loss": 3.561252190857692, + "tokens_seen": 1300365312 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003060782347041123, + "loss": 1.352, + "theoretical_loss": 3.5612359299428338, + "tokens_seen": 1300430848 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030606820461384155, + "loss": 1.5318, + "theoretical_loss": 3.5612196700768743, + "tokens_seen": 1300496384 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030605817452357073, + "loss": 1.2587, + "theoretical_loss": 3.5612034112596933, + "tokens_seen": 1300561920 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003060481444332999, + "loss": 1.3525, + "theoretical_loss": 3.5611871534911703, + "tokens_seen": 1300627456 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003060381143430291, + "loss": 1.3671, + "theoretical_loss": 3.561170896771184, + "tokens_seen": 1300692992 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003060280842527583, + "loss": 1.5023, + "theoretical_loss": 3.5611546410996153, + "tokens_seen": 1300758528 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030601805416248745, + "loss": 1.5216, + "theoretical_loss": 3.561138386476343, + "tokens_seen": 1300824064 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": 0.0006661753286607563, + "objective/train/docs_used": 737292, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4573729038238525, + "objective/train/original_loss": 2.4573729038238525, + "objective/train/theoretical_loss": 3.5611221329012466, + "objective/train/tokens_used": 1321349600, + "objective/train/value_avg": -0.00499725341796875, + "objective/train/value_loss": 3.350884435349144e-05, + "objective/train/value_max": -0.0005011558532714844, + "objective/train/value_min": -0.07159423828125, + "objective/train/value_reward_corr": 0.34994675755111165, + "objective/train/value_std": 0.00377655029296875, + "objective/train/weight_avg": 1.008323311805725, + "objective/train/weighted_lm_loss": 2.469413995742798, + "objective/train/weights_max": 1.848900318145752, + "objective/train/weights_min": 0.6609131693840027, + "theoretical_loss": 3.5611221329012466, + "tokens_seen": 1300889600 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003060080240722167, + "loss": 1.4041, + "theoretical_loss": 3.5611221329012466, + "tokens_seen": 1300889600 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003059979939819458, + "loss": 1.4303, + "theoretical_loss": 3.561105880374206, + "tokens_seen": 1300955136 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030598796389167505, + "loss": 1.4024, + "theoretical_loss": 3.5610896288951004, + "tokens_seen": 1301020672 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030597793380140423, + "loss": 1.3809, + "theoretical_loss": 3.5610733784638104, + "tokens_seen": 1301086208 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003059679037111334, + "loss": 1.4479, + "theoretical_loss": 3.561057129080215, + "tokens_seen": 1301151744 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003059578736208626, + "loss": 1.3109, + "theoretical_loss": 3.5610408807441933, + "tokens_seen": 1301217280 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003059478435305918, + "loss": 1.3451, + "theoretical_loss": 3.5610246334556255, + "tokens_seen": 1301282816 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030593781344032096, + "loss": 1.457, + "theoretical_loss": 3.5610083872143923, + "tokens_seen": 1301348352 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003059277833500502, + "loss": 1.5584, + "theoretical_loss": 3.560992142020372, + "tokens_seen": 1301413888 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003059177532597793, + "loss": 1.3097, + "theoretical_loss": 3.560975897873445, + "tokens_seen": 1301479424 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030590772316950855, + "loss": 1.4656, + "theoretical_loss": 3.5609596547734914, + "tokens_seen": 1301544960 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003058976930792377, + "loss": 1.4117, + "theoretical_loss": 3.5609434127203903, + "tokens_seen": 1301610496 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003058876629889669, + "loss": 1.3719, + "theoretical_loss": 3.560927171714022, + "tokens_seen": 1301676032 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003058776328986961, + "loss": 1.3903, + "theoretical_loss": 3.5609109317542664, + "tokens_seen": 1301741568 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003058676028084253, + "loss": 1.4408, + "theoretical_loss": 3.5608946928410035, + "tokens_seen": 1301807104 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030585757271815446, + "loss": 1.4678, + "theoretical_loss": 3.5608784549741124, + "tokens_seen": 1301872640 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003058475426278837, + "loss": 1.4915, + "theoretical_loss": 3.5608622181534733, + "tokens_seen": 1301938176 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003058375125376128, + "loss": 1.3297, + "theoretical_loss": 3.560845982378967, + "tokens_seen": 1302003712 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030582748244734206, + "loss": 1.3797, + "theoretical_loss": 3.560829747650472, + "tokens_seen": 1302069248 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003058174523570712, + "loss": 1.3788, + "theoretical_loss": 3.5608135139678696, + "tokens_seen": 1302134784 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003058074222668004, + "loss": 1.4718, + "theoretical_loss": 3.5607972813310393, + "tokens_seen": 1302200320 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003057973921765296, + "loss": 1.3089, + "theoretical_loss": 3.5607810497398606, + "tokens_seen": 1302265856 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003057873620862588, + "loss": 1.318, + "theoretical_loss": 3.5607648191942145, + "tokens_seen": 1302331392 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030577733199598796, + "loss": 1.4261, + "theoretical_loss": 3.56074858969398, + "tokens_seen": 1302396928 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030576730190571714, + "loss": 1.4059, + "theoretical_loss": 3.560732361239038, + "tokens_seen": 1302462464 + }, + { + "epoch": 0.39, + "objective/train/advantage_avg": -5.000009332434274e-05, + "objective/train/docs_used": 737760, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0377941131591797, + "objective/train/original_loss": 3.0377938747406006, + "objective/train/theoretical_loss": 3.5607161338292683, + "objective/train/tokens_used": 1322988000, + "objective/train/value_avg": -0.003910064697265625, + "objective/train/value_loss": 4.432622154126875e-05, + "objective/train/value_max": -0.00035834312438964844, + "objective/train/value_min": -0.034698486328125, + "objective/train/value_reward_corr": 0.3308248658064557, + "objective/train/value_std": 0.0029277801513671875, + "objective/train/weight_avg": 1.0015881061553955, + "objective/train/weighted_lm_loss": 3.053602695465088, + "objective/train/weights_max": 1.3425359725952148, + "objective/train/weights_min": 0.4297851026058197, + "theoretical_loss": 3.5607161338292683, + "tokens_seen": 1302528000 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003057572718154463, + "loss": 1.4731, + "theoretical_loss": 3.5607161338292683, + "tokens_seen": 1302528000 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030574724172517556, + "loss": 1.3051, + "theoretical_loss": 3.5606999074645507, + "tokens_seen": 1302593536 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003057372116349047, + "loss": 1.3712, + "theoretical_loss": 3.5606836821447656, + "tokens_seen": 1302659072 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003057271815446339, + "loss": 1.358, + "theoretical_loss": 3.560667457869793, + "tokens_seen": 1302724608 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030571715145436305, + "loss": 1.391, + "theoretical_loss": 3.5606512346395136, + "tokens_seen": 1302790144 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003057071213640923, + "loss": 1.4395, + "theoretical_loss": 3.5606350124538073, + "tokens_seen": 1302855680 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030569709127382147, + "loss": 1.3775, + "theoretical_loss": 3.5606187913125535, + "tokens_seen": 1302921216 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030568706118355065, + "loss": 1.4073, + "theoretical_loss": 3.5606025712156333, + "tokens_seen": 1302986752 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030567703109327983, + "loss": 1.4117, + "theoretical_loss": 3.560586352162927, + "tokens_seen": 1303052288 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030566700100300906, + "loss": 1.3609, + "theoretical_loss": 3.5605701341543146, + "tokens_seen": 1303117824 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003056569709127382, + "loss": 1.3708, + "theoretical_loss": 3.5605539171896767, + "tokens_seen": 1303183360 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003056469408224674, + "loss": 1.4471, + "theoretical_loss": 3.560537701268893, + "tokens_seen": 1303248896 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030563691073219655, + "loss": 1.3933, + "theoretical_loss": 3.5605214863918437, + "tokens_seen": 1303314432 + }, + { + "epoch": 0.39, + "learning_rate": 0.0003056268806419258, + "loss": 1.3625, + "theoretical_loss": 3.56050527255841, + "tokens_seen": 1303379968 + }, + { + "epoch": 0.39, + "learning_rate": 0.00030561685055165497, + "loss": 1.4572, + "theoretical_loss": 3.5604890597684715, + "tokens_seen": 1303445504 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030560682046138415, + "loss": 1.3531, + "theoretical_loss": 3.5604728480219094, + "tokens_seen": 1303511040 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030559679037111333, + "loss": 1.4226, + "theoretical_loss": 3.5604566373186035, + "tokens_seen": 1303576576 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003055867602808425, + "loss": 1.4082, + "theoretical_loss": 3.560440427658434, + "tokens_seen": 1303642112 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003055767301905717, + "loss": 1.2471, + "theoretical_loss": 3.560424219041282, + "tokens_seen": 1303707648 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030556670010030093, + "loss": 1.4497, + "theoretical_loss": 3.560408011467027, + "tokens_seen": 1303773184 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030555667001003006, + "loss": 1.2953, + "theoretical_loss": 3.560391804935551, + "tokens_seen": 1303838720 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003055466399197593, + "loss": 1.4171, + "theoretical_loss": 3.5603755994467328, + "tokens_seen": 1303904256 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003055366098294885, + "loss": 1.4206, + "theoretical_loss": 3.5603593950004546, + "tokens_seen": 1303969792 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030552657973921765, + "loss": 1.3635, + "theoretical_loss": 3.5603431915965955, + "tokens_seen": 1304035328 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003055165496489469, + "loss": 1.4173, + "theoretical_loss": 3.5603269892350364, + "tokens_seen": 1304100864 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": 0.0001686174509814009, + "objective/train/docs_used": 739145, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8617215156555176, + "objective/train/original_loss": 2.8617212772369385, + "objective/train/theoretical_loss": 3.5603107879156584, + "objective/train/tokens_used": 1324626400, + "objective/train/value_avg": -0.00507354736328125, + "objective/train/value_loss": 5.505370063474402e-05, + "objective/train/value_max": -0.0003249645233154297, + "objective/train/value_min": -0.0699462890625, + "objective/train/value_reward_corr": 0.46310393255764254, + "objective/train/value_std": 0.00461578369140625, + "objective/train/weight_avg": 1.0042614936828613, + "objective/train/weighted_lm_loss": 2.8786120414733887, + "objective/train/weights_max": 1.712446689605713, + "objective/train/weights_min": 0.43080514669418335, + "theoretical_loss": 3.5603107879156584, + "tokens_seen": 1304166400 + }, + { + "epoch": 0.4, + "learning_rate": 0.000305506519558676, + "loss": 1.4776, + "theoretical_loss": 3.5603107879156584, + "tokens_seen": 1304166400 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030549648946840525, + "loss": 1.4567, + "theoretical_loss": 3.5602945876383423, + "tokens_seen": 1304231936 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030548645937813443, + "loss": 1.3328, + "theoretical_loss": 3.5602783884029674, + "tokens_seen": 1304297472 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003054764292878636, + "loss": 1.4427, + "theoretical_loss": 3.5602621902094156, + "tokens_seen": 1304363008 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003054663991975928, + "loss": 1.3451, + "theoretical_loss": 3.560245993057567, + "tokens_seen": 1304428544 + }, + { + "epoch": 0.4, + "learning_rate": 0.000305456369107322, + "loss": 1.3171, + "theoretical_loss": 3.5602297969473025, + "tokens_seen": 1304494080 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030544633901705116, + "loss": 1.462, + "theoretical_loss": 3.5602136018785027, + "tokens_seen": 1304559616 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003054363089267804, + "loss": 1.3184, + "theoretical_loss": 3.5601974078510477, + "tokens_seen": 1304625152 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003054262788365095, + "loss": 1.4566, + "theoretical_loss": 3.5601812148648193, + "tokens_seen": 1304690688 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030541624874623875, + "loss": 1.3971, + "theoretical_loss": 3.5601650229196977, + "tokens_seen": 1304756224 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003054062186559679, + "loss": 1.3319, + "theoretical_loss": 3.560148832015564, + "tokens_seen": 1304821760 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003053961885656971, + "loss": 1.3615, + "theoretical_loss": 3.5601326421522987, + "tokens_seen": 1304887296 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003053861584754263, + "loss": 1.4478, + "theoretical_loss": 3.5601164533297824, + "tokens_seen": 1304952832 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003053761283851555, + "loss": 1.4603, + "theoretical_loss": 3.560100265547897, + "tokens_seen": 1305018368 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030536609829488466, + "loss": 1.3946, + "theoretical_loss": 3.560084078806522, + "tokens_seen": 1305083904 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003053560682046139, + "loss": 1.4025, + "theoretical_loss": 3.5600678931055385, + "tokens_seen": 1305149440 + }, + { + "epoch": 0.4, + "learning_rate": 0.000305346038114343, + "loss": 1.3746, + "theoretical_loss": 3.560051708444828, + "tokens_seen": 1305214976 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030533600802407226, + "loss": 1.3804, + "theoretical_loss": 3.560035524824271, + "tokens_seen": 1305280512 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003053259779338014, + "loss": 1.3716, + "theoretical_loss": 3.5600193422437494, + "tokens_seen": 1305346048 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003053159478435306, + "loss": 1.3879, + "theoretical_loss": 3.5600031607031424, + "tokens_seen": 1305411584 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003053059177532598, + "loss": 1.5045, + "theoretical_loss": 3.5599869802023325, + "tokens_seen": 1305477120 + }, + { + "epoch": 0.4, + "learning_rate": 0.000305295887662989, + "loss": 1.2968, + "theoretical_loss": 3.5599708007412, + "tokens_seen": 1305542656 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030528585757271816, + "loss": 1.193, + "theoretical_loss": 3.5599546223196255, + "tokens_seen": 1305608192 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030527582748244734, + "loss": 1.4294, + "theoretical_loss": 3.559938444937491, + "tokens_seen": 1305673728 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003052657973921765, + "loss": 1.2779, + "theoretical_loss": 3.5599222685946765, + "tokens_seen": 1305739264 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.0012420525308698416, + "objective/train/docs_used": 739729, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5409059524536133, + "objective/train/original_loss": 2.540905714035034, + "objective/train/theoretical_loss": 3.5599060932910644, + "objective/train/tokens_used": 1326264800, + "objective/train/value_avg": -0.00608062744140625, + "objective/train/value_loss": 7.339736475842074e-05, + "objective/train/value_max": -0.00036978721618652344, + "objective/train/value_min": -0.089599609375, + "objective/train/value_reward_corr": 0.6297087453095223, + "objective/train/value_std": 0.006317138671875, + "objective/train/weight_avg": 0.9910397529602051, + "objective/train/weighted_lm_loss": 2.5236823558807373, + "objective/train/weights_max": 1.7278450727462769, + "objective/train/weights_min": 0.5393592119216919, + "theoretical_loss": 3.5599060932910644, + "tokens_seen": 1305804800 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030525576730190576, + "loss": 1.3438, + "theoretical_loss": 3.5599060932910644, + "tokens_seen": 1305804800 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003052457372116349, + "loss": 1.3757, + "theoretical_loss": 3.559889919026535, + "tokens_seen": 1305870336 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003052357071213641, + "loss": 1.4018, + "theoretical_loss": 3.559873745800969, + "tokens_seen": 1305935872 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030522567703109325, + "loss": 1.3743, + "theoretical_loss": 3.559857573614248, + "tokens_seen": 1306001408 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003052156469408225, + "loss": 1.3979, + "theoretical_loss": 3.5598414024662537, + "tokens_seen": 1306066944 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030520561685055167, + "loss": 1.2773, + "theoretical_loss": 3.559825232356866, + "tokens_seen": 1306132480 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030519558676028085, + "loss": 1.2851, + "theoretical_loss": 3.559809063285967, + "tokens_seen": 1306198016 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030518555667001003, + "loss": 1.4405, + "theoretical_loss": 3.559792895253438, + "tokens_seen": 1306263552 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030517552657973926, + "loss": 1.3802, + "theoretical_loss": 3.5597767282591595, + "tokens_seen": 1306329088 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003051654964894684, + "loss": 1.3198, + "theoretical_loss": 3.5597605623030133, + "tokens_seen": 1306394624 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003051554663991976, + "loss": 1.3072, + "theoretical_loss": 3.559744397384881, + "tokens_seen": 1306460160 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030514543630892675, + "loss": 1.4113, + "theoretical_loss": 3.5597282335046425, + "tokens_seen": 1306525696 + }, + { + "epoch": 0.4, + "learning_rate": 0.000305135406218656, + "loss": 1.3587, + "theoretical_loss": 3.5597120706621803, + "tokens_seen": 1306591232 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030512537612838517, + "loss": 1.3927, + "theoretical_loss": 3.5596959088573756, + "tokens_seen": 1306656768 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030511534603811435, + "loss": 1.4511, + "theoretical_loss": 3.55967974809011, + "tokens_seen": 1306722304 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030510531594784353, + "loss": 1.3861, + "theoretical_loss": 3.559663588360264, + "tokens_seen": 1306787840 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003050952858575727, + "loss": 1.3712, + "theoretical_loss": 3.559647429667719, + "tokens_seen": 1306853376 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003050852557673019, + "loss": 1.4301, + "theoretical_loss": 3.559631272012357, + "tokens_seen": 1306918912 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030507522567703113, + "loss": 1.4059, + "theoretical_loss": 3.5596151153940596, + "tokens_seen": 1306984448 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030506519558676026, + "loss": 1.3285, + "theoretical_loss": 3.559598959812708, + "tokens_seen": 1307049984 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003050551654964895, + "loss": 1.3907, + "theoretical_loss": 3.5595828052681835, + "tokens_seen": 1307115520 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003050451354062186, + "loss": 1.4851, + "theoretical_loss": 3.5595666517603677, + "tokens_seen": 1307181056 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030503510531594785, + "loss": 1.4147, + "theoretical_loss": 3.5595504992891414, + "tokens_seen": 1307246592 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030502507522567703, + "loss": 1.4999, + "theoretical_loss": 3.5595343478543873, + "tokens_seen": 1307312128 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003050150451354062, + "loss": 1.4332, + "theoretical_loss": 3.559518197455986, + "tokens_seen": 1307377664 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": 0.000664729974232614, + "objective/train/docs_used": 740313, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4946823120117188, + "objective/train/original_loss": 2.494682550430298, + "objective/train/theoretical_loss": 3.5595020480938198, + "objective/train/tokens_used": 1327903200, + "objective/train/value_avg": -0.00582122802734375, + "objective/train/value_loss": 4.656162127503194e-05, + "objective/train/value_max": -0.00046372413635253906, + "objective/train/value_min": -0.05645751953125, + "objective/train/value_reward_corr": 0.4568269584084218, + "objective/train/value_std": 0.005279541015625, + "objective/train/weight_avg": 1.0089410543441772, + "objective/train/weighted_lm_loss": 2.522176742553711, + "objective/train/weights_max": 1.7278450727462769, + "objective/train/weights_min": 0.44418537616729736, + "theoretical_loss": 3.5595020480938198, + "tokens_seen": 1307443200 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003050050150451354, + "loss": 1.3185, + "theoretical_loss": 3.5595020480938198, + "tokens_seen": 1307443200 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030499498495486463, + "loss": 1.4578, + "theoretical_loss": 3.5594858997677696, + "tokens_seen": 1307508736 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030498495486459376, + "loss": 1.3727, + "theoretical_loss": 3.5594697524777175, + "tokens_seen": 1307574272 + }, + { + "epoch": 0.4, + "learning_rate": 0.000304974924774323, + "loss": 1.4217, + "theoretical_loss": 3.5594536062235447, + "tokens_seen": 1307639808 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003049648946840521, + "loss": 1.3637, + "theoretical_loss": 3.5594374610051336, + "tokens_seen": 1307705344 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030495486459378136, + "loss": 1.4198, + "theoretical_loss": 3.5594213168223647, + "tokens_seen": 1307770880 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030494483450351054, + "loss": 1.4142, + "theoretical_loss": 3.5594051736751204, + "tokens_seen": 1307836416 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003049348044132397, + "loss": 1.3682, + "theoretical_loss": 3.5593890315632826, + "tokens_seen": 1307901952 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003049247743229689, + "loss": 1.4644, + "theoretical_loss": 3.5593728904867326, + "tokens_seen": 1307967488 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003049147442326981, + "loss": 1.4543, + "theoretical_loss": 3.5593567504453523, + "tokens_seen": 1308033024 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030490471414242726, + "loss": 1.4296, + "theoretical_loss": 3.559340611439023, + "tokens_seen": 1308098560 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003048946840521565, + "loss": 1.3553, + "theoretical_loss": 3.559324473467627, + "tokens_seen": 1308164096 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003048846539618856, + "loss": 1.3701, + "theoretical_loss": 3.5593083365310463, + "tokens_seen": 1308229632 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030487462387161486, + "loss": 1.3849, + "theoretical_loss": 3.5592922006291623, + "tokens_seen": 1308295168 + }, + { + "epoch": 0.4, + "learning_rate": 0.000304864593781344, + "loss": 1.4344, + "theoretical_loss": 3.559276065761856, + "tokens_seen": 1308360704 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003048545636910732, + "loss": 1.4136, + "theoretical_loss": 3.5592599319290112, + "tokens_seen": 1308426240 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003048445336008024, + "loss": 1.265, + "theoretical_loss": 3.5592437991305084, + "tokens_seen": 1308491776 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003048345035105316, + "loss": 1.3862, + "theoretical_loss": 3.5592276673662298, + "tokens_seen": 1308557312 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030482447342026076, + "loss": 1.4009, + "theoretical_loss": 3.559211536636057, + "tokens_seen": 1308622848 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030481444332999, + "loss": 1.3737, + "theoretical_loss": 3.559195406939873, + "tokens_seen": 1308688384 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003048044132397192, + "loss": 1.3158, + "theoretical_loss": 3.5591792782775578, + "tokens_seen": 1308753920 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030479438314944836, + "loss": 1.3795, + "theoretical_loss": 3.5591631506489954, + "tokens_seen": 1308819456 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030478435305917754, + "loss": 1.512, + "theoretical_loss": 3.5591470240540666, + "tokens_seen": 1308884992 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003047743229689067, + "loss": 1.3613, + "theoretical_loss": 3.5591308984926537, + "tokens_seen": 1308950528 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030476429287863596, + "loss": 1.3935, + "theoretical_loss": 3.5591147739646387, + "tokens_seen": 1309016064 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.0006393615622073412, + "objective/train/docs_used": 741446, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4115633964538574, + "objective/train/original_loss": 2.4115636348724365, + "objective/train/theoretical_loss": 3.5590986504699043, + "objective/train/tokens_used": 1329541600, + "objective/train/value_avg": -0.005176544189453125, + "objective/train/value_loss": 4.5045220758765936e-05, + "objective/train/value_max": -0.00033020973205566406, + "objective/train/value_min": -0.06500244140625, + "objective/train/value_reward_corr": 0.4835156318680401, + "objective/train/value_std": 0.0045013427734375, + "objective/train/weight_avg": 0.9957542419433594, + "objective/train/weighted_lm_loss": 2.398571014404297, + "objective/train/weights_max": 1.9155875444412231, + "objective/train/weights_min": 0.4503137469291687, + "theoretical_loss": 3.5590986504699043, + "tokens_seen": 1309081600 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003047542627883651, + "loss": 1.2918, + "theoretical_loss": 3.5590986504699043, + "tokens_seen": 1309081600 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003047442326980943, + "loss": 1.4747, + "theoretical_loss": 3.5590825280083314, + "tokens_seen": 1309147136 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030473420260782345, + "loss": 1.4231, + "theoretical_loss": 3.5590664065798023, + "tokens_seen": 1309212672 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003047241725175527, + "loss": 1.4082, + "theoretical_loss": 3.5590502861842, + "tokens_seen": 1309278208 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030471414242728187, + "loss": 1.4223, + "theoretical_loss": 3.559034166821406, + "tokens_seen": 1309343744 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030470411233701105, + "loss": 1.3368, + "theoretical_loss": 3.5590180484913025, + "tokens_seen": 1309409280 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030469408224674023, + "loss": 1.254, + "theoretical_loss": 3.5590019311937713, + "tokens_seen": 1309474816 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030468405215646946, + "loss": 1.2061, + "theoretical_loss": 3.5589858149286955, + "tokens_seen": 1309540352 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003046740220661986, + "loss": 1.3528, + "theoretical_loss": 3.5589696996959566, + "tokens_seen": 1309605888 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003046639919759278, + "loss": 1.4064, + "theoretical_loss": 3.5589535854954364, + "tokens_seen": 1309671424 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030465396188565695, + "loss": 1.3437, + "theoretical_loss": 3.558937472327018, + "tokens_seen": 1309736960 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003046439317953862, + "loss": 1.3953, + "theoretical_loss": 3.558921360190584, + "tokens_seen": 1309802496 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030463390170511537, + "loss": 1.3654, + "theoretical_loss": 3.558905249086015, + "tokens_seen": 1309868032 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030462387161484455, + "loss": 1.349, + "theoretical_loss": 3.5588891390131945, + "tokens_seen": 1309933568 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030461384152457373, + "loss": 1.3567, + "theoretical_loss": 3.558873029972005, + "tokens_seen": 1309999104 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003046038114343029, + "loss": 1.4825, + "theoretical_loss": 3.5588569219623283, + "tokens_seen": 1310064640 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003045937813440321, + "loss": 1.4765, + "theoretical_loss": 3.5588408149840474, + "tokens_seen": 1310130176 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030458375125376133, + "loss": 1.295, + "theoretical_loss": 3.5588247090370437, + "tokens_seen": 1310195712 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030457372116349046, + "loss": 1.4092, + "theoretical_loss": 3.5588086041212, + "tokens_seen": 1310261248 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003045636910732197, + "loss": 1.3154, + "theoretical_loss": 3.558792500236399, + "tokens_seen": 1310326784 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003045536609829488, + "loss": 1.4242, + "theoretical_loss": 3.5587763973825224, + "tokens_seen": 1310392320 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030454363089267805, + "loss": 1.3146, + "theoretical_loss": 3.5587602955594537, + "tokens_seen": 1310457856 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030453360080240723, + "loss": 1.3116, + "theoretical_loss": 3.558744194767075, + "tokens_seen": 1310523392 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003045235707121364, + "loss": 1.3306, + "theoretical_loss": 3.558728095005268, + "tokens_seen": 1310588928 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003045135406218656, + "loss": 1.4295, + "theoretical_loss": 3.5587119962739164, + "tokens_seen": 1310654464 + }, + { + "debugging/Self-BLEU-5": 0.0033272052399666366, + "debugging/distinct-1-grams": 0.667789757412399, + "debugging/distinct-2-grams": 0.9952380952380953, + "debugging/entropy-1-grams": 4.425504236146162, + "debugging/entropy-2-grams": 5.021377173934136, + "debugging/length": 422.0, + "debugging/num_segments": 2, + "debugging/raw_token_scores_avg": 0.005191398318856955, + "debugging/raw_token_scores_std": 0.009463910013437271, + "debugging/score": 0.002307692307692308, + "debugging/score_std": 0.002307692307692308, + "epoch": 0.4, + "objective/train/advantage_avg": -0.0007081179646775126, + "objective/train/docs_used": 742066, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8782236576080322, + "objective/train/original_loss": 2.8782236576080322, + "objective/train/theoretical_loss": 3.5586958985729016, + "objective/train/tokens_used": 1331180000, + "objective/train/value_avg": -0.004482269287109375, + "objective/train/value_loss": 7.981698581716046e-05, + "objective/train/value_max": -0.0002892017364501953, + "objective/train/value_min": -0.080078125, + "objective/train/value_reward_corr": 0.34481037634643097, + "objective/train/value_std": 0.0038623809814453125, + "objective/train/weight_avg": 0.9964691400527954, + "objective/train/weighted_lm_loss": 2.866584300994873, + "objective/train/weights_max": 1.9930579662322998, + "objective/train/weights_min": 0.345547080039978, + "theoretical_loss": 3.5586958985729016, + "tokens_seen": 1310720000 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030450351053159483, + "loss": 1.4147, + "theoretical_loss": 3.5586958985729016, + "tokens_seen": 1310720000 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030449348044132396, + "loss": 1.4877, + "theoretical_loss": 3.558679801902107, + "tokens_seen": 1310785536 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003044834503510532, + "loss": 1.5067, + "theoretical_loss": 3.558663706261415, + "tokens_seen": 1310851072 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003044734202607823, + "loss": 1.3864, + "theoretical_loss": 3.558647611650708, + "tokens_seen": 1310916608 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030446339017051156, + "loss": 1.4204, + "theoretical_loss": 3.558631518069869, + "tokens_seen": 1310982144 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030445336008024074, + "loss": 1.3858, + "theoretical_loss": 3.5586154255187794, + "tokens_seen": 1311047680 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003044433299899699, + "loss": 1.3471, + "theoretical_loss": 3.558599333997323, + "tokens_seen": 1311113216 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003044332998996991, + "loss": 1.2869, + "theoretical_loss": 3.5585832435053826, + "tokens_seen": 1311178752 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003044232698094283, + "loss": 1.4605, + "theoretical_loss": 3.55856715404284, + "tokens_seen": 1311244288 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030441323971915746, + "loss": 1.3312, + "theoretical_loss": 3.5585510656095787, + "tokens_seen": 1311309824 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003044032096288867, + "loss": 1.3838, + "theoretical_loss": 3.558534978205481, + "tokens_seen": 1311375360 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003043931795386158, + "loss": 1.463, + "theoretical_loss": 3.5585188918304294, + "tokens_seen": 1311440896 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030438314944834506, + "loss": 1.3538, + "theoretical_loss": 3.558502806484307, + "tokens_seen": 1311506432 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003043731193580742, + "loss": 1.4261, + "theoretical_loss": 3.5584867221669967, + "tokens_seen": 1311571968 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003043630892678034, + "loss": 1.3512, + "theoretical_loss": 3.558470638878381, + "tokens_seen": 1311637504 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003043530591775326, + "loss": 1.3695, + "theoretical_loss": 3.5584545566183436, + "tokens_seen": 1311703040 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003043430290872618, + "loss": 1.2918, + "theoretical_loss": 3.558438475386766, + "tokens_seen": 1311768576 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030433299899699097, + "loss": 1.3771, + "theoretical_loss": 3.5584223951835314, + "tokens_seen": 1311834112 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003043229689067202, + "loss": 1.3835, + "theoretical_loss": 3.558406316008523, + "tokens_seen": 1311899648 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030431293881644933, + "loss": 1.3612, + "theoretical_loss": 3.558390237861624, + "tokens_seen": 1311965184 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030430290872617856, + "loss": 1.3707, + "theoretical_loss": 3.5583741607427166, + "tokens_seen": 1312030720 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003042928786359077, + "loss": 1.3472, + "theoretical_loss": 3.558358084651684, + "tokens_seen": 1312096256 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003042828485456369, + "loss": 1.3371, + "theoretical_loss": 3.5583420095884097, + "tokens_seen": 1312161792 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003042728184553661, + "loss": 1.4118, + "theoretical_loss": 3.5583259355527757, + "tokens_seen": 1312227328 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003042627883650953, + "loss": 1.3462, + "theoretical_loss": 3.5583098625446654, + "tokens_seen": 1312292864 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.0003217071352992207, + "objective/train/docs_used": 743478, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5826923847198486, + "objective/train/original_loss": 2.5826921463012695, + "objective/train/theoretical_loss": 3.558293790563962, + "objective/train/tokens_used": 1332818400, + "objective/train/value_avg": -0.004543304443359375, + "objective/train/value_loss": 6.081031824578531e-05, + "objective/train/value_max": -0.0002269744873046875, + "objective/train/value_min": -0.0753173828125, + "objective/train/value_reward_corr": 0.3111543998391164, + "objective/train/value_std": 0.004055023193359375, + "objective/train/weight_avg": 0.9996057748794556, + "objective/train/weighted_lm_loss": 2.5762381553649902, + "objective/train/weights_max": 2.0258796215057373, + "objective/train/weights_min": 0.34662967920303345, + "theoretical_loss": 3.558293790563962, + "tokens_seen": 1312358400 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030425275827482447, + "loss": 1.3142, + "theoretical_loss": 3.558293790563962, + "tokens_seen": 1312358400 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030424272818455365, + "loss": 1.3633, + "theoretical_loss": 3.5582777196105484, + "tokens_seen": 1312423936 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030423269809428283, + "loss": 1.466, + "theoretical_loss": 3.5582616496843076, + "tokens_seen": 1312489472 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030422266800401207, + "loss": 1.2967, + "theoretical_loss": 3.5582455807851225, + "tokens_seen": 1312555008 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003042126379137412, + "loss": 1.4122, + "theoretical_loss": 3.5582295129128765, + "tokens_seen": 1312620544 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030420260782347043, + "loss": 1.3757, + "theoretical_loss": 3.5582134460674526, + "tokens_seen": 1312686080 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003041925777331996, + "loss": 1.3552, + "theoretical_loss": 3.558197380248734, + "tokens_seen": 1312751616 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003041825476429288, + "loss": 1.37, + "theoretical_loss": 3.5581813154566038, + "tokens_seen": 1312817152 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030417251755265797, + "loss": 1.3651, + "theoretical_loss": 3.5581652516909448, + "tokens_seen": 1312882688 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030416248746238715, + "loss": 1.447, + "theoretical_loss": 3.558149188951641, + "tokens_seen": 1312948224 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030415245737211633, + "loss": 1.4364, + "theoretical_loss": 3.5581331272385746, + "tokens_seen": 1313013760 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030414242728184557, + "loss": 1.3693, + "theoretical_loss": 3.5581170665516293, + "tokens_seen": 1313079296 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003041323971915747, + "loss": 1.3337, + "theoretical_loss": 3.558101006890688, + "tokens_seen": 1313144832 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030412236710130393, + "loss": 1.3415, + "theoretical_loss": 3.558084948255635, + "tokens_seen": 1313210368 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030411233701103306, + "loss": 1.3518, + "theoretical_loss": 3.5580688906463527, + "tokens_seen": 1313275904 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003041023069207623, + "loss": 1.3166, + "theoretical_loss": 3.558052834062724, + "tokens_seen": 1313341440 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003040922768304915, + "loss": 1.3122, + "theoretical_loss": 3.558036778504633, + "tokens_seen": 1313406976 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030408224674022066, + "loss": 1.3955, + "theoretical_loss": 3.5580207239719632, + "tokens_seen": 1313472512 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030407221664994984, + "loss": 1.2327, + "theoretical_loss": 3.558004670464597, + "tokens_seen": 1313538048 + }, + { + "epoch": 0.4, + "learning_rate": 0.000304062186559679, + "loss": 1.3506, + "theoretical_loss": 3.557988617982419, + "tokens_seen": 1313603584 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030405215646940825, + "loss": 1.3818, + "theoretical_loss": 3.5579725665253115, + "tokens_seen": 1313669120 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030404212637913743, + "loss": 1.4188, + "theoretical_loss": 3.5579565160931583, + "tokens_seen": 1313734656 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003040320962888666, + "loss": 1.3195, + "theoretical_loss": 3.5579404666858427, + "tokens_seen": 1313800192 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003040220661985958, + "loss": 1.3175, + "theoretical_loss": 3.5579244183032483, + "tokens_seen": 1313865728 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030401203610832503, + "loss": 1.4535, + "theoretical_loss": 3.557908370945259, + "tokens_seen": 1313931264 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.0014573795488104224, + "objective/train/docs_used": 744202, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.940551280975342, + "objective/train/original_loss": 2.940551519393921, + "objective/train/theoretical_loss": 3.5578923246117578, + "objective/train/tokens_used": 1334456800, + "objective/train/value_avg": -0.005428314208984375, + "objective/train/value_loss": 7.938718044897541e-05, + "objective/train/value_max": -0.0002982616424560547, + "objective/train/value_min": -0.0784912109375, + "objective/train/value_reward_corr": 0.40089502916297315, + "objective/train/value_std": 0.004863739013671875, + "objective/train/weight_avg": 0.9891196489334106, + "objective/train/weighted_lm_loss": 2.9214961528778076, + "objective/train/weights_max": 1.6918874979019165, + "objective/train/weights_min": 0.3007955551147461, + "theoretical_loss": 3.5578923246117578, + "tokens_seen": 1313996800 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030400200601805416, + "loss": 1.4406, + "theoretical_loss": 3.5578923246117578, + "tokens_seen": 1313996800 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003039919759277834, + "loss": 1.4253, + "theoretical_loss": 3.5578762793026275, + "tokens_seen": 1314062336 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003039819458375125, + "loss": 1.4153, + "theoretical_loss": 3.5578602350177526, + "tokens_seen": 1314127872 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030397191574724176, + "loss": 1.2272, + "theoretical_loss": 3.5578441917570167, + "tokens_seen": 1314193408 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030396188565697094, + "loss": 1.4949, + "theoretical_loss": 3.557828149520303, + "tokens_seen": 1314258944 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003039518555667001, + "loss": 1.362, + "theoretical_loss": 3.557812108307495, + "tokens_seen": 1314324480 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003039418254764293, + "loss": 1.3278, + "theoretical_loss": 3.557796068118477, + "tokens_seen": 1314390016 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003039317953861585, + "loss": 1.4335, + "theoretical_loss": 3.557780028953131, + "tokens_seen": 1314455552 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030392176529588766, + "loss": 1.3308, + "theoretical_loss": 3.5577639908113428, + "tokens_seen": 1314521088 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003039117352056169, + "loss": 1.3805, + "theoretical_loss": 3.557747953692994, + "tokens_seen": 1314586624 + }, + { + "epoch": 0.4, + "learning_rate": 0.000303901705115346, + "loss": 1.3416, + "theoretical_loss": 3.5577319175979696, + "tokens_seen": 1314652160 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030389167502507526, + "loss": 1.3796, + "theoretical_loss": 3.5577158825261535, + "tokens_seen": 1314717696 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003038816449348044, + "loss": 1.3935, + "theoretical_loss": 3.5576998484774283, + "tokens_seen": 1314783232 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003038716148445336, + "loss": 1.3414, + "theoretical_loss": 3.5576838154516777, + "tokens_seen": 1314848768 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003038615847542628, + "loss": 1.4197, + "theoretical_loss": 3.557667783448787, + "tokens_seen": 1314914304 + }, + { + "epoch": 0.4, + "learning_rate": 0.000303851554663992, + "loss": 1.3766, + "theoretical_loss": 3.5576517524686384, + "tokens_seen": 1314979840 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030384152457372117, + "loss": 1.4164, + "theoretical_loss": 3.557635722511116, + "tokens_seen": 1315045376 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003038314944834504, + "loss": 1.4322, + "theoretical_loss": 3.557619693576105, + "tokens_seen": 1315110912 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030382146439317953, + "loss": 1.3558, + "theoretical_loss": 3.5576036656634873, + "tokens_seen": 1315176448 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030381143430290876, + "loss": 1.3277, + "theoretical_loss": 3.557587638773147, + "tokens_seen": 1315241984 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003038014042126379, + "loss": 1.3191, + "theoretical_loss": 3.5575716129049697, + "tokens_seen": 1315307520 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003037913741223671, + "loss": 1.3275, + "theoretical_loss": 3.5575555880588374, + "tokens_seen": 1315373056 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003037813440320963, + "loss": 1.439, + "theoretical_loss": 3.557539564234635, + "tokens_seen": 1315438592 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003037713139418255, + "loss": 1.403, + "theoretical_loss": 3.557523541432246, + "tokens_seen": 1315504128 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030376128385155467, + "loss": 1.4205, + "theoretical_loss": 3.5575075196515544, + "tokens_seen": 1315569664 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": 0.00043641714728437364, + "objective/train/docs_used": 745555, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1677615642547607, + "objective/train/original_loss": 3.16776180267334, + "objective/train/theoretical_loss": 3.5574914988924444, + "objective/train/tokens_used": 1336095200, + "objective/train/value_avg": -0.004604339599609375, + "objective/train/value_loss": 5.036233415012248e-05, + "objective/train/value_max": -0.000537872314453125, + "objective/train/value_min": -0.0472412109375, + "objective/train/value_reward_corr": 0.35422296066841136, + "objective/train/value_std": 0.003513336181640625, + "objective/train/weight_avg": 1.0067075490951538, + "objective/train/weighted_lm_loss": 3.189713716506958, + "objective/train/weights_max": 1.4916703701019287, + "objective/train/weights_min": 0.20572447776794434, + "theoretical_loss": 3.5574914988924444, + "tokens_seen": 1315635200 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030375125376128385, + "loss": 1.3987, + "theoretical_loss": 3.5574914988924444, + "tokens_seen": 1315635200 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030374122367101303, + "loss": 1.3577, + "theoretical_loss": 3.5574754791547996, + "tokens_seen": 1315700736 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030373119358074227, + "loss": 1.2277, + "theoretical_loss": 3.557459460438504, + "tokens_seen": 1315766272 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003037211634904714, + "loss": 1.4534, + "theoretical_loss": 3.557443442743442, + "tokens_seen": 1315831808 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030371113340020063, + "loss": 1.2924, + "theoretical_loss": 3.5574274260694976, + "tokens_seen": 1315897344 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003037011033099298, + "loss": 1.3426, + "theoretical_loss": 3.5574114104165546, + "tokens_seen": 1315962880 + }, + { + "epoch": 0.4, + "learning_rate": 0.000303691073219659, + "loss": 1.3682, + "theoretical_loss": 3.5573953957844973, + "tokens_seen": 1316028416 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030368104312938817, + "loss": 1.3038, + "theoretical_loss": 3.5573793821732096, + "tokens_seen": 1316093952 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030367101303911735, + "loss": 1.2695, + "theoretical_loss": 3.5573633695825757, + "tokens_seen": 1316159488 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030366098294884653, + "loss": 1.3974, + "theoretical_loss": 3.5573473580124797, + "tokens_seen": 1316225024 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030365095285857577, + "loss": 1.3598, + "theoretical_loss": 3.5573313474628057, + "tokens_seen": 1316290560 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003036409227683049, + "loss": 1.4874, + "theoretical_loss": 3.5573153379334377, + "tokens_seen": 1316356096 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030363089267803413, + "loss": 1.389, + "theoretical_loss": 3.5572993294242607, + "tokens_seen": 1316421632 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030362086258776326, + "loss": 1.3716, + "theoretical_loss": 3.557283321935158, + "tokens_seen": 1316487168 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003036108324974925, + "loss": 1.5188, + "theoretical_loss": 3.557267315466014, + "tokens_seen": 1316552704 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003036008024072217, + "loss": 1.2908, + "theoretical_loss": 3.5572513100167136, + "tokens_seen": 1316618240 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030359077231695086, + "loss": 1.2977, + "theoretical_loss": 3.55723530558714, + "tokens_seen": 1316683776 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030358074222668004, + "loss": 1.408, + "theoretical_loss": 3.5572193021771783, + "tokens_seen": 1316749312 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003035707121364092, + "loss": 1.4468, + "theoretical_loss": 3.5572032997867122, + "tokens_seen": 1316814848 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003035606820461384, + "loss": 1.4143, + "theoretical_loss": 3.5571872984156268, + "tokens_seen": 1316880384 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030355065195586763, + "loss": 1.398, + "theoretical_loss": 3.557171298063806, + "tokens_seen": 1316945920 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030354062186559676, + "loss": 1.2921, + "theoretical_loss": 3.557155298731134, + "tokens_seen": 1317011456 + }, + { + "epoch": 0.4, + "learning_rate": 0.000303530591775326, + "loss": 1.5, + "theoretical_loss": 3.5571393004174947, + "tokens_seen": 1317076992 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003035205616850552, + "loss": 1.3656, + "theoretical_loss": 3.557123303122774, + "tokens_seen": 1317142528 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030351053159478436, + "loss": 1.4093, + "theoretical_loss": 3.557107306846855, + "tokens_seen": 1317208064 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.0009422223083674908, + "objective/train/docs_used": 746236, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.747063159942627, + "objective/train/original_loss": 2.747063159942627, + "objective/train/theoretical_loss": 3.5570913115896228, + "objective/train/tokens_used": 1337733600, + "objective/train/value_avg": -0.006072998046875, + "objective/train/value_loss": 0.00010480949276825413, + "objective/train/value_max": -0.00021660327911376953, + "objective/train/value_min": -0.07159423828125, + "objective/train/value_reward_corr": 0.47972828634660947, + "objective/train/value_std": 0.005084991455078125, + "objective/train/weight_avg": 0.995235025882721, + "objective/train/weighted_lm_loss": 2.738213062286377, + "objective/train/weights_max": 1.7413350343704224, + "objective/train/weights_min": 0.1497088074684143, + "theoretical_loss": 3.5570913115896228, + "tokens_seen": 1317273600 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030350050150451354, + "loss": 1.3837, + "theoretical_loss": 3.5570913115896228, + "tokens_seen": 1317273600 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003034904714142427, + "loss": 1.3845, + "theoretical_loss": 3.5570753173509617, + "tokens_seen": 1317339136 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003034804413239719, + "loss": 1.3924, + "theoretical_loss": 3.5570593241307558, + "tokens_seen": 1317404672 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030347041123370114, + "loss": 1.4583, + "theoretical_loss": 3.55704333192889, + "tokens_seen": 1317470208 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030346038114343026, + "loss": 1.3888, + "theoretical_loss": 3.5570273407452486, + "tokens_seen": 1317535744 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003034503510531595, + "loss": 1.3361, + "theoretical_loss": 3.557011350579716, + "tokens_seen": 1317601280 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003034403209628886, + "loss": 1.5078, + "theoretical_loss": 3.556995361432177, + "tokens_seen": 1317666816 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030343029087261786, + "loss": 1.2932, + "theoretical_loss": 3.5569793733025166, + "tokens_seen": 1317732352 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030342026078234704, + "loss": 1.3929, + "theoretical_loss": 3.5569633861906187, + "tokens_seen": 1317797888 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003034102306920762, + "loss": 1.4096, + "theoretical_loss": 3.5569474000963677, + "tokens_seen": 1317863424 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003034002006018054, + "loss": 1.3674, + "theoretical_loss": 3.556931415019649, + "tokens_seen": 1317928960 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003033901705115346, + "loss": 1.4194, + "theoretical_loss": 3.5569154309603466, + "tokens_seen": 1317994496 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030338014042126377, + "loss": 1.4135, + "theoretical_loss": 3.5568994479183456, + "tokens_seen": 1318060032 + }, + { + "epoch": 0.4, + "learning_rate": 0.000303370110330993, + "loss": 1.3454, + "theoretical_loss": 3.556883465893531, + "tokens_seen": 1318125568 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030336008024072213, + "loss": 1.3183, + "theoretical_loss": 3.556867484885786, + "tokens_seen": 1318191104 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030335005015045137, + "loss": 1.3683, + "theoretical_loss": 3.556851504894997, + "tokens_seen": 1318256640 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030334002006018055, + "loss": 1.2883, + "theoretical_loss": 3.556835525921048, + "tokens_seen": 1318322176 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030332998996990973, + "loss": 1.3899, + "theoretical_loss": 3.5568195479638236, + "tokens_seen": 1318387712 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003033199598796389, + "loss": 1.2944, + "theoretical_loss": 3.5568035710232087, + "tokens_seen": 1318453248 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003033099297893681, + "loss": 1.3522, + "theoretical_loss": 3.556787595099088, + "tokens_seen": 1318518784 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003032998996990973, + "loss": 1.4393, + "theoretical_loss": 3.556771620191347, + "tokens_seen": 1318584320 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003032898696088265, + "loss": 1.3194, + "theoretical_loss": 3.55675564629987, + "tokens_seen": 1318649856 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003032798395185557, + "loss": 1.2049, + "theoretical_loss": 3.5567396734245413, + "tokens_seen": 1318715392 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030326980942828487, + "loss": 1.3925, + "theoretical_loss": 3.5567237015652466, + "tokens_seen": 1318780928 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030325977933801405, + "loss": 1.3514, + "theoretical_loss": 3.556707730721871, + "tokens_seen": 1318846464 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.001187819754704833, + "objective/train/docs_used": 747487, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8383235931396484, + "objective/train/original_loss": 2.8383238315582275, + "objective/train/theoretical_loss": 3.5566917608942976, + "objective/train/tokens_used": 1339372000, + "objective/train/value_avg": -0.005069732666015625, + "objective/train/value_loss": 5.6084700190695e-05, + "objective/train/value_max": -0.00020992755889892578, + "objective/train/value_min": -0.080078125, + "objective/train/value_reward_corr": 0.5205832420573283, + "objective/train/value_std": 0.004955291748046875, + "objective/train/weight_avg": 0.990775465965271, + "objective/train/weighted_lm_loss": 2.8180277347564697, + "objective/train/weights_max": 1.59145188331604, + "objective/train/weights_min": 0.46329575777053833, + "theoretical_loss": 3.5566917608942976, + "tokens_seen": 1318912000 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030324974924774323, + "loss": 1.2896, + "theoretical_loss": 3.5566917608942976, + "tokens_seen": 1318912000 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030323971915747247, + "loss": 1.4064, + "theoretical_loss": 3.5566757920824132, + "tokens_seen": 1318977536 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003032296890672016, + "loss": 1.4831, + "theoretical_loss": 3.5566598242861027, + "tokens_seen": 1319043072 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030321965897693083, + "loss": 1.4239, + "theoretical_loss": 3.55664385750525, + "tokens_seen": 1319108608 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030320962888666, + "loss": 1.3532, + "theoretical_loss": 3.5566278917397405, + "tokens_seen": 1319174144 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003031995987963892, + "loss": 1.4437, + "theoretical_loss": 3.5566119269894596, + "tokens_seen": 1319239680 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030318956870611837, + "loss": 1.4968, + "theoretical_loss": 3.556595963254292, + "tokens_seen": 1319305216 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030317953861584755, + "loss": 1.3309, + "theoretical_loss": 3.5565800005341224, + "tokens_seen": 1319370752 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030316950852557673, + "loss": 1.4204, + "theoretical_loss": 3.556564038828837, + "tokens_seen": 1319436288 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030315947843530597, + "loss": 1.3532, + "theoretical_loss": 3.556548078138319, + "tokens_seen": 1319501824 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003031494483450351, + "loss": 1.3603, + "theoretical_loss": 3.5565321184624548, + "tokens_seen": 1319567360 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030313941825476433, + "loss": 1.4286, + "theoretical_loss": 3.55651615980113, + "tokens_seen": 1319632896 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030312938816449346, + "loss": 1.2972, + "theoretical_loss": 3.5565002021542282, + "tokens_seen": 1319698432 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003031193580742227, + "loss": 1.2714, + "theoretical_loss": 3.5564842455216352, + "tokens_seen": 1319763968 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003031093279839519, + "loss": 1.459, + "theoretical_loss": 3.556468289903237, + "tokens_seen": 1319829504 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030309929789368106, + "loss": 1.358, + "theoretical_loss": 3.5564523352989177, + "tokens_seen": 1319895040 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030308926780341024, + "loss": 1.3678, + "theoretical_loss": 3.556436381708563, + "tokens_seen": 1319960576 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003030792377131394, + "loss": 1.3477, + "theoretical_loss": 3.556420429132058, + "tokens_seen": 1320026112 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003030692076228686, + "loss": 1.4388, + "theoretical_loss": 3.556404477569288, + "tokens_seen": 1320091648 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030305917753259783, + "loss": 1.2281, + "theoretical_loss": 3.556388527020138, + "tokens_seen": 1320157184 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030304914744232696, + "loss": 1.4519, + "theoretical_loss": 3.556372577484493, + "tokens_seen": 1320222720 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003030391173520562, + "loss": 1.4481, + "theoretical_loss": 3.5563566289622397, + "tokens_seen": 1320288256 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003030290872617854, + "loss": 1.2714, + "theoretical_loss": 3.556340681453262, + "tokens_seen": 1320353792 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030301905717151456, + "loss": 1.4403, + "theoretical_loss": 3.5563247349574456, + "tokens_seen": 1320419328 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030300902708124374, + "loss": 1.3826, + "theoretical_loss": 3.5563087894746763, + "tokens_seen": 1320484864 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.0009264076361432672, + "objective/train/docs_used": 748180, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9996256828308105, + "objective/train/original_loss": 2.9996259212493896, + "objective/train/theoretical_loss": 3.5562928450048386, + "objective/train/tokens_used": 1341010400, + "objective/train/value_avg": -0.004608154296875, + "objective/train/value_loss": 7.800839375704527e-05, + "objective/train/value_max": -0.00031757354736328125, + "objective/train/value_min": -0.0560302734375, + "objective/train/value_reward_corr": 0.26410990909893467, + "objective/train/value_std": 0.0033168792724609375, + "objective/train/weight_avg": 0.993853747844696, + "objective/train/weighted_lm_loss": 2.976853132247925, + "objective/train/weights_max": 1.7512025833129883, + "objective/train/weights_min": 0.13826273381710052, + "theoretical_loss": 3.5562928450048386, + "tokens_seen": 1320550400 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003029989969909729, + "loss": 1.4388, + "theoretical_loss": 3.5562928450048386, + "tokens_seen": 1320550400 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003029889669007021, + "loss": 1.3863, + "theoretical_loss": 3.556276901547819, + "tokens_seen": 1320615936 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030297893681043134, + "loss": 1.4812, + "theoretical_loss": 3.5562609591035024, + "tokens_seen": 1320681472 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030296890672016046, + "loss": 1.3792, + "theoretical_loss": 3.556245017671774, + "tokens_seen": 1320747008 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003029588766298897, + "loss": 1.3656, + "theoretical_loss": 3.556229077252519, + "tokens_seen": 1320812544 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003029488465396188, + "loss": 1.4763, + "theoretical_loss": 3.556213137845624, + "tokens_seen": 1320878080 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030293881644934806, + "loss": 1.4712, + "theoretical_loss": 3.556197199450973, + "tokens_seen": 1320943616 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030292878635907724, + "loss": 1.2802, + "theoretical_loss": 3.5561812620684528, + "tokens_seen": 1321009152 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003029187562688064, + "loss": 1.3282, + "theoretical_loss": 3.5561653256979477, + "tokens_seen": 1321074688 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003029087261785356, + "loss": 1.4061, + "theoretical_loss": 3.556149390339345, + "tokens_seen": 1321140224 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003028986960882648, + "loss": 1.4171, + "theoretical_loss": 3.556133455992528, + "tokens_seen": 1321205760 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030288866599799397, + "loss": 1.3824, + "theoretical_loss": 3.5561175226573845, + "tokens_seen": 1321271296 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003028786359077232, + "loss": 1.4281, + "theoretical_loss": 3.5561015903337987, + "tokens_seen": 1321336832 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030286860581745233, + "loss": 1.4635, + "theoretical_loss": 3.5560856590216563, + "tokens_seen": 1321402368 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030285857572718157, + "loss": 1.4891, + "theoretical_loss": 3.5560697287208427, + "tokens_seen": 1321467904 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030284854563691075, + "loss": 1.3503, + "theoretical_loss": 3.556053799431245, + "tokens_seen": 1321533440 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030283851554663993, + "loss": 1.3078, + "theoretical_loss": 3.556037871152747, + "tokens_seen": 1321598976 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003028284854563691, + "loss": 1.3652, + "theoretical_loss": 3.556021943885236, + "tokens_seen": 1321664512 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003028184553660983, + "loss": 1.4763, + "theoretical_loss": 3.556006017628597, + "tokens_seen": 1321730048 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030280842527582747, + "loss": 1.3465, + "theoretical_loss": 3.5559900923827152, + "tokens_seen": 1321795584 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003027983951855567, + "loss": 1.4385, + "theoretical_loss": 3.5559741681474772, + "tokens_seen": 1321861120 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030278836509528583, + "loss": 1.3515, + "theoretical_loss": 3.5559582449227687, + "tokens_seen": 1321926656 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030277833500501507, + "loss": 1.3083, + "theoretical_loss": 3.555942322708475, + "tokens_seen": 1321992192 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003027683049147442, + "loss": 1.3647, + "theoretical_loss": 3.5559264015044816, + "tokens_seen": 1322057728 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030275827482447343, + "loss": 1.3056, + "theoretical_loss": 3.5559104813106757, + "tokens_seen": 1322123264 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.00027307134587317705, + "objective/train/docs_used": 749233, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5339694023132324, + "objective/train/original_loss": 2.5339691638946533, + "objective/train/theoretical_loss": 3.5558945621269413, + "objective/train/tokens_used": 1342648800, + "objective/train/value_avg": -0.0048065185546875, + "objective/train/value_loss": 4.268142583896406e-05, + "objective/train/value_max": -0.0003712177276611328, + "objective/train/value_min": -0.090087890625, + "objective/train/value_reward_corr": 0.32577214934870263, + "objective/train/value_std": 0.003421783447265625, + "objective/train/weight_avg": 0.999281644821167, + "objective/train/weighted_lm_loss": 2.538689136505127, + "objective/train/weights_max": 1.9091185331344604, + "objective/train/weights_min": 0.37862154841423035, + "theoretical_loss": 3.5558945621269413, + "tokens_seen": 1322188800 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003027482447342026, + "loss": 1.3426, + "theoretical_loss": 3.5558945621269413, + "tokens_seen": 1322188800 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003027382146439318, + "loss": 1.3303, + "theoretical_loss": 3.5558786439531653, + "tokens_seen": 1322254336 + }, + { + "epoch": 0.4, + "learning_rate": 0.000302728184553661, + "loss": 1.3929, + "theoretical_loss": 3.5558627267892344, + "tokens_seen": 1322319872 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003027181544633902, + "loss": 1.4332, + "theoretical_loss": 3.5558468106350327, + "tokens_seen": 1322385408 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030270812437311934, + "loss": 1.3319, + "theoretical_loss": 3.5558308954904474, + "tokens_seen": 1322450944 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030269809428284857, + "loss": 1.3433, + "theoretical_loss": 3.555814981355364, + "tokens_seen": 1322516480 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003026880641925777, + "loss": 1.3957, + "theoretical_loss": 3.555799068229668, + "tokens_seen": 1322582016 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030267803410230693, + "loss": 1.3596, + "theoretical_loss": 3.5557831561132462, + "tokens_seen": 1322647552 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003026680040120361, + "loss": 1.3869, + "theoretical_loss": 3.5557672450059843, + "tokens_seen": 1322713088 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003026579739217653, + "loss": 1.346, + "theoretical_loss": 3.555751334907768, + "tokens_seen": 1322778624 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003026479438314945, + "loss": 1.3911, + "theoretical_loss": 3.555735425818484, + "tokens_seen": 1322844160 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030263791374122366, + "loss": 1.2592, + "theoretical_loss": 3.555719517738017, + "tokens_seen": 1322909696 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030262788365095284, + "loss": 1.2981, + "theoretical_loss": 3.5557036106662547, + "tokens_seen": 1322975232 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003026178535606821, + "loss": 1.2472, + "theoretical_loss": 3.555687704603082, + "tokens_seen": 1323040768 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003026078234704112, + "loss": 1.445, + "theoretical_loss": 3.5556717995483855, + "tokens_seen": 1323106304 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030259779338014044, + "loss": 1.3701, + "theoretical_loss": 3.555655895502051, + "tokens_seen": 1323171840 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030258776328986956, + "loss": 1.4609, + "theoretical_loss": 3.555639992463965, + "tokens_seen": 1323237376 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003025777331995988, + "loss": 1.3944, + "theoretical_loss": 3.555624090434014, + "tokens_seen": 1323302912 + }, + { + "epoch": 0.4, + "learning_rate": 0.000302567703109328, + "loss": 1.4055, + "theoretical_loss": 3.555608189412083, + "tokens_seen": 1323368448 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030255767301905716, + "loss": 1.367, + "theoretical_loss": 3.555592289398059, + "tokens_seen": 1323433984 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003025476429287864, + "loss": 1.3962, + "theoretical_loss": 3.555576390391828, + "tokens_seen": 1323499520 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003025376128385156, + "loss": 1.2552, + "theoretical_loss": 3.555560492393276, + "tokens_seen": 1323565056 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030252758274824476, + "loss": 1.3666, + "theoretical_loss": 3.5555445954022904, + "tokens_seen": 1323630592 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030251755265797394, + "loss": 1.4982, + "theoretical_loss": 3.555528699418756, + "tokens_seen": 1323696128 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003025075225677031, + "loss": 1.4602, + "theoretical_loss": 3.55551280444256, + "tokens_seen": 1323761664 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.0015765336574986577, + "objective/train/docs_used": 749806, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.638338327407837, + "objective/train/original_loss": 2.6383378505706787, + "objective/train/theoretical_loss": 3.555496910473588, + "objective/train/tokens_used": 1344287200, + "objective/train/value_avg": -0.00530242919921875, + "objective/train/value_loss": 7.007757085375488e-05, + "objective/train/value_max": -0.00019562244415283203, + "objective/train/value_min": -0.05108642578125, + "objective/train/value_reward_corr": 0.3707164272014695, + "objective/train/value_std": 0.004184722900390625, + "objective/train/weight_avg": 0.987445056438446, + "objective/train/weighted_lm_loss": 2.6152780055999756, + "objective/train/weights_max": 1.4155209064483643, + "objective/train/weights_min": 0.44777941703796387, + "theoretical_loss": 3.555496910473588, + "tokens_seen": 1323827200 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003024974924774323, + "loss": 1.3674, + "theoretical_loss": 3.555496910473588, + "tokens_seen": 1323827200 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030248746238716154, + "loss": 1.2529, + "theoretical_loss": 3.5554810175117266, + "tokens_seen": 1323892736 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030247743229689066, + "loss": 1.3812, + "theoretical_loss": 3.5554651255568626, + "tokens_seen": 1323958272 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003024674022066199, + "loss": 1.4177, + "theoretical_loss": 3.555449234608882, + "tokens_seen": 1324023808 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030245737211634903, + "loss": 1.3898, + "theoretical_loss": 3.5554333446676707, + "tokens_seen": 1324089344 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030244734202607826, + "loss": 1.318, + "theoretical_loss": 3.5554174557331155, + "tokens_seen": 1324154880 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030243731193580744, + "loss": 1.3368, + "theoretical_loss": 3.555401567805103, + "tokens_seen": 1324220416 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003024272818455366, + "loss": 1.2578, + "theoretical_loss": 3.5553856808835196, + "tokens_seen": 1324285952 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003024172517552658, + "loss": 1.2849, + "theoretical_loss": 3.555369794968252, + "tokens_seen": 1324351488 + }, + { + "epoch": 0.4, + "learning_rate": 0.000302407221664995, + "loss": 1.2586, + "theoretical_loss": 3.5553539100591856, + "tokens_seen": 1324417024 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030239719157472417, + "loss": 1.3649, + "theoretical_loss": 3.5553380261562078, + "tokens_seen": 1324482560 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003023871614844534, + "loss": 1.4137, + "theoretical_loss": 3.555322143259205, + "tokens_seen": 1324548096 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030237713139418253, + "loss": 1.3067, + "theoretical_loss": 3.5553062613680635, + "tokens_seen": 1324613632 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030236710130391177, + "loss": 1.3858, + "theoretical_loss": 3.5552903804826697, + "tokens_seen": 1324679168 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030235707121364095, + "loss": 1.3627, + "theoretical_loss": 3.555274500602911, + "tokens_seen": 1324744704 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030234704112337013, + "loss": 1.2489, + "theoretical_loss": 3.5552586217286724, + "tokens_seen": 1324810240 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003023370110330993, + "loss": 1.4912, + "theoretical_loss": 3.555242743859842, + "tokens_seen": 1324875776 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003023269809428285, + "loss": 1.2837, + "theoretical_loss": 3.555226866996306, + "tokens_seen": 1324941312 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030231695085255767, + "loss": 1.2656, + "theoretical_loss": 3.555210991137951, + "tokens_seen": 1325006848 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003023069207622869, + "loss": 1.5491, + "theoretical_loss": 3.555195116284663, + "tokens_seen": 1325072384 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030229689067201603, + "loss": 1.3935, + "theoretical_loss": 3.5551792424363295, + "tokens_seen": 1325137920 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030228686058174527, + "loss": 1.2689, + "theoretical_loss": 3.5551633695928366, + "tokens_seen": 1325203456 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003022768304914744, + "loss": 1.3355, + "theoretical_loss": 3.5551474977540716, + "tokens_seen": 1325268992 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030226680040120363, + "loss": 1.3504, + "theoretical_loss": 3.555131626919921, + "tokens_seen": 1325334528 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003022567703109328, + "loss": 1.4347, + "theoretical_loss": 3.555115757090271, + "tokens_seen": 1325400064 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.0010800380259752274, + "objective/train/docs_used": 750658, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8289244174957275, + "objective/train/original_loss": 2.8289244174957275, + "objective/train/theoretical_loss": 3.5550998882650093, + "objective/train/tokens_used": 1345925600, + "objective/train/value_avg": -0.005840301513671875, + "objective/train/value_loss": 0.00011416893539717421, + "objective/train/value_max": -0.0003845691680908203, + "objective/train/value_min": -0.07330322265625, + "objective/train/value_reward_corr": 0.4771336458334516, + "objective/train/value_std": 0.00548553466796875, + "objective/train/weight_avg": 0.9943448305130005, + "objective/train/weighted_lm_loss": 2.8205819129943848, + "objective/train/weights_max": 1.8885549306869507, + "objective/train/weights_min": 0.255083829164505, + "theoretical_loss": 3.5550998882650093, + "tokens_seen": 1325465600 + }, + { + "epoch": 0.4, + "learning_rate": 0.000302246740220662, + "loss": 1.4133, + "theoretical_loss": 3.5550998882650093, + "tokens_seen": 1325465600 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003022367101303912, + "loss": 1.4509, + "theoretical_loss": 3.5550840204440224, + "tokens_seen": 1325531136 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003022266800401204, + "loss": 1.4529, + "theoretical_loss": 3.555068153627196, + "tokens_seen": 1325596672 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030221664994984954, + "loss": 1.3233, + "theoretical_loss": 3.5550522878144184, + "tokens_seen": 1325662208 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030220661985957877, + "loss": 1.3919, + "theoretical_loss": 3.5550364230055758, + "tokens_seen": 1325727744 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003021965897693079, + "loss": 1.437, + "theoretical_loss": 3.555020559200555, + "tokens_seen": 1325793280 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030218655967903713, + "loss": 1.4401, + "theoretical_loss": 3.555004696399243, + "tokens_seen": 1325858816 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003021765295887663, + "loss": 1.4344, + "theoretical_loss": 3.5549888346015264, + "tokens_seen": 1325924352 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003021664994984955, + "loss": 1.3497, + "theoretical_loss": 3.554972973807293, + "tokens_seen": 1325989888 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003021564694082247, + "loss": 1.3552, + "theoretical_loss": 3.554957114016428, + "tokens_seen": 1326055424 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030214643931795386, + "loss": 1.2894, + "theoretical_loss": 3.5549412552288207, + "tokens_seen": 1326120960 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030213640922768304, + "loss": 1.2862, + "theoretical_loss": 3.5549253974443555, + "tokens_seen": 1326186496 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003021263791374123, + "loss": 1.3598, + "theoretical_loss": 3.554909540662922, + "tokens_seen": 1326252032 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003021163490471414, + "loss": 1.295, + "theoretical_loss": 3.554893684884405, + "tokens_seen": 1326317568 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030210631895687064, + "loss": 1.2681, + "theoretical_loss": 3.5548778301086923, + "tokens_seen": 1326383104 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030209628886659976, + "loss": 1.3112, + "theoretical_loss": 3.554861976335671, + "tokens_seen": 1326448640 + }, + { + "epoch": 0.4, + "learning_rate": 0.000302086258776329, + "loss": 1.3812, + "theoretical_loss": 3.5548461235652287, + "tokens_seen": 1326514176 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003020762286860582, + "loss": 1.4377, + "theoretical_loss": 3.5548302717972513, + "tokens_seen": 1326579712 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030206619859578736, + "loss": 1.3539, + "theoretical_loss": 3.554814421031627, + "tokens_seen": 1326645248 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030205616850551654, + "loss": 1.2736, + "theoretical_loss": 3.554798571268242, + "tokens_seen": 1326710784 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003020461384152458, + "loss": 1.5237, + "theoretical_loss": 3.554782722506984, + "tokens_seen": 1326776320 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003020361083249749, + "loss": 1.2865, + "theoretical_loss": 3.55476687474774, + "tokens_seen": 1326841856 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030202607823470414, + "loss": 1.3842, + "theoretical_loss": 3.554751027990397, + "tokens_seen": 1326907392 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030201604814443327, + "loss": 1.3897, + "theoretical_loss": 3.5547351822348423, + "tokens_seen": 1326972928 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003020060180541625, + "loss": 1.3742, + "theoretical_loss": 3.5547193374809636, + "tokens_seen": 1327038464 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.0004422456549946219, + "objective/train/docs_used": 751457, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4948196411132812, + "objective/train/original_loss": 2.494819402694702, + "objective/train/theoretical_loss": 3.5547034937286472, + "objective/train/tokens_used": 1347564000, + "objective/train/value_avg": -0.005695343017578125, + "objective/train/value_loss": 6.664610555162653e-05, + "objective/train/value_max": -0.00042557716369628906, + "objective/train/value_min": -0.059326171875, + "objective/train/value_reward_corr": 0.4660038660855782, + "objective/train/value_std": 0.0045318603515625, + "objective/train/weight_avg": 0.9984939694404602, + "objective/train/weighted_lm_loss": 2.4925010204315186, + "objective/train/weights_max": 1.7126208543777466, + "objective/train/weights_min": 0.08438848704099655, + "theoretical_loss": 3.5547034937286472, + "tokens_seen": 1327104000 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003019959879638917, + "loss": 1.3762, + "theoretical_loss": 3.5547034937286472, + "tokens_seen": 1327104000 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030198595787362087, + "loss": 1.3598, + "theoretical_loss": 3.554687650977781, + "tokens_seen": 1327169536 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030197592778335005, + "loss": 1.3065, + "theoretical_loss": 3.5546718092282514, + "tokens_seen": 1327235072 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030196589769307923, + "loss": 1.4567, + "theoretical_loss": 3.5546559684799472, + "tokens_seen": 1327300608 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003019558676028084, + "loss": 1.3669, + "theoretical_loss": 3.5546401287327543, + "tokens_seen": 1327366144 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030194583751253764, + "loss": 1.3358, + "theoretical_loss": 3.5546242899865605, + "tokens_seen": 1327431680 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030193580742226677, + "loss": 1.3422, + "theoretical_loss": 3.5546084522412533, + "tokens_seen": 1327497216 + }, + { + "epoch": 0.4, + "learning_rate": 0.000301925777331996, + "loss": 1.4215, + "theoretical_loss": 3.55459261549672, + "tokens_seen": 1327562752 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030191574724172513, + "loss": 1.3493, + "theoretical_loss": 3.554576779752848, + "tokens_seen": 1327628288 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030190571715145437, + "loss": 1.4076, + "theoretical_loss": 3.554560945009524, + "tokens_seen": 1327693824 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030189568706118355, + "loss": 1.4182, + "theoretical_loss": 3.554545111266636, + "tokens_seen": 1327759360 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030188565697091273, + "loss": 1.32, + "theoretical_loss": 3.5545292785240714, + "tokens_seen": 1327824896 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003018756268806419, + "loss": 1.2851, + "theoretical_loss": 3.5545134467817174, + "tokens_seen": 1327890432 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030186559679037115, + "loss": 1.4119, + "theoretical_loss": 3.554497616039462, + "tokens_seen": 1327955968 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003018555667001003, + "loss": 1.2971, + "theoretical_loss": 3.554481786297192, + "tokens_seen": 1328021504 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003018455366098295, + "loss": 1.3427, + "theoretical_loss": 3.5544659575547954, + "tokens_seen": 1328087040 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030183550651955864, + "loss": 1.3643, + "theoretical_loss": 3.5544501298121594, + "tokens_seen": 1328152576 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030182547642928787, + "loss": 1.4439, + "theoretical_loss": 3.5544343030691716, + "tokens_seen": 1328218112 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030181544633901705, + "loss": 1.2541, + "theoretical_loss": 3.5544184773257195, + "tokens_seen": 1328283648 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030180541624874623, + "loss": 1.3607, + "theoretical_loss": 3.5544026525816905, + "tokens_seen": 1328349184 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030179538615847547, + "loss": 1.4267, + "theoretical_loss": 3.5543868288369733, + "tokens_seen": 1328414720 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003017853560682046, + "loss": 1.3807, + "theoretical_loss": 3.5543710060914537, + "tokens_seen": 1328480256 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030177532597793383, + "loss": 1.3596, + "theoretical_loss": 3.5543551843450203, + "tokens_seen": 1328545792 + }, + { + "epoch": 0.4, + "learning_rate": 0.000301765295887663, + "loss": 1.4141, + "theoretical_loss": 3.5543393635975615, + "tokens_seen": 1328611328 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003017552657973922, + "loss": 1.3265, + "theoretical_loss": 3.5543235438489633, + "tokens_seen": 1328676864 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": -0.0003839919518213719, + "objective/train/docs_used": 752138, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0370640754699707, + "objective/train/original_loss": 3.0370638370513916, + "objective/train/theoretical_loss": 3.5543077250991146, + "objective/train/tokens_used": 1349202400, + "objective/train/value_avg": -0.005138397216796875, + "objective/train/value_loss": 5.1246563089080155e-05, + "objective/train/value_max": -0.0003654956817626953, + "objective/train/value_min": -0.061309814453125, + "objective/train/value_reward_corr": 0.3580939337139976, + "objective/train/value_std": 0.00408172607421875, + "objective/train/weight_avg": 0.998633861541748, + "objective/train/weighted_lm_loss": 3.044532537460327, + "objective/train/weights_max": 1.598183035850525, + "objective/train/weights_min": 0.6348018050193787, + "theoretical_loss": 3.5543077250991146, + "tokens_seen": 1328742400 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003017452357071214, + "loss": 1.4767, + "theoretical_loss": 3.5543077250991146, + "tokens_seen": 1328742400 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003017352056168506, + "loss": 1.3419, + "theoretical_loss": 3.5542919073479022, + "tokens_seen": 1328807936 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030172517552657974, + "loss": 1.3603, + "theoretical_loss": 3.5542760905952147, + "tokens_seen": 1328873472 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030171514543630897, + "loss": 1.3023, + "theoretical_loss": 3.5542602748409395, + "tokens_seen": 1328939008 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003017051153460381, + "loss": 1.3066, + "theoretical_loss": 3.5542444600849636, + "tokens_seen": 1329004544 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030169508525576733, + "loss": 1.4434, + "theoretical_loss": 3.554228646327176, + "tokens_seen": 1329070080 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003016850551654965, + "loss": 1.4042, + "theoretical_loss": 3.5542128335674636, + "tokens_seen": 1329135616 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003016750250752257, + "loss": 1.4004, + "theoretical_loss": 3.554197021805715, + "tokens_seen": 1329201152 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003016649949849549, + "loss": 1.3035, + "theoretical_loss": 3.5541812110418167, + "tokens_seen": 1329266688 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030165496489468406, + "loss": 1.3948, + "theoretical_loss": 3.5541654012756583, + "tokens_seen": 1329332224 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030164493480441324, + "loss": 1.3778, + "theoretical_loss": 3.554149592507126, + "tokens_seen": 1329397760 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003016349047141425, + "loss": 1.3365, + "theoretical_loss": 3.5541337847361087, + "tokens_seen": 1329463296 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003016248746238716, + "loss": 1.3395, + "theoretical_loss": 3.554117977962494, + "tokens_seen": 1329528832 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030161484453360084, + "loss": 1.1943, + "theoretical_loss": 3.5541021721861696, + "tokens_seen": 1329594368 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030160481444332996, + "loss": 1.2284, + "theoretical_loss": 3.5540863674070238, + "tokens_seen": 1329659904 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003015947843530592, + "loss": 1.4375, + "theoretical_loss": 3.5540705636249443, + "tokens_seen": 1329725440 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003015847542627884, + "loss": 1.4062, + "theoretical_loss": 3.554054760839819, + "tokens_seen": 1329790976 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030157472417251756, + "loss": 1.3906, + "theoretical_loss": 3.554038959051536, + "tokens_seen": 1329856512 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030156469408224674, + "loss": 1.3074, + "theoretical_loss": 3.5540231582599833, + "tokens_seen": 1329922048 + }, + { + "epoch": 0.4, + "learning_rate": 0.000301554663991976, + "loss": 1.2929, + "theoretical_loss": 3.554007358465049, + "tokens_seen": 1329987584 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003015446339017051, + "loss": 1.3866, + "theoretical_loss": 3.553991559666621, + "tokens_seen": 1330053120 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030153460381143434, + "loss": 1.3672, + "theoretical_loss": 3.5539757618645873, + "tokens_seen": 1330118656 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030152457372116347, + "loss": 1.352, + "theoretical_loss": 3.5539599650588363, + "tokens_seen": 1330184192 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003015145436308927, + "loss": 1.2546, + "theoretical_loss": 3.553944169249256, + "tokens_seen": 1330249728 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003015045135406219, + "loss": 1.4412, + "theoretical_loss": 3.553928374435734, + "tokens_seen": 1330315264 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": 0.0008486735168844461, + "objective/train/docs_used": 753163, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.606360912322998, + "objective/train/original_loss": 2.606361150741577, + "objective/train/theoretical_loss": 3.5539125806181584, + "objective/train/tokens_used": 1350840800, + "objective/train/value_avg": -0.00525665283203125, + "objective/train/value_loss": 3.4082077036146075e-05, + "objective/train/value_max": -0.0003726482391357422, + "objective/train/value_min": -0.0469970703125, + "objective/train/value_reward_corr": 0.38681435382209217, + "objective/train/value_std": 0.00408935546875, + "objective/train/weight_avg": 1.0101655721664429, + "objective/train/weighted_lm_loss": 2.6289734840393066, + "objective/train/weights_max": 1.364703893661499, + "objective/train/weights_min": 0.5625985264778137, + "theoretical_loss": 3.5539125806181584, + "tokens_seen": 1330380800 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030149448345035107, + "loss": 1.3596, + "theoretical_loss": 3.5539125806181584, + "tokens_seen": 1330380800 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030148445336008025, + "loss": 1.2808, + "theoretical_loss": 3.5538967877964183, + "tokens_seen": 1330446336 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030147442326980943, + "loss": 1.3415, + "theoretical_loss": 3.553880995970401, + "tokens_seen": 1330511872 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003014643931795386, + "loss": 1.414, + "theoretical_loss": 3.5538652051399957, + "tokens_seen": 1330577408 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030145436308926784, + "loss": 1.3396, + "theoretical_loss": 3.5538494153050895, + "tokens_seen": 1330642944 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030144433299899697, + "loss": 1.2189, + "theoretical_loss": 3.5538336264655705, + "tokens_seen": 1330708480 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003014343029087262, + "loss": 1.3501, + "theoretical_loss": 3.5538178386213275, + "tokens_seen": 1330774016 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030142427281845533, + "loss": 1.3274, + "theoretical_loss": 3.5538020517722493, + "tokens_seen": 1330839552 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030141424272818457, + "loss": 1.4661, + "theoretical_loss": 3.553786265918223, + "tokens_seen": 1330905088 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030140421263791375, + "loss": 1.393, + "theoretical_loss": 3.5537704810591375, + "tokens_seen": 1330970624 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030139418254764293, + "loss": 1.4407, + "theoretical_loss": 3.5537546971948815, + "tokens_seen": 1331036160 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003013841524573721, + "loss": 1.4094, + "theoretical_loss": 3.5537389143253426, + "tokens_seen": 1331101696 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030137412236710135, + "loss": 1.2492, + "theoretical_loss": 3.5537231324504095, + "tokens_seen": 1331167232 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003013640922768305, + "loss": 1.3893, + "theoretical_loss": 3.5537073515699698, + "tokens_seen": 1331232768 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003013540621865597, + "loss": 1.3746, + "theoretical_loss": 3.5536915716839133, + "tokens_seen": 1331298304 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030134403209628884, + "loss": 1.3574, + "theoretical_loss": 3.553675792792127, + "tokens_seen": 1331363840 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030133400200601807, + "loss": 1.3345, + "theoretical_loss": 3.5536600148945006, + "tokens_seen": 1331429376 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030132397191574725, + "loss": 1.4235, + "theoretical_loss": 3.5536442379909214, + "tokens_seen": 1331494912 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030131394182547643, + "loss": 1.4219, + "theoretical_loss": 3.553628462081279, + "tokens_seen": 1331560448 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003013039117352056, + "loss": 1.4036, + "theoretical_loss": 3.5536126871654607, + "tokens_seen": 1331625984 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003012938816449348, + "loss": 1.2907, + "theoretical_loss": 3.5535969132433554, + "tokens_seen": 1331691520 + }, + { + "epoch": 0.4, + "learning_rate": 0.000301283851554664, + "loss": 1.3883, + "theoretical_loss": 3.5535811403148516, + "tokens_seen": 1331757056 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003012738214643932, + "loss": 1.343, + "theoretical_loss": 3.5535653683798385, + "tokens_seen": 1331822592 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030126379137412234, + "loss": 1.3112, + "theoretical_loss": 3.5535495974382036, + "tokens_seen": 1331888128 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003012537612838516, + "loss": 1.3737, + "theoretical_loss": 3.553533827489836, + "tokens_seen": 1331953664 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": 4.511533916229382e-06, + "objective/train/docs_used": 753870, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.61496901512146, + "objective/train/original_loss": 2.614969491958618, + "objective/train/theoretical_loss": 3.553518058534624, + "objective/train/tokens_used": 1352479200, + "objective/train/value_avg": -0.004367828369140625, + "objective/train/value_loss": 4.0958355384645984e-05, + "objective/train/value_max": -0.00018668174743652344, + "objective/train/value_min": -0.1114501953125, + "objective/train/value_reward_corr": 0.42485649621074173, + "objective/train/value_std": 0.0036449432373046875, + "objective/train/weight_avg": 1.0019737482070923, + "objective/train/weighted_lm_loss": 2.6255698204040527, + "objective/train/weights_max": 2.8971877098083496, + "objective/train/weights_min": 0.3801526427268982, + "theoretical_loss": 3.553518058534624, + "tokens_seen": 1332019200 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003012437311935807, + "loss": 1.4103, + "theoretical_loss": 3.553518058534624, + "tokens_seen": 1332019200 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030123370110330994, + "loss": 1.2993, + "theoretical_loss": 3.5535022905724567, + "tokens_seen": 1332084736 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003012236710130391, + "loss": 1.3817, + "theoretical_loss": 3.5534865236032225, + "tokens_seen": 1332150272 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003012136409227683, + "loss": 1.4025, + "theoretical_loss": 3.5534707576268096, + "tokens_seen": 1332215808 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003012036108324975, + "loss": 1.3036, + "theoretical_loss": 3.553454992643107, + "tokens_seen": 1332281344 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003011935807422267, + "loss": 1.4999, + "theoretical_loss": 3.5534392286520036, + "tokens_seen": 1332346880 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030118355065195584, + "loss": 1.3477, + "theoretical_loss": 3.5534234656533874, + "tokens_seen": 1332412416 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003011735205616851, + "loss": 1.3615, + "theoretical_loss": 3.553407703647148, + "tokens_seen": 1332477952 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003011634904714142, + "loss": 1.3857, + "theoretical_loss": 3.5533919426331737, + "tokens_seen": 1332543488 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030115346038114344, + "loss": 1.426, + "theoretical_loss": 3.553376182611353, + "tokens_seen": 1332609024 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003011434302908726, + "loss": 1.4301, + "theoretical_loss": 3.553360423581575, + "tokens_seen": 1332674560 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003011334002006018, + "loss": 1.3315, + "theoretical_loss": 3.5533446655437277, + "tokens_seen": 1332740096 + }, + { + "epoch": 0.4, + "learning_rate": 0.000301123370110331, + "loss": 1.4032, + "theoretical_loss": 3.5533289084977007, + "tokens_seen": 1332805632 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030111334002006016, + "loss": 1.4398, + "theoretical_loss": 3.553313152443383, + "tokens_seen": 1332871168 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030110330992978935, + "loss": 1.4821, + "theoretical_loss": 3.5532973973806627, + "tokens_seen": 1332936704 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003010932798395186, + "loss": 1.4456, + "theoretical_loss": 3.553281643309429, + "tokens_seen": 1333002240 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003010832497492477, + "loss": 1.3914, + "theoretical_loss": 3.553265890229571, + "tokens_seen": 1333067776 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030107321965897694, + "loss": 1.3957, + "theoretical_loss": 3.553250138140977, + "tokens_seen": 1333133312 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030106318956870607, + "loss": 1.2726, + "theoretical_loss": 3.5532343870435366, + "tokens_seen": 1333198848 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003010531594784353, + "loss": 1.4647, + "theoretical_loss": 3.5532186369371384, + "tokens_seen": 1333264384 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030104312938816454, + "loss": 1.32, + "theoretical_loss": 3.553202887821671, + "tokens_seen": 1333329920 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030103309929789367, + "loss": 1.4354, + "theoretical_loss": 3.553187139697024, + "tokens_seen": 1333395456 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003010230692076229, + "loss": 1.3361, + "theoretical_loss": 3.5531713925630855, + "tokens_seen": 1333460992 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003010130391173521, + "loss": 1.4414, + "theoretical_loss": 3.5531556464197456, + "tokens_seen": 1333526528 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030100300902708127, + "loss": 1.2812, + "theoretical_loss": 3.553139901266892, + "tokens_seen": 1333592064 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": 0.0007176174549385905, + "objective/train/docs_used": 755212, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.976012706756592, + "objective/train/original_loss": 2.976012706756592, + "objective/train/theoretical_loss": 3.5531241571044148, + "objective/train/tokens_used": 1354117600, + "objective/train/value_avg": -0.00469207763671875, + "objective/train/value_loss": 4.1423365473747253e-05, + "objective/train/value_max": -0.0004839897155761719, + "objective/train/value_min": -0.054412841796875, + "objective/train/value_reward_corr": 0.36600345565447573, + "objective/train/value_std": 0.0032901763916015625, + "objective/train/weight_avg": 1.009182095527649, + "objective/train/weighted_lm_loss": 3.0027377605438232, + "objective/train/weights_max": 1.468602180480957, + "objective/train/weights_min": 0.6783351898193359, + "theoretical_loss": 3.5531241571044148, + "tokens_seen": 1333657600 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030099297893681045, + "loss": 1.3599, + "theoretical_loss": 3.5531241571044148, + "tokens_seen": 1333657600 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030098294884653963, + "loss": 1.3616, + "theoretical_loss": 3.5531084139322027, + "tokens_seen": 1333723136 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003009729187562688, + "loss": 1.4261, + "theoretical_loss": 3.5530926717501448, + "tokens_seen": 1333788672 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030096288866599804, + "loss": 1.1713, + "theoretical_loss": 3.55307693055813, + "tokens_seen": 1333854208 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030095285857572717, + "loss": 1.3651, + "theoretical_loss": 3.5530611903560474, + "tokens_seen": 1333919744 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003009428284854564, + "loss": 1.4353, + "theoretical_loss": 3.5530454511437863, + "tokens_seen": 1333985280 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030093279839518553, + "loss": 1.3609, + "theoretical_loss": 3.5530297129212354, + "tokens_seen": 1334050816 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030092276830491477, + "loss": 1.4384, + "theoretical_loss": 3.553013975688285, + "tokens_seen": 1334116352 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030091273821464395, + "loss": 1.4276, + "theoretical_loss": 3.5529982394448227, + "tokens_seen": 1334181888 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030090270812437313, + "loss": 1.4164, + "theoretical_loss": 3.552982504190739, + "tokens_seen": 1334247424 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003008926780341023, + "loss": 1.2641, + "theoretical_loss": 3.5529667699259226, + "tokens_seen": 1334312960 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030088264794383155, + "loss": 1.3754, + "theoretical_loss": 3.552951036650262, + "tokens_seen": 1334378496 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003008726178535607, + "loss": 1.368, + "theoretical_loss": 3.552935304363648, + "tokens_seen": 1334444032 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003008625877632899, + "loss": 1.3675, + "theoretical_loss": 3.552919573065968, + "tokens_seen": 1334509568 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030085255767301904, + "loss": 1.4171, + "theoretical_loss": 3.5529038427571127, + "tokens_seen": 1334575104 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030084252758274827, + "loss": 1.3293, + "theoretical_loss": 3.552888113436971, + "tokens_seen": 1334640640 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030083249749247745, + "loss": 1.4422, + "theoretical_loss": 3.5528723851054327, + "tokens_seen": 1334706176 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030082246740220663, + "loss": 1.3422, + "theoretical_loss": 3.552856657762386, + "tokens_seen": 1334771712 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003008124373119358, + "loss": 1.4364, + "theoretical_loss": 3.5528409314077205, + "tokens_seen": 1334837248 + }, + { + "epoch": 0.4, + "learning_rate": 0.000300802407221665, + "loss": 1.3235, + "theoretical_loss": 3.5528252060413266, + "tokens_seen": 1334902784 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003007923771313942, + "loss": 1.446, + "theoretical_loss": 3.5528094816630924, + "tokens_seen": 1334968320 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003007823470411234, + "loss": 1.3782, + "theoretical_loss": 3.5527937582729083, + "tokens_seen": 1335033856 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030077231695085254, + "loss": 1.3723, + "theoretical_loss": 3.552778035870663, + "tokens_seen": 1335099392 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003007622868605818, + "loss": 1.362, + "theoretical_loss": 3.5527623144562464, + "tokens_seen": 1335164928 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003007522567703109, + "loss": 1.3471, + "theoretical_loss": 3.5527465940295473, + "tokens_seen": 1335230464 + }, + { + "epoch": 0.4, + "objective/train/advantage_avg": 0.0010266726603731513, + "objective/train/docs_used": 755960, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6580464839935303, + "objective/train/original_loss": 2.6580464839935303, + "objective/train/theoretical_loss": 3.5527308745904564, + "objective/train/tokens_used": 1355756000, + "objective/train/value_avg": -0.005413055419921875, + "objective/train/value_loss": 3.276567076682113e-05, + "objective/train/value_max": -0.0003077983856201172, + "objective/train/value_min": -0.051544189453125, + "objective/train/value_reward_corr": 0.46888606802358346, + "objective/train/value_std": 0.00429534912109375, + "objective/train/weight_avg": 1.011886715888977, + "objective/train/weighted_lm_loss": 2.689018726348877, + "objective/train/weights_max": 1.5414742231369019, + "objective/train/weights_min": 0.5847765803337097, + "theoretical_loss": 3.5527308745904564, + "tokens_seen": 1335296000 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030074222668004014, + "loss": 1.4486, + "theoretical_loss": 3.5527308745904564, + "tokens_seen": 1335296000 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003007321965897693, + "loss": 1.3061, + "theoretical_loss": 3.552715156138862, + "tokens_seen": 1335361536 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003007221664994985, + "loss": 1.3082, + "theoretical_loss": 3.5526994386746544, + "tokens_seen": 1335427072 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003007121364092277, + "loss": 1.4217, + "theoretical_loss": 3.552683722197722, + "tokens_seen": 1335492608 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003007021063189569, + "loss": 1.3313, + "theoretical_loss": 3.5526680067079557, + "tokens_seen": 1335558144 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030069207622868604, + "loss": 1.3242, + "theoretical_loss": 3.552652292205244, + "tokens_seen": 1335623680 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003006820461384153, + "loss": 1.4032, + "theoretical_loss": 3.552636578689478, + "tokens_seen": 1335689216 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003006720160481444, + "loss": 1.3371, + "theoretical_loss": 3.552620866160545, + "tokens_seen": 1335754752 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030066198595787364, + "loss": 1.3033, + "theoretical_loss": 3.5526051546183366, + "tokens_seen": 1335820288 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003006519558676028, + "loss": 1.3275, + "theoretical_loss": 3.5525894440627415, + "tokens_seen": 1335885824 + }, + { + "epoch": 0.4, + "learning_rate": 0.000300641925777332, + "loss": 1.3762, + "theoretical_loss": 3.5525737344936497, + "tokens_seen": 1335951360 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003006318956870612, + "loss": 1.33, + "theoretical_loss": 3.55255802591095, + "tokens_seen": 1336016896 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030062186559679036, + "loss": 1.374, + "theoretical_loss": 3.5525423183145337, + "tokens_seen": 1336082432 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030061183550651955, + "loss": 1.4829, + "theoretical_loss": 3.5525266117042893, + "tokens_seen": 1336147968 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003006018054162488, + "loss": 1.3824, + "theoretical_loss": 3.5525109060801063, + "tokens_seen": 1336213504 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003005917753259779, + "loss": 1.2639, + "theoretical_loss": 3.552495201441875, + "tokens_seen": 1336279040 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030058174523570714, + "loss": 1.3888, + "theoretical_loss": 3.552479497789486, + "tokens_seen": 1336344576 + }, + { + "epoch": 0.4, + "learning_rate": 0.00030057171514543627, + "loss": 1.4471, + "theoretical_loss": 3.5524637951228275, + "tokens_seen": 1336410112 + }, + { + "epoch": 0.4, + "learning_rate": 0.0003005616850551655, + "loss": 1.3877, + "theoretical_loss": 3.55244809344179, + "tokens_seen": 1336475648 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003005516549648947, + "loss": 1.2634, + "theoretical_loss": 3.5524323927462635, + "tokens_seen": 1336541184 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030054162487462387, + "loss": 1.3691, + "theoretical_loss": 3.552416693036137, + "tokens_seen": 1336606720 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030053159478435305, + "loss": 1.4321, + "theoretical_loss": 3.5524009943113013, + "tokens_seen": 1336672256 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003005215646940823, + "loss": 1.439, + "theoretical_loss": 3.5523852965716456, + "tokens_seen": 1336737792 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003005115346038114, + "loss": 1.3874, + "theoretical_loss": 3.5523695998170606, + "tokens_seen": 1336803328 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030050150451354065, + "loss": 1.3476, + "theoretical_loss": 3.5523539040474352, + "tokens_seen": 1336868864 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": 0.0004674735537264496, + "objective/train/docs_used": 757325, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6668858528137207, + "objective/train/original_loss": 2.6668860912323, + "objective/train/theoretical_loss": 3.5523382092626603, + "objective/train/tokens_used": 1357394400, + "objective/train/value_avg": -0.00555419921875, + "objective/train/value_loss": 6.245707481866702e-05, + "objective/train/value_max": -0.0004513263702392578, + "objective/train/value_min": -0.0816650390625, + "objective/train/value_reward_corr": 0.30656844657104143, + "objective/train/value_std": 0.0042877197265625, + "objective/train/weight_avg": 1.007513165473938, + "objective/train/weighted_lm_loss": 2.6785733699798584, + "objective/train/weights_max": 2.262907028198242, + "objective/train/weights_min": 0.13907471299171448, + "theoretical_loss": 3.5523382092626603, + "tokens_seen": 1336934400 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003004914744232698, + "loss": 1.3414, + "theoretical_loss": 3.5523382092626603, + "tokens_seen": 1336934400 + }, + { + "epoch": 0.41, + "learning_rate": 0.000300481444332999, + "loss": 1.3154, + "theoretical_loss": 3.552322515462625, + "tokens_seen": 1336999936 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003004714142427282, + "loss": 1.4025, + "theoretical_loss": 3.5523068226472194, + "tokens_seen": 1337065472 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030046138415245737, + "loss": 1.3573, + "theoretical_loss": 3.552291130816334, + "tokens_seen": 1337131008 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030045135406218655, + "loss": 1.4554, + "theoretical_loss": 3.5522754399698586, + "tokens_seen": 1337196544 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030044132397191573, + "loss": 1.4069, + "theoretical_loss": 3.552259750107682, + "tokens_seen": 1337262080 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003004312938816449, + "loss": 1.2406, + "theoretical_loss": 3.5522440612296964, + "tokens_seen": 1337327616 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030042126379137415, + "loss": 1.3936, + "theoretical_loss": 3.5522283733357902, + "tokens_seen": 1337393152 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003004112337011033, + "loss": 1.3765, + "theoretical_loss": 3.552212686425854, + "tokens_seen": 1337458688 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003004012036108325, + "loss": 1.3455, + "theoretical_loss": 3.5521970004997776, + "tokens_seen": 1337524224 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003003911735205617, + "loss": 1.4167, + "theoretical_loss": 3.5521813155574513, + "tokens_seen": 1337589760 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003003811434302909, + "loss": 1.3154, + "theoretical_loss": 3.5521656315987657, + "tokens_seen": 1337655296 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030037111334002006, + "loss": 1.3764, + "theoretical_loss": 3.55214994862361, + "tokens_seen": 1337720832 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030036108324974924, + "loss": 1.4135, + "theoretical_loss": 3.5521342666318745, + "tokens_seen": 1337786368 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003003510531594784, + "loss": 1.3957, + "theoretical_loss": 3.5521185856234503, + "tokens_seen": 1337851904 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030034102306920765, + "loss": 1.4677, + "theoretical_loss": 3.5521029055982263, + "tokens_seen": 1337917440 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003003309929789368, + "loss": 1.3334, + "theoretical_loss": 3.552087226556094, + "tokens_seen": 1337982976 + }, + { + "epoch": 0.41, + "learning_rate": 0.000300320962888666, + "loss": 1.2939, + "theoretical_loss": 3.5520715484969427, + "tokens_seen": 1338048512 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030031093279839514, + "loss": 1.3444, + "theoretical_loss": 3.552055871420662, + "tokens_seen": 1338114048 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003003009027081244, + "loss": 1.3772, + "theoretical_loss": 3.5520401953271445, + "tokens_seen": 1338179584 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003002908726178536, + "loss": 1.1847, + "theoretical_loss": 3.552024520216278, + "tokens_seen": 1338245120 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030028084252758274, + "loss": 1.4214, + "theoretical_loss": 3.5520088460879538, + "tokens_seen": 1338310656 + }, + { + "epoch": 0.41, + "learning_rate": 0.000300270812437312, + "loss": 1.3339, + "theoretical_loss": 3.5519931729420624, + "tokens_seen": 1338376192 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003002607823470411, + "loss": 1.2546, + "theoretical_loss": 3.551977500778494, + "tokens_seen": 1338441728 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030025075225677034, + "loss": 1.4146, + "theoretical_loss": 3.551961829597139, + "tokens_seen": 1338507264 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.00010054895392386243, + "objective/train/docs_used": 758060, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4988319873809814, + "objective/train/original_loss": 2.498831272125244, + "objective/train/theoretical_loss": 3.551946159397887, + "objective/train/tokens_used": 1359032800, + "objective/train/value_avg": -0.00542449951171875, + "objective/train/value_loss": 5.592636080109514e-05, + "objective/train/value_max": -0.000415802001953125, + "objective/train/value_min": -0.07342529296875, + "objective/train/value_reward_corr": 0.3130929501404723, + "objective/train/value_std": 0.00406646728515625, + "objective/train/weight_avg": 1.001599669456482, + "objective/train/weighted_lm_loss": 2.500227212905884, + "objective/train/weights_max": 1.5750278234481812, + "objective/train/weights_min": 0.3779217004776001, + "theoretical_loss": 3.551946159397887, + "tokens_seen": 1338572800 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003002407221664995, + "loss": 1.3056, + "theoretical_loss": 3.551946159397887, + "tokens_seen": 1338572800 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003002306920762287, + "loss": 1.3062, + "theoretical_loss": 3.551930490180629, + "tokens_seen": 1338638336 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003002206619859579, + "loss": 1.3091, + "theoretical_loss": 3.551914821945256, + "tokens_seen": 1338703872 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003002106318956871, + "loss": 1.3481, + "theoretical_loss": 3.551899154691657, + "tokens_seen": 1338769408 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030020060180541624, + "loss": 1.3989, + "theoretical_loss": 3.5518834884197235, + "tokens_seen": 1338834944 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003001905717151455, + "loss": 1.4053, + "theoretical_loss": 3.5518678231293457, + "tokens_seen": 1338900480 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003001805416248746, + "loss": 1.3763, + "theoretical_loss": 3.551852158820414, + "tokens_seen": 1338966016 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030017051153460384, + "loss": 1.3236, + "theoretical_loss": 3.5518364954928185, + "tokens_seen": 1339031552 + }, + { + "epoch": 0.41, + "learning_rate": 0.000300160481444333, + "loss": 1.2386, + "theoretical_loss": 3.5518208331464507, + "tokens_seen": 1339097088 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003001504513540622, + "loss": 1.2833, + "theoretical_loss": 3.5518051717812, + "tokens_seen": 1339162624 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003001404212637914, + "loss": 1.3828, + "theoretical_loss": 3.5517895113969575, + "tokens_seen": 1339228160 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030013039117352056, + "loss": 1.4081, + "theoretical_loss": 3.5517738519936133, + "tokens_seen": 1339293696 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030012036108324975, + "loss": 1.4292, + "theoretical_loss": 3.551758193571059, + "tokens_seen": 1339359232 + }, + { + "epoch": 0.41, + "learning_rate": 0.000300110330992979, + "loss": 1.3793, + "theoretical_loss": 3.551742536129184, + "tokens_seen": 1339424768 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003001003009027081, + "loss": 1.3888, + "theoretical_loss": 3.55172687966788, + "tokens_seen": 1339490304 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030009027081243734, + "loss": 1.3484, + "theoretical_loss": 3.5517112241870366, + "tokens_seen": 1339555840 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030008024072216647, + "loss": 1.3182, + "theoretical_loss": 3.5516955696865447, + "tokens_seen": 1339621376 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003000702106318957, + "loss": 1.3862, + "theoretical_loss": 3.5516799161662957, + "tokens_seen": 1339686912 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003000601805416249, + "loss": 1.3766, + "theoretical_loss": 3.5516642636261793, + "tokens_seen": 1339752448 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030005015045135407, + "loss": 1.4152, + "theoretical_loss": 3.551648612066087, + "tokens_seen": 1339817984 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030004012036108325, + "loss": 1.4222, + "theoretical_loss": 3.5516329614859083, + "tokens_seen": 1339883520 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003000300902708125, + "loss": 1.3854, + "theoretical_loss": 3.551617311885535, + "tokens_seen": 1339949056 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003000200601805416, + "loss": 1.4004, + "theoretical_loss": 3.551601663264858, + "tokens_seen": 1340014592 + }, + { + "epoch": 0.41, + "learning_rate": 0.00030001003009027085, + "loss": 1.271, + "theoretical_loss": 3.551586015623767, + "tokens_seen": 1340080128 + }, + { + "epoch": 0.41, + "learning_rate": 0.0003, + "loss": 1.2828, + "theoretical_loss": 3.5515703689621536, + "tokens_seen": 1340145664 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.0005568142514675856, + "objective/train/docs_used": 759300, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8669803142547607, + "objective/train/original_loss": 2.8669798374176025, + "objective/train/theoretical_loss": 3.5515547232799087, + "objective/train/tokens_used": 1360671200, + "objective/train/value_avg": -0.005199432373046875, + "objective/train/value_loss": 7.415046275127679e-05, + "objective/train/value_max": -0.0004012584686279297, + "objective/train/value_min": -0.049224853515625, + "objective/train/value_reward_corr": 0.32926394871785214, + "objective/train/value_std": 0.004077911376953125, + "objective/train/weight_avg": 0.9978125095367432, + "objective/train/weighted_lm_loss": 2.8623857498168945, + "objective/train/weights_max": 1.464309573173523, + "objective/train/weights_min": 0.3393098711967468, + "theoretical_loss": 3.5515547232799087, + "tokens_seen": 1340211200 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002999899699097292, + "loss": 1.4203, + "theoretical_loss": 3.5515547232799087, + "tokens_seen": 1340211200 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002999799398194584, + "loss": 1.3392, + "theoretical_loss": 3.5515390785769227, + "tokens_seen": 1340276736 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029996990972918757, + "loss": 1.3446, + "theoretical_loss": 3.5515234348530864, + "tokens_seen": 1340342272 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029995987963891675, + "loss": 1.3133, + "theoretical_loss": 3.5515077921082905, + "tokens_seen": 1340407808 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029994984954864593, + "loss": 1.2153, + "theoretical_loss": 3.5514921503424266, + "tokens_seen": 1340473344 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002999398194583751, + "loss": 1.2858, + "theoretical_loss": 3.5514765095553846, + "tokens_seen": 1340538880 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029992978936810435, + "loss": 1.3886, + "theoretical_loss": 3.551460869747056, + "tokens_seen": 1340604416 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002999197592778335, + "loss": 1.2707, + "theoretical_loss": 3.5514452309173326, + "tokens_seen": 1340669952 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002999097291875627, + "loss": 1.3315, + "theoretical_loss": 3.5514295930661035, + "tokens_seen": 1340735488 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002998996990972919, + "loss": 1.3615, + "theoretical_loss": 3.5514139561932607, + "tokens_seen": 1340801024 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002998896690070211, + "loss": 1.3312, + "theoretical_loss": 3.551398320298695, + "tokens_seen": 1340866560 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029987963891675026, + "loss": 1.2451, + "theoretical_loss": 3.551382685382298, + "tokens_seen": 1340932096 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029986960882647944, + "loss": 1.2642, + "theoretical_loss": 3.5513670514439593, + "tokens_seen": 1340997632 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002998595787362086, + "loss": 1.3455, + "theoretical_loss": 3.551351418483571, + "tokens_seen": 1341063168 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029984954864593785, + "loss": 1.391, + "theoretical_loss": 3.5513357865010233, + "tokens_seen": 1341128704 + }, + { + "epoch": 0.41, + "learning_rate": 0.000299839518555667, + "loss": 1.305, + "theoretical_loss": 3.5513201554962084, + "tokens_seen": 1341194240 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002998294884653962, + "loss": 1.3278, + "theoretical_loss": 3.5513045254690168, + "tokens_seen": 1341259776 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029981945837512534, + "loss": 1.3453, + "theoretical_loss": 3.551288896419339, + "tokens_seen": 1341325312 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002998094282848546, + "loss": 1.1973, + "theoretical_loss": 3.5512732683470674, + "tokens_seen": 1341390848 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029979939819458376, + "loss": 1.4972, + "theoretical_loss": 3.551257641252092, + "tokens_seen": 1341456384 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029978936810431294, + "loss": 1.3207, + "theoretical_loss": 3.5512420151343047, + "tokens_seen": 1341521920 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002997793380140421, + "loss": 1.2927, + "theoretical_loss": 3.5512263899935963, + "tokens_seen": 1341587456 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002997693079237713, + "loss": 1.2748, + "theoretical_loss": 3.5512107658298575, + "tokens_seen": 1341652992 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002997592778335005, + "loss": 1.3699, + "theoretical_loss": 3.55119514264298, + "tokens_seen": 1341718528 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002997492477432297, + "loss": 1.4207, + "theoretical_loss": 3.5511795204328553, + "tokens_seen": 1341784064 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.0014922281261533499, + "objective/train/docs_used": 759950, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.440171718597412, + "objective/train/original_loss": 2.440171718597412, + "objective/train/theoretical_loss": 3.5511638991993744, + "objective/train/tokens_used": 1362309600, + "objective/train/value_avg": -0.005901336669921875, + "objective/train/value_loss": 6.811714411014691e-05, + "objective/train/value_max": -0.0004494190216064453, + "objective/train/value_min": -0.056854248046875, + "objective/train/value_reward_corr": 0.5716887793176229, + "objective/train/value_std": 0.004932403564453125, + "objective/train/weight_avg": 0.9882935285568237, + "objective/train/weighted_lm_loss": 2.4146528244018555, + "objective/train/weights_max": 1.4733549356460571, + "objective/train/weights_min": 0.31409311294555664, + "theoretical_loss": 3.5511638991993744, + "tokens_seen": 1341849600 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029973921765295885, + "loss": 1.3887, + "theoretical_loss": 3.5511638991993744, + "tokens_seen": 1341849600 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002997291875626881, + "loss": 1.2875, + "theoretical_loss": 3.5511482789424287, + "tokens_seen": 1341915136 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029971915747241726, + "loss": 1.2896, + "theoretical_loss": 3.551132659661909, + "tokens_seen": 1341980672 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029970912738214644, + "loss": 1.3665, + "theoretical_loss": 3.551117041357707, + "tokens_seen": 1342046208 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002996990972918756, + "loss": 1.315, + "theoretical_loss": 3.5511014240297136, + "tokens_seen": 1342111744 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002996890672016048, + "loss": 1.3488, + "theoretical_loss": 3.5510858076778202, + "tokens_seen": 1342177280 + }, + { + "epoch": 0.41, + "learning_rate": 0.000299679037111334, + "loss": 1.4576, + "theoretical_loss": 3.5510701923019186, + "tokens_seen": 1342242816 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002996690070210632, + "loss": 1.3194, + "theoretical_loss": 3.5510545779019, + "tokens_seen": 1342308352 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029965897693079235, + "loss": 1.3204, + "theoretical_loss": 3.551038964477655, + "tokens_seen": 1342373888 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002996489468405216, + "loss": 1.4048, + "theoretical_loss": 3.551023352029076, + "tokens_seen": 1342439424 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002996389167502507, + "loss": 1.4074, + "theoretical_loss": 3.551007740556054, + "tokens_seen": 1342504960 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029962888665997995, + "loss": 1.4636, + "theoretical_loss": 3.5509921300584804, + "tokens_seen": 1342570496 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029961885656970913, + "loss": 1.403, + "theoretical_loss": 3.5509765205362465, + "tokens_seen": 1342636032 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002996088264794383, + "loss": 1.3513, + "theoretical_loss": 3.5509609119892445, + "tokens_seen": 1342701568 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002995987963891675, + "loss": 1.4189, + "theoretical_loss": 3.5509453044173647, + "tokens_seen": 1342767104 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029958876629889667, + "loss": 1.3301, + "theoretical_loss": 3.5509296978204996, + "tokens_seen": 1342832640 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029957873620862585, + "loss": 1.3802, + "theoretical_loss": 3.55091409219854, + "tokens_seen": 1342898176 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002995687061183551, + "loss": 1.3008, + "theoretical_loss": 3.550898487551378, + "tokens_seen": 1342963712 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002995586760280842, + "loss": 1.2525, + "theoretical_loss": 3.550882883878905, + "tokens_seen": 1343029248 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029954864593781345, + "loss": 1.3318, + "theoretical_loss": 3.550867281181012, + "tokens_seen": 1343094784 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002995386158475427, + "loss": 1.3515, + "theoretical_loss": 3.5508516794575913, + "tokens_seen": 1343160320 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002995285857572718, + "loss": 1.3145, + "theoretical_loss": 3.5508360787085342, + "tokens_seen": 1343225856 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029951855566700105, + "loss": 1.2915, + "theoretical_loss": 3.5508204789337325, + "tokens_seen": 1343291392 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002995085255767302, + "loss": 1.3885, + "theoretical_loss": 3.550804880133077, + "tokens_seen": 1343356928 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002994984954864594, + "loss": 1.3969, + "theoretical_loss": 3.550789282306461, + "tokens_seen": 1343422464 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.0003260862722527236, + "objective/train/docs_used": 761102, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.760524034500122, + "objective/train/original_loss": 2.760523557662964, + "objective/train/theoretical_loss": 3.550773685453774, + "objective/train/tokens_used": 1363948000, + "objective/train/value_avg": -0.0053863525390625, + "objective/train/value_loss": 6.370035407599062e-05, + "objective/train/value_max": -0.0003514289855957031, + "objective/train/value_min": -0.06597900390625, + "objective/train/value_reward_corr": 0.41008612516462184, + "objective/train/value_std": 0.00469207763671875, + "objective/train/weight_avg": 0.9997183680534363, + "objective/train/weighted_lm_loss": 2.7683308124542236, + "objective/train/weights_max": 1.6583672761917114, + "objective/train/weights_min": 0.34770235419273376, + "theoretical_loss": 3.550773685453774, + "tokens_seen": 1343488000 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002994884653961886, + "loss": 1.3501, + "theoretical_loss": 3.550773685453774, + "tokens_seen": 1343488000 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029947843530591777, + "loss": 1.3835, + "theoretical_loss": 3.55075808957491, + "tokens_seen": 1343553536 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029946840521564695, + "loss": 1.2687, + "theoretical_loss": 3.550742494669759, + "tokens_seen": 1343619072 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029945837512537613, + "loss": 1.367, + "theoretical_loss": 3.550726900738213, + "tokens_seen": 1343684608 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002994483450351053, + "loss": 1.2236, + "theoretical_loss": 3.5507113077801646, + "tokens_seen": 1343750144 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029943831494483455, + "loss": 1.2687, + "theoretical_loss": 3.550695715795505, + "tokens_seen": 1343815680 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002994282848545637, + "loss": 1.3329, + "theoretical_loss": 3.5506801247841255, + "tokens_seen": 1343881216 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002994182547642929, + "loss": 1.2314, + "theoretical_loss": 3.5506645347459185, + "tokens_seen": 1343946752 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002994082246740221, + "loss": 1.3153, + "theoretical_loss": 3.5506489456807757, + "tokens_seen": 1344012288 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002993981945837513, + "loss": 1.3764, + "theoretical_loss": 3.550633357588589, + "tokens_seen": 1344077824 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029938816449348046, + "loss": 1.3248, + "theoretical_loss": 3.55061777046925, + "tokens_seen": 1344143360 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029937813440320964, + "loss": 1.2578, + "theoretical_loss": 3.55060218432265, + "tokens_seen": 1344208896 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002993681043129388, + "loss": 1.3444, + "theoretical_loss": 3.5505865991486827, + "tokens_seen": 1344274432 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029935807422266805, + "loss": 1.4165, + "theoretical_loss": 3.550571014947238, + "tokens_seen": 1344339968 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002993480441323972, + "loss": 1.4079, + "theoretical_loss": 3.550555431718209, + "tokens_seen": 1344405504 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002993380140421264, + "loss": 1.3934, + "theoretical_loss": 3.5505398494614866, + "tokens_seen": 1344471040 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029932798395185554, + "loss": 1.2795, + "theoretical_loss": 3.550524268176964, + "tokens_seen": 1344536576 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002993179538615848, + "loss": 1.4435, + "theoretical_loss": 3.550508687864532, + "tokens_seen": 1344602112 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029930792377131396, + "loss": 1.3829, + "theoretical_loss": 3.550493108524084, + "tokens_seen": 1344667648 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029929789368104314, + "loss": 1.3692, + "theoretical_loss": 3.5504775301555105, + "tokens_seen": 1344733184 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002992878635907723, + "loss": 1.31, + "theoretical_loss": 3.5504619527587042, + "tokens_seen": 1344798720 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002992778335005015, + "loss": 1.2573, + "theoretical_loss": 3.550446376333557, + "tokens_seen": 1344864256 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002992678034102307, + "loss": 1.2473, + "theoretical_loss": 3.5504308008799614, + "tokens_seen": 1344929792 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002992577733199599, + "loss": 1.3255, + "theoretical_loss": 3.550415226397809, + "tokens_seen": 1344995328 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029924774322968905, + "loss": 1.4004, + "theoretical_loss": 3.5503996528869917, + "tokens_seen": 1345060864 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.00034903112100437284, + "objective/train/docs_used": 761644, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.152174711227417, + "objective/train/original_loss": 2.152174472808838, + "objective/train/theoretical_loss": 3.550384080347402, + "objective/train/tokens_used": 1365586400, + "objective/train/value_avg": -0.0038928985595703125, + "objective/train/value_loss": 4.269912460586056e-05, + "objective/train/value_max": -0.00018525123596191406, + "objective/train/value_min": -0.1414794921875, + "objective/train/value_reward_corr": 0.1998924387061232, + "objective/train/value_std": 0.00304412841796875, + "objective/train/weight_avg": 0.9984341859817505, + "objective/train/weighted_lm_loss": 2.1607375144958496, + "objective/train/weights_max": 3.9198758602142334, + "objective/train/weights_min": 0.2940843999385834, + "theoretical_loss": 3.550384080347402, + "tokens_seen": 1345126400 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002992377131394183, + "loss": 1.2042, + "theoretical_loss": 3.550384080347402, + "tokens_seen": 1345126400 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029922768304914746, + "loss": 1.3188, + "theoretical_loss": 3.5503685087789316, + "tokens_seen": 1345191936 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029921765295887664, + "loss": 1.26, + "theoretical_loss": 3.550352938181473, + "tokens_seen": 1345257472 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002992076228686058, + "loss": 1.2999, + "theoretical_loss": 3.5503373685549184, + "tokens_seen": 1345323008 + }, + { + "epoch": 0.41, + "learning_rate": 0.000299197592778335, + "loss": 1.4079, + "theoretical_loss": 3.5503217998991596, + "tokens_seen": 1345388544 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002991875626880642, + "loss": 1.3301, + "theoretical_loss": 3.5503062322140897, + "tokens_seen": 1345454080 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002991775325977934, + "loss": 1.2609, + "theoretical_loss": 3.5502906654996, + "tokens_seen": 1345519616 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029916750250752255, + "loss": 1.3699, + "theoretical_loss": 3.550275099755582, + "tokens_seen": 1345585152 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002991574724172518, + "loss": 1.293, + "theoretical_loss": 3.5502595349819295, + "tokens_seen": 1345650688 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002991474423269809, + "loss": 1.3812, + "theoretical_loss": 3.550243971178534, + "tokens_seen": 1345716224 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029913741223671015, + "loss": 1.4181, + "theoretical_loss": 3.550228408345288, + "tokens_seen": 1345781760 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029912738214643933, + "loss": 1.255, + "theoretical_loss": 3.5502128464820837, + "tokens_seen": 1345847296 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002991173520561685, + "loss": 1.3188, + "theoretical_loss": 3.5501972855888138, + "tokens_seen": 1345912832 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002991073219658977, + "loss": 1.2394, + "theoretical_loss": 3.5501817256653694, + "tokens_seen": 1345978368 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029909729187562687, + "loss": 1.3718, + "theoretical_loss": 3.550166166711644, + "tokens_seen": 1346043904 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029908726178535605, + "loss": 1.3692, + "theoretical_loss": 3.550150608727529, + "tokens_seen": 1346109440 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002990772316950853, + "loss": 1.2607, + "theoretical_loss": 3.5501350517129184, + "tokens_seen": 1346174976 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002990672016048144, + "loss": 1.3272, + "theoretical_loss": 3.550119495667703, + "tokens_seen": 1346240512 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029905717151454365, + "loss": 1.2956, + "theoretical_loss": 3.550103940591776, + "tokens_seen": 1346306048 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029904714142427283, + "loss": 1.2966, + "theoretical_loss": 3.5500883864850294, + "tokens_seen": 1346371584 + }, + { + "epoch": 0.41, + "learning_rate": 0.000299037111334002, + "loss": 1.2675, + "theoretical_loss": 3.5500728333473557, + "tokens_seen": 1346437120 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002990270812437312, + "loss": 1.3445, + "theoretical_loss": 3.550057281178648, + "tokens_seen": 1346502656 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002990170511534604, + "loss": 1.3768, + "theoretical_loss": 3.5500417299787976, + "tokens_seen": 1346568192 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029900702106318956, + "loss": 1.3651, + "theoretical_loss": 3.5500261797476984, + "tokens_seen": 1346633728 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002989969909729188, + "loss": 1.187, + "theoretical_loss": 3.5500106304852412, + "tokens_seen": 1346699264 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": 4.373329284135252e-05, + "objective/train/docs_used": 762571, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0976967811584473, + "objective/train/original_loss": 3.0976967811584473, + "objective/train/theoretical_loss": 3.5499950821913204, + "objective/train/tokens_used": 1367224800, + "objective/train/value_avg": -0.005123138427734375, + "objective/train/value_loss": 6.048174691386521e-05, + "objective/train/value_max": -0.0002846717834472656, + "objective/train/value_min": -0.07977294921875, + "objective/train/value_reward_corr": 0.4287212445031494, + "objective/train/value_std": 0.00417327880859375, + "objective/train/weight_avg": 1.0032237768173218, + "objective/train/weighted_lm_loss": 3.1026620864868164, + "objective/train/weights_max": 1.9500584602355957, + "objective/train/weights_min": 0.2610293924808502, + "theoretical_loss": 3.5499950821913204, + "tokens_seen": 1346764800 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002989869608826479, + "loss": 1.37, + "theoretical_loss": 3.5499950821913204, + "tokens_seen": 1346764800 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029897693079237715, + "loss": 1.3282, + "theoretical_loss": 3.5499795348658276, + "tokens_seen": 1346830336 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002989669007021063, + "loss": 1.4475, + "theoretical_loss": 3.5499639885086554, + "tokens_seen": 1346895872 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002989568706118355, + "loss": 1.3038, + "theoretical_loss": 3.549948443119696, + "tokens_seen": 1346961408 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002989468405215647, + "loss": 1.2583, + "theoretical_loss": 3.5499328986988425, + "tokens_seen": 1347026944 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002989368104312939, + "loss": 1.4266, + "theoretical_loss": 3.549917355245987, + "tokens_seen": 1347092480 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029892678034102306, + "loss": 1.2664, + "theoretical_loss": 3.549901812761023, + "tokens_seen": 1347158016 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002989167502507523, + "loss": 1.3908, + "theoretical_loss": 3.5498862712438433, + "tokens_seen": 1347223552 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002989067201604814, + "loss": 1.352, + "theoretical_loss": 3.549870730694339, + "tokens_seen": 1347289088 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029889669007021066, + "loss": 1.4114, + "theoretical_loss": 3.549855191112404, + "tokens_seen": 1347354624 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002988866599799398, + "loss": 1.3339, + "theoretical_loss": 3.5498396524979308, + "tokens_seen": 1347420160 + }, + { + "epoch": 0.41, + "learning_rate": 0.000298876629889669, + "loss": 1.4385, + "theoretical_loss": 3.549824114850812, + "tokens_seen": 1347485696 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002988665997993982, + "loss": 1.4327, + "theoretical_loss": 3.5498085781709405, + "tokens_seen": 1347551232 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002988565697091274, + "loss": 1.3197, + "theoretical_loss": 3.5497930424582087, + "tokens_seen": 1347616768 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029884653961885656, + "loss": 1.3105, + "theoretical_loss": 3.54977750771251, + "tokens_seen": 1347682304 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029883650952858574, + "loss": 1.3234, + "theoretical_loss": 3.5497619739337365, + "tokens_seen": 1347747840 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002988264794383149, + "loss": 1.3336, + "theoretical_loss": 3.549746441121781, + "tokens_seen": 1347813376 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029881644934804416, + "loss": 1.4043, + "theoretical_loss": 3.549730909276537, + "tokens_seen": 1347878912 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002988064192577733, + "loss": 1.4597, + "theoretical_loss": 3.549715378397897, + "tokens_seen": 1347944448 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002987963891675025, + "loss": 1.4007, + "theoretical_loss": 3.549699848485753, + "tokens_seen": 1348009984 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002987863590772317, + "loss": 1.4234, + "theoretical_loss": 3.54968431954, + "tokens_seen": 1348075520 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002987763289869609, + "loss": 1.5243, + "theoretical_loss": 3.5496687915605287, + "tokens_seen": 1348141056 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002987662988966901, + "loss": 1.3729, + "theoretical_loss": 3.549653264547233, + "tokens_seen": 1348206592 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029875626880641925, + "loss": 1.4302, + "theoretical_loss": 3.5496377385000057, + "tokens_seen": 1348272128 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002987462387161485, + "loss": 1.3684, + "theoretical_loss": 3.54962221341874, + "tokens_seen": 1348337664 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.00031850836239755154, + "objective/train/docs_used": 763291, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.481006145477295, + "objective/train/original_loss": 2.481005907058716, + "objective/train/theoretical_loss": 3.549606689303328, + "objective/train/tokens_used": 1368863200, + "objective/train/value_avg": -0.005687713623046875, + "objective/train/value_loss": 7.049384294077754e-05, + "objective/train/value_max": -0.00032258033752441406, + "objective/train/value_min": -0.07318115234375, + "objective/train/value_reward_corr": 0.5312122594427167, + "objective/train/value_std": 0.005664825439453125, + "objective/train/weight_avg": 1.000009536743164, + "objective/train/weighted_lm_loss": 2.483508586883545, + "objective/train/weights_max": 1.6933753490447998, + "objective/train/weights_min": 0.32261791825294495, + "theoretical_loss": 3.549606689303328, + "tokens_seen": 1348403200 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029873620862587766, + "loss": 1.3289, + "theoretical_loss": 3.549606689303328, + "tokens_seen": 1348403200 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029872617853560684, + "loss": 1.3378, + "theoretical_loss": 3.5495911661536637, + "tokens_seen": 1348468736 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029872617853560684, + "loss": 1.3348, + "theoretical_loss": 3.5495756439696393, + "tokens_seen": 1348534272 + }, + { + "epoch": 0.41, + "learning_rate": 0.000298716148445336, + "loss": 1.2817, + "theoretical_loss": 3.5495601227511484, + "tokens_seen": 1348599808 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002987061183550652, + "loss": 1.3342, + "theoretical_loss": 3.5495446024980835, + "tokens_seen": 1348665344 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002986960882647944, + "loss": 1.4293, + "theoretical_loss": 3.5495290832103383, + "tokens_seen": 1348730880 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002986860581745236, + "loss": 1.427, + "theoretical_loss": 3.549513564887805, + "tokens_seen": 1348796416 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029867602808425275, + "loss": 1.4314, + "theoretical_loss": 3.5494980475303777, + "tokens_seen": 1348861952 + }, + { + "epoch": 0.41, + "learning_rate": 0.000298665997993982, + "loss": 1.2982, + "theoretical_loss": 3.549482531137949, + "tokens_seen": 1348927488 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002986559679037111, + "loss": 1.5014, + "theoretical_loss": 3.5494670157104116, + "tokens_seen": 1348993024 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029864593781344035, + "loss": 1.2911, + "theoretical_loss": 3.549451501247659, + "tokens_seen": 1349058560 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029863590772316953, + "loss": 1.3109, + "theoretical_loss": 3.5494359877495842, + "tokens_seen": 1349124096 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002986258776328987, + "loss": 1.4826, + "theoretical_loss": 3.5494204752160803, + "tokens_seen": 1349189632 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002986158475426279, + "loss": 1.3632, + "theoretical_loss": 3.5494049636470413, + "tokens_seen": 1349255168 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029860581745235707, + "loss": 1.2866, + "theoretical_loss": 3.5493894530423593, + "tokens_seen": 1349320704 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029859578736208625, + "loss": 1.4346, + "theoretical_loss": 3.5493739434019282, + "tokens_seen": 1349386240 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002985857572718155, + "loss": 1.2985, + "theoretical_loss": 3.5493584347256406, + "tokens_seen": 1349451776 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002985757271815446, + "loss": 1.3999, + "theoretical_loss": 3.5493429270133907, + "tokens_seen": 1349517312 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029856569709127385, + "loss": 1.4291, + "theoretical_loss": 3.5493274202650706, + "tokens_seen": 1349582848 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029855566700100303, + "loss": 1.3754, + "theoretical_loss": 3.5493119144805743, + "tokens_seen": 1349648384 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002985456369107322, + "loss": 1.3913, + "theoretical_loss": 3.5492964096597954, + "tokens_seen": 1349713920 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002985356068204614, + "loss": 1.3705, + "theoretical_loss": 3.5492809058026262, + "tokens_seen": 1349779456 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002985255767301906, + "loss": 1.464, + "theoretical_loss": 3.5492654029089614, + "tokens_seen": 1349844992 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029851554663991976, + "loss": 1.3954, + "theoretical_loss": 3.549249900978693, + "tokens_seen": 1349910528 + }, + { + "epoch": 0.41, + "learning_rate": 0.000298505516549649, + "loss": 1.3088, + "theoretical_loss": 3.5492344000117146, + "tokens_seen": 1349976064 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.000181753232027404, + "objective/train/docs_used": 764652, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.593167304992676, + "objective/train/original_loss": 2.5931668281555176, + "objective/train/theoretical_loss": 3.549218900007921, + "objective/train/tokens_used": 1370501600, + "objective/train/value_avg": -0.005420684814453125, + "objective/train/value_loss": 6.366815796354786e-05, + "objective/train/value_max": -0.0003077983856201172, + "objective/train/value_min": -0.06439208984375, + "objective/train/value_reward_corr": 0.5637575866605893, + "objective/train/value_std": 0.004886627197265625, + "objective/train/weight_avg": 1.0010827779769897, + "objective/train/weighted_lm_loss": 2.5902676582336426, + "objective/train/weights_max": 1.5700411796569824, + "objective/train/weights_min": 0.2942976653575897, + "theoretical_loss": 3.549218900007921, + "tokens_seen": 1350041600 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002984954864593781, + "loss": 1.2893, + "theoretical_loss": 3.549218900007921, + "tokens_seen": 1350041600 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029848545636910735, + "loss": 1.3435, + "theoretical_loss": 3.5492034009672038, + "tokens_seen": 1350107136 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002984754262788365, + "loss": 1.3606, + "theoretical_loss": 3.5491879028894573, + "tokens_seen": 1350172672 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002984653961885657, + "loss": 1.3256, + "theoretical_loss": 3.5491724057745744, + "tokens_seen": 1350238208 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002984553660982949, + "loss": 1.232, + "theoretical_loss": 3.5491569096224493, + "tokens_seen": 1350303744 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002984453360080241, + "loss": 1.3537, + "theoretical_loss": 3.5491414144329747, + "tokens_seen": 1350369280 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029843530591775326, + "loss": 1.4274, + "theoretical_loss": 3.549125920206045, + "tokens_seen": 1350434816 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002984252758274825, + "loss": 1.3853, + "theoretical_loss": 3.5491104269415525, + "tokens_seen": 1350500352 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002984152457372116, + "loss": 1.293, + "theoretical_loss": 3.549094934639392, + "tokens_seen": 1350565888 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029840521564694086, + "loss": 1.327, + "theoretical_loss": 3.549079443299456, + "tokens_seen": 1350631424 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029839518555667, + "loss": 1.3996, + "theoretical_loss": 3.549063952921639, + "tokens_seen": 1350696960 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002983851554663992, + "loss": 1.2823, + "theoretical_loss": 3.5490484635058337, + "tokens_seen": 1350762496 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002983751253761284, + "loss": 1.4305, + "theoretical_loss": 3.549032975051934, + "tokens_seen": 1350828032 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002983650952858576, + "loss": 1.4358, + "theoretical_loss": 3.549017487559833, + "tokens_seen": 1350893568 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029835506519558676, + "loss": 1.3581, + "theoretical_loss": 3.549002001029425, + "tokens_seen": 1350959104 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029834503510531594, + "loss": 1.277, + "theoretical_loss": 3.548986515460604, + "tokens_seen": 1351024640 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002983350050150451, + "loss": 1.4708, + "theoretical_loss": 3.548971030853263, + "tokens_seen": 1351090176 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029832497492477436, + "loss": 1.2263, + "theoretical_loss": 3.548955547207296, + "tokens_seen": 1351155712 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002983149448345035, + "loss": 1.3357, + "theoretical_loss": 3.548940064522596, + "tokens_seen": 1351221248 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002983049147442327, + "loss": 1.4148, + "theoretical_loss": 3.5489245827990574, + "tokens_seen": 1351286784 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029829488465396185, + "loss": 1.3351, + "theoretical_loss": 3.5489091020365735, + "tokens_seen": 1351352320 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002982848545636911, + "loss": 1.4274, + "theoretical_loss": 3.5488936222350382, + "tokens_seen": 1351417856 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029827482447342026, + "loss": 1.3483, + "theoretical_loss": 3.5488781433943455, + "tokens_seen": 1351483392 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029826479438314945, + "loss": 1.3125, + "theoretical_loss": 3.548862665514389, + "tokens_seen": 1351548928 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002982547642928786, + "loss": 1.4124, + "theoretical_loss": 3.5488471885950625, + "tokens_seen": 1351614464 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.0008193800458684564, + "objective/train/docs_used": 765372, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0008389949798584, + "objective/train/original_loss": 3.0008389949798584, + "objective/train/theoretical_loss": 3.5488317126362596, + "objective/train/tokens_used": 1372140000, + "objective/train/value_avg": -0.0045318603515625, + "objective/train/value_loss": 5.814000542159192e-05, + "objective/train/value_max": -0.00023055076599121094, + "objective/train/value_min": -0.079345703125, + "objective/train/value_reward_corr": 0.32282465402167027, + "objective/train/value_std": 0.00360107421875, + "objective/train/weight_avg": 0.9945103526115417, + "objective/train/weighted_lm_loss": 2.9876115322113037, + "objective/train/weights_max": 1.973523736000061, + "objective/train/weights_min": 0.37673419713974, + "theoretical_loss": 3.5488317126362596, + "tokens_seen": 1351680000 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029824473420260786, + "loss": 1.4155, + "theoretical_loss": 3.5488317126362596, + "tokens_seen": 1351680000 + }, + { + "epoch": 0.41, + "learning_rate": 0.000298234704112337, + "loss": 1.3004, + "theoretical_loss": 3.5488162376378742, + "tokens_seen": 1351745536 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002982246740220662, + "loss": 1.3137, + "theoretical_loss": 3.5488007635998007, + "tokens_seen": 1351811072 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029821464393179535, + "loss": 1.37, + "theoretical_loss": 3.5487852905219324, + "tokens_seen": 1351876608 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002982046138415246, + "loss": 1.4442, + "theoretical_loss": 3.5487698184041627, + "tokens_seen": 1351942144 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029819458375125377, + "loss": 1.4208, + "theoretical_loss": 3.548754347246386, + "tokens_seen": 1352007680 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029818455366098295, + "loss": 1.3022, + "theoretical_loss": 3.548738877048497, + "tokens_seen": 1352073216 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029817452357071213, + "loss": 1.336, + "theoretical_loss": 3.5487234078103884, + "tokens_seen": 1352138752 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002981644934804413, + "loss": 1.2597, + "theoretical_loss": 3.5487079395319547, + "tokens_seen": 1352204288 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002981544633901705, + "loss": 1.4184, + "theoretical_loss": 3.54869247221309, + "tokens_seen": 1352269824 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029814443329989973, + "loss": 1.3201, + "theoretical_loss": 3.5486770058536874, + "tokens_seen": 1352335360 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029813440320962885, + "loss": 1.4076, + "theoretical_loss": 3.548661540453642, + "tokens_seen": 1352400896 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002981243731193581, + "loss": 1.2857, + "theoretical_loss": 3.5486460760128473, + "tokens_seen": 1352466432 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002981143430290872, + "loss": 1.461, + "theoretical_loss": 3.548630612531197, + "tokens_seen": 1352531968 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029810431293881645, + "loss": 1.2371, + "theoretical_loss": 3.548615150008586, + "tokens_seen": 1352597504 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029809428284854563, + "loss": 1.398, + "theoretical_loss": 3.5485996884449076, + "tokens_seen": 1352663040 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002980842527582748, + "loss": 1.3428, + "theoretical_loss": 3.548584227840056, + "tokens_seen": 1352728576 + }, + { + "epoch": 0.41, + "learning_rate": 0.000298074222668004, + "loss": 1.3655, + "theoretical_loss": 3.5485687681939257, + "tokens_seen": 1352794112 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029806419257773323, + "loss": 1.3322, + "theoretical_loss": 3.5485533095064103, + "tokens_seen": 1352859648 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029805416248746236, + "loss": 1.3215, + "theoretical_loss": 3.548537851777404, + "tokens_seen": 1352925184 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002980441323971916, + "loss": 1.2846, + "theoretical_loss": 3.5485223950068017, + "tokens_seen": 1352990720 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002980341023069208, + "loss": 1.3703, + "theoretical_loss": 3.5485069391944966, + "tokens_seen": 1353056256 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029802407221664996, + "loss": 1.3495, + "theoretical_loss": 3.548491484340383, + "tokens_seen": 1353121792 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002980140421263792, + "loss": 1.3596, + "theoretical_loss": 3.5484760304443554, + "tokens_seen": 1353187328 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002980040120361083, + "loss": 1.3658, + "theoretical_loss": 3.5484605775063085, + "tokens_seen": 1353252864 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.00044691600487567484, + "objective/train/docs_used": 766365, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4681403636932373, + "objective/train/original_loss": 2.4681406021118164, + "objective/train/theoretical_loss": 3.5484451255261353, + "objective/train/tokens_used": 1373778400, + "objective/train/value_avg": -0.004604339599609375, + "objective/train/value_loss": 5.9141075325896963e-05, + "objective/train/value_max": -0.0001926422119140625, + "objective/train/value_min": -0.047515869140625, + "objective/train/value_reward_corr": 0.34273047373659327, + "objective/train/value_std": 0.0037136077880859375, + "objective/train/weight_avg": 0.9983102083206177, + "objective/train/weighted_lm_loss": 2.4722886085510254, + "objective/train/weights_max": 1.5028992891311646, + "objective/train/weights_min": 0.48496273159980774, + "theoretical_loss": 3.5484451255261353, + "tokens_seen": 1353318400 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029799398194583755, + "loss": 1.3338, + "theoretical_loss": 3.5484451255261353, + "tokens_seen": 1353318400 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002979839518555667, + "loss": 1.3126, + "theoretical_loss": 3.548429674503731, + "tokens_seen": 1353383936 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002979739217652959, + "loss": 1.3021, + "theoretical_loss": 3.54841422443899, + "tokens_seen": 1353449472 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002979638916750251, + "loss": 1.2737, + "theoretical_loss": 3.5483987753318056, + "tokens_seen": 1353515008 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002979538615847543, + "loss": 1.4448, + "theoretical_loss": 3.548383327182073, + "tokens_seen": 1353580544 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029794383149448346, + "loss": 1.3771, + "theoretical_loss": 3.5483678799896854, + "tokens_seen": 1353646080 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002979338014042127, + "loss": 1.4448, + "theoretical_loss": 3.5483524337545385, + "tokens_seen": 1353711616 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002979237713139418, + "loss": 1.3269, + "theoretical_loss": 3.5483369884765255, + "tokens_seen": 1353777152 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029791374122367106, + "loss": 1.4449, + "theoretical_loss": 3.5483215441555416, + "tokens_seen": 1353842688 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002979037111334002, + "loss": 1.3681, + "theoretical_loss": 3.548306100791481, + "tokens_seen": 1353908224 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002978936810431294, + "loss": 1.324, + "theoretical_loss": 3.548290658384238, + "tokens_seen": 1353973760 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002978836509528586, + "loss": 1.4542, + "theoretical_loss": 3.548275216933707, + "tokens_seen": 1354039296 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002978736208625878, + "loss": 1.3436, + "theoretical_loss": 3.5482597764397816, + "tokens_seen": 1354104832 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029786359077231696, + "loss": 1.3286, + "theoretical_loss": 3.548244336902358, + "tokens_seen": 1354170368 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029785356068204614, + "loss": 1.3103, + "theoretical_loss": 3.548228898321329, + "tokens_seen": 1354235904 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002978435305917753, + "loss": 1.2983, + "theoretical_loss": 3.54821346069659, + "tokens_seen": 1354301440 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029783350050150456, + "loss": 1.4922, + "theoretical_loss": 3.548198024028035, + "tokens_seen": 1354366976 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002978234704112337, + "loss": 1.281, + "theoretical_loss": 3.5481825883155595, + "tokens_seen": 1354432512 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002978134403209629, + "loss": 1.299, + "theoretical_loss": 3.5481671535590564, + "tokens_seen": 1354498048 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029780341023069205, + "loss": 1.312, + "theoretical_loss": 3.5481517197584216, + "tokens_seen": 1354563584 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002977933801404213, + "loss": 1.4931, + "theoretical_loss": 3.5481362869135493, + "tokens_seen": 1354629120 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029778335005015046, + "loss": 1.324, + "theoretical_loss": 3.548120855024334, + "tokens_seen": 1354694656 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029777331995987965, + "loss": 1.3001, + "theoretical_loss": 3.5481054240906698, + "tokens_seen": 1354760192 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029776328986960883, + "loss": 1.4141, + "theoretical_loss": 3.5480899941124524, + "tokens_seen": 1354825728 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029775325977933806, + "loss": 1.26, + "theoretical_loss": 3.5480745650895757, + "tokens_seen": 1354891264 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.0007971188751980662, + "objective/train/docs_used": 767078, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3615305423736572, + "objective/train/original_loss": 2.3615307807922363, + "objective/train/theoretical_loss": 3.5480591370219345, + "objective/train/tokens_used": 1375416800, + "objective/train/value_avg": -0.0047149658203125, + "objective/train/value_loss": 8.396933844778687e-05, + "objective/train/value_max": -0.00025725364685058594, + "objective/train/value_min": -0.0838623046875, + "objective/train/value_reward_corr": 0.29578602157015105, + "objective/train/value_std": 0.0037899017333984375, + "objective/train/weight_avg": 0.9957172870635986, + "objective/train/weighted_lm_loss": 2.345637798309326, + "objective/train/weights_max": 2.2311606407165527, + "objective/train/weights_min": 0.2496829330921173, + "theoretical_loss": 3.5480591370219345, + "tokens_seen": 1354956800 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002977432296890672, + "loss": 1.2418, + "theoretical_loss": 3.5480591370219345, + "tokens_seen": 1354956800 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002977331995987964, + "loss": 1.4091, + "theoretical_loss": 3.5480437099094226, + "tokens_seen": 1355022336 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029772316950852555, + "loss": 1.3386, + "theoretical_loss": 3.5480282837519366, + "tokens_seen": 1355087872 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002977131394182548, + "loss": 1.355, + "theoretical_loss": 3.5480128585493693, + "tokens_seen": 1355153408 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029770310932798397, + "loss": 1.4702, + "theoretical_loss": 3.5479974343016165, + "tokens_seen": 1355218944 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029769307923771315, + "loss": 1.2836, + "theoretical_loss": 3.547982011008573, + "tokens_seen": 1355284480 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029768304914744233, + "loss": 1.4159, + "theoretical_loss": 3.547966588670133, + "tokens_seen": 1355350016 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002976730190571715, + "loss": 1.4494, + "theoretical_loss": 3.5479511672861914, + "tokens_seen": 1355415552 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002976629889669007, + "loss": 1.257, + "theoretical_loss": 3.5479357468566426, + "tokens_seen": 1355481088 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029765295887662993, + "loss": 1.3174, + "theoretical_loss": 3.547920327381382, + "tokens_seen": 1355546624 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029764292878635905, + "loss": 1.3507, + "theoretical_loss": 3.5479049088603043, + "tokens_seen": 1355612160 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002976328986960883, + "loss": 1.3484, + "theoretical_loss": 3.5478894912933043, + "tokens_seen": 1355677696 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002976228686058174, + "loss": 1.2609, + "theoretical_loss": 3.5478740746802773, + "tokens_seen": 1355743232 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029761283851554665, + "loss": 1.3117, + "theoretical_loss": 3.547858659021117, + "tokens_seen": 1355808768 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029760280842527583, + "loss": 1.4292, + "theoretical_loss": 3.547843244315719, + "tokens_seen": 1355874304 + }, + { + "epoch": 0.41, + "learning_rate": 0.000297592778335005, + "loss": 1.3416, + "theoretical_loss": 3.5478278305639783, + "tokens_seen": 1355939840 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002975827482447342, + "loss": 1.4336, + "theoretical_loss": 3.54781241776579, + "tokens_seen": 1356005376 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029757271815446343, + "loss": 1.3018, + "theoretical_loss": 3.547797005921048, + "tokens_seen": 1356070912 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029756268806419256, + "loss": 1.2816, + "theoretical_loss": 3.5477815950296483, + "tokens_seen": 1356136448 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002975526579739218, + "loss": 1.1974, + "theoretical_loss": 3.5477661850914854, + "tokens_seen": 1356201984 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002975426278836509, + "loss": 1.2664, + "theoretical_loss": 3.5477507761064544, + "tokens_seen": 1356267520 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029753259779338016, + "loss": 1.2789, + "theoretical_loss": 3.5477353680744503, + "tokens_seen": 1356333056 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029752256770310934, + "loss": 1.2381, + "theoretical_loss": 3.5477199609953685, + "tokens_seen": 1356398592 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002975125376128385, + "loss": 1.363, + "theoretical_loss": 3.547704554869103, + "tokens_seen": 1356464128 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002975025075225677, + "loss": 1.3848, + "theoretical_loss": 3.5476891496955494, + "tokens_seen": 1356529664 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": 0.0006630669231526554, + "objective/train/docs_used": 768221, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9265618324279785, + "objective/train/original_loss": 2.9265615940093994, + "objective/train/theoretical_loss": 3.5476737454746035, + "objective/train/tokens_used": 1377055200, + "objective/train/value_avg": -0.005245208740234375, + "objective/train/value_loss": 3.9179871237138286e-05, + "objective/train/value_max": -0.00033545494079589844, + "objective/train/value_min": -0.06634521484375, + "objective/train/value_reward_corr": 0.5433427281208943, + "objective/train/value_std": 0.00511932373046875, + "objective/train/weight_avg": 1.0085656642913818, + "objective/train/weighted_lm_loss": 2.9470620155334473, + "objective/train/weights_max": 1.606262445449829, + "objective/train/weights_min": 0.6283432245254517, + "theoretical_loss": 3.5476737454746035, + "tokens_seen": 1356595200 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002974924774322969, + "loss": 1.3724, + "theoretical_loss": 3.5476737454746035, + "tokens_seen": 1356595200 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029748244734202606, + "loss": 1.418, + "theoretical_loss": 3.5476583422061587, + "tokens_seen": 1356660736 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002974724172517553, + "loss": 1.2463, + "theoretical_loss": 3.5476429398901117, + "tokens_seen": 1356726272 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002974623871614844, + "loss": 1.3695, + "theoretical_loss": 3.547627538526357, + "tokens_seen": 1356791808 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029745235707121366, + "loss": 1.318, + "theoretical_loss": 3.5476121381147894, + "tokens_seen": 1356857344 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002974423269809428, + "loss": 1.233, + "theoretical_loss": 3.5475967386553044, + "tokens_seen": 1356922880 + }, + { + "epoch": 0.41, + "learning_rate": 0.000297432296890672, + "loss": 1.3276, + "theoretical_loss": 3.5475813401477976, + "tokens_seen": 1356988416 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002974222668004012, + "loss": 1.4568, + "theoretical_loss": 3.547565942592163, + "tokens_seen": 1357053952 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002974122367101304, + "loss": 1.387, + "theoretical_loss": 3.5475505459882974, + "tokens_seen": 1357119488 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029740220661985956, + "loss": 1.42, + "theoretical_loss": 3.5475351503360946, + "tokens_seen": 1357185024 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002973921765295888, + "loss": 1.3127, + "theoretical_loss": 3.5475197556354505, + "tokens_seen": 1357250560 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002973821464393179, + "loss": 1.381, + "theoretical_loss": 3.5475043618862596, + "tokens_seen": 1357316096 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029737211634904716, + "loss": 1.361, + "theoretical_loss": 3.5474889690884184, + "tokens_seen": 1357381632 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002973620862587763, + "loss": 1.395, + "theoretical_loss": 3.5474735772418216, + "tokens_seen": 1357447168 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002973520561685055, + "loss": 1.3137, + "theoretical_loss": 3.547458186346364, + "tokens_seen": 1357512704 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002973420260782347, + "loss": 1.3432, + "theoretical_loss": 3.547442796401941, + "tokens_seen": 1357578240 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002973319959879639, + "loss": 1.3264, + "theoretical_loss": 3.547427407408449, + "tokens_seen": 1357643776 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029732196589769307, + "loss": 1.2803, + "theoretical_loss": 3.5474120193657823, + "tokens_seen": 1357709312 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029731193580742225, + "loss": 1.3925, + "theoretical_loss": 3.547396632273837, + "tokens_seen": 1357774848 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029730190571715143, + "loss": 1.3997, + "theoretical_loss": 3.547381246132507, + "tokens_seen": 1357840384 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029729187562688067, + "loss": 1.3146, + "theoretical_loss": 3.5473658609416896, + "tokens_seen": 1357905920 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029728184553660985, + "loss": 1.1528, + "theoretical_loss": 3.547350476701279, + "tokens_seen": 1357971456 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029727181544633903, + "loss": 1.3728, + "theoretical_loss": 3.547335093411171, + "tokens_seen": 1358036992 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029726178535606826, + "loss": 1.2428, + "theoretical_loss": 3.5473197110712613, + "tokens_seen": 1358102528 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002972517552657974, + "loss": 1.2863, + "theoretical_loss": 3.547304329681445, + "tokens_seen": 1358168064 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.0010514226742088795, + "objective/train/docs_used": 768734, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.448707342147827, + "objective/train/original_loss": 2.448707103729248, + "objective/train/theoretical_loss": 3.5472889492416173, + "objective/train/tokens_used": 1378693600, + "objective/train/value_avg": -0.0050506591796875, + "objective/train/value_loss": 8.569018973503262e-05, + "objective/train/value_max": -0.0002002716064453125, + "objective/train/value_min": -0.061981201171875, + "objective/train/value_reward_corr": 0.46999638509252123, + "objective/train/value_std": 0.004436492919921875, + "objective/train/weight_avg": 0.9933367371559143, + "objective/train/weighted_lm_loss": 2.4437451362609863, + "objective/train/weights_max": 1.5223071575164795, + "objective/train/weights_min": 0.3519727885723114, + "theoretical_loss": 3.5472889492416173, + "tokens_seen": 1358233600 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002972417251755266, + "loss": 1.3805, + "theoretical_loss": 3.5472889492416173, + "tokens_seen": 1358233600 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029723169508525575, + "loss": 1.3389, + "theoretical_loss": 3.547273569751674, + "tokens_seen": 1358299136 + }, + { + "epoch": 0.41, + "learning_rate": 0.000297221664994985, + "loss": 1.3107, + "theoretical_loss": 3.547258191211511, + "tokens_seen": 1358364672 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029721163490471417, + "loss": 1.3327, + "theoretical_loss": 3.5472428136210237, + "tokens_seen": 1358430208 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029720160481444335, + "loss": 1.4847, + "theoretical_loss": 3.5472274369801067, + "tokens_seen": 1358495744 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029719157472417253, + "loss": 1.4015, + "theoretical_loss": 3.547212061288657, + "tokens_seen": 1358561280 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002971815446339017, + "loss": 1.2898, + "theoretical_loss": 3.547196686546569, + "tokens_seen": 1358626816 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002971715145436309, + "loss": 1.3179, + "theoretical_loss": 3.5471813127537386, + "tokens_seen": 1358692352 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029716148445336013, + "loss": 1.2976, + "theoretical_loss": 3.547165939910062, + "tokens_seen": 1358757888 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029715145436308926, + "loss": 1.3326, + "theoretical_loss": 3.547150568015434, + "tokens_seen": 1358823424 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002971414242728185, + "loss": 1.2766, + "theoretical_loss": 3.547135197069751, + "tokens_seen": 1358888960 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002971313941825476, + "loss": 1.3402, + "theoretical_loss": 3.5471198270729083, + "tokens_seen": 1358954496 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029712136409227685, + "loss": 1.3048, + "theoretical_loss": 3.5471044580248012, + "tokens_seen": 1359020032 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029711133400200603, + "loss": 1.2822, + "theoretical_loss": 3.5470890899253256, + "tokens_seen": 1359085568 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002971013039117352, + "loss": 1.4123, + "theoretical_loss": 3.547073722774378, + "tokens_seen": 1359151104 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002970912738214644, + "loss": 1.3382, + "theoretical_loss": 3.547058356571853, + "tokens_seen": 1359216640 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029708124373119363, + "loss": 1.2993, + "theoretical_loss": 3.547042991317647, + "tokens_seen": 1359282176 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029707121364092276, + "loss": 1.3961, + "theoretical_loss": 3.547027627011656, + "tokens_seen": 1359347712 + }, + { + "epoch": 0.41, + "learning_rate": 0.000297061183550652, + "loss": 1.3964, + "theoretical_loss": 3.5470122636537744, + "tokens_seen": 1359413248 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002970511534603811, + "loss": 1.3643, + "theoretical_loss": 3.5469969012438995, + "tokens_seen": 1359478784 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029704112337011036, + "loss": 1.3483, + "theoretical_loss": 3.5469815397819264, + "tokens_seen": 1359544320 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029703109327983954, + "loss": 1.3249, + "theoretical_loss": 3.546966179267751, + "tokens_seen": 1359609856 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002970210631895687, + "loss": 1.3909, + "theoretical_loss": 3.5469508197012694, + "tokens_seen": 1359675392 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002970110330992979, + "loss": 1.4029, + "theoretical_loss": 3.5469354610823776, + "tokens_seen": 1359740928 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002970010030090271, + "loss": 1.439, + "theoretical_loss": 3.5469201034109705, + "tokens_seen": 1359806464 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.0011098915711045265, + "objective/train/docs_used": 769315, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.1347150802612305, + "objective/train/original_loss": 2.1347150802612305, + "objective/train/theoretical_loss": 3.5469047466869448, + "objective/train/tokens_used": 1380332000, + "objective/train/value_avg": -0.0042724609375, + "objective/train/value_loss": 8.74572288012132e-05, + "objective/train/value_max": -0.00022518634796142578, + "objective/train/value_min": -0.07879638671875, + "objective/train/value_reward_corr": 0.4100164855211496, + "objective/train/value_std": 0.003986358642578125, + "objective/train/weight_avg": 0.9925225377082825, + "objective/train/weighted_lm_loss": 2.131476640701294, + "objective/train/weights_max": 1.693010926246643, + "objective/train/weights_min": 0.15246368944644928, + "theoretical_loss": 3.5469047466869448, + "tokens_seen": 1359872000 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029699097291875626, + "loss": 1.2477, + "theoretical_loss": 3.5469047466869448, + "tokens_seen": 1359872000 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002969809428284855, + "loss": 1.3548, + "theoretical_loss": 3.546889390910197, + "tokens_seen": 1359937536 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002969709127382146, + "loss": 1.3759, + "theoretical_loss": 3.546874036080621, + "tokens_seen": 1360003072 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029696088264794386, + "loss": 1.3695, + "theoretical_loss": 3.5468586821981147, + "tokens_seen": 1360068608 + }, + { + "epoch": 0.41, + "learning_rate": 0.000296950852557673, + "loss": 1.346, + "theoretical_loss": 3.546843329262573, + "tokens_seen": 1360134144 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002969408224674022, + "loss": 1.2512, + "theoretical_loss": 3.5468279772738924, + "tokens_seen": 1360199680 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002969307923771314, + "loss": 1.3074, + "theoretical_loss": 3.5468126262319686, + "tokens_seen": 1360265216 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002969207622868606, + "loss": 1.5186, + "theoretical_loss": 3.546797276136698, + "tokens_seen": 1360330752 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029691073219658976, + "loss": 1.3139, + "theoretical_loss": 3.5467819269879763, + "tokens_seen": 1360396288 + }, + { + "epoch": 0.41, + "learning_rate": 0.000296900702106319, + "loss": 1.4751, + "theoretical_loss": 3.5467665787856997, + "tokens_seen": 1360461824 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002968906720160481, + "loss": 1.317, + "theoretical_loss": 3.546751231529764, + "tokens_seen": 1360527360 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029688064192577736, + "loss": 1.4375, + "theoretical_loss": 3.546735885220065, + "tokens_seen": 1360592896 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002968706118355065, + "loss": 1.4134, + "theoretical_loss": 3.5467205398564996, + "tokens_seen": 1360658432 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002968605817452357, + "loss": 1.3041, + "theoretical_loss": 3.546705195438963, + "tokens_seen": 1360723968 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002968505516549649, + "loss": 1.2163, + "theoretical_loss": 3.5466898519673524, + "tokens_seen": 1360789504 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002968405215646941, + "loss": 1.278, + "theoretical_loss": 3.546674509441563, + "tokens_seen": 1360855040 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029683049147442327, + "loss": 1.2802, + "theoretical_loss": 3.546659167861491, + "tokens_seen": 1360920576 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029682046138415245, + "loss": 1.2952, + "theoretical_loss": 3.5466438272270335, + "tokens_seen": 1360986112 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029681043129388163, + "loss": 1.3759, + "theoretical_loss": 3.5466284875380856, + "tokens_seen": 1361051648 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029680040120361087, + "loss": 1.3475, + "theoretical_loss": 3.546613148794544, + "tokens_seen": 1361117184 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029679037111334, + "loss": 1.4467, + "theoretical_loss": 3.546597810996305, + "tokens_seen": 1361182720 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029678034102306923, + "loss": 1.3182, + "theoretical_loss": 3.546582474143265, + "tokens_seen": 1361248256 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002967703109327984, + "loss": 1.4152, + "theoretical_loss": 3.546567138235319, + "tokens_seen": 1361313792 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002967602808425276, + "loss": 1.2509, + "theoretical_loss": 3.546551803272365, + "tokens_seen": 1361379328 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029675025075225677, + "loss": 1.388, + "theoretical_loss": 3.546536469254298, + "tokens_seen": 1361444864 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.0005003847763873637, + "objective/train/docs_used": 770761, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.726062297821045, + "objective/train/original_loss": 2.726062536239624, + "objective/train/theoretical_loss": 3.546521136181015, + "objective/train/tokens_used": 1381970400, + "objective/train/value_avg": -0.004913330078125, + "objective/train/value_loss": 7.576790812890977e-05, + "objective/train/value_max": -0.0002913475036621094, + "objective/train/value_min": -0.06939697265625, + "objective/train/value_reward_corr": 0.30710249272120954, + "objective/train/value_std": 0.004306793212890625, + "objective/train/weight_avg": 0.9981938600540161, + "objective/train/weighted_lm_loss": 2.738096237182617, + "objective/train/weights_max": 1.8648282289505005, + "objective/train/weights_min": 0.08690565824508667, + "theoretical_loss": 3.546521136181015, + "tokens_seen": 1361510400 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029674022066198595, + "loss": 1.4456, + "theoretical_loss": 3.546521136181015, + "tokens_seen": 1361510400 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029673019057171513, + "loss": 1.3812, + "theoretical_loss": 3.546505804052413, + "tokens_seen": 1361575936 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029672016048144437, + "loss": 1.3437, + "theoretical_loss": 3.546490472868386, + "tokens_seen": 1361641472 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002967101303911735, + "loss": 1.3627, + "theoretical_loss": 3.5464751426288323, + "tokens_seen": 1361707008 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029670010030090273, + "loss": 1.4393, + "theoretical_loss": 3.546459813333648, + "tokens_seen": 1361772544 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029669007021063186, + "loss": 1.2978, + "theoretical_loss": 3.5464444849827292, + "tokens_seen": 1361838080 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002966800401203611, + "loss": 1.312, + "theoretical_loss": 3.5464291575759725, + "tokens_seen": 1361903616 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002966700100300903, + "loss": 1.2967, + "theoretical_loss": 3.5464138311132736, + "tokens_seen": 1361969152 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029665997993981946, + "loss": 1.3373, + "theoretical_loss": 3.5463985055945297, + "tokens_seen": 1362034688 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029664994984954864, + "loss": 1.3429, + "theoretical_loss": 3.546383181019637, + "tokens_seen": 1362100224 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002966399197592778, + "loss": 1.357, + "theoretical_loss": 3.546367857388492, + "tokens_seen": 1362165760 + }, + { + "epoch": 0.41, + "learning_rate": 0.000296629889669007, + "loss": 1.3262, + "theoretical_loss": 3.546352534700991, + "tokens_seen": 1362231296 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029661985957873623, + "loss": 1.3182, + "theoretical_loss": 3.546337212957031, + "tokens_seen": 1362296832 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029660982948846536, + "loss": 1.3893, + "theoretical_loss": 3.546321892156508, + "tokens_seen": 1362362368 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002965997993981946, + "loss": 1.339, + "theoretical_loss": 3.5463065722993186, + "tokens_seen": 1362427904 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002965897693079238, + "loss": 1.3336, + "theoretical_loss": 3.54629125338536, + "tokens_seen": 1362493440 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029657973921765296, + "loss": 1.3741, + "theoretical_loss": 3.5462759354145277, + "tokens_seen": 1362558976 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029656970912738214, + "loss": 1.2715, + "theoretical_loss": 3.546260618386719, + "tokens_seen": 1362624512 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002965596790371113, + "loss": 1.3779, + "theoretical_loss": 3.5462453023018297, + "tokens_seen": 1362690048 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002965496489468405, + "loss": 1.2877, + "theoretical_loss": 3.546229987159758, + "tokens_seen": 1362755584 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029653961885656974, + "loss": 1.4043, + "theoretical_loss": 3.5462146729603985, + "tokens_seen": 1362821120 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002965295887662989, + "loss": 1.2729, + "theoretical_loss": 3.546199359703649, + "tokens_seen": 1362886656 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002965195586760281, + "loss": 1.3027, + "theoretical_loss": 3.5461840473894064, + "tokens_seen": 1362952192 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002965095285857573, + "loss": 1.4038, + "theoretical_loss": 3.546168736017567, + "tokens_seen": 1363017728 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029649949849548646, + "loss": 1.3019, + "theoretical_loss": 3.5461534255880274, + "tokens_seen": 1363083264 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.002163119148463011, + "objective/train/docs_used": 771430, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.716097354888916, + "objective/train/original_loss": 2.716097354888916, + "objective/train/theoretical_loss": 3.5461381161006846, + "objective/train/tokens_used": 1383608800, + "objective/train/value_avg": -0.00557708740234375, + "objective/train/value_loss": 0.00020285736536607146, + "objective/train/value_max": -0.0002453327178955078, + "objective/train/value_min": -0.0604248046875, + "objective/train/value_reward_corr": 0.5274279471126763, + "objective/train/value_std": 0.004364013671875, + "objective/train/weight_avg": 0.9867398142814636, + "objective/train/weighted_lm_loss": 2.6876442432403564, + "objective/train/weights_max": 1.6799733638763428, + "objective/train/weights_min": 0.39559224247932434, + "theoretical_loss": 3.5461381161006846, + "tokens_seen": 1363148800 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002964894684052157, + "loss": 1.3339, + "theoretical_loss": 3.5461381161006846, + "tokens_seen": 1363148800 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002964794383149448, + "loss": 1.3613, + "theoretical_loss": 3.5461228075554345, + "tokens_seen": 1363214336 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029646940822467406, + "loss": 1.3643, + "theoretical_loss": 3.546107499952175, + "tokens_seen": 1363279872 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002964593781344032, + "loss": 1.303, + "theoretical_loss": 3.546092193290802, + "tokens_seen": 1363345408 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002964493480441324, + "loss": 1.3424, + "theoretical_loss": 3.5460768875712128, + "tokens_seen": 1363410944 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002964393179538616, + "loss": 1.2712, + "theoretical_loss": 3.546061582793304, + "tokens_seen": 1363476480 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002964292878635908, + "loss": 1.3573, + "theoretical_loss": 3.5460462789569718, + "tokens_seen": 1363542016 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029641925777331996, + "loss": 1.2097, + "theoretical_loss": 3.5460309760621143, + "tokens_seen": 1363607552 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002964092276830492, + "loss": 1.314, + "theoretical_loss": 3.5460156741086273, + "tokens_seen": 1363673088 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002963991975927783, + "loss": 1.456, + "theoretical_loss": 3.5460003730964083, + "tokens_seen": 1363738624 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029638916750250756, + "loss": 1.427, + "theoretical_loss": 3.5459850730253537, + "tokens_seen": 1363804160 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002963791374122367, + "loss": 1.324, + "theoretical_loss": 3.54596977389536, + "tokens_seen": 1363869696 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002963691073219659, + "loss": 1.2951, + "theoretical_loss": 3.545954475706325, + "tokens_seen": 1363935232 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002963590772316951, + "loss": 1.3904, + "theoretical_loss": 3.5459391784581458, + "tokens_seen": 1364000768 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002963490471414243, + "loss": 1.262, + "theoretical_loss": 3.545923882150718, + "tokens_seen": 1364066304 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029633901705115347, + "loss": 1.2983, + "theoretical_loss": 3.54590858678394, + "tokens_seen": 1364131840 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029632898696088265, + "loss": 1.4654, + "theoretical_loss": 3.5458932923577082, + "tokens_seen": 1364197376 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029631895687061183, + "loss": 1.2949, + "theoretical_loss": 3.545877998871919, + "tokens_seen": 1364262912 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029630892678034107, + "loss": 1.4016, + "theoretical_loss": 3.54586270632647, + "tokens_seen": 1364328448 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002962988966900702, + "loss": 1.2327, + "theoretical_loss": 3.5458474147212584, + "tokens_seen": 1364393984 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029628886659979943, + "loss": 1.32, + "theoretical_loss": 3.5458321240561803, + "tokens_seen": 1364459520 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002962788365095286, + "loss": 1.2955, + "theoretical_loss": 3.5458168343311343, + "tokens_seen": 1364525056 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002962688064192578, + "loss": 1.3897, + "theoretical_loss": 3.5458015455460155, + "tokens_seen": 1364590592 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029625877632898697, + "loss": 1.4539, + "theoretical_loss": 3.545786257700723, + "tokens_seen": 1364656128 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029624874623871615, + "loss": 1.3219, + "theoretical_loss": 3.545770970795153, + "tokens_seen": 1364721664 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.0008425674168393016, + "objective/train/docs_used": 772731, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5964303016662598, + "objective/train/original_loss": 2.596430540084839, + "objective/train/theoretical_loss": 3.545755684829202, + "objective/train/tokens_used": 1385247200, + "objective/train/value_avg": -0.00569915771484375, + "objective/train/value_loss": 8.818400965537876e-05, + "objective/train/value_max": -0.000507354736328125, + "objective/train/value_min": -0.09588623046875, + "objective/train/value_reward_corr": 0.5412026917795998, + "objective/train/value_std": 0.005619049072265625, + "objective/train/weight_avg": 0.9954269528388977, + "objective/train/weighted_lm_loss": 2.5811216831207275, + "objective/train/weights_max": 2.42079496383667, + "objective/train/weights_min": 0.20951010286808014, + "theoretical_loss": 3.545755684829202, + "tokens_seen": 1364787200 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029623871614844533, + "loss": 1.3536, + "theoretical_loss": 3.545755684829202, + "tokens_seen": 1364787200 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029622868605817457, + "loss": 1.3335, + "theoretical_loss": 3.5457403998027677, + "tokens_seen": 1364852736 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002962186559679037, + "loss": 1.405, + "theoretical_loss": 3.5457251157157472, + "tokens_seen": 1364918272 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029620862587763293, + "loss": 1.225, + "theoretical_loss": 3.545709832568038, + "tokens_seen": 1364983808 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029619859578736206, + "loss": 1.205, + "theoretical_loss": 3.545694550359537, + "tokens_seen": 1365049344 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002961885656970913, + "loss": 1.2854, + "theoretical_loss": 3.5456792690901415, + "tokens_seen": 1365114880 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002961785356068205, + "loss": 1.2814, + "theoretical_loss": 3.545663988759748, + "tokens_seen": 1365180416 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029616850551654966, + "loss": 1.3981, + "theoretical_loss": 3.5456487093682547, + "tokens_seen": 1365245952 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029615847542627884, + "loss": 1.378, + "theoretical_loss": 3.545633430915559, + "tokens_seen": 1365311488 + }, + { + "epoch": 0.41, + "learning_rate": 0.000296148445336008, + "loss": 1.3689, + "theoretical_loss": 3.5456181534015565, + "tokens_seen": 1365377024 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002961384152457372, + "loss": 1.4338, + "theoretical_loss": 3.5456028768261465, + "tokens_seen": 1365442560 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029612838515546643, + "loss": 1.3468, + "theoretical_loss": 3.545587601189225, + "tokens_seen": 1365508096 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029611835506519556, + "loss": 1.3842, + "theoretical_loss": 3.5455723264906895, + "tokens_seen": 1365573632 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002961083249749248, + "loss": 1.3528, + "theoretical_loss": 3.545557052730438, + "tokens_seen": 1365639168 + }, + { + "epoch": 0.41, + "learning_rate": 0.000296098294884654, + "loss": 1.3821, + "theoretical_loss": 3.545541779908367, + "tokens_seen": 1365704704 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029608826479438316, + "loss": 1.3232, + "theoretical_loss": 3.5455265080243743, + "tokens_seen": 1365770240 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029607823470411234, + "loss": 1.2921, + "theoretical_loss": 3.5455112370783572, + "tokens_seen": 1365835776 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002960682046138415, + "loss": 1.3033, + "theoretical_loss": 3.545495967070213, + "tokens_seen": 1365901312 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002960581745235707, + "loss": 1.3662, + "theoretical_loss": 3.5454806979998392, + "tokens_seen": 1365966848 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029604814443329994, + "loss": 1.2995, + "theoretical_loss": 3.545465429867133, + "tokens_seen": 1366032384 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029603811434302906, + "loss": 1.2307, + "theoretical_loss": 3.5454501626719925, + "tokens_seen": 1366097920 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002960280842527583, + "loss": 1.4189, + "theoretical_loss": 3.5454348964143145, + "tokens_seen": 1366163456 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002960180541624874, + "loss": 1.4173, + "theoretical_loss": 3.5454196310939965, + "tokens_seen": 1366228992 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029600802407221666, + "loss": 1.4081, + "theoretical_loss": 3.5454043667109367, + "tokens_seen": 1366294528 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029599799398194584, + "loss": 1.3555, + "theoretical_loss": 3.5453891032650313, + "tokens_seen": 1366360064 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": -0.0003309218445792794, + "objective/train/docs_used": 773461, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4450771808624268, + "objective/train/original_loss": 2.4450771808624268, + "objective/train/theoretical_loss": 3.545373840756179, + "objective/train/tokens_used": 1386885600, + "objective/train/value_avg": -0.0046844482421875, + "objective/train/value_loss": 4.919776984024793e-05, + "objective/train/value_max": -0.0002491474151611328, + "objective/train/value_min": -0.0706787109375, + "objective/train/value_reward_corr": 0.4154007332015986, + "objective/train/value_std": 0.0037555694580078125, + "objective/train/weight_avg": 0.9990046620368958, + "objective/train/weighted_lm_loss": 2.4310295581817627, + "objective/train/weights_max": 1.6959035396575928, + "objective/train/weights_min": 0.4157238304615021, + "theoretical_loss": 3.545373840756179, + "tokens_seen": 1366425600 + }, + { + "epoch": 0.41, + "learning_rate": 0.000295987963891675, + "loss": 1.3023, + "theoretical_loss": 3.545373840756179, + "tokens_seen": 1366425600 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002959779338014042, + "loss": 1.3599, + "theoretical_loss": 3.5453585791842768, + "tokens_seen": 1366491136 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002959679037111334, + "loss": 1.3988, + "theoretical_loss": 3.5453433185492225, + "tokens_seen": 1366556672 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029595787362086257, + "loss": 1.4869, + "theoretical_loss": 3.545328058850913, + "tokens_seen": 1366622208 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002959478435305918, + "loss": 1.3895, + "theoretical_loss": 3.5453128000892473, + "tokens_seen": 1366687744 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029593781344032093, + "loss": 1.249, + "theoretical_loss": 3.5452975422641213, + "tokens_seen": 1366753280 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029592778335005016, + "loss": 1.3118, + "theoretical_loss": 3.5452822853754338, + "tokens_seen": 1366818816 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029591775325977935, + "loss": 1.4195, + "theoretical_loss": 3.5452670294230817, + "tokens_seen": 1366884352 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029590772316950853, + "loss": 1.3622, + "theoretical_loss": 3.5452517744069634, + "tokens_seen": 1366949888 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002958976930792377, + "loss": 1.3356, + "theoretical_loss": 3.545236520326976, + "tokens_seen": 1367015424 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002958876629889669, + "loss": 1.3663, + "theoretical_loss": 3.5452212671830168, + "tokens_seen": 1367080960 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029587763289869607, + "loss": 1.244, + "theoretical_loss": 3.545206014974985, + "tokens_seen": 1367146496 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002958676028084253, + "loss": 1.2497, + "theoretical_loss": 3.5451907637027764, + "tokens_seen": 1367212032 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029585757271815443, + "loss": 1.3236, + "theoretical_loss": 3.5451755133662903, + "tokens_seen": 1367277568 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029584754262788367, + "loss": 1.4583, + "theoretical_loss": 3.545160263965424, + "tokens_seen": 1367343104 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002958375125376128, + "loss": 1.4087, + "theoretical_loss": 3.5451450155000748, + "tokens_seen": 1367408640 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029582748244734203, + "loss": 1.3822, + "theoretical_loss": 3.5451297679701406, + "tokens_seen": 1367474176 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002958174523570712, + "loss": 1.3937, + "theoretical_loss": 3.545114521375519, + "tokens_seen": 1367539712 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002958074222668004, + "loss": 1.2563, + "theoretical_loss": 3.545099275716109, + "tokens_seen": 1367605248 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002957973921765296, + "loss": 1.3786, + "theoretical_loss": 3.545084030991807, + "tokens_seen": 1367670784 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002957873620862588, + "loss": 1.3599, + "theoretical_loss": 3.5450687872025113, + "tokens_seen": 1367736320 + }, + { + "epoch": 0.41, + "learning_rate": 0.000295777331995988, + "loss": 1.149, + "theoretical_loss": 3.54505354434812, + "tokens_seen": 1367801856 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029576730190571717, + "loss": 1.4198, + "theoretical_loss": 3.545038302428531, + "tokens_seen": 1367867392 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029575727181544635, + "loss": 1.311, + "theoretical_loss": 3.5450230614436418, + "tokens_seen": 1367932928 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029574724172517553, + "loss": 1.3754, + "theoretical_loss": 3.5450078213933507, + "tokens_seen": 1367998464 + }, + { + "epoch": 0.41, + "objective/train/advantage_avg": 0.000197978486539796, + "objective/train/docs_used": 774789, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.207716464996338, + "objective/train/original_loss": 3.207716226577759, + "objective/train/theoretical_loss": 3.5449925822775556, + "objective/train/tokens_used": 1388524000, + "objective/train/value_avg": -0.00577545166015625, + "objective/train/value_loss": 7.085037213983014e-05, + "objective/train/value_max": -0.00033283233642578125, + "objective/train/value_min": -0.06744384765625, + "objective/train/value_reward_corr": 0.39297890976239797, + "objective/train/value_std": 0.005466461181640625, + "objective/train/weight_avg": 1.0053706169128418, + "objective/train/weighted_lm_loss": 3.2335379123687744, + "objective/train/weights_max": 1.88888418674469, + "objective/train/weights_min": 0.476481556892395, + "theoretical_loss": 3.5449925822775556, + "tokens_seen": 1368064000 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029573721163490477, + "loss": 1.501, + "theoretical_loss": 3.5449925822775556, + "tokens_seen": 1368064000 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002957271815446339, + "loss": 1.3321, + "theoretical_loss": 3.544977344096154, + "tokens_seen": 1368129536 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029571715145436313, + "loss": 1.4688, + "theoretical_loss": 3.5449621068490442, + "tokens_seen": 1368195072 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029570712136409226, + "loss": 1.4551, + "theoretical_loss": 3.5449468705361245, + "tokens_seen": 1368260608 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002956970912738215, + "loss": 1.423, + "theoretical_loss": 3.5449316351572917, + "tokens_seen": 1368326144 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002956870611835507, + "loss": 1.2825, + "theoretical_loss": 3.544916400712445, + "tokens_seen": 1368391680 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029567703109327986, + "loss": 1.371, + "theoretical_loss": 3.5449011672014823, + "tokens_seen": 1368457216 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029566700100300904, + "loss": 1.4231, + "theoretical_loss": 3.544885934624301, + "tokens_seen": 1368522752 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002956569709127382, + "loss": 1.3085, + "theoretical_loss": 3.5448707029807993, + "tokens_seen": 1368588288 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002956469408224674, + "loss": 1.3477, + "theoretical_loss": 3.544855472270876, + "tokens_seen": 1368653824 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029563691073219663, + "loss": 1.3081, + "theoretical_loss": 3.544840242494428, + "tokens_seen": 1368719360 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029562688064192576, + "loss": 1.3304, + "theoretical_loss": 3.544825013651354, + "tokens_seen": 1368784896 + }, + { + "epoch": 0.41, + "learning_rate": 0.000295616850551655, + "loss": 1.3825, + "theoretical_loss": 3.544809785741553, + "tokens_seen": 1368850432 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002956068204613842, + "loss": 1.3341, + "theoretical_loss": 3.544794558764922, + "tokens_seen": 1368915968 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029559679037111336, + "loss": 1.2872, + "theoretical_loss": 3.5447793327213586, + "tokens_seen": 1368981504 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029558676028084254, + "loss": 1.3239, + "theoretical_loss": 3.5447641076107623, + "tokens_seen": 1369047040 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002955767301905717, + "loss": 1.459, + "theoretical_loss": 3.544748883433031, + "tokens_seen": 1369112576 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002955667001003009, + "loss": 1.2529, + "theoretical_loss": 3.5447336601880624, + "tokens_seen": 1369178112 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029555667001003014, + "loss": 1.314, + "theoretical_loss": 3.544718437875755, + "tokens_seen": 1369243648 + }, + { + "epoch": 0.41, + "learning_rate": 0.00029554663991975926, + "loss": 1.2813, + "theoretical_loss": 3.5447032164960066, + "tokens_seen": 1369309184 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002955366098294885, + "loss": 1.2721, + "theoretical_loss": 3.544687996048716, + "tokens_seen": 1369374720 + }, + { + "epoch": 0.41, + "learning_rate": 0.0002955265797392176, + "loss": 1.2885, + "theoretical_loss": 3.5446727765337815, + "tokens_seen": 1369440256 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029551654964894686, + "loss": 1.292, + "theoretical_loss": 3.5446575579511013, + "tokens_seen": 1369505792 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029550651955867604, + "loss": 1.3874, + "theoretical_loss": 3.544642340300573, + "tokens_seen": 1369571328 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002954964894684052, + "loss": 1.4033, + "theoretical_loss": 3.544627123582096, + "tokens_seen": 1369636864 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.0005356296896934509, + "objective/train/docs_used": 775516, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9146337509155273, + "objective/train/original_loss": 2.9146335124969482, + "objective/train/theoretical_loss": 3.5446119077955673, + "objective/train/tokens_used": 1390162400, + "objective/train/value_avg": -0.005962371826171875, + "objective/train/value_loss": 6.629277049796656e-05, + "objective/train/value_max": -0.0003612041473388672, + "objective/train/value_min": -0.07794189453125, + "objective/train/value_reward_corr": 0.4162289369504385, + "objective/train/value_std": 0.005191802978515625, + "objective/train/weight_avg": 0.9976552724838257, + "objective/train/weighted_lm_loss": 2.9020028114318848, + "objective/train/weights_max": 1.7421408891677856, + "objective/train/weights_min": 0.09331729263067245, + "theoretical_loss": 3.5446119077955673, + "tokens_seen": 1369702400 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002954864593781344, + "loss": 1.3962, + "theoretical_loss": 3.5446119077955673, + "tokens_seen": 1369702400 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002954764292878636, + "loss": 1.3897, + "theoretical_loss": 3.5445966929408863, + "tokens_seen": 1369767936 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029546639919759277, + "loss": 1.448, + "theoretical_loss": 3.5445814790179515, + "tokens_seen": 1369833472 + }, + { + "epoch": 0.42, + "learning_rate": 0.000295456369107322, + "loss": 1.3224, + "theoretical_loss": 3.5445662660266604, + "tokens_seen": 1369899008 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029544633901705113, + "loss": 1.1326, + "theoretical_loss": 3.544551053966912, + "tokens_seen": 1369964544 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029543630892678037, + "loss": 1.3194, + "theoretical_loss": 3.544535842838604, + "tokens_seen": 1370030080 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029542627883650955, + "loss": 1.3605, + "theoretical_loss": 3.544520632641636, + "tokens_seen": 1370095616 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029541624874623873, + "loss": 1.3779, + "theoretical_loss": 3.544505423375905, + "tokens_seen": 1370161152 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002954062186559679, + "loss": 1.2453, + "theoretical_loss": 3.5444902150413107, + "tokens_seen": 1370226688 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002953961885656971, + "loss": 1.3558, + "theoretical_loss": 3.544475007637751, + "tokens_seen": 1370292224 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029538615847542627, + "loss": 1.3412, + "theoretical_loss": 3.5444598011651243, + "tokens_seen": 1370357760 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002953761283851555, + "loss": 1.2687, + "theoretical_loss": 3.5444445956233293, + "tokens_seen": 1370423296 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029536609829488463, + "loss": 1.2843, + "theoretical_loss": 3.5444293910122644, + "tokens_seen": 1370488832 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029535606820461387, + "loss": 1.3548, + "theoretical_loss": 3.544414187331828, + "tokens_seen": 1370554368 + }, + { + "epoch": 0.42, + "learning_rate": 0.000295346038114343, + "loss": 1.3514, + "theoretical_loss": 3.5443989845819193, + "tokens_seen": 1370619904 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029533600802407223, + "loss": 1.4316, + "theoretical_loss": 3.544383782762436, + "tokens_seen": 1370685440 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002953259779338014, + "loss": 1.3168, + "theoretical_loss": 3.544368581873277, + "tokens_seen": 1370750976 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002953159478435306, + "loss": 1.2079, + "theoretical_loss": 3.544353381914341, + "tokens_seen": 1370816512 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002953059177532598, + "loss": 1.3477, + "theoretical_loss": 3.5443381828855265, + "tokens_seen": 1370882048 + }, + { + "epoch": 0.42, + "learning_rate": 0.000295295887662989, + "loss": 1.2894, + "theoretical_loss": 3.5443229847867324, + "tokens_seen": 1370947584 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029528585757271814, + "loss": 1.1993, + "theoretical_loss": 3.5443077876178566, + "tokens_seen": 1371013120 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029527582748244737, + "loss": 1.2872, + "theoretical_loss": 3.544292591378799, + "tokens_seen": 1371078656 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002952657973921765, + "loss": 1.2554, + "theoretical_loss": 3.5442773960694565, + "tokens_seen": 1371144192 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029525576730190573, + "loss": 1.3136, + "theoretical_loss": 3.5442622016897296, + "tokens_seen": 1371209728 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002952457372116349, + "loss": 1.3738, + "theoretical_loss": 3.5442470082395157, + "tokens_seen": 1371275264 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.0009736904175952077, + "objective/train/docs_used": 776338, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8554773330688477, + "objective/train/original_loss": 2.8554773330688477, + "objective/train/theoretical_loss": 3.5442318157187147, + "objective/train/tokens_used": 1391800800, + "objective/train/value_avg": -0.005954742431640625, + "objective/train/value_loss": 6.924451736267656e-05, + "objective/train/value_max": -0.0002453327178955078, + "objective/train/value_min": -0.06451416015625, + "objective/train/value_reward_corr": 0.33615146842316446, + "objective/train/value_std": 0.004169464111328125, + "objective/train/weight_avg": 0.9934540390968323, + "objective/train/weighted_lm_loss": 2.8459250926971436, + "objective/train/weights_max": 1.7848817110061646, + "objective/train/weights_min": 0.3385083079338074, + "theoretical_loss": 3.5442318157187147, + "tokens_seen": 1371340800 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002952357071213641, + "loss": 1.3708, + "theoretical_loss": 3.5442318157187147, + "tokens_seen": 1371340800 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002952256770310933, + "loss": 1.3304, + "theoretical_loss": 3.544216624127224, + "tokens_seen": 1371406336 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029521564694082246, + "loss": 1.3426, + "theoretical_loss": 3.544201433464943, + "tokens_seen": 1371471872 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029520561685055164, + "loss": 1.3611, + "theoretical_loss": 3.544186243731771, + "tokens_seen": 1371537408 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002951955867602809, + "loss": 1.4017, + "theoretical_loss": 3.544171054927606, + "tokens_seen": 1371602944 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029518555667001, + "loss": 1.3696, + "theoretical_loss": 3.544155867052347, + "tokens_seen": 1371668480 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029517552657973924, + "loss": 1.2109, + "theoretical_loss": 3.544140680105893, + "tokens_seen": 1371734016 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029516549648946836, + "loss": 1.3142, + "theoretical_loss": 3.5441254940881426, + "tokens_seen": 1371799552 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002951554663991976, + "loss": 1.3529, + "theoretical_loss": 3.5441103089989947, + "tokens_seen": 1371865088 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002951454363089268, + "loss": 1.2892, + "theoretical_loss": 3.5440951248383485, + "tokens_seen": 1371930624 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029513540621865596, + "loss": 1.399, + "theoretical_loss": 3.5440799416061024, + "tokens_seen": 1371996160 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029512537612838514, + "loss": 1.3032, + "theoretical_loss": 3.544064759302155, + "tokens_seen": 1372061696 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002951153460381144, + "loss": 1.4277, + "theoretical_loss": 3.5440495779264065, + "tokens_seen": 1372127232 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002951053159478435, + "loss": 1.4755, + "theoretical_loss": 3.5440343974787547, + "tokens_seen": 1372192768 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029509528585757274, + "loss": 1.3915, + "theoretical_loss": 3.544019217959099, + "tokens_seen": 1372258304 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029508525576730187, + "loss": 1.4207, + "theoretical_loss": 3.5440040393673375, + "tokens_seen": 1372323840 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002950752256770311, + "loss": 1.3403, + "theoretical_loss": 3.5439888617033706, + "tokens_seen": 1372389376 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002950651955867603, + "loss": 1.3369, + "theoretical_loss": 3.543973684967096, + "tokens_seen": 1372454912 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029505516549648946, + "loss": 1.3008, + "theoretical_loss": 3.5439585091584136, + "tokens_seen": 1372520448 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029504513540621865, + "loss": 1.3841, + "theoretical_loss": 3.5439433342772224, + "tokens_seen": 1372585984 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002950351053159478, + "loss": 1.2461, + "theoretical_loss": 3.5439281603234205, + "tokens_seen": 1372651520 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029502507522567706, + "loss": 1.2828, + "theoretical_loss": 3.543912987296908, + "tokens_seen": 1372717056 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029501504513540624, + "loss": 1.3382, + "theoretical_loss": 3.5438978151975835, + "tokens_seen": 1372782592 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002950050150451354, + "loss": 1.3413, + "theoretical_loss": 3.5438826440253455, + "tokens_seen": 1372848128 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002949949849548646, + "loss": 1.4336, + "theoretical_loss": 3.543867473780094, + "tokens_seen": 1372913664 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -7.635955262230709e-05, + "objective/train/docs_used": 777606, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.618417978286743, + "objective/train/original_loss": 2.618417978286743, + "objective/train/theoretical_loss": 3.543852304461728, + "objective/train/tokens_used": 1393439200, + "objective/train/value_avg": -0.003902435302734375, + "objective/train/value_loss": 5.134617458679713e-05, + "objective/train/value_max": -0.0002269744873046875, + "objective/train/value_min": -0.0765380859375, + "objective/train/value_reward_corr": 0.34447903571331595, + "objective/train/value_std": 0.0034694671630859375, + "objective/train/weight_avg": 1.001583218574524, + "objective/train/weighted_lm_loss": 2.6207664012908936, + "objective/train/weights_max": 1.969077706336975, + "objective/train/weights_min": 0.34868526458740234, + "theoretical_loss": 3.543852304461728, + "tokens_seen": 1372979200 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002949849548645938, + "loss": 1.365, + "theoretical_loss": 3.543852304461728, + "tokens_seen": 1372979200 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029497492477432297, + "loss": 1.3869, + "theoretical_loss": 3.543837136070147, + "tokens_seen": 1373044736 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002949648946840522, + "loss": 1.3703, + "theoretical_loss": 3.543821968605249, + "tokens_seen": 1373110272 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029495486459378133, + "loss": 1.2879, + "theoretical_loss": 3.5438068020669338, + "tokens_seen": 1373175808 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029494483450351057, + "loss": 1.4397, + "theoretical_loss": 3.5437916364551008, + "tokens_seen": 1373241344 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029493480441323975, + "loss": 1.3465, + "theoretical_loss": 3.5437764717696485, + "tokens_seen": 1373306880 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029492477432296893, + "loss": 1.3647, + "theoretical_loss": 3.5437613080104766, + "tokens_seen": 1373372416 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002949147442326981, + "loss": 1.3732, + "theoretical_loss": 3.5437461451774843, + "tokens_seen": 1373437952 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002949047141424273, + "loss": 1.2692, + "theoretical_loss": 3.543730983270571, + "tokens_seen": 1373503488 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029489468405215647, + "loss": 1.2417, + "theoretical_loss": 3.5437158222896357, + "tokens_seen": 1373569024 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002948846539618857, + "loss": 1.3542, + "theoretical_loss": 3.5437006622345777, + "tokens_seen": 1373634560 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029487462387161483, + "loss": 1.3073, + "theoretical_loss": 3.543685503105296, + "tokens_seen": 1373700096 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029486459378134407, + "loss": 1.3895, + "theoretical_loss": 3.543670344901691, + "tokens_seen": 1373765632 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002948545636910732, + "loss": 1.3047, + "theoretical_loss": 3.543655187623661, + "tokens_seen": 1373831168 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029484453360080243, + "loss": 1.307, + "theoretical_loss": 3.543640031271105, + "tokens_seen": 1373896704 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002948345035105316, + "loss": 1.3979, + "theoretical_loss": 3.5436248758439235, + "tokens_seen": 1373962240 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002948244734202608, + "loss": 1.3684, + "theoretical_loss": 3.543609721342015, + "tokens_seen": 1374027776 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029481444332999, + "loss": 1.4265, + "theoretical_loss": 3.5435945677652794, + "tokens_seen": 1374093312 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002948044132397192, + "loss": 1.3127, + "theoretical_loss": 3.5435794151136157, + "tokens_seen": 1374158848 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029479438314944834, + "loss": 1.3659, + "theoretical_loss": 3.5435642633869238, + "tokens_seen": 1374224384 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029478435305917757, + "loss": 1.3443, + "theoretical_loss": 3.543549112585102, + "tokens_seen": 1374289920 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002947743229689067, + "loss": 1.3876, + "theoretical_loss": 3.5435339627080515, + "tokens_seen": 1374355456 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029476429287863593, + "loss": 1.2966, + "theoretical_loss": 3.5435188137556706, + "tokens_seen": 1374420992 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002947542627883651, + "loss": 1.3582, + "theoretical_loss": 3.543503665727859, + "tokens_seen": 1374486528 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002947442326980943, + "loss": 1.3602, + "theoretical_loss": 3.5434885186245157, + "tokens_seen": 1374552064 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.00035679570282809436, + "objective/train/docs_used": 778164, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6433377265930176, + "objective/train/original_loss": 2.6433374881744385, + "objective/train/theoretical_loss": 3.5434733724455407, + "objective/train/tokens_used": 1395077600, + "objective/train/value_avg": -0.00492095947265625, + "objective/train/value_loss": 4.351153256720863e-05, + "objective/train/value_max": -0.00031757354736328125, + "objective/train/value_min": -0.052337646484375, + "objective/train/value_reward_corr": 0.26397415783561384, + "objective/train/value_std": 0.00368499755859375, + "objective/train/weight_avg": 0.9985454678535461, + "objective/train/weighted_lm_loss": 2.6389381885528564, + "objective/train/weights_max": 1.4055795669555664, + "objective/train/weights_min": 0.5301655530929565, + "theoretical_loss": 3.5434733724455407, + "tokens_seen": 1374617600 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002947342026078235, + "loss": 1.3247, + "theoretical_loss": 3.5434733724455407, + "tokens_seen": 1374617600 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029472417251755266, + "loss": 1.3868, + "theoretical_loss": 3.5434582271908344, + "tokens_seen": 1374683136 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029471414242728184, + "loss": 1.4233, + "theoretical_loss": 3.5434430828602945, + "tokens_seen": 1374748672 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002947041123370111, + "loss": 1.2441, + "theoretical_loss": 3.5434279394538217, + "tokens_seen": 1374814208 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002946940822467402, + "loss": 1.4169, + "theoretical_loss": 3.543412796971316, + "tokens_seen": 1374879744 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029468405215646944, + "loss": 1.4415, + "theoretical_loss": 3.543397655412676, + "tokens_seen": 1374945280 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029467402206619856, + "loss": 1.3797, + "theoretical_loss": 3.5433825147778015, + "tokens_seen": 1375010816 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002946639919759278, + "loss": 1.358, + "theoretical_loss": 3.543367375066593, + "tokens_seen": 1375076352 + }, + { + "epoch": 0.42, + "learning_rate": 0.000294653961885657, + "loss": 1.2999, + "theoretical_loss": 3.5433522362789485, + "tokens_seen": 1375141888 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029464393179538616, + "loss": 1.2941, + "theoretical_loss": 3.543337098414769, + "tokens_seen": 1375207424 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029463390170511534, + "loss": 1.2502, + "theoretical_loss": 3.5433219614739535, + "tokens_seen": 1375272960 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002946238716148446, + "loss": 1.37, + "theoretical_loss": 3.543306825456402, + "tokens_seen": 1375338496 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002946138415245737, + "loss": 1.4251, + "theoretical_loss": 3.5432916903620146, + "tokens_seen": 1375404032 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029460381143430294, + "loss": 1.2719, + "theoretical_loss": 3.54327655619069, + "tokens_seen": 1375469568 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029459378134403207, + "loss": 1.3824, + "theoretical_loss": 3.5432614229423285, + "tokens_seen": 1375535104 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002945837512537613, + "loss": 1.3026, + "theoretical_loss": 3.54324629061683, + "tokens_seen": 1375600640 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002945737211634905, + "loss": 1.3991, + "theoretical_loss": 3.543231159214094, + "tokens_seen": 1375666176 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029456369107321966, + "loss": 1.4363, + "theoretical_loss": 3.5432160287340206, + "tokens_seen": 1375731712 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029455366098294885, + "loss": 1.392, + "theoretical_loss": 3.543200899176509, + "tokens_seen": 1375797248 + }, + { + "epoch": 0.42, + "learning_rate": 0.000294543630892678, + "loss": 1.3923, + "theoretical_loss": 3.5431857705414593, + "tokens_seen": 1375862784 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002945336008024072, + "loss": 1.3398, + "theoretical_loss": 3.5431706428287715, + "tokens_seen": 1375928320 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029452357071213644, + "loss": 1.3358, + "theoretical_loss": 3.543155516038345, + "tokens_seen": 1375993856 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029451354062186557, + "loss": 1.4287, + "theoretical_loss": 3.5431403901700804, + "tokens_seen": 1376059392 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002945035105315948, + "loss": 1.4625, + "theoretical_loss": 3.5431252652238765, + "tokens_seen": 1376124928 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029449348044132393, + "loss": 1.3344, + "theoretical_loss": 3.543110141199634, + "tokens_seen": 1376190464 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": 0.0005963516305200756, + "objective/train/docs_used": 779255, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7059783935546875, + "objective/train/original_loss": 2.70597767829895, + "objective/train/theoretical_loss": 3.5430950180972527, + "objective/train/tokens_used": 1396716000, + "objective/train/value_avg": -0.005706787109375, + "objective/train/value_loss": 5.780037463409826e-05, + "objective/train/value_max": -0.0004858970642089844, + "objective/train/value_min": -0.07781982421875, + "objective/train/value_reward_corr": 0.29864625472672957, + "objective/train/value_std": 0.004772186279296875, + "objective/train/weight_avg": 1.0087594985961914, + "objective/train/weighted_lm_loss": 2.7456717491149902, + "objective/train/weights_max": 2.078640937805176, + "objective/train/weights_min": 0.39524656534194946, + "theoretical_loss": 3.5430950180972527, + "tokens_seen": 1376256000 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029448345035105317, + "loss": 1.3614, + "theoretical_loss": 3.5430950180972527, + "tokens_seen": 1376256000 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029447342026078235, + "loss": 1.3395, + "theoretical_loss": 3.543079895916633, + "tokens_seen": 1376321536 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029446339017051153, + "loss": 1.3272, + "theoretical_loss": 3.5430647746576733, + "tokens_seen": 1376387072 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002944533600802407, + "loss": 1.3375, + "theoretical_loss": 3.543049654320275, + "tokens_seen": 1376452608 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029444332998996995, + "loss": 1.4319, + "theoretical_loss": 3.5430345349043373, + "tokens_seen": 1376518144 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002944332998996991, + "loss": 1.3869, + "theoretical_loss": 3.54301941640976, + "tokens_seen": 1376583680 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002944232698094283, + "loss": 1.3157, + "theoretical_loss": 3.5430042988364443, + "tokens_seen": 1376649216 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029441323971915744, + "loss": 1.4575, + "theoretical_loss": 3.5429891821842894, + "tokens_seen": 1376714752 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029440320962888667, + "loss": 1.4626, + "theoretical_loss": 3.542974066453195, + "tokens_seen": 1376780288 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029439317953861585, + "loss": 1.3357, + "theoretical_loss": 3.5429589516430617, + "tokens_seen": 1376845824 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029438314944834503, + "loss": 1.3915, + "theoretical_loss": 3.542943837753789, + "tokens_seen": 1376911360 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002943731193580742, + "loss": 1.4657, + "theoretical_loss": 3.542928724785278, + "tokens_seen": 1376976896 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002943630892678034, + "loss": 1.4957, + "theoretical_loss": 3.5429136127374274, + "tokens_seen": 1377042432 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002943530591775326, + "loss": 1.2745, + "theoretical_loss": 3.5428985016101384, + "tokens_seen": 1377107968 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002943430290872618, + "loss": 1.3191, + "theoretical_loss": 3.5428833914033104, + "tokens_seen": 1377173504 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029433299899699094, + "loss": 1.3828, + "theoretical_loss": 3.542868282116844, + "tokens_seen": 1377239040 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002943229689067202, + "loss": 1.3439, + "theoretical_loss": 3.5428531737506397, + "tokens_seen": 1377304576 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002943129388164493, + "loss": 1.3341, + "theoretical_loss": 3.542838066304596, + "tokens_seen": 1377370112 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029430290872617854, + "loss": 1.3884, + "theoretical_loss": 3.5428229597786154, + "tokens_seen": 1377435648 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029429287863590777, + "loss": 1.3836, + "theoretical_loss": 3.5428078541725965, + "tokens_seen": 1377501184 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002942828485456369, + "loss": 1.3088, + "theoretical_loss": 3.54279274948644, + "tokens_seen": 1377566720 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029427281845536613, + "loss": 1.3714, + "theoretical_loss": 3.5427776457200464, + "tokens_seen": 1377632256 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002942627883650953, + "loss": 1.3414, + "theoretical_loss": 3.542762542873315, + "tokens_seen": 1377697792 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002942527582748245, + "loss": 1.4023, + "theoretical_loss": 3.5427474409461475, + "tokens_seen": 1377763328 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002942427281845537, + "loss": 1.3702, + "theoretical_loss": 3.5427323399384427, + "tokens_seen": 1377828864 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": 0.00034320849226787686, + "objective/train/docs_used": 780014, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.418775796890259, + "objective/train/original_loss": 2.4187755584716797, + "objective/train/theoretical_loss": 3.5427172398501012, + "objective/train/tokens_used": 1398354400, + "objective/train/value_avg": -0.004909515380859375, + "objective/train/value_loss": 3.4424887417117134e-05, + "objective/train/value_max": -0.0002532005310058594, + "objective/train/value_min": -0.042724609375, + "objective/train/value_reward_corr": 0.44554032226871365, + "objective/train/value_std": 0.0038127899169921875, + "objective/train/weight_avg": 1.0051064491271973, + "objective/train/weighted_lm_loss": 2.421550750732422, + "objective/train/weights_max": 1.4317346811294556, + "objective/train/weights_min": 0.6492063403129578, + "theoretical_loss": 3.5427172398501012, + "tokens_seen": 1377894400 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029423269809428286, + "loss": 1.3957, + "theoretical_loss": 3.5427172398501012, + "tokens_seen": 1377894400 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029422266800401204, + "loss": 1.3193, + "theoretical_loss": 3.5427021406810244, + "tokens_seen": 1377959936 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002942126379137413, + "loss": 1.4498, + "theoretical_loss": 3.542687042431112, + "tokens_seen": 1378025472 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002942026078234704, + "loss": 1.3636, + "theoretical_loss": 3.5426719451002637, + "tokens_seen": 1378091008 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029419257773319964, + "loss": 1.2643, + "theoretical_loss": 3.5426568486883805, + "tokens_seen": 1378156544 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029418254764292876, + "loss": 1.3797, + "theoretical_loss": 3.5426417531953627, + "tokens_seen": 1378222080 + }, + { + "epoch": 0.42, + "learning_rate": 0.000294172517552658, + "loss": 1.4554, + "theoretical_loss": 3.542626658621111, + "tokens_seen": 1378287616 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002941624874623872, + "loss": 1.445, + "theoretical_loss": 3.542611564965525, + "tokens_seen": 1378353152 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029415245737211636, + "loss": 1.332, + "theoretical_loss": 3.5425964722285053, + "tokens_seen": 1378418688 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029414242728184554, + "loss": 1.2848, + "theoretical_loss": 3.5425813804099526, + "tokens_seen": 1378484224 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002941323971915748, + "loss": 1.2693, + "theoretical_loss": 3.542566289509768, + "tokens_seen": 1378549760 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002941223671013039, + "loss": 1.3673, + "theoretical_loss": 3.5425511995278507, + "tokens_seen": 1378615296 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029411233701103314, + "loss": 1.3662, + "theoretical_loss": 3.542536110464102, + "tokens_seen": 1378680832 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029410230692076227, + "loss": 1.281, + "theoretical_loss": 3.5425210223184216, + "tokens_seen": 1378746368 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002940922768304915, + "loss": 1.3268, + "theoretical_loss": 3.542505935090711, + "tokens_seen": 1378811904 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002940822467402207, + "loss": 1.3414, + "theoretical_loss": 3.5424908487808704, + "tokens_seen": 1378877440 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029407221664994986, + "loss": 1.3257, + "theoretical_loss": 3.5424757633887998, + "tokens_seen": 1378942976 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029406218655967905, + "loss": 1.3666, + "theoretical_loss": 3.5424606789144004, + "tokens_seen": 1379008512 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002940521564694082, + "loss": 1.3282, + "theoretical_loss": 3.5424455953575724, + "tokens_seen": 1379074048 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002940421263791374, + "loss": 1.3581, + "theoretical_loss": 3.5424305127182167, + "tokens_seen": 1379139584 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029403209628886664, + "loss": 1.3077, + "theoretical_loss": 3.5424154309962335, + "tokens_seen": 1379205120 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029402206619859577, + "loss": 1.2951, + "theoretical_loss": 3.5424003501915236, + "tokens_seen": 1379270656 + }, + { + "epoch": 0.42, + "learning_rate": 0.000294012036108325, + "loss": 1.3228, + "theoretical_loss": 3.542385270303988, + "tokens_seen": 1379336192 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029400200601805413, + "loss": 1.4857, + "theoretical_loss": 3.542370191333527, + "tokens_seen": 1379401728 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029399197592778337, + "loss": 1.3964, + "theoretical_loss": 3.5423551132800406, + "tokens_seen": 1379467264 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.0002705657097976655, + "objective/train/docs_used": 781246, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6886651515960693, + "objective/train/original_loss": 2.6886651515960693, + "objective/train/theoretical_loss": 3.5423400361434307, + "objective/train/tokens_used": 1399992800, + "objective/train/value_avg": -0.004344940185546875, + "objective/train/value_loss": 4.010374323115684e-05, + "objective/train/value_max": -0.00034332275390625, + "objective/train/value_min": -0.06231689453125, + "objective/train/value_reward_corr": 0.2605116622904157, + "objective/train/value_std": 0.0029773712158203125, + "objective/train/weight_avg": 0.9992139339447021, + "objective/train/weighted_lm_loss": 2.6850199699401855, + "objective/train/weights_max": 1.8648282289505005, + "objective/train/weights_min": 0.47700896859169006, + "theoretical_loss": 3.5423400361434307, + "tokens_seen": 1379532800 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029398194583751255, + "loss": 1.4189, + "theoretical_loss": 3.5423400361434307, + "tokens_seen": 1379532800 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029397191574724173, + "loss": 1.333, + "theoretical_loss": 3.5423249599235973, + "tokens_seen": 1379598336 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002939618856569709, + "loss": 1.3723, + "theoretical_loss": 3.5423098846204413, + "tokens_seen": 1379663872 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029395185556670015, + "loss": 1.2496, + "theoretical_loss": 3.542294810233863, + "tokens_seen": 1379729408 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002939418254764293, + "loss": 1.2584, + "theoretical_loss": 3.5422797367637635, + "tokens_seen": 1379794944 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002939317953861585, + "loss": 1.297, + "theoretical_loss": 3.542264664210044, + "tokens_seen": 1379860480 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029392176529588764, + "loss": 1.3619, + "theoretical_loss": 3.542249592572605, + "tokens_seen": 1379926016 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029391173520561687, + "loss": 1.4591, + "theoretical_loss": 3.5422345218513467, + "tokens_seen": 1379991552 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029390170511534605, + "loss": 1.371, + "theoretical_loss": 3.5422194520461705, + "tokens_seen": 1380057088 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029389167502507523, + "loss": 1.2525, + "theoretical_loss": 3.5422043831569767, + "tokens_seen": 1380122624 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002938816449348044, + "loss": 1.3493, + "theoretical_loss": 3.542189315183667, + "tokens_seen": 1380188160 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002938716148445336, + "loss": 1.3276, + "theoretical_loss": 3.542174248126141, + "tokens_seen": 1380253696 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002938615847542628, + "loss": 1.3424, + "theoretical_loss": 3.5421591819843012, + "tokens_seen": 1380319232 + }, + { + "epoch": 0.42, + "learning_rate": 0.000293851554663992, + "loss": 1.3115, + "theoretical_loss": 3.542144116758047, + "tokens_seen": 1380384768 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029384152457372114, + "loss": 1.2902, + "theoretical_loss": 3.5421290524472795, + "tokens_seen": 1380450304 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002938314944834504, + "loss": 1.2665, + "theoretical_loss": 3.5421139890519004, + "tokens_seen": 1380515840 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002938214643931795, + "loss": 1.2524, + "theoretical_loss": 3.5420989265718106, + "tokens_seen": 1380581376 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029381143430290874, + "loss": 1.4458, + "theoretical_loss": 3.54208386500691, + "tokens_seen": 1380646912 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002938014042126379, + "loss": 1.2945, + "theoretical_loss": 3.5420688043571005, + "tokens_seen": 1380712448 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002937913741223671, + "loss": 1.3742, + "theoretical_loss": 3.542053744622282, + "tokens_seen": 1380777984 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002937813440320963, + "loss": 1.218, + "theoretical_loss": 3.5420386858023574, + "tokens_seen": 1380843520 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002937713139418255, + "loss": 1.498, + "theoretical_loss": 3.542023627897226, + "tokens_seen": 1380909056 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029376128385155464, + "loss": 1.4965, + "theoretical_loss": 3.5420085709067894, + "tokens_seen": 1380974592 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002937512537612839, + "loss": 1.3762, + "theoretical_loss": 3.5419935148309483, + "tokens_seen": 1381040128 + }, + { + "epoch": 0.42, + "learning_rate": 0.000293741223671013, + "loss": 1.3539, + "theoretical_loss": 3.5419784596696045, + "tokens_seen": 1381105664 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": 0.0001850074768299237, + "objective/train/docs_used": 781914, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6459145545959473, + "objective/train/original_loss": 2.6459150314331055, + "objective/train/theoretical_loss": 3.541963405422658, + "objective/train/tokens_used": 1401631200, + "objective/train/value_avg": -0.005878448486328125, + "objective/train/value_loss": 7.561954407719895e-05, + "objective/train/value_max": -0.00047469139099121094, + "objective/train/value_min": -0.11126708984375, + "objective/train/value_reward_corr": 0.4374404386979189, + "objective/train/value_std": 0.005016326904296875, + "objective/train/weight_avg": 1.005266785621643, + "objective/train/weighted_lm_loss": 2.6491806507110596, + "objective/train/weights_max": 2.7369682788848877, + "objective/train/weights_min": 0.14946913719177246, + "theoretical_loss": 3.541963405422658, + "tokens_seen": 1381171200 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029373119358074224, + "loss": 1.353, + "theoretical_loss": 3.541963405422658, + "tokens_seen": 1381171200 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002937211634904714, + "loss": 1.3394, + "theoretical_loss": 3.541948352090011, + "tokens_seen": 1381236736 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002937111334002006, + "loss": 1.3742, + "theoretical_loss": 3.5419332996715642, + "tokens_seen": 1381302272 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002937011033099298, + "loss": 1.4195, + "theoretical_loss": 3.5419182481672182, + "tokens_seen": 1381367808 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029369107321965896, + "loss": 1.4115, + "theoretical_loss": 3.5419031975768744, + "tokens_seen": 1381433344 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029368104312938815, + "loss": 1.3847, + "theoretical_loss": 3.5418881479004343, + "tokens_seen": 1381498880 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002936710130391174, + "loss": 1.2909, + "theoretical_loss": 3.5418730991377987, + "tokens_seen": 1381564416 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002936609829488465, + "loss": 1.3707, + "theoretical_loss": 3.5418580512888687, + "tokens_seen": 1381629952 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029365095285857574, + "loss": 1.3257, + "theoretical_loss": 3.541843004353546, + "tokens_seen": 1381695488 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029364092276830487, + "loss": 1.4033, + "theoretical_loss": 3.5418279583317314, + "tokens_seen": 1381761024 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002936308926780341, + "loss": 1.2617, + "theoretical_loss": 3.541812913223326, + "tokens_seen": 1381826560 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002936208625877633, + "loss": 1.3452, + "theoretical_loss": 3.5417978690282315, + "tokens_seen": 1381892096 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029361083249749247, + "loss": 1.3468, + "theoretical_loss": 3.5417828257463486, + "tokens_seen": 1381957632 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029360080240722165, + "loss": 1.344, + "theoretical_loss": 3.541767783377579, + "tokens_seen": 1382023168 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002935907723169509, + "loss": 1.3211, + "theoretical_loss": 3.5417527419218233, + "tokens_seen": 1382088704 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029358074222668, + "loss": 1.2723, + "theoretical_loss": 3.5417377013789837, + "tokens_seen": 1382154240 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029357071213640925, + "loss": 1.4121, + "theoretical_loss": 3.5417226617489614, + "tokens_seen": 1382219776 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002935606820461384, + "loss": 1.3724, + "theoretical_loss": 3.541707623031657, + "tokens_seen": 1382285312 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002935506519558676, + "loss": 1.3359, + "theoretical_loss": 3.5416925852269725, + "tokens_seen": 1382350848 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029354062186559684, + "loss": 1.3072, + "theoretical_loss": 3.5416775483348086, + "tokens_seen": 1382416384 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029353059177532597, + "loss": 1.3565, + "theoretical_loss": 3.541662512355068, + "tokens_seen": 1382481920 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002935205616850552, + "loss": 1.2932, + "theoretical_loss": 3.5416474772876505, + "tokens_seen": 1382547456 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029351053159478433, + "loss": 1.271, + "theoretical_loss": 3.541632443132458, + "tokens_seen": 1382612992 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029350050150451357, + "loss": 1.3176, + "theoretical_loss": 3.541617409889392, + "tokens_seen": 1382678528 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029349047141424275, + "loss": 1.4085, + "theoretical_loss": 3.541602377558355, + "tokens_seen": 1382744064 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": 0.0006855725659988821, + "objective/train/docs_used": 783265, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7433369159698486, + "objective/train/original_loss": 2.7433371543884277, + "objective/train/theoretical_loss": 3.541587346139247, + "objective/train/tokens_used": 1403269600, + "objective/train/value_avg": -0.00501251220703125, + "objective/train/value_loss": 5.614881229121238e-05, + "objective/train/value_max": -0.0002892017364501953, + "objective/train/value_min": -0.06658935546875, + "objective/train/value_reward_corr": 0.26129259987170733, + "objective/train/value_std": 0.0038280487060546875, + "objective/train/weight_avg": 1.0091716051101685, + "objective/train/weighted_lm_loss": 2.7651431560516357, + "objective/train/weights_max": 1.9462287425994873, + "objective/train/weights_min": 0.0842919647693634, + "theoretical_loss": 3.541587346139247, + "tokens_seen": 1382809600 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029348044132397193, + "loss": 1.4627, + "theoretical_loss": 3.541587346139247, + "tokens_seen": 1382809600 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002934704112337011, + "loss": 1.4181, + "theoretical_loss": 3.54157231563197, + "tokens_seen": 1382875136 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029346038114343035, + "loss": 1.3669, + "theoretical_loss": 3.541557286036425, + "tokens_seen": 1382940672 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002934503510531595, + "loss": 1.3419, + "theoretical_loss": 3.541542257352514, + "tokens_seen": 1383006208 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002934403209628887, + "loss": 1.2823, + "theoretical_loss": 3.5415272295801388, + "tokens_seen": 1383071744 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029343029087261784, + "loss": 1.4133, + "theoretical_loss": 3.5415122027192005, + "tokens_seen": 1383137280 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029342026078234707, + "loss": 1.4665, + "theoretical_loss": 3.5414971767696004, + "tokens_seen": 1383202816 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029341023069207625, + "loss": 1.3626, + "theoretical_loss": 3.5414821517312403, + "tokens_seen": 1383268352 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029340020060180543, + "loss": 1.2803, + "theoretical_loss": 3.5414671276040224, + "tokens_seen": 1383333888 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002933901705115346, + "loss": 1.456, + "theoretical_loss": 3.5414521043878473, + "tokens_seen": 1383399424 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002933801404212638, + "loss": 1.4754, + "theoretical_loss": 3.5414370820826164, + "tokens_seen": 1383464960 + }, + { + "epoch": 0.42, + "learning_rate": 0.000293370110330993, + "loss": 1.3271, + "theoretical_loss": 3.541422060688233, + "tokens_seen": 1383530496 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002933600802407222, + "loss": 1.3435, + "theoretical_loss": 3.541407040204597, + "tokens_seen": 1383596032 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029335005015045134, + "loss": 1.4616, + "theoretical_loss": 3.5413920206316107, + "tokens_seen": 1383661568 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002933400200601806, + "loss": 1.4107, + "theoretical_loss": 3.5413770019691757, + "tokens_seen": 1383727104 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002933299899699097, + "loss": 1.279, + "theoretical_loss": 3.541361984217194, + "tokens_seen": 1383792640 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029331995987963894, + "loss": 1.3622, + "theoretical_loss": 3.541346967375567, + "tokens_seen": 1383858176 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002933099297893681, + "loss": 1.3694, + "theoretical_loss": 3.5413319514441963, + "tokens_seen": 1383923712 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002932998996990973, + "loss": 1.471, + "theoretical_loss": 3.5413169364229837, + "tokens_seen": 1383989248 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002932898696088265, + "loss": 1.3356, + "theoretical_loss": 3.541301922311831, + "tokens_seen": 1384054784 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002932798395185557, + "loss": 1.34, + "theoretical_loss": 3.5412869091106405, + "tokens_seen": 1384120320 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029326980942828484, + "loss": 1.2806, + "theoretical_loss": 3.541271896819313, + "tokens_seen": 1384185856 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002932597793380141, + "loss": 1.3002, + "theoretical_loss": 3.5412568854377504, + "tokens_seen": 1384251392 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002932497492477432, + "loss": 1.3521, + "theoretical_loss": 3.5412418749658547, + "tokens_seen": 1384316928 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029323971915747244, + "loss": 1.3707, + "theoretical_loss": 3.5412268654035284, + "tokens_seen": 1384382464 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.0011285095242783427, + "objective/train/docs_used": 783958, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.760974407196045, + "objective/train/original_loss": 2.760974884033203, + "objective/train/theoretical_loss": 3.5412118567506727, + "objective/train/tokens_used": 1404908000, + "objective/train/value_avg": -0.005466461181640625, + "objective/train/value_loss": 6.747973384335637e-05, + "objective/train/value_max": -0.00044417381286621094, + "objective/train/value_min": -0.05865478515625, + "objective/train/value_reward_corr": 0.3436343566601887, + "objective/train/value_std": 0.004703521728515625, + "objective/train/weight_avg": 0.991776168346405, + "objective/train/weighted_lm_loss": 2.7417259216308594, + "objective/train/weights_max": 1.7977715730667114, + "objective/train/weights_min": 0.2899157404899597, + "theoretical_loss": 3.5412118567506727, + "tokens_seen": 1384448000 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002932296890672016, + "loss": 1.2964, + "theoretical_loss": 3.5412118567506727, + "tokens_seen": 1384448000 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002932196589769308, + "loss": 1.2804, + "theoretical_loss": 3.5411968490071892, + "tokens_seen": 1384513536 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029320962888666, + "loss": 1.4265, + "theoretical_loss": 3.54118184217298, + "tokens_seen": 1384579072 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029319959879638916, + "loss": 1.4, + "theoretical_loss": 3.541166836247947, + "tokens_seen": 1384644608 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029318956870611835, + "loss": 1.441, + "theoretical_loss": 3.5411518312319923, + "tokens_seen": 1384710144 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002931795386158476, + "loss": 1.4269, + "theoretical_loss": 3.541136827125017, + "tokens_seen": 1384775680 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002931695085255767, + "loss": 1.424, + "theoretical_loss": 3.541121823926925, + "tokens_seen": 1384841216 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029315947843530594, + "loss": 1.1727, + "theoretical_loss": 3.541106821637616, + "tokens_seen": 1384906752 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029314944834503507, + "loss": 1.3364, + "theoretical_loss": 3.5410918202569928, + "tokens_seen": 1384972288 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002931394182547643, + "loss": 1.4306, + "theoretical_loss": 3.541076819784957, + "tokens_seen": 1385037824 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002931293881644935, + "loss": 1.2777, + "theoretical_loss": 3.5410618202214117, + "tokens_seen": 1385103360 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029311935807422267, + "loss": 1.3279, + "theoretical_loss": 3.541046821566258, + "tokens_seen": 1385168896 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029310932798395185, + "loss": 1.3448, + "theoretical_loss": 3.541031823819398, + "tokens_seen": 1385234432 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002930992978936811, + "loss": 1.2407, + "theoretical_loss": 3.5410168269807336, + "tokens_seen": 1385299968 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002930892678034102, + "loss": 1.4412, + "theoretical_loss": 3.541001831050167, + "tokens_seen": 1385365504 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029307923771313945, + "loss": 1.3128, + "theoretical_loss": 3.5409868360276007, + "tokens_seen": 1385431040 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002930692076228686, + "loss": 1.3067, + "theoretical_loss": 3.540971841912936, + "tokens_seen": 1385496576 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002930591775325978, + "loss": 1.258, + "theoretical_loss": 3.5409568487060756, + "tokens_seen": 1385562112 + }, + { + "epoch": 0.42, + "learning_rate": 0.000293049147442327, + "loss": 1.2672, + "theoretical_loss": 3.5409418564069215, + "tokens_seen": 1385627648 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029303911735205617, + "loss": 1.3234, + "theoretical_loss": 3.5409268650153756, + "tokens_seen": 1385693184 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029302908726178535, + "loss": 1.2788, + "theoretical_loss": 3.54091187453134, + "tokens_seen": 1385758720 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029301905717151453, + "loss": 1.3037, + "theoretical_loss": 3.540896884954717, + "tokens_seen": 1385824256 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002930090270812437, + "loss": 1.3511, + "theoretical_loss": 3.5408818962854087, + "tokens_seen": 1385889792 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029299899699097295, + "loss": 1.3477, + "theoretical_loss": 3.5408669085233173, + "tokens_seen": 1385955328 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002929889669007021, + "loss": 1.3411, + "theoretical_loss": 3.540851921668345, + "tokens_seen": 1386020864 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.0021150840912014246, + "objective/train/docs_used": 785103, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7472846508026123, + "objective/train/original_loss": 2.7472846508026123, + "objective/train/theoretical_loss": 3.540836935720394, + "objective/train/tokens_used": 1406546400, + "objective/train/value_avg": -0.005908966064453125, + "objective/train/value_loss": 0.00016067444812506437, + "objective/train/value_max": -0.00021660327911376953, + "objective/train/value_min": -0.05389404296875, + "objective/train/value_reward_corr": 0.3210341366094674, + "objective/train/value_std": 0.004833221435546875, + "objective/train/weight_avg": 0.9852107763290405, + "objective/train/weighted_lm_loss": 2.7112326622009277, + "objective/train/weights_max": 1.6225658655166626, + "objective/train/weights_min": 0.08750110864639282, + "theoretical_loss": 3.540836935720394, + "tokens_seen": 1386086400 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002929789368104313, + "loss": 1.393, + "theoretical_loss": 3.540836935720394, + "tokens_seen": 1386086400 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002929689067201605, + "loss": 1.2973, + "theoretical_loss": 3.5408219506793666, + "tokens_seen": 1386151936 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002929588766298897, + "loss": 1.2699, + "theoretical_loss": 3.5408069665451656, + "tokens_seen": 1386217472 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029294884653961885, + "loss": 1.3849, + "theoretical_loss": 3.5407919833176917, + "tokens_seen": 1386283008 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029293881644934804, + "loss": 1.2412, + "theoretical_loss": 3.5407770009968487, + "tokens_seen": 1386348544 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002929287863590772, + "loss": 1.3974, + "theoretical_loss": 3.5407620195825382, + "tokens_seen": 1386414080 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029291875626880645, + "loss": 1.3009, + "theoretical_loss": 3.5407470390746623, + "tokens_seen": 1386479616 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002929087261785356, + "loss": 1.4648, + "theoretical_loss": 3.5407320594731235, + "tokens_seen": 1386545152 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002928986960882648, + "loss": 1.431, + "theoretical_loss": 3.5407170807778243, + "tokens_seen": 1386610688 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029288866599799394, + "loss": 1.2786, + "theoretical_loss": 3.5407021029886674, + "tokens_seen": 1386676224 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002928786359077232, + "loss": 1.2855, + "theoretical_loss": 3.5406871261055546, + "tokens_seen": 1386741760 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029286860581745236, + "loss": 1.2369, + "theoretical_loss": 3.5406721501283887, + "tokens_seen": 1386807296 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029285857572718154, + "loss": 1.3304, + "theoretical_loss": 3.5406571750570714, + "tokens_seen": 1386872832 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002928485456369107, + "loss": 1.4253, + "theoretical_loss": 3.5406422008915053, + "tokens_seen": 1386938368 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002928385155466399, + "loss": 1.2054, + "theoretical_loss": 3.5406272276315933, + "tokens_seen": 1387003904 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002928284854563691, + "loss": 1.3406, + "theoretical_loss": 3.540612255277238, + "tokens_seen": 1387069440 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002928184553660983, + "loss": 1.3099, + "theoretical_loss": 3.5405972838283413, + "tokens_seen": 1387134976 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029280842527582744, + "loss": 1.2562, + "theoretical_loss": 3.540582313284805, + "tokens_seen": 1387200512 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002927983951855567, + "loss": 1.369, + "theoretical_loss": 3.5405673436465332, + "tokens_seen": 1387266048 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002927883650952859, + "loss": 1.3809, + "theoretical_loss": 3.5405523749134273, + "tokens_seen": 1387331584 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029277833500501504, + "loss": 1.3239, + "theoretical_loss": 3.54053740708539, + "tokens_seen": 1387397120 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002927683049147443, + "loss": 1.4199, + "theoretical_loss": 3.540522440162324, + "tokens_seen": 1387462656 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002927582748244734, + "loss": 1.2457, + "theoretical_loss": 3.540507474144132, + "tokens_seen": 1387528192 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029274824473420264, + "loss": 1.3156, + "theoretical_loss": 3.5404925090307158, + "tokens_seen": 1387593728 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002927382146439318, + "loss": 1.3676, + "theoretical_loss": 3.5404775448219787, + "tokens_seen": 1387659264 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.0008739037439227104, + "objective/train/docs_used": 785794, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.379692792892456, + "objective/train/original_loss": 2.379693031311035, + "objective/train/theoretical_loss": 3.540462581517824, + "objective/train/tokens_used": 1408184800, + "objective/train/value_avg": -0.005435943603515625, + "objective/train/value_loss": 7.261631253641099e-05, + "objective/train/value_max": -0.00023233890533447266, + "objective/train/value_min": -0.06304931640625, + "objective/train/value_reward_corr": 0.28802792560245605, + "objective/train/value_std": 0.004405975341796875, + "objective/train/weight_avg": 0.9946014881134033, + "objective/train/weighted_lm_loss": 2.375328779220581, + "objective/train/weights_max": 1.8785368204116821, + "objective/train/weights_min": 0.2518447935581207, + "theoretical_loss": 3.540462581517824, + "tokens_seen": 1387724800 + }, + { + "epoch": 0.42, + "learning_rate": 0.000292728184553661, + "loss": 1.4568, + "theoretical_loss": 3.540462581517824, + "tokens_seen": 1387724800 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002927181544633902, + "loss": 1.3154, + "theoretical_loss": 3.540447619118152, + "tokens_seen": 1387790336 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029270812437311936, + "loss": 1.3234, + "theoretical_loss": 3.5404326576228673, + "tokens_seen": 1387855872 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029269809428284855, + "loss": 1.3503, + "theoretical_loss": 3.540417697031872, + "tokens_seen": 1387921408 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002926880641925778, + "loss": 1.3217, + "theoretical_loss": 3.540402737345069, + "tokens_seen": 1387986944 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002926780341023069, + "loss": 1.3783, + "theoretical_loss": 3.54038777856236, + "tokens_seen": 1388052480 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029266800401203614, + "loss": 1.3629, + "theoretical_loss": 3.5403728206836487, + "tokens_seen": 1388118016 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029265797392176527, + "loss": 1.3717, + "theoretical_loss": 3.540357863708837, + "tokens_seen": 1388183552 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002926479438314945, + "loss": 1.4072, + "theoretical_loss": 3.5403429076378283, + "tokens_seen": 1388249088 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002926379137412237, + "loss": 1.261, + "theoretical_loss": 3.540327952470525, + "tokens_seen": 1388314624 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029262788365095287, + "loss": 1.3779, + "theoretical_loss": 3.54031299820683, + "tokens_seen": 1388380160 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029261785356068205, + "loss": 1.499, + "theoretical_loss": 3.540298044846646, + "tokens_seen": 1388445696 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002926078234704113, + "loss": 1.434, + "theoretical_loss": 3.540283092389876, + "tokens_seen": 1388511232 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002925977933801404, + "loss": 1.4572, + "theoretical_loss": 3.540268140836422, + "tokens_seen": 1388576768 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029258776328986965, + "loss": 1.2738, + "theoretical_loss": 3.540253190186187, + "tokens_seen": 1388642304 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002925777331995988, + "loss": 1.3062, + "theoretical_loss": 3.5402382404390744, + "tokens_seen": 1388707840 + }, + { + "epoch": 0.42, + "learning_rate": 0.000292567703109328, + "loss": 1.4315, + "theoretical_loss": 3.540223291594987, + "tokens_seen": 1388773376 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002925576730190572, + "loss": 1.4404, + "theoretical_loss": 3.540208343653827, + "tokens_seen": 1388838912 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029254764292878637, + "loss": 1.2403, + "theoretical_loss": 3.540193396615498, + "tokens_seen": 1388904448 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029253761283851555, + "loss": 1.3344, + "theoretical_loss": 3.5401784504799023, + "tokens_seen": 1388969984 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029252758274824473, + "loss": 1.3466, + "theoretical_loss": 3.540163505246943, + "tokens_seen": 1389035520 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002925175526579739, + "loss": 1.4968, + "theoretical_loss": 3.5401485609165233, + "tokens_seen": 1389101056 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029250752256770315, + "loss": 1.369, + "theoretical_loss": 3.5401336174885456, + "tokens_seen": 1389166592 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002924974924774323, + "loss": 1.3898, + "theoretical_loss": 3.5401186749629128, + "tokens_seen": 1389232128 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002924874623871615, + "loss": 1.2303, + "theoretical_loss": 3.540103733339528, + "tokens_seen": 1389297664 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": 0.0002697068848647177, + "objective/train/docs_used": 786491, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6072394847869873, + "objective/train/original_loss": 2.6072394847869873, + "objective/train/theoretical_loss": 3.5400887926182953, + "objective/train/tokens_used": 1409823200, + "objective/train/value_avg": -0.004756927490234375, + "objective/train/value_loss": 3.953425766667351e-05, + "objective/train/value_max": -0.0003998279571533203, + "objective/train/value_min": -0.06622314453125, + "objective/train/value_reward_corr": 0.3963998537345339, + "objective/train/value_std": 0.004222869873046875, + "objective/train/weight_avg": 1.0045878887176514, + "objective/train/weighted_lm_loss": 2.6213862895965576, + "objective/train/weights_max": 1.9391144514083862, + "objective/train/weights_min": 0.5682604312896729, + "theoretical_loss": 3.5400887926182953, + "tokens_seen": 1389363200 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002924774322968907, + "loss": 1.3867, + "theoretical_loss": 3.5400887926182953, + "tokens_seen": 1389363200 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002924674022066199, + "loss": 1.3608, + "theoretical_loss": 3.5400738527991154, + "tokens_seen": 1389428736 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029245737211634906, + "loss": 1.3604, + "theoretical_loss": 3.5400589138818934, + "tokens_seen": 1389494272 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029244734202607824, + "loss": 1.2805, + "theoretical_loss": 3.540043975866531, + "tokens_seen": 1389559808 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002924373119358074, + "loss": 1.3679, + "theoretical_loss": 3.540029038752932, + "tokens_seen": 1389625344 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029242728184553665, + "loss": 1.3464, + "theoretical_loss": 3.540014102540999, + "tokens_seen": 1389690880 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002924172517552658, + "loss": 1.2788, + "theoretical_loss": 3.539999167230635, + "tokens_seen": 1389756416 + }, + { + "epoch": 0.42, + "learning_rate": 0.000292407221664995, + "loss": 1.2989, + "theoretical_loss": 3.5399842328217437, + "tokens_seen": 1389821952 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029239719157472414, + "loss": 1.3322, + "theoretical_loss": 3.539969299314227, + "tokens_seen": 1389887488 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002923871614844534, + "loss": 1.4158, + "theoretical_loss": 3.539954366707989, + "tokens_seen": 1389953024 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029237713139418256, + "loss": 1.3285, + "theoretical_loss": 3.5399394350029327, + "tokens_seen": 1390018560 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029236710130391174, + "loss": 1.3577, + "theoretical_loss": 3.5399245041989613, + "tokens_seen": 1390084096 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002923570712136409, + "loss": 1.2955, + "theoretical_loss": 3.5399095742959776, + "tokens_seen": 1390149632 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002923470411233701, + "loss": 1.3527, + "theoretical_loss": 3.5398946452938844, + "tokens_seen": 1390215168 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002923370110330993, + "loss": 1.3941, + "theoretical_loss": 3.5398797171925858, + "tokens_seen": 1390280704 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002923269809428285, + "loss": 1.3922, + "theoretical_loss": 3.5398647899919844, + "tokens_seen": 1390346240 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029231695085255765, + "loss": 1.474, + "theoretical_loss": 3.5398498636919835, + "tokens_seen": 1390411776 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002923069207622869, + "loss": 1.434, + "theoretical_loss": 3.5398349382924863, + "tokens_seen": 1390477312 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029229689067201606, + "loss": 1.39, + "theoretical_loss": 3.5398200137933964, + "tokens_seen": 1390542848 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029228686058174524, + "loss": 1.4863, + "theoretical_loss": 3.5398050901946165, + "tokens_seen": 1390608384 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002922768304914744, + "loss": 1.3233, + "theoretical_loss": 3.53979016749605, + "tokens_seen": 1390673920 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002922668004012036, + "loss": 1.3794, + "theoretical_loss": 3.5397752456976006, + "tokens_seen": 1390739456 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002922567703109328, + "loss": 1.3747, + "theoretical_loss": 3.539760324799171, + "tokens_seen": 1390804992 + }, + { + "epoch": 0.42, + "learning_rate": 0.000292246740220662, + "loss": 1.3544, + "theoretical_loss": 3.539745404800665, + "tokens_seen": 1390870528 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029223671013039115, + "loss": 1.3889, + "theoretical_loss": 3.539730485701985, + "tokens_seen": 1390936064 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.0003267198917455971, + "objective/train/docs_used": 787676, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.720022439956665, + "objective/train/original_loss": 2.720022439956665, + "objective/train/theoretical_loss": 3.539715567503036, + "objective/train/tokens_used": 1411461600, + "objective/train/value_avg": -0.0056915283203125, + "objective/train/value_loss": 4.245503805577755e-05, + "objective/train/value_max": -0.0003249645233154297, + "objective/train/value_min": -0.09088134765625, + "objective/train/value_reward_corr": 0.4482572790776903, + "objective/train/value_std": 0.00464630126953125, + "objective/train/weight_avg": 0.9988225102424622, + "objective/train/weighted_lm_loss": 2.7141470909118652, + "objective/train/weights_max": 2.220433473587036, + "objective/train/weights_min": 0.602498471736908, + "theoretical_loss": 3.539715567503036, + "tokens_seen": 1391001600 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002922266800401204, + "loss": 1.3095, + "theoretical_loss": 3.539715567503036, + "tokens_seen": 1391001600 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002922166499498495, + "loss": 1.2055, + "theoretical_loss": 3.53970065020372, + "tokens_seen": 1391067136 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029220661985957875, + "loss": 1.2781, + "theoretical_loss": 3.5396857338039407, + "tokens_seen": 1391132672 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002921965897693079, + "loss": 1.235, + "theoretical_loss": 3.5396708183036014, + "tokens_seen": 1391198208 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002921865596790371, + "loss": 1.3337, + "theoretical_loss": 3.539655903702606, + "tokens_seen": 1391263744 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002921765295887663, + "loss": 1.258, + "theoretical_loss": 3.5396409900008576, + "tokens_seen": 1391329280 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029216649949849547, + "loss": 1.4029, + "theoretical_loss": 3.539626077198259, + "tokens_seen": 1391394816 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029215646940822465, + "loss": 1.4265, + "theoretical_loss": 3.5396111652947146, + "tokens_seen": 1391460352 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002921464393179539, + "loss": 1.3219, + "theoretical_loss": 3.539596254290128, + "tokens_seen": 1391525888 + }, + { + "epoch": 0.42, + "learning_rate": 0.000292136409227683, + "loss": 1.4662, + "theoretical_loss": 3.5395813441844015, + "tokens_seen": 1391591424 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029212637913741225, + "loss": 1.3445, + "theoretical_loss": 3.5395664349774396, + "tokens_seen": 1391656960 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029211634904714143, + "loss": 1.4218, + "theoretical_loss": 3.539551526669145, + "tokens_seen": 1391722496 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002921063189568706, + "loss": 1.2803, + "theoretical_loss": 3.5395366192594224, + "tokens_seen": 1391788032 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002920962888665998, + "loss": 1.397, + "theoretical_loss": 3.539521712748174, + "tokens_seen": 1391853568 + }, + { + "epoch": 0.42, + "learning_rate": 0.000292086258776329, + "loss": 1.2652, + "theoretical_loss": 3.5395068071353037, + "tokens_seen": 1391919104 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029207622868605815, + "loss": 1.35, + "theoretical_loss": 3.539491902420716, + "tokens_seen": 1391984640 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002920661985957874, + "loss": 1.282, + "theoretical_loss": 3.5394769986043135, + "tokens_seen": 1392050176 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002920561685055165, + "loss": 1.3874, + "theoretical_loss": 3.5394620956860003, + "tokens_seen": 1392115712 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029204613841524575, + "loss": 1.381, + "theoretical_loss": 3.5394471936656795, + "tokens_seen": 1392181248 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029203610832497493, + "loss": 1.4213, + "theoretical_loss": 3.539432292543255, + "tokens_seen": 1392246784 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002920260782347041, + "loss": 1.2976, + "theoretical_loss": 3.5394173923186307, + "tokens_seen": 1392312320 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029201604814443335, + "loss": 1.4031, + "theoretical_loss": 3.53940249299171, + "tokens_seen": 1392377856 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002920060180541625, + "loss": 1.3201, + "theoretical_loss": 3.5393875945623963, + "tokens_seen": 1392443392 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002919959879638917, + "loss": 1.3086, + "theoretical_loss": 3.539372697030594, + "tokens_seen": 1392508928 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002919859578736209, + "loss": 1.3123, + "theoretical_loss": 3.539357800396206, + "tokens_seen": 1392574464 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.00031404144829139113, + "objective/train/docs_used": 788318, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.228451728820801, + "objective/train/original_loss": 2.2284512519836426, + "objective/train/theoretical_loss": 3.5393429046591365, + "objective/train/tokens_used": 1413100000, + "objective/train/value_avg": -0.00534820556640625, + "objective/train/value_loss": 5.4431995522463694e-05, + "objective/train/value_max": -0.00047278404235839844, + "objective/train/value_min": -0.061187744140625, + "objective/train/value_reward_corr": 0.4688153428701195, + "objective/train/value_std": 0.004283905029296875, + "objective/train/weight_avg": 0.999444305896759, + "objective/train/weighted_lm_loss": 2.209174394607544, + "objective/train/weights_max": 1.7770898342132568, + "objective/train/weights_min": 0.5037786960601807, + "theoretical_loss": 3.5393429046591365, + "tokens_seen": 1392640000 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002919759277833501, + "loss": 1.34, + "theoretical_loss": 3.5393429046591365, + "tokens_seen": 1392640000 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029196589769307926, + "loss": 1.3574, + "theoretical_loss": 3.5393280098192887, + "tokens_seen": 1392705536 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029195586760280844, + "loss": 1.2736, + "theoretical_loss": 3.539313115876567, + "tokens_seen": 1392771072 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002919458375125376, + "loss": 1.3285, + "theoretical_loss": 3.539298222830875, + "tokens_seen": 1392836608 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029193580742226685, + "loss": 1.4656, + "theoretical_loss": 3.5392833306821165, + "tokens_seen": 1392902144 + }, + { + "epoch": 0.42, + "learning_rate": 0.000291925777331996, + "loss": 1.3959, + "theoretical_loss": 3.5392684394301943, + "tokens_seen": 1392967680 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002919157472417252, + "loss": 1.4019, + "theoretical_loss": 3.539253549075014, + "tokens_seen": 1393033216 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029190571715145434, + "loss": 1.3484, + "theoretical_loss": 3.539238659616478, + "tokens_seen": 1393098752 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002918956870611836, + "loss": 1.3424, + "theoretical_loss": 3.5392237710544907, + "tokens_seen": 1393164288 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029188565697091276, + "loss": 1.296, + "theoretical_loss": 3.5392088833889557, + "tokens_seen": 1393229824 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029187562688064194, + "loss": 1.4045, + "theoretical_loss": 3.539193996619777, + "tokens_seen": 1393295360 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002918655967903711, + "loss": 1.3658, + "theoretical_loss": 3.5391791107468586, + "tokens_seen": 1393360896 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002918555667001003, + "loss": 1.3747, + "theoretical_loss": 3.539164225770104, + "tokens_seen": 1393426432 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002918455366098295, + "loss": 1.3179, + "theoretical_loss": 3.5391493416894173, + "tokens_seen": 1393491968 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002918355065195587, + "loss": 1.3487, + "theoretical_loss": 3.5391344585047024, + "tokens_seen": 1393557504 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029182547642928785, + "loss": 1.4021, + "theoretical_loss": 3.539119576215864, + "tokens_seen": 1393623040 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002918154463390171, + "loss": 1.3534, + "theoretical_loss": 3.5391046948228047, + "tokens_seen": 1393688576 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029180541624874626, + "loss": 1.2955, + "theoretical_loss": 3.53908981432543, + "tokens_seen": 1393754112 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029179538615847544, + "loss": 1.2955, + "theoretical_loss": 3.539074934723642, + "tokens_seen": 1393819648 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002917853560682046, + "loss": 1.353, + "theoretical_loss": 3.5390600560173455, + "tokens_seen": 1393885184 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002917753259779338, + "loss": 1.2649, + "theoretical_loss": 3.539045178206446, + "tokens_seen": 1393950720 + }, + { + "epoch": 0.42, + "learning_rate": 0.000291765295887663, + "loss": 1.2818, + "theoretical_loss": 3.539030301290845, + "tokens_seen": 1394016256 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002917552657973922, + "loss": 1.3742, + "theoretical_loss": 3.539015425270448, + "tokens_seen": 1394081792 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029174523570712135, + "loss": 1.3608, + "theoretical_loss": 3.5390005501451594, + "tokens_seen": 1394147328 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002917352056168506, + "loss": 1.3741, + "theoretical_loss": 3.538985675914882, + "tokens_seen": 1394212864 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.0004957691417075694, + "objective/train/docs_used": 789787, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.044466972351074, + "objective/train/original_loss": 3.0444674491882324, + "objective/train/theoretical_loss": 3.538970802579521, + "objective/train/tokens_used": 1414738400, + "objective/train/value_avg": -0.004955291748046875, + "objective/train/value_loss": 5.026054714107886e-05, + "objective/train/value_max": -0.0002434253692626953, + "objective/train/value_min": -0.060089111328125, + "objective/train/value_reward_corr": 0.4383443841599405, + "objective/train/value_std": 0.00409698486328125, + "objective/train/weight_avg": 0.9974206686019897, + "objective/train/weighted_lm_loss": 3.0355136394500732, + "objective/train/weights_max": 1.6546136140823364, + "objective/train/weights_min": 0.41221022605895996, + "theoretical_loss": 3.538970802579521, + "tokens_seen": 1394278400 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002917251755265797, + "loss": 1.4009, + "theoretical_loss": 3.538970802579521, + "tokens_seen": 1394278400 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029171514543630895, + "loss": 1.3895, + "theoretical_loss": 3.5389559301389797, + "tokens_seen": 1394343936 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002917051153460381, + "loss": 1.3993, + "theoretical_loss": 3.5389410585931627, + "tokens_seen": 1394409472 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002916950852557673, + "loss": 1.2293, + "theoretical_loss": 3.538926187941974, + "tokens_seen": 1394475008 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002916850551654965, + "loss": 1.2765, + "theoretical_loss": 3.538911318185318, + "tokens_seen": 1394540544 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029167502507522567, + "loss": 1.3667, + "theoretical_loss": 3.538896449323098, + "tokens_seen": 1394606080 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029166499498495485, + "loss": 1.3782, + "theoretical_loss": 3.538881581355219, + "tokens_seen": 1394671616 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002916549648946841, + "loss": 1.2705, + "theoretical_loss": 3.5388667142815855, + "tokens_seen": 1394737152 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002916449348044132, + "loss": 1.2645, + "theoretical_loss": 3.5388518481021003, + "tokens_seen": 1394802688 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029163490471414245, + "loss": 1.2381, + "theoretical_loss": 3.538836982816669, + "tokens_seen": 1394868224 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029162487462387163, + "loss": 1.2984, + "theoretical_loss": 3.5388221184251947, + "tokens_seen": 1394933760 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002916148445336008, + "loss": 1.3717, + "theoretical_loss": 3.538807254927583, + "tokens_seen": 1394999296 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029160481444333, + "loss": 1.2664, + "theoretical_loss": 3.538792392323737, + "tokens_seen": 1395064832 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002915947843530592, + "loss": 1.1054, + "theoretical_loss": 3.538777530613561, + "tokens_seen": 1395130368 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029158475426278835, + "loss": 1.395, + "theoretical_loss": 3.5387626697969603, + "tokens_seen": 1395195904 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002915747241725176, + "loss": 1.3292, + "theoretical_loss": 3.538747809873838, + "tokens_seen": 1395261440 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002915646940822467, + "loss": 1.3775, + "theoretical_loss": 3.5387329508440994, + "tokens_seen": 1395326976 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029155466399197595, + "loss": 1.3334, + "theoretical_loss": 3.538718092707648, + "tokens_seen": 1395392512 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002915446339017051, + "loss": 1.4157, + "theoretical_loss": 3.5387032354643893, + "tokens_seen": 1395458048 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002915346038114343, + "loss": 1.3819, + "theoretical_loss": 3.538688379114226, + "tokens_seen": 1395523584 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002915245737211635, + "loss": 1.2517, + "theoretical_loss": 3.538673523657064, + "tokens_seen": 1395589120 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002915145436308927, + "loss": 1.3868, + "theoretical_loss": 3.538658669092807, + "tokens_seen": 1395654656 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029150451354062186, + "loss": 1.3243, + "theoretical_loss": 3.5386438154213593, + "tokens_seen": 1395720192 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002914944834503511, + "loss": 1.3998, + "theoretical_loss": 3.5386289626426253, + "tokens_seen": 1395785728 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002914844533600802, + "loss": 1.3532, + "theoretical_loss": 3.53861411075651, + "tokens_seen": 1395851264 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": 0.00028351510991342366, + "objective/train/docs_used": 790377, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6869778633117676, + "objective/train/original_loss": 2.6869773864746094, + "objective/train/theoretical_loss": 3.5385992597629174, + "objective/train/tokens_used": 1416376800, + "objective/train/value_avg": -0.005901336669921875, + "objective/train/value_loss": 4.265982352080755e-05, + "objective/train/value_max": -0.0003249645233154297, + "objective/train/value_min": -0.0560302734375, + "objective/train/value_reward_corr": 0.4688471743534392, + "objective/train/value_std": 0.004566192626953125, + "objective/train/weight_avg": 1.0049128532409668, + "objective/train/weighted_lm_loss": 2.6993513107299805, + "objective/train/weights_max": 1.3945690393447876, + "objective/train/weights_min": 0.597406268119812, + "theoretical_loss": 3.5385992597629174, + "tokens_seen": 1395916800 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029147442326980946, + "loss": 1.2437, + "theoretical_loss": 3.5385992597629174, + "tokens_seen": 1395916800 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002914643931795386, + "loss": 1.335, + "theoretical_loss": 3.5385844096617523, + "tokens_seen": 1395982336 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002914543630892678, + "loss": 1.3305, + "theoretical_loss": 3.5385695604529186, + "tokens_seen": 1396047872 + }, + { + "epoch": 0.42, + "learning_rate": 0.000291444332998997, + "loss": 1.2804, + "theoretical_loss": 3.538554712136321, + "tokens_seen": 1396113408 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002914343029087262, + "loss": 1.4022, + "theoretical_loss": 3.538539864711865, + "tokens_seen": 1396178944 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029142427281845536, + "loss": 1.4709, + "theoretical_loss": 3.5385250181794534, + "tokens_seen": 1396244480 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029141424272818454, + "loss": 1.3849, + "theoretical_loss": 3.5385101725389916, + "tokens_seen": 1396310016 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002914042126379137, + "loss": 1.3351, + "theoretical_loss": 3.5384953277903843, + "tokens_seen": 1396375552 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029139418254764296, + "loss": 1.3603, + "theoretical_loss": 3.5384804839335366, + "tokens_seen": 1396441088 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002913841524573721, + "loss": 1.3194, + "theoretical_loss": 3.5384656409683517, + "tokens_seen": 1396506624 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002913741223671013, + "loss": 1.3777, + "theoretical_loss": 3.5384507988947353, + "tokens_seen": 1396572160 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029136409227683045, + "loss": 1.2334, + "theoretical_loss": 3.5384359577125912, + "tokens_seen": 1396637696 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002913540621865597, + "loss": 1.3808, + "theoretical_loss": 3.5384211174218247, + "tokens_seen": 1396703232 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029134403209628886, + "loss": 1.2904, + "theoretical_loss": 3.5384062780223404, + "tokens_seen": 1396768768 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029133400200601805, + "loss": 1.3094, + "theoretical_loss": 3.538391439514042, + "tokens_seen": 1396834304 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002913239719157472, + "loss": 1.3095, + "theoretical_loss": 3.538376601896836, + "tokens_seen": 1396899840 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029131394182547646, + "loss": 1.3772, + "theoretical_loss": 3.538361765170625, + "tokens_seen": 1396965376 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002913039117352056, + "loss": 1.3579, + "theoretical_loss": 3.538346929335315, + "tokens_seen": 1397030912 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002912938816449348, + "loss": 1.2288, + "theoretical_loss": 3.5383320943908108, + "tokens_seen": 1397096448 + }, + { + "epoch": 0.42, + "learning_rate": 0.000291283851554664, + "loss": 1.424, + "theoretical_loss": 3.5383172603370165, + "tokens_seen": 1397161984 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002912738214643932, + "loss": 1.2825, + "theoretical_loss": 3.538302427173837, + "tokens_seen": 1397227520 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002912637913741224, + "loss": 1.3598, + "theoretical_loss": 3.538287594901177, + "tokens_seen": 1397293056 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029125376128385155, + "loss": 1.3412, + "theoretical_loss": 3.538272763518942, + "tokens_seen": 1397358592 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002912437311935808, + "loss": 1.3307, + "theoretical_loss": 3.538257933027036, + "tokens_seen": 1397424128 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002912337011033099, + "loss": 1.3274, + "theoretical_loss": 3.538243103425364, + "tokens_seen": 1397489664 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": 0.00033362608519382775, + "objective/train/docs_used": 791674, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8487918376922607, + "objective/train/original_loss": 2.84879207611084, + "objective/train/theoretical_loss": 3.5382282747138305, + "objective/train/tokens_used": 1418015200, + "objective/train/value_avg": -0.0052490234375, + "objective/train/value_loss": 5.404349940363318e-05, + "objective/train/value_max": -0.00021827220916748047, + "objective/train/value_min": -0.07611083984375, + "objective/train/value_reward_corr": 0.5038729562737606, + "objective/train/value_std": 0.00485992431640625, + "objective/train/weight_avg": 1.0058915615081787, + "objective/train/weighted_lm_loss": 2.863621473312378, + "objective/train/weights_max": 1.7700260877609253, + "objective/train/weights_min": 0.29566484689712524, + "theoretical_loss": 3.5382282747138305, + "tokens_seen": 1397555200 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029122367101303915, + "loss": 1.3063, + "theoretical_loss": 3.5382282747138305, + "tokens_seen": 1397555200 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029121364092276833, + "loss": 1.4059, + "theoretical_loss": 3.5382134468923407, + "tokens_seen": 1397620736 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002912036108324975, + "loss": 1.3877, + "theoretical_loss": 3.5381986199607995, + "tokens_seen": 1397686272 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002911935807422267, + "loss": 1.3427, + "theoretical_loss": 3.538183793919112, + "tokens_seen": 1397751808 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029118355065195587, + "loss": 1.4425, + "theoretical_loss": 3.5381689687671822, + "tokens_seen": 1397817344 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029117352056168505, + "loss": 1.3352, + "theoretical_loss": 3.538154144504916, + "tokens_seen": 1397882880 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002911634904714143, + "loss": 1.3653, + "theoretical_loss": 3.538139321132218, + "tokens_seen": 1397948416 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002911534603811434, + "loss": 1.3543, + "theoretical_loss": 3.538124498648992, + "tokens_seen": 1398013952 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029114343029087265, + "loss": 1.383, + "theoretical_loss": 3.538109677055145, + "tokens_seen": 1398079488 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029113340020060183, + "loss": 1.3823, + "theoretical_loss": 3.5380948563505803, + "tokens_seen": 1398145024 + }, + { + "epoch": 0.42, + "learning_rate": 0.000291123370110331, + "loss": 1.4219, + "theoretical_loss": 3.538080036535203, + "tokens_seen": 1398210560 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002911133400200602, + "loss": 1.3355, + "theoretical_loss": 3.5380652176089193, + "tokens_seen": 1398276096 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002911033099297894, + "loss": 1.3136, + "theoretical_loss": 3.5380503995716333, + "tokens_seen": 1398341632 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029109327983951855, + "loss": 1.3016, + "theoretical_loss": 3.53803558242325, + "tokens_seen": 1398407168 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002910832497492478, + "loss": 1.2178, + "theoretical_loss": 3.5380207661636742, + "tokens_seen": 1398472704 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002910732196589769, + "loss": 1.2454, + "theoretical_loss": 3.538005950792811, + "tokens_seen": 1398538240 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029106318956870615, + "loss": 1.3191, + "theoretical_loss": 3.5379911363105663, + "tokens_seen": 1398603776 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002910531594784353, + "loss": 1.3909, + "theoretical_loss": 3.537976322716845, + "tokens_seen": 1398669312 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002910431293881645, + "loss": 1.3436, + "theoretical_loss": 3.5379615100115505, + "tokens_seen": 1398734848 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002910330992978937, + "loss": 1.3565, + "theoretical_loss": 3.53794669819459, + "tokens_seen": 1398800384 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002910230692076229, + "loss": 1.4652, + "theoretical_loss": 3.5379318872658674, + "tokens_seen": 1398865920 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029101303911735206, + "loss": 1.3157, + "theoretical_loss": 3.5379170772252886, + "tokens_seen": 1398931456 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002910030090270813, + "loss": 1.2419, + "theoretical_loss": 3.5379022680727576, + "tokens_seen": 1398996992 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002909929789368104, + "loss": 1.355, + "theoretical_loss": 3.537887459808181, + "tokens_seen": 1399062528 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029098294884653966, + "loss": 1.4274, + "theoretical_loss": 3.5378726524314628, + "tokens_seen": 1399128064 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": 0.0007006326923146844, + "objective/train/docs_used": 792315, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8502957820892334, + "objective/train/original_loss": 2.8502955436706543, + "objective/train/theoretical_loss": 3.5378578459425087, + "objective/train/tokens_used": 1419653600, + "objective/train/value_avg": -0.005374908447265625, + "objective/train/value_loss": 4.9657232011668384e-05, + "objective/train/value_max": -0.000415802001953125, + "objective/train/value_min": -0.06304931640625, + "objective/train/value_reward_corr": 0.39409374844926226, + "objective/train/value_std": 0.004268646240234375, + "objective/train/weight_avg": 1.0093505382537842, + "objective/train/weighted_lm_loss": 2.8765716552734375, + "objective/train/weights_max": 1.5609707832336426, + "objective/train/weights_min": 0.19936609268188477, + "theoretical_loss": 3.5378578459425087, + "tokens_seen": 1399193600 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002909729187562688, + "loss": 1.3754, + "theoretical_loss": 3.5378578459425087, + "tokens_seen": 1399193600 + }, + { + "epoch": 0.42, + "learning_rate": 0.000290962888665998, + "loss": 1.3209, + "theoretical_loss": 3.537843040341224, + "tokens_seen": 1399259136 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002909528585757272, + "loss": 1.3486, + "theoretical_loss": 3.5378282356275137, + "tokens_seen": 1399324672 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002909428284854564, + "loss": 1.4568, + "theoretical_loss": 3.5378134318012826, + "tokens_seen": 1399390208 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029093279839518556, + "loss": 1.3985, + "theoretical_loss": 3.5377986288624372, + "tokens_seen": 1399455744 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029092276830491474, + "loss": 1.3468, + "theoretical_loss": 3.537783826810881, + "tokens_seen": 1399521280 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002909127382146439, + "loss": 1.33, + "theoretical_loss": 3.5377690256465213, + "tokens_seen": 1399586816 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029090270812437316, + "loss": 1.3525, + "theoretical_loss": 3.5377542253692615, + "tokens_seen": 1399652352 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002908926780341023, + "loss": 1.3305, + "theoretical_loss": 3.537739425979008, + "tokens_seen": 1399717888 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002908826479438315, + "loss": 1.3782, + "theoretical_loss": 3.537724627475666, + "tokens_seen": 1399783424 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029087261785356065, + "loss": 1.3241, + "theoretical_loss": 3.53770982985914, + "tokens_seen": 1399848960 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002908625877632899, + "loss": 1.383, + "theoretical_loss": 3.537695033129336, + "tokens_seen": 1399914496 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029085255767301906, + "loss": 1.4338, + "theoretical_loss": 3.5376802372861604, + "tokens_seen": 1399980032 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029084252758274825, + "loss": 1.309, + "theoretical_loss": 3.5376654423295166, + "tokens_seen": 1400045568 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002908324974924774, + "loss": 1.3194, + "theoretical_loss": 3.5376506482593113, + "tokens_seen": 1400111104 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029082246740220666, + "loss": 1.3778, + "theoretical_loss": 3.537635855075449, + "tokens_seen": 1400176640 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002908124373119358, + "loss": 1.3645, + "theoretical_loss": 3.537621062777836, + "tokens_seen": 1400242176 + }, + { + "epoch": 0.42, + "learning_rate": 0.000290802407221665, + "loss": 1.3371, + "theoretical_loss": 3.5376062713663776, + "tokens_seen": 1400307712 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029079237713139415, + "loss": 1.2081, + "theoretical_loss": 3.5375914808409785, + "tokens_seen": 1400373248 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002907823470411234, + "loss": 1.4157, + "theoretical_loss": 3.5375766912015445, + "tokens_seen": 1400438784 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029077231695085257, + "loss": 1.4268, + "theoretical_loss": 3.537561902447982, + "tokens_seen": 1400504320 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029076228686058175, + "loss": 1.3335, + "theoretical_loss": 3.537547114580195, + "tokens_seen": 1400569856 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029075225677031093, + "loss": 1.4573, + "theoretical_loss": 3.53753232759809, + "tokens_seen": 1400635392 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002907422266800401, + "loss": 1.3851, + "theoretical_loss": 3.5375175415015723, + "tokens_seen": 1400700928 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002907321965897693, + "loss": 1.3091, + "theoretical_loss": 3.5375027562905474, + "tokens_seen": 1400766464 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.0010856686858460307, + "objective/train/docs_used": 793814, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8064305782318115, + "objective/train/original_loss": 2.8064305782318115, + "objective/train/theoretical_loss": 3.5374879719649206, + "objective/train/tokens_used": 1421292000, + "objective/train/value_avg": -0.0052642822265625, + "objective/train/value_loss": 6.557050801347941e-05, + "objective/train/value_max": -0.0004239082336425781, + "objective/train/value_min": -0.05450439453125, + "objective/train/value_reward_corr": 0.25521951340322896, + "objective/train/value_std": 0.0038585662841796875, + "objective/train/weight_avg": 0.9921828508377075, + "objective/train/weighted_lm_loss": 2.7885446548461914, + "objective/train/weights_max": 1.660638451576233, + "objective/train/weights_min": 0.42266377806663513, + "theoretical_loss": 3.5374879719649206, + "tokens_seen": 1400832000 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029072216649949853, + "loss": 1.3716, + "theoretical_loss": 3.5374879719649206, + "tokens_seen": 1400832000 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029071213640922765, + "loss": 1.3284, + "theoretical_loss": 3.537473188524598, + "tokens_seen": 1400897536 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002907021063189569, + "loss": 1.434, + "theoretical_loss": 3.5374584059694847, + "tokens_seen": 1400963072 + }, + { + "epoch": 0.42, + "learning_rate": 0.000290692076228686, + "loss": 1.2772, + "theoretical_loss": 3.5374436242994864, + "tokens_seen": 1401028608 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029068204613841525, + "loss": 1.3565, + "theoretical_loss": 3.5374288435145083, + "tokens_seen": 1401094144 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029067201604814443, + "loss": 1.3581, + "theoretical_loss": 3.537414063614457, + "tokens_seen": 1401159680 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002906619859578736, + "loss": 1.2867, + "theoretical_loss": 3.5373992845992372, + "tokens_seen": 1401225216 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002906519558676028, + "loss": 1.3979, + "theoretical_loss": 3.5373845064687552, + "tokens_seen": 1401290752 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029064192577733203, + "loss": 1.3394, + "theoretical_loss": 3.5373697292229167, + "tokens_seen": 1401356288 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029063189568706116, + "loss": 1.3572, + "theoretical_loss": 3.5373549528616266, + "tokens_seen": 1401421824 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002906218655967904, + "loss": 1.3702, + "theoretical_loss": 3.5373401773847912, + "tokens_seen": 1401487360 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002906118355065195, + "loss": 1.3889, + "theoretical_loss": 3.537325402792316, + "tokens_seen": 1401552896 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029060180541624876, + "loss": 1.307, + "theoretical_loss": 3.5373106290841068, + "tokens_seen": 1401618432 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029059177532597794, + "loss": 1.3158, + "theoretical_loss": 3.5372958562600694, + "tokens_seen": 1401683968 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002905817452357071, + "loss": 1.3707, + "theoretical_loss": 3.5372810843201097, + "tokens_seen": 1401749504 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002905717151454363, + "loss": 1.3635, + "theoretical_loss": 3.5372663132641327, + "tokens_seen": 1401815040 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002905616850551655, + "loss": 1.2991, + "theoretical_loss": 3.537251543092045, + "tokens_seen": 1401880576 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029055165496489466, + "loss": 1.3586, + "theoretical_loss": 3.5372367738037527, + "tokens_seen": 1401946112 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002905416248746239, + "loss": 1.2007, + "theoretical_loss": 3.53722200539916, + "tokens_seen": 1402011648 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002905315947843531, + "loss": 1.3969, + "theoretical_loss": 3.537207237878174, + "tokens_seen": 1402077184 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029052156469408226, + "loss": 1.1946, + "theoretical_loss": 3.537192471240701, + "tokens_seen": 1402142720 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002905115346038115, + "loss": 1.3114, + "theoretical_loss": 3.537177705486645, + "tokens_seen": 1402208256 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002905015045135406, + "loss": 1.3448, + "theoretical_loss": 3.5371629406159135, + "tokens_seen": 1402273792 + }, + { + "epoch": 0.42, + "learning_rate": 0.00029049147442326986, + "loss": 1.2947, + "theoretical_loss": 3.537148176628412, + "tokens_seen": 1402339328 + }, + { + "epoch": 0.42, + "learning_rate": 0.000290481444332999, + "loss": 1.3638, + "theoretical_loss": 3.537133413524046, + "tokens_seen": 1402404864 + }, + { + "epoch": 0.42, + "objective/train/advantage_avg": -0.0003912511747330427, + "objective/train/docs_used": 794381, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.73787784576416, + "objective/train/original_loss": 2.737877130508423, + "objective/train/theoretical_loss": 3.537118651302722, + "objective/train/tokens_used": 1422930400, + "objective/train/value_avg": -0.004421234130859375, + "objective/train/value_loss": 3.810754787991755e-05, + "objective/train/value_max": -0.00019407272338867188, + "objective/train/value_min": -0.040618896484375, + "objective/train/value_reward_corr": 0.39846004198399765, + "objective/train/value_std": 0.0033512115478515625, + "objective/train/weight_avg": 0.9979098439216614, + "objective/train/weighted_lm_loss": 2.7451295852661133, + "objective/train/weights_max": 1.3060780763626099, + "objective/train/weights_min": 0.45167794823646545, + "theoretical_loss": 3.537118651302722, + "tokens_seen": 1402470400 + }, + { + "epoch": 0.42, + "learning_rate": 0.0002904714142427282, + "loss": 1.3196, + "theoretical_loss": 3.537118651302722, + "tokens_seen": 1402470400 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002904613841524574, + "loss": 1.3765, + "theoretical_loss": 3.537103889964345, + "tokens_seen": 1402535936 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002904513540621866, + "loss": 1.4138, + "theoretical_loss": 3.5370891295088223, + "tokens_seen": 1402601472 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029044132397191576, + "loss": 1.2381, + "theoretical_loss": 3.5370743699360583, + "tokens_seen": 1402667008 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029043129388164494, + "loss": 1.3796, + "theoretical_loss": 3.5370596112459607, + "tokens_seen": 1402732544 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002904212637913741, + "loss": 1.4297, + "theoretical_loss": 3.5370448534384336, + "tokens_seen": 1402798080 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029041123370110336, + "loss": 1.4548, + "theoretical_loss": 3.537030096513384, + "tokens_seen": 1402863616 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002904012036108325, + "loss": 1.4391, + "theoretical_loss": 3.537015340470718, + "tokens_seen": 1402929152 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002903911735205617, + "loss": 1.2535, + "theoretical_loss": 3.5370005853103414, + "tokens_seen": 1402994688 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029038114343029085, + "loss": 1.3927, + "theoretical_loss": 3.5369858310321605, + "tokens_seen": 1403060224 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002903711133400201, + "loss": 1.3781, + "theoretical_loss": 3.536971077636081, + "tokens_seen": 1403125760 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029036108324974926, + "loss": 1.3158, + "theoretical_loss": 3.536956325122009, + "tokens_seen": 1403191296 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029035105315947845, + "loss": 1.4294, + "theoretical_loss": 3.536941573489851, + "tokens_seen": 1403256832 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002903410230692076, + "loss": 1.4689, + "theoretical_loss": 3.5369268227395123, + "tokens_seen": 1403322368 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029033099297893686, + "loss": 1.3686, + "theoretical_loss": 3.5369120728708996, + "tokens_seen": 1403387904 + }, + { + "epoch": 0.43, + "learning_rate": 0.000290320962888666, + "loss": 1.259, + "theoretical_loss": 3.5368973238839185, + "tokens_seen": 1403453440 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002903109327983952, + "loss": 1.3337, + "theoretical_loss": 3.5368825757784763, + "tokens_seen": 1403518976 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029030090270812435, + "loss": 1.3349, + "theoretical_loss": 3.536867828554478, + "tokens_seen": 1403584512 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002902908726178536, + "loss": 1.394, + "theoretical_loss": 3.53685308221183, + "tokens_seen": 1403650048 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029028084252758277, + "loss": 1.3118, + "theoretical_loss": 3.536838336750439, + "tokens_seen": 1403715584 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029027081243731195, + "loss": 1.3443, + "theoretical_loss": 3.5368235921702107, + "tokens_seen": 1403781120 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029026078234704113, + "loss": 1.4098, + "theoretical_loss": 3.5368088484710514, + "tokens_seen": 1403846656 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002902507522567703, + "loss": 1.3386, + "theoretical_loss": 3.5367941056528673, + "tokens_seen": 1403912192 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002902407221664995, + "loss": 1.3336, + "theoretical_loss": 3.536779363715565, + "tokens_seen": 1403977728 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029023069207622873, + "loss": 1.2801, + "theoretical_loss": 3.5367646226590503, + "tokens_seen": 1404043264 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0008476286311633885, + "objective/train/docs_used": 795506, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5866925716400146, + "objective/train/original_loss": 2.5866923332214355, + "objective/train/theoretical_loss": 3.536749882483229, + "objective/train/tokens_used": 1424568800, + "objective/train/value_avg": -0.00656890869140625, + "objective/train/value_loss": 0.0001241923018824309, + "objective/train/value_max": -0.00030541419982910156, + "objective/train/value_min": -0.09368896484375, + "objective/train/value_reward_corr": 0.607929357390809, + "objective/train/value_std": 0.006847381591796875, + "objective/train/weight_avg": 0.9970483779907227, + "objective/train/weighted_lm_loss": 2.5769426822662354, + "objective/train/weights_max": 2.4344475269317627, + "objective/train/weights_min": 0.25372514128685, + "theoretical_loss": 3.536749882483229, + "tokens_seen": 1404108800 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029022066198595785, + "loss": 1.3454, + "theoretical_loss": 3.536749882483229, + "tokens_seen": 1404108800 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002902106318956871, + "loss": 1.3365, + "theoretical_loss": 3.536735143188009, + "tokens_seen": 1404174336 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002902006018054162, + "loss": 1.3306, + "theoretical_loss": 3.536720404773295, + "tokens_seen": 1404239872 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029019057171514545, + "loss": 1.3606, + "theoretical_loss": 3.5367056672389943, + "tokens_seen": 1404305408 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029018054162487463, + "loss": 1.3882, + "theoretical_loss": 3.5366909305850127, + "tokens_seen": 1404370944 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002901705115346038, + "loss": 1.4128, + "theoretical_loss": 3.5366761948112564, + "tokens_seen": 1404436480 + }, + { + "epoch": 0.43, + "learning_rate": 0.000290160481444333, + "loss": 1.352, + "theoretical_loss": 3.536661459917632, + "tokens_seen": 1404502016 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029015045135406223, + "loss": 1.3726, + "theoretical_loss": 3.5366467259040464, + "tokens_seen": 1404567552 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029014042126379136, + "loss": 1.4234, + "theoretical_loss": 3.536631992770405, + "tokens_seen": 1404633088 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002901303911735206, + "loss": 1.3353, + "theoretical_loss": 3.536617260516615, + "tokens_seen": 1404698624 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002901203610832497, + "loss": 1.344, + "theoretical_loss": 3.536602529142582, + "tokens_seen": 1404764160 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029011033099297896, + "loss": 1.3913, + "theoretical_loss": 3.5365877986482133, + "tokens_seen": 1404829696 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029010030090270814, + "loss": 1.2312, + "theoretical_loss": 3.5365730690334147, + "tokens_seen": 1404895232 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002900902708124373, + "loss": 1.3704, + "theoretical_loss": 3.536558340298093, + "tokens_seen": 1404960768 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002900802407221665, + "loss": 1.3818, + "theoretical_loss": 3.5365436124421548, + "tokens_seen": 1405026304 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002900702106318957, + "loss": 1.3031, + "theoretical_loss": 3.536528885465506, + "tokens_seen": 1405091840 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029006018054162486, + "loss": 1.3457, + "theoretical_loss": 3.5365141593680534, + "tokens_seen": 1405157376 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002900501504513541, + "loss": 1.3725, + "theoretical_loss": 3.536499434149704, + "tokens_seen": 1405222912 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002900401203610832, + "loss": 1.4683, + "theoretical_loss": 3.536484709810364, + "tokens_seen": 1405288448 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029003009027081246, + "loss": 1.3673, + "theoretical_loss": 3.536469986349939, + "tokens_seen": 1405353984 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002900200601805416, + "loss": 1.2719, + "theoretical_loss": 3.536455263768337, + "tokens_seen": 1405419520 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002900100300902708, + "loss": 1.3931, + "theoretical_loss": 3.5364405420654634, + "tokens_seen": 1405485056 + }, + { + "epoch": 0.43, + "learning_rate": 0.00029, + "loss": 1.3818, + "theoretical_loss": 3.5364258212412256, + "tokens_seen": 1405550592 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002899899699097292, + "loss": 1.31, + "theoretical_loss": 3.5364111012955304, + "tokens_seen": 1405616128 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028997993981945836, + "loss": 1.4485, + "theoretical_loss": 3.5363963822282836, + "tokens_seen": 1405681664 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0015632803551852703, + "objective/train/docs_used": 796159, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.694598436355591, + "objective/train/original_loss": 2.69459867477417, + "objective/train/theoretical_loss": 3.536381664039392, + "objective/train/tokens_used": 1426207200, + "objective/train/value_avg": -0.004131317138671875, + "objective/train/value_loss": 5.786851397715509e-05, + "objective/train/value_max": -0.0003101825714111328, + "objective/train/value_min": -0.0526123046875, + "objective/train/value_reward_corr": 0.26320533367289967, + "objective/train/value_std": 0.003002166748046875, + "objective/train/weight_avg": 0.9871290326118469, + "objective/train/weighted_lm_loss": 2.6863858699798584, + "objective/train/weights_max": 1.4609168767929077, + "objective/train/weights_min": 0.6037638783454895, + "theoretical_loss": 3.536381664039392, + "tokens_seen": 1405747200 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002899699097291876, + "loss": 1.3971, + "theoretical_loss": 3.536381664039392, + "tokens_seen": 1405747200 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002899598796389167, + "loss": 1.3236, + "theoretical_loss": 3.536366946728762, + "tokens_seen": 1405812736 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028994984954864596, + "loss": 1.4147, + "theoretical_loss": 3.5363522302963015, + "tokens_seen": 1405878272 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002899398194583751, + "loss": 1.34, + "theoretical_loss": 3.536337514741916, + "tokens_seen": 1405943808 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002899297893681043, + "loss": 1.4274, + "theoretical_loss": 3.5363228000655123, + "tokens_seen": 1406009344 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002899197592778335, + "loss": 1.42, + "theoretical_loss": 3.536308086266997, + "tokens_seen": 1406074880 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002899097291875627, + "loss": 1.3838, + "theoretical_loss": 3.536293373346278, + "tokens_seen": 1406140416 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028989969909729187, + "loss": 1.3332, + "theoretical_loss": 3.5362786613032604, + "tokens_seen": 1406205952 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028988966900702105, + "loss": 1.4204, + "theoretical_loss": 3.536263950137852, + "tokens_seen": 1406271488 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028987963891675023, + "loss": 1.3176, + "theoretical_loss": 3.536249239849959, + "tokens_seen": 1406337024 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028986960882647946, + "loss": 1.2499, + "theoretical_loss": 3.536234530439488, + "tokens_seen": 1406402560 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002898595787362086, + "loss": 1.4156, + "theoretical_loss": 3.536219821906347, + "tokens_seen": 1406468096 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002898495486459378, + "loss": 1.3228, + "theoretical_loss": 3.536205114250442, + "tokens_seen": 1406533632 + }, + { + "epoch": 0.43, + "learning_rate": 0.000289839518555667, + "loss": 1.3182, + "theoretical_loss": 3.536190407471679, + "tokens_seen": 1406599168 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002898294884653962, + "loss": 1.3306, + "theoretical_loss": 3.536175701569966, + "tokens_seen": 1406664704 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028981945837512537, + "loss": 1.3814, + "theoretical_loss": 3.5361609965452097, + "tokens_seen": 1406730240 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028980942828485455, + "loss": 1.3908, + "theoretical_loss": 3.5361462923973166, + "tokens_seen": 1406795776 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028979939819458373, + "loss": 1.4682, + "theoretical_loss": 3.5361315891261933, + "tokens_seen": 1406861312 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028978936810431297, + "loss": 1.3032, + "theoretical_loss": 3.5361168867317474, + "tokens_seen": 1406926848 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028977933801404215, + "loss": 1.1961, + "theoretical_loss": 3.5361021852138848, + "tokens_seen": 1406992384 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028976930792377133, + "loss": 1.3067, + "theoretical_loss": 3.5360874845725134, + "tokens_seen": 1407057920 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002897592778335005, + "loss": 1.392, + "theoretical_loss": 3.5360727848075397, + "tokens_seen": 1407123456 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002897492477432297, + "loss": 1.398, + "theoretical_loss": 3.5360580859188713, + "tokens_seen": 1407188992 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028973921765295893, + "loss": 1.3459, + "theoretical_loss": 3.536043387906414, + "tokens_seen": 1407254528 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028972918756268805, + "loss": 1.3707, + "theoretical_loss": 3.5360286907700758, + "tokens_seen": 1407320064 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0003224987303838134, + "objective/train/docs_used": 797613, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4252493381500244, + "objective/train/original_loss": 2.4252493381500244, + "objective/train/theoretical_loss": 3.536013994509762, + "objective/train/tokens_used": 1427845600, + "objective/train/value_avg": -0.004276275634765625, + "objective/train/value_loss": 4.827817974728532e-05, + "objective/train/value_max": -0.0002892017364501953, + "objective/train/value_min": -0.1619873046875, + "objective/train/value_reward_corr": 0.3797905333099932, + "objective/train/value_std": 0.003498077392578125, + "objective/train/weight_avg": 0.9990419745445251, + "objective/train/weighted_lm_loss": 2.436353921890259, + "objective/train/weights_max": 4.337080001831055, + "objective/train/weights_min": 0.25179678201675415, + "theoretical_loss": 3.536013994509762, + "tokens_seen": 1407385600 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002897191574724173, + "loss": 1.2569, + "theoretical_loss": 3.536013994509762, + "tokens_seen": 1407385600 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002897091273821464, + "loss": 1.3811, + "theoretical_loss": 3.5359992991253817, + "tokens_seen": 1407451136 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028969909729187565, + "loss": 1.3423, + "theoretical_loss": 3.535984604616841, + "tokens_seen": 1407516672 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028968906720160483, + "loss": 1.3879, + "theoretical_loss": 3.535969910984047, + "tokens_seen": 1407582208 + }, + { + "epoch": 0.43, + "learning_rate": 0.000289679037111334, + "loss": 1.4015, + "theoretical_loss": 3.535955218226907, + "tokens_seen": 1407647744 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002896690070210632, + "loss": 1.3575, + "theoretical_loss": 3.5359405263453274, + "tokens_seen": 1407713280 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028965897693079243, + "loss": 1.3496, + "theoretical_loss": 3.5359258353392153, + "tokens_seen": 1407778816 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028964894684052156, + "loss": 1.3686, + "theoretical_loss": 3.535911145208478, + "tokens_seen": 1407844352 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002896389167502508, + "loss": 1.2583, + "theoretical_loss": 3.535896455953023, + "tokens_seen": 1407909888 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002896288866599799, + "loss": 1.4322, + "theoretical_loss": 3.535881767572757, + "tokens_seen": 1407975424 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028961885656970916, + "loss": 1.3517, + "theoretical_loss": 3.535867080067587, + "tokens_seen": 1408040960 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028960882647943834, + "loss": 1.3972, + "theoretical_loss": 3.5358523934374206, + "tokens_seen": 1408106496 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002895987963891675, + "loss": 1.4276, + "theoretical_loss": 3.5358377076821643, + "tokens_seen": 1408172032 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002895887662988967, + "loss": 1.3433, + "theoretical_loss": 3.5358230228017264, + "tokens_seen": 1408237568 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002895787362086259, + "loss": 1.3407, + "theoretical_loss": 3.535808338796013, + "tokens_seen": 1408303104 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028956870611835506, + "loss": 1.3063, + "theoretical_loss": 3.535793655664931, + "tokens_seen": 1408368640 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002895586760280843, + "loss": 1.4543, + "theoretical_loss": 3.535778973408389, + "tokens_seen": 1408434176 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002895486459378134, + "loss": 1.4042, + "theoretical_loss": 3.5357642920262933, + "tokens_seen": 1408499712 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028953861584754266, + "loss": 1.27, + "theoretical_loss": 3.535749611518551, + "tokens_seen": 1408565248 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002895285857572718, + "loss": 1.3174, + "theoretical_loss": 3.53573493188507, + "tokens_seen": 1408630784 + }, + { + "epoch": 0.43, + "learning_rate": 0.000289518555667001, + "loss": 1.3325, + "theoretical_loss": 3.535720253125757, + "tokens_seen": 1408696320 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002895085255767302, + "loss": 1.3681, + "theoretical_loss": 3.5357055752405193, + "tokens_seen": 1408761856 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002894984954864594, + "loss": 1.4074, + "theoretical_loss": 3.535690898229265, + "tokens_seen": 1408827392 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028948846539618856, + "loss": 1.3012, + "theoretical_loss": 3.5356762220919, + "tokens_seen": 1408892928 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002894784353059178, + "loss": 1.4801, + "theoretical_loss": 3.535661546828333, + "tokens_seen": 1408958464 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": 0.0007056426838971674, + "objective/train/docs_used": 798401, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.048619031906128, + "objective/train/original_loss": 3.0486185550689697, + "objective/train/theoretical_loss": 3.5356468724384706, + "objective/train/tokens_used": 1429484000, + "objective/train/value_avg": -0.005512237548828125, + "objective/train/value_loss": 7.173923222580925e-05, + "objective/train/value_max": -0.00027370452880859375, + "objective/train/value_min": -0.06903076171875, + "objective/train/value_reward_corr": 0.4487086205245295, + "objective/train/value_std": 0.004673004150390625, + "objective/train/weight_avg": 1.0101070404052734, + "objective/train/weighted_lm_loss": 3.082043170928955, + "objective/train/weights_max": 1.6627776622772217, + "objective/train/weights_min": 0.11283545196056366, + "theoretical_loss": 3.5356468724384706, + "tokens_seen": 1409024000 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002894684052156469, + "loss": 1.3012, + "theoretical_loss": 3.5356468724384706, + "tokens_seen": 1409024000 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028945837512537616, + "loss": 1.4054, + "theoretical_loss": 3.5356321989222197, + "tokens_seen": 1409089536 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002894483450351053, + "loss": 1.3749, + "theoretical_loss": 3.535617526279489, + "tokens_seen": 1409155072 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002894383149448345, + "loss": 1.2827, + "theoretical_loss": 3.5356028545101847, + "tokens_seen": 1409220608 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002894282848545637, + "loss": 1.3561, + "theoretical_loss": 3.535588183614215, + "tokens_seen": 1409286144 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002894182547642929, + "loss": 1.3605, + "theoretical_loss": 3.535573513591487, + "tokens_seen": 1409351680 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028940822467402207, + "loss": 1.4164, + "theoretical_loss": 3.5355588444419075, + "tokens_seen": 1409417216 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028939819458375125, + "loss": 1.4227, + "theoretical_loss": 3.5355441761653847, + "tokens_seen": 1409482752 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028938816449348043, + "loss": 1.4111, + "theoretical_loss": 3.535529508761826, + "tokens_seen": 1409548288 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028937813440320966, + "loss": 1.4742, + "theoretical_loss": 3.535514842231138, + "tokens_seen": 1409613824 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002893681043129388, + "loss": 1.2921, + "theoretical_loss": 3.53550017657323, + "tokens_seen": 1409679360 + }, + { + "epoch": 0.43, + "learning_rate": 0.000289358074222668, + "loss": 1.4256, + "theoretical_loss": 3.535485511788008, + "tokens_seen": 1409744896 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002893480441323972, + "loss": 1.3315, + "theoretical_loss": 3.53547084787538, + "tokens_seen": 1409810432 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002893380140421264, + "loss": 1.266, + "theoretical_loss": 3.535456184835253, + "tokens_seen": 1409875968 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028932798395185557, + "loss": 1.2382, + "theoretical_loss": 3.5354415226675346, + "tokens_seen": 1409941504 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028931795386158475, + "loss": 1.3841, + "theoretical_loss": 3.535426861372133, + "tokens_seen": 1410007040 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028930792377131393, + "loss": 1.4499, + "theoretical_loss": 3.5354122009489553, + "tokens_seen": 1410072576 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028929789368104317, + "loss": 1.3743, + "theoretical_loss": 3.5353975413979093, + "tokens_seen": 1410138112 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002892878635907723, + "loss": 1.4525, + "theoretical_loss": 3.5353828827189027, + "tokens_seen": 1410203648 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028927783350050153, + "loss": 1.3375, + "theoretical_loss": 3.535368224911843, + "tokens_seen": 1410269184 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028926780341023066, + "loss": 1.3988, + "theoretical_loss": 3.535353567976637, + "tokens_seen": 1410334720 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002892577733199599, + "loss": 1.2395, + "theoretical_loss": 3.5353389119131937, + "tokens_seen": 1410400256 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002892477432296891, + "loss": 1.3597, + "theoretical_loss": 3.5353242567214203, + "tokens_seen": 1410465792 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028923771313941825, + "loss": 1.4267, + "theoretical_loss": 3.535309602401224, + "tokens_seen": 1410531328 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028922768304914744, + "loss": 1.4437, + "theoretical_loss": 3.535294948952512, + "tokens_seen": 1410596864 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0007814686978235841, + "objective/train/docs_used": 799700, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.78079891204834, + "objective/train/original_loss": 2.780799150466919, + "objective/train/theoretical_loss": 3.5352802963751935, + "objective/train/tokens_used": 1431122400, + "objective/train/value_avg": -0.004970550537109375, + "objective/train/value_loss": 6.702192331431434e-05, + "objective/train/value_max": -0.0003006458282470703, + "objective/train/value_min": -0.06903076171875, + "objective/train/value_reward_corr": 0.4939722813426459, + "objective/train/value_std": 0.004138946533203125, + "objective/train/weight_avg": 0.9952037930488586, + "objective/train/weighted_lm_loss": 2.7727298736572266, + "objective/train/weights_max": 1.3428575992584229, + "objective/train/weights_min": 0.2923228144645691, + "theoretical_loss": 3.5352802963751935, + "tokens_seen": 1410662400 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002892176529588766, + "loss": 1.2895, + "theoretical_loss": 3.5352802963751935, + "tokens_seen": 1410662400 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002892076228686058, + "loss": 1.3231, + "theoretical_loss": 3.535265644669175, + "tokens_seen": 1410727936 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028919759277833503, + "loss": 1.3, + "theoretical_loss": 3.5352509938343646, + "tokens_seen": 1410793472 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028918756268806416, + "loss": 1.365, + "theoretical_loss": 3.535236343870671, + "tokens_seen": 1410859008 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002891775325977934, + "loss": 1.4006, + "theoretical_loss": 3.5352216947779995, + "tokens_seen": 1410924544 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002891675025075226, + "loss": 1.3796, + "theoretical_loss": 3.5352070465562604, + "tokens_seen": 1410990080 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028915747241725176, + "loss": 1.3709, + "theoretical_loss": 3.5351923992053598, + "tokens_seen": 1411055616 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028914744232698094, + "loss": 1.3043, + "theoretical_loss": 3.5351777527252066, + "tokens_seen": 1411121152 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002891374122367101, + "loss": 1.2896, + "theoretical_loss": 3.535163107115708, + "tokens_seen": 1411186688 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002891273821464393, + "loss": 1.359, + "theoretical_loss": 3.535148462376772, + "tokens_seen": 1411252224 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028911735205616854, + "loss": 1.2541, + "theoretical_loss": 3.5351338185083065, + "tokens_seen": 1411317760 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028910732196589766, + "loss": 1.354, + "theoretical_loss": 3.5351191755102187, + "tokens_seen": 1411383296 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002890972918756269, + "loss": 1.4687, + "theoretical_loss": 3.5351045333824174, + "tokens_seen": 1411448832 + }, + { + "epoch": 0.43, + "learning_rate": 0.000289087261785356, + "loss": 1.3134, + "theoretical_loss": 3.5350898921248097, + "tokens_seen": 1411514368 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028907723169508526, + "loss": 1.3158, + "theoretical_loss": 3.5350752517373043, + "tokens_seen": 1411579904 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028906720160481444, + "loss": 1.4164, + "theoretical_loss": 3.535060612219808, + "tokens_seen": 1411645440 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002890571715145436, + "loss": 1.3942, + "theoretical_loss": 3.53504597357223, + "tokens_seen": 1411710976 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002890471414242728, + "loss": 1.3683, + "theoretical_loss": 3.535031335794477, + "tokens_seen": 1411776512 + }, + { + "epoch": 0.43, + "learning_rate": 0.000289037111334002, + "loss": 1.3518, + "theoretical_loss": 3.5350166988864578, + "tokens_seen": 1411842048 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002890270812437312, + "loss": 1.3532, + "theoretical_loss": 3.53500206284808, + "tokens_seen": 1411907584 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002890170511534604, + "loss": 1.3323, + "theoretical_loss": 3.5349874276792512, + "tokens_seen": 1411973120 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002890070210631896, + "loss": 1.3626, + "theoretical_loss": 3.5349727933798807, + "tokens_seen": 1412038656 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028899699097291876, + "loss": 1.3212, + "theoretical_loss": 3.5349581599498747, + "tokens_seen": 1412104192 + }, + { + "epoch": 0.43, + "learning_rate": 0.000288986960882648, + "loss": 1.4003, + "theoretical_loss": 3.5349435273891423, + "tokens_seen": 1412169728 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002889769307923771, + "loss": 1.2441, + "theoretical_loss": 3.5349288956975915, + "tokens_seen": 1412235264 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0001438494073227048, + "objective/train/docs_used": 800754, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7171859741210938, + "objective/train/original_loss": 2.7171857357025146, + "objective/train/theoretical_loss": 3.5349142648751304, + "objective/train/tokens_used": 1432760800, + "objective/train/value_avg": -0.004970550537109375, + "objective/train/value_loss": 5.923142816754989e-05, + "objective/train/value_max": -0.00027370452880859375, + "objective/train/value_min": -0.0509033203125, + "objective/train/value_reward_corr": 0.330435274472879, + "objective/train/value_std": 0.0037097930908203125, + "objective/train/weight_avg": 1.001343846321106, + "objective/train/weighted_lm_loss": 2.7157421112060547, + "objective/train/weights_max": 1.5010861158370972, + "objective/train/weights_min": 0.49448713660240173, + "theoretical_loss": 3.5349142648751304, + "tokens_seen": 1412300800 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028896690070210636, + "loss": 1.4211, + "theoretical_loss": 3.5349142648751304, + "tokens_seen": 1412300800 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002889568706118355, + "loss": 1.2667, + "theoretical_loss": 3.5348996349216666, + "tokens_seen": 1412366336 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002889468405215647, + "loss": 1.366, + "theoretical_loss": 3.534885005837108, + "tokens_seen": 1412431872 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002889368104312939, + "loss": 1.3375, + "theoretical_loss": 3.5348703776213632, + "tokens_seen": 1412497408 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002889267803410231, + "loss": 1.3254, + "theoretical_loss": 3.5348557502743407, + "tokens_seen": 1412562944 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028891675025075227, + "loss": 1.3578, + "theoretical_loss": 3.534841123795948, + "tokens_seen": 1412628480 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028890672016048145, + "loss": 1.4807, + "theoretical_loss": 3.534826498186093, + "tokens_seen": 1412694016 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028889669007021063, + "loss": 1.3459, + "theoretical_loss": 3.534811873444684, + "tokens_seen": 1412759552 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028888665997993986, + "loss": 1.3071, + "theoretical_loss": 3.5347972495716293, + "tokens_seen": 1412825088 + }, + { + "epoch": 0.43, + "learning_rate": 0.000288876629889669, + "loss": 1.3449, + "theoretical_loss": 3.5347826265668374, + "tokens_seen": 1412890624 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028886659979939823, + "loss": 1.301, + "theoretical_loss": 3.534768004430216, + "tokens_seen": 1412956160 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002888565697091274, + "loss": 1.4567, + "theoretical_loss": 3.5347533831616733, + "tokens_seen": 1413021696 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002888465396188566, + "loss": 1.3935, + "theoretical_loss": 3.5347387627611173, + "tokens_seen": 1413087232 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028883650952858577, + "loss": 1.4241, + "theoretical_loss": 3.5347241432284573, + "tokens_seen": 1413152768 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028882647943831495, + "loss": 1.3403, + "theoretical_loss": 3.5347095245636, + "tokens_seen": 1413218304 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028881644934804413, + "loss": 1.2684, + "theoretical_loss": 3.534694906766455, + "tokens_seen": 1413283840 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028880641925777337, + "loss": 1.3812, + "theoretical_loss": 3.5346802898369303, + "tokens_seen": 1413349376 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002887963891675025, + "loss": 1.4738, + "theoretical_loss": 3.534665673774933, + "tokens_seen": 1413414912 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028878635907723173, + "loss": 1.2374, + "theoretical_loss": 3.5346510585803728, + "tokens_seen": 1413480448 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028877632898696086, + "loss": 1.2095, + "theoretical_loss": 3.534636444253157, + "tokens_seen": 1413545984 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002887662988966901, + "loss": 1.3348, + "theoretical_loss": 3.5346218307931947, + "tokens_seen": 1413611520 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002887562688064193, + "loss": 1.302, + "theoretical_loss": 3.534607218200394, + "tokens_seen": 1413677056 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028874623871614845, + "loss": 1.2887, + "theoretical_loss": 3.5345926064746633, + "tokens_seen": 1413742592 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028873620862587764, + "loss": 1.3925, + "theoretical_loss": 3.5345779956159102, + "tokens_seen": 1413808128 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002887261785356068, + "loss": 1.4708, + "theoretical_loss": 3.534563385624044, + "tokens_seen": 1413873664 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0031044345814734697, + "objective/train/docs_used": 801653, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8275861740112305, + "objective/train/original_loss": 2.8275856971740723, + "objective/train/theoretical_loss": 3.5345487764989727, + "objective/train/tokens_used": 1434399200, + "objective/train/value_avg": -0.005126953125, + "objective/train/value_loss": 0.00011763369548134506, + "objective/train/value_max": -0.0003275871276855469, + "objective/train/value_min": -0.08990478515625, + "objective/train/value_reward_corr": 0.43628214768712903, + "objective/train/value_std": 0.004405975341796875, + "objective/train/weight_avg": 0.9743637442588806, + "objective/train/weighted_lm_loss": 2.8051955699920654, + "objective/train/weights_max": 1.943854570388794, + "objective/train/weights_min": 0.3827384412288666, + "theoretical_loss": 3.5345487764989727, + "tokens_seen": 1413939200 + }, + { + "epoch": 0.43, + "learning_rate": 0.000288716148445336, + "loss": 1.3504, + "theoretical_loss": 3.5345487764989727, + "tokens_seen": 1413939200 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028870611835506523, + "loss": 1.337, + "theoretical_loss": 3.5345341682406053, + "tokens_seen": 1414004736 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028869608826479436, + "loss": 1.3149, + "theoretical_loss": 3.534519560848849, + "tokens_seen": 1414070272 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002886860581745236, + "loss": 1.2074, + "theoretical_loss": 3.534504954323613, + "tokens_seen": 1414135808 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002886760280842528, + "loss": 1.3859, + "theoretical_loss": 3.5344903486648063, + "tokens_seen": 1414201344 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028866599799398196, + "loss": 1.3448, + "theoretical_loss": 3.534475743872336, + "tokens_seen": 1414266880 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028865596790371114, + "loss": 1.3949, + "theoretical_loss": 3.534461139946112, + "tokens_seen": 1414332416 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002886459378134403, + "loss": 1.2944, + "theoretical_loss": 3.5344465368860414, + "tokens_seen": 1414397952 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002886359077231695, + "loss": 1.4469, + "theoretical_loss": 3.5344319346920336, + "tokens_seen": 1414463488 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028862587763289874, + "loss": 1.3322, + "theoretical_loss": 3.534417333363997, + "tokens_seen": 1414529024 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028861584754262786, + "loss": 1.3753, + "theoretical_loss": 3.53440273290184, + "tokens_seen": 1414594560 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002886058174523571, + "loss": 1.2706, + "theoretical_loss": 3.5343881333054714, + "tokens_seen": 1414660096 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002885957873620862, + "loss": 1.3972, + "theoretical_loss": 3.534373534574799, + "tokens_seen": 1414725632 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028858575727181546, + "loss": 1.3861, + "theoretical_loss": 3.534358936709732, + "tokens_seen": 1414791168 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028857572718154464, + "loss": 1.3598, + "theoretical_loss": 3.5343443397101795, + "tokens_seen": 1414856704 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002885656970912738, + "loss": 1.3578, + "theoretical_loss": 3.5343297435760492, + "tokens_seen": 1414922240 + }, + { + "epoch": 0.43, + "learning_rate": 0.000288555667001003, + "loss": 1.3974, + "theoretical_loss": 3.5343151483072495, + "tokens_seen": 1414987776 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002885456369107322, + "loss": 1.2229, + "theoretical_loss": 3.5343005539036896, + "tokens_seen": 1415053312 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028853560682046137, + "loss": 1.4084, + "theoretical_loss": 3.5342859603652785, + "tokens_seen": 1415118848 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002885255767301906, + "loss": 1.3673, + "theoretical_loss": 3.5342713676919235, + "tokens_seen": 1415184384 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028851554663991973, + "loss": 1.3688, + "theoretical_loss": 3.5342567758835353, + "tokens_seen": 1415249920 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028850551654964896, + "loss": 1.4805, + "theoretical_loss": 3.5342421849400205, + "tokens_seen": 1415315456 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028849548645937815, + "loss": 1.3049, + "theoretical_loss": 3.534227594861289, + "tokens_seen": 1415380992 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002884854563691073, + "loss": 1.4076, + "theoretical_loss": 3.5342130056472496, + "tokens_seen": 1415446528 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002884754262788365, + "loss": 1.381, + "theoretical_loss": 3.5341984172978105, + "tokens_seen": 1415512064 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0024693526793271303, + "objective/train/docs_used": 802505, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4770567417144775, + "objective/train/original_loss": 2.4770569801330566, + "objective/train/theoretical_loss": 3.5341838298128803, + "objective/train/tokens_used": 1436037600, + "objective/train/value_avg": -0.0056915283203125, + "objective/train/value_loss": 9.077771392185241e-05, + "objective/train/value_max": -0.00036835670471191406, + "objective/train/value_min": -0.080078125, + "objective/train/value_reward_corr": 0.36218318825870555, + "objective/train/value_std": 0.00409698486328125, + "objective/train/weight_avg": 0.9792460203170776, + "objective/train/weighted_lm_loss": 2.431955575942993, + "objective/train/weights_max": 1.934668779373169, + "objective/train/weights_min": 0.08417306840419769, + "theoretical_loss": 3.5341838298128803, + "tokens_seen": 1415577600 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002884653961885657, + "loss": 1.2944, + "theoretical_loss": 3.5341838298128803, + "tokens_seen": 1415577600 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028845536609829487, + "loss": 1.3526, + "theoretical_loss": 3.534169243192368, + "tokens_seen": 1415643136 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002884453360080241, + "loss": 1.3753, + "theoretical_loss": 3.534154657436183, + "tokens_seen": 1415708672 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028843530591775323, + "loss": 1.3461, + "theoretical_loss": 3.5341400725442327, + "tokens_seen": 1415774208 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028842527582748247, + "loss": 1.288, + "theoretical_loss": 3.5341254885164273, + "tokens_seen": 1415839744 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002884152457372116, + "loss": 1.3944, + "theoretical_loss": 3.5341109053526747, + "tokens_seen": 1415905280 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028840521564694083, + "loss": 1.3718, + "theoretical_loss": 3.534096323052884, + "tokens_seen": 1415970816 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028839518555667, + "loss": 1.3918, + "theoretical_loss": 3.534081741616964, + "tokens_seen": 1416036352 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002883851554663992, + "loss": 1.3854, + "theoretical_loss": 3.5340671610448235, + "tokens_seen": 1416101888 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002883751253761284, + "loss": 1.4591, + "theoretical_loss": 3.534052581336372, + "tokens_seen": 1416167424 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002883650952858576, + "loss": 1.4493, + "theoretical_loss": 3.534038002491517, + "tokens_seen": 1416232960 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028835506519558674, + "loss": 1.3648, + "theoretical_loss": 3.5340234245101687, + "tokens_seen": 1416298496 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028834503510531597, + "loss": 1.2984, + "theoretical_loss": 3.5340088473922355, + "tokens_seen": 1416364032 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002883350050150451, + "loss": 1.3523, + "theoretical_loss": 3.533994271137626, + "tokens_seen": 1416429568 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028832497492477433, + "loss": 1.3979, + "theoretical_loss": 3.5339796957462495, + "tokens_seen": 1416495104 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002883149448345035, + "loss": 1.3839, + "theoretical_loss": 3.5339651212180145, + "tokens_seen": 1416560640 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002883049147442327, + "loss": 1.4392, + "theoretical_loss": 3.5339505475528314, + "tokens_seen": 1416626176 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002882948846539619, + "loss": 1.358, + "theoretical_loss": 3.533935974750607, + "tokens_seen": 1416691712 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028828485456369106, + "loss": 1.3519, + "theoretical_loss": 3.533921402811252, + "tokens_seen": 1416757248 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002882748244734203, + "loss": 1.447, + "theoretical_loss": 3.533906831734675, + "tokens_seen": 1416822784 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002882647943831495, + "loss": 1.2694, + "theoretical_loss": 3.5338922615207844, + "tokens_seen": 1416888320 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028825476429287866, + "loss": 1.321, + "theoretical_loss": 3.533877692169489, + "tokens_seen": 1416953856 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028824473420260784, + "loss": 1.4861, + "theoretical_loss": 3.5338631236806997, + "tokens_seen": 1417019392 + }, + { + "epoch": 0.43, + "learning_rate": 0.000288234704112337, + "loss": 1.2587, + "theoretical_loss": 3.533848556054324, + "tokens_seen": 1417084928 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002882246740220662, + "loss": 1.3816, + "theoretical_loss": 3.5338339892902706, + "tokens_seen": 1417150464 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": 0.0002964813611470163, + "objective/train/docs_used": 803171, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2104296684265137, + "objective/train/original_loss": 3.2104294300079346, + "objective/train/theoretical_loss": 3.53381942338845, + "objective/train/tokens_used": 1437676000, + "objective/train/value_avg": -0.006252288818359375, + "objective/train/value_loss": 8.180701115634292e-05, + "objective/train/value_max": -0.0003859996795654297, + "objective/train/value_min": -0.06964111328125, + "objective/train/value_reward_corr": 0.43560346135881456, + "objective/train/value_std": 0.0051116943359375, + "objective/train/weight_avg": 1.0067694187164307, + "objective/train/weighted_lm_loss": 3.23797869682312, + "objective/train/weights_max": 1.547546625137329, + "objective/train/weights_min": 0.3064942955970764, + "theoretical_loss": 3.53381942338845, + "tokens_seen": 1417216000 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028821464393179543, + "loss": 1.4645, + "theoretical_loss": 3.53381942338845, + "tokens_seen": 1417216000 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028820461384152456, + "loss": 1.3584, + "theoretical_loss": 3.5338048583487702, + "tokens_seen": 1417281536 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002881945837512538, + "loss": 1.3952, + "theoretical_loss": 3.5337902941711405, + "tokens_seen": 1417347072 + }, + { + "epoch": 0.43, + "learning_rate": 0.000288184553660983, + "loss": 1.3793, + "theoretical_loss": 3.5337757308554707, + "tokens_seen": 1417412608 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028817452357071216, + "loss": 1.3727, + "theoretical_loss": 3.533761168401669, + "tokens_seen": 1417478144 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028816449348044134, + "loss": 1.3083, + "theoretical_loss": 3.533746606809645, + "tokens_seen": 1417543680 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002881544633901705, + "loss": 1.3234, + "theoretical_loss": 3.533732046079308, + "tokens_seen": 1417609216 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002881444332998997, + "loss": 1.4178, + "theoretical_loss": 3.533717486210567, + "tokens_seen": 1417674752 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028813440320962894, + "loss": 1.4204, + "theoretical_loss": 3.533702927203332, + "tokens_seen": 1417740288 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028812437311935806, + "loss": 1.3629, + "theoretical_loss": 3.5336883690575105, + "tokens_seen": 1417805824 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002881143430290873, + "loss": 1.3729, + "theoretical_loss": 3.533673811773012, + "tokens_seen": 1417871360 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002881043129388164, + "loss": 1.4683, + "theoretical_loss": 3.5336592553497477, + "tokens_seen": 1417936896 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028809428284854566, + "loss": 1.237, + "theoretical_loss": 3.533644699787625, + "tokens_seen": 1418002432 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028808425275827484, + "loss": 1.1778, + "theoretical_loss": 3.533630145086553, + "tokens_seen": 1418067968 + }, + { + "epoch": 0.43, + "learning_rate": 0.000288074222668004, + "loss": 1.2979, + "theoretical_loss": 3.533615591246442, + "tokens_seen": 1418133504 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002880641925777332, + "loss": 1.3422, + "theoretical_loss": 3.533601038267202, + "tokens_seen": 1418199040 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002880541624874624, + "loss": 1.2738, + "theoretical_loss": 3.53358648614874, + "tokens_seen": 1418264576 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028804413239719157, + "loss": 1.3746, + "theoretical_loss": 3.5335719348909667, + "tokens_seen": 1418330112 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002880341023069208, + "loss": 1.3534, + "theoretical_loss": 3.5335573844937915, + "tokens_seen": 1418395648 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028802407221664993, + "loss": 1.3772, + "theoretical_loss": 3.5335428349571236, + "tokens_seen": 1418461184 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028801404212637916, + "loss": 1.3295, + "theoretical_loss": 3.5335282862808723, + "tokens_seen": 1418526720 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028800401203610835, + "loss": 1.3552, + "theoretical_loss": 3.533513738464947, + "tokens_seen": 1418592256 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002879939819458375, + "loss": 1.3647, + "theoretical_loss": 3.5334991915092564, + "tokens_seen": 1418657792 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002879839518555667, + "loss": 1.454, + "theoretical_loss": 3.5334846454137114, + "tokens_seen": 1418723328 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002879739217652959, + "loss": 1.3443, + "theoretical_loss": 3.53347010017822, + "tokens_seen": 1418788864 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0014089815085753798, + "objective/train/docs_used": 804514, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8982527256011963, + "objective/train/original_loss": 2.8982527256011963, + "objective/train/theoretical_loss": 3.533455555802692, + "objective/train/tokens_used": 1439314400, + "objective/train/value_avg": -0.005260467529296875, + "objective/train/value_loss": 0.00010663033026503399, + "objective/train/value_max": -0.0003407001495361328, + "objective/train/value_min": -0.061309814453125, + "objective/train/value_reward_corr": 0.6549664955333433, + "objective/train/value_std": 0.004985809326171875, + "objective/train/weight_avg": 0.9907052516937256, + "objective/train/weighted_lm_loss": 2.8778724670410156, + "objective/train/weights_max": 1.3946473598480225, + "objective/train/weights_min": 0.4457384943962097, + "theoretical_loss": 3.533455555802692, + "tokens_seen": 1418854400 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028796389167502507, + "loss": 1.3071, + "theoretical_loss": 3.533455555802692, + "tokens_seen": 1418854400 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002879538615847543, + "loss": 1.3052, + "theoretical_loss": 3.5334410122870374, + "tokens_seen": 1418919936 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028794383149448343, + "loss": 1.4172, + "theoretical_loss": 3.5334264696311655, + "tokens_seen": 1418985472 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028793380140421267, + "loss": 1.392, + "theoretical_loss": 3.533411927834985, + "tokens_seen": 1419051008 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002879237713139418, + "loss": 1.4172, + "theoretical_loss": 3.5333973868984057, + "tokens_seen": 1419116544 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028791374122367103, + "loss": 1.3916, + "theoretical_loss": 3.533382846821338, + "tokens_seen": 1419182080 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002879037111334002, + "loss": 1.3481, + "theoretical_loss": 3.5333683076036904, + "tokens_seen": 1419247616 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002878936810431294, + "loss": 1.4306, + "theoretical_loss": 3.5333537692453723, + "tokens_seen": 1419313152 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002878836509528586, + "loss": 1.3901, + "theoretical_loss": 3.533339231746294, + "tokens_seen": 1419378688 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002878736208625878, + "loss": 1.4254, + "theoretical_loss": 3.5333246951063644, + "tokens_seen": 1419444224 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028786359077231694, + "loss": 1.4222, + "theoretical_loss": 3.533310159325494, + "tokens_seen": 1419509760 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028785356068204617, + "loss": 1.3504, + "theoretical_loss": 3.533295624403591, + "tokens_seen": 1419575296 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002878435305917753, + "loss": 1.4228, + "theoretical_loss": 3.5332810903405663, + "tokens_seen": 1419640832 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028783350050150453, + "loss": 1.3851, + "theoretical_loss": 3.5332665571363284, + "tokens_seen": 1419706368 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002878234704112337, + "loss": 1.2881, + "theoretical_loss": 3.533252024790788, + "tokens_seen": 1419771904 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002878134403209629, + "loss": 1.3393, + "theoretical_loss": 3.5332374933038535, + "tokens_seen": 1419837440 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002878034102306921, + "loss": 1.4136, + "theoretical_loss": 3.5332229626754357, + "tokens_seen": 1419902976 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028779338014042126, + "loss": 1.3524, + "theoretical_loss": 3.533208432905444, + "tokens_seen": 1419968512 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028778335005015044, + "loss": 1.2365, + "theoretical_loss": 3.533193903993787, + "tokens_seen": 1420034048 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002877733199598797, + "loss": 1.3701, + "theoretical_loss": 3.5331793759403762, + "tokens_seen": 1420099584 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002877632898696088, + "loss": 1.2891, + "theoretical_loss": 3.5331648487451197, + "tokens_seen": 1420165120 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028775325977933804, + "loss": 1.3244, + "theoretical_loss": 3.5331503224079284, + "tokens_seen": 1420230656 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028774322968906716, + "loss": 1.2882, + "theoretical_loss": 3.533135796928711, + "tokens_seen": 1420296192 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002877331995987964, + "loss": 1.3202, + "theoretical_loss": 3.533121272307378, + "tokens_seen": 1420361728 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002877231695085256, + "loss": 1.3039, + "theoretical_loss": 3.5331067485438385, + "tokens_seen": 1420427264 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": 0.0004130412416998297, + "objective/train/docs_used": 804990, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.480898380279541, + "objective/train/original_loss": 2.480898141860962, + "objective/train/theoretical_loss": 3.5330922256380024, + "objective/train/tokens_used": 1440952800, + "objective/train/value_avg": -0.005084991455078125, + "objective/train/value_loss": 5.077890818938613e-05, + "objective/train/value_max": -0.0004107952117919922, + "objective/train/value_min": -0.059326171875, + "objective/train/value_reward_corr": 0.33886621077970275, + "objective/train/value_std": 0.0037899017333984375, + "objective/train/weight_avg": 1.0065457820892334, + "objective/train/weighted_lm_loss": 2.4983510971069336, + "objective/train/weights_max": 1.550267219543457, + "objective/train/weights_min": 0.4411492645740509, + "theoretical_loss": 3.5330922256380024, + "tokens_seen": 1420492800 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028771313941825476, + "loss": 1.3068, + "theoretical_loss": 3.5330922256380024, + "tokens_seen": 1420492800 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028770310932798394, + "loss": 1.3956, + "theoretical_loss": 3.5330777035897807, + "tokens_seen": 1420558336 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002876930792377132, + "loss": 1.4265, + "theoretical_loss": 3.5330631823990815, + "tokens_seen": 1420623872 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002876830491474423, + "loss": 1.3697, + "theoretical_loss": 3.5330486620658155, + "tokens_seen": 1420689408 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028767301905717154, + "loss": 1.2843, + "theoretical_loss": 3.533034142589892, + "tokens_seen": 1420754944 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028766298896690067, + "loss": 1.2967, + "theoretical_loss": 3.5330196239712217, + "tokens_seen": 1420820480 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002876529588766299, + "loss": 1.3374, + "theoretical_loss": 3.533005106209713, + "tokens_seen": 1420886016 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002876429287863591, + "loss": 1.2814, + "theoretical_loss": 3.5329905893052778, + "tokens_seen": 1420951552 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028763289869608826, + "loss": 1.3995, + "theoretical_loss": 3.5329760732578244, + "tokens_seen": 1421017088 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028762286860581745, + "loss": 1.4262, + "theoretical_loss": 3.532961558067263, + "tokens_seen": 1421082624 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002876128385155466, + "loss": 1.5136, + "theoretical_loss": 3.532947043733504, + "tokens_seen": 1421148160 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002876028084252758, + "loss": 1.4346, + "theoretical_loss": 3.5329325302564563, + "tokens_seen": 1421213696 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028759277833500504, + "loss": 1.3053, + "theoretical_loss": 3.532918017636031, + "tokens_seen": 1421279232 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028758274824473417, + "loss": 1.3851, + "theoretical_loss": 3.532903505872137, + "tokens_seen": 1421344768 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002875727181544634, + "loss": 1.2926, + "theoretical_loss": 3.5328889949646856, + "tokens_seen": 1421410304 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028756268806419253, + "loss": 1.3158, + "theoretical_loss": 3.532874484913586, + "tokens_seen": 1421475840 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028755265797392177, + "loss": 1.4581, + "theoretical_loss": 3.5328599757187478, + "tokens_seen": 1421541376 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028754262788365095, + "loss": 1.2901, + "theoretical_loss": 3.5328454673800813, + "tokens_seen": 1421606912 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028753259779338013, + "loss": 1.3443, + "theoretical_loss": 3.5328309598974963, + "tokens_seen": 1421672448 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028752256770310936, + "loss": 1.2991, + "theoretical_loss": 3.5328164532709034, + "tokens_seen": 1421737984 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028751253761283855, + "loss": 1.3497, + "theoretical_loss": 3.5328019475002126, + "tokens_seen": 1421803520 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002875025075225677, + "loss": 1.3885, + "theoretical_loss": 3.532787442585333, + "tokens_seen": 1421869056 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002874924774322969, + "loss": 1.3484, + "theoretical_loss": 3.532772938526176, + "tokens_seen": 1421934592 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002874824473420261, + "loss": 1.2626, + "theoretical_loss": 3.53275843532265, + "tokens_seen": 1422000128 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028747241725175527, + "loss": 1.384, + "theoretical_loss": 3.5327439329746673, + "tokens_seen": 1422065664 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.00020909438899252564, + "objective/train/docs_used": 806295, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0342588424682617, + "objective/train/original_loss": 3.0342588424682617, + "objective/train/theoretical_loss": 3.5327294314821365, + "objective/train/tokens_used": 1442591200, + "objective/train/value_avg": -0.005157470703125, + "objective/train/value_loss": 5.1330862333998084e-05, + "objective/train/value_max": -0.000415802001953125, + "objective/train/value_min": -0.04327392578125, + "objective/train/value_reward_corr": 0.32643699361222644, + "objective/train/value_std": 0.0035839080810546875, + "objective/train/weight_avg": 1.000361442565918, + "objective/train/weighted_lm_loss": 3.031498432159424, + "objective/train/weights_max": 1.4135005474090576, + "objective/train/weights_min": 0.480460524559021, + "theoretical_loss": 3.5327294314821365, + "tokens_seen": 1422131200 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002874623871614845, + "loss": 1.4037, + "theoretical_loss": 3.5327294314821365, + "tokens_seen": 1422131200 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028745235707121363, + "loss": 1.4614, + "theoretical_loss": 3.5327149308449677, + "tokens_seen": 1422196736 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028744232698094287, + "loss": 1.4164, + "theoretical_loss": 3.532700431063072, + "tokens_seen": 1422262272 + }, + { + "epoch": 0.43, + "learning_rate": 0.000287432296890672, + "loss": 1.4621, + "theoretical_loss": 3.532685932136358, + "tokens_seen": 1422327808 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028742226680040123, + "loss": 1.3661, + "theoretical_loss": 3.5326714340647376, + "tokens_seen": 1422393344 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002874122367101304, + "loss": 1.3245, + "theoretical_loss": 3.53265693684812, + "tokens_seen": 1422458880 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002874022066198596, + "loss": 1.2816, + "theoretical_loss": 3.5326424404864154, + "tokens_seen": 1422524416 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002873921765295888, + "loss": 1.3532, + "theoretical_loss": 3.5326279449795344, + "tokens_seen": 1422589952 + }, + { + "epoch": 0.43, + "learning_rate": 0.000287382146439318, + "loss": 1.309, + "theoretical_loss": 3.532613450327387, + "tokens_seen": 1422655488 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028737211634904714, + "loss": 1.4073, + "theoretical_loss": 3.532598956529883, + "tokens_seen": 1422721024 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028736208625877637, + "loss": 1.1403, + "theoretical_loss": 3.5325844635869337, + "tokens_seen": 1422786560 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002873520561685055, + "loss": 1.4153, + "theoretical_loss": 3.5325699714984484, + "tokens_seen": 1422852096 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028734202607823473, + "loss": 1.3549, + "theoretical_loss": 3.5325554802643375, + "tokens_seen": 1422917632 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002873319959879639, + "loss": 1.2697, + "theoretical_loss": 3.5325409898845113, + "tokens_seen": 1422983168 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002873219658976931, + "loss": 1.3819, + "theoretical_loss": 3.5325265003588804, + "tokens_seen": 1423048704 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002873119358074223, + "loss": 1.259, + "theoretical_loss": 3.5325120116873556, + "tokens_seen": 1423114240 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028730190571715146, + "loss": 1.2964, + "theoretical_loss": 3.5324975238698455, + "tokens_seen": 1423179776 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028729187562688064, + "loss": 1.3714, + "theoretical_loss": 3.5324830369062625, + "tokens_seen": 1423245312 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002872818455366099, + "loss": 1.3067, + "theoretical_loss": 3.5324685507965157, + "tokens_seen": 1423310848 + }, + { + "epoch": 0.43, + "learning_rate": 0.000287271815446339, + "loss": 1.4048, + "theoretical_loss": 3.5324540655405157, + "tokens_seen": 1423376384 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028726178535606824, + "loss": 1.3026, + "theoretical_loss": 3.5324395811381724, + "tokens_seen": 1423441920 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028725175526579736, + "loss": 1.3587, + "theoretical_loss": 3.5324250975893974, + "tokens_seen": 1423507456 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002872417251755266, + "loss": 1.4153, + "theoretical_loss": 3.5324106148941, + "tokens_seen": 1423572992 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002872316950852558, + "loss": 1.4899, + "theoretical_loss": 3.5323961330521914, + "tokens_seen": 1423638528 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028722166499498496, + "loss": 1.2564, + "theoretical_loss": 3.532381652063582, + "tokens_seen": 1423704064 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0017469447338953614, + "objective/train/docs_used": 806909, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0434799194335938, + "objective/train/original_loss": 3.043480157852173, + "objective/train/theoretical_loss": 3.532367171928181, + "objective/train/tokens_used": 1444229600, + "objective/train/value_avg": -0.005603790283203125, + "objective/train/value_loss": 8.413185423705727e-05, + "objective/train/value_max": -0.00032258033752441406, + "objective/train/value_min": -0.06695556640625, + "objective/train/value_reward_corr": 0.19644033949206188, + "objective/train/value_std": 0.00426483154296875, + "objective/train/weight_avg": 0.9865029454231262, + "objective/train/weighted_lm_loss": 3.0273773670196533, + "objective/train/weights_max": 1.3878015279769897, + "objective/train/weights_min": 0.5309143662452698, + "theoretical_loss": 3.532367171928181, + "tokens_seen": 1423769600 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028721163490471414, + "loss": 1.3467, + "theoretical_loss": 3.532367171928181, + "tokens_seen": 1423769600 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002872016048144434, + "loss": 1.2957, + "theoretical_loss": 3.5323526926459, + "tokens_seen": 1423835136 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002871915747241725, + "loss": 1.3023, + "theoretical_loss": 3.53233821421665, + "tokens_seen": 1423900672 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028718154463390174, + "loss": 1.3042, + "theoretical_loss": 3.5323237366403397, + "tokens_seen": 1423966208 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028717151454363087, + "loss": 1.3259, + "theoretical_loss": 3.5323092599168815, + "tokens_seen": 1424031744 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002871614844533601, + "loss": 1.3658, + "theoretical_loss": 3.5322947840461847, + "tokens_seen": 1424097280 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002871514543630893, + "loss": 1.2248, + "theoretical_loss": 3.5322803090281605, + "tokens_seen": 1424162816 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028714142427281846, + "loss": 1.3691, + "theoretical_loss": 3.5322658348627187, + "tokens_seen": 1424228352 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028713139418254765, + "loss": 1.2657, + "theoretical_loss": 3.5322513615497706, + "tokens_seen": 1424293888 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002871213640922768, + "loss": 1.1895, + "theoretical_loss": 3.5322368890892264, + "tokens_seen": 1424359424 + }, + { + "epoch": 0.43, + "learning_rate": 0.000287111334002006, + "loss": 1.3793, + "theoretical_loss": 3.5322224174809964, + "tokens_seen": 1424424960 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028710130391173524, + "loss": 1.4359, + "theoretical_loss": 3.532207946724992, + "tokens_seen": 1424490496 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028709127382146437, + "loss": 1.3941, + "theoretical_loss": 3.532193476821123, + "tokens_seen": 1424556032 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002870812437311936, + "loss": 1.3836, + "theoretical_loss": 3.5321790077693005, + "tokens_seen": 1424621568 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028707121364092273, + "loss": 1.4121, + "theoretical_loss": 3.5321645395694348, + "tokens_seen": 1424687104 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028706118355065197, + "loss": 1.3092, + "theoretical_loss": 3.532150072221437, + "tokens_seen": 1424752640 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028705115346038115, + "loss": 1.3275, + "theoretical_loss": 3.5321356057252173, + "tokens_seen": 1424818176 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028704112337011033, + "loss": 1.2676, + "theoretical_loss": 3.5321211400806867, + "tokens_seen": 1424883712 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002870310932798395, + "loss": 1.2767, + "theoretical_loss": 3.5321066752877557, + "tokens_seen": 1424949248 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028702106318956875, + "loss": 1.4401, + "theoretical_loss": 3.532092211346335, + "tokens_seen": 1425014784 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002870110330992979, + "loss": 1.2501, + "theoretical_loss": 3.5320777482563352, + "tokens_seen": 1425080320 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002870010030090271, + "loss": 1.2706, + "theoretical_loss": 3.5320632860176677, + "tokens_seen": 1425145856 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028699097291875624, + "loss": 1.4079, + "theoretical_loss": 3.532048824630243, + "tokens_seen": 1425211392 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028698094282848547, + "loss": 1.386, + "theoretical_loss": 3.532034364093971, + "tokens_seen": 1425276928 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028697091273821465, + "loss": 1.4523, + "theoretical_loss": 3.532019904408763, + "tokens_seen": 1425342464 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0006440713186748326, + "objective/train/docs_used": 808391, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.659785747528076, + "objective/train/original_loss": 2.659785509109497, + "objective/train/theoretical_loss": 3.5320054455745304, + "objective/train/tokens_used": 1445868000, + "objective/train/value_avg": -0.004634857177734375, + "objective/train/value_loss": 5.692466220352799e-05, + "objective/train/value_max": -0.0002846717834472656, + "objective/train/value_min": -0.08599853515625, + "objective/train/value_reward_corr": 0.4249714942460647, + "objective/train/value_std": 0.0037631988525390625, + "objective/train/weight_avg": 0.9962445497512817, + "objective/train/weighted_lm_loss": 2.6559090614318848, + "objective/train/weights_max": 1.7682840824127197, + "objective/train/weights_min": 0.5254842042922974, + "theoretical_loss": 3.5320054455745304, + "tokens_seen": 1425408000 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028696088264794383, + "loss": 1.3526, + "theoretical_loss": 3.5320054455745304, + "tokens_seen": 1425408000 + }, + { + "epoch": 0.43, + "learning_rate": 0.000286950852557673, + "loss": 1.4692, + "theoretical_loss": 3.5319909875911835, + "tokens_seen": 1425473536 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002869408224674022, + "loss": 1.366, + "theoretical_loss": 3.5319765304586324, + "tokens_seen": 1425539072 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002869307923771314, + "loss": 1.3362, + "theoretical_loss": 3.531962074176789, + "tokens_seen": 1425604608 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002869207622868606, + "loss": 1.3786, + "theoretical_loss": 3.531947618745564, + "tokens_seen": 1425670144 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028691073219658974, + "loss": 1.3904, + "theoretical_loss": 3.5319331641648675, + "tokens_seen": 1425735680 + }, + { + "epoch": 0.43, + "learning_rate": 0.000286900702106319, + "loss": 1.38, + "theoretical_loss": 3.531918710434611, + "tokens_seen": 1425801216 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002868906720160481, + "loss": 1.2807, + "theoretical_loss": 3.5319042575547055, + "tokens_seen": 1425866752 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028688064192577734, + "loss": 1.2743, + "theoretical_loss": 3.5318898055250614, + "tokens_seen": 1425932288 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002868706118355065, + "loss": 1.3163, + "theoretical_loss": 3.5318753543455896, + "tokens_seen": 1425997824 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002868605817452357, + "loss": 1.4161, + "theoretical_loss": 3.5318609040162015, + "tokens_seen": 1426063360 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002868505516549649, + "loss": 1.2684, + "theoretical_loss": 3.531846454536808, + "tokens_seen": 1426128896 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002868405215646941, + "loss": 1.3066, + "theoretical_loss": 3.5318320059073196, + "tokens_seen": 1426194432 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028683049147442324, + "loss": 1.384, + "theoretical_loss": 3.5318175581276474, + "tokens_seen": 1426259968 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002868204613841525, + "loss": 1.3625, + "theoretical_loss": 3.5318031111977026, + "tokens_seen": 1426325504 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002868104312938816, + "loss": 1.4101, + "theoretical_loss": 3.5317886651173955, + "tokens_seen": 1426391040 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028680040120361084, + "loss": 1.4058, + "theoretical_loss": 3.531774219886638, + "tokens_seen": 1426456576 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028679037111334, + "loss": 1.4067, + "theoretical_loss": 3.5317597755053414, + "tokens_seen": 1426522112 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002867803410230692, + "loss": 1.3125, + "theoretical_loss": 3.5317453319734153, + "tokens_seen": 1426587648 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028677031093279844, + "loss": 1.4502, + "theoretical_loss": 3.5317308892907717, + "tokens_seen": 1426653184 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028676028084252756, + "loss": 1.3713, + "theoretical_loss": 3.5317164474573217, + "tokens_seen": 1426718720 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002867502507522568, + "loss": 1.3394, + "theoretical_loss": 3.531702006472976, + "tokens_seen": 1426784256 + }, + { + "epoch": 0.43, + "learning_rate": 0.000286740220661986, + "loss": 1.4555, + "theoretical_loss": 3.5316875663376455, + "tokens_seen": 1426849792 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028673019057171516, + "loss": 1.3534, + "theoretical_loss": 3.531673127051242, + "tokens_seen": 1426915328 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028672016048144434, + "loss": 1.3527, + "theoretical_loss": 3.5316586886136756, + "tokens_seen": 1426980864 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -7.953577733132988e-05, + "objective/train/docs_used": 809012, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.592853546142578, + "objective/train/original_loss": 2.592853307723999, + "objective/train/theoretical_loss": 3.5316442510248587, + "objective/train/tokens_used": 1447506400, + "objective/train/value_avg": -0.0050201416015625, + "objective/train/value_loss": 5.0126720452681184e-05, + "objective/train/value_max": -0.0003249645233154297, + "objective/train/value_min": -0.0552978515625, + "objective/train/value_reward_corr": 0.3851962778818321, + "objective/train/value_std": 0.004024505615234375, + "objective/train/weight_avg": 1.0016063451766968, + "objective/train/weighted_lm_loss": 2.600778579711914, + "objective/train/weights_max": 1.738423228263855, + "objective/train/weights_min": 0.5311411619186401, + "theoretical_loss": 3.5316442510248587, + "tokens_seen": 1427046400 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002867101303911736, + "loss": 1.3436, + "theoretical_loss": 3.5316442510248587, + "tokens_seen": 1427046400 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002867001003009027, + "loss": 1.3952, + "theoretical_loss": 3.5316298142847016, + "tokens_seen": 1427111936 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028669007021063194, + "loss": 1.3277, + "theoretical_loss": 3.5316153783931155, + "tokens_seen": 1427177472 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028668004012036107, + "loss": 1.3443, + "theoretical_loss": 3.5316009433500115, + "tokens_seen": 1427243008 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002866700100300903, + "loss": 1.328, + "theoretical_loss": 3.531586509155301, + "tokens_seen": 1427308544 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002866599799398195, + "loss": 1.4796, + "theoretical_loss": 3.5315720758088953, + "tokens_seen": 1427374080 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028664994984954866, + "loss": 1.2611, + "theoretical_loss": 3.5315576433107054, + "tokens_seen": 1427439616 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028663991975927785, + "loss": 1.3768, + "theoretical_loss": 3.5315432116606424, + "tokens_seen": 1427505152 + }, + { + "epoch": 0.43, + "learning_rate": 0.000286629889669007, + "loss": 1.3488, + "theoretical_loss": 3.5315287808586175, + "tokens_seen": 1427570688 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002866198595787362, + "loss": 1.4094, + "theoretical_loss": 3.5315143509045424, + "tokens_seen": 1427636224 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028660982948846544, + "loss": 1.4148, + "theoretical_loss": 3.5314999217983285, + "tokens_seen": 1427701760 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028659979939819457, + "loss": 1.3942, + "theoretical_loss": 3.531485493539886, + "tokens_seen": 1427767296 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002865897693079238, + "loss": 1.4287, + "theoretical_loss": 3.531471066129127, + "tokens_seen": 1427832832 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028657973921765293, + "loss": 1.35, + "theoretical_loss": 3.5314566395659623, + "tokens_seen": 1427898368 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028656970912738217, + "loss": 1.3418, + "theoretical_loss": 3.5314422138503034, + "tokens_seen": 1427963904 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028655967903711135, + "loss": 1.3658, + "theoretical_loss": 3.5314277889820618, + "tokens_seen": 1428029440 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028654964894684053, + "loss": 1.2943, + "theoretical_loss": 3.531413364961149, + "tokens_seen": 1428094976 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002865396188565697, + "loss": 1.3395, + "theoretical_loss": 3.5313989417874763, + "tokens_seen": 1428160512 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028652958876629895, + "loss": 1.3751, + "theoretical_loss": 3.5313845194609543, + "tokens_seen": 1428226048 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002865195586760281, + "loss": 1.4753, + "theoretical_loss": 3.5313700979814957, + "tokens_seen": 1428291584 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002865095285857573, + "loss": 1.2712, + "theoretical_loss": 3.5313556773490102, + "tokens_seen": 1428357120 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028649949849548644, + "loss": 1.3321, + "theoretical_loss": 3.531341257563411, + "tokens_seen": 1428422656 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028648946840521567, + "loss": 1.3849, + "theoretical_loss": 3.531326838624608, + "tokens_seen": 1428488192 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028647943831494485, + "loss": 1.4887, + "theoretical_loss": 3.5313124205325135, + "tokens_seen": 1428553728 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028646940822467403, + "loss": 1.4067, + "theoretical_loss": 3.5312980032870387, + "tokens_seen": 1428619264 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0021094868425279856, + "objective/train/docs_used": 809683, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.581800699234009, + "objective/train/original_loss": 2.581800699234009, + "objective/train/theoretical_loss": 3.5312835868880947, + "objective/train/tokens_used": 1449144800, + "objective/train/value_avg": -0.006946563720703125, + "objective/train/value_loss": 0.00014884794654790312, + "objective/train/value_max": -0.0003981590270996094, + "objective/train/value_min": -0.07598876953125, + "objective/train/value_reward_corr": 0.580695804686818, + "objective/train/value_std": 0.005580902099609375, + "objective/train/weight_avg": 0.9856070876121521, + "objective/train/weighted_lm_loss": 2.565943717956543, + "objective/train/weights_max": 1.6385338306427002, + "objective/train/weights_min": 0.23453639447689056, + "theoretical_loss": 3.5312835868880947, + "tokens_seen": 1428684800 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002864593781344032, + "loss": 1.2787, + "theoretical_loss": 3.5312835868880947, + "tokens_seen": 1428684800 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002864493480441324, + "loss": 1.363, + "theoretical_loss": 3.5312691713355937, + "tokens_seen": 1428750336 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002864393179538616, + "loss": 1.2524, + "theoretical_loss": 3.5312547566294463, + "tokens_seen": 1428815872 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002864292878635908, + "loss": 1.427, + "theoretical_loss": 3.531240342769565, + "tokens_seen": 1428881408 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028641925777331994, + "loss": 1.3655, + "theoretical_loss": 3.531225929755861, + "tokens_seen": 1428946944 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002864092276830492, + "loss": 1.323, + "theoretical_loss": 3.531211517588245, + "tokens_seen": 1429012480 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002863991975927783, + "loss": 1.4007, + "theoretical_loss": 3.5311971062666294, + "tokens_seen": 1429078016 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028638916750250754, + "loss": 1.3357, + "theoretical_loss": 3.5311826957909256, + "tokens_seen": 1429143552 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002863791374122367, + "loss": 1.3389, + "theoretical_loss": 3.5311682861610447, + "tokens_seen": 1429209088 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002863691073219659, + "loss": 1.3699, + "theoretical_loss": 3.531153877376899, + "tokens_seen": 1429274624 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002863590772316951, + "loss": 1.2443, + "theoretical_loss": 3.5311394694383997, + "tokens_seen": 1429340160 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002863490471414243, + "loss": 1.3294, + "theoretical_loss": 3.531125062345458, + "tokens_seen": 1429405696 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028633901705115344, + "loss": 1.3002, + "theoretical_loss": 3.531110656097986, + "tokens_seen": 1429471232 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002863289869608827, + "loss": 1.3409, + "theoretical_loss": 3.531096250695896, + "tokens_seen": 1429536768 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002863189568706118, + "loss": 1.4228, + "theoretical_loss": 3.531081846139098, + "tokens_seen": 1429602304 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028630892678034104, + "loss": 1.4031, + "theoretical_loss": 3.5310674424275046, + "tokens_seen": 1429667840 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002862988966900702, + "loss": 1.382, + "theoretical_loss": 3.5310530395610273, + "tokens_seen": 1429733376 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002862888665997994, + "loss": 1.3786, + "theoretical_loss": 3.531038637539579, + "tokens_seen": 1429798912 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002862788365095286, + "loss": 1.2442, + "theoretical_loss": 3.531024236363069, + "tokens_seen": 1429864448 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028626880641925776, + "loss": 1.3972, + "theoretical_loss": 3.531009836031411, + "tokens_seen": 1429929984 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028625877632898694, + "loss": 1.3157, + "theoretical_loss": 3.530995436544516, + "tokens_seen": 1429995520 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002862487462387162, + "loss": 1.3407, + "theoretical_loss": 3.5309810379022952, + "tokens_seen": 1430061056 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002862387161484453, + "loss": 1.3172, + "theoretical_loss": 3.530966640104661, + "tokens_seen": 1430126592 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028622868605817454, + "loss": 1.427, + "theoretical_loss": 3.5309522431515257, + "tokens_seen": 1430192128 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028621865596790367, + "loss": 1.2606, + "theoretical_loss": 3.5309378470427997, + "tokens_seen": 1430257664 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0014301985502243042, + "objective/train/docs_used": 810937, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.471269130706787, + "objective/train/original_loss": 2.471269369125366, + "objective/train/theoretical_loss": 3.5309234517783956, + "objective/train/tokens_used": 1450783200, + "objective/train/value_avg": -0.00540924072265625, + "objective/train/value_loss": 7.25018689990975e-05, + "objective/train/value_max": -0.0001767873764038086, + "objective/train/value_min": -0.05877685546875, + "objective/train/value_reward_corr": 0.4070727155886973, + "objective/train/value_std": 0.0038433074951171875, + "objective/train/weight_avg": 0.9890615344047546, + "objective/train/weighted_lm_loss": 2.4637420177459717, + "objective/train/weights_max": 1.5480741262435913, + "objective/train/weights_min": 0.3494042754173279, + "theoretical_loss": 3.5309234517783956, + "tokens_seen": 1430323200 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002862086258776329, + "loss": 1.4015, + "theoretical_loss": 3.5309234517783956, + "tokens_seen": 1430323200 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002861985957873621, + "loss": 1.3536, + "theoretical_loss": 3.530909057358225, + "tokens_seen": 1430388736 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028618856569709127, + "loss": 1.3226, + "theoretical_loss": 3.5308946637822003, + "tokens_seen": 1430454272 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028617853560682045, + "loss": 1.4823, + "theoretical_loss": 3.5308802710502327, + "tokens_seen": 1430519808 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002861685055165497, + "loss": 1.3367, + "theoretical_loss": 3.530865879162234, + "tokens_seen": 1430585344 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002861584754262788, + "loss": 1.3825, + "theoretical_loss": 3.530851488118116, + "tokens_seen": 1430650880 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028614844533600805, + "loss": 1.47, + "theoretical_loss": 3.530837097917791, + "tokens_seen": 1430716416 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028613841524573717, + "loss": 1.311, + "theoretical_loss": 3.530822708561171, + "tokens_seen": 1430781952 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002861283851554664, + "loss": 1.3735, + "theoretical_loss": 3.530808320048167, + "tokens_seen": 1430847488 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002861183550651956, + "loss": 1.4423, + "theoretical_loss": 3.5307939323786917, + "tokens_seen": 1430913024 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028610832497492477, + "loss": 1.267, + "theoretical_loss": 3.5307795455526567, + "tokens_seen": 1430978560 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028609829488465395, + "loss": 1.3356, + "theoretical_loss": 3.5307651595699743, + "tokens_seen": 1431044096 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028608826479438313, + "loss": 1.3999, + "theoretical_loss": 3.5307507744305555, + "tokens_seen": 1431109632 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002860782347041123, + "loss": 1.2933, + "theoretical_loss": 3.5307363901343134, + "tokens_seen": 1431175168 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028606820461384155, + "loss": 1.376, + "theoretical_loss": 3.5307220066811595, + "tokens_seen": 1431240704 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002860581745235707, + "loss": 1.4547, + "theoretical_loss": 3.530707624071006, + "tokens_seen": 1431306240 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002860481444332999, + "loss": 1.3718, + "theoretical_loss": 3.530693242303764, + "tokens_seen": 1431371776 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002860381143430291, + "loss": 1.3698, + "theoretical_loss": 3.5306788613793465, + "tokens_seen": 1431437312 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002860280842527583, + "loss": 1.2907, + "theoretical_loss": 3.5306644812976655, + "tokens_seen": 1431502848 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002860180541624875, + "loss": 1.3434, + "theoretical_loss": 3.5306501020586323, + "tokens_seen": 1431568384 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028600802407221664, + "loss": 1.3029, + "theoretical_loss": 3.5306357236621597, + "tokens_seen": 1431633920 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028599799398194587, + "loss": 1.3454, + "theoretical_loss": 3.530621346108159, + "tokens_seen": 1431699456 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028598796389167505, + "loss": 1.4543, + "theoretical_loss": 3.5306069693965427, + "tokens_seen": 1431764992 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028597793380140423, + "loss": 1.3734, + "theoretical_loss": 3.5305925935272233, + "tokens_seen": 1431830528 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002859679037111334, + "loss": 1.3328, + "theoretical_loss": 3.5305782185001124, + "tokens_seen": 1431896064 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.000503413612022996, + "objective/train/docs_used": 811574, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7090227603912354, + "objective/train/original_loss": 2.7090229988098145, + "objective/train/theoretical_loss": 3.530563844315122, + "objective/train/tokens_used": 1452421600, + "objective/train/value_avg": -0.007587432861328125, + "objective/train/value_loss": 0.00011736081796698272, + "objective/train/value_max": -0.00047469139099121094, + "objective/train/value_min": -0.0830078125, + "objective/train/value_reward_corr": 0.5023641387410163, + "objective/train/value_std": 0.0065765380859375, + "objective/train/weight_avg": 1.0004148483276367, + "objective/train/weighted_lm_loss": 2.70167875289917, + "objective/train/weights_max": 1.7939351797103882, + "objective/train/weights_min": 0.3554268777370453, + "theoretical_loss": 3.530563844315122, + "tokens_seen": 1431961600 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002859578736208626, + "loss": 1.4015, + "theoretical_loss": 3.530563844315122, + "tokens_seen": 1431961600 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002859478435305918, + "loss": 1.312, + "theoretical_loss": 3.530549470972165, + "tokens_seen": 1432027136 + }, + { + "epoch": 0.43, + "learning_rate": 0.000285937813440321, + "loss": 1.2269, + "theoretical_loss": 3.5305350984711525, + "tokens_seen": 1432092672 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028592778335005014, + "loss": 1.3874, + "theoretical_loss": 3.5305207268119974, + "tokens_seen": 1432158208 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002859177532597794, + "loss": 1.3804, + "theoretical_loss": 3.5305063559946115, + "tokens_seen": 1432223744 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002859077231695085, + "loss": 1.3777, + "theoretical_loss": 3.530491986018907, + "tokens_seen": 1432289280 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028589769307923774, + "loss": 1.3037, + "theoretical_loss": 3.5304776168847964, + "tokens_seen": 1432354816 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002858876629889669, + "loss": 1.4065, + "theoretical_loss": 3.5304632485921914, + "tokens_seen": 1432420352 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002858776328986961, + "loss": 1.3873, + "theoretical_loss": 3.530448881141005, + "tokens_seen": 1432485888 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002858676028084253, + "loss": 1.3743, + "theoretical_loss": 3.530434514531149, + "tokens_seen": 1432551424 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002858575727181545, + "loss": 1.3352, + "theoretical_loss": 3.530420148762535, + "tokens_seen": 1432616960 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028584754262788364, + "loss": 1.4153, + "theoretical_loss": 3.530405783835076, + "tokens_seen": 1432682496 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002858375125376129, + "loss": 1.3234, + "theoretical_loss": 3.5303914197486845, + "tokens_seen": 1432748032 + }, + { + "epoch": 0.43, + "learning_rate": 0.000285827482447342, + "loss": 1.2634, + "theoretical_loss": 3.5303770565032724, + "tokens_seen": 1432813568 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028581745235707124, + "loss": 1.356, + "theoretical_loss": 3.530362694098752, + "tokens_seen": 1432879104 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002858074222668004, + "loss": 1.3828, + "theoretical_loss": 3.530348332535035, + "tokens_seen": 1432944640 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002857973921765296, + "loss": 1.3229, + "theoretical_loss": 3.530333971812035, + "tokens_seen": 1433010176 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002857873620862588, + "loss": 1.3856, + "theoretical_loss": 3.530319611929664, + "tokens_seen": 1433075712 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028577733199598796, + "loss": 1.3299, + "theoretical_loss": 3.5303052528878336, + "tokens_seen": 1433141248 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028576730190571714, + "loss": 1.3244, + "theoretical_loss": 3.530290894686457, + "tokens_seen": 1433206784 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002857572718154464, + "loss": 1.39, + "theoretical_loss": 3.530276537325446, + "tokens_seen": 1433272320 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002857472417251755, + "loss": 1.3906, + "theoretical_loss": 3.530262180804713, + "tokens_seen": 1433337856 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028573721163490474, + "loss": 1.3639, + "theoretical_loss": 3.530247825124171, + "tokens_seen": 1433403392 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028572718154463387, + "loss": 1.4385, + "theoretical_loss": 3.530233470283732, + "tokens_seen": 1433468928 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002857171514543631, + "loss": 1.3751, + "theoretical_loss": 3.530219116283308, + "tokens_seen": 1433534464 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": -0.0008962763822637498, + "objective/train/docs_used": 812859, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.792703151702881, + "objective/train/original_loss": 2.792703151702881, + "objective/train/theoretical_loss": 3.5302047631228124, + "objective/train/tokens_used": 1454060000, + "objective/train/value_avg": -0.0052337646484375, + "objective/train/value_loss": 5.389629222918302e-05, + "objective/train/value_max": -0.0003077983856201172, + "objective/train/value_min": -0.08819580078125, + "objective/train/value_reward_corr": 0.43415665373010326, + "objective/train/value_std": 0.00447845458984375, + "objective/train/weight_avg": 0.9935721158981323, + "objective/train/weighted_lm_loss": 2.7815310955047607, + "objective/train/weights_max": 1.6826401948928833, + "objective/train/weights_min": 0.24541936814785004, + "theoretical_loss": 3.5302047631228124, + "tokens_seen": 1433600000 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002857071213640923, + "loss": 1.4817, + "theoretical_loss": 3.5302047631228124, + "tokens_seen": 1433600000 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028569709127382147, + "loss": 1.4002, + "theoretical_loss": 3.530190410802157, + "tokens_seen": 1433665536 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028568706118355065, + "loss": 1.5215, + "theoretical_loss": 3.5301760593212537, + "tokens_seen": 1433731072 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002856770310932799, + "loss": 1.3604, + "theoretical_loss": 3.5301617086800166, + "tokens_seen": 1433796608 + }, + { + "epoch": 0.43, + "learning_rate": 0.000285667001003009, + "loss": 1.4101, + "theoretical_loss": 3.530147358878357, + "tokens_seen": 1433862144 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028565697091273825, + "loss": 1.2324, + "theoretical_loss": 3.530133009916188, + "tokens_seen": 1433927680 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028564694082246737, + "loss": 1.3404, + "theoretical_loss": 3.5301186617934217, + "tokens_seen": 1433993216 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002856369107321966, + "loss": 1.308, + "theoretical_loss": 3.5301043145099706, + "tokens_seen": 1434058752 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002856268806419258, + "loss": 1.3547, + "theoretical_loss": 3.530089968065748, + "tokens_seen": 1434124288 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028561685055165497, + "loss": 1.2758, + "theoretical_loss": 3.5300756224606653, + "tokens_seen": 1434189824 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028560682046138415, + "loss": 1.2498, + "theoretical_loss": 3.530061277694636, + "tokens_seen": 1434255360 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028559679037111333, + "loss": 1.2386, + "theoretical_loss": 3.530046933767572, + "tokens_seen": 1434320896 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002855867602808425, + "loss": 1.3533, + "theoretical_loss": 3.5300325906793866, + "tokens_seen": 1434386432 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028557673019057175, + "loss": 1.3454, + "theoretical_loss": 3.530018248429992, + "tokens_seen": 1434451968 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002855667001003009, + "loss": 1.3891, + "theoretical_loss": 3.530003907019301, + "tokens_seen": 1434517504 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002855566700100301, + "loss": 1.286, + "theoretical_loss": 3.5299895664472256, + "tokens_seen": 1434583040 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002855466399197593, + "loss": 1.2958, + "theoretical_loss": 3.5299752267136797, + "tokens_seen": 1434648576 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002855366098294885, + "loss": 1.3534, + "theoretical_loss": 3.529960887818575, + "tokens_seen": 1434714112 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028552657973921765, + "loss": 1.292, + "theoretical_loss": 3.5299465497618243, + "tokens_seen": 1434779648 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028551654964894684, + "loss": 1.3954, + "theoretical_loss": 3.5299322125433408, + "tokens_seen": 1434845184 + }, + { + "epoch": 0.43, + "learning_rate": 0.000285506519558676, + "loss": 1.2791, + "theoretical_loss": 3.5299178761630365, + "tokens_seen": 1434910720 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028549648946840525, + "loss": 1.4583, + "theoretical_loss": 3.529903540620825, + "tokens_seen": 1434976256 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002854864593781344, + "loss": 1.3667, + "theoretical_loss": 3.5298892059166183, + "tokens_seen": 1435041792 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002854764292878636, + "loss": 1.5104, + "theoretical_loss": 3.5298748720503292, + "tokens_seen": 1435107328 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028546639919759274, + "loss": 1.3881, + "theoretical_loss": 3.5298605390218705, + "tokens_seen": 1435172864 + }, + { + "epoch": 0.43, + "objective/train/advantage_avg": 0.0003240380610805005, + "objective/train/docs_used": 813625, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.349976062774658, + "objective/train/original_loss": 2.3499763011932373, + "objective/train/theoretical_loss": 3.5298462068311554, + "objective/train/tokens_used": 1455698400, + "objective/train/value_avg": -0.00522613525390625, + "objective/train/value_loss": 5.610838343272917e-05, + "objective/train/value_max": -0.00038909912109375, + "objective/train/value_min": -0.07696533203125, + "objective/train/value_reward_corr": 0.2822562371031193, + "objective/train/value_std": 0.00405120849609375, + "objective/train/weight_avg": 1.005942940711975, + "objective/train/weighted_lm_loss": 2.3686704635620117, + "objective/train/weights_max": 2.159017562866211, + "objective/train/weights_min": 0.37085655331611633, + "theoretical_loss": 3.5298462068311554, + "tokens_seen": 1435238400 + }, + { + "epoch": 0.43, + "learning_rate": 0.000285456369107322, + "loss": 1.3313, + "theoretical_loss": 3.5298462068311554, + "tokens_seen": 1435238400 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028544633901705116, + "loss": 1.386, + "theoretical_loss": 3.529831875478096, + "tokens_seen": 1435303936 + }, + { + "epoch": 0.43, + "learning_rate": 0.00028543630892678034, + "loss": 1.4546, + "theoretical_loss": 3.5298175449626057, + "tokens_seen": 1435369472 + }, + { + "epoch": 0.43, + "learning_rate": 0.0002854262788365095, + "loss": 1.3376, + "theoretical_loss": 3.529803215284597, + "tokens_seen": 1435435008 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002854162487462387, + "loss": 1.3025, + "theoretical_loss": 3.529788886443983, + "tokens_seen": 1435500544 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002854062186559679, + "loss": 1.4304, + "theoretical_loss": 3.5297745584406766, + "tokens_seen": 1435566080 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002853961885656971, + "loss": 1.4512, + "theoretical_loss": 3.5297602312745897, + "tokens_seen": 1435631616 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028538615847542624, + "loss": 1.3237, + "theoretical_loss": 3.5297459049456363, + "tokens_seen": 1435697152 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002853761283851555, + "loss": 1.2878, + "theoretical_loss": 3.5297315794537285, + "tokens_seen": 1435762688 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028536609829488466, + "loss": 1.3668, + "theoretical_loss": 3.52971725479878, + "tokens_seen": 1435828224 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028535606820461384, + "loss": 1.2745, + "theoretical_loss": 3.5297029309807026, + "tokens_seen": 1435893760 + }, + { + "epoch": 0.44, + "learning_rate": 0.000285346038114343, + "loss": 1.4294, + "theoretical_loss": 3.5296886079994105, + "tokens_seen": 1435959296 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002853360080240722, + "loss": 1.3752, + "theoretical_loss": 3.529674285854816, + "tokens_seen": 1436024832 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002853259779338014, + "loss": 1.535, + "theoretical_loss": 3.5296599645468314, + "tokens_seen": 1436090368 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002853159478435306, + "loss": 1.2868, + "theoretical_loss": 3.5296456440753707, + "tokens_seen": 1436155904 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028530591775325975, + "loss": 1.4367, + "theoretical_loss": 3.5296313244403463, + "tokens_seen": 1436221440 + }, + { + "epoch": 0.44, + "learning_rate": 0.000285295887662989, + "loss": 1.2665, + "theoretical_loss": 3.529617005641671, + "tokens_seen": 1436286976 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002852858575727181, + "loss": 1.3611, + "theoretical_loss": 3.529602687679259, + "tokens_seen": 1436352512 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028527582748244735, + "loss": 1.4383, + "theoretical_loss": 3.529588370553022, + "tokens_seen": 1436418048 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002852657973921766, + "loss": 1.3952, + "theoretical_loss": 3.529574054262873, + "tokens_seen": 1436483584 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002852557673019057, + "loss": 1.2696, + "theoretical_loss": 3.529559738808726, + "tokens_seen": 1436549120 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028524573721163494, + "loss": 1.3537, + "theoretical_loss": 3.529545424190493, + "tokens_seen": 1436614656 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028523570712136407, + "loss": 1.3056, + "theoretical_loss": 3.5295311104080884, + "tokens_seen": 1436680192 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002852256770310933, + "loss": 1.3417, + "theoretical_loss": 3.529516797461424, + "tokens_seen": 1436745728 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002852156469408225, + "loss": 1.3009, + "theoretical_loss": 3.529502485350413, + "tokens_seen": 1436811264 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": 0.00018793682102113962, + "objective/train/docs_used": 814744, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4950809478759766, + "objective/train/original_loss": 2.4950807094573975, + "objective/train/theoretical_loss": 3.5294881740749693, + "objective/train/tokens_used": 1457336800, + "objective/train/value_avg": -0.00516510009765625, + "objective/train/value_loss": 0.00010534859029576182, + "objective/train/value_max": -0.00031757354736328125, + "objective/train/value_min": -0.04913330078125, + "objective/train/value_reward_corr": 0.2015412005767197, + "objective/train/value_std": 0.0038089752197265625, + "objective/train/weight_avg": 1.0059187412261963, + "objective/train/weighted_lm_loss": 2.516106128692627, + "objective/train/weights_max": 1.4238914251327515, + "objective/train/weights_min": 0.13943061232566833, + "theoretical_loss": 3.5294881740749693, + "tokens_seen": 1436876800 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028520561685055167, + "loss": 1.3176, + "theoretical_loss": 3.5294881740749693, + "tokens_seen": 1436876800 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028519558676028085, + "loss": 1.3695, + "theoretical_loss": 3.5294738636350056, + "tokens_seen": 1436942336 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002851855566700101, + "loss": 1.3388, + "theoretical_loss": 3.5294595540304345, + "tokens_seen": 1437007872 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002851755265797392, + "loss": 1.2806, + "theoretical_loss": 3.52944524526117, + "tokens_seen": 1437073408 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028516549648946845, + "loss": 1.3674, + "theoretical_loss": 3.529430937327125, + "tokens_seen": 1437138944 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028515546639919757, + "loss": 1.2828, + "theoretical_loss": 3.529416630228212, + "tokens_seen": 1437204480 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002851454363089268, + "loss": 1.2734, + "theoretical_loss": 3.5294023239643453, + "tokens_seen": 1437270016 + }, + { + "epoch": 0.44, + "learning_rate": 0.000285135406218656, + "loss": 1.4052, + "theoretical_loss": 3.529388018535437, + "tokens_seen": 1437335552 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028512537612838517, + "loss": 1.4014, + "theoretical_loss": 3.5293737139414008, + "tokens_seen": 1437401088 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028511534603811435, + "loss": 1.3085, + "theoretical_loss": 3.52935941018215, + "tokens_seen": 1437466624 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028510531594784353, + "loss": 1.3396, + "theoretical_loss": 3.5293451072575976, + "tokens_seen": 1437532160 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002850952858575727, + "loss": 1.3511, + "theoretical_loss": 3.5293308051676573, + "tokens_seen": 1437597696 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028508525576730195, + "loss": 1.3707, + "theoretical_loss": 3.529316503912242, + "tokens_seen": 1437663232 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002850752256770311, + "loss": 1.3462, + "theoretical_loss": 3.529302203491265, + "tokens_seen": 1437728768 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002850651955867603, + "loss": 1.2157, + "theoretical_loss": 3.529287903904639, + "tokens_seen": 1437794304 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002850551654964895, + "loss": 1.4695, + "theoretical_loss": 3.529273605152279, + "tokens_seen": 1437859840 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002850451354062187, + "loss": 1.4399, + "theoretical_loss": 3.529259307234096, + "tokens_seen": 1437925376 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028503510531594785, + "loss": 1.4032, + "theoretical_loss": 3.5292450101500052, + "tokens_seen": 1437990912 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028502507522567704, + "loss": 1.2119, + "theoretical_loss": 3.529230713899919, + "tokens_seen": 1438056448 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002850150451354062, + "loss": 1.334, + "theoretical_loss": 3.529216418483751, + "tokens_seen": 1438121984 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028500501504513545, + "loss": 1.3826, + "theoretical_loss": 3.529202123901414, + "tokens_seen": 1438187520 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002849949849548646, + "loss": 1.303, + "theoretical_loss": 3.5291878301528232, + "tokens_seen": 1438253056 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002849849548645938, + "loss": 1.3676, + "theoretical_loss": 3.52917353723789, + "tokens_seen": 1438318592 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028497492477432294, + "loss": 1.1834, + "theoretical_loss": 3.529159245156528, + "tokens_seen": 1438384128 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002849648946840522, + "loss": 1.3619, + "theoretical_loss": 3.5291449539086512, + "tokens_seen": 1438449664 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.000518488755915314, + "objective/train/docs_used": 815309, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8959696292877197, + "objective/train/original_loss": 2.8959696292877197, + "objective/train/theoretical_loss": 3.5291306634941737, + "objective/train/tokens_used": 1458975200, + "objective/train/value_avg": -0.005615234375, + "objective/train/value_loss": 7.655517401872203e-05, + "objective/train/value_max": -0.00030303001403808594, + "objective/train/value_min": -0.09912109375, + "objective/train/value_reward_corr": 0.4909756035442817, + "objective/train/value_std": 0.0052642822265625, + "objective/train/weight_avg": 0.9983570575714111, + "objective/train/weighted_lm_loss": 2.8834502696990967, + "objective/train/weights_max": 2.19246244430542, + "objective/train/weights_min": 0.3191172778606415, + "theoretical_loss": 3.5291306634941737, + "tokens_seen": 1438515200 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028495486459378136, + "loss": 1.3977, + "theoretical_loss": 3.5291306634941737, + "tokens_seen": 1438515200 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028494483450351054, + "loss": 1.2797, + "theoretical_loss": 3.5291163739130074, + "tokens_seen": 1438580736 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002849348044132397, + "loss": 1.3721, + "theoretical_loss": 3.529102085165067, + "tokens_seen": 1438646272 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002849247743229689, + "loss": 1.3354, + "theoretical_loss": 3.5290877972502646, + "tokens_seen": 1438711808 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002849147442326981, + "loss": 1.3279, + "theoretical_loss": 3.529073510168516, + "tokens_seen": 1438777344 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002849047141424273, + "loss": 1.2584, + "theoretical_loss": 3.529059223919732, + "tokens_seen": 1438842880 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028489468405215644, + "loss": 1.2521, + "theoretical_loss": 3.5290449385038274, + "tokens_seen": 1438908416 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002848846539618857, + "loss": 1.3484, + "theoretical_loss": 3.529030653920716, + "tokens_seen": 1438973952 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028487462387161486, + "loss": 1.3198, + "theoretical_loss": 3.529016370170311, + "tokens_seen": 1439039488 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028486459378134404, + "loss": 1.5275, + "theoretical_loss": 3.5290020872525254, + "tokens_seen": 1439105024 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002848545636910732, + "loss": 1.3089, + "theoretical_loss": 3.5289878051672736, + "tokens_seen": 1439170560 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002848445336008024, + "loss": 1.3364, + "theoretical_loss": 3.528973523914469, + "tokens_seen": 1439236096 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002848345035105316, + "loss": 1.419, + "theoretical_loss": 3.5289592434940245, + "tokens_seen": 1439301632 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002848244734202608, + "loss": 1.3723, + "theoretical_loss": 3.5289449639058543, + "tokens_seen": 1439367168 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028481444332998995, + "loss": 1.311, + "theoretical_loss": 3.5289306851498723, + "tokens_seen": 1439432704 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002848044132397192, + "loss": 1.2704, + "theoretical_loss": 3.5289164072259913, + "tokens_seen": 1439498240 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002847943831494483, + "loss": 1.2839, + "theoretical_loss": 3.5289021301341252, + "tokens_seen": 1439563776 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028478435305917755, + "loss": 1.3645, + "theoretical_loss": 3.5288878538741884, + "tokens_seen": 1439629312 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002847743229689067, + "loss": 1.2206, + "theoretical_loss": 3.5288735784460936, + "tokens_seen": 1439694848 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002847642928786359, + "loss": 1.3749, + "theoretical_loss": 3.5288593038497544, + "tokens_seen": 1439760384 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002847542627883651, + "loss": 1.3029, + "theoretical_loss": 3.528845030085085, + "tokens_seen": 1439825920 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028474423269809427, + "loss": 1.2756, + "theoretical_loss": 3.528830757152, + "tokens_seen": 1439891456 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028473420260782345, + "loss": 1.2992, + "theoretical_loss": 3.528816485050411, + "tokens_seen": 1439956992 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002847241725175527, + "loss": 1.3161, + "theoretical_loss": 3.528802213780233, + "tokens_seen": 1440022528 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002847141424272818, + "loss": 1.116, + "theoretical_loss": 3.5287879433413796, + "tokens_seen": 1440088064 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.0013436770532280207, + "objective/train/docs_used": 816515, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8265609741210938, + "objective/train/original_loss": 2.826561212539673, + "objective/train/theoretical_loss": 3.528773673733764, + "objective/train/tokens_used": 1460613600, + "objective/train/value_avg": -0.0061492919921875, + "objective/train/value_loss": 7.008708053035662e-05, + "objective/train/value_max": -0.0004029273986816406, + "objective/train/value_min": -0.06658935546875, + "objective/train/value_reward_corr": 0.4334784471965483, + "objective/train/value_std": 0.005176544189453125, + "objective/train/weight_avg": 0.9899373650550842, + "objective/train/weighted_lm_loss": 2.796156644821167, + "objective/train/weights_max": 1.9462287425994873, + "objective/train/weights_min": 0.5851883888244629, + "theoretical_loss": 3.528773673733764, + "tokens_seen": 1440153600 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028470411233701105, + "loss": 1.4196, + "theoretical_loss": 3.528773673733764, + "tokens_seen": 1440153600 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028469408224674023, + "loss": 1.3511, + "theoretical_loss": 3.5287594049573014, + "tokens_seen": 1440219136 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002846840521564694, + "loss": 1.3271, + "theoretical_loss": 3.528745137011904, + "tokens_seen": 1440284672 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002846740220661986, + "loss": 1.4088, + "theoretical_loss": 3.528730869897486, + "tokens_seen": 1440350208 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002846639919759278, + "loss": 1.4069, + "theoretical_loss": 3.5287166036139617, + "tokens_seen": 1440415744 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028465396188565695, + "loss": 1.4146, + "theoretical_loss": 3.528702338161245, + "tokens_seen": 1440481280 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002846439317953862, + "loss": 1.2668, + "theoretical_loss": 3.5286880735392483, + "tokens_seen": 1440546816 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002846339017051153, + "loss": 1.3653, + "theoretical_loss": 3.528673809747887, + "tokens_seen": 1440612352 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028462387161484455, + "loss": 1.3473, + "theoretical_loss": 3.5286595467870745, + "tokens_seen": 1440677888 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002846138415245737, + "loss": 1.265, + "theoretical_loss": 3.5286452846567244, + "tokens_seen": 1440743424 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002846038114343029, + "loss": 1.3964, + "theoretical_loss": 3.528631023356751, + "tokens_seen": 1440808960 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002845937813440321, + "loss": 1.4451, + "theoretical_loss": 3.5286167628870677, + "tokens_seen": 1440874496 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002845837512537613, + "loss": 1.3391, + "theoretical_loss": 3.5286025032475883, + "tokens_seen": 1440940032 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028457372116349046, + "loss": 1.4081, + "theoretical_loss": 3.5285882444382275, + "tokens_seen": 1441005568 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002845636910732197, + "loss": 1.309, + "theoretical_loss": 3.5285739864588983, + "tokens_seen": 1441071104 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002845536609829488, + "loss": 1.357, + "theoretical_loss": 3.5285597293095154, + "tokens_seen": 1441136640 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028454363089267805, + "loss": 1.3242, + "theoretical_loss": 3.5285454729899923, + "tokens_seen": 1441202176 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028453360080240724, + "loss": 1.4458, + "theoretical_loss": 3.5285312175002432, + "tokens_seen": 1441267712 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002845235707121364, + "loss": 1.3261, + "theoretical_loss": 3.528516962840182, + "tokens_seen": 1441333248 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028451354062186565, + "loss": 1.3107, + "theoretical_loss": 3.5285027090097225, + "tokens_seen": 1441398784 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002845035105315948, + "loss": 1.3072, + "theoretical_loss": 3.5284884560087795, + "tokens_seen": 1441464320 + }, + { + "epoch": 0.44, + "learning_rate": 0.000284493480441324, + "loss": 1.4009, + "theoretical_loss": 3.5284742038372654, + "tokens_seen": 1441529856 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028448345035105314, + "loss": 1.4096, + "theoretical_loss": 3.528459952495096, + "tokens_seen": 1441595392 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002844734202607824, + "loss": 1.3471, + "theoretical_loss": 3.528445701982184, + "tokens_seen": 1441660928 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028446339017051156, + "loss": 1.3542, + "theoretical_loss": 3.528431452298445, + "tokens_seen": 1441726464 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.0014218437718227506, + "objective/train/docs_used": 817059, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6817209720611572, + "objective/train/original_loss": 2.6817209720611572, + "objective/train/theoretical_loss": 3.528417203443791, + "objective/train/tokens_used": 1462252000, + "objective/train/value_avg": -0.005275726318359375, + "objective/train/value_loss": 0.0001048085032380186, + "objective/train/value_max": -0.0002269744873046875, + "objective/train/value_min": -0.04986572265625, + "objective/train/value_reward_corr": 0.48704083319728037, + "objective/train/value_std": 0.004505157470703125, + "objective/train/weight_avg": 0.9905960559844971, + "objective/train/weighted_lm_loss": 2.657762050628662, + "objective/train/weights_max": 1.4934436082839966, + "objective/train/weights_min": 0.48446348309516907, + "theoretical_loss": 3.528417203443791, + "tokens_seen": 1441792000 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028445336008024074, + "loss": 1.3723, + "theoretical_loss": 3.528417203443791, + "tokens_seen": 1441792000 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002844433299899699, + "loss": 1.3489, + "theoretical_loss": 3.5284029554181373, + "tokens_seen": 1441857536 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002844332998996991, + "loss": 1.2296, + "theoretical_loss": 3.5283887082213985, + "tokens_seen": 1441923072 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002844232698094283, + "loss": 1.351, + "theoretical_loss": 3.5283744618534874, + "tokens_seen": 1441988608 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002844132397191575, + "loss": 1.2838, + "theoretical_loss": 3.5283602163143186, + "tokens_seen": 1442054144 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028440320962888664, + "loss": 1.3765, + "theoretical_loss": 3.528345971603807, + "tokens_seen": 1442119680 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002843931795386159, + "loss": 1.4172, + "theoretical_loss": 3.5283317277218664, + "tokens_seen": 1442185216 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028438314944834506, + "loss": 1.4178, + "theoretical_loss": 3.5283174846684098, + "tokens_seen": 1442250752 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028437311935807424, + "loss": 1.4431, + "theoretical_loss": 3.5283032424433527, + "tokens_seen": 1442316288 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002843630892678034, + "loss": 1.3273, + "theoretical_loss": 3.528289001046609, + "tokens_seen": 1442381824 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002843530591775326, + "loss": 1.3841, + "theoretical_loss": 3.5282747604780926, + "tokens_seen": 1442447360 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002843430290872618, + "loss": 1.2926, + "theoretical_loss": 3.528260520737718, + "tokens_seen": 1442512896 + }, + { + "epoch": 0.44, + "learning_rate": 0.000284332998996991, + "loss": 1.2766, + "theoretical_loss": 3.528246281825399, + "tokens_seen": 1442578432 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028432296890672015, + "loss": 1.3786, + "theoretical_loss": 3.52823204374105, + "tokens_seen": 1442643968 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002843129388164494, + "loss": 1.4775, + "theoretical_loss": 3.5282178064845855, + "tokens_seen": 1442709504 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002843029087261785, + "loss": 1.3911, + "theoretical_loss": 3.52820357005592, + "tokens_seen": 1442775040 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028429287863590775, + "loss": 1.3682, + "theoretical_loss": 3.528189334454967, + "tokens_seen": 1442840576 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002842828485456369, + "loss": 1.3807, + "theoretical_loss": 3.528175099681641, + "tokens_seen": 1442906112 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002842728184553661, + "loss": 1.3524, + "theoretical_loss": 3.528160865735857, + "tokens_seen": 1442971648 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002842627883650953, + "loss": 1.3012, + "theoretical_loss": 3.528146632617528, + "tokens_seen": 1443037184 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028425275827482447, + "loss": 1.244, + "theoretical_loss": 3.5281324003265695, + "tokens_seen": 1443102720 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028424272818455365, + "loss": 1.4126, + "theoretical_loss": 3.5281181688628953, + "tokens_seen": 1443168256 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002842326980942829, + "loss": 1.3224, + "theoretical_loss": 3.5281039382264203, + "tokens_seen": 1443233792 + }, + { + "epoch": 0.44, + "learning_rate": 0.000284222668004012, + "loss": 1.3712, + "theoretical_loss": 3.528089708417058, + "tokens_seen": 1443299328 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028421263791374125, + "loss": 1.4062, + "theoretical_loss": 3.5280754794347233, + "tokens_seen": 1443364864 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.0007357292342931032, + "objective/train/docs_used": 818508, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7948403358459473, + "objective/train/original_loss": 2.7948405742645264, + "objective/train/theoretical_loss": 3.528061251279331, + "objective/train/tokens_used": 1463890400, + "objective/train/value_avg": -0.0055389404296875, + "objective/train/value_loss": 6.500659947050735e-05, + "objective/train/value_max": -0.0003380775451660156, + "objective/train/value_min": -0.09637451171875, + "objective/train/value_reward_corr": 0.42221707994100344, + "objective/train/value_std": 0.004665374755859375, + "objective/train/weight_avg": 0.9956671595573425, + "objective/train/weighted_lm_loss": 2.7764830589294434, + "objective/train/weights_max": 2.3704936504364014, + "objective/train/weights_min": 0.30437374114990234, + "theoretical_loss": 3.528061251279331, + "tokens_seen": 1443430400 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028420260782347043, + "loss": 1.3421, + "theoretical_loss": 3.528061251279331, + "tokens_seen": 1443430400 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002841925777331996, + "loss": 1.3091, + "theoretical_loss": 3.528047023950794, + "tokens_seen": 1443495936 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002841825476429288, + "loss": 1.3458, + "theoretical_loss": 3.528032797449028, + "tokens_seen": 1443561472 + }, + { + "epoch": 0.44, + "learning_rate": 0.000284172517552658, + "loss": 1.4438, + "theoretical_loss": 3.5280185717739476, + "tokens_seen": 1443627008 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028416248746238715, + "loss": 1.3235, + "theoretical_loss": 3.5280043469254663, + "tokens_seen": 1443692544 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002841524573721164, + "loss": 1.3254, + "theoretical_loss": 3.5279901229034993, + "tokens_seen": 1443758080 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002841424272818455, + "loss": 1.4168, + "theoretical_loss": 3.5279758997079607, + "tokens_seen": 1443823616 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028413239719157475, + "loss": 1.4467, + "theoretical_loss": 3.527961677338765, + "tokens_seen": 1443889152 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002841223671013039, + "loss": 1.4113, + "theoretical_loss": 3.527947455795827, + "tokens_seen": 1443954688 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002841123370110331, + "loss": 1.3041, + "theoretical_loss": 3.52793323507906, + "tokens_seen": 1444020224 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002841023069207623, + "loss": 1.3725, + "theoretical_loss": 3.5279190151883806, + "tokens_seen": 1444085760 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002840922768304915, + "loss": 1.3614, + "theoretical_loss": 3.5279047961237016, + "tokens_seen": 1444151296 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028408224674022066, + "loss": 1.2961, + "theoretical_loss": 3.5278905778849383, + "tokens_seen": 1444216832 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002840722166499499, + "loss": 1.2264, + "theoretical_loss": 3.527876360472005, + "tokens_seen": 1444282368 + }, + { + "epoch": 0.44, + "learning_rate": 0.000284062186559679, + "loss": 1.4393, + "theoretical_loss": 3.5278621438848163, + "tokens_seen": 1444347904 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028405215646940825, + "loss": 1.3368, + "theoretical_loss": 3.527847928123287, + "tokens_seen": 1444413440 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002840421263791374, + "loss": 1.4063, + "theoretical_loss": 3.5278337131873307, + "tokens_seen": 1444478976 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002840320962888666, + "loss": 1.2733, + "theoretical_loss": 3.5278194990768634, + "tokens_seen": 1444544512 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002840220661985958, + "loss": 1.2514, + "theoretical_loss": 3.5278052857917994, + "tokens_seen": 1444610048 + }, + { + "epoch": 0.44, + "learning_rate": 0.000284012036108325, + "loss": 1.4371, + "theoretical_loss": 3.527791073332053, + "tokens_seen": 1444675584 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028400200601805416, + "loss": 1.4345, + "theoretical_loss": 3.5277768616975385, + "tokens_seen": 1444741120 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028399197592778334, + "loss": 1.4192, + "theoretical_loss": 3.527762650888171, + "tokens_seen": 1444806656 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002839819458375125, + "loss": 1.4029, + "theoretical_loss": 3.5277484409038653, + "tokens_seen": 1444872192 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028397191574724176, + "loss": 1.3557, + "theoretical_loss": 3.5277342317445353, + "tokens_seen": 1444937728 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002839618856569709, + "loss": 1.4144, + "theoretical_loss": 3.5277200234100965, + "tokens_seen": 1445003264 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.0007292371592484415, + "objective/train/docs_used": 819330, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.0144717693328857, + "objective/train/original_loss": 2.014472007751465, + "objective/train/theoretical_loss": 3.5277058159004637, + "objective/train/tokens_used": 1465528800, + "objective/train/value_avg": -0.005096435546875, + "objective/train/value_loss": 4.792102845385671e-05, + "objective/train/value_max": -0.00046563148498535156, + "objective/train/value_min": -0.0753173828125, + "objective/train/value_reward_corr": 0.2705389895999796, + "objective/train/value_std": 0.0037689208984375, + "objective/train/weight_avg": 0.9949880838394165, + "objective/train/weighted_lm_loss": 2.0105977058410645, + "objective/train/weights_max": 1.6117359399795532, + "objective/train/weights_min": 0.40678587555885315, + "theoretical_loss": 3.5277058159004637, + "tokens_seen": 1445068800 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002839518555667001, + "loss": 1.3419, + "theoretical_loss": 3.5277058159004637, + "tokens_seen": 1445068800 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028394182547642925, + "loss": 1.3621, + "theoretical_loss": 3.5276916092155512, + "tokens_seen": 1445134336 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002839317953861585, + "loss": 1.3965, + "theoretical_loss": 3.527677403355274, + "tokens_seen": 1445199872 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028392176529588766, + "loss": 1.4212, + "theoretical_loss": 3.527663198319546, + "tokens_seen": 1445265408 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028391173520561684, + "loss": 1.2042, + "theoretical_loss": 3.527648994108283, + "tokens_seen": 1445330944 + }, + { + "epoch": 0.44, + "learning_rate": 0.000283901705115346, + "loss": 1.3665, + "theoretical_loss": 3.5276347907213994, + "tokens_seen": 1445396480 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028389167502507526, + "loss": 1.3315, + "theoretical_loss": 3.5276205881588103, + "tokens_seen": 1445462016 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002838816449348044, + "loss": 1.3559, + "theoretical_loss": 3.5276063864204295, + "tokens_seen": 1445527552 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002838716148445336, + "loss": 1.2985, + "theoretical_loss": 3.527592185506173, + "tokens_seen": 1445593088 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028386158475426275, + "loss": 1.3867, + "theoretical_loss": 3.527577985415955, + "tokens_seen": 1445658624 + }, + { + "epoch": 0.44, + "learning_rate": 0.000283851554663992, + "loss": 1.3908, + "theoretical_loss": 3.52756378614969, + "tokens_seen": 1445724160 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028384152457372117, + "loss": 1.341, + "theoretical_loss": 3.5275495877072935, + "tokens_seen": 1445789696 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028383149448345035, + "loss": 1.5026, + "theoretical_loss": 3.52753539008868, + "tokens_seen": 1445855232 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028382146439317953, + "loss": 1.462, + "theoretical_loss": 3.5275211932937647, + "tokens_seen": 1445920768 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002838114343029087, + "loss": 1.3748, + "theoretical_loss": 3.5275069973224618, + "tokens_seen": 1445986304 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002838014042126379, + "loss": 1.2794, + "theoretical_loss": 3.527492802174687, + "tokens_seen": 1446051840 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002837913741223671, + "loss": 1.3983, + "theoretical_loss": 3.5274786078503553, + "tokens_seen": 1446117376 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002837813440320963, + "loss": 1.3844, + "theoretical_loss": 3.5274644143493807, + "tokens_seen": 1446182912 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002837713139418255, + "loss": 1.3656, + "theoretical_loss": 3.5274502216716783, + "tokens_seen": 1446248448 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028376128385155467, + "loss": 1.4132, + "theoretical_loss": 3.5274360298171636, + "tokens_seen": 1446313984 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028375125376128385, + "loss": 1.3804, + "theoretical_loss": 3.5274218387857514, + "tokens_seen": 1446379520 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002837412236710131, + "loss": 1.3893, + "theoretical_loss": 3.5274076485773564, + "tokens_seen": 1446445056 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002837311935807422, + "loss": 1.4517, + "theoretical_loss": 3.5273934591918943, + "tokens_seen": 1446510592 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028372116349047145, + "loss": 1.3097, + "theoretical_loss": 3.527379270629279, + "tokens_seen": 1446576128 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028371113340020063, + "loss": 1.231, + "theoretical_loss": 3.527365082889426, + "tokens_seen": 1446641664 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.0005933484644629061, + "objective/train/docs_used": 820636, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7461273670196533, + "objective/train/original_loss": 2.746127128601074, + "objective/train/theoretical_loss": 3.5273508959722504, + "objective/train/tokens_used": 1467167200, + "objective/train/value_avg": -0.005504608154296875, + "objective/train/value_loss": 6.890024815220386e-05, + "objective/train/value_max": -0.00039196014404296875, + "objective/train/value_min": -0.06890869140625, + "objective/train/value_reward_corr": 0.34997415103028223, + "objective/train/value_std": 0.004100799560546875, + "objective/train/weight_avg": 0.9972702264785767, + "objective/train/weighted_lm_loss": 2.7362124919891357, + "objective/train/weights_max": 1.5607781410217285, + "objective/train/weights_min": 0.3387666642665863, + "theoretical_loss": 3.5273508959722504, + "tokens_seen": 1446707200 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002837011033099298, + "loss": 1.389, + "theoretical_loss": 3.5273508959722504, + "tokens_seen": 1446707200 + }, + { + "epoch": 0.44, + "learning_rate": 0.000283691073219659, + "loss": 1.2723, + "theoretical_loss": 3.527336709877668, + "tokens_seen": 1446772736 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002836810431293882, + "loss": 1.3488, + "theoretical_loss": 3.527322524605592, + "tokens_seen": 1446838272 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028367101303911735, + "loss": 1.3685, + "theoretical_loss": 3.527308340155939, + "tokens_seen": 1446903808 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002836609829488466, + "loss": 1.4329, + "theoretical_loss": 3.5272941565286238, + "tokens_seen": 1446969344 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002836509528585757, + "loss": 1.5085, + "theoretical_loss": 3.527279973723561, + "tokens_seen": 1447034880 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028364092276830495, + "loss": 1.2885, + "theoretical_loss": 3.527265791740666, + "tokens_seen": 1447100416 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002836308926780341, + "loss": 1.3411, + "theoretical_loss": 3.527251610579854, + "tokens_seen": 1447165952 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002836208625877633, + "loss": 1.2902, + "theoretical_loss": 3.5272374302410396, + "tokens_seen": 1447231488 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002836108324974925, + "loss": 1.3346, + "theoretical_loss": 3.5272232507241386, + "tokens_seen": 1447297024 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002836008024072217, + "loss": 1.2562, + "theoretical_loss": 3.527209072029066, + "tokens_seen": 1447362560 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028359077231695086, + "loss": 1.3726, + "theoretical_loss": 3.5271948941557367, + "tokens_seen": 1447428096 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002835807422266801, + "loss": 1.3874, + "theoretical_loss": 3.527180717104066, + "tokens_seen": 1447493632 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002835707121364092, + "loss": 1.3544, + "theoretical_loss": 3.527166540873969, + "tokens_seen": 1447559168 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028356068204613846, + "loss": 1.3276, + "theoretical_loss": 3.527152365465361, + "tokens_seen": 1447624704 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002835506519558676, + "loss": 1.2905, + "theoretical_loss": 3.5271381908781576, + "tokens_seen": 1447690240 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002835406218655968, + "loss": 1.3916, + "theoretical_loss": 3.527124017112273, + "tokens_seen": 1447755776 + }, + { + "epoch": 0.44, + "learning_rate": 0.000283530591775326, + "loss": 1.3594, + "theoretical_loss": 3.527109844167623, + "tokens_seen": 1447821312 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002835205616850552, + "loss": 1.278, + "theoretical_loss": 3.5270956720441236, + "tokens_seen": 1447886848 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028351053159478436, + "loss": 1.4524, + "theoretical_loss": 3.527081500741689, + "tokens_seen": 1447952384 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028350050150451354, + "loss": 1.3995, + "theoretical_loss": 3.527067330260235, + "tokens_seen": 1448017920 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002834904714142427, + "loss": 1.3134, + "theoretical_loss": 3.5270531605996767, + "tokens_seen": 1448083456 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028348044132397196, + "loss": 1.4368, + "theoretical_loss": 3.5270389917599294, + "tokens_seen": 1448148992 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002834704112337011, + "loss": 1.3951, + "theoretical_loss": 3.5270248237409083, + "tokens_seen": 1448214528 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002834603811434303, + "loss": 1.3168, + "theoretical_loss": 3.5270106565425285, + "tokens_seen": 1448280064 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.00025031817494891584, + "objective/train/docs_used": 821297, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.073176860809326, + "objective/train/original_loss": 3.0731770992279053, + "objective/train/theoretical_loss": 3.5269964901647066, + "objective/train/tokens_used": 1468805600, + "objective/train/value_avg": -0.005779266357421875, + "objective/train/value_loss": 5.7712531997822225e-05, + "objective/train/value_max": -0.00034332275390625, + "objective/train/value_min": -0.1131591796875, + "objective/train/value_reward_corr": 0.41502830671007546, + "objective/train/value_std": 0.0054779052734375, + "objective/train/weight_avg": 1.0002609491348267, + "objective/train/weighted_lm_loss": 3.080570697784424, + "objective/train/weights_max": 2.659287691116333, + "objective/train/weights_min": 0.3115394413471222, + "theoretical_loss": 3.5269964901647066, + "tokens_seen": 1448345600 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028345035105315945, + "loss": 1.3824, + "theoretical_loss": 3.5269964901647066, + "tokens_seen": 1448345600 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002834403209628887, + "loss": 1.2509, + "theoretical_loss": 3.5269823246073564, + "tokens_seen": 1448411136 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028343029087261786, + "loss": 1.3559, + "theoretical_loss": 3.5269681598703935, + "tokens_seen": 1448476672 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028342026078234705, + "loss": 1.4, + "theoretical_loss": 3.5269539959537344, + "tokens_seen": 1448542208 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002834102306920762, + "loss": 1.2542, + "theoretical_loss": 3.526939832857294, + "tokens_seen": 1448607744 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028340020060180546, + "loss": 1.2994, + "theoretical_loss": 3.5269256705809866, + "tokens_seen": 1448673280 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002833901705115346, + "loss": 1.3435, + "theoretical_loss": 3.5269115091247287, + "tokens_seen": 1448738816 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002833801404212638, + "loss": 1.4177, + "theoretical_loss": 3.5268973484884354, + "tokens_seen": 1448804352 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028337011033099295, + "loss": 1.3091, + "theoretical_loss": 3.5268831886720227, + "tokens_seen": 1448869888 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002833600802407222, + "loss": 1.3096, + "theoretical_loss": 3.526869029675405, + "tokens_seen": 1448935424 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028335005015045137, + "loss": 1.3157, + "theoretical_loss": 3.5268548714984984, + "tokens_seen": 1449000960 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028334002006018055, + "loss": 1.3856, + "theoretical_loss": 3.5268407141412186, + "tokens_seen": 1449066496 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028332998996990973, + "loss": 1.377, + "theoretical_loss": 3.5268265576034805, + "tokens_seen": 1449132032 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002833199598796389, + "loss": 1.326, + "theoretical_loss": 3.5268124018852003, + "tokens_seen": 1449197568 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002833099297893681, + "loss": 1.2707, + "theoretical_loss": 3.526798246986293, + "tokens_seen": 1449263104 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002832998996990973, + "loss": 1.3131, + "theoretical_loss": 3.5267840929066736, + "tokens_seen": 1449328640 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028328986960882645, + "loss": 1.3933, + "theoretical_loss": 3.5267699396462593, + "tokens_seen": 1449394176 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002832798395185557, + "loss": 1.2633, + "theoretical_loss": 3.526755787204964, + "tokens_seen": 1449459712 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002832698094282848, + "loss": 1.2859, + "theoretical_loss": 3.5267416355827033, + "tokens_seen": 1449525248 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028325977933801405, + "loss": 1.3502, + "theoretical_loss": 3.5267274847793946, + "tokens_seen": 1449590784 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028324974924774323, + "loss": 1.4592, + "theoretical_loss": 3.526713334794951, + "tokens_seen": 1449656320 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002832397191574724, + "loss": 1.2944, + "theoretical_loss": 3.52669918562929, + "tokens_seen": 1449721856 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002832296890672016, + "loss": 1.4368, + "theoretical_loss": 3.526685037282326, + "tokens_seen": 1449787392 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028321965897693083, + "loss": 1.4277, + "theoretical_loss": 3.526670889753976, + "tokens_seen": 1449852928 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028320962888665996, + "loss": 1.3809, + "theoretical_loss": 3.526656743044154, + "tokens_seen": 1449918464 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": 0.00044940871885046363, + "objective/train/docs_used": 822517, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9187607765197754, + "objective/train/original_loss": 2.918760299682617, + "objective/train/theoretical_loss": 3.526642597152777, + "objective/train/tokens_used": 1470444000, + "objective/train/value_avg": -0.004817962646484375, + "objective/train/value_loss": 3.3014570362865925e-05, + "objective/train/value_max": -0.0002892017364501953, + "objective/train/value_min": -0.07696533203125, + "objective/train/value_reward_corr": 0.4018750120250697, + "objective/train/value_std": 0.004180908203125, + "objective/train/weight_avg": 1.006130337715149, + "objective/train/weighted_lm_loss": 2.9363863468170166, + "objective/train/weights_max": 1.9080041646957397, + "objective/train/weights_min": 0.7104275226593018, + "theoretical_loss": 3.526642597152777, + "tokens_seen": 1449984000 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002831995987963892, + "loss": 1.3292, + "theoretical_loss": 3.526642597152777, + "tokens_seen": 1449984000 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002831895687061183, + "loss": 1.3732, + "theoretical_loss": 3.52662845207976, + "tokens_seen": 1450049536 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028317953861584755, + "loss": 1.3868, + "theoretical_loss": 3.5266143078250183, + "tokens_seen": 1450115072 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028316950852557674, + "loss": 1.38, + "theoretical_loss": 3.5266001643884684, + "tokens_seen": 1450180608 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002831594784353059, + "loss": 1.4039, + "theoretical_loss": 3.526586021770026, + "tokens_seen": 1450246144 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002831494483450351, + "loss": 1.2636, + "theoretical_loss": 3.526571879969606, + "tokens_seen": 1450311680 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002831394182547643, + "loss": 1.392, + "theoretical_loss": 3.5265577389871248, + "tokens_seen": 1450377216 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028312938816449346, + "loss": 1.3675, + "theoretical_loss": 3.5265435988224976, + "tokens_seen": 1450442752 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002831193580742227, + "loss": 1.3138, + "theoretical_loss": 3.5265294594756407, + "tokens_seen": 1450508288 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002831093279839518, + "loss": 1.2991, + "theoretical_loss": 3.52651532094647, + "tokens_seen": 1450573824 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028309929789368106, + "loss": 1.3659, + "theoretical_loss": 3.5265011832349007, + "tokens_seen": 1450639360 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002830892678034102, + "loss": 1.3339, + "theoretical_loss": 3.526487046340849, + "tokens_seen": 1450704896 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002830792377131394, + "loss": 1.2857, + "theoretical_loss": 3.5264729102642294, + "tokens_seen": 1450770432 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002830692076228686, + "loss": 1.4023, + "theoretical_loss": 3.5264587750049596, + "tokens_seen": 1450835968 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002830591775325978, + "loss": 1.3473, + "theoretical_loss": 3.526444640562955, + "tokens_seen": 1450901504 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028304914744232696, + "loss": 1.4445, + "theoretical_loss": 3.5264305069381305, + "tokens_seen": 1450967040 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002830391173520562, + "loss": 1.4417, + "theoretical_loss": 3.526416374130403, + "tokens_seen": 1451032576 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002830290872617854, + "loss": 1.3538, + "theoretical_loss": 3.5264022421396874, + "tokens_seen": 1451098112 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028301905717151456, + "loss": 1.3769, + "theoretical_loss": 3.5263881109659003, + "tokens_seen": 1451163648 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028300902708124374, + "loss": 1.3464, + "theoretical_loss": 3.526373980608957, + "tokens_seen": 1451229184 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002829989969909729, + "loss": 1.4367, + "theoretical_loss": 3.526359851068774, + "tokens_seen": 1451294720 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028298896690070216, + "loss": 1.4009, + "theoretical_loss": 3.526345722345267, + "tokens_seen": 1451360256 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002829789368104313, + "loss": 1.4327, + "theoretical_loss": 3.5263315944383518, + "tokens_seen": 1451425792 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002829689067201605, + "loss": 1.4235, + "theoretical_loss": 3.5263174673479445, + "tokens_seen": 1451491328 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028295887662988965, + "loss": 1.3391, + "theoretical_loss": 3.5263033410739606, + "tokens_seen": 1451556864 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.0014240649761632085, + "objective/train/docs_used": 823297, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.365222692489624, + "objective/train/original_loss": 2.365222692489624, + "objective/train/theoretical_loss": 3.526289215616317, + "objective/train/tokens_used": 1472082400, + "objective/train/value_avg": -0.0052490234375, + "objective/train/value_loss": 5.70763586438261e-05, + "objective/train/value_max": -0.00015592575073242188, + "objective/train/value_min": -0.07080078125, + "objective/train/value_reward_corr": 0.6272126315024507, + "objective/train/value_std": 0.005924224853515625, + "objective/train/weight_avg": 0.988467276096344, + "objective/train/weighted_lm_loss": 2.3325326442718506, + "objective/train/weights_max": 1.501397728919983, + "objective/train/weights_min": 0.4278087913990021, + "theoretical_loss": 3.526289215616317, + "tokens_seen": 1451622400 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002829488465396189, + "loss": 1.3017, + "theoretical_loss": 3.526289215616317, + "tokens_seen": 1451622400 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028293881644934806, + "loss": 1.3953, + "theoretical_loss": 3.5262750909749285, + "tokens_seen": 1451687936 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028292878635907725, + "loss": 1.3357, + "theoretical_loss": 3.526260967149712, + "tokens_seen": 1451753472 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002829187562688064, + "loss": 1.3447, + "theoretical_loss": 3.5262468441405828, + "tokens_seen": 1451819008 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028290872617853566, + "loss": 1.2899, + "theoretical_loss": 3.5262327219474576, + "tokens_seen": 1451884544 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002828986960882648, + "loss": 1.3286, + "theoretical_loss": 3.5262186005702523, + "tokens_seen": 1451950080 + }, + { + "epoch": 0.44, + "learning_rate": 0.000282888665997994, + "loss": 1.3398, + "theoretical_loss": 3.5262044800088823, + "tokens_seen": 1452015616 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028287863590772315, + "loss": 1.2868, + "theoretical_loss": 3.5261903602632643, + "tokens_seen": 1452081152 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002828686058174524, + "loss": 1.3489, + "theoretical_loss": 3.526176241333314, + "tokens_seen": 1452146688 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028285857572718157, + "loss": 1.105, + "theoretical_loss": 3.526162123218948, + "tokens_seen": 1452212224 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028284854563691075, + "loss": 1.3162, + "theoretical_loss": 3.5261480059200814, + "tokens_seen": 1452277760 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028283851554663993, + "loss": 1.3794, + "theoretical_loss": 3.5261338894366316, + "tokens_seen": 1452343296 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002828284854563691, + "loss": 1.36, + "theoretical_loss": 3.5261197737685137, + "tokens_seen": 1452408832 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002828184553660983, + "loss": 1.3431, + "theoretical_loss": 3.526105658915644, + "tokens_seen": 1452474368 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002828084252758275, + "loss": 1.3625, + "theoretical_loss": 3.5260915448779384, + "tokens_seen": 1452539904 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028279839518555665, + "loss": 1.343, + "theoretical_loss": 3.526077431655314, + "tokens_seen": 1452605440 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002827883650952859, + "loss": 1.4498, + "theoretical_loss": 3.5260633192476862, + "tokens_seen": 1452670976 + }, + { + "epoch": 0.44, + "learning_rate": 0.000282778335005015, + "loss": 1.4091, + "theoretical_loss": 3.5260492076549714, + "tokens_seen": 1452736512 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028276830491474425, + "loss": 1.3624, + "theoretical_loss": 3.5260350968770857, + "tokens_seen": 1452802048 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028275827482447343, + "loss": 1.3029, + "theoretical_loss": 3.526020986913945, + "tokens_seen": 1452867584 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002827482447342026, + "loss": 1.4066, + "theoretical_loss": 3.5260068777654663, + "tokens_seen": 1452933120 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002827382146439318, + "loss": 1.2753, + "theoretical_loss": 3.525992769431565, + "tokens_seen": 1452998656 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028272818455366103, + "loss": 1.369, + "theoretical_loss": 3.5259786619121583, + "tokens_seen": 1453064192 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028271815446339016, + "loss": 1.3989, + "theoretical_loss": 3.5259645552071612, + "tokens_seen": 1453129728 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002827081243731194, + "loss": 1.3051, + "theoretical_loss": 3.5259504493164906, + "tokens_seen": 1453195264 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.0010362555040046573, + "objective/train/docs_used": 824661, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.856130361557007, + "objective/train/original_loss": 2.856130599975586, + "objective/train/theoretical_loss": 3.525936344240063, + "objective/train/tokens_used": 1473720800, + "objective/train/value_avg": -0.006011962890625, + "objective/train/value_loss": 9.333712660009041e-05, + "objective/train/value_max": -0.0002694129943847656, + "objective/train/value_min": -0.053192138671875, + "objective/train/value_reward_corr": 0.4180323609558514, + "objective/train/value_std": 0.0048675537109375, + "objective/train/weight_avg": 0.9939176440238953, + "objective/train/weighted_lm_loss": 2.8296079635620117, + "objective/train/weights_max": 1.4943825006484985, + "objective/train/weights_min": 0.3623541295528412, + "theoretical_loss": 3.525936344240063, + "tokens_seen": 1453260800 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002826980942828485, + "loss": 1.345, + "theoretical_loss": 3.525936344240063, + "tokens_seen": 1453260800 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028268806419257775, + "loss": 1.3462, + "theoretical_loss": 3.5259222399777945, + "tokens_seen": 1453326336 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028267803410230694, + "loss": 1.3937, + "theoretical_loss": 3.5259081365296012, + "tokens_seen": 1453391872 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002826680040120361, + "loss": 1.3425, + "theoretical_loss": 3.5258940338953995, + "tokens_seen": 1453457408 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002826579739217653, + "loss": 1.3158, + "theoretical_loss": 3.5258799320751057, + "tokens_seen": 1453522944 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002826479438314945, + "loss": 1.3354, + "theoretical_loss": 3.5258658310686366, + "tokens_seen": 1453588480 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028263791374122366, + "loss": 1.3738, + "theoretical_loss": 3.5258517308759076, + "tokens_seen": 1453654016 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002826278836509529, + "loss": 1.3022, + "theoretical_loss": 3.525837631496836, + "tokens_seen": 1453719552 + }, + { + "epoch": 0.44, + "learning_rate": 0.000282617853560682, + "loss": 1.3928, + "theoretical_loss": 3.525823532931337, + "tokens_seen": 1453785088 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028260782347041126, + "loss": 1.2739, + "theoretical_loss": 3.5258094351793288, + "tokens_seen": 1453850624 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002825977933801404, + "loss": 1.3795, + "theoretical_loss": 3.5257953382407266, + "tokens_seen": 1453916160 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002825877632898696, + "loss": 1.369, + "theoretical_loss": 3.5257812421154466, + "tokens_seen": 1453981696 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002825777331995988, + "loss": 1.3437, + "theoretical_loss": 3.5257671468034055, + "tokens_seen": 1454047232 + }, + { + "epoch": 0.44, + "learning_rate": 0.000282567703109328, + "loss": 1.3233, + "theoretical_loss": 3.5257530523045197, + "tokens_seen": 1454112768 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028255767301905716, + "loss": 1.4159, + "theoretical_loss": 3.5257389586187067, + "tokens_seen": 1454178304 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002825476429287864, + "loss": 1.3062, + "theoretical_loss": 3.525724865745881, + "tokens_seen": 1454243840 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002825376128385155, + "loss": 1.3078, + "theoretical_loss": 3.5257107736859603, + "tokens_seen": 1454309376 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028252758274824476, + "loss": 1.2866, + "theoretical_loss": 3.525696682438861, + "tokens_seen": 1454374912 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002825175526579739, + "loss": 1.3989, + "theoretical_loss": 3.5256825920044994, + "tokens_seen": 1454440448 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002825075225677031, + "loss": 1.3075, + "theoretical_loss": 3.525668502382792, + "tokens_seen": 1454505984 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002824974924774323, + "loss": 1.4527, + "theoretical_loss": 3.5256544135736556, + "tokens_seen": 1454571520 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002824874623871615, + "loss": 1.3673, + "theoretical_loss": 3.5256403255770064, + "tokens_seen": 1454637056 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028247743229689067, + "loss": 1.4331, + "theoretical_loss": 3.5256262383927606, + "tokens_seen": 1454702592 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028246740220661985, + "loss": 1.4535, + "theoretical_loss": 3.5256121520208357, + "tokens_seen": 1454768128 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028245737211634903, + "loss": 1.3965, + "theoretical_loss": 3.525598066461148, + "tokens_seen": 1454833664 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": 0.00015884294407442212, + "objective/train/docs_used": 824875, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3233911991119385, + "objective/train/original_loss": 2.3233911991119385, + "objective/train/theoretical_loss": 3.525583981713613, + "objective/train/tokens_used": 1475359200, + "objective/train/value_avg": -0.004764556884765625, + "objective/train/value_loss": 7.0102461904753e-05, + "objective/train/value_max": -0.0002415180206298828, + "objective/train/value_min": -0.07196044921875, + "objective/train/value_reward_corr": 0.19737011027362727, + "objective/train/value_std": 0.003993988037109375, + "objective/train/weight_avg": 1.0047516822814941, + "objective/train/weighted_lm_loss": 2.343076467514038, + "objective/train/weights_max": 1.8735994100570679, + "objective/train/weights_min": 0.3502885103225708, + "theoretical_loss": 3.525583981713613, + "tokens_seen": 1454899200 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028244734202607826, + "loss": 1.323, + "theoretical_loss": 3.525583981713613, + "tokens_seen": 1454899200 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002824373119358074, + "loss": 1.375, + "theoretical_loss": 3.5255698977781487, + "tokens_seen": 1454964736 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002824272818455366, + "loss": 1.4103, + "theoretical_loss": 3.5255558146546715, + "tokens_seen": 1455030272 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002824172517552658, + "loss": 1.4544, + "theoretical_loss": 3.525541732343097, + "tokens_seen": 1455095808 + }, + { + "epoch": 0.44, + "learning_rate": 0.000282407221664995, + "loss": 1.2723, + "theoretical_loss": 3.5255276508433426, + "tokens_seen": 1455161344 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028239719157472417, + "loss": 1.4042, + "theoretical_loss": 3.5255135701553253, + "tokens_seen": 1455226880 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028238716148445335, + "loss": 1.3919, + "theoretical_loss": 3.525499490278961, + "tokens_seen": 1455292416 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028237713139418253, + "loss": 1.4102, + "theoretical_loss": 3.5254854112141665, + "tokens_seen": 1455357952 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028236710130391177, + "loss": 1.3933, + "theoretical_loss": 3.5254713329608585, + "tokens_seen": 1455423488 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002823570712136409, + "loss": 1.4356, + "theoretical_loss": 3.5254572555189543, + "tokens_seen": 1455489024 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028234704112337013, + "loss": 1.2839, + "theoretical_loss": 3.52544317888837, + "tokens_seen": 1455554560 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028233701103309926, + "loss": 1.2876, + "theoretical_loss": 3.525429103069022, + "tokens_seen": 1455620096 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002823269809428285, + "loss": 1.2085, + "theoretical_loss": 3.525415028060828, + "tokens_seen": 1455685632 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002823169508525577, + "loss": 1.4278, + "theoretical_loss": 3.525400953863704, + "tokens_seen": 1455751168 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028230692076228685, + "loss": 1.3404, + "theoretical_loss": 3.5253868804775665, + "tokens_seen": 1455816704 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028229689067201604, + "loss": 1.3915, + "theoretical_loss": 3.5253728079023334, + "tokens_seen": 1455882240 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002822868605817452, + "loss": 1.4429, + "theoretical_loss": 3.5253587361379206, + "tokens_seen": 1455947776 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028227683049147445, + "loss": 1.3764, + "theoretical_loss": 3.525344665184245, + "tokens_seen": 1456013312 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028226680040120363, + "loss": 1.3632, + "theoretical_loss": 3.5253305950412237, + "tokens_seen": 1456078848 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002822567703109328, + "loss": 1.3684, + "theoretical_loss": 3.525316525708773, + "tokens_seen": 1456144384 + }, + { + "epoch": 0.44, + "learning_rate": 0.000282246740220662, + "loss": 1.4538, + "theoretical_loss": 3.52530245718681, + "tokens_seen": 1456209920 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028223671013039123, + "loss": 1.3006, + "theoretical_loss": 3.5252883894752522, + "tokens_seen": 1456275456 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028222668004012036, + "loss": 1.3532, + "theoretical_loss": 3.525274322574015, + "tokens_seen": 1456340992 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002822166499498496, + "loss": 1.4539, + "theoretical_loss": 3.5252602564830164, + "tokens_seen": 1456406528 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002822066198595787, + "loss": 1.3014, + "theoretical_loss": 3.5252461912021733, + "tokens_seen": 1456472064 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.000395592040149495, + "objective/train/docs_used": 826174, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.047271966934204, + "objective/train/original_loss": 3.047271728515625, + "objective/train/theoretical_loss": 3.525232126731402, + "objective/train/tokens_used": 1476997600, + "objective/train/value_avg": -0.005035400390625, + "objective/train/value_loss": 4.2744362872326747e-05, + "objective/train/value_max": -0.0002199411392211914, + "objective/train/value_min": -0.060089111328125, + "objective/train/value_reward_corr": 0.3697681239564147, + "objective/train/value_std": 0.0034694671630859375, + "objective/train/weight_avg": 0.9981060028076172, + "objective/train/weighted_lm_loss": 3.05100417137146, + "objective/train/weights_max": 1.6925973892211914, + "objective/train/weights_min": 0.5638715028762817, + "theoretical_loss": 3.525232126731402, + "tokens_seen": 1456537600 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028219658976930795, + "loss": 1.382, + "theoretical_loss": 3.525232126731402, + "tokens_seen": 1456537600 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028218655967903714, + "loss": 1.3376, + "theoretical_loss": 3.5252180630706196, + "tokens_seen": 1456603136 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002821765295887663, + "loss": 1.3558, + "theoretical_loss": 3.525204000219743, + "tokens_seen": 1456668672 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002821664994984955, + "loss": 1.4427, + "theoretical_loss": 3.5251899381786895, + "tokens_seen": 1456734208 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002821564694082247, + "loss": 1.3659, + "theoretical_loss": 3.525175876947375, + "tokens_seen": 1456799744 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028214643931795386, + "loss": 1.3781, + "theoretical_loss": 3.525161816525718, + "tokens_seen": 1456865280 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002821364092276831, + "loss": 1.3268, + "theoretical_loss": 3.5251477569136336, + "tokens_seen": 1456930816 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002821263791374122, + "loss": 1.4153, + "theoretical_loss": 3.525133698111041, + "tokens_seen": 1456996352 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028211634904714146, + "loss": 1.428, + "theoretical_loss": 3.525119640117855, + "tokens_seen": 1457061888 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002821063189568706, + "loss": 1.2979, + "theoretical_loss": 3.525105582933994, + "tokens_seen": 1457127424 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002820962888665998, + "loss": 1.331, + "theoretical_loss": 3.5250915265593745, + "tokens_seen": 1457192960 + }, + { + "epoch": 0.44, + "learning_rate": 0.000282086258776329, + "loss": 1.4248, + "theoretical_loss": 3.5250774709939137, + "tokens_seen": 1457258496 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002820762286860582, + "loss": 1.3872, + "theoretical_loss": 3.5250634162375283, + "tokens_seen": 1457324032 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028206619859578736, + "loss": 1.4237, + "theoretical_loss": 3.5250493622901358, + "tokens_seen": 1457389568 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002820561685055166, + "loss": 1.3902, + "theoretical_loss": 3.525035309151653, + "tokens_seen": 1457455104 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002820461384152457, + "loss": 1.467, + "theoretical_loss": 3.525021256821997, + "tokens_seen": 1457520640 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028203610832497496, + "loss": 1.395, + "theoretical_loss": 3.525007205301085, + "tokens_seen": 1457586176 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002820260782347041, + "loss": 1.3642, + "theoretical_loss": 3.524993154588834, + "tokens_seen": 1457651712 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002820160481444333, + "loss": 1.3115, + "theoretical_loss": 3.524979104685161, + "tokens_seen": 1457717248 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002820060180541625, + "loss": 1.3993, + "theoretical_loss": 3.524965055589983, + "tokens_seen": 1457782784 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002819959879638917, + "loss": 1.3673, + "theoretical_loss": 3.524951007303218, + "tokens_seen": 1457848320 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028198595787362087, + "loss": 1.4498, + "theoretical_loss": 3.524936959824782, + "tokens_seen": 1457913856 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028197592778335005, + "loss": 1.2508, + "theoretical_loss": 3.524922913154593, + "tokens_seen": 1457979392 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028196589769307923, + "loss": 1.2971, + "theoretical_loss": 3.5249088672925675, + "tokens_seen": 1458044928 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028195586760280846, + "loss": 1.3622, + "theoretical_loss": 3.524894822238623, + "tokens_seen": 1458110464 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.0001795592688722536, + "objective/train/docs_used": 826810, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.480132579803467, + "objective/train/original_loss": 2.4801323413848877, + "objective/train/theoretical_loss": 3.524880777992677, + "objective/train/tokens_used": 1478636000, + "objective/train/value_avg": -0.004180908203125, + "objective/train/value_loss": 3.753750934265554e-05, + "objective/train/value_max": -0.00043392181396484375, + "objective/train/value_min": -0.03448486328125, + "objective/train/value_reward_corr": 0.37769791943891484, + "objective/train/value_std": 0.003032684326171875, + "objective/train/weight_avg": 0.9999316334724426, + "objective/train/weighted_lm_loss": 2.4836313724517822, + "objective/train/weights_max": 1.3512848615646362, + "objective/train/weights_min": 0.2994903028011322, + "theoretical_loss": 3.524880777992677, + "tokens_seen": 1458176000 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002819458375125376, + "loss": 1.3756, + "theoretical_loss": 3.524880777992677, + "tokens_seen": 1458176000 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002819358074222668, + "loss": 1.3563, + "theoretical_loss": 3.524866734554646, + "tokens_seen": 1458241536 + }, + { + "epoch": 0.44, + "learning_rate": 0.000281925777331996, + "loss": 1.3381, + "theoretical_loss": 3.524852691924448, + "tokens_seen": 1458307072 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002819157472417252, + "loss": 1.3464, + "theoretical_loss": 3.5248386501019997, + "tokens_seen": 1458372608 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028190571715145437, + "loss": 1.3826, + "theoretical_loss": 3.5248246090872186, + "tokens_seen": 1458438144 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028189568706118355, + "loss": 1.3141, + "theoretical_loss": 3.5248105688800218, + "tokens_seen": 1458503680 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028188565697091273, + "loss": 1.4655, + "theoretical_loss": 3.5247965294803265, + "tokens_seen": 1458569216 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028187562688064197, + "loss": 1.3179, + "theoretical_loss": 3.52478249088805, + "tokens_seen": 1458634752 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002818655967903711, + "loss": 1.3895, + "theoretical_loss": 3.5247684531031096, + "tokens_seen": 1458700288 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028185556670010033, + "loss": 1.3328, + "theoretical_loss": 3.524754416125423, + "tokens_seen": 1458765824 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028184553660982946, + "loss": 1.3159, + "theoretical_loss": 3.524740379954907, + "tokens_seen": 1458831360 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002818355065195587, + "loss": 1.3016, + "theoretical_loss": 3.524726344591479, + "tokens_seen": 1458896896 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002818254764292879, + "loss": 1.3066, + "theoretical_loss": 3.5247123100350572, + "tokens_seen": 1458962432 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028181544633901705, + "loss": 1.3755, + "theoretical_loss": 3.524698276285558, + "tokens_seen": 1459027968 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028180541624874624, + "loss": 1.378, + "theoretical_loss": 3.5246842433428984, + "tokens_seen": 1459093504 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002817953861584754, + "loss": 1.3669, + "theoretical_loss": 3.5246702112069963, + "tokens_seen": 1459159040 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002817853560682046, + "loss": 1.336, + "theoretical_loss": 3.52465617987777, + "tokens_seen": 1459224576 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028177532597793383, + "loss": 1.3489, + "theoretical_loss": 3.5246421493551354, + "tokens_seen": 1459290112 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028176529588766296, + "loss": 1.2824, + "theoretical_loss": 3.5246281196390106, + "tokens_seen": 1459355648 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002817552657973922, + "loss": 1.4525, + "theoretical_loss": 3.5246140907293135, + "tokens_seen": 1459421184 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002817452357071214, + "loss": 1.3333, + "theoretical_loss": 3.5246000626259604, + "tokens_seen": 1459486720 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028173520561685056, + "loss": 1.4456, + "theoretical_loss": 3.52458603532887, + "tokens_seen": 1459552256 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028172517552657974, + "loss": 1.3231, + "theoretical_loss": 3.524572008837958, + "tokens_seen": 1459617792 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002817151454363089, + "loss": 1.3829, + "theoretical_loss": 3.524557983153144, + "tokens_seen": 1459683328 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002817051153460381, + "loss": 1.4836, + "theoretical_loss": 3.5245439582743443, + "tokens_seen": 1459748864 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.0004995877388864756, + "objective/train/docs_used": 827996, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.800590753555298, + "objective/train/original_loss": 2.800590753555298, + "objective/train/theoretical_loss": 3.5245299342014764, + "objective/train/tokens_used": 1480274400, + "objective/train/value_avg": -0.005157470703125, + "objective/train/value_loss": 4.968069333699532e-05, + "objective/train/value_max": -0.00040602684020996094, + "objective/train/value_min": -0.06451416015625, + "objective/train/value_reward_corr": 0.4942182695923293, + "objective/train/value_std": 0.00440216064453125, + "objective/train/weight_avg": 0.9972912669181824, + "objective/train/weighted_lm_loss": 2.798768997192383, + "objective/train/weights_max": 1.7564753293991089, + "objective/train/weights_min": 0.3217330276966095, + "theoretical_loss": 3.5245299342014764, + "tokens_seen": 1459814400 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028169508525576734, + "loss": 1.3975, + "theoretical_loss": 3.5245299342014764, + "tokens_seen": 1459814400 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028168505516549646, + "loss": 1.3585, + "theoretical_loss": 3.524515910934458, + "tokens_seen": 1459879936 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002816750250752257, + "loss": 1.4088, + "theoretical_loss": 3.524501888473207, + "tokens_seen": 1459945472 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002816649949849548, + "loss": 1.4863, + "theoretical_loss": 3.52448786681764, + "tokens_seen": 1460011008 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028165496489468406, + "loss": 1.4333, + "theoretical_loss": 3.5244738459676754, + "tokens_seen": 1460076544 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028164493480441324, + "loss": 1.3964, + "theoretical_loss": 3.5244598259232305, + "tokens_seen": 1460142080 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002816349047141424, + "loss": 1.503, + "theoretical_loss": 3.5244458066842226, + "tokens_seen": 1460207616 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002816248746238716, + "loss": 1.3523, + "theoretical_loss": 3.5244317882505696, + "tokens_seen": 1460273152 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002816148445336008, + "loss": 1.3358, + "theoretical_loss": 3.524417770622189, + "tokens_seen": 1460338688 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028160481444332997, + "loss": 1.3355, + "theoretical_loss": 3.5244037537989987, + "tokens_seen": 1460404224 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002815947843530592, + "loss": 1.3737, + "theoretical_loss": 3.5243897377809157, + "tokens_seen": 1460469760 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028158475426278833, + "loss": 1.3462, + "theoretical_loss": 3.524375722567858, + "tokens_seen": 1460535296 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028157472417251756, + "loss": 1.3945, + "theoretical_loss": 3.524361708159743, + "tokens_seen": 1460600832 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028156469408224674, + "loss": 1.3575, + "theoretical_loss": 3.5243476945564893, + "tokens_seen": 1460666368 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002815546639919759, + "loss": 1.3936, + "theoretical_loss": 3.524333681758013, + "tokens_seen": 1460731904 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002815446339017051, + "loss": 1.3671, + "theoretical_loss": 3.524319669764233, + "tokens_seen": 1460797440 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002815346038114343, + "loss": 1.3705, + "theoretical_loss": 3.524305658575067, + "tokens_seen": 1460862976 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002815245737211635, + "loss": 1.3767, + "theoretical_loss": 3.524291648190432, + "tokens_seen": 1460928512 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002815145436308927, + "loss": 1.3575, + "theoretical_loss": 3.5242776386102452, + "tokens_seen": 1460994048 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002815045135406219, + "loss": 1.416, + "theoretical_loss": 3.524263629834426, + "tokens_seen": 1461059584 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028149448345035107, + "loss": 1.268, + "theoretical_loss": 3.524249621862891, + "tokens_seen": 1461125120 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028148445336008025, + "loss": 1.3934, + "theoretical_loss": 3.5242356146955585, + "tokens_seen": 1461190656 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028147442326980943, + "loss": 1.3117, + "theoretical_loss": 3.524221608332346, + "tokens_seen": 1461256192 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028146439317953866, + "loss": 1.3809, + "theoretical_loss": 3.524207602773171, + "tokens_seen": 1461321728 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002814543630892678, + "loss": 1.288, + "theoretical_loss": 3.524193598017952, + "tokens_seen": 1461387264 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": 0.0003962115151807666, + "objective/train/docs_used": 828576, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.2701945304870605, + "objective/train/original_loss": 3.2701942920684814, + "objective/train/theoretical_loss": 3.524179594066606, + "objective/train/tokens_used": 1481912800, + "objective/train/value_avg": -0.004241943359375, + "objective/train/value_loss": 4.163167977822013e-05, + "objective/train/value_max": -0.0004029273986816406, + "objective/train/value_min": -0.04168701171875, + "objective/train/value_reward_corr": 0.2651410568460188, + "objective/train/value_std": 0.00301361083984375, + "objective/train/weight_avg": 1.0058506727218628, + "objective/train/weighted_lm_loss": 3.2859413623809814, + "objective/train/weights_max": 1.408333659172058, + "objective/train/weights_min": 0.24722348153591156, + "theoretical_loss": 3.524179594066606, + "tokens_seen": 1461452800 + }, + { + "epoch": 0.44, + "learning_rate": 0.000281444332998997, + "loss": 1.4295, + "theoretical_loss": 3.524179594066606, + "tokens_seen": 1461452800 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002814343029087262, + "loss": 1.4586, + "theoretical_loss": 3.524165590919051, + "tokens_seen": 1461518336 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002814242728184554, + "loss": 1.4919, + "theoretical_loss": 3.5241515885752053, + "tokens_seen": 1461583872 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028141424272818457, + "loss": 1.3428, + "theoretical_loss": 3.5241375870349865, + "tokens_seen": 1461649408 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028140421263791375, + "loss": 1.4028, + "theoretical_loss": 3.524123586298312, + "tokens_seen": 1461714944 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028139418254764293, + "loss": 1.3827, + "theoretical_loss": 3.5241095863651006, + "tokens_seen": 1461780480 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028138415245737217, + "loss": 1.3744, + "theoretical_loss": 3.5240955872352693, + "tokens_seen": 1461846016 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002813741223671013, + "loss": 1.5167, + "theoretical_loss": 3.5240815889087362, + "tokens_seen": 1461911552 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028136409227683053, + "loss": 1.4101, + "theoretical_loss": 3.5240675913854194, + "tokens_seen": 1461977088 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028135406218655966, + "loss": 1.3509, + "theoretical_loss": 3.524053594665237, + "tokens_seen": 1462042624 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002813440320962889, + "loss": 1.3291, + "theoretical_loss": 3.5240395987481064, + "tokens_seen": 1462108160 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002813340020060181, + "loss": 1.4027, + "theoretical_loss": 3.5240256036339455, + "tokens_seen": 1462173696 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028132397191574725, + "loss": 1.3857, + "theoretical_loss": 3.524011609322673, + "tokens_seen": 1462239232 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028131394182547644, + "loss": 1.3505, + "theoretical_loss": 3.5239976158142063, + "tokens_seen": 1462304768 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002813039117352056, + "loss": 1.3167, + "theoretical_loss": 3.523983623108463, + "tokens_seen": 1462370304 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002812938816449348, + "loss": 1.3881, + "theoretical_loss": 3.523969631205362, + "tokens_seen": 1462435840 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028128385155466403, + "loss": 1.5034, + "theoretical_loss": 3.5239556401048207, + "tokens_seen": 1462501376 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028127382146439316, + "loss": 1.409, + "theoretical_loss": 3.523941649806757, + "tokens_seen": 1462566912 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002812637913741224, + "loss": 1.3398, + "theoretical_loss": 3.5239276603110894, + "tokens_seen": 1462632448 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002812537612838516, + "loss": 1.2768, + "theoretical_loss": 3.5239136716177355, + "tokens_seen": 1462697984 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028124373119358076, + "loss": 1.375, + "theoretical_loss": 3.5238996837266137, + "tokens_seen": 1462763520 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028123370110330994, + "loss": 1.4067, + "theoretical_loss": 3.5238856966376417, + "tokens_seen": 1462829056 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002812236710130391, + "loss": 1.2671, + "theoretical_loss": 3.523871710350738, + "tokens_seen": 1462894592 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002812136409227683, + "loss": 1.3631, + "theoretical_loss": 3.5238577248658203, + "tokens_seen": 1462960128 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028120361083249754, + "loss": 1.412, + "theoretical_loss": 3.523843740182807, + "tokens_seen": 1463025664 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -8.877593063516542e-05, + "objective/train/docs_used": 830024, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9041588306427, + "objective/train/original_loss": 2.904158115386963, + "objective/train/theoretical_loss": 3.523829756301615, + "objective/train/tokens_used": 1483551200, + "objective/train/value_avg": -0.004642486572265625, + "objective/train/value_loss": 4.4208067265572026e-05, + "objective/train/value_max": -0.00034880638122558594, + "objective/train/value_min": -0.038909912109375, + "objective/train/value_reward_corr": 0.34115320396335086, + "objective/train/value_std": 0.0032520294189453125, + "objective/train/weight_avg": 1.0012176036834717, + "objective/train/weighted_lm_loss": 2.9172537326812744, + "objective/train/weights_max": 1.3583134412765503, + "objective/train/weights_min": 0.455354243516922, + "theoretical_loss": 3.523829756301615, + "tokens_seen": 1463091200 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028119358074222666, + "loss": 1.3689, + "theoretical_loss": 3.523829756301615, + "tokens_seen": 1463091200 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002811835506519559, + "loss": 1.4089, + "theoretical_loss": 3.5238157732221644, + "tokens_seen": 1463156736 + }, + { + "epoch": 0.44, + "learning_rate": 0.000281173520561685, + "loss": 1.4057, + "theoretical_loss": 3.523801790944372, + "tokens_seen": 1463222272 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028116349047141426, + "loss": 1.5302, + "theoretical_loss": 3.523787809468156, + "tokens_seen": 1463287808 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028115346038114344, + "loss": 1.3768, + "theoretical_loss": 3.5237738287934355, + "tokens_seen": 1463353344 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002811434302908726, + "loss": 1.4033, + "theoretical_loss": 3.523759848920128, + "tokens_seen": 1463418880 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002811334002006018, + "loss": 1.3558, + "theoretical_loss": 3.523745869848151, + "tokens_seen": 1463484416 + }, + { + "epoch": 0.44, + "learning_rate": 0.000281123370110331, + "loss": 1.2816, + "theoretical_loss": 3.523731891577424, + "tokens_seen": 1463549952 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028111334002006017, + "loss": 1.2838, + "theoretical_loss": 3.523717914107864, + "tokens_seen": 1463615488 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002811033099297894, + "loss": 1.3866, + "theoretical_loss": 3.5237039374393904, + "tokens_seen": 1463681024 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028109327983951853, + "loss": 1.3617, + "theoretical_loss": 3.52368996157192, + "tokens_seen": 1463746560 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028108324974924776, + "loss": 1.3807, + "theoretical_loss": 3.5236759865053724, + "tokens_seen": 1463812096 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028107321965897695, + "loss": 1.2734, + "theoretical_loss": 3.523662012239665, + "tokens_seen": 1463877632 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002810631895687061, + "loss": 1.2909, + "theoretical_loss": 3.5236480387747164, + "tokens_seen": 1463943168 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002810531594784353, + "loss": 1.3278, + "theoretical_loss": 3.5236340661104446, + "tokens_seen": 1464008704 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002810431293881645, + "loss": 1.3295, + "theoretical_loss": 3.523620094246769, + "tokens_seen": 1464074240 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028103309929789367, + "loss": 1.2228, + "theoretical_loss": 3.5236061231836056, + "tokens_seen": 1464139776 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002810230692076229, + "loss": 1.2376, + "theoretical_loss": 3.523592152920875, + "tokens_seen": 1464205312 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028101303911735203, + "loss": 1.3954, + "theoretical_loss": 3.5235781834584943, + "tokens_seen": 1464270848 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028100300902708127, + "loss": 1.4337, + "theoretical_loss": 3.523564214796382, + "tokens_seen": 1464336384 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002809929789368104, + "loss": 1.4043, + "theoretical_loss": 3.523550246934457, + "tokens_seen": 1464401920 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028098294884653963, + "loss": 1.27, + "theoretical_loss": 3.523536279872637, + "tokens_seen": 1464467456 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002809729187562688, + "loss": 1.4075, + "theoretical_loss": 3.5235223136108402, + "tokens_seen": 1464532992 + }, + { + "epoch": 0.44, + "learning_rate": 0.000280962888665998, + "loss": 1.3837, + "theoretical_loss": 3.5235083481489857, + "tokens_seen": 1464598528 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028095285857572717, + "loss": 1.2876, + "theoretical_loss": 3.523494383486992, + "tokens_seen": 1464664064 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.0005117039545439184, + "objective/train/docs_used": 830651, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.3287546634674072, + "objective/train/original_loss": 3.328754186630249, + "objective/train/theoretical_loss": 3.5234804196247764, + "objective/train/tokens_used": 1485189600, + "objective/train/value_avg": -0.00543975830078125, + "objective/train/value_loss": 7.042023935355246e-05, + "objective/train/value_max": -0.00033020973205566406, + "objective/train/value_min": -0.1278076171875, + "objective/train/value_reward_corr": 0.6106893742157299, + "objective/train/value_std": 0.00595855712890625, + "objective/train/weight_avg": 0.9982035756111145, + "objective/train/weighted_lm_loss": 3.3234286308288574, + "objective/train/weights_max": 2.5520315170288086, + "objective/train/weights_min": 0.42529797554016113, + "theoretical_loss": 3.5234804196247764, + "tokens_seen": 1464729600 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028094282848545635, + "loss": 1.4328, + "theoretical_loss": 3.5234804196247764, + "tokens_seen": 1464729600 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028093279839518553, + "loss": 1.4477, + "theoretical_loss": 3.523466456562258, + "tokens_seen": 1464795136 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028092276830491477, + "loss": 1.3287, + "theoretical_loss": 3.523452494299356, + "tokens_seen": 1464860672 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002809127382146439, + "loss": 1.2954, + "theoretical_loss": 3.523438532835987, + "tokens_seen": 1464926208 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028090270812437313, + "loss": 1.2837, + "theoretical_loss": 3.523424572172071, + "tokens_seen": 1464991744 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002808926780341023, + "loss": 1.2716, + "theoretical_loss": 3.5234106123075257, + "tokens_seen": 1465057280 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002808826479438315, + "loss": 1.3907, + "theoretical_loss": 3.52339665324227, + "tokens_seen": 1465122816 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002808726178535607, + "loss": 1.3334, + "theoretical_loss": 3.523382694976222, + "tokens_seen": 1465188352 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028086258776328986, + "loss": 1.3396, + "theoretical_loss": 3.5233687375093012, + "tokens_seen": 1465253888 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028085255767301904, + "loss": 1.35, + "theoretical_loss": 3.5233547808414247, + "tokens_seen": 1465319424 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002808425275827483, + "loss": 1.3946, + "theoretical_loss": 3.5233408249725113, + "tokens_seen": 1465384960 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002808324974924774, + "loss": 1.4246, + "theoretical_loss": 3.5233268699024807, + "tokens_seen": 1465450496 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028082246740220664, + "loss": 1.3935, + "theoretical_loss": 3.5233129156312497, + "tokens_seen": 1465516032 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028081243731193576, + "loss": 1.3106, + "theoretical_loss": 3.523298962158738, + "tokens_seen": 1465581568 + }, + { + "epoch": 0.44, + "learning_rate": 0.000280802407221665, + "loss": 1.2906, + "theoretical_loss": 3.523285009484865, + "tokens_seen": 1465647104 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002807923771313942, + "loss": 1.2936, + "theoretical_loss": 3.523271057609547, + "tokens_seen": 1465712640 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028078234704112336, + "loss": 1.3318, + "theoretical_loss": 3.5232571065327045, + "tokens_seen": 1465778176 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002807723169508526, + "loss": 1.3483, + "theoretical_loss": 3.5232431562542548, + "tokens_seen": 1465843712 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002807622868605818, + "loss": 1.2091, + "theoretical_loss": 3.5232292067741176, + "tokens_seen": 1465909248 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028075225677031096, + "loss": 1.4174, + "theoretical_loss": 3.523215258092211, + "tokens_seen": 1465974784 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028074222668004014, + "loss": 1.3341, + "theoretical_loss": 3.523201310208454, + "tokens_seen": 1466040320 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002807321965897693, + "loss": 1.3631, + "theoretical_loss": 3.5231873631227644, + "tokens_seen": 1466105856 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002807221664994985, + "loss": 1.3099, + "theoretical_loss": 3.5231734168350615, + "tokens_seen": 1466171392 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028071213640922774, + "loss": 1.3345, + "theoretical_loss": 3.5231594713452647, + "tokens_seen": 1466236928 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028070210631895686, + "loss": 1.3751, + "theoretical_loss": 3.523145526653291, + "tokens_seen": 1466302464 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": 0.000357199547579512, + "objective/train/docs_used": 831840, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8030154705047607, + "objective/train/original_loss": 2.8030154705047607, + "objective/train/theoretical_loss": 3.5231315827590604, + "objective/train/tokens_used": 1486828000, + "objective/train/value_avg": -0.00548553466796875, + "objective/train/value_loss": 6.269430741667747e-05, + "objective/train/value_max": -0.0004494190216064453, + "objective/train/value_min": -0.06396484375, + "objective/train/value_reward_corr": 0.3378613107715117, + "objective/train/value_std": 0.0037708282470703125, + "objective/train/weight_avg": 1.006455898284912, + "objective/train/weighted_lm_loss": 2.8149805068969727, + "objective/train/weights_max": 1.8958141803741455, + "objective/train/weights_min": 0.29626888036727905, + "theoretical_loss": 3.5231315827590604, + "tokens_seen": 1466368000 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002806920762286861, + "loss": 1.4446, + "theoretical_loss": 3.5231315827590604, + "tokens_seen": 1466368000 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002806820461384152, + "loss": 1.3994, + "theoretical_loss": 3.523117639662491, + "tokens_seen": 1466433536 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028067201604814446, + "loss": 1.3976, + "theoretical_loss": 3.523103697363502, + "tokens_seen": 1466499072 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028066198595787364, + "loss": 1.3521, + "theoretical_loss": 3.523089755862012, + "tokens_seen": 1466564608 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002806519558676028, + "loss": 1.3485, + "theoretical_loss": 3.523075815157939, + "tokens_seen": 1466630144 + }, + { + "epoch": 0.44, + "learning_rate": 0.000280641925777332, + "loss": 1.2938, + "theoretical_loss": 3.5230618752512024, + "tokens_seen": 1466695680 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002806318956870612, + "loss": 1.3588, + "theoretical_loss": 3.5230479361417215, + "tokens_seen": 1466761216 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028062186559679037, + "loss": 1.294, + "theoretical_loss": 3.5230339978294145, + "tokens_seen": 1466826752 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002806118355065196, + "loss": 1.244, + "theoretical_loss": 3.5230200603142, + "tokens_seen": 1466892288 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028060180541624873, + "loss": 1.3534, + "theoretical_loss": 3.523006123595997, + "tokens_seen": 1466957824 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028059177532597796, + "loss": 1.3093, + "theoretical_loss": 3.522992187674725, + "tokens_seen": 1467023360 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028058174523570715, + "loss": 1.4166, + "theoretical_loss": 3.5229782525503013, + "tokens_seen": 1467088896 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002805717151454363, + "loss": 1.3659, + "theoretical_loss": 3.5229643182226464, + "tokens_seen": 1467154432 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002805616850551655, + "loss": 1.2476, + "theoretical_loss": 3.5229503846916774, + "tokens_seen": 1467219968 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002805516549648947, + "loss": 1.3225, + "theoretical_loss": 3.5229364519573148, + "tokens_seen": 1467285504 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028054162487462387, + "loss": 1.2992, + "theoretical_loss": 3.522922520019477, + "tokens_seen": 1467351040 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002805315947843531, + "loss": 1.3444, + "theoretical_loss": 3.5229085888780824, + "tokens_seen": 1467416576 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028052156469408223, + "loss": 1.452, + "theoretical_loss": 3.5228946585330507, + "tokens_seen": 1467482112 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028051153460381147, + "loss": 1.1726, + "theoretical_loss": 3.5228807289843003, + "tokens_seen": 1467547648 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002805015045135406, + "loss": 1.478, + "theoretical_loss": 3.52286680023175, + "tokens_seen": 1467613184 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028049147442326983, + "loss": 1.3134, + "theoretical_loss": 3.522852872275318, + "tokens_seen": 1467678720 + }, + { + "epoch": 0.44, + "learning_rate": 0.000280481444332999, + "loss": 1.4264, + "theoretical_loss": 3.5228389451149256, + "tokens_seen": 1467744256 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002804714142427282, + "loss": 1.381, + "theoretical_loss": 3.5228250187504897, + "tokens_seen": 1467809792 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028046138415245737, + "loss": 1.3385, + "theoretical_loss": 3.5228110931819296, + "tokens_seen": 1467875328 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028045135406218655, + "loss": 1.4016, + "theoretical_loss": 3.5227971684091655, + "tokens_seen": 1467940864 + }, + { + "epoch": 0.44, + "objective/train/advantage_avg": -0.0005592018715105951, + "objective/train/docs_used": 832610, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6082191467285156, + "objective/train/original_loss": 2.6082193851470947, + "objective/train/theoretical_loss": 3.522783244432115, + "objective/train/tokens_used": 1488466400, + "objective/train/value_avg": -0.0044708251953125, + "objective/train/value_loss": 5.482080450747162e-05, + "objective/train/value_max": -0.0003275871276855469, + "objective/train/value_min": -0.06732177734375, + "objective/train/value_reward_corr": 0.37388034378497426, + "objective/train/value_std": 0.003749847412109375, + "objective/train/weight_avg": 0.9967919588088989, + "objective/train/weighted_lm_loss": 2.602451801300049, + "objective/train/weights_max": 1.5646370649337769, + "objective/train/weights_min": 0.1415717750787735, + "theoretical_loss": 3.522783244432115, + "tokens_seen": 1468006400 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028044132397191574, + "loss": 1.4205, + "theoretical_loss": 3.522783244432115, + "tokens_seen": 1468006400 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028043129388164497, + "loss": 1.3721, + "theoretical_loss": 3.5227693212506974, + "tokens_seen": 1468071936 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002804212637913741, + "loss": 1.2906, + "theoretical_loss": 3.522755398864832, + "tokens_seen": 1468137472 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028041123370110333, + "loss": 1.2842, + "theoretical_loss": 3.5227414772744376, + "tokens_seen": 1468203008 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002804012036108325, + "loss": 1.3481, + "theoretical_loss": 3.522727556479434, + "tokens_seen": 1468268544 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002803911735205617, + "loss": 1.2459, + "theoretical_loss": 3.5227136364797387, + "tokens_seen": 1468334080 + }, + { + "epoch": 0.44, + "learning_rate": 0.0002803811434302909, + "loss": 1.3181, + "theoretical_loss": 3.5226997172752723, + "tokens_seen": 1468399616 + }, + { + "epoch": 0.44, + "learning_rate": 0.00028037111334002006, + "loss": 1.4156, + "theoretical_loss": 3.5226857988659535, + "tokens_seen": 1468465152 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028036108324974924, + "loss": 1.3633, + "theoretical_loss": 3.522671881251701, + "tokens_seen": 1468530688 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002803510531594785, + "loss": 1.4589, + "theoretical_loss": 3.522657964432434, + "tokens_seen": 1468596224 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002803410230692076, + "loss": 1.3545, + "theoretical_loss": 3.522644048408072, + "tokens_seen": 1468661760 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028033099297893684, + "loss": 1.3398, + "theoretical_loss": 3.5226301331785335, + "tokens_seen": 1468727296 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028032096288866596, + "loss": 1.3777, + "theoretical_loss": 3.522616218743738, + "tokens_seen": 1468792832 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002803109327983952, + "loss": 1.2919, + "theoretical_loss": 3.522602305103605, + "tokens_seen": 1468858368 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002803009027081244, + "loss": 1.4096, + "theoretical_loss": 3.5225883922580534, + "tokens_seen": 1468923904 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028029087261785356, + "loss": 1.3193, + "theoretical_loss": 3.522574480207002, + "tokens_seen": 1468989440 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028028084252758274, + "loss": 1.2964, + "theoretical_loss": 3.52256056895037, + "tokens_seen": 1469054976 + }, + { + "epoch": 0.45, + "learning_rate": 0.000280270812437312, + "loss": 1.3715, + "theoretical_loss": 3.522546658488077, + "tokens_seen": 1469120512 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002802607823470411, + "loss": 1.324, + "theoretical_loss": 3.5225327488200424, + "tokens_seen": 1469186048 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028025075225677034, + "loss": 1.2986, + "theoretical_loss": 3.522518839946185, + "tokens_seen": 1469251584 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028024072216649947, + "loss": 1.2724, + "theoretical_loss": 3.5225049318664237, + "tokens_seen": 1469317120 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002802306920762287, + "loss": 1.3801, + "theoretical_loss": 3.522491024580679, + "tokens_seen": 1469382656 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002802206619859579, + "loss": 1.3501, + "theoretical_loss": 3.5224771180888688, + "tokens_seen": 1469448192 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028021063189568706, + "loss": 1.3479, + "theoretical_loss": 3.522463212390913, + "tokens_seen": 1469513728 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028020060180541624, + "loss": 1.3344, + "theoretical_loss": 3.5224493074867307, + "tokens_seen": 1469579264 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.0003296186914667487, + "objective/train/docs_used": 834187, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.370718479156494, + "objective/train/original_loss": 2.370718479156494, + "objective/train/theoretical_loss": 3.5224354033762415, + "objective/train/tokens_used": 1490104800, + "objective/train/value_avg": -0.005825042724609375, + "objective/train/value_loss": 5.791274816147052e-05, + "objective/train/value_max": -0.0004093647003173828, + "objective/train/value_min": -0.07452392578125, + "objective/train/value_reward_corr": 0.421342606332736, + "objective/train/value_std": 0.005756378173828125, + "objective/train/weight_avg": 0.9994285106658936, + "objective/train/weighted_lm_loss": 2.3833460807800293, + "objective/train/weights_max": 1.6909204721450806, + "objective/train/weights_min": 0.3004629909992218, + "theoretical_loss": 3.5224354033762415, + "tokens_seen": 1469644800 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002801905717151454, + "loss": 1.3815, + "theoretical_loss": 3.5224354033762415, + "tokens_seen": 1469644800 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002801805416248746, + "loss": 1.3805, + "theoretical_loss": 3.522421500059364, + "tokens_seen": 1469710336 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028017051153460384, + "loss": 1.3688, + "theoretical_loss": 3.5224075975360187, + "tokens_seen": 1469775872 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028016048144433297, + "loss": 1.4043, + "theoretical_loss": 3.522393695806124, + "tokens_seen": 1469841408 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002801504513540622, + "loss": 1.3578, + "theoretical_loss": 3.5223797948695994, + "tokens_seen": 1469906944 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028014042126379133, + "loss": 1.3535, + "theoretical_loss": 3.5223658947263643, + "tokens_seen": 1469972480 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028013039117352057, + "loss": 1.3399, + "theoretical_loss": 3.5223519953763383, + "tokens_seen": 1470038016 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028012036108324975, + "loss": 1.2375, + "theoretical_loss": 3.5223380968194404, + "tokens_seen": 1470103552 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028011033099297893, + "loss": 1.275, + "theoretical_loss": 3.52232419905559, + "tokens_seen": 1470169088 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002801003009027081, + "loss": 1.2776, + "theoretical_loss": 3.5223103020847075, + "tokens_seen": 1470234624 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028009027081243735, + "loss": 1.4309, + "theoretical_loss": 3.522296405906711, + "tokens_seen": 1470300160 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028008024072216647, + "loss": 1.3107, + "theoretical_loss": 3.52228251052152, + "tokens_seen": 1470365696 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002800702106318957, + "loss": 1.3853, + "theoretical_loss": 3.5222686159290544, + "tokens_seen": 1470431232 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028006018054162483, + "loss": 1.4341, + "theoretical_loss": 3.522254722129234, + "tokens_seen": 1470496768 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028005015045135407, + "loss": 1.4368, + "theoretical_loss": 3.5222408291219773, + "tokens_seen": 1470562304 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028004012036108325, + "loss": 1.4624, + "theoretical_loss": 3.522226936907205, + "tokens_seen": 1470627840 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028003009027081243, + "loss": 1.3539, + "theoretical_loss": 3.5222130454848353, + "tokens_seen": 1470693376 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028002006018054167, + "loss": 1.3697, + "theoretical_loss": 3.5221991548547886, + "tokens_seen": 1470758912 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002800100300902708, + "loss": 1.3296, + "theoretical_loss": 3.5221852650169843, + "tokens_seen": 1470824448 + }, + { + "epoch": 0.45, + "learning_rate": 0.00028000000000000003, + "loss": 1.2698, + "theoretical_loss": 3.5221713759713413, + "tokens_seen": 1470889984 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002799899699097292, + "loss": 1.4093, + "theoretical_loss": 3.52215748771778, + "tokens_seen": 1470955520 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002799799398194584, + "loss": 1.3424, + "theoretical_loss": 3.5221436002562188, + "tokens_seen": 1471021056 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002799699097291876, + "loss": 1.3831, + "theoretical_loss": 3.522129713586579, + "tokens_seen": 1471086592 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027995987963891675, + "loss": 1.3048, + "theoretical_loss": 3.522115827708778, + "tokens_seen": 1471152128 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027994984954864594, + "loss": 1.3997, + "theoretical_loss": 3.522101942622737, + "tokens_seen": 1471217664 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": 0.000559692271053791, + "objective/train/docs_used": 834912, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7659127712249756, + "objective/train/original_loss": 2.7659127712249756, + "objective/train/theoretical_loss": 3.522088058328375, + "objective/train/tokens_used": 1491743200, + "objective/train/value_avg": -0.004913330078125, + "objective/train/value_loss": 6.717603537254035e-05, + "objective/train/value_max": -0.0002033710479736328, + "objective/train/value_min": -0.06500244140625, + "objective/train/value_reward_corr": 0.4049106901241786, + "objective/train/value_std": 0.004055023193359375, + "objective/train/weight_avg": 1.0086299180984497, + "objective/train/weighted_lm_loss": 2.7977652549743652, + "objective/train/weights_max": 1.6661807298660278, + "objective/train/weights_min": 0.25155675411224365, + "theoretical_loss": 3.522088058328375, + "tokens_seen": 1471283200 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027993981945837517, + "loss": 1.4, + "theoretical_loss": 3.522088058328375, + "tokens_seen": 1471283200 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002799297893681043, + "loss": 1.2902, + "theoretical_loss": 3.5220741748256117, + "tokens_seen": 1471348736 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027991975927783353, + "loss": 1.3324, + "theoretical_loss": 3.522060292114366, + "tokens_seen": 1471414272 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002799097291875627, + "loss": 1.3561, + "theoretical_loss": 3.522046410194559, + "tokens_seen": 1471479808 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002798996990972919, + "loss": 1.2776, + "theoretical_loss": 3.52203252906611, + "tokens_seen": 1471545344 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002798896690070211, + "loss": 1.4884, + "theoretical_loss": 3.5220186487289373, + "tokens_seen": 1471610880 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027987963891675026, + "loss": 1.2818, + "theoretical_loss": 3.5220047691829617, + "tokens_seen": 1471676416 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027986960882647944, + "loss": 1.297, + "theoretical_loss": 3.5219908904281025, + "tokens_seen": 1471741952 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002798595787362087, + "loss": 1.4064, + "theoretical_loss": 3.52197701246428, + "tokens_seen": 1471807488 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002798495486459378, + "loss": 1.3178, + "theoretical_loss": 3.521963135291413, + "tokens_seen": 1471873024 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027983951855566704, + "loss": 1.3381, + "theoretical_loss": 3.5219492589094217, + "tokens_seen": 1471938560 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027982948846539616, + "loss": 1.3633, + "theoretical_loss": 3.5219353833182256, + "tokens_seen": 1472004096 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002798194583751254, + "loss": 1.2965, + "theoretical_loss": 3.5219215085177447, + "tokens_seen": 1472069632 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002798094282848546, + "loss": 1.292, + "theoretical_loss": 3.5219076345078983, + "tokens_seen": 1472135168 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027979939819458376, + "loss": 1.2828, + "theoretical_loss": 3.5218937612886068, + "tokens_seen": 1472200704 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027978936810431294, + "loss": 1.3299, + "theoretical_loss": 3.5218798888597895, + "tokens_seen": 1472266240 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002797793380140422, + "loss": 1.3218, + "theoretical_loss": 3.5218660172213663, + "tokens_seen": 1472331776 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002797693079237713, + "loss": 1.2709, + "theoretical_loss": 3.5218521463732566, + "tokens_seen": 1472397312 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027975927783350054, + "loss": 1.3784, + "theoretical_loss": 3.521838276315381, + "tokens_seen": 1472462848 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027974924774322967, + "loss": 1.2184, + "theoretical_loss": 3.521824407047659, + "tokens_seen": 1472528384 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002797392176529589, + "loss": 1.4077, + "theoretical_loss": 3.5218105385700103, + "tokens_seen": 1472593920 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002797291875626881, + "loss": 1.3481, + "theoretical_loss": 3.5217966708823543, + "tokens_seen": 1472659456 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027971915747241726, + "loss": 1.4369, + "theoretical_loss": 3.5217828039846113, + "tokens_seen": 1472724992 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027970912738214644, + "loss": 1.2805, + "theoretical_loss": 3.5217689378767014, + "tokens_seen": 1472790528 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002796990972918756, + "loss": 1.4258, + "theoretical_loss": 3.521755072558544, + "tokens_seen": 1472856064 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.0018488341011106968, + "objective/train/docs_used": 835528, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.847113847732544, + "objective/train/original_loss": 2.847113609313965, + "objective/train/theoretical_loss": 3.5217412080300594, + "objective/train/tokens_used": 1493381600, + "objective/train/value_avg": -0.004833221435546875, + "objective/train/value_loss": 7.593969348818064e-05, + "objective/train/value_max": -0.0004172325134277344, + "objective/train/value_min": -0.051666259765625, + "objective/train/value_reward_corr": 0.4575688958885449, + "objective/train/value_std": 0.004161834716796875, + "objective/train/weight_avg": 0.9848887920379639, + "objective/train/weighted_lm_loss": 2.816926956176758, + "objective/train/weights_max": 1.4230226278305054, + "objective/train/weights_min": 0.14489367604255676, + "theoretical_loss": 3.5217412080300594, + "tokens_seen": 1472921600 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002796890672016048, + "loss": 1.2952, + "theoretical_loss": 3.5217412080300594, + "tokens_seen": 1472921600 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027967903711133404, + "loss": 1.267, + "theoretical_loss": 3.5217273442911674, + "tokens_seen": 1472987136 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027966900702106317, + "loss": 1.3665, + "theoretical_loss": 3.5217134813417874, + "tokens_seen": 1473052672 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002796589769307924, + "loss": 1.3807, + "theoretical_loss": 3.52169961918184, + "tokens_seen": 1473118208 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027964894684052153, + "loss": 1.3643, + "theoretical_loss": 3.5216857578112446, + "tokens_seen": 1473183744 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027963891675025077, + "loss": 1.282, + "theoretical_loss": 3.521671897229922, + "tokens_seen": 1473249280 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027962888665997995, + "loss": 1.377, + "theoretical_loss": 3.521658037437791, + "tokens_seen": 1473314816 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027961885656970913, + "loss": 1.3125, + "theoretical_loss": 3.521644178434772, + "tokens_seen": 1473380352 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002796088264794383, + "loss": 1.3774, + "theoretical_loss": 3.521630320220785, + "tokens_seen": 1473445888 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027959879638916755, + "loss": 1.3634, + "theoretical_loss": 3.5216164627957505, + "tokens_seen": 1473511424 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027958876629889667, + "loss": 1.4371, + "theoretical_loss": 3.521602606159588, + "tokens_seen": 1473576960 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002795787362086259, + "loss": 1.3946, + "theoretical_loss": 3.521588750312217, + "tokens_seen": 1473642496 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027956870611835503, + "loss": 1.3584, + "theoretical_loss": 3.5215748952535586, + "tokens_seen": 1473708032 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027955867602808427, + "loss": 1.2424, + "theoretical_loss": 3.5215610409835323, + "tokens_seen": 1473773568 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027954864593781345, + "loss": 1.3521, + "theoretical_loss": 3.521547187502058, + "tokens_seen": 1473839104 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027953861584754263, + "loss": 1.4273, + "theoretical_loss": 3.521533334809056, + "tokens_seen": 1473904640 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002795285857572718, + "loss": 1.2772, + "theoretical_loss": 3.521519482904446, + "tokens_seen": 1473970176 + }, + { + "epoch": 0.45, + "learning_rate": 0.000279518555667001, + "loss": 1.2198, + "theoretical_loss": 3.5215056317881483, + "tokens_seen": 1474035712 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002795085255767302, + "loss": 1.2898, + "theoretical_loss": 3.5214917814600835, + "tokens_seen": 1474101248 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002794984954864594, + "loss": 1.2986, + "theoretical_loss": 3.5214779319201712, + "tokens_seen": 1474166784 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027948846539618854, + "loss": 1.2447, + "theoretical_loss": 3.5214640831683317, + "tokens_seen": 1474232320 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002794784353059178, + "loss": 1.2677, + "theoretical_loss": 3.521450235204485, + "tokens_seen": 1474297856 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002794684052156469, + "loss": 1.3573, + "theoretical_loss": 3.5214363880285506, + "tokens_seen": 1474363392 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027945837512537614, + "loss": 1.2586, + "theoretical_loss": 3.5214225416404497, + "tokens_seen": 1474428928 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002794483450351053, + "loss": 1.4434, + "theoretical_loss": 3.521408696040102, + "tokens_seen": 1474494464 + }, + { + "debugging/Self-BLEU-5": 0.365615496353209, + "debugging/distinct-1-grams": 0.7956029334862484, + "debugging/distinct-2-grams": 0.9610780075485269, + "debugging/entropy-1-grams": 5.809781272019493, + "debugging/entropy-2-grams": 6.493642411046769, + "debugging/length": 443.8181818181818, + "debugging/num_segments": 11, + "debugging/raw_token_scores_avg": 0.00662244250997901, + "debugging/raw_token_scores_std": 0.007457161322236061, + "debugging/score": 0.006146315175700628, + "debugging/score_std": 0.005962009299771863, + "epoch": 0.45, + "objective/train/advantage_avg": -0.0011339924531057477, + "objective/train/docs_used": 836610, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8118937015533447, + "objective/train/original_loss": 2.8118932247161865, + "objective/train/theoretical_loss": 3.521394851227428, + "objective/train/tokens_used": 1495020000, + "objective/train/value_avg": -0.005489349365234375, + "objective/train/value_loss": 5.42011002835352e-05, + "objective/train/value_max": -0.00037407875061035156, + "objective/train/value_min": -0.04840087890625, + "objective/train/value_reward_corr": 0.311265131150338, + "objective/train/value_std": 0.003955841064453125, + "objective/train/weight_avg": 0.9912189245223999, + "objective/train/weighted_lm_loss": 2.8029346466064453, + "objective/train/weights_max": 1.4156591892242432, + "objective/train/weights_min": 0.19344022870063782, + "theoretical_loss": 3.521394851227428, + "tokens_seen": 1474560000 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002794383149448345, + "loss": 1.3797, + "theoretical_loss": 3.521394851227428, + "tokens_seen": 1474560000 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002794282848545637, + "loss": 1.3972, + "theoretical_loss": 3.521381007202347, + "tokens_seen": 1474625536 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002794182547642929, + "loss": 1.2395, + "theoretical_loss": 3.5213671639647806, + "tokens_seen": 1474691072 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027940822467402204, + "loss": 1.1641, + "theoretical_loss": 3.5213533215146477, + "tokens_seen": 1474756608 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002793981945837513, + "loss": 1.2388, + "theoretical_loss": 3.521339479851869, + "tokens_seen": 1474822144 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002793881644934804, + "loss": 1.2947, + "theoretical_loss": 3.521325638976365, + "tokens_seen": 1474887680 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027937813440320964, + "loss": 1.4272, + "theoretical_loss": 3.5213117988880556, + "tokens_seen": 1474953216 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002793681043129388, + "loss": 1.2726, + "theoretical_loss": 3.521297959586861, + "tokens_seen": 1475018752 + }, + { + "epoch": 0.45, + "learning_rate": 0.000279358074222668, + "loss": 1.2953, + "theoretical_loss": 3.5212841210727017, + "tokens_seen": 1475084288 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002793480441323972, + "loss": 1.2443, + "theoretical_loss": 3.5212702833454985, + "tokens_seen": 1475149824 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027933801404212636, + "loss": 1.3018, + "theoretical_loss": 3.5212564464051703, + "tokens_seen": 1475215360 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027932798395185554, + "loss": 1.3961, + "theoretical_loss": 3.5212426102516385, + "tokens_seen": 1475280896 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002793179538615848, + "loss": 1.3768, + "theoretical_loss": 3.521228774884823, + "tokens_seen": 1475346432 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002793079237713139, + "loss": 1.3943, + "theoretical_loss": 3.5212149403046444, + "tokens_seen": 1475411968 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027929789368104314, + "loss": 1.2431, + "theoretical_loss": 3.5212011065110227, + "tokens_seen": 1475477504 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027928786359077227, + "loss": 1.5076, + "theoretical_loss": 3.5211872735038785, + "tokens_seen": 1475543040 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002792778335005015, + "loss": 1.3072, + "theoretical_loss": 3.521173441283132, + "tokens_seen": 1475608576 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027926780341023074, + "loss": 1.3087, + "theoretical_loss": 3.521159609848704, + "tokens_seen": 1475674112 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027925777331995987, + "loss": 1.4211, + "theoretical_loss": 3.521145779200514, + "tokens_seen": 1475739648 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002792477432296891, + "loss": 1.3969, + "theoretical_loss": 3.5211319493384834, + "tokens_seen": 1475805184 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002792377131394183, + "loss": 1.3634, + "theoretical_loss": 3.521118120262532, + "tokens_seen": 1475870720 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027922768304914746, + "loss": 1.353, + "theoretical_loss": 3.52110429197258, + "tokens_seen": 1475936256 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027921765295887664, + "loss": 1.3411, + "theoretical_loss": 3.521090464468548, + "tokens_seen": 1476001792 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002792076228686058, + "loss": 1.3688, + "theoretical_loss": 3.5210766377503573, + "tokens_seen": 1476067328 + }, + { + "epoch": 0.45, + "learning_rate": 0.000279197592778335, + "loss": 1.361, + "theoretical_loss": 3.5210628118179272, + "tokens_seen": 1476132864 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.0008457493386231363, + "objective/train/docs_used": 837052, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5496017932891846, + "objective/train/original_loss": 2.5496022701263428, + "objective/train/theoretical_loss": 3.521048986671179, + "objective/train/tokens_used": 1496658400, + "objective/train/value_avg": -0.005420684814453125, + "objective/train/value_loss": 6.966645014472306e-05, + "objective/train/value_max": -0.0002491474151611328, + "objective/train/value_min": -0.0631103515625, + "objective/train/value_reward_corr": 0.3281514238382601, + "objective/train/value_std": 0.00411224365234375, + "objective/train/weight_avg": 0.9948558211326599, + "objective/train/weighted_lm_loss": 2.531805992126465, + "objective/train/weights_max": 1.7922935485839844, + "objective/train/weights_min": 0.46235567331314087, + "theoretical_loss": 3.521048986671179, + "tokens_seen": 1476198400 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027918756268806424, + "loss": 1.286, + "theoretical_loss": 3.521048986671179, + "tokens_seen": 1476198400 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027917753259779337, + "loss": 1.4403, + "theoretical_loss": 3.5210351623100324, + "tokens_seen": 1476263936 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002791675025075226, + "loss": 1.3681, + "theoretical_loss": 3.5210213387344087, + "tokens_seen": 1476329472 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027915747241725173, + "loss": 1.2709, + "theoretical_loss": 3.521007515944228, + "tokens_seen": 1476395008 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027914744232698097, + "loss": 1.4586, + "theoretical_loss": 3.5209936939394106, + "tokens_seen": 1476460544 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027913741223671015, + "loss": 1.2466, + "theoretical_loss": 3.520979872719877, + "tokens_seen": 1476526080 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027912738214643933, + "loss": 1.3417, + "theoretical_loss": 3.520966052285548, + "tokens_seen": 1476591616 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002791173520561685, + "loss": 1.3178, + "theoretical_loss": 3.5209522326363443, + "tokens_seen": 1476657152 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027910732196589775, + "loss": 1.4695, + "theoretical_loss": 3.5209384137721864, + "tokens_seen": 1476722688 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027909729187562687, + "loss": 1.3263, + "theoretical_loss": 3.5209245956929944, + "tokens_seen": 1476788224 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002790872617853561, + "loss": 1.2988, + "theoretical_loss": 3.5209107783986893, + "tokens_seen": 1476853760 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027907723169508523, + "loss": 1.464, + "theoretical_loss": 3.5208969618891914, + "tokens_seen": 1476919296 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027906720160481447, + "loss": 1.3884, + "theoretical_loss": 3.5208831461644214, + "tokens_seen": 1476984832 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027905717151454365, + "loss": 1.3743, + "theoretical_loss": 3.5208693312243007, + "tokens_seen": 1477050368 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027904714142427283, + "loss": 1.3861, + "theoretical_loss": 3.5208555170687488, + "tokens_seen": 1477115904 + }, + { + "epoch": 0.45, + "learning_rate": 0.000279037111334002, + "loss": 1.2796, + "theoretical_loss": 3.5208417036976867, + "tokens_seen": 1477181440 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002790270812437312, + "loss": 1.3091, + "theoretical_loss": 3.520827891111035, + "tokens_seen": 1477246976 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002790170511534604, + "loss": 1.3624, + "theoretical_loss": 3.520814079308715, + "tokens_seen": 1477312512 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002790070210631896, + "loss": 1.3557, + "theoretical_loss": 3.520800268290646, + "tokens_seen": 1477378048 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027899699097291874, + "loss": 1.2244, + "theoretical_loss": 3.52078645805675, + "tokens_seen": 1477443584 + }, + { + "epoch": 0.45, + "learning_rate": 0.000278986960882648, + "loss": 1.387, + "theoretical_loss": 3.5207726486069473, + "tokens_seen": 1477509120 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002789769307923771, + "loss": 1.4299, + "theoretical_loss": 3.5207588399411582, + "tokens_seen": 1477574656 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027896690070210634, + "loss": 1.3452, + "theoretical_loss": 3.5207450320593043, + "tokens_seen": 1477640192 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002789568706118355, + "loss": 1.3828, + "theoretical_loss": 3.520731224961305, + "tokens_seen": 1477705728 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002789468405215647, + "loss": 1.1476, + "theoretical_loss": 3.5207174186470827, + "tokens_seen": 1477771264 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.0002029037568718195, + "objective/train/docs_used": 837925, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.902714252471924, + "objective/train/original_loss": 2.902714252471924, + "objective/train/theoretical_loss": 3.5207036131165568, + "objective/train/tokens_used": 1498296800, + "objective/train/value_avg": -0.004940032958984375, + "objective/train/value_loss": 4.765613266499713e-05, + "objective/train/value_max": -0.0003726482391357422, + "objective/train/value_min": -0.08648681640625, + "objective/train/value_reward_corr": 0.47703743012579003, + "objective/train/value_std": 0.004322052001953125, + "objective/train/weight_avg": 1.0002291202545166, + "objective/train/weighted_lm_loss": 2.9108779430389404, + "objective/train/weights_max": 1.7126208543777466, + "objective/train/weights_min": 0.42227697372436523, + "theoretical_loss": 3.5207036131165568, + "tokens_seen": 1477836800 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002789368104312939, + "loss": 1.3648, + "theoretical_loss": 3.5207036131165568, + "tokens_seen": 1477836800 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002789267803410231, + "loss": 1.3744, + "theoretical_loss": 3.5206898083696485, + "tokens_seen": 1477902336 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027891675025075224, + "loss": 1.4165, + "theoretical_loss": 3.520676004406279, + "tokens_seen": 1477967872 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002789067201604815, + "loss": 1.3701, + "theoretical_loss": 3.520662201226368, + "tokens_seen": 1478033408 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002788966900702106, + "loss": 1.271, + "theoretical_loss": 3.5206483988298376, + "tokens_seen": 1478098944 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027888665997993984, + "loss": 1.3514, + "theoretical_loss": 3.520634597216608, + "tokens_seen": 1478164480 + }, + { + "epoch": 0.45, + "learning_rate": 0.000278876629889669, + "loss": 1.4903, + "theoretical_loss": 3.5206207963865994, + "tokens_seen": 1478230016 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002788665997993982, + "loss": 1.2891, + "theoretical_loss": 3.5206069963397333, + "tokens_seen": 1478295552 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002788565697091274, + "loss": 1.2776, + "theoretical_loss": 3.5205931970759314, + "tokens_seen": 1478361088 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027884653961885656, + "loss": 1.3451, + "theoretical_loss": 3.520579398595113, + "tokens_seen": 1478426624 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027883650952858574, + "loss": 1.3656, + "theoretical_loss": 3.5205656008972, + "tokens_seen": 1478492160 + }, + { + "epoch": 0.45, + "learning_rate": 0.000278826479438315, + "loss": 1.3968, + "theoretical_loss": 3.520551803982112, + "tokens_seen": 1478557696 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002788164493480441, + "loss": 1.3873, + "theoretical_loss": 3.5205380078497717, + "tokens_seen": 1478623232 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027880641925777334, + "loss": 1.2802, + "theoretical_loss": 3.5205242125000984, + "tokens_seen": 1478688768 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027879638916750247, + "loss": 1.355, + "theoretical_loss": 3.5205104179330142, + "tokens_seen": 1478754304 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002787863590772317, + "loss": 1.3818, + "theoretical_loss": 3.5204966241484397, + "tokens_seen": 1478819840 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002787763289869609, + "loss": 1.4302, + "theoretical_loss": 3.5204828311462952, + "tokens_seen": 1478885376 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027876629889669007, + "loss": 1.4455, + "theoretical_loss": 3.5204690389265023, + "tokens_seen": 1478950912 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027875626880641925, + "loss": 1.3752, + "theoretical_loss": 3.520455247488982, + "tokens_seen": 1479016448 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002787462387161485, + "loss": 1.4041, + "theoretical_loss": 3.520441456833655, + "tokens_seen": 1479081984 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002787362086258776, + "loss": 1.2705, + "theoretical_loss": 3.520427666960442, + "tokens_seen": 1479147520 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027872617853560685, + "loss": 1.4137, + "theoretical_loss": 3.5204138778692644, + "tokens_seen": 1479213056 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027871614844533597, + "loss": 1.3925, + "theoretical_loss": 3.5204000895600434, + "tokens_seen": 1479278592 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002787061183550652, + "loss": 1.386, + "theoretical_loss": 3.5203863020326995, + "tokens_seen": 1479344128 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002786960882647944, + "loss": 1.3663, + "theoretical_loss": 3.520372515287154, + "tokens_seen": 1479409664 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.0006117890006862581, + "objective/train/docs_used": 838738, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8439199924468994, + "objective/train/original_loss": 2.843919515609741, + "objective/train/theoretical_loss": 3.520358729323328, + "objective/train/tokens_used": 1499935200, + "objective/train/value_avg": -0.00495147705078125, + "objective/train/value_loss": 7.201625703601167e-05, + "objective/train/value_max": -0.00020503997802734375, + "objective/train/value_min": -0.055816650390625, + "objective/train/value_reward_corr": 0.2777273468972459, + "objective/train/value_std": 0.004253387451171875, + "objective/train/weight_avg": 0.9972162246704102, + "objective/train/weighted_lm_loss": 2.8327696323394775, + "objective/train/weights_max": 1.6195976734161377, + "objective/train/weights_min": 0.37666237354278564, + "theoretical_loss": 3.520358729323328, + "tokens_seen": 1479475200 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027868605817452357, + "loss": 1.3701, + "theoretical_loss": 3.520358729323328, + "tokens_seen": 1479475200 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027867602808425275, + "loss": 1.3912, + "theoretical_loss": 3.5203449441411423, + "tokens_seen": 1479540736 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027866599799398193, + "loss": 1.3846, + "theoretical_loss": 3.5203311597405182, + "tokens_seen": 1479606272 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002786559679037111, + "loss": 1.3456, + "theoretical_loss": 3.520317376121377, + "tokens_seen": 1479671808 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027864593781344035, + "loss": 1.2047, + "theoretical_loss": 3.5203035932836393, + "tokens_seen": 1479737344 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002786359077231695, + "loss": 1.2497, + "theoretical_loss": 3.520289811227226, + "tokens_seen": 1479802880 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002786258776328987, + "loss": 1.2995, + "theoretical_loss": 3.520276029952059, + "tokens_seen": 1479868416 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002786158475426279, + "loss": 1.3133, + "theoretical_loss": 3.520262249458059, + "tokens_seen": 1479933952 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027860581745235707, + "loss": 1.1922, + "theoretical_loss": 3.5202484697451473, + "tokens_seen": 1479999488 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027859578736208625, + "loss": 1.2887, + "theoretical_loss": 3.5202346908132442, + "tokens_seen": 1480065024 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027858575727181544, + "loss": 1.5077, + "theoretical_loss": 3.520220912662272, + "tokens_seen": 1480130560 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002785757271815446, + "loss": 1.3732, + "theoretical_loss": 3.520207135292152, + "tokens_seen": 1480196096 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027856569709127385, + "loss": 1.3418, + "theoretical_loss": 3.5201933587028034, + "tokens_seen": 1480261632 + }, + { + "epoch": 0.45, + "learning_rate": 0.000278555667001003, + "loss": 1.4344, + "theoretical_loss": 3.5201795828941496, + "tokens_seen": 1480327168 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002785456369107322, + "loss": 1.2079, + "theoretical_loss": 3.5201658078661113, + "tokens_seen": 1480392704 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027853560682046134, + "loss": 1.2176, + "theoretical_loss": 3.5201520336186087, + "tokens_seen": 1480458240 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002785255767301906, + "loss": 1.3815, + "theoretical_loss": 3.5201382601515636, + "tokens_seen": 1480523776 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002785155466399198, + "loss": 1.3607, + "theoretical_loss": 3.5201244874648983, + "tokens_seen": 1480589312 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027850551654964894, + "loss": 1.3482, + "theoretical_loss": 3.5201107155585323, + "tokens_seen": 1480654848 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002784954864593782, + "loss": 1.4356, + "theoretical_loss": 3.520096944432388, + "tokens_seen": 1480720384 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002784854563691073, + "loss": 1.3933, + "theoretical_loss": 3.5200831740863863, + "tokens_seen": 1480785920 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027847542627883654, + "loss": 1.3405, + "theoretical_loss": 3.520069404520448, + "tokens_seen": 1480851456 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002784653961885657, + "loss": 1.401, + "theoretical_loss": 3.5200556357344954, + "tokens_seen": 1480916992 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002784553660982949, + "loss": 1.3341, + "theoretical_loss": 3.520041867728449, + "tokens_seen": 1480982528 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002784453360080241, + "loss": 1.3648, + "theoretical_loss": 3.5200281005022305, + "tokens_seen": 1481048064 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.0006699150544591248, + "objective/train/docs_used": 839976, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5591745376586914, + "objective/train/original_loss": 2.5591747760772705, + "objective/train/theoretical_loss": 3.5200143340557615, + "objective/train/tokens_used": 1501573600, + "objective/train/value_avg": -0.005886077880859375, + "objective/train/value_loss": 5.033959314459935e-05, + "objective/train/value_max": -0.00042057037353515625, + "objective/train/value_min": -0.081787109375, + "objective/train/value_reward_corr": 0.4580982783865395, + "objective/train/value_std": 0.005214691162109375, + "objective/train/weight_avg": 0.9957640767097473, + "objective/train/weighted_lm_loss": 2.548064947128296, + "objective/train/weights_max": 1.8272595405578613, + "objective/train/weights_min": 0.5674589276313782, + "theoretical_loss": 3.5200143340557615, + "tokens_seen": 1481113600 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002784353059177533, + "loss": 1.3455, + "theoretical_loss": 3.5200143340557615, + "tokens_seen": 1481113600 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027842527582748244, + "loss": 1.3029, + "theoretical_loss": 3.520000568388962, + "tokens_seen": 1481179136 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002784152457372117, + "loss": 1.3518, + "theoretical_loss": 3.519986803501755, + "tokens_seen": 1481244672 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002784052156469408, + "loss": 1.3654, + "theoretical_loss": 3.5199730393940616, + "tokens_seen": 1481310208 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027839518555667004, + "loss": 1.4101, + "theoretical_loss": 3.5199592760658023, + "tokens_seen": 1481375744 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002783851554663992, + "loss": 1.4457, + "theoretical_loss": 3.5199455135168987, + "tokens_seen": 1481441280 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002783751253761284, + "loss": 1.3734, + "theoretical_loss": 3.5199317517472726, + "tokens_seen": 1481506816 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002783650952858576, + "loss": 1.4061, + "theoretical_loss": 3.519917990756845, + "tokens_seen": 1481572352 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027835506519558676, + "loss": 1.2438, + "theoretical_loss": 3.519904230545538, + "tokens_seen": 1481637888 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027834503510531594, + "loss": 1.4156, + "theoretical_loss": 3.519890471113272, + "tokens_seen": 1481703424 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002783350050150452, + "loss": 1.3744, + "theoretical_loss": 3.51987671245997, + "tokens_seen": 1481768960 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002783249749247743, + "loss": 1.3735, + "theoretical_loss": 3.5198629545855518, + "tokens_seen": 1481834496 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027831494483450354, + "loss": 1.4394, + "theoretical_loss": 3.5198491974899397, + "tokens_seen": 1481900032 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027830491474423267, + "loss": 1.3858, + "theoretical_loss": 3.5198354411730546, + "tokens_seen": 1481965568 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002782948846539619, + "loss": 1.284, + "theoretical_loss": 3.5198216856348186, + "tokens_seen": 1482031104 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002782848545636911, + "loss": 1.3587, + "theoretical_loss": 3.519807930875153, + "tokens_seen": 1482096640 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027827482447342027, + "loss": 1.3599, + "theoretical_loss": 3.5197941768939796, + "tokens_seen": 1482162176 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027826479438314945, + "loss": 1.3862, + "theoretical_loss": 3.519780423691219, + "tokens_seen": 1482227712 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002782547642928787, + "loss": 1.3888, + "theoretical_loss": 3.519766671266794, + "tokens_seen": 1482293248 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002782447342026078, + "loss": 1.3458, + "theoretical_loss": 3.519752919620625, + "tokens_seen": 1482358784 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027823470411233705, + "loss": 1.3397, + "theoretical_loss": 3.519739168752634, + "tokens_seen": 1482424320 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027822467402206617, + "loss": 1.3994, + "theoretical_loss": 3.519725418662743, + "tokens_seen": 1482489856 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002782146439317954, + "loss": 1.3326, + "theoretical_loss": 3.519711669350873, + "tokens_seen": 1482555392 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002782046138415246, + "loss": 1.4493, + "theoretical_loss": 3.5196979208169457, + "tokens_seen": 1482620928 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027819458375125377, + "loss": 1.2103, + "theoretical_loss": 3.5196841730608828, + "tokens_seen": 1482686464 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.0013120669173076749, + "objective/train/docs_used": 840589, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.335341453552246, + "objective/train/original_loss": 2.335341453552246, + "objective/train/theoretical_loss": 3.519670426082606, + "objective/train/tokens_used": 1503212000, + "objective/train/value_avg": -0.00499725341796875, + "objective/train/value_loss": 6.349405157379806e-05, + "objective/train/value_max": -0.0004012584686279297, + "objective/train/value_min": -0.04132080078125, + "objective/train/value_reward_corr": 0.34058843851569187, + "objective/train/value_std": 0.0034618377685546875, + "objective/train/weight_avg": 0.9898606538772583, + "objective/train/weighted_lm_loss": 2.3294546604156494, + "objective/train/weights_max": 1.4329637289047241, + "objective/train/weights_min": 0.4973342716693878, + "theoretical_loss": 3.519670426082606, + "tokens_seen": 1482752000 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027818455366098295, + "loss": 1.2495, + "theoretical_loss": 3.519670426082606, + "tokens_seen": 1482752000 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027817452357071213, + "loss": 1.3315, + "theoretical_loss": 3.519656679882037, + "tokens_seen": 1482817536 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002781644934804413, + "loss": 1.4332, + "theoretical_loss": 3.519642934459097, + "tokens_seen": 1482883072 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027815446339017055, + "loss": 1.288, + "theoretical_loss": 3.5196291898137075, + "tokens_seen": 1482948608 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002781444332998997, + "loss": 1.3813, + "theoretical_loss": 3.5196154459457905, + "tokens_seen": 1483014144 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002781344032096289, + "loss": 1.288, + "theoretical_loss": 3.519601702855268, + "tokens_seen": 1483079680 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002781243731193581, + "loss": 1.3496, + "theoretical_loss": 3.519587960542062, + "tokens_seen": 1483145216 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002781143430290873, + "loss": 1.3341, + "theoretical_loss": 3.519574219006093, + "tokens_seen": 1483210752 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027810431293881645, + "loss": 1.2975, + "theoretical_loss": 3.519560478247283, + "tokens_seen": 1483276288 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027809428284854564, + "loss": 1.396, + "theoretical_loss": 3.5195467382655545, + "tokens_seen": 1483341824 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002780842527582748, + "loss": 1.3652, + "theoretical_loss": 3.5195329990608286, + "tokens_seen": 1483407360 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027807422266800405, + "loss": 1.2519, + "theoretical_loss": 3.5195192606330266, + "tokens_seen": 1483472896 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002780641925777332, + "loss": 1.4311, + "theoretical_loss": 3.5195055229820715, + "tokens_seen": 1483538432 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002780541624874624, + "loss": 1.2381, + "theoretical_loss": 3.519491786107884, + "tokens_seen": 1483603968 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027804413239719154, + "loss": 1.2775, + "theoretical_loss": 3.5194780500103864, + "tokens_seen": 1483669504 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002780341023069208, + "loss": 1.3343, + "theoretical_loss": 3.5194643146895, + "tokens_seen": 1483735040 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027802407221664996, + "loss": 1.301, + "theoretical_loss": 3.519450580145147, + "tokens_seen": 1483800576 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027801404212637914, + "loss": 1.3754, + "theoretical_loss": 3.519436846377249, + "tokens_seen": 1483866112 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002780040120361083, + "loss": 1.385, + "theoretical_loss": 3.519423113385728, + "tokens_seen": 1483931648 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002779939819458375, + "loss": 1.2844, + "theoretical_loss": 3.5194093811705063, + "tokens_seen": 1483997184 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002779839518555667, + "loss": 1.3389, + "theoretical_loss": 3.5193956497315044, + "tokens_seen": 1484062720 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002779739217652959, + "loss": 1.3555, + "theoretical_loss": 3.5193819190686453, + "tokens_seen": 1484128256 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027796389167502504, + "loss": 1.2359, + "theoretical_loss": 3.5193681891818507, + "tokens_seen": 1484193792 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002779538615847543, + "loss": 1.4616, + "theoretical_loss": 3.5193544600710416, + "tokens_seen": 1484259328 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027794383149448346, + "loss": 1.3098, + "theoretical_loss": 3.5193407317361407, + "tokens_seen": 1484324864 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": 0.00028647526050917804, + "objective/train/docs_used": 841910, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5858993530273438, + "objective/train/original_loss": 2.585899591445923, + "objective/train/theoretical_loss": 3.5193270041770703, + "objective/train/tokens_used": 1504850400, + "objective/train/value_avg": -0.004772186279296875, + "objective/train/value_loss": 4.474067100090906e-05, + "objective/train/value_max": -0.0002472400665283203, + "objective/train/value_min": -0.06573486328125, + "objective/train/value_reward_corr": 0.4295297385767808, + "objective/train/value_std": 0.00395965576171875, + "objective/train/weight_avg": 1.004889965057373, + "objective/train/weighted_lm_loss": 2.5931107997894287, + "objective/train/weights_max": 1.5584888458251953, + "objective/train/weights_min": 0.13895803689956665, + "theoretical_loss": 3.5193270041770703, + "tokens_seen": 1484390400 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027793380140421264, + "loss": 1.2072, + "theoretical_loss": 3.5193270041770703, + "tokens_seen": 1484390400 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002779237713139418, + "loss": 1.3483, + "theoretical_loss": 3.5193132773937514, + "tokens_seen": 1484455936 + }, + { + "epoch": 0.45, + "learning_rate": 0.000277913741223671, + "loss": 1.335, + "theoretical_loss": 3.519299551386106, + "tokens_seen": 1484521472 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002779037111334002, + "loss": 1.2806, + "theoretical_loss": 3.5192858261540567, + "tokens_seen": 1484587008 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002778936810431294, + "loss": 1.4465, + "theoretical_loss": 3.519272101697525, + "tokens_seen": 1484652544 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027788365095285855, + "loss": 1.4356, + "theoretical_loss": 3.519258378016432, + "tokens_seen": 1484718080 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002778736208625878, + "loss": 1.3256, + "theoretical_loss": 3.5192446551107017, + "tokens_seen": 1484783616 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002778635907723169, + "loss": 1.3676, + "theoretical_loss": 3.5192309329802542, + "tokens_seen": 1484849152 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027785356068204614, + "loss": 1.2876, + "theoretical_loss": 3.519217211625013, + "tokens_seen": 1484914688 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002778435305917753, + "loss": 1.2449, + "theoretical_loss": 3.5192034910448986, + "tokens_seen": 1484980224 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002778335005015045, + "loss": 1.3454, + "theoretical_loss": 3.5191897712398337, + "tokens_seen": 1485045760 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002778234704112337, + "loss": 1.3517, + "theoretical_loss": 3.5191760522097404, + "tokens_seen": 1485111296 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027781344032096287, + "loss": 1.2935, + "theoretical_loss": 3.519162333954541, + "tokens_seen": 1485176832 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027780341023069205, + "loss": 1.4257, + "theoretical_loss": 3.5191486164741566, + "tokens_seen": 1485242368 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002777933801404213, + "loss": 1.317, + "theoretical_loss": 3.51913489976851, + "tokens_seen": 1485307904 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002777833500501504, + "loss": 1.3884, + "theoretical_loss": 3.519121183837523, + "tokens_seen": 1485373440 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027777331995987965, + "loss": 1.4087, + "theoretical_loss": 3.5191074686811183, + "tokens_seen": 1485438976 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002777632898696089, + "loss": 1.4525, + "theoretical_loss": 3.519093754299217, + "tokens_seen": 1485504512 + }, + { + "epoch": 0.45, + "learning_rate": 0.000277753259779338, + "loss": 1.3748, + "theoretical_loss": 3.519080040691742, + "tokens_seen": 1485570048 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027774322968906725, + "loss": 1.3695, + "theoretical_loss": 3.5190663278586145, + "tokens_seen": 1485635584 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027773319959879637, + "loss": 1.406, + "theoretical_loss": 3.519052615799758, + "tokens_seen": 1485701120 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002777231695085256, + "loss": 1.3533, + "theoretical_loss": 3.519038904515093, + "tokens_seen": 1485766656 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002777131394182548, + "loss": 1.2254, + "theoretical_loss": 3.519025194004543, + "tokens_seen": 1485832192 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027770310932798397, + "loss": 1.3764, + "theoretical_loss": 3.5190114842680296, + "tokens_seen": 1485897728 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027769307923771315, + "loss": 1.233, + "theoretical_loss": 3.5189977753054746, + "tokens_seen": 1485963264 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": 0.00019671676272992045, + "objective/train/docs_used": 842517, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5162813663482666, + "objective/train/original_loss": 2.5162811279296875, + "objective/train/theoretical_loss": 3.5189840671168007, + "objective/train/tokens_used": 1506488800, + "objective/train/value_avg": -0.00626373291015625, + "objective/train/value_loss": 6.57694399706088e-05, + "objective/train/value_max": -0.0004029273986816406, + "objective/train/value_min": -0.085693359375, + "objective/train/value_reward_corr": 0.5143217192077872, + "objective/train/value_std": 0.0057525634765625, + "objective/train/weight_avg": 1.0051100254058838, + "objective/train/weighted_lm_loss": 2.5253982543945312, + "objective/train/weights_max": 1.6125013828277588, + "objective/train/weights_min": 0.5366052985191345, + "theoretical_loss": 3.5189840671168007, + "tokens_seen": 1486028800 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027768304914744233, + "loss": 1.3843, + "theoretical_loss": 3.5189840671168007, + "tokens_seen": 1486028800 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002776730190571715, + "loss": 1.3738, + "theoretical_loss": 3.51897035970193, + "tokens_seen": 1486094336 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027766298896690075, + "loss": 1.3393, + "theoretical_loss": 3.518956653060785, + "tokens_seen": 1486159872 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002776529588766299, + "loss": 1.4161, + "theoretical_loss": 3.5189429471932874, + "tokens_seen": 1486225408 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002776429287863591, + "loss": 1.3504, + "theoretical_loss": 3.518929242099359, + "tokens_seen": 1486290944 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002776328986960883, + "loss": 1.4601, + "theoretical_loss": 3.5189155377789234, + "tokens_seen": 1486356480 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002776228686058175, + "loss": 1.2929, + "theoretical_loss": 3.5189018342319014, + "tokens_seen": 1486422016 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027761283851554665, + "loss": 1.3947, + "theoretical_loss": 3.5188881314582163, + "tokens_seen": 1486487552 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027760280842527584, + "loss": 1.4167, + "theoretical_loss": 3.5188744294577896, + "tokens_seen": 1486553088 + }, + { + "epoch": 0.45, + "learning_rate": 0.000277592778335005, + "loss": 1.3667, + "theoretical_loss": 3.5188607282305444, + "tokens_seen": 1486618624 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027758274824473425, + "loss": 1.2544, + "theoretical_loss": 3.518847027776402, + "tokens_seen": 1486684160 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002775727181544634, + "loss": 1.3202, + "theoretical_loss": 3.518833328095286, + "tokens_seen": 1486749696 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002775626880641926, + "loss": 1.332, + "theoretical_loss": 3.518819629187117, + "tokens_seen": 1486815232 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027755265797392174, + "loss": 1.3169, + "theoretical_loss": 3.518805931051819, + "tokens_seen": 1486880768 + }, + { + "epoch": 0.45, + "learning_rate": 0.000277542627883651, + "loss": 1.4097, + "theoretical_loss": 3.5187922336893136, + "tokens_seen": 1486946304 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027753259779338016, + "loss": 1.344, + "theoretical_loss": 3.518778537099523, + "tokens_seen": 1487011840 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027752256770310934, + "loss": 1.2262, + "theoretical_loss": 3.518764841282369, + "tokens_seen": 1487077376 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002775125376128385, + "loss": 1.3294, + "theoretical_loss": 3.5187511462377756, + "tokens_seen": 1487142912 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002775025075225677, + "loss": 1.3214, + "theoretical_loss": 3.518737451965664, + "tokens_seen": 1487208448 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002774924774322969, + "loss": 1.3521, + "theoretical_loss": 3.5187237584659568, + "tokens_seen": 1487273984 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002774824473420261, + "loss": 1.3706, + "theoretical_loss": 3.5187100657385764, + "tokens_seen": 1487339520 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027747241725175524, + "loss": 1.3631, + "theoretical_loss": 3.5186963737834454, + "tokens_seen": 1487405056 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002774623871614845, + "loss": 1.2644, + "theoretical_loss": 3.5186826826004856, + "tokens_seen": 1487470592 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027745235707121366, + "loss": 1.3139, + "theoretical_loss": 3.5186689921896206, + "tokens_seen": 1487536128 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027744232698094284, + "loss": 1.2904, + "theoretical_loss": 3.518655302550772, + "tokens_seen": 1487601664 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.0017998875118792057, + "objective/train/docs_used": 843832, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.808987617492676, + "objective/train/original_loss": 2.808987617492676, + "objective/train/theoretical_loss": 3.518641613683862, + "objective/train/tokens_used": 1508127200, + "objective/train/value_avg": -0.007354736328125, + "objective/train/value_loss": 0.00012099300511181355, + "objective/train/value_max": -0.00026726722717285156, + "objective/train/value_min": -0.06280517578125, + "objective/train/value_reward_corr": 0.5237203414438844, + "objective/train/value_std": 0.00649261474609375, + "objective/train/weight_avg": 0.9875363111495972, + "objective/train/weighted_lm_loss": 2.7857112884521484, + "objective/train/weights_max": 1.5615516901016235, + "objective/train/weights_min": 0.31319406628608704, + "theoretical_loss": 3.518641613683862, + "tokens_seen": 1487667200 + }, + { + "epoch": 0.45, + "learning_rate": 0.000277432296890672, + "loss": 1.3264, + "theoretical_loss": 3.518641613683862, + "tokens_seen": 1487667200 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002774222668004012, + "loss": 1.3271, + "theoretical_loss": 3.518627925588814, + "tokens_seen": 1487732736 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002774122367101304, + "loss": 1.409, + "theoretical_loss": 3.5186142382655494, + "tokens_seen": 1487798272 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002774022066198596, + "loss": 1.3724, + "theoretical_loss": 3.518600551713991, + "tokens_seen": 1487863808 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027739217652958875, + "loss": 1.3292, + "theoretical_loss": 3.5185868659340627, + "tokens_seen": 1487929344 + }, + { + "epoch": 0.45, + "learning_rate": 0.000277382146439318, + "loss": 1.4429, + "theoretical_loss": 3.518573180925685, + "tokens_seen": 1487994880 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002773721163490471, + "loss": 1.3518, + "theoretical_loss": 3.5185594966887814, + "tokens_seen": 1488060416 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027736208625877634, + "loss": 1.3818, + "theoretical_loss": 3.518545813223275, + "tokens_seen": 1488125952 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002773520561685055, + "loss": 1.2296, + "theoretical_loss": 3.5185321305290866, + "tokens_seen": 1488191488 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002773420260782347, + "loss": 1.3733, + "theoretical_loss": 3.518518448606141, + "tokens_seen": 1488257024 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002773319959879639, + "loss": 1.4299, + "theoretical_loss": 3.518504767454359, + "tokens_seen": 1488322560 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027732196589769307, + "loss": 1.4188, + "theoretical_loss": 3.5184910870736634, + "tokens_seen": 1488388096 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027731193580742225, + "loss": 1.3098, + "theoretical_loss": 3.5184774074639775, + "tokens_seen": 1488453632 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002773019057171515, + "loss": 1.2923, + "theoretical_loss": 3.518463728625224, + "tokens_seen": 1488519168 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002772918756268806, + "loss": 1.4017, + "theoretical_loss": 3.518450050557324, + "tokens_seen": 1488584704 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027728184553660985, + "loss": 1.3504, + "theoretical_loss": 3.5184363732602026, + "tokens_seen": 1488650240 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027727181544633903, + "loss": 1.2781, + "theoretical_loss": 3.51842269673378, + "tokens_seen": 1488715776 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002772617853560682, + "loss": 1.3091, + "theoretical_loss": 3.5184090209779804, + "tokens_seen": 1488781312 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002772517552657974, + "loss": 1.4448, + "theoretical_loss": 3.518395345992726, + "tokens_seen": 1488846848 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027724172517552657, + "loss": 1.421, + "theoretical_loss": 3.518381671777939, + "tokens_seen": 1488912384 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027723169508525575, + "loss": 1.3142, + "theoretical_loss": 3.518367998333543, + "tokens_seen": 1488977920 + }, + { + "epoch": 0.45, + "learning_rate": 0.000277221664994985, + "loss": 1.3539, + "theoretical_loss": 3.51835432565946, + "tokens_seen": 1489043456 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002772116349047141, + "loss": 1.3696, + "theoretical_loss": 3.5183406537556126, + "tokens_seen": 1489108992 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027720160481444335, + "loss": 1.3929, + "theoretical_loss": 3.5183269826219243, + "tokens_seen": 1489174528 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002771915747241725, + "loss": 1.2745, + "theoretical_loss": 3.518313312258317, + "tokens_seen": 1489240064 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": 4.759352304972708e-05, + "objective/train/docs_used": 844519, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4528965950012207, + "objective/train/original_loss": 2.4528965950012207, + "objective/train/theoretical_loss": 3.5182996426647137, + "objective/train/tokens_used": 1509765600, + "objective/train/value_avg": -0.005451202392578125, + "objective/train/value_loss": 5.1591661758720875e-05, + "objective/train/value_max": -0.00048041343688964844, + "objective/train/value_min": -0.06304931640625, + "objective/train/value_reward_corr": 0.30315256978610644, + "objective/train/value_std": 0.0040435791015625, + "objective/train/weight_avg": 1.0029215812683105, + "objective/train/weighted_lm_loss": 2.467534303665161, + "objective/train/weights_max": 1.6841150522232056, + "objective/train/weights_min": 0.2975884675979614, + "theoretical_loss": 3.5182996426647137, + "tokens_seen": 1489305600 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002771815446339017, + "loss": 1.4072, + "theoretical_loss": 3.5182996426647137, + "tokens_seen": 1489305600 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002771715145436309, + "loss": 1.2053, + "theoretical_loss": 3.5182859738410373, + "tokens_seen": 1489371136 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002771614844533601, + "loss": 1.3975, + "theoretical_loss": 3.5182723057872107, + "tokens_seen": 1489436672 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027715145436308926, + "loss": 1.3354, + "theoretical_loss": 3.5182586385031565, + "tokens_seen": 1489502208 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002771414242728185, + "loss": 1.4404, + "theoretical_loss": 3.518244971988797, + "tokens_seen": 1489567744 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002771313941825476, + "loss": 1.338, + "theoretical_loss": 3.518231306244056, + "tokens_seen": 1489633280 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027712136409227685, + "loss": 1.4043, + "theoretical_loss": 3.5182176412688553, + "tokens_seen": 1489698816 + }, + { + "epoch": 0.45, + "learning_rate": 0.000277111334002006, + "loss": 1.3397, + "theoretical_loss": 3.518203977063118, + "tokens_seen": 1489764352 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002771013039117352, + "loss": 1.3764, + "theoretical_loss": 3.5181903136267674, + "tokens_seen": 1489829888 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002770912738214644, + "loss": 1.4151, + "theoretical_loss": 3.5181766509597256, + "tokens_seen": 1489895424 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002770812437311936, + "loss": 1.3642, + "theoretical_loss": 3.5181629890619166, + "tokens_seen": 1489960960 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027707121364092276, + "loss": 1.165, + "theoretical_loss": 3.518149327933262, + "tokens_seen": 1490026496 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027706118355065194, + "loss": 1.3102, + "theoretical_loss": 3.5181356675736852, + "tokens_seen": 1490092032 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002770511534603811, + "loss": 1.3381, + "theoretical_loss": 3.5181220079831093, + "tokens_seen": 1490157568 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027704112337011036, + "loss": 1.3414, + "theoretical_loss": 3.518108349161457, + "tokens_seen": 1490223104 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002770310932798395, + "loss": 1.4269, + "theoretical_loss": 3.518094691108651, + "tokens_seen": 1490288640 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002770210631895687, + "loss": 1.3443, + "theoretical_loss": 3.5180810338246142, + "tokens_seen": 1490354176 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002770110330992979, + "loss": 1.3635, + "theoretical_loss": 3.5180673773092694, + "tokens_seen": 1490419712 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002770010030090271, + "loss": 1.3885, + "theoretical_loss": 3.5180537215625405, + "tokens_seen": 1490485248 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002769909729187563, + "loss": 1.2959, + "theoretical_loss": 3.5180400665843496, + "tokens_seen": 1490550784 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027698094282848544, + "loss": 1.3005, + "theoretical_loss": 3.5180264123746197, + "tokens_seen": 1490616320 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002769709127382147, + "loss": 1.3498, + "theoretical_loss": 3.518012758933274, + "tokens_seen": 1490681856 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027696088264794386, + "loss": 1.4259, + "theoretical_loss": 3.517999106260236, + "tokens_seen": 1490747392 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027695085255767304, + "loss": 1.3107, + "theoretical_loss": 3.5179854543554274, + "tokens_seen": 1490812928 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002769408224674022, + "loss": 1.4797, + "theoretical_loss": 3.5179718032187717, + "tokens_seen": 1490878464 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.0002507980680093169, + "objective/train/docs_used": 845874, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6979098320007324, + "objective/train/original_loss": 2.6979098320007324, + "objective/train/theoretical_loss": 3.517958152850192, + "objective/train/tokens_used": 1511404000, + "objective/train/value_avg": -0.00555419921875, + "objective/train/value_loss": 7.85633092164062e-05, + "objective/train/value_max": -0.00011235475540161133, + "objective/train/value_min": -0.06488037109375, + "objective/train/value_reward_corr": 0.4380958890105412, + "objective/train/value_std": 0.00490570068359375, + "objective/train/weight_avg": 1.0009872913360596, + "objective/train/weighted_lm_loss": 2.7198989391326904, + "objective/train/weights_max": 1.6683204174041748, + "objective/train/weights_min": 0.34273919463157654, + "theoretical_loss": 3.517958152850192, + "tokens_seen": 1490944000 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002769307923771314, + "loss": 1.3322, + "theoretical_loss": 3.517958152850192, + "tokens_seen": 1490944000 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002769207622868606, + "loss": 1.3577, + "theoretical_loss": 3.5179445032496126, + "tokens_seen": 1491009536 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002769107321965898, + "loss": 1.3199, + "theoretical_loss": 3.5179308544169543, + "tokens_seen": 1491075072 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027690070210631895, + "loss": 1.3614, + "theoretical_loss": 3.5179172063521413, + "tokens_seen": 1491140608 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002768906720160482, + "loss": 1.4425, + "theoretical_loss": 3.517903559055097, + "tokens_seen": 1491206144 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002768806419257773, + "loss": 1.3281, + "theoretical_loss": 3.517889912525744, + "tokens_seen": 1491271680 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027687061183550655, + "loss": 1.326, + "theoretical_loss": 3.5178762667640053, + "tokens_seen": 1491337216 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002768605817452357, + "loss": 1.2076, + "theoretical_loss": 3.5178626217698037, + "tokens_seen": 1491402752 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002768505516549649, + "loss": 1.2726, + "theoretical_loss": 3.5178489775430632, + "tokens_seen": 1491468288 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002768405215646941, + "loss": 1.2921, + "theoretical_loss": 3.5178353340837063, + "tokens_seen": 1491533824 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027683049147442327, + "loss": 1.2752, + "theoretical_loss": 3.5178216913916565, + "tokens_seen": 1491599360 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027682046138415245, + "loss": 1.3036, + "theoretical_loss": 3.5178080494668365, + "tokens_seen": 1491664896 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002768104312938817, + "loss": 1.4429, + "theoretical_loss": 3.51779440830917, + "tokens_seen": 1491730432 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002768004012036108, + "loss": 1.445, + "theoretical_loss": 3.5177807679185795, + "tokens_seen": 1491795968 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027679037111334005, + "loss": 1.3754, + "theoretical_loss": 3.517767128294988, + "tokens_seen": 1491861504 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027678034102306923, + "loss": 1.3713, + "theoretical_loss": 3.51775348943832, + "tokens_seen": 1491927040 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002767703109327984, + "loss": 1.3768, + "theoretical_loss": 3.517739851348497, + "tokens_seen": 1491992576 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002767602808425276, + "loss": 1.3311, + "theoretical_loss": 3.517726214025444, + "tokens_seen": 1492058112 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027675025075225677, + "loss": 1.3071, + "theoretical_loss": 3.5177125774690827, + "tokens_seen": 1492123648 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027674022066198595, + "loss": 1.3023, + "theoretical_loss": 3.517698941679337, + "tokens_seen": 1492189184 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002767301905717152, + "loss": 1.3842, + "theoretical_loss": 3.51768530665613, + "tokens_seen": 1492254720 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002767201604814443, + "loss": 1.3063, + "theoretical_loss": 3.517671672399385, + "tokens_seen": 1492320256 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027671013039117355, + "loss": 1.3855, + "theoretical_loss": 3.5176580389090253, + "tokens_seen": 1492385792 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002767001003009027, + "loss": 1.4169, + "theoretical_loss": 3.517644406184974, + "tokens_seen": 1492451328 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002766900702106319, + "loss": 1.2629, + "theoretical_loss": 3.5176307742271544, + "tokens_seen": 1492516864 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -7.959605864016339e-05, + "objective/train/docs_used": 846561, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6431827545166016, + "objective/train/original_loss": 2.6431822776794434, + "objective/train/theoretical_loss": 3.5176171430354897, + "objective/train/tokens_used": 1513042400, + "objective/train/value_avg": -0.00457763671875, + "objective/train/value_loss": 4.484792589209974e-05, + "objective/train/value_max": -0.00039196014404296875, + "objective/train/value_min": -0.062103271484375, + "objective/train/value_reward_corr": 0.400961007101351, + "objective/train/value_std": 0.003612518310546875, + "objective/train/weight_avg": 1.001295566558838, + "objective/train/weighted_lm_loss": 2.657761812210083, + "objective/train/weights_max": 1.6097922325134277, + "objective/train/weights_min": 0.24676179885864258, + "theoretical_loss": 3.5176171430354897, + "tokens_seen": 1492582400 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002766800401203611, + "loss": 1.3899, + "theoretical_loss": 3.5176171430354897, + "tokens_seen": 1492582400 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002766700100300903, + "loss": 1.3136, + "theoretical_loss": 3.517603512609904, + "tokens_seen": 1492647936 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027665997993981946, + "loss": 1.2755, + "theoretical_loss": 3.517589882950319, + "tokens_seen": 1492713472 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002766499498495487, + "loss": 1.3548, + "theoretical_loss": 3.51757625405666, + "tokens_seen": 1492779008 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002766399197592778, + "loss": 1.4204, + "theoretical_loss": 3.517562625928849, + "tokens_seen": 1492844544 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027662988966900705, + "loss": 1.3311, + "theoretical_loss": 3.51754899856681, + "tokens_seen": 1492910080 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002766198595787362, + "loss": 1.3927, + "theoretical_loss": 3.5175353719704656, + "tokens_seen": 1492975616 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002766098294884654, + "loss": 1.2945, + "theoretical_loss": 3.5175217461397397, + "tokens_seen": 1493041152 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002765997993981946, + "loss": 1.2362, + "theoretical_loss": 3.5175081210745556, + "tokens_seen": 1493106688 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002765897693079238, + "loss": 1.3771, + "theoretical_loss": 3.517494496774837, + "tokens_seen": 1493172224 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027657973921765296, + "loss": 1.3113, + "theoretical_loss": 3.5174808732405065, + "tokens_seen": 1493237760 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027656970912738214, + "loss": 1.4389, + "theoretical_loss": 3.5174672504714883, + "tokens_seen": 1493303296 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002765596790371113, + "loss": 1.4429, + "theoretical_loss": 3.5174536284677056, + "tokens_seen": 1493368832 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027654964894684056, + "loss": 1.2596, + "theoretical_loss": 3.517440007229082, + "tokens_seen": 1493434368 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002765396188565697, + "loss": 1.3604, + "theoretical_loss": 3.51742638675554, + "tokens_seen": 1493499904 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002765295887662989, + "loss": 1.4304, + "theoretical_loss": 3.5174127670470043, + "tokens_seen": 1493565440 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027651955867602805, + "loss": 1.2553, + "theoretical_loss": 3.5173991481033977, + "tokens_seen": 1493630976 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002765095285857573, + "loss": 1.3556, + "theoretical_loss": 3.5173855299246437, + "tokens_seen": 1493696512 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027649949849548646, + "loss": 1.4027, + "theoretical_loss": 3.517371912510666, + "tokens_seen": 1493762048 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027648946840521564, + "loss": 1.3401, + "theoretical_loss": 3.5173582958613885, + "tokens_seen": 1493827584 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002764794383149448, + "loss": 1.4082, + "theoretical_loss": 3.5173446799767336, + "tokens_seen": 1493893120 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027646940822467406, + "loss": 1.4189, + "theoretical_loss": 3.5173310648566254, + "tokens_seen": 1493958656 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002764593781344032, + "loss": 1.3149, + "theoretical_loss": 3.517317450500988, + "tokens_seen": 1494024192 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002764493480441324, + "loss": 1.3442, + "theoretical_loss": 3.5173038369097442, + "tokens_seen": 1494089728 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027643931795386155, + "loss": 1.3117, + "theoretical_loss": 3.517290224082817, + "tokens_seen": 1494155264 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.0017223914619535208, + "objective/train/docs_used": 848095, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8200342655181885, + "objective/train/original_loss": 2.8200340270996094, + "objective/train/theoretical_loss": 3.517276612020132, + "objective/train/tokens_used": 1514680800, + "objective/train/value_avg": -0.005405426025390625, + "objective/train/value_loss": 6.262039096327499e-05, + "objective/train/value_max": -0.00033664703369140625, + "objective/train/value_min": -0.1666259765625, + "objective/train/value_reward_corr": 0.4876224481474351, + "objective/train/value_std": 0.0044097900390625, + "objective/train/weight_avg": 0.9857997894287109, + "objective/train/weighted_lm_loss": 2.768656015396118, + "objective/train/weights_max": 4.828406810760498, + "objective/train/weights_min": 0.43251532316207886, + "theoretical_loss": 3.517276612020132, + "tokens_seen": 1494220800 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002764292878635908, + "loss": 1.2625, + "theoretical_loss": 3.517276612020132, + "tokens_seen": 1494220800 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027641925777331997, + "loss": 1.2509, + "theoretical_loss": 3.517263000721611, + "tokens_seen": 1494286336 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027640922768304915, + "loss": 1.296, + "theoretical_loss": 3.517249390187178, + "tokens_seen": 1494351872 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027639919759277833, + "loss": 1.4403, + "theoretical_loss": 3.517235780416757, + "tokens_seen": 1494417408 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002763891675025075, + "loss": 1.4117, + "theoretical_loss": 3.517222171410271, + "tokens_seen": 1494482944 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002763791374122367, + "loss": 1.43, + "theoretical_loss": 3.517208563167644, + "tokens_seen": 1494548480 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002763691073219659, + "loss": 1.2816, + "theoretical_loss": 3.5171949556888, + "tokens_seen": 1494614016 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027635907723169505, + "loss": 1.4132, + "theoretical_loss": 3.5171813489736614, + "tokens_seen": 1494679552 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002763490471414243, + "loss": 1.3408, + "theoretical_loss": 3.5171677430221533, + "tokens_seen": 1494745088 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002763390170511534, + "loss": 1.3772, + "theoretical_loss": 3.517154137834199, + "tokens_seen": 1494810624 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027632898696088265, + "loss": 1.3678, + "theoretical_loss": 3.517140533409721, + "tokens_seen": 1494876160 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027631895687061183, + "loss": 1.3371, + "theoretical_loss": 3.5171269297486445, + "tokens_seen": 1494941696 + }, + { + "epoch": 0.45, + "learning_rate": 0.000276308926780341, + "loss": 1.2657, + "theoretical_loss": 3.5171133268508927, + "tokens_seen": 1495007232 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002762988966900702, + "loss": 1.4052, + "theoretical_loss": 3.517099724716389, + "tokens_seen": 1495072768 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027628886659979943, + "loss": 1.3054, + "theoretical_loss": 3.5170861233450577, + "tokens_seen": 1495138304 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027627883650952856, + "loss": 1.3098, + "theoretical_loss": 3.5170725227368216, + "tokens_seen": 1495203840 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002762688064192578, + "loss": 1.4727, + "theoretical_loss": 3.5170589228916054, + "tokens_seen": 1495269376 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027625877632898697, + "loss": 1.4741, + "theoretical_loss": 3.5170453238093327, + "tokens_seen": 1495334912 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027624874623871615, + "loss": 1.4206, + "theoretical_loss": 3.517031725489926, + "tokens_seen": 1495400448 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002762387161484454, + "loss": 1.3887, + "theoretical_loss": 3.5170181279333113, + "tokens_seen": 1495465984 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002762286860581745, + "loss": 1.3196, + "theoretical_loss": 3.5170045311394107, + "tokens_seen": 1495531520 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027621865596790375, + "loss": 1.336, + "theoretical_loss": 3.516990935108148, + "tokens_seen": 1495597056 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002762086258776329, + "loss": 1.3387, + "theoretical_loss": 3.5169773398394484, + "tokens_seen": 1495662592 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002761985957873621, + "loss": 1.3747, + "theoretical_loss": 3.5169637453332343, + "tokens_seen": 1495728128 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002761885656970913, + "loss": 1.3062, + "theoretical_loss": 3.5169501515894304, + "tokens_seen": 1495793664 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.001216553500853479, + "objective/train/docs_used": 848766, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4589216709136963, + "objective/train/original_loss": 2.4589219093322754, + "objective/train/theoretical_loss": 3.5169365586079597, + "objective/train/tokens_used": 1516319200, + "objective/train/value_avg": -0.00620269775390625, + "objective/train/value_loss": 7.380575698334724e-05, + "objective/train/value_max": -0.00039196014404296875, + "objective/train/value_min": -0.05975341796875, + "objective/train/value_reward_corr": 0.4081879015278531, + "objective/train/value_std": 0.004974365234375, + "objective/train/weight_avg": 0.9913321733474731, + "objective/train/weighted_lm_loss": 2.438804864883423, + "objective/train/weights_max": 1.7619236707687378, + "objective/train/weights_min": 0.46826642751693726, + "theoretical_loss": 3.5169365586079597, + "tokens_seen": 1495859200 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002761785356068205, + "loss": 1.3262, + "theoretical_loss": 3.5169365586079597, + "tokens_seen": 1495859200 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027616850551654966, + "loss": 1.3304, + "theoretical_loss": 3.516922966388747, + "tokens_seen": 1495924736 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002761584754262789, + "loss": 1.313, + "theoretical_loss": 3.516909374931715, + "tokens_seen": 1495990272 + }, + { + "epoch": 0.45, + "learning_rate": 0.000276148445336008, + "loss": 1.2915, + "theoretical_loss": 3.516895784236789, + "tokens_seen": 1496055808 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027613841524573725, + "loss": 1.2755, + "theoretical_loss": 3.516882194303892, + "tokens_seen": 1496121344 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002761283851554664, + "loss": 1.3761, + "theoretical_loss": 3.516868605132948, + "tokens_seen": 1496186880 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002761183550651956, + "loss": 1.3907, + "theoretical_loss": 3.516855016723881, + "tokens_seen": 1496252416 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002761083249749248, + "loss": 1.3903, + "theoretical_loss": 3.516841429076615, + "tokens_seen": 1496317952 + }, + { + "epoch": 0.45, + "learning_rate": 0.000276098294884654, + "loss": 1.3017, + "theoretical_loss": 3.5168278421910735, + "tokens_seen": 1496383488 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027608826479438316, + "loss": 1.2193, + "theoretical_loss": 3.516814256067181, + "tokens_seen": 1496449024 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027607823470411234, + "loss": 1.4582, + "theoretical_loss": 3.5168006707048614, + "tokens_seen": 1496514560 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002760682046138415, + "loss": 1.3497, + "theoretical_loss": 3.516787086104038, + "tokens_seen": 1496580096 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027605817452357076, + "loss": 1.2983, + "theoretical_loss": 3.5167735022646354, + "tokens_seen": 1496645632 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002760481444332999, + "loss": 1.3027, + "theoretical_loss": 3.5167599191865775, + "tokens_seen": 1496711168 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002760381143430291, + "loss": 1.3774, + "theoretical_loss": 3.5167463368697884, + "tokens_seen": 1496776704 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027602808425275825, + "loss": 1.3431, + "theoretical_loss": 3.516732755314192, + "tokens_seen": 1496842240 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002760180541624875, + "loss": 1.2913, + "theoretical_loss": 3.5167191745197117, + "tokens_seen": 1496907776 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027600802407221666, + "loss": 1.3107, + "theoretical_loss": 3.516705594486272, + "tokens_seen": 1496973312 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027599799398194584, + "loss": 1.42, + "theoretical_loss": 3.5166920152137973, + "tokens_seen": 1497038848 + }, + { + "epoch": 0.45, + "learning_rate": 0.000275987963891675, + "loss": 1.419, + "theoretical_loss": 3.516678436702212, + "tokens_seen": 1497104384 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027597793380140426, + "loss": 1.393, + "theoretical_loss": 3.5166648589514384, + "tokens_seen": 1497169920 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002759679037111334, + "loss": 1.3138, + "theoretical_loss": 3.516651281961402, + "tokens_seen": 1497235456 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002759578736208626, + "loss": 1.2283, + "theoretical_loss": 3.516637705732027, + "tokens_seen": 1497300992 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027594784353059175, + "loss": 1.3752, + "theoretical_loss": 3.516624130263237, + "tokens_seen": 1497366528 + }, + { + "epoch": 0.45, + "learning_rate": 0.000275937813440321, + "loss": 1.3796, + "theoretical_loss": 3.516610555554956, + "tokens_seen": 1497432064 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -0.0008201107266359031, + "objective/train/docs_used": 849888, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3545782566070557, + "objective/train/original_loss": 2.3545782566070557, + "objective/train/theoretical_loss": 3.5165969816071083, + "objective/train/tokens_used": 1517957600, + "objective/train/value_avg": -0.005229949951171875, + "objective/train/value_loss": 6.192749424371868e-05, + "objective/train/value_max": -0.0003981590270996094, + "objective/train/value_min": -0.1456298828125, + "objective/train/value_reward_corr": 0.5327920568325143, + "objective/train/value_std": 0.006313323974609375, + "objective/train/weight_avg": 0.994703471660614, + "objective/train/weighted_lm_loss": 2.340620279312134, + "objective/train/weights_max": 3.038825750350952, + "objective/train/weights_min": 0.3107560873031616, + "theoretical_loss": 3.5165969816071083, + "tokens_seen": 1497497600 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027592778335005017, + "loss": 1.2096, + "theoretical_loss": 3.5165969816071083, + "tokens_seen": 1497497600 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027591775325977935, + "loss": 1.4626, + "theoretical_loss": 3.516583408419618, + "tokens_seen": 1497563136 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027590772316950853, + "loss": 1.3134, + "theoretical_loss": 3.516569835992409, + "tokens_seen": 1497628672 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002758976930792377, + "loss": 1.1856, + "theoretical_loss": 3.5165562643254056, + "tokens_seen": 1497694208 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002758876629889669, + "loss": 1.3212, + "theoretical_loss": 3.5165426934185327, + "tokens_seen": 1497759744 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002758776328986961, + "loss": 1.346, + "theoretical_loss": 3.516529123271713, + "tokens_seen": 1497825280 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027586760280842525, + "loss": 1.3035, + "theoretical_loss": 3.5165155538848722, + "tokens_seen": 1497890816 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002758575727181545, + "loss": 1.2719, + "theoretical_loss": 3.5165019852579333, + "tokens_seen": 1497956352 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002758475426278836, + "loss": 1.3189, + "theoretical_loss": 3.5164884173908213, + "tokens_seen": 1498021888 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027583751253761285, + "loss": 1.4051, + "theoretical_loss": 3.51647485028346, + "tokens_seen": 1498087424 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027582748244734203, + "loss": 1.3385, + "theoretical_loss": 3.516461283935774, + "tokens_seen": 1498152960 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002758174523570712, + "loss": 1.319, + "theoretical_loss": 3.5164477183476865, + "tokens_seen": 1498218496 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002758074222668004, + "loss": 1.2985, + "theoretical_loss": 3.516434153519123, + "tokens_seen": 1498284032 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027579739217652963, + "loss": 1.4405, + "theoretical_loss": 3.516420589450007, + "tokens_seen": 1498349568 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027578736208625876, + "loss": 1.2918, + "theoretical_loss": 3.5164070261402633, + "tokens_seen": 1498415104 + }, + { + "epoch": 0.45, + "learning_rate": 0.000275777331995988, + "loss": 1.3085, + "theoretical_loss": 3.516393463589816, + "tokens_seen": 1498480640 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002757673019057171, + "loss": 1.2955, + "theoretical_loss": 3.5163799017985893, + "tokens_seen": 1498546176 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027575727181544635, + "loss": 1.3802, + "theoretical_loss": 3.516366340766507, + "tokens_seen": 1498611712 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027574724172517554, + "loss": 1.2139, + "theoretical_loss": 3.516352780493494, + "tokens_seen": 1498677248 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002757372116349047, + "loss": 1.3672, + "theoretical_loss": 3.516339220979475, + "tokens_seen": 1498742784 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002757271815446339, + "loss": 1.391, + "theoretical_loss": 3.5163256622243733, + "tokens_seen": 1498808320 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002757171514543631, + "loss": 1.3233, + "theoretical_loss": 3.5163121042281142, + "tokens_seen": 1498873856 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027570712136409226, + "loss": 1.347, + "theoretical_loss": 3.5162985469906216, + "tokens_seen": 1498939392 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002756970912738215, + "loss": 1.4183, + "theoretical_loss": 3.51628499051182, + "tokens_seen": 1499004928 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002756870611835506, + "loss": 1.3033, + "theoretical_loss": 3.5162714347916335, + "tokens_seen": 1499070464 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": -1.739674189593643e-05, + "objective/train/docs_used": 850610, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.754971504211426, + "objective/train/original_loss": 2.754971742630005, + "objective/train/theoretical_loss": 3.516257879829986, + "objective/train/tokens_used": 1519596000, + "objective/train/value_avg": -0.005092620849609375, + "objective/train/value_loss": 5.753243021899834e-05, + "objective/train/value_max": -0.0003101825714111328, + "objective/train/value_min": -0.0701904296875, + "objective/train/value_reward_corr": 0.3917164199286444, + "objective/train/value_std": 0.0043487548828125, + "objective/train/weight_avg": 1.0024789571762085, + "objective/train/weighted_lm_loss": 2.765697956085205, + "objective/train/weights_max": 1.6803690195083618, + "objective/train/weights_min": 0.13936680555343628, + "theoretical_loss": 3.516257879829986, + "tokens_seen": 1499136000 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027567703109327986, + "loss": 1.248, + "theoretical_loss": 3.516257879829986, + "tokens_seen": 1499136000 + }, + { + "epoch": 0.45, + "learning_rate": 0.000275667001003009, + "loss": 1.309, + "theoretical_loss": 3.5162443256268032, + "tokens_seen": 1499201536 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002756569709127382, + "loss": 1.2891, + "theoretical_loss": 3.516230772182009, + "tokens_seen": 1499267072 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002756469408224674, + "loss": 1.285, + "theoretical_loss": 3.5162172194955277, + "tokens_seen": 1499332608 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002756369107321966, + "loss": 1.3393, + "theoretical_loss": 3.516203667567283, + "tokens_seen": 1499398144 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027562688064192576, + "loss": 1.3321, + "theoretical_loss": 3.516190116397201, + "tokens_seen": 1499463680 + }, + { + "epoch": 0.45, + "learning_rate": 0.000275616850551655, + "loss": 1.2348, + "theoretical_loss": 3.516176565985205, + "tokens_seen": 1499529216 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002756068204613841, + "loss": 1.4773, + "theoretical_loss": 3.5161630163312196, + "tokens_seen": 1499594752 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027559679037111336, + "loss": 1.3823, + "theoretical_loss": 3.516149467435169, + "tokens_seen": 1499660288 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002755867602808425, + "loss": 1.3401, + "theoretical_loss": 3.5161359192969783, + "tokens_seen": 1499725824 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002755767301905717, + "loss": 1.4031, + "theoretical_loss": 3.516122371916572, + "tokens_seen": 1499791360 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002755667001003009, + "loss": 1.3459, + "theoretical_loss": 3.516108825293874, + "tokens_seen": 1499856896 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002755566700100301, + "loss": 1.3156, + "theoretical_loss": 3.516095279428809, + "tokens_seen": 1499922432 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027554663991975927, + "loss": 1.2948, + "theoretical_loss": 3.516081734321302, + "tokens_seen": 1499987968 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027553660982948845, + "loss": 1.2659, + "theoretical_loss": 3.516068189971277, + "tokens_seen": 1500053504 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027552657973921763, + "loss": 1.3748, + "theoretical_loss": 3.516054646378659, + "tokens_seen": 1500119040 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027551654964894686, + "loss": 1.2996, + "theoretical_loss": 3.5160411035433725, + "tokens_seen": 1500184576 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027550651955867604, + "loss": 1.4022, + "theoretical_loss": 3.516027561465341, + "tokens_seen": 1500250112 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002754964894684052, + "loss": 1.3208, + "theoretical_loss": 3.5160140201444907, + "tokens_seen": 1500315648 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027548645937813446, + "loss": 1.2848, + "theoretical_loss": 3.5160004795807454, + "tokens_seen": 1500381184 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002754764292878636, + "loss": 1.3345, + "theoretical_loss": 3.5159869397740295, + "tokens_seen": 1500446720 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002754663991975928, + "loss": 1.3217, + "theoretical_loss": 3.5159734007242682, + "tokens_seen": 1500512256 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027545636910732195, + "loss": 1.2859, + "theoretical_loss": 3.5159598624313855, + "tokens_seen": 1500577792 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002754463390170512, + "loss": 1.2671, + "theoretical_loss": 3.5159463248953067, + "tokens_seen": 1500643328 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027543630892678037, + "loss": 1.3932, + "theoretical_loss": 3.515932788115956, + "tokens_seen": 1500708864 + }, + { + "epoch": 0.45, + "objective/train/advantage_avg": 0.00038928110734559596, + "objective/train/docs_used": 851359, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.567326545715332, + "objective/train/original_loss": 2.567326307296753, + "objective/train/theoretical_loss": 3.5159192520932576, + "objective/train/tokens_used": 1521234400, + "objective/train/value_avg": -0.00514984130859375, + "objective/train/value_loss": 5.5564960348419845e-05, + "objective/train/value_max": -0.0004673004150390625, + "objective/train/value_min": -0.04010009765625, + "objective/train/value_reward_corr": 0.3612654718062561, + "objective/train/value_std": 0.0037708282470703125, + "objective/train/weight_avg": 1.0063068866729736, + "objective/train/weighted_lm_loss": 2.5760064125061035, + "objective/train/weights_max": 1.3952717781066895, + "objective/train/weights_min": 0.13890239596366882, + "theoretical_loss": 3.5159192520932576, + "tokens_seen": 1500774400 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027542627883650955, + "loss": 1.3453, + "theoretical_loss": 3.5159192520932576, + "tokens_seen": 1500774400 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027541624874623873, + "loss": 1.3455, + "theoretical_loss": 3.5159057168271373, + "tokens_seen": 1500839936 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002754062186559679, + "loss": 1.3282, + "theoretical_loss": 3.5158921823175184, + "tokens_seen": 1500905472 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002753961885656971, + "loss": 1.4315, + "theoretical_loss": 3.5158786485643274, + "tokens_seen": 1500971008 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002753861584754263, + "loss": 1.3002, + "theoretical_loss": 3.5158651155674874, + "tokens_seen": 1501036544 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027537612838515545, + "loss": 1.365, + "theoretical_loss": 3.5158515833269237, + "tokens_seen": 1501102080 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002753660982948847, + "loss": 1.2968, + "theoretical_loss": 3.515838051842561, + "tokens_seen": 1501167616 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002753560682046138, + "loss": 1.4073, + "theoretical_loss": 3.5158245211143244, + "tokens_seen": 1501233152 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027534603811434305, + "loss": 1.3445, + "theoretical_loss": 3.515810991142138, + "tokens_seen": 1501298688 + }, + { + "epoch": 0.45, + "learning_rate": 0.00027533600802407223, + "loss": 1.3496, + "theoretical_loss": 3.515797461925927, + "tokens_seen": 1501364224 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002753259779338014, + "loss": 1.3111, + "theoretical_loss": 3.5157839334656154, + "tokens_seen": 1501429760 + }, + { + "epoch": 0.45, + "learning_rate": 0.0002753159478435306, + "loss": 1.3976, + "theoretical_loss": 3.515770405761129, + "tokens_seen": 1501495296 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027530591775325983, + "loss": 1.4062, + "theoretical_loss": 3.5157568788123923, + "tokens_seen": 1501560832 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027529588766298896, + "loss": 1.2974, + "theoretical_loss": 3.51574335261933, + "tokens_seen": 1501626368 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002752858575727182, + "loss": 1.4186, + "theoretical_loss": 3.5157298271818664, + "tokens_seen": 1501691904 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002752758274824473, + "loss": 1.3267, + "theoretical_loss": 3.5157163024999267, + "tokens_seen": 1501757440 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027526579739217655, + "loss": 1.3166, + "theoretical_loss": 3.5157027785734365, + "tokens_seen": 1501822976 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027525576730190574, + "loss": 1.337, + "theoretical_loss": 3.5156892554023194, + "tokens_seen": 1501888512 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002752457372116349, + "loss": 1.461, + "theoretical_loss": 3.5156757329865007, + "tokens_seen": 1501954048 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002752357071213641, + "loss": 1.2339, + "theoretical_loss": 3.515662211325906, + "tokens_seen": 1502019584 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002752256770310933, + "loss": 1.3497, + "theoretical_loss": 3.515648690420459, + "tokens_seen": 1502085120 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027521564694082246, + "loss": 1.3796, + "theoretical_loss": 3.5156351702700857, + "tokens_seen": 1502150656 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002752056168505517, + "loss": 1.3419, + "theoretical_loss": 3.5156216508747096, + "tokens_seen": 1502216192 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002751955867602808, + "loss": 1.3358, + "theoretical_loss": 3.5156081322342567, + "tokens_seen": 1502281728 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027518555667001006, + "loss": 1.3823, + "theoretical_loss": 3.515594614348652, + "tokens_seen": 1502347264 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.0006425058818422258, + "objective/train/docs_used": 852603, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.799377918243408, + "objective/train/original_loss": 2.7993781566619873, + "objective/train/theoretical_loss": 3.5155810972178196, + "objective/train/tokens_used": 1522872800, + "objective/train/value_avg": -0.0049285888671875, + "objective/train/value_loss": 4.961540616932325e-05, + "objective/train/value_max": -0.0001609325408935547, + "objective/train/value_min": -0.05645751953125, + "objective/train/value_reward_corr": 0.2831316335943326, + "objective/train/value_std": 0.003612518310546875, + "objective/train/weight_avg": 0.995943009853363, + "objective/train/weighted_lm_loss": 2.7831931114196777, + "objective/train/weights_max": 1.7126208543777466, + "objective/train/weights_min": 0.4469388425350189, + "theoretical_loss": 3.5155810972178196, + "tokens_seen": 1502412800 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002751755265797392, + "loss": 1.3595, + "theoretical_loss": 3.5155810972178196, + "tokens_seen": 1502412800 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002751654964894684, + "loss": 1.3038, + "theoretical_loss": 3.515567580841685, + "tokens_seen": 1502478336 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002751554663991976, + "loss": 1.2017, + "theoretical_loss": 3.515554065220173, + "tokens_seen": 1502543872 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002751454363089268, + "loss": 1.3609, + "theoretical_loss": 3.515540550353209, + "tokens_seen": 1502609408 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027513540621865596, + "loss": 1.2269, + "theoretical_loss": 3.515527036240717, + "tokens_seen": 1502674944 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002751253761283852, + "loss": 1.191, + "theoretical_loss": 3.515513522882623, + "tokens_seen": 1502740480 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002751153460381143, + "loss": 1.2342, + "theoretical_loss": 3.515500010278851, + "tokens_seen": 1502806016 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027510531594784356, + "loss": 1.2286, + "theoretical_loss": 3.5154864984293273, + "tokens_seen": 1502871552 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002750952858575727, + "loss": 1.2299, + "theoretical_loss": 3.5154729873339754, + "tokens_seen": 1502937088 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002750852557673019, + "loss": 1.2716, + "theoretical_loss": 3.5154594769927217, + "tokens_seen": 1503002624 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002750752256770311, + "loss": 1.2719, + "theoretical_loss": 3.5154459674054896, + "tokens_seen": 1503068160 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002750651955867603, + "loss": 1.3059, + "theoretical_loss": 3.515432458572206, + "tokens_seen": 1503133696 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027505516549648947, + "loss": 1.4465, + "theoretical_loss": 3.515418950492795, + "tokens_seen": 1503199232 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027504513540621865, + "loss": 1.3835, + "theoretical_loss": 3.5154054431671815, + "tokens_seen": 1503264768 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027503510531594783, + "loss": 1.2993, + "theoretical_loss": 3.5153919365952913, + "tokens_seen": 1503330304 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027502507522567706, + "loss": 1.3034, + "theoretical_loss": 3.515378430777049, + "tokens_seen": 1503395840 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002750150451354062, + "loss": 1.4063, + "theoretical_loss": 3.5153649257123787, + "tokens_seen": 1503461376 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002750050150451354, + "loss": 1.3698, + "theoretical_loss": 3.515351421401207, + "tokens_seen": 1503526912 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027499498495486455, + "loss": 1.3748, + "theoretical_loss": 3.515337917843459, + "tokens_seen": 1503592448 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002749849548645938, + "loss": 1.2216, + "theoretical_loss": 3.5153244150390597, + "tokens_seen": 1503657984 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027497492477432297, + "loss": 1.3347, + "theoretical_loss": 3.515310912987933, + "tokens_seen": 1503723520 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027496489468405215, + "loss": 1.3864, + "theoretical_loss": 3.5152974116900055, + "tokens_seen": 1503789056 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027495486459378133, + "loss": 1.4197, + "theoretical_loss": 3.5152839111452017, + "tokens_seen": 1503854592 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027494483450351057, + "loss": 1.3157, + "theoretical_loss": 3.5152704113534465, + "tokens_seen": 1503920128 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002749348044132397, + "loss": 1.246, + "theoretical_loss": 3.5152569123146664, + "tokens_seen": 1503985664 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": 0.00041763531044125557, + "objective/train/docs_used": 853123, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.594728469848633, + "objective/train/original_loss": 2.594728708267212, + "objective/train/theoretical_loss": 3.515243414028785, + "objective/train/tokens_used": 1524511200, + "objective/train/value_avg": -0.00505828857421875, + "objective/train/value_loss": 5.666246579494327e-05, + "objective/train/value_max": -0.0004565715789794922, + "objective/train/value_min": -0.04510498046875, + "objective/train/value_reward_corr": 0.4084341355865735, + "objective/train/value_std": 0.004016876220703125, + "objective/train/weight_avg": 1.0068442821502686, + "objective/train/weighted_lm_loss": 2.614215135574341, + "objective/train/weights_max": 1.4617563486099243, + "objective/train/weights_min": 0.5360580086708069, + "theoretical_loss": 3.515243414028785, + "tokens_seen": 1504051200 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027492477432296893, + "loss": 1.2867, + "theoretical_loss": 3.515243414028785, + "tokens_seen": 1504051200 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027491474423269806, + "loss": 1.2761, + "theoretical_loss": 3.5152299164957284, + "tokens_seen": 1504116736 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002749047141424273, + "loss": 1.3675, + "theoretical_loss": 3.515216419715421, + "tokens_seen": 1504182272 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027489468405215647, + "loss": 1.4423, + "theoretical_loss": 3.5152029236877897, + "tokens_seen": 1504247808 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027488465396188565, + "loss": 1.2814, + "theoretical_loss": 3.515189428412758, + "tokens_seen": 1504313344 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027487462387161483, + "loss": 1.2952, + "theoretical_loss": 3.5151759338902515, + "tokens_seen": 1504378880 + }, + { + "epoch": 0.46, + "learning_rate": 0.000274864593781344, + "loss": 1.3593, + "theoretical_loss": 3.5151624401201964, + "tokens_seen": 1504444416 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002748545636910732, + "loss": 1.3419, + "theoretical_loss": 3.5151489471025172, + "tokens_seen": 1504509952 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027484453360080243, + "loss": 1.2871, + "theoretical_loss": 3.5151354548371394, + "tokens_seen": 1504575488 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027483450351053156, + "loss": 1.431, + "theoretical_loss": 3.5151219633239874, + "tokens_seen": 1504641024 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002748244734202608, + "loss": 1.4231, + "theoretical_loss": 3.5151084725629884, + "tokens_seen": 1504706560 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027481444332999, + "loss": 1.4355, + "theoretical_loss": 3.515094982554066, + "tokens_seen": 1504772096 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027480441323971916, + "loss": 1.2748, + "theoretical_loss": 3.515081493297146, + "tokens_seen": 1504837632 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027479438314944834, + "loss": 1.4638, + "theoretical_loss": 3.5150680047921545, + "tokens_seen": 1504903168 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002747843530591775, + "loss": 1.29, + "theoretical_loss": 3.5150545170390153, + "tokens_seen": 1504968704 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027477432296890675, + "loss": 1.3322, + "theoretical_loss": 3.515041030037655, + "tokens_seen": 1505034240 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027476429287863594, + "loss": 1.3527, + "theoretical_loss": 3.5150275437879994, + "tokens_seen": 1505099776 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002747542627883651, + "loss": 1.3984, + "theoretical_loss": 3.5150140582899723, + "tokens_seen": 1505165312 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002747442326980943, + "loss": 1.3428, + "theoretical_loss": 3.5150005735435004, + "tokens_seen": 1505230848 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002747342026078235, + "loss": 1.4289, + "theoretical_loss": 3.514987089548508, + "tokens_seen": 1505296384 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027472417251755266, + "loss": 1.3449, + "theoretical_loss": 3.5149736063049213, + "tokens_seen": 1505361920 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002747141424272819, + "loss": 1.318, + "theoretical_loss": 3.5149601238126653, + "tokens_seen": 1505427456 + }, + { + "epoch": 0.46, + "learning_rate": 0.000274704112337011, + "loss": 1.3632, + "theoretical_loss": 3.514946642071666, + "tokens_seen": 1505492992 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027469408224674026, + "loss": 1.3482, + "theoretical_loss": 3.5149331610818484, + "tokens_seen": 1505558528 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002746840521564694, + "loss": 1.2479, + "theoretical_loss": 3.514919680843138, + "tokens_seen": 1505624064 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.000406974577344954, + "objective/train/docs_used": 854391, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.85479474067688, + "objective/train/original_loss": 2.8547942638397217, + "objective/train/theoretical_loss": 3.51490620135546, + "objective/train/tokens_used": 1526149600, + "objective/train/value_avg": -0.00589752197265625, + "objective/train/value_loss": 9.322638652520254e-05, + "objective/train/value_max": -0.0002613067626953125, + "objective/train/value_min": -0.07586669921875, + "objective/train/value_reward_corr": 0.28080106184007597, + "objective/train/value_std": 0.005359649658203125, + "objective/train/weight_avg": 1.0002896785736084, + "objective/train/weighted_lm_loss": 2.874056339263916, + "objective/train/weights_max": 2.135427713394165, + "objective/train/weights_min": 0.15568441152572632, + "theoretical_loss": 3.51490620135546, + "tokens_seen": 1505689600 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002746740220661986, + "loss": 1.3748, + "theoretical_loss": 3.51490620135546, + "tokens_seen": 1505689600 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002746639919759278, + "loss": 1.2937, + "theoretical_loss": 3.5148927226187405, + "tokens_seen": 1505755136 + }, + { + "epoch": 0.46, + "learning_rate": 0.000274653961885657, + "loss": 1.4295, + "theoretical_loss": 3.5148792446329047, + "tokens_seen": 1505820672 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027464393179538616, + "loss": 1.2622, + "theoretical_loss": 3.514865767397878, + "tokens_seen": 1505886208 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002746339017051154, + "loss": 1.3564, + "theoretical_loss": 3.514852290913586, + "tokens_seen": 1505951744 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002746238716148445, + "loss": 1.2619, + "theoretical_loss": 3.514838815179954, + "tokens_seen": 1506017280 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027461384152457376, + "loss": 1.3361, + "theoretical_loss": 3.5148253401969076, + "tokens_seen": 1506082816 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002746038114343029, + "loss": 1.3389, + "theoretical_loss": 3.514811865964372, + "tokens_seen": 1506148352 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002745937813440321, + "loss": 1.4, + "theoretical_loss": 3.514798392482274, + "tokens_seen": 1506213888 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002745837512537613, + "loss": 1.3816, + "theoretical_loss": 3.514784919750538, + "tokens_seen": 1506279424 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002745737211634905, + "loss": 1.2425, + "theoretical_loss": 3.51477144776909, + "tokens_seen": 1506344960 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027456369107321967, + "loss": 1.3399, + "theoretical_loss": 3.514757976537855, + "tokens_seen": 1506410496 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027455366098294885, + "loss": 1.413, + "theoretical_loss": 3.5147445060567595, + "tokens_seen": 1506476032 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027454363089267803, + "loss": 1.324, + "theoretical_loss": 3.514731036325728, + "tokens_seen": 1506541568 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027453360080240726, + "loss": 1.3401, + "theoretical_loss": 3.5147175673446873, + "tokens_seen": 1506607104 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002745235707121364, + "loss": 1.4125, + "theoretical_loss": 3.5147040991135623, + "tokens_seen": 1506672640 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002745135406218656, + "loss": 1.3436, + "theoretical_loss": 3.514690631632279, + "tokens_seen": 1506738176 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027450351053159475, + "loss": 1.326, + "theoretical_loss": 3.514677164900762, + "tokens_seen": 1506803712 + }, + { + "epoch": 0.46, + "learning_rate": 0.000274493480441324, + "loss": 1.3138, + "theoretical_loss": 3.514663698918939, + "tokens_seen": 1506869248 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027448345035105317, + "loss": 1.3332, + "theoretical_loss": 3.5146502336867336, + "tokens_seen": 1506934784 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027447342026078235, + "loss": 1.408, + "theoretical_loss": 3.5146367692040728, + "tokens_seen": 1507000320 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027446339017051153, + "loss": 1.4863, + "theoretical_loss": 3.5146233054708818, + "tokens_seen": 1507065856 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027445336008024077, + "loss": 1.2991, + "theoretical_loss": 3.5146098424870855, + "tokens_seen": 1507131392 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002744433299899699, + "loss": 1.4395, + "theoretical_loss": 3.5145963802526112, + "tokens_seen": 1507196928 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027443329989969913, + "loss": 1.4141, + "theoretical_loss": 3.514582918767384, + "tokens_seen": 1507262464 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.0009278397774323821, + "objective/train/docs_used": 855086, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.831824779510498, + "objective/train/original_loss": 2.831824541091919, + "objective/train/theoretical_loss": 3.5145694580313287, + "objective/train/tokens_used": 1527788000, + "objective/train/value_avg": -0.005832672119140625, + "objective/train/value_loss": 7.552171155111864e-05, + "objective/train/value_max": -0.0004897117614746094, + "objective/train/value_min": -0.06475830078125, + "objective/train/value_reward_corr": 0.51325986100791, + "objective/train/value_std": 0.00537872314453125, + "objective/train/weight_avg": 0.9942311644554138, + "objective/train/weighted_lm_loss": 2.830439805984497, + "objective/train/weights_max": 1.634827733039856, + "objective/train/weights_min": 0.33453837037086487, + "theoretical_loss": 3.5145694580313287, + "tokens_seen": 1507328000 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027442326980942826, + "loss": 1.4737, + "theoretical_loss": 3.5145694580313287, + "tokens_seen": 1507328000 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002744132397191575, + "loss": 1.3787, + "theoretical_loss": 3.514555998044372, + "tokens_seen": 1507393536 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027440320962888667, + "loss": 1.4184, + "theoretical_loss": 3.51454253880644, + "tokens_seen": 1507459072 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027439317953861585, + "loss": 1.2692, + "theoretical_loss": 3.514529080317457, + "tokens_seen": 1507524608 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027438314944834503, + "loss": 1.3374, + "theoretical_loss": 3.5145156225773504, + "tokens_seen": 1507590144 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002743731193580742, + "loss": 1.291, + "theoretical_loss": 3.5145021655860447, + "tokens_seen": 1507655680 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002743630892678034, + "loss": 1.3468, + "theoretical_loss": 3.5144887093434667, + "tokens_seen": 1507721216 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027435305917753263, + "loss": 1.4485, + "theoretical_loss": 3.5144752538495414, + "tokens_seen": 1507786752 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027434302908726176, + "loss": 1.2107, + "theoretical_loss": 3.514461799104195, + "tokens_seen": 1507852288 + }, + { + "epoch": 0.46, + "learning_rate": 0.000274332998996991, + "loss": 1.3956, + "theoretical_loss": 3.514448345107353, + "tokens_seen": 1507917824 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002743229689067202, + "loss": 1.3625, + "theoretical_loss": 3.514434891858942, + "tokens_seen": 1507983360 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027431293881644936, + "loss": 1.3864, + "theoretical_loss": 3.514421439358887, + "tokens_seen": 1508048896 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027430290872617854, + "loss": 1.3331, + "theoretical_loss": 3.514407987607114, + "tokens_seen": 1508114432 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002742928786359077, + "loss": 1.457, + "theoretical_loss": 3.5143945366035494, + "tokens_seen": 1508179968 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002742828485456369, + "loss": 1.3736, + "theoretical_loss": 3.5143810863481186, + "tokens_seen": 1508245504 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027427281845536614, + "loss": 1.2719, + "theoretical_loss": 3.5143676368407477, + "tokens_seen": 1508311040 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027426278836509526, + "loss": 1.4145, + "theoretical_loss": 3.5143541880813625, + "tokens_seen": 1508376576 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002742527582748245, + "loss": 1.407, + "theoretical_loss": 3.5143407400698887, + "tokens_seen": 1508442112 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002742427281845536, + "loss": 1.22, + "theoretical_loss": 3.514327292806253, + "tokens_seen": 1508507648 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027423269809428286, + "loss": 1.3718, + "theoretical_loss": 3.5143138462903796, + "tokens_seen": 1508573184 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027422266800401204, + "loss": 1.2887, + "theoretical_loss": 3.5143004005221963, + "tokens_seen": 1508638720 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002742126379137412, + "loss": 1.2798, + "theoretical_loss": 3.5142869555016283, + "tokens_seen": 1508704256 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002742026078234704, + "loss": 1.2473, + "theoretical_loss": 3.5142735112286014, + "tokens_seen": 1508769792 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002741925777331996, + "loss": 1.3701, + "theoretical_loss": 3.514260067703042, + "tokens_seen": 1508835328 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027418254764292877, + "loss": 1.3728, + "theoretical_loss": 3.5142466249248754, + "tokens_seen": 1508900864 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.0004903155495412648, + "objective/train/docs_used": 856541, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.912665843963623, + "objective/train/original_loss": 2.912665605545044, + "objective/train/theoretical_loss": 3.5142331828940288, + "objective/train/tokens_used": 1529426400, + "objective/train/value_avg": -0.005767822265625, + "objective/train/value_loss": 7.347849168581888e-05, + "objective/train/value_max": -0.0004355907440185547, + "objective/train/value_min": -0.057708740234375, + "objective/train/value_reward_corr": 0.41575093964448767, + "objective/train/value_std": 0.00496673583984375, + "objective/train/weight_avg": 0.998535692691803, + "objective/train/weighted_lm_loss": 2.925887107849121, + "objective/train/weights_max": 1.560882568359375, + "objective/train/weights_min": 0.3001307547092438, + "theoretical_loss": 3.5142331828940288, + "tokens_seen": 1508966400 + }, + { + "epoch": 0.46, + "learning_rate": 0.000274172517552658, + "loss": 1.2963, + "theoretical_loss": 3.5142331828940288, + "tokens_seen": 1508966400 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027416248746238713, + "loss": 1.3032, + "theoretical_loss": 3.514219741610427, + "tokens_seen": 1509031936 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027415245737211636, + "loss": 1.3744, + "theoretical_loss": 3.514206301073996, + "tokens_seen": 1509097472 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027414242728184554, + "loss": 1.2877, + "theoretical_loss": 3.5141928612846627, + "tokens_seen": 1509163008 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002741323971915747, + "loss": 1.3929, + "theoretical_loss": 3.5141794222423526, + "tokens_seen": 1509228544 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002741223671013039, + "loss": 1.3456, + "theoretical_loss": 3.514165983946992, + "tokens_seen": 1509294080 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002741123370110331, + "loss": 1.1789, + "theoretical_loss": 3.5141525463985066, + "tokens_seen": 1509359616 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027410230692076227, + "loss": 1.3442, + "theoretical_loss": 3.5141391095968224, + "tokens_seen": 1509425152 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002740922768304915, + "loss": 1.317, + "theoretical_loss": 3.5141256735418662, + "tokens_seen": 1509490688 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027408224674022063, + "loss": 1.2471, + "theoretical_loss": 3.5141122382335634, + "tokens_seen": 1509556224 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027407221664994987, + "loss": 1.2636, + "theoretical_loss": 3.514098803671841, + "tokens_seen": 1509621760 + }, + { + "epoch": 0.46, + "learning_rate": 0.000274062186559679, + "loss": 1.2888, + "theoretical_loss": 3.5140853698566232, + "tokens_seen": 1509687296 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027405215646940823, + "loss": 1.2311, + "theoretical_loss": 3.514071936787838, + "tokens_seen": 1509752832 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002740421263791374, + "loss": 1.3603, + "theoretical_loss": 3.5140585044654102, + "tokens_seen": 1509818368 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002740320962888666, + "loss": 1.4977, + "theoretical_loss": 3.5140450728892674, + "tokens_seen": 1509883904 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002740220661985958, + "loss": 1.2773, + "theoretical_loss": 3.5140316420593347, + "tokens_seen": 1509949440 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027401203610832495, + "loss": 1.3229, + "theoretical_loss": 3.5140182119755385, + "tokens_seen": 1510014976 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002740020060180542, + "loss": 1.2773, + "theoretical_loss": 3.5140047826378047, + "tokens_seen": 1510080512 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027399197592778337, + "loss": 1.2769, + "theoretical_loss": 3.51399135404606, + "tokens_seen": 1510146048 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027398194583751255, + "loss": 1.33, + "theoretical_loss": 3.51397792620023, + "tokens_seen": 1510211584 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027397191574724173, + "loss": 1.3044, + "theoretical_loss": 3.5139644991002417, + "tokens_seen": 1510277120 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027396188565697097, + "loss": 1.2818, + "theoretical_loss": 3.5139510727460204, + "tokens_seen": 1510342656 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002739518555667001, + "loss": 1.2622, + "theoretical_loss": 3.5139376471374932, + "tokens_seen": 1510408192 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027394182547642933, + "loss": 1.3142, + "theoretical_loss": 3.5139242222745857, + "tokens_seen": 1510473728 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027393179538615846, + "loss": 1.325, + "theoretical_loss": 3.513910798157224, + "tokens_seen": 1510539264 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.000804048147983849, + "objective/train/docs_used": 857299, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.276914596557617, + "objective/train/original_loss": 2.276914596557617, + "objective/train/theoretical_loss": 3.5138973747853353, + "objective/train/tokens_used": 1531064800, + "objective/train/value_avg": -0.00482940673828125, + "objective/train/value_loss": 3.666316843009554e-05, + "objective/train/value_max": -0.00044417381286621094, + "objective/train/value_min": -0.08929443359375, + "objective/train/value_reward_corr": 0.3521112614193997, + "objective/train/value_std": 0.0035858154296875, + "objective/train/weight_avg": 0.9937489032745361, + "objective/train/weighted_lm_loss": 2.2677037715911865, + "objective/train/weights_max": 2.2811620235443115, + "objective/train/weights_min": 0.5786400437355042, + "theoretical_loss": 3.5138973747853353, + "tokens_seen": 1510604800 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002739217652958877, + "loss": 1.25, + "theoretical_loss": 3.5138973747853353, + "tokens_seen": 1510604800 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027391173520561687, + "loss": 1.4104, + "theoretical_loss": 3.513883952158845, + "tokens_seen": 1510670336 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027390170511534605, + "loss": 1.327, + "theoretical_loss": 3.5138705302776794, + "tokens_seen": 1510735872 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027389167502507524, + "loss": 1.3381, + "theoretical_loss": 3.513857109141765, + "tokens_seen": 1510801408 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002738816449348044, + "loss": 1.2945, + "theoretical_loss": 3.5138436887510283, + "tokens_seen": 1510866944 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002738716148445336, + "loss": 1.2728, + "theoretical_loss": 3.5138302691053953, + "tokens_seen": 1510932480 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027386158475426283, + "loss": 1.3728, + "theoretical_loss": 3.513816850204793, + "tokens_seen": 1510998016 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027385155466399196, + "loss": 1.3623, + "theoretical_loss": 3.5138034320491465, + "tokens_seen": 1511063552 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002738415245737212, + "loss": 1.3554, + "theoretical_loss": 3.513790014638383, + "tokens_seen": 1511129088 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002738314944834504, + "loss": 1.2787, + "theoretical_loss": 3.513776597972429, + "tokens_seen": 1511194624 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027382146439317956, + "loss": 1.3749, + "theoretical_loss": 3.5137631820512096, + "tokens_seen": 1511260160 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027381143430290874, + "loss": 1.3773, + "theoretical_loss": 3.5137497668746525, + "tokens_seen": 1511325696 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002738014042126379, + "loss": 1.3612, + "theoretical_loss": 3.5137363524426837, + "tokens_seen": 1511391232 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002737913741223671, + "loss": 1.4102, + "theoretical_loss": 3.5137229387552296, + "tokens_seen": 1511456768 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027378134403209634, + "loss": 1.4151, + "theoretical_loss": 3.5137095258122164, + "tokens_seen": 1511522304 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027377131394182546, + "loss": 1.333, + "theoretical_loss": 3.513696113613571, + "tokens_seen": 1511587840 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002737612838515547, + "loss": 1.2333, + "theoretical_loss": 3.513682702159219, + "tokens_seen": 1511653376 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002737512537612838, + "loss": 1.4703, + "theoretical_loss": 3.5136692914490877, + "tokens_seen": 1511718912 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027374122367101306, + "loss": 1.3316, + "theoretical_loss": 3.513655881483103, + "tokens_seen": 1511784448 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027373119358074224, + "loss": 1.4164, + "theoretical_loss": 3.513642472261191, + "tokens_seen": 1511849984 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002737211634904714, + "loss": 1.2823, + "theoretical_loss": 3.513629063783279, + "tokens_seen": 1511915520 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002737111334002006, + "loss": 1.3271, + "theoretical_loss": 3.5136156560492937, + "tokens_seen": 1511981056 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002737011033099298, + "loss": 1.2986, + "theoretical_loss": 3.5136022490591605, + "tokens_seen": 1512046592 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027369107321965897, + "loss": 1.3242, + "theoretical_loss": 3.513588842812806, + "tokens_seen": 1512112128 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002736810431293882, + "loss": 1.4519, + "theoretical_loss": 3.5135754373101578, + "tokens_seen": 1512177664 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.0006654870230704546, + "objective/train/docs_used": 858636, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.66447114944458, + "objective/train/original_loss": 2.66447114944458, + "objective/train/theoretical_loss": 3.513562032551141, + "objective/train/tokens_used": 1532703200, + "objective/train/value_avg": -0.004978179931640625, + "objective/train/value_loss": 5.7375815231353045e-05, + "objective/train/value_max": -0.0004582405090332031, + "objective/train/value_min": -0.04986572265625, + "objective/train/value_reward_corr": 0.33172273789813034, + "objective/train/value_std": 0.0036373138427734375, + "objective/train/weight_avg": 0.9960423111915588, + "objective/train/weighted_lm_loss": 2.658586025238037, + "objective/train/weights_max": 1.504130482673645, + "objective/train/weights_min": 0.579215407371521, + "theoretical_loss": 3.513562032551141, + "tokens_seen": 1512243200 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027367101303911733, + "loss": 1.2103, + "theoretical_loss": 3.513562032551141, + "tokens_seen": 1512243200 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027366098294884656, + "loss": 1.3693, + "theoretical_loss": 3.513548628535683, + "tokens_seen": 1512308736 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027365095285857574, + "loss": 1.2698, + "theoretical_loss": 3.513535225263711, + "tokens_seen": 1512374272 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002736409227683049, + "loss": 1.4128, + "theoretical_loss": 3.51352182273515, + "tokens_seen": 1512439808 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002736308926780341, + "loss": 1.3703, + "theoretical_loss": 3.513508420949927, + "tokens_seen": 1512505344 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002736208625877633, + "loss": 1.3917, + "theoretical_loss": 3.5134950199079698, + "tokens_seen": 1512570880 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027361083249749247, + "loss": 1.3262, + "theoretical_loss": 3.5134816196092036, + "tokens_seen": 1512636416 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002736008024072217, + "loss": 1.2883, + "theoretical_loss": 3.513468220053555, + "tokens_seen": 1512701952 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027359077231695083, + "loss": 1.2689, + "theoretical_loss": 3.5134548212409515, + "tokens_seen": 1512767488 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027358074222668007, + "loss": 1.2665, + "theoretical_loss": 3.5134414231713187, + "tokens_seen": 1512833024 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002735707121364092, + "loss": 1.4154, + "theoretical_loss": 3.5134280258445845, + "tokens_seen": 1512898560 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027356068204613843, + "loss": 1.4695, + "theoretical_loss": 3.5134146292606743, + "tokens_seen": 1512964096 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002735506519558676, + "loss": 1.2978, + "theoretical_loss": 3.513401233419515, + "tokens_seen": 1513029632 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002735406218655968, + "loss": 1.4504, + "theoretical_loss": 3.5133878383210337, + "tokens_seen": 1513095168 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027353059177532597, + "loss": 1.3524, + "theoretical_loss": 3.513374443965157, + "tokens_seen": 1513160704 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027352056168505515, + "loss": 1.3591, + "theoretical_loss": 3.5133610503518113, + "tokens_seen": 1513226240 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027351053159478433, + "loss": 1.3152, + "theoretical_loss": 3.513347657480923, + "tokens_seen": 1513291776 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027350050150451357, + "loss": 1.5052, + "theoretical_loss": 3.5133342653524195, + "tokens_seen": 1513357312 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002734904714142427, + "loss": 1.4239, + "theoretical_loss": 3.5133208739662267, + "tokens_seen": 1513422848 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027348044132397193, + "loss": 1.358, + "theoretical_loss": 3.5133074833222726, + "tokens_seen": 1513488384 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002734704112337011, + "loss": 1.3694, + "theoretical_loss": 3.5132940934204826, + "tokens_seen": 1513553920 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002734603811434303, + "loss": 1.2016, + "theoretical_loss": 3.513280704260784, + "tokens_seen": 1513619456 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002734503510531595, + "loss": 1.3875, + "theoretical_loss": 3.5132673158431036, + "tokens_seen": 1513684992 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027344032096288866, + "loss": 1.3402, + "theoretical_loss": 3.5132539281673676, + "tokens_seen": 1513750528 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027343029087261784, + "loss": 1.2592, + "theoretical_loss": 3.513240541233504, + "tokens_seen": 1513816064 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.00031348023912869394, + "objective/train/docs_used": 859135, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7072391510009766, + "objective/train/original_loss": 2.7072391510009766, + "objective/train/theoretical_loss": 3.513227155041438, + "objective/train/tokens_used": 1534341600, + "objective/train/value_avg": -0.005435943603515625, + "objective/train/value_loss": 4.6429395297309384e-05, + "objective/train/value_max": -0.00019109249114990234, + "objective/train/value_min": -0.05059814453125, + "objective/train/value_reward_corr": 0.353181709523405, + "objective/train/value_std": 0.0036411285400390625, + "objective/train/weight_avg": 0.999075710773468, + "objective/train/weighted_lm_loss": 2.698431968688965, + "objective/train/weights_max": 1.6109546422958374, + "objective/train/weights_min": 0.5335944890975952, + "theoretical_loss": 3.513227155041438, + "tokens_seen": 1513881600 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002734202607823471, + "loss": 1.3223, + "theoretical_loss": 3.513227155041438, + "tokens_seen": 1513881600 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002734102306920762, + "loss": 1.3431, + "theoretical_loss": 3.5132137695910974, + "tokens_seen": 1513947136 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027340020060180544, + "loss": 1.3765, + "theoretical_loss": 3.513200384882409, + "tokens_seen": 1514012672 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027339017051153456, + "loss": 1.4018, + "theoretical_loss": 3.5131870009152992, + "tokens_seen": 1514078208 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002733801404212638, + "loss": 1.3891, + "theoretical_loss": 3.513173617689695, + "tokens_seen": 1514143744 + }, + { + "epoch": 0.46, + "learning_rate": 0.000273370110330993, + "loss": 1.2001, + "theoretical_loss": 3.513160235205523, + "tokens_seen": 1514209280 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027336008024072216, + "loss": 1.3561, + "theoretical_loss": 3.51314685346271, + "tokens_seen": 1514274816 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027335005015045134, + "loss": 1.3123, + "theoretical_loss": 3.513133472461184, + "tokens_seen": 1514340352 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002733400200601806, + "loss": 1.3791, + "theoretical_loss": 3.5131200922008703, + "tokens_seen": 1514405888 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002733299899699097, + "loss": 1.3203, + "theoretical_loss": 3.5131067126816964, + "tokens_seen": 1514471424 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027331995987963894, + "loss": 1.333, + "theoretical_loss": 3.5130933339035897, + "tokens_seen": 1514536960 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027330992978936807, + "loss": 1.3288, + "theoretical_loss": 3.513079955866476, + "tokens_seen": 1514602496 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002732998996990973, + "loss": 1.3889, + "theoretical_loss": 3.513066578570283, + "tokens_seen": 1514668032 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002732898696088265, + "loss": 1.2381, + "theoretical_loss": 3.5130532020149374, + "tokens_seen": 1514733568 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027327983951855566, + "loss": 1.4471, + "theoretical_loss": 3.513039826200366, + "tokens_seen": 1514799104 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002732698094282849, + "loss": 1.3729, + "theoretical_loss": 3.513026451126496, + "tokens_seen": 1514864640 + }, + { + "epoch": 0.46, + "learning_rate": 0.000273259779338014, + "loss": 1.2487, + "theoretical_loss": 3.5130130767932544, + "tokens_seen": 1514930176 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027324974924774326, + "loss": 1.3073, + "theoretical_loss": 3.5129997032005673, + "tokens_seen": 1514995712 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027323971915747244, + "loss": 1.4137, + "theoretical_loss": 3.5129863303483626, + "tokens_seen": 1515061248 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002732296890672016, + "loss": 1.2539, + "theoretical_loss": 3.512972958236567, + "tokens_seen": 1515126784 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002732196589769308, + "loss": 1.351, + "theoretical_loss": 3.512959586865108, + "tokens_seen": 1515192320 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027320962888666, + "loss": 1.2554, + "theoretical_loss": 3.512946216233911, + "tokens_seen": 1515257856 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027319959879638917, + "loss": 1.4047, + "theoretical_loss": 3.5129328463429053, + "tokens_seen": 1515323392 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002731895687061184, + "loss": 1.3172, + "theoretical_loss": 3.512919477192016, + "tokens_seen": 1515388928 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027317953861584753, + "loss": 1.3528, + "theoretical_loss": 3.5129061087811704, + "tokens_seen": 1515454464 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.0019470712868496776, + "objective/train/docs_used": 860397, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8746964931488037, + "objective/train/original_loss": 2.8746964931488037, + "objective/train/theoretical_loss": 3.5128927411102966, + "objective/train/tokens_used": 1535980000, + "objective/train/value_avg": -0.004482269287109375, + "objective/train/value_loss": 8.096267265500501e-05, + "objective/train/value_max": -0.0002397298812866211, + "objective/train/value_min": -0.03424072265625, + "objective/train/value_reward_corr": 0.27274559569725476, + "objective/train/value_std": 0.003314971923828125, + "objective/train/weight_avg": 0.9841191172599792, + "objective/train/weighted_lm_loss": 2.839481830596924, + "objective/train/weights_max": 1.2983287572860718, + "objective/train/weights_min": 0.09083762764930725, + "theoretical_loss": 3.5128927411102966, + "tokens_seen": 1515520000 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027316950852557676, + "loss": 1.3368, + "theoretical_loss": 3.5128927411102966, + "tokens_seen": 1515520000 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027315947843530594, + "loss": 1.2564, + "theoretical_loss": 3.5128793741793203, + "tokens_seen": 1515585536 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002731494483450351, + "loss": 1.2068, + "theoretical_loss": 3.51286600798817, + "tokens_seen": 1515651072 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002731394182547643, + "loss": 1.2604, + "theoretical_loss": 3.5128526425367714, + "tokens_seen": 1515716608 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002731293881644935, + "loss": 1.331, + "theoretical_loss": 3.5128392778250523, + "tokens_seen": 1515782144 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027311935807422267, + "loss": 1.3655, + "theoretical_loss": 3.5128259138529394, + "tokens_seen": 1515847680 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002731093279839519, + "loss": 1.4043, + "theoretical_loss": 3.5128125506203607, + "tokens_seen": 1515913216 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027309929789368103, + "loss": 1.4334, + "theoretical_loss": 3.512799188127242, + "tokens_seen": 1515978752 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027308926780341027, + "loss": 1.2058, + "theoretical_loss": 3.512785826373512, + "tokens_seen": 1516044288 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002730792377131394, + "loss": 1.3449, + "theoretical_loss": 3.512772465359096, + "tokens_seen": 1516109824 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027306920762286863, + "loss": 1.3825, + "theoretical_loss": 3.512759105083922, + "tokens_seen": 1516175360 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002730591775325978, + "loss": 1.2478, + "theoretical_loss": 3.512745745547918, + "tokens_seen": 1516240896 + }, + { + "epoch": 0.46, + "learning_rate": 0.000273049147442327, + "loss": 1.3163, + "theoretical_loss": 3.5127323867510096, + "tokens_seen": 1516306432 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027303911735205617, + "loss": 1.3243, + "theoretical_loss": 3.512719028693125, + "tokens_seen": 1516371968 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027302908726178535, + "loss": 1.2644, + "theoretical_loss": 3.5127056713741913, + "tokens_seen": 1516437504 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027301905717151453, + "loss": 1.3477, + "theoretical_loss": 3.512692314794135, + "tokens_seen": 1516503040 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027300902708124377, + "loss": 1.3899, + "theoretical_loss": 3.512678958952884, + "tokens_seen": 1516568576 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002729989969909729, + "loss": 1.3996, + "theoretical_loss": 3.5126656038503654, + "tokens_seen": 1516634112 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027298896690070213, + "loss": 1.4172, + "theoretical_loss": 3.5126522494865062, + "tokens_seen": 1516699648 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002729789368104313, + "loss": 1.3847, + "theoretical_loss": 3.512638895861234, + "tokens_seen": 1516765184 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002729689067201605, + "loss": 1.3057, + "theoretical_loss": 3.5126255429744755, + "tokens_seen": 1516830720 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002729588766298897, + "loss": 1.3707, + "theoretical_loss": 3.5126121908261583, + "tokens_seen": 1516896256 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027294884653961886, + "loss": 1.462, + "theoretical_loss": 3.5125988394162095, + "tokens_seen": 1516961792 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027293881644934804, + "loss": 1.4335, + "theoretical_loss": 3.5125854887445565, + "tokens_seen": 1517027328 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002729287863590773, + "loss": 1.2873, + "theoretical_loss": 3.5125721388111266, + "tokens_seen": 1517092864 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": 0.00030165936914272606, + "objective/train/docs_used": 860980, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.904541254043579, + "objective/train/original_loss": 2.904541492462158, + "objective/train/theoretical_loss": 3.5125587896158477, + "objective/train/tokens_used": 1537618400, + "objective/train/value_avg": -0.005153656005859375, + "objective/train/value_loss": 6.334056524792686e-05, + "objective/train/value_max": -0.00043892860412597656, + "objective/train/value_min": -0.07598876953125, + "objective/train/value_reward_corr": 0.34289993170863803, + "objective/train/value_std": 0.004001617431640625, + "objective/train/weight_avg": 1.005852222442627, + "objective/train/weighted_lm_loss": 2.921332359313965, + "objective/train/weights_max": 2.0579330921173096, + "objective/train/weights_min": 0.3003483712673187, + "theoretical_loss": 3.5125587896158477, + "tokens_seen": 1517158400 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002729187562688064, + "loss": 1.4006, + "theoretical_loss": 3.5125587896158477, + "tokens_seen": 1517158400 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027290872617853564, + "loss": 1.2624, + "theoretical_loss": 3.5125454411586454, + "tokens_seen": 1517223936 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027289869608826476, + "loss": 1.3, + "theoretical_loss": 3.5125320934394484, + "tokens_seen": 1517289472 + }, + { + "epoch": 0.46, + "learning_rate": 0.000272888665997994, + "loss": 1.2528, + "theoretical_loss": 3.512518746458184, + "tokens_seen": 1517355008 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002728786359077232, + "loss": 1.3701, + "theoretical_loss": 3.512505400214779, + "tokens_seen": 1517420544 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027286860581745236, + "loss": 1.2968, + "theoretical_loss": 3.5124920547091607, + "tokens_seen": 1517486080 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027285857572718154, + "loss": 1.3603, + "theoretical_loss": 3.5124787099412575, + "tokens_seen": 1517551616 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002728485456369108, + "loss": 1.4152, + "theoretical_loss": 3.5124653659109955, + "tokens_seen": 1517617152 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002728385155466399, + "loss": 1.3756, + "theoretical_loss": 3.5124520226183025, + "tokens_seen": 1517682688 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027282848545636914, + "loss": 1.3615, + "theoretical_loss": 3.512438680063106, + "tokens_seen": 1517748224 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027281845536609827, + "loss": 1.3011, + "theoretical_loss": 3.512425338245333, + "tokens_seen": 1517813760 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002728084252758275, + "loss": 1.3833, + "theoretical_loss": 3.5124119971649117, + "tokens_seen": 1517879296 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002727983951855567, + "loss": 1.2848, + "theoretical_loss": 3.5123986568217687, + "tokens_seen": 1517944832 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027278836509528586, + "loss": 1.52, + "theoretical_loss": 3.5123853172158324, + "tokens_seen": 1518010368 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027277833500501504, + "loss": 1.4191, + "theoretical_loss": 3.512371978347029, + "tokens_seen": 1518075904 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002727683049147442, + "loss": 1.359, + "theoretical_loss": 3.512358640215287, + "tokens_seen": 1518141440 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002727582748244734, + "loss": 1.3377, + "theoretical_loss": 3.512345302820533, + "tokens_seen": 1518206976 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027274824473420264, + "loss": 1.3039, + "theoretical_loss": 3.5123319661626953, + "tokens_seen": 1518272512 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027273821464393177, + "loss": 1.3968, + "theoretical_loss": 3.5123186302417007, + "tokens_seen": 1518338048 + }, + { + "epoch": 0.46, + "learning_rate": 0.000272728184553661, + "loss": 1.352, + "theoretical_loss": 3.512305295057477, + "tokens_seen": 1518403584 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027271815446339013, + "loss": 1.3462, + "theoretical_loss": 3.512291960609952, + "tokens_seen": 1518469120 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027270812437311937, + "loss": 1.4446, + "theoretical_loss": 3.5122786268990525, + "tokens_seen": 1518534656 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027269809428284855, + "loss": 1.3626, + "theoretical_loss": 3.5122652939247065, + "tokens_seen": 1518600192 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027268806419257773, + "loss": 1.3995, + "theoretical_loss": 3.5122519616868413, + "tokens_seen": 1518665728 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002726780341023069, + "loss": 1.3218, + "theoretical_loss": 3.5122386301853847, + "tokens_seen": 1518731264 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.0001803163904696703, + "objective/train/docs_used": 862373, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.110654592514038, + "objective/train/original_loss": 3.110654592514038, + "objective/train/theoretical_loss": 3.512225299420264, + "objective/train/tokens_used": 1539256800, + "objective/train/value_avg": -0.005199432373046875, + "objective/train/value_loss": 6.17879704805091e-05, + "objective/train/value_max": -0.0004076957702636719, + "objective/train/value_min": -0.0526123046875, + "objective/train/value_reward_corr": 0.4216308600733051, + "objective/train/value_std": 0.004058837890625, + "objective/train/weight_avg": 1.0009686946868896, + "objective/train/weighted_lm_loss": 3.1217801570892334, + "objective/train/weights_max": 1.692358374595642, + "objective/train/weights_min": 0.08790592849254608, + "theoretical_loss": 3.512225299420264, + "tokens_seen": 1518796800 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027266800401203614, + "loss": 1.3546, + "theoretical_loss": 3.512225299420264, + "tokens_seen": 1518796800 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027265797392176527, + "loss": 1.2503, + "theoretical_loss": 3.512211969391407, + "tokens_seen": 1518862336 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002726479438314945, + "loss": 1.2393, + "theoretical_loss": 3.5121986400987413, + "tokens_seen": 1518927872 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027263791374122363, + "loss": 1.3366, + "theoretical_loss": 3.512185311542194, + "tokens_seen": 1518993408 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027262788365095287, + "loss": 1.3338, + "theoretical_loss": 3.512171983721693, + "tokens_seen": 1519058944 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027261785356068205, + "loss": 1.302, + "theoretical_loss": 3.5121586566371663, + "tokens_seen": 1519124480 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027260782347041123, + "loss": 1.2589, + "theoretical_loss": 3.512145330288541, + "tokens_seen": 1519190016 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002725977933801404, + "loss": 1.3998, + "theoretical_loss": 3.5121320046757445, + "tokens_seen": 1519255552 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002725877632898696, + "loss": 1.4783, + "theoretical_loss": 3.512118679798705, + "tokens_seen": 1519321088 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002725777331995988, + "loss": 1.305, + "theoretical_loss": 3.51210535565735, + "tokens_seen": 1519386624 + }, + { + "epoch": 0.46, + "learning_rate": 0.000272567703109328, + "loss": 1.4049, + "theoretical_loss": 3.5120920322516067, + "tokens_seen": 1519452160 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027255767301905714, + "loss": 1.4416, + "theoretical_loss": 3.5120787095814032, + "tokens_seen": 1519517696 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027254764292878637, + "loss": 1.3799, + "theoretical_loss": 3.5120653876466674, + "tokens_seen": 1519583232 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002725376128385155, + "loss": 1.338, + "theoretical_loss": 3.5120520664473265, + "tokens_seen": 1519648768 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027252758274824473, + "loss": 1.3821, + "theoretical_loss": 3.5120387459833085, + "tokens_seen": 1519714304 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027251755265797397, + "loss": 1.2472, + "theoretical_loss": 3.5120254262545405, + "tokens_seen": 1519779840 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002725075225677031, + "loss": 1.213, + "theoretical_loss": 3.5120121072609516, + "tokens_seen": 1519845376 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027249749247743233, + "loss": 1.3561, + "theoretical_loss": 3.511998789002468, + "tokens_seen": 1519910912 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002724874623871615, + "loss": 1.3521, + "theoretical_loss": 3.5119854714790177, + "tokens_seen": 1519976448 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002724774322968907, + "loss": 1.2985, + "theoretical_loss": 3.51197215469053, + "tokens_seen": 1520041984 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002724674022066199, + "loss": 1.3728, + "theoretical_loss": 3.5119588386369305, + "tokens_seen": 1520107520 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027245737211634906, + "loss": 1.3841, + "theoretical_loss": 3.511945523318148, + "tokens_seen": 1520173056 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027244734202607824, + "loss": 1.2696, + "theoretical_loss": 3.51193220873411, + "tokens_seen": 1520238592 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002724373119358075, + "loss": 1.3119, + "theoretical_loss": 3.511918894884745, + "tokens_seen": 1520304128 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002724272818455366, + "loss": 1.4172, + "theoretical_loss": 3.5119055817699802, + "tokens_seen": 1520369664 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": 0.0005355306202545762, + "objective/train/docs_used": 862945, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0267045497894287, + "objective/train/original_loss": 3.0267040729522705, + "objective/train/theoretical_loss": 3.511892269389743, + "objective/train/tokens_used": 1540895200, + "objective/train/value_avg": -0.00548553466796875, + "objective/train/value_loss": 4.403730054036714e-05, + "objective/train/value_max": -0.0001323223114013672, + "objective/train/value_min": -0.06549072265625, + "objective/train/value_reward_corr": 0.47472908803290087, + "objective/train/value_std": 0.00507354736328125, + "objective/train/weight_avg": 1.0075297355651855, + "objective/train/weighted_lm_loss": 3.0466649532318115, + "objective/train/weights_max": 1.6833703517913818, + "objective/train/weights_min": 0.6556447148323059, + "theoretical_loss": 3.511892269389743, + "tokens_seen": 1520435200 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027241725175526584, + "loss": 1.4101, + "theoretical_loss": 3.511892269389743, + "tokens_seen": 1520435200 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027240722166499496, + "loss": 1.3621, + "theoretical_loss": 3.511878957743962, + "tokens_seen": 1520500736 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002723971915747242, + "loss": 1.2832, + "theoretical_loss": 3.511865646832565, + "tokens_seen": 1520566272 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002723871614844534, + "loss": 1.3431, + "theoretical_loss": 3.5118523366554792, + "tokens_seen": 1520631808 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027237713139418256, + "loss": 1.2963, + "theoretical_loss": 3.511839027212633, + "tokens_seen": 1520697344 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027236710130391174, + "loss": 1.3329, + "theoretical_loss": 3.511825718503954, + "tokens_seen": 1520762880 + }, + { + "epoch": 0.46, + "learning_rate": 0.000272357071213641, + "loss": 1.3426, + "theoretical_loss": 3.51181241052937, + "tokens_seen": 1520828416 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002723470411233701, + "loss": 1.4112, + "theoretical_loss": 3.511799103288809, + "tokens_seen": 1520893952 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027233701103309934, + "loss": 1.3456, + "theoretical_loss": 3.511785796782199, + "tokens_seen": 1520959488 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027232698094282847, + "loss": 1.3776, + "theoretical_loss": 3.5117724910094674, + "tokens_seen": 1521025024 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002723169508525577, + "loss": 1.4026, + "theoretical_loss": 3.511759185970543, + "tokens_seen": 1521090560 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002723069207622869, + "loss": 1.4237, + "theoretical_loss": 3.511745881665353, + "tokens_seen": 1521156096 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027229689067201606, + "loss": 1.3567, + "theoretical_loss": 3.5117325780938256, + "tokens_seen": 1521221632 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027228686058174524, + "loss": 1.3235, + "theoretical_loss": 3.5117192752558886, + "tokens_seen": 1521287168 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002722768304914744, + "loss": 1.3472, + "theoretical_loss": 3.51170597315147, + "tokens_seen": 1521352704 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002722668004012036, + "loss": 1.3968, + "theoretical_loss": 3.511692671780498, + "tokens_seen": 1521418240 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027225677031093284, + "loss": 1.3696, + "theoretical_loss": 3.5116793711429004, + "tokens_seen": 1521483776 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027224674022066197, + "loss": 1.3073, + "theoretical_loss": 3.5116660712386047, + "tokens_seen": 1521549312 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002722367101303912, + "loss": 1.3114, + "theoretical_loss": 3.5116527720675395, + "tokens_seen": 1521614848 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027222668004012033, + "loss": 1.2207, + "theoretical_loss": 3.5116394736296326, + "tokens_seen": 1521680384 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027221664994984957, + "loss": 1.2056, + "theoretical_loss": 3.511626175924812, + "tokens_seen": 1521745920 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027220661985957875, + "loss": 1.3187, + "theoretical_loss": 3.511612878953006, + "tokens_seen": 1521811456 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027219658976930793, + "loss": 1.3777, + "theoretical_loss": 3.511599582714142, + "tokens_seen": 1521876992 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002721865596790371, + "loss": 1.3612, + "theoretical_loss": 3.5115862872081487, + "tokens_seen": 1521942528 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027217652958876635, + "loss": 1.3476, + "theoretical_loss": 3.511572992434954, + "tokens_seen": 1522008064 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.000459633010905236, + "objective/train/docs_used": 863367, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4866740703582764, + "objective/train/original_loss": 2.4866743087768555, + "objective/train/theoretical_loss": 3.5115596983944855, + "objective/train/tokens_used": 1542533600, + "objective/train/value_avg": -0.005031585693359375, + "objective/train/value_loss": 4.252339931554161e-05, + "objective/train/value_max": -0.0003936290740966797, + "objective/train/value_min": -0.052825927734375, + "objective/train/value_reward_corr": 0.4692553805799342, + "objective/train/value_std": 0.004467010498046875, + "objective/train/weight_avg": 0.9974629878997803, + "objective/train/weighted_lm_loss": 2.48685359954834, + "objective/train/weights_max": 1.5041238069534302, + "objective/train/weights_min": 0.5678053498268127, + "theoretical_loss": 3.5115596983944855, + "tokens_seen": 1522073600 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027216649949849547, + "loss": 1.2518, + "theoretical_loss": 3.5115596983944855, + "tokens_seen": 1522073600 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002721564694082247, + "loss": 1.3403, + "theoretical_loss": 3.5115464050866714, + "tokens_seen": 1522139136 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027214643931795383, + "loss": 1.3441, + "theoretical_loss": 3.51153311251144, + "tokens_seen": 1522204672 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027213640922768307, + "loss": 1.3279, + "theoretical_loss": 3.51151982066872, + "tokens_seen": 1522270208 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027212637913741225, + "loss": 1.2887, + "theoretical_loss": 3.5115065295584382, + "tokens_seen": 1522335744 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027211634904714143, + "loss": 1.3107, + "theoretical_loss": 3.511493239180524, + "tokens_seen": 1522401280 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002721063189568706, + "loss": 1.2247, + "theoretical_loss": 3.5114799495349045, + "tokens_seen": 1522466816 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002720962888665998, + "loss": 1.3318, + "theoretical_loss": 3.511466660621508, + "tokens_seen": 1522532352 + }, + { + "epoch": 0.46, + "learning_rate": 0.000272086258776329, + "loss": 1.3765, + "theoretical_loss": 3.511453372440263, + "tokens_seen": 1522597888 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002720762286860582, + "loss": 1.3752, + "theoretical_loss": 3.511440084991098, + "tokens_seen": 1522663424 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027206619859578734, + "loss": 1.3539, + "theoretical_loss": 3.5114267982739404, + "tokens_seen": 1522728960 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027205616850551657, + "loss": 1.2714, + "theoretical_loss": 3.5114135122887182, + "tokens_seen": 1522794496 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002720461384152457, + "loss": 1.3039, + "theoretical_loss": 3.5114002270353604, + "tokens_seen": 1522860032 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027203610832497493, + "loss": 1.3059, + "theoretical_loss": 3.511386942513795, + "tokens_seen": 1522925568 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002720260782347041, + "loss": 1.2381, + "theoretical_loss": 3.51137365872395, + "tokens_seen": 1522991104 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002720160481444333, + "loss": 1.3428, + "theoretical_loss": 3.5113603756657534, + "tokens_seen": 1523056640 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002720060180541625, + "loss": 1.2175, + "theoretical_loss": 3.5113470933391335, + "tokens_seen": 1523122176 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002719959879638917, + "loss": 1.2511, + "theoretical_loss": 3.511333811744019, + "tokens_seen": 1523187712 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027198595787362084, + "loss": 1.3462, + "theoretical_loss": 3.5113205308803384, + "tokens_seen": 1523253248 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002719759277833501, + "loss": 1.2861, + "theoretical_loss": 3.5113072507480183, + "tokens_seen": 1523318784 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002719658976930792, + "loss": 1.2057, + "theoretical_loss": 3.511293971346989, + "tokens_seen": 1523384320 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027195586760280844, + "loss": 1.3107, + "theoretical_loss": 3.511280692677177, + "tokens_seen": 1523449856 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002719458375125376, + "loss": 1.2337, + "theoretical_loss": 3.511267414738512, + "tokens_seen": 1523515392 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002719358074222668, + "loss": 1.2104, + "theoretical_loss": 3.5112541375309214, + "tokens_seen": 1523580928 + }, + { + "epoch": 0.46, + "learning_rate": 0.000271925777331996, + "loss": 1.2506, + "theoretical_loss": 3.511240861054334, + "tokens_seen": 1523646464 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.00035184354055672884, + "objective/train/docs_used": 864510, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.01428484916687, + "objective/train/original_loss": 3.014284372329712, + "objective/train/theoretical_loss": 3.511227585308678, + "objective/train/tokens_used": 1544172000, + "objective/train/value_avg": -0.00543975830078125, + "objective/train/value_loss": 5.6171083997469395e-05, + "objective/train/value_max": -0.0004189014434814453, + "objective/train/value_min": -0.05511474609375, + "objective/train/value_reward_corr": 0.40832642220416354, + "objective/train/value_std": 0.004276275634765625, + "objective/train/weight_avg": 0.9991597533226013, + "objective/train/weighted_lm_loss": 3.0194787979125977, + "objective/train/weights_max": 1.5870260000228882, + "objective/train/weights_min": 0.4853143095970154, + "theoretical_loss": 3.511227585308678, + "tokens_seen": 1523712000 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027191574724172516, + "loss": 1.3224, + "theoretical_loss": 3.511227585308678, + "tokens_seen": 1523712000 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027190571715145434, + "loss": 1.4398, + "theoretical_loss": 3.511214310293881, + "tokens_seen": 1523777536 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002718956870611836, + "loss": 1.2632, + "theoretical_loss": 3.511201036009872, + "tokens_seen": 1523843072 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002718856569709127, + "loss": 1.1788, + "theoretical_loss": 3.51118776245658, + "tokens_seen": 1523908608 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027187562688064194, + "loss": 1.3063, + "theoretical_loss": 3.511174489633933, + "tokens_seen": 1523974144 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027186559679037107, + "loss": 1.3065, + "theoretical_loss": 3.5111612175418583, + "tokens_seen": 1524039680 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002718555667001003, + "loss": 1.3804, + "theoretical_loss": 3.511147946180285, + "tokens_seen": 1524105216 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002718455366098295, + "loss": 1.212, + "theoretical_loss": 3.5111346755491417, + "tokens_seen": 1524170752 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027183550651955867, + "loss": 1.2702, + "theoretical_loss": 3.511121405648357, + "tokens_seen": 1524236288 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027182547642928785, + "loss": 1.3125, + "theoretical_loss": 3.511108136477858, + "tokens_seen": 1524301824 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002718154463390171, + "loss": 1.3616, + "theoretical_loss": 3.511094868037574, + "tokens_seen": 1524367360 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002718054162487462, + "loss": 1.368, + "theoretical_loss": 3.5110816003274348, + "tokens_seen": 1524432896 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027179538615847544, + "loss": 1.1587, + "theoretical_loss": 3.5110683333473665, + "tokens_seen": 1524498432 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027178535606820457, + "loss": 1.2185, + "theoretical_loss": 3.511055067097298, + "tokens_seen": 1524563968 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002717753259779338, + "loss": 1.2756, + "theoretical_loss": 3.511041801577159, + "tokens_seen": 1524629504 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027176529588766304, + "loss": 1.2003, + "theoretical_loss": 3.511028536786877, + "tokens_seen": 1524695040 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027175526579739217, + "loss": 1.2831, + "theoretical_loss": 3.511015272726381, + "tokens_seen": 1524760576 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002717452357071214, + "loss": 1.4855, + "theoretical_loss": 3.511002009395599, + "tokens_seen": 1524826112 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027173520561685053, + "loss": 1.4144, + "theoretical_loss": 3.5109887467944594, + "tokens_seen": 1524891648 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027172517552657977, + "loss": 1.2661, + "theoretical_loss": 3.5109754849228914, + "tokens_seen": 1524957184 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027171514543630895, + "loss": 1.3572, + "theoretical_loss": 3.5109622237808233, + "tokens_seen": 1525022720 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027170511534603813, + "loss": 1.2298, + "theoretical_loss": 3.510948963368183, + "tokens_seen": 1525088256 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002716950852557673, + "loss": 1.381, + "theoretical_loss": 3.5109357036848996, + "tokens_seen": 1525153792 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027168505516549655, + "loss": 1.2705, + "theoretical_loss": 3.5109224447309013, + "tokens_seen": 1525219328 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027167502507522567, + "loss": 1.3429, + "theoretical_loss": 3.510909186506117, + "tokens_seen": 1525284864 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": 0.0005466070724651217, + "objective/train/docs_used": 865194, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5877432823181152, + "objective/train/original_loss": 2.587743043899536, + "objective/train/theoretical_loss": 3.5108959290104753, + "objective/train/tokens_used": 1545810400, + "objective/train/value_avg": -0.005279541015625, + "objective/train/value_loss": 5.2969102398492396e-05, + "objective/train/value_max": -0.0002779960632324219, + "objective/train/value_min": -0.05145263671875, + "objective/train/value_reward_corr": 0.4654392567363029, + "objective/train/value_std": 0.00458526611328125, + "objective/train/weight_avg": 1.0079376697540283, + "objective/train/weighted_lm_loss": 2.6021006107330322, + "objective/train/weights_max": 1.6112169027328491, + "objective/train/weights_min": 0.3619396686553955, + "theoretical_loss": 3.5108959290104753, + "tokens_seen": 1525350400 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002716649949849549, + "loss": 1.2823, + "theoretical_loss": 3.5108959290104753, + "tokens_seen": 1525350400 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027165496489468403, + "loss": 1.2308, + "theoretical_loss": 3.5108826722439046, + "tokens_seen": 1525415936 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027164493480441327, + "loss": 1.3147, + "theoretical_loss": 3.5108694162063334, + "tokens_seen": 1525481472 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027163490471414245, + "loss": 1.3349, + "theoretical_loss": 3.5108561608976903, + "tokens_seen": 1525547008 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027162487462387163, + "loss": 1.3004, + "theoretical_loss": 3.5108429063179045, + "tokens_seen": 1525612544 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002716148445336008, + "loss": 1.3397, + "theoretical_loss": 3.510829652466904, + "tokens_seen": 1525678080 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027160481444333, + "loss": 1.2846, + "theoretical_loss": 3.510816399344617, + "tokens_seen": 1525743616 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002715947843530592, + "loss": 1.2665, + "theoretical_loss": 3.510803146950973, + "tokens_seen": 1525809152 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002715847542627884, + "loss": 1.302, + "theoretical_loss": 3.5107898952859, + "tokens_seen": 1525874688 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027157472417251754, + "loss": 1.3764, + "theoretical_loss": 3.510776644349327, + "tokens_seen": 1525940224 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002715646940822468, + "loss": 1.2624, + "theoretical_loss": 3.510763394141183, + "tokens_seen": 1526005760 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002715546639919759, + "loss": 1.1849, + "theoretical_loss": 3.5107501446613965, + "tokens_seen": 1526071296 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027154463390170514, + "loss": 1.2804, + "theoretical_loss": 3.5107368959098957, + "tokens_seen": 1526136832 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002715346038114343, + "loss": 1.3182, + "theoretical_loss": 3.5107236478866097, + "tokens_seen": 1526202368 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002715245737211635, + "loss": 1.3493, + "theoretical_loss": 3.510710400591467, + "tokens_seen": 1526267904 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002715145436308927, + "loss": 1.3917, + "theoretical_loss": 3.5106971540243963, + "tokens_seen": 1526333440 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002715045135406219, + "loss": 1.3567, + "theoretical_loss": 3.5106839081853267, + "tokens_seen": 1526398976 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027149448345035104, + "loss": 1.3216, + "theoretical_loss": 3.510670663074187, + "tokens_seen": 1526464512 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002714844533600803, + "loss": 1.2834, + "theoretical_loss": 3.510657418690905, + "tokens_seen": 1526530048 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002714744232698094, + "loss": 1.2366, + "theoretical_loss": 3.5106441750354103, + "tokens_seen": 1526595584 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027146439317953864, + "loss": 1.3216, + "theoretical_loss": 3.5106309321076314, + "tokens_seen": 1526661120 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002714543630892678, + "loss": 1.297, + "theoretical_loss": 3.5106176899074972, + "tokens_seen": 1526726656 + }, + { + "epoch": 0.46, + "learning_rate": 0.000271444332998997, + "loss": 1.3313, + "theoretical_loss": 3.5106044484349366, + "tokens_seen": 1526792192 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002714343029087262, + "loss": 1.3456, + "theoretical_loss": 3.5105912076898784, + "tokens_seen": 1526857728 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027142427281845536, + "loss": 1.2835, + "theoretical_loss": 3.510577967672251, + "tokens_seen": 1526923264 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.0004572533944156021, + "objective/train/docs_used": 866480, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.0657284259796143, + "objective/train/original_loss": 2.0657284259796143, + "objective/train/theoretical_loss": 3.510564728381983, + "objective/train/tokens_used": 1547448800, + "objective/train/value_avg": -0.005748748779296875, + "objective/train/value_loss": 5.226362554822117e-05, + "objective/train/value_max": -0.00037860870361328125, + "objective/train/value_min": -0.06982421875, + "objective/train/value_reward_corr": 0.4361926757282285, + "objective/train/value_std": 0.005340576171875, + "objective/train/weight_avg": 0.9979428648948669, + "objective/train/weighted_lm_loss": 2.0538642406463623, + "objective/train/weights_max": 1.6898201704025269, + "objective/train/weights_min": 0.3930363655090332, + "theoretical_loss": 3.510564728381983, + "tokens_seen": 1526988800 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027141424272818454, + "loss": 1.2075, + "theoretical_loss": 3.510564728381983, + "tokens_seen": 1526988800 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002714042126379138, + "loss": 1.3055, + "theoretical_loss": 3.5105514898190044, + "tokens_seen": 1527054336 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002713941825476429, + "loss": 1.4038, + "theoretical_loss": 3.5105382519832427, + "tokens_seen": 1527119872 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027138415245737214, + "loss": 1.3396, + "theoretical_loss": 3.5105250148746276, + "tokens_seen": 1527185408 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027137412236710127, + "loss": 1.2829, + "theoretical_loss": 3.510511778493088, + "tokens_seen": 1527250944 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002713640922768305, + "loss": 1.3967, + "theoretical_loss": 3.510498542838552, + "tokens_seen": 1527316480 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002713540621865597, + "loss": 1.388, + "theoretical_loss": 3.5104853079109493, + "tokens_seen": 1527382016 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027134403209628887, + "loss": 1.2938, + "theoretical_loss": 3.510472073710208, + "tokens_seen": 1527447552 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027133400200601805, + "loss": 1.2762, + "theoretical_loss": 3.510458840236258, + "tokens_seen": 1527513088 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002713239719157473, + "loss": 1.2982, + "theoretical_loss": 3.510445607489028, + "tokens_seen": 1527578624 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002713139418254764, + "loss": 1.2541, + "theoretical_loss": 3.5104323754684454, + "tokens_seen": 1527644160 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027130391173520564, + "loss": 1.3347, + "theoretical_loss": 3.510419144174441, + "tokens_seen": 1527709696 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027129388164493477, + "loss": 1.3202, + "theoretical_loss": 3.510405913606943, + "tokens_seen": 1527775232 + }, + { + "epoch": 0.46, + "learning_rate": 0.000271283851554664, + "loss": 1.4128, + "theoretical_loss": 3.51039268376588, + "tokens_seen": 1527840768 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002712738214643932, + "loss": 1.2845, + "theoretical_loss": 3.510379454651182, + "tokens_seen": 1527906304 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027126379137412237, + "loss": 1.2332, + "theoretical_loss": 3.510366226262777, + "tokens_seen": 1527971840 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027125376128385155, + "loss": 1.4362, + "theoretical_loss": 3.510352998600594, + "tokens_seen": 1528037376 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027124373119358073, + "loss": 1.2741, + "theoretical_loss": 3.5103397716645626, + "tokens_seen": 1528102912 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002712337011033099, + "loss": 1.2657, + "theoretical_loss": 3.510326545454611, + "tokens_seen": 1528168448 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027122367101303915, + "loss": 1.1893, + "theoretical_loss": 3.510313319970669, + "tokens_seen": 1528233984 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002712136409227683, + "loss": 1.216, + "theoretical_loss": 3.5103000952126653, + "tokens_seen": 1528299520 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002712036108324975, + "loss": 1.4158, + "theoretical_loss": 3.510286871180529, + "tokens_seen": 1528365056 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002711935807422267, + "loss": 1.3222, + "theoretical_loss": 3.5102736478741887, + "tokens_seen": 1528430592 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027118355065195587, + "loss": 1.2732, + "theoretical_loss": 3.5102604252935743, + "tokens_seen": 1528496128 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027117352056168505, + "loss": 1.3373, + "theoretical_loss": 3.510247203438614, + "tokens_seen": 1528561664 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": 0.0002188266080338508, + "objective/train/docs_used": 867112, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.746248483657837, + "objective/train/original_loss": 2.746248483657837, + "objective/train/theoretical_loss": 3.510233982309237, + "objective/train/tokens_used": 1549087200, + "objective/train/value_avg": -0.00534820556640625, + "objective/train/value_loss": 4.273887680028565e-05, + "objective/train/value_max": -0.00025725364685058594, + "objective/train/value_min": -0.0635986328125, + "objective/train/value_reward_corr": 0.4511663798761907, + "objective/train/value_std": 0.004180908203125, + "objective/train/weight_avg": 1.0042659044265747, + "objective/train/weighted_lm_loss": 2.756417751312256, + "objective/train/weights_max": 1.658612608909607, + "objective/train/weights_min": 0.5772180557250977, + "theoretical_loss": 3.510233982309237, + "tokens_seen": 1528627200 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027116349047141423, + "loss": 1.3898, + "theoretical_loss": 3.510233982309237, + "tokens_seen": 1528627200 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002711534603811434, + "loss": 1.3123, + "theoretical_loss": 3.510220761905373, + "tokens_seen": 1528692736 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027114343029087265, + "loss": 1.3785, + "theoretical_loss": 3.5102075422269507, + "tokens_seen": 1528758272 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002711334002006018, + "loss": 1.2374, + "theoretical_loss": 3.510194323273899, + "tokens_seen": 1528823808 + }, + { + "epoch": 0.46, + "learning_rate": 0.000271123370110331, + "loss": 1.2426, + "theoretical_loss": 3.510181105046147, + "tokens_seen": 1528889344 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027111334002006014, + "loss": 1.3335, + "theoretical_loss": 3.5101678875436244, + "tokens_seen": 1528954880 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002711033099297894, + "loss": 1.3294, + "theoretical_loss": 3.51015467076626, + "tokens_seen": 1529020416 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027109327983951856, + "loss": 1.2368, + "theoretical_loss": 3.5101414547139824, + "tokens_seen": 1529085952 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027108324974924774, + "loss": 1.2929, + "theoretical_loss": 3.5101282393867215, + "tokens_seen": 1529151488 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002710732196589769, + "loss": 1.3103, + "theoretical_loss": 3.510115024784406, + "tokens_seen": 1529217024 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002710631895687061, + "loss": 1.3235, + "theoretical_loss": 3.5101018109069653, + "tokens_seen": 1529282560 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002710531594784353, + "loss": 1.2294, + "theoretical_loss": 3.5100885977543284, + "tokens_seen": 1529348096 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002710431293881645, + "loss": 1.2685, + "theoretical_loss": 3.510075385326425, + "tokens_seen": 1529413632 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027103309929789364, + "loss": 1.3318, + "theoretical_loss": 3.510062173623184, + "tokens_seen": 1529479168 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002710230692076229, + "loss": 1.2643, + "theoretical_loss": 3.5100489626445337, + "tokens_seen": 1529544704 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002710130391173521, + "loss": 1.334, + "theoretical_loss": 3.5100357523904044, + "tokens_seen": 1529610240 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027100300902708124, + "loss": 1.2801, + "theoretical_loss": 3.5100225428607255, + "tokens_seen": 1529675776 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002709929789368105, + "loss": 1.2521, + "theoretical_loss": 3.5100093340554253, + "tokens_seen": 1529741312 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002709829488465396, + "loss": 1.3841, + "theoretical_loss": 3.5099961259744337, + "tokens_seen": 1529806848 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027097291875626884, + "loss": 1.1865, + "theoretical_loss": 3.5099829186176796, + "tokens_seen": 1529872384 + }, + { + "epoch": 0.46, + "learning_rate": 0.000270962888665998, + "loss": 1.3137, + "theoretical_loss": 3.509969711985093, + "tokens_seen": 1529937920 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002709528585757272, + "loss": 1.25, + "theoretical_loss": 3.509956506076602, + "tokens_seen": 1530003456 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002709428284854564, + "loss": 1.3123, + "theoretical_loss": 3.509943300892137, + "tokens_seen": 1530068992 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027093279839518556, + "loss": 1.3499, + "theoretical_loss": 3.5099300964316265, + "tokens_seen": 1530134528 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027092276830491474, + "loss": 1.4434, + "theoretical_loss": 3.509916892695, + "tokens_seen": 1530200064 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": 0.0003841529251076281, + "objective/train/docs_used": 868323, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.915659189224243, + "objective/train/original_loss": 2.915658712387085, + "objective/train/theoretical_loss": 3.5099036896821874, + "objective/train/tokens_used": 1550725600, + "objective/train/value_avg": -0.006053924560546875, + "objective/train/value_loss": 6.81367891957052e-05, + "objective/train/value_max": -0.00046372413635253906, + "objective/train/value_min": -0.11126708984375, + "objective/train/value_reward_corr": 0.4394519331576489, + "objective/train/value_std": 0.005279541015625, + "objective/train/weight_avg": 1.0069602727890015, + "objective/train/weighted_lm_loss": 2.9389595985412598, + "objective/train/weights_max": 2.4154887199401855, + "objective/train/weights_min": 0.2934008836746216, + "theoretical_loss": 3.5099036896821874, + "tokens_seen": 1530265600 + }, + { + "epoch": 0.46, + "learning_rate": 0.000270912738214644, + "loss": 1.4689, + "theoretical_loss": 3.5099036896821874, + "tokens_seen": 1530265600 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002709027081243731, + "loss": 1.1639, + "theoretical_loss": 3.509890487393117, + "tokens_seen": 1530331136 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027089267803410234, + "loss": 1.2814, + "theoretical_loss": 3.509877285827719, + "tokens_seen": 1530396672 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027088264794383147, + "loss": 1.3536, + "theoretical_loss": 3.5098640849859226, + "tokens_seen": 1530462208 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002708726178535607, + "loss": 1.2973, + "theoretical_loss": 3.5098508848676566, + "tokens_seen": 1530527744 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002708625877632899, + "loss": 1.3038, + "theoretical_loss": 3.509837685472851, + "tokens_seen": 1530593280 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027085255767301907, + "loss": 1.4607, + "theoretical_loss": 3.5098244868014348, + "tokens_seen": 1530658816 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027084252758274825, + "loss": 1.2843, + "theoretical_loss": 3.5098112888533373, + "tokens_seen": 1530724352 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002708324974924775, + "loss": 1.3815, + "theoretical_loss": 3.5097980916284883, + "tokens_seen": 1530789888 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002708224674022066, + "loss": 1.2492, + "theoretical_loss": 3.509784895126817, + "tokens_seen": 1530855424 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027081243731193584, + "loss": 1.2205, + "theoretical_loss": 3.509771699348253, + "tokens_seen": 1530920960 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027080240722166497, + "loss": 1.3142, + "theoretical_loss": 3.5097585042927255, + "tokens_seen": 1530986496 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002707923771313942, + "loss": 1.2882, + "theoretical_loss": 3.509745309960164, + "tokens_seen": 1531052032 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002707823470411234, + "loss": 1.1937, + "theoretical_loss": 3.5097321163504978, + "tokens_seen": 1531117568 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027077231695085257, + "loss": 1.3434, + "theoretical_loss": 3.509718923463656, + "tokens_seen": 1531183104 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027076228686058175, + "loss": 1.3181, + "theoretical_loss": 3.5097057312995696, + "tokens_seen": 1531248640 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027075225677031093, + "loss": 1.3002, + "theoretical_loss": 3.5096925398581664, + "tokens_seen": 1531314176 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002707422266800401, + "loss": 1.2454, + "theoretical_loss": 3.5096793491393763, + "tokens_seen": 1531379712 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027073219658976935, + "loss": 1.3471, + "theoretical_loss": 3.5096661591431295, + "tokens_seen": 1531445248 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002707221664994985, + "loss": 1.4181, + "theoretical_loss": 3.5096529698693546, + "tokens_seen": 1531510784 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002707121364092277, + "loss": 1.2688, + "theoretical_loss": 3.509639781317982, + "tokens_seen": 1531576320 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002707021063189569, + "loss": 1.3906, + "theoretical_loss": 3.5096265934889406, + "tokens_seen": 1531641856 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027069207622868607, + "loss": 1.3701, + "theoretical_loss": 3.50961340638216, + "tokens_seen": 1531707392 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027068204613841525, + "loss": 1.327, + "theoretical_loss": 3.5096002199975693, + "tokens_seen": 1531772928 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027067201604814443, + "loss": 1.3087, + "theoretical_loss": 3.509587034335099, + "tokens_seen": 1531838464 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": -0.0011475011706352234, + "objective/train/docs_used": 869121, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5183069705963135, + "objective/train/original_loss": 2.5183064937591553, + "objective/train/theoretical_loss": 3.5095738493946786, + "objective/train/tokens_used": 1552364000, + "objective/train/value_avg": -0.005794525146484375, + "objective/train/value_loss": 6.880563887534663e-05, + "objective/train/value_max": -0.0002359151840209961, + "objective/train/value_min": -0.07354736328125, + "objective/train/value_reward_corr": 0.4320824044062958, + "objective/train/value_std": 0.004817962646484375, + "objective/train/weight_avg": 0.9917244911193848, + "objective/train/weighted_lm_loss": 2.5016074180603027, + "objective/train/weights_max": 1.7802315950393677, + "objective/train/weights_min": 0.32232266664505005, + "theoretical_loss": 3.5095738493946786, + "tokens_seen": 1531904000 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002706619859578736, + "loss": 1.3822, + "theoretical_loss": 3.5095738493946786, + "tokens_seen": 1531904000 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027065195586760285, + "loss": 1.2509, + "theoretical_loss": 3.5095606651762368, + "tokens_seen": 1531969536 + }, + { + "epoch": 0.46, + "learning_rate": 0.000270641925777332, + "loss": 1.3471, + "theoretical_loss": 3.509547481679704, + "tokens_seen": 1532035072 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002706318956870612, + "loss": 1.2471, + "theoretical_loss": 3.509534298905009, + "tokens_seen": 1532100608 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027062186559679034, + "loss": 1.3432, + "theoretical_loss": 3.5095211168520826, + "tokens_seen": 1532166144 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002706118355065196, + "loss": 1.367, + "theoretical_loss": 3.509507935520853, + "tokens_seen": 1532231680 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027060180541624876, + "loss": 1.3408, + "theoretical_loss": 3.509494754911251, + "tokens_seen": 1532297216 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027059177532597794, + "loss": 1.2585, + "theoretical_loss": 3.509481575023206, + "tokens_seen": 1532362752 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002705817452357071, + "loss": 1.221, + "theoretical_loss": 3.509468395856647, + "tokens_seen": 1532428288 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002705717151454363, + "loss": 1.3905, + "theoretical_loss": 3.509455217411504, + "tokens_seen": 1532493824 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002705616850551655, + "loss": 1.2399, + "theoretical_loss": 3.5094420396877064, + "tokens_seen": 1532559360 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002705516549648947, + "loss": 1.3304, + "theoretical_loss": 3.509428862685185, + "tokens_seen": 1532624896 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027054162487462384, + "loss": 1.266, + "theoretical_loss": 3.509415686403868, + "tokens_seen": 1532690432 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002705315947843531, + "loss": 1.3613, + "theoretical_loss": 3.509402510843686, + "tokens_seen": 1532755968 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027052156469408226, + "loss": 1.3523, + "theoretical_loss": 3.5093893360045687, + "tokens_seen": 1532821504 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027051153460381144, + "loss": 1.2827, + "theoretical_loss": 3.509376161886445, + "tokens_seen": 1532887040 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002705015045135406, + "loss": 1.3484, + "theoretical_loss": 3.5093629884892454, + "tokens_seen": 1532952576 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002704914744232698, + "loss": 1.3326, + "theoretical_loss": 3.5093498158128997, + "tokens_seen": 1533018112 + }, + { + "epoch": 0.46, + "learning_rate": 0.000270481444332999, + "loss": 1.3172, + "theoretical_loss": 3.5093366438573375, + "tokens_seen": 1533083648 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002704714142427282, + "loss": 1.2919, + "theoretical_loss": 3.509323472622488, + "tokens_seen": 1533149184 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027046138415245735, + "loss": 1.4006, + "theoretical_loss": 3.5093103021082817, + "tokens_seen": 1533214720 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002704513540621866, + "loss": 1.3867, + "theoretical_loss": 3.5092971323146473, + "tokens_seen": 1533280256 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002704413239719157, + "loss": 1.3097, + "theoretical_loss": 3.509283963241516, + "tokens_seen": 1533345792 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027043129388164494, + "loss": 1.4028, + "theoretical_loss": 3.509270794888817, + "tokens_seen": 1533411328 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002704212637913741, + "loss": 1.3939, + "theoretical_loss": 3.5092576272564795, + "tokens_seen": 1533476864 + }, + { + "epoch": 0.46, + "objective/train/advantage_avg": 0.0005004322738386691, + "objective/train/docs_used": 869825, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.2789528369903564, + "objective/train/original_loss": 2.2789530754089355, + "objective/train/theoretical_loss": 3.5092444603444344, + "objective/train/tokens_used": 1554002400, + "objective/train/value_avg": -0.004962921142578125, + "objective/train/value_loss": 5.937218156759627e-05, + "objective/train/value_max": -0.00029587745666503906, + "objective/train/value_min": -0.080078125, + "objective/train/value_reward_corr": 0.3234044444530334, + "objective/train/value_std": 0.0037326812744140625, + "objective/train/weight_avg": 1.0076789855957031, + "objective/train/weighted_lm_loss": 2.2873117923736572, + "objective/train/weights_max": 2.227280378341675, + "objective/train/weights_min": 0.34799429774284363, + "theoretical_loss": 3.5092444603444344, + "tokens_seen": 1533542400 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002704112337011033, + "loss": 1.3354, + "theoretical_loss": 3.5092444603444344, + "tokens_seen": 1533542400 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002704012036108325, + "loss": 1.2456, + "theoretical_loss": 3.5092312941526105, + "tokens_seen": 1533607936 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027039117352056167, + "loss": 1.208, + "theoretical_loss": 3.509218128680938, + "tokens_seen": 1533673472 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027038114343029085, + "loss": 1.3519, + "theoretical_loss": 3.509204963929347, + "tokens_seen": 1533739008 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002703711133400201, + "loss": 1.3939, + "theoretical_loss": 3.509191799897767, + "tokens_seen": 1533804544 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002703610832497492, + "loss": 1.4263, + "theoretical_loss": 3.5091786365861286, + "tokens_seen": 1533870080 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027035105315947845, + "loss": 1.3693, + "theoretical_loss": 3.5091654739943605, + "tokens_seen": 1533935616 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027034102306920763, + "loss": 1.3417, + "theoretical_loss": 3.5091523121223935, + "tokens_seen": 1534001152 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002703309929789368, + "loss": 1.2513, + "theoretical_loss": 3.509139150970157, + "tokens_seen": 1534066688 + }, + { + "epoch": 0.46, + "learning_rate": 0.000270320962888666, + "loss": 1.2956, + "theoretical_loss": 3.509125990537581, + "tokens_seen": 1534132224 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027031093279839517, + "loss": 1.373, + "theoretical_loss": 3.509112830824596, + "tokens_seen": 1534197760 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027030090270812435, + "loss": 1.2695, + "theoretical_loss": 3.5090996718311307, + "tokens_seen": 1534263296 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002702908726178536, + "loss": 1.2694, + "theoretical_loss": 3.509086513557116, + "tokens_seen": 1534328832 + }, + { + "epoch": 0.46, + "learning_rate": 0.0002702808425275827, + "loss": 1.2526, + "theoretical_loss": 3.5090733560024816, + "tokens_seen": 1534394368 + }, + { + "epoch": 0.46, + "learning_rate": 0.00027027081243731195, + "loss": 1.3199, + "theoretical_loss": 3.5090601991671573, + "tokens_seen": 1534459904 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027026078234704113, + "loss": 1.3182, + "theoretical_loss": 3.5090470430510727, + "tokens_seen": 1534525440 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002702507522567703, + "loss": 1.2657, + "theoretical_loss": 3.5090338876541587, + "tokens_seen": 1534590976 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027024072216649955, + "loss": 1.2714, + "theoretical_loss": 3.509020732976345, + "tokens_seen": 1534656512 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002702306920762287, + "loss": 1.3303, + "theoretical_loss": 3.509007579017561, + "tokens_seen": 1534722048 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002702206619859579, + "loss": 1.3765, + "theoretical_loss": 3.5089944257777375, + "tokens_seen": 1534787584 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002702106318956871, + "loss": 1.3649, + "theoretical_loss": 3.5089812732568038, + "tokens_seen": 1534853120 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027020060180541627, + "loss": 1.3239, + "theoretical_loss": 3.50896812145469, + "tokens_seen": 1534918656 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027019057171514545, + "loss": 1.346, + "theoretical_loss": 3.5089549703713265, + "tokens_seen": 1534984192 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027018054162487463, + "loss": 1.3454, + "theoretical_loss": 3.508941820006643, + "tokens_seen": 1535049728 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002701705115346038, + "loss": 1.2965, + "theoretical_loss": 3.5089286703605698, + "tokens_seen": 1535115264 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.0013044662773609161, + "objective/train/docs_used": 871111, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.1680846214294434, + "objective/train/original_loss": 3.1680848598480225, + "objective/train/theoretical_loss": 3.508915521433037, + "objective/train/tokens_used": 1555640800, + "objective/train/value_avg": -0.004573822021484375, + "objective/train/value_loss": 8.783876546658576e-05, + "objective/train/value_max": -0.0004322528839111328, + "objective/train/value_min": -0.0546875, + "objective/train/value_reward_corr": 0.37996012401907886, + "objective/train/value_std": 0.0036716461181640625, + "objective/train/weight_avg": 0.9909074306488037, + "objective/train/weighted_lm_loss": 3.1673743724823, + "objective/train/weights_max": 1.3588981628417969, + "objective/train/weights_min": 0.29873722791671753, + "theoretical_loss": 3.508915521433037, + "tokens_seen": 1535180800 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027016048144433305, + "loss": 1.3696, + "theoretical_loss": 3.508915521433037, + "tokens_seen": 1535180800 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002701504513540622, + "loss": 1.3018, + "theoretical_loss": 3.508902373223975, + "tokens_seen": 1535246336 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002701404212637914, + "loss": 1.267, + "theoretical_loss": 3.508889225733313, + "tokens_seen": 1535311872 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027013039117352054, + "loss": 1.2528, + "theoretical_loss": 3.5088760789609807, + "tokens_seen": 1535377408 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002701203610832498, + "loss": 1.343, + "theoretical_loss": 3.5088629329069096, + "tokens_seen": 1535442944 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027011033099297896, + "loss": 1.2805, + "theoretical_loss": 3.5088497875710294, + "tokens_seen": 1535508480 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027010030090270814, + "loss": 1.3106, + "theoretical_loss": 3.50883664295327, + "tokens_seen": 1535574016 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002700902708124373, + "loss": 1.2528, + "theoretical_loss": 3.5088234990535616, + "tokens_seen": 1535639552 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002700802407221665, + "loss": 1.3301, + "theoretical_loss": 3.5088103558718338, + "tokens_seen": 1535705088 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002700702106318957, + "loss": 1.3252, + "theoretical_loss": 3.5087972134080174, + "tokens_seen": 1535770624 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002700601805416249, + "loss": 1.3485, + "theoretical_loss": 3.5087840716620424, + "tokens_seen": 1535836160 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027005015045135404, + "loss": 1.3546, + "theoretical_loss": 3.508770930633839, + "tokens_seen": 1535901696 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002700401203610833, + "loss": 1.3631, + "theoretical_loss": 3.508757790323337, + "tokens_seen": 1535967232 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027003009027081246, + "loss": 1.209, + "theoretical_loss": 3.508744650730467, + "tokens_seen": 1536032768 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027002006018054164, + "loss": 1.3912, + "theoretical_loss": 3.508731511855159, + "tokens_seen": 1536098304 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002700100300902708, + "loss": 1.2468, + "theoretical_loss": 3.5087183736973437, + "tokens_seen": 1536163840 + }, + { + "epoch": 0.47, + "learning_rate": 0.00027, + "loss": 1.2694, + "theoretical_loss": 3.508705236256951, + "tokens_seen": 1536229376 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002699899699097292, + "loss": 1.2921, + "theoretical_loss": 3.50869209953391, + "tokens_seen": 1536294912 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002699799398194584, + "loss": 1.285, + "theoretical_loss": 3.5086789635281526, + "tokens_seen": 1536360448 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026996990972918755, + "loss": 1.2686, + "theoretical_loss": 3.508665828239608, + "tokens_seen": 1536425984 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002699598796389168, + "loss": 1.2293, + "theoretical_loss": 3.508652693668207, + "tokens_seen": 1536491520 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002699498495486459, + "loss": 1.2914, + "theoretical_loss": 3.50863955981388, + "tokens_seen": 1536557056 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026993981945837514, + "loss": 1.2876, + "theoretical_loss": 3.5086264266765568, + "tokens_seen": 1536622592 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002699297893681043, + "loss": 1.3158, + "theoretical_loss": 3.5086132942561674, + "tokens_seen": 1536688128 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002699197592778335, + "loss": 1.3635, + "theoretical_loss": 3.508600162552643, + "tokens_seen": 1536753664 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": 0.00038945776759646833, + "objective/train/docs_used": 871860, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9068500995635986, + "objective/train/original_loss": 2.9068500995635986, + "objective/train/theoretical_loss": 3.5085870315659133, + "objective/train/tokens_used": 1557279200, + "objective/train/value_avg": -0.00464630126953125, + "objective/train/value_loss": 3.211842704331502e-05, + "objective/train/value_max": -0.00035834312438964844, + "objective/train/value_min": -0.0635986328125, + "objective/train/value_reward_corr": 0.32337005279443376, + "objective/train/value_std": 0.0033473968505859375, + "objective/train/weight_avg": 1.0054727792739868, + "objective/train/weighted_lm_loss": 2.9237821102142334, + "objective/train/weights_max": 1.7726831436157227, + "objective/train/weights_min": 0.43152916431427, + "theoretical_loss": 3.5085870315659133, + "tokens_seen": 1536819200 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002699097291875627, + "loss": 1.3152, + "theoretical_loss": 3.5085870315659133, + "tokens_seen": 1536819200 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026989969909729187, + "loss": 1.3925, + "theoretical_loss": 3.508573901295909, + "tokens_seen": 1536884736 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026988966900702105, + "loss": 1.327, + "theoretical_loss": 3.5085607717425598, + "tokens_seen": 1536950272 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002698796389167503, + "loss": 1.2817, + "theoretical_loss": 3.5085476429057962, + "tokens_seen": 1537015808 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002698696088264794, + "loss": 1.3046, + "theoretical_loss": 3.5085345147855493, + "tokens_seen": 1537081344 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026985957873620865, + "loss": 1.3307, + "theoretical_loss": 3.508521387381748, + "tokens_seen": 1537146880 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026984954864593783, + "loss": 1.2934, + "theoretical_loss": 3.5085082606943243, + "tokens_seen": 1537212416 + }, + { + "epoch": 0.47, + "learning_rate": 0.000269839518555667, + "loss": 1.3826, + "theoretical_loss": 3.5084951347232076, + "tokens_seen": 1537277952 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002698294884653962, + "loss": 1.3214, + "theoretical_loss": 3.5084820094683282, + "tokens_seen": 1537343488 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026981945837512537, + "loss": 1.3033, + "theoretical_loss": 3.5084688849296173, + "tokens_seen": 1537409024 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026980942828485455, + "loss": 1.2809, + "theoretical_loss": 3.508455761107004, + "tokens_seen": 1537474560 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002697993981945838, + "loss": 1.3558, + "theoretical_loss": 3.50844263800042, + "tokens_seen": 1537540096 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002697893681043129, + "loss": 1.3473, + "theoretical_loss": 3.5084295156097953, + "tokens_seen": 1537605632 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026977933801404215, + "loss": 1.313, + "theoretical_loss": 3.50841639393506, + "tokens_seen": 1537671168 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002697693079237713, + "loss": 1.3857, + "theoretical_loss": 3.5084032729761443, + "tokens_seen": 1537736704 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002697592778335005, + "loss": 1.2979, + "theoretical_loss": 3.5083901527329795, + "tokens_seen": 1537802240 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002697492477432297, + "loss": 1.3342, + "theoretical_loss": 3.508377033205496, + "tokens_seen": 1537867776 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002697392176529589, + "loss": 1.235, + "theoretical_loss": 3.508363914393623, + "tokens_seen": 1537933312 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026972918756268806, + "loss": 1.3124, + "theoretical_loss": 3.508350796297293, + "tokens_seen": 1537998848 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002697191574724173, + "loss": 1.3013, + "theoretical_loss": 3.508337678916434, + "tokens_seen": 1538064384 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002697091273821464, + "loss": 1.3437, + "theoretical_loss": 3.508324562250979, + "tokens_seen": 1538129920 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026969909729187565, + "loss": 1.3154, + "theoretical_loss": 3.508311446300857, + "tokens_seen": 1538195456 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002696890672016048, + "loss": 1.3475, + "theoretical_loss": 3.508298331065999, + "tokens_seen": 1538260992 + }, + { + "epoch": 0.47, + "learning_rate": 0.000269679037111334, + "loss": 1.348, + "theoretical_loss": 3.5082852165463354, + "tokens_seen": 1538326528 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002696690070210632, + "loss": 1.373, + "theoretical_loss": 3.5082721027417962, + "tokens_seen": 1538392064 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.00014305856893770397, + "objective/train/docs_used": 873345, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.513871908187866, + "objective/train/original_loss": 2.5138721466064453, + "objective/train/theoretical_loss": 3.508258989652313, + "objective/train/tokens_used": 1558917600, + "objective/train/value_avg": -0.00482940673828125, + "objective/train/value_loss": 5.308675463311374e-05, + "objective/train/value_max": -0.00015723705291748047, + "objective/train/value_min": -0.1842041015625, + "objective/train/value_reward_corr": 0.309205179696507, + "objective/train/value_std": 0.004352569580078125, + "objective/train/weight_avg": 1.0011249780654907, + "objective/train/weighted_lm_loss": 2.518028974533081, + "objective/train/weights_max": 5.076640605926514, + "objective/train/weights_min": 0.19383171200752258, + "theoretical_loss": 3.508258989652313, + "tokens_seen": 1538457600 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002696589769307924, + "loss": 1.3321, + "theoretical_loss": 3.508258989652313, + "tokens_seen": 1538457600 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026964894684052156, + "loss": 1.2605, + "theoretical_loss": 3.508245877277816, + "tokens_seen": 1538523136 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026963891675025074, + "loss": 1.4449, + "theoretical_loss": 3.508232765618235, + "tokens_seen": 1538588672 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002696288866599799, + "loss": 1.3375, + "theoretical_loss": 3.5082196546735016, + "tokens_seen": 1538654208 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026961885656970916, + "loss": 1.3854, + "theoretical_loss": 3.5082065444435457, + "tokens_seen": 1538719744 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002696088264794383, + "loss": 1.3993, + "theoretical_loss": 3.508193434928298, + "tokens_seen": 1538785280 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002695987963891675, + "loss": 1.3634, + "theoretical_loss": 3.5081803261276896, + "tokens_seen": 1538850816 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026958876629889665, + "loss": 1.3175, + "theoretical_loss": 3.5081672180416503, + "tokens_seen": 1538916352 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002695787362086259, + "loss": 1.3881, + "theoretical_loss": 3.5081541106701115, + "tokens_seen": 1538981888 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026956870611835506, + "loss": 1.3721, + "theoretical_loss": 3.5081410040130034, + "tokens_seen": 1539047424 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026955867602808424, + "loss": 1.3815, + "theoretical_loss": 3.508127898070257, + "tokens_seen": 1539112960 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002695486459378134, + "loss": 1.302, + "theoretical_loss": 3.5081147928418024, + "tokens_seen": 1539178496 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026953861584754266, + "loss": 1.29, + "theoretical_loss": 3.508101688327571, + "tokens_seen": 1539244032 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002695285857572718, + "loss": 1.258, + "theoretical_loss": 3.508088584527492, + "tokens_seen": 1539309568 + }, + { + "epoch": 0.47, + "learning_rate": 0.000269518555667001, + "loss": 1.2289, + "theoretical_loss": 3.508075481441498, + "tokens_seen": 1539375104 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002695085255767302, + "loss": 1.4421, + "theoretical_loss": 3.5080623790695182, + "tokens_seen": 1539440640 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002694984954864594, + "loss": 1.4607, + "theoretical_loss": 3.508049277411484, + "tokens_seen": 1539506176 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002694884653961886, + "loss": 1.3857, + "theoretical_loss": 3.5080361764673267, + "tokens_seen": 1539571712 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026947843530591775, + "loss": 1.3449, + "theoretical_loss": 3.5080230762369755, + "tokens_seen": 1539637248 + }, + { + "epoch": 0.47, + "learning_rate": 0.000269468405215647, + "loss": 1.2771, + "theoretical_loss": 3.508009976720362, + "tokens_seen": 1539702784 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002694583751253761, + "loss": 1.3388, + "theoretical_loss": 3.507996877917417, + "tokens_seen": 1539768320 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026944834503510534, + "loss": 1.3462, + "theoretical_loss": 3.507983779828071, + "tokens_seen": 1539833856 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002694383149448345, + "loss": 1.3005, + "theoretical_loss": 3.507970682452255, + "tokens_seen": 1539899392 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002694282848545637, + "loss": 1.2107, + "theoretical_loss": 3.5079575857898995, + "tokens_seen": 1539964928 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002694182547642929, + "loss": 1.3531, + "theoretical_loss": 3.507944489840936, + "tokens_seen": 1540030464 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.0006937590078450739, + "objective/train/docs_used": 874032, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.671658515930176, + "objective/train/original_loss": 2.6716582775115967, + "objective/train/theoretical_loss": 3.507931394605294, + "objective/train/tokens_used": 1560556000, + "objective/train/value_avg": -0.00458526611328125, + "objective/train/value_loss": 4.3314550566719845e-05, + "objective/train/value_max": -0.0003199577331542969, + "objective/train/value_min": -0.061981201171875, + "objective/train/value_reward_corr": 0.36451801971196945, + "objective/train/value_std": 0.0037174224853515625, + "objective/train/weight_avg": 0.995152473449707, + "objective/train/weighted_lm_loss": 2.659318685531616, + "objective/train/weights_max": 1.4671193361282349, + "objective/train/weights_min": 0.4976806342601776, + "theoretical_loss": 3.507931394605294, + "tokens_seen": 1540096000 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026940822467402207, + "loss": 1.3262, + "theoretical_loss": 3.507931394605294, + "tokens_seen": 1540096000 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026939819458375125, + "loss": 1.4082, + "theoretical_loss": 3.5079183000829053, + "tokens_seen": 1540161536 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002693881644934805, + "loss": 1.2679, + "theoretical_loss": 3.5079052062737004, + "tokens_seen": 1540227072 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002693781344032096, + "loss": 1.3969, + "theoretical_loss": 3.5078921131776095, + "tokens_seen": 1540292608 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026936810431293885, + "loss": 1.3611, + "theoretical_loss": 3.5078790207945647, + "tokens_seen": 1540358144 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026935807422266803, + "loss": 1.3923, + "theoretical_loss": 3.5078659291244962, + "tokens_seen": 1540423680 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002693480441323972, + "loss": 1.2867, + "theoretical_loss": 3.5078528381673344, + "tokens_seen": 1540489216 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002693380140421264, + "loss": 1.3379, + "theoretical_loss": 3.507839747923011, + "tokens_seen": 1540554752 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026932798395185557, + "loss": 1.2852, + "theoretical_loss": 3.5078266583914557, + "tokens_seen": 1540620288 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026931795386158475, + "loss": 1.2853, + "theoretical_loss": 3.5078135695726007, + "tokens_seen": 1540685824 + }, + { + "epoch": 0.47, + "learning_rate": 0.000269307923771314, + "loss": 1.3606, + "theoretical_loss": 3.507800481466376, + "tokens_seen": 1540751360 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002692978936810431, + "loss": 1.3387, + "theoretical_loss": 3.507787394072713, + "tokens_seen": 1540816896 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026928786359077235, + "loss": 1.17, + "theoretical_loss": 3.507774307391542, + "tokens_seen": 1540882432 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002692778335005015, + "loss": 1.2792, + "theoretical_loss": 3.507761221422795, + "tokens_seen": 1540947968 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002692678034102307, + "loss": 1.2298, + "theoretical_loss": 3.5077481361664016, + "tokens_seen": 1541013504 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002692577733199599, + "loss": 1.2956, + "theoretical_loss": 3.5077350516222934, + "tokens_seen": 1541079040 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002692477432296891, + "loss": 1.2296, + "theoretical_loss": 3.507721967790401, + "tokens_seen": 1541144576 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026923771313941826, + "loss": 1.3739, + "theoretical_loss": 3.507708884670656, + "tokens_seen": 1541210112 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002692276830491475, + "loss": 1.2807, + "theoretical_loss": 3.5076958022629885, + "tokens_seen": 1541275648 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002692176529588766, + "loss": 1.2815, + "theoretical_loss": 3.5076827205673307, + "tokens_seen": 1541341184 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026920762286860585, + "loss": 1.2972, + "theoretical_loss": 3.507669639583612, + "tokens_seen": 1541406720 + }, + { + "epoch": 0.47, + "learning_rate": 0.000269197592778335, + "loss": 1.3486, + "theoretical_loss": 3.507656559311765, + "tokens_seen": 1541472256 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002691875626880642, + "loss": 1.2561, + "theoretical_loss": 3.507643479751719, + "tokens_seen": 1541537792 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002691775325977934, + "loss": 1.3522, + "theoretical_loss": 3.5076304009034067, + "tokens_seen": 1541603328 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002691675025075226, + "loss": 1.3005, + "theoretical_loss": 3.507617322766758, + "tokens_seen": 1541668864 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -5.598132020168123e-07, + "objective/train/docs_used": 874753, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.397055149078369, + "objective/train/original_loss": 2.397055149078369, + "objective/train/theoretical_loss": 3.5076042453417045, + "objective/train/tokens_used": 1562194400, + "objective/train/value_avg": -0.00402069091796875, + "objective/train/value_loss": 5.050000618211925e-05, + "objective/train/value_max": -0.00015234947204589844, + "objective/train/value_min": -0.05291748046875, + "objective/train/value_reward_corr": 0.21485456256157967, + "objective/train/value_std": 0.0029888153076171875, + "objective/train/weight_avg": 1.002299427986145, + "objective/train/weighted_lm_loss": 2.410358190536499, + "objective/train/weights_max": 1.6618324518203735, + "objective/train/weights_min": 0.2922782003879547, + "theoretical_loss": 3.5076042453417045, + "tokens_seen": 1541734400 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026915747241725176, + "loss": 1.2764, + "theoretical_loss": 3.5076042453417045, + "tokens_seen": 1541734400 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026914744232698094, + "loss": 1.2713, + "theoretical_loss": 3.507591168628177, + "tokens_seen": 1541799936 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002691374122367101, + "loss": 1.3031, + "theoretical_loss": 3.507578092626106, + "tokens_seen": 1541865472 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026912738214643936, + "loss": 1.3352, + "theoretical_loss": 3.5075650173354234, + "tokens_seen": 1541931008 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002691173520561685, + "loss": 1.2851, + "theoretical_loss": 3.5075519427560597, + "tokens_seen": 1541996544 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002691073219658977, + "loss": 1.2924, + "theoretical_loss": 3.5075388688879467, + "tokens_seen": 1542062080 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026909729187562685, + "loss": 1.4422, + "theoretical_loss": 3.5075257957310146, + "tokens_seen": 1542127616 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002690872617853561, + "loss": 1.307, + "theoretical_loss": 3.507512723285195, + "tokens_seen": 1542193152 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026907723169508526, + "loss": 1.28, + "theoretical_loss": 3.507499651550419, + "tokens_seen": 1542258688 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026906720160481444, + "loss": 1.3875, + "theoretical_loss": 3.507486580526617, + "tokens_seen": 1542324224 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002690571715145436, + "loss": 1.28, + "theoretical_loss": 3.5074735102137216, + "tokens_seen": 1542389760 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026904714142427286, + "loss": 1.3105, + "theoretical_loss": 3.5074604406116627, + "tokens_seen": 1542455296 + }, + { + "epoch": 0.47, + "learning_rate": 0.000269037111334002, + "loss": 1.362, + "theoretical_loss": 3.507447371720372, + "tokens_seen": 1542520832 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002690270812437312, + "loss": 1.3195, + "theoretical_loss": 3.50743430353978, + "tokens_seen": 1542586368 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026901705115346035, + "loss": 1.3702, + "theoretical_loss": 3.5074212360698187, + "tokens_seen": 1542651904 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002690070210631896, + "loss": 1.2503, + "theoretical_loss": 3.507408169310419, + "tokens_seen": 1542717440 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026899699097291877, + "loss": 1.3166, + "theoretical_loss": 3.507395103261511, + "tokens_seen": 1542782976 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026898696088264795, + "loss": 1.3413, + "theoretical_loss": 3.507382037923028, + "tokens_seen": 1542848512 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026897693079237713, + "loss": 1.3703, + "theoretical_loss": 3.5073689732948994, + "tokens_seen": 1542914048 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002689669007021063, + "loss": 1.3981, + "theoretical_loss": 3.5073559093770577, + "tokens_seen": 1542979584 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002689568706118355, + "loss": 1.3851, + "theoretical_loss": 3.5073428461694327, + "tokens_seen": 1543045120 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002689468405215647, + "loss": 1.3583, + "theoretical_loss": 3.507329783671957, + "tokens_seen": 1543110656 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026893681043129385, + "loss": 1.2726, + "theoretical_loss": 3.507316721884561, + "tokens_seen": 1543176192 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002689267803410231, + "loss": 1.4027, + "theoretical_loss": 3.5073036608071764, + "tokens_seen": 1543241728 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002689167502507522, + "loss": 1.3305, + "theoretical_loss": 3.5072906004397337, + "tokens_seen": 1543307264 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.0015451116487383842, + "objective/train/docs_used": 875990, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.760725975036621, + "objective/train/original_loss": 2.760725736618042, + "objective/train/theoretical_loss": 3.507277540782165, + "objective/train/tokens_used": 1563832800, + "objective/train/value_avg": -0.004955291748046875, + "objective/train/value_loss": 5.3862921049585566e-05, + "objective/train/value_max": -0.00042557716369628906, + "objective/train/value_min": -0.04931640625, + "objective/train/value_reward_corr": 0.4365325261498278, + "objective/train/value_std": 0.0040130615234375, + "objective/train/weight_avg": 0.9871395230293274, + "objective/train/weighted_lm_loss": 2.726952075958252, + "objective/train/weights_max": 1.485138177871704, + "objective/train/weights_min": 0.6165010333061218, + "theoretical_loss": 3.507277540782165, + "tokens_seen": 1543372800 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026890672016048145, + "loss": 1.3313, + "theoretical_loss": 3.507277540782165, + "tokens_seen": 1543372800 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026889669007021063, + "loss": 1.3187, + "theoretical_loss": 3.507264481834402, + "tokens_seen": 1543438336 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002688866599799398, + "loss": 1.371, + "theoretical_loss": 3.507251423596374, + "tokens_seen": 1543503872 + }, + { + "epoch": 0.47, + "learning_rate": 0.000268876629889669, + "loss": 1.4068, + "theoretical_loss": 3.507238366068014, + "tokens_seen": 1543569408 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026886659979939823, + "loss": 1.262, + "theoretical_loss": 3.507225309249254, + "tokens_seen": 1543634944 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026885656970912736, + "loss": 1.373, + "theoretical_loss": 3.5072122531400227, + "tokens_seen": 1543700480 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002688465396188566, + "loss": 1.2966, + "theoretical_loss": 3.5071991977402535, + "tokens_seen": 1543766016 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002688365095285857, + "loss": 1.2642, + "theoretical_loss": 3.5071861430498767, + "tokens_seen": 1543831552 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026882647943831495, + "loss": 1.2863, + "theoretical_loss": 3.5071730890688246, + "tokens_seen": 1543897088 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026881644934804413, + "loss": 1.4534, + "theoretical_loss": 3.5071600357970283, + "tokens_seen": 1543962624 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002688064192577733, + "loss": 1.391, + "theoretical_loss": 3.5071469832344184, + "tokens_seen": 1544028160 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002687963891675025, + "loss": 1.2838, + "theoretical_loss": 3.5071339313809267, + "tokens_seen": 1544093696 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002687863590772317, + "loss": 1.2692, + "theoretical_loss": 3.507120880236485, + "tokens_seen": 1544159232 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026877632898696086, + "loss": 1.2619, + "theoretical_loss": 3.507107829801024, + "tokens_seen": 1544224768 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002687662988966901, + "loss": 1.409, + "theoretical_loss": 3.5070947800744756, + "tokens_seen": 1544290304 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002687562688064193, + "loss": 1.2916, + "theoretical_loss": 3.507081731056771, + "tokens_seen": 1544355840 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026874623871614846, + "loss": 1.2803, + "theoretical_loss": 3.5070686827478417, + "tokens_seen": 1544421376 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002687362086258777, + "loss": 1.3343, + "theoretical_loss": 3.507055635147619, + "tokens_seen": 1544486912 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002687261785356068, + "loss": 1.2985, + "theoretical_loss": 3.5070425882560343, + "tokens_seen": 1544552448 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026871614844533605, + "loss": 1.2325, + "theoretical_loss": 3.5070295420730195, + "tokens_seen": 1544617984 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002687061183550652, + "loss": 1.4405, + "theoretical_loss": 3.5070164965985056, + "tokens_seen": 1544683520 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002686960882647944, + "loss": 1.2591, + "theoretical_loss": 3.507003451832424, + "tokens_seen": 1544749056 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002686860581745236, + "loss": 1.3139, + "theoretical_loss": 3.5069904077747065, + "tokens_seen": 1544814592 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002686760280842528, + "loss": 1.3734, + "theoretical_loss": 3.5069773644252846, + "tokens_seen": 1544880128 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026866599799398196, + "loss": 1.2648, + "theoretical_loss": 3.50696432178409, + "tokens_seen": 1544945664 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.0016885806107893586, + "objective/train/docs_used": 876526, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.870736837387085, + "objective/train/original_loss": 2.870736837387085, + "objective/train/theoretical_loss": 3.5069512798510534, + "objective/train/tokens_used": 1565471200, + "objective/train/value_avg": -0.006877899169921875, + "objective/train/value_loss": 7.956966874189675e-05, + "objective/train/value_max": -0.00040459632873535156, + "objective/train/value_min": -0.07183837890625, + "objective/train/value_reward_corr": 0.4788950314614871, + "objective/train/value_std": 0.00531005859375, + "objective/train/weight_avg": 0.9868391752243042, + "objective/train/weighted_lm_loss": 2.8485286235809326, + "objective/train/weights_max": 1.8338974714279175, + "objective/train/weights_min": 0.298839807510376, + "theoretical_loss": 3.5069512798510534, + "tokens_seen": 1545011200 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026865596790371114, + "loss": 1.3622, + "theoretical_loss": 3.5069512798510534, + "tokens_seen": 1545011200 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002686459378134403, + "loss": 1.2728, + "theoretical_loss": 3.5069382386261068, + "tokens_seen": 1545076736 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026863590772316956, + "loss": 1.2966, + "theoretical_loss": 3.5069251981091814, + "tokens_seen": 1545142272 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002686258776328987, + "loss": 1.209, + "theoretical_loss": 3.5069121583002096, + "tokens_seen": 1545207808 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002686158475426279, + "loss": 1.3468, + "theoretical_loss": 3.5068991191991223, + "tokens_seen": 1545273344 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026860581745235705, + "loss": 1.3551, + "theoretical_loss": 3.5068860808058506, + "tokens_seen": 1545338880 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002685957873620863, + "loss": 1.2529, + "theoretical_loss": 3.506873043120327, + "tokens_seen": 1545404416 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026858575727181546, + "loss": 1.2902, + "theoretical_loss": 3.5068600061424826, + "tokens_seen": 1545469952 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026857572718154464, + "loss": 1.2132, + "theoretical_loss": 3.5068469698722486, + "tokens_seen": 1545535488 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002685656970912738, + "loss": 1.1457, + "theoretical_loss": 3.506833934309558, + "tokens_seen": 1545601024 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026855566700100306, + "loss": 1.3729, + "theoretical_loss": 3.5068208994543406, + "tokens_seen": 1545666560 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002685456369107322, + "loss": 1.2794, + "theoretical_loss": 3.5068078653065293, + "tokens_seen": 1545732096 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002685356068204614, + "loss": 1.3013, + "theoretical_loss": 3.506794831866055, + "tokens_seen": 1545797632 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026852557673019055, + "loss": 1.3064, + "theoretical_loss": 3.50678179913285, + "tokens_seen": 1545863168 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002685155466399198, + "loss": 1.3267, + "theoretical_loss": 3.506768767106845, + "tokens_seen": 1545928704 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026850551654964897, + "loss": 1.2358, + "theoretical_loss": 3.506755735787972, + "tokens_seen": 1545994240 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026849548645937815, + "loss": 1.2806, + "theoretical_loss": 3.5067427051761633, + "tokens_seen": 1546059776 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026848545636910733, + "loss": 1.3126, + "theoretical_loss": 3.50672967527135, + "tokens_seen": 1546125312 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002684754262788365, + "loss": 1.313, + "theoretical_loss": 3.506716646073464, + "tokens_seen": 1546190848 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002684653961885657, + "loss": 1.4034, + "theoretical_loss": 3.5067036175824366, + "tokens_seen": 1546256384 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002684553660982949, + "loss": 1.3517, + "theoretical_loss": 3.5066905897981995, + "tokens_seen": 1546321920 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026844533600802405, + "loss": 1.4053, + "theoretical_loss": 3.506677562720685, + "tokens_seen": 1546387456 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002684353059177533, + "loss": 1.2915, + "theoretical_loss": 3.506664536349825, + "tokens_seen": 1546452992 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002684252758274824, + "loss": 1.3092, + "theoretical_loss": 3.5066515106855496, + "tokens_seen": 1546518528 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026841524573721165, + "loss": 1.3545, + "theoretical_loss": 3.5066384857277924, + "tokens_seen": 1546584064 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": 0.00029018428176641464, + "objective/train/docs_used": 877732, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4631850719451904, + "objective/train/original_loss": 2.4631850719451904, + "objective/train/theoretical_loss": 3.5066254614764842, + "objective/train/tokens_used": 1567109600, + "objective/train/value_avg": -0.004764556884765625, + "objective/train/value_loss": 3.6296445614425465e-05, + "objective/train/value_max": -0.0004897117614746094, + "objective/train/value_min": -0.06231689453125, + "objective/train/value_reward_corr": 0.42983137078912, + "objective/train/value_std": 0.0038127899169921875, + "objective/train/weight_avg": 1.0046100616455078, + "objective/train/weighted_lm_loss": 2.469862699508667, + "objective/train/weights_max": 1.7225468158721924, + "objective/train/weights_min": 0.25339630246162415, + "theoretical_loss": 3.5066254614764842, + "tokens_seen": 1546649600 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026840521564694083, + "loss": 1.3158, + "theoretical_loss": 3.5066254614764842, + "tokens_seen": 1546649600 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026839518555667, + "loss": 1.3116, + "theoretical_loss": 3.5066124379315564, + "tokens_seen": 1546715136 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002683851554663992, + "loss": 1.281, + "theoretical_loss": 3.506599415092942, + "tokens_seen": 1546780672 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026837512537612843, + "loss": 1.2389, + "theoretical_loss": 3.506586392960572, + "tokens_seen": 1546846208 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026836509528585756, + "loss": 1.3704, + "theoretical_loss": 3.506573371534378, + "tokens_seen": 1546911744 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002683550651955868, + "loss": 1.2259, + "theoretical_loss": 3.5065603508142917, + "tokens_seen": 1546977280 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002683450351053159, + "loss": 1.2822, + "theoretical_loss": 3.5065473308002453, + "tokens_seen": 1547042816 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026833500501504515, + "loss": 1.3662, + "theoretical_loss": 3.506534311492171, + "tokens_seen": 1547108352 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026832497492477433, + "loss": 1.4495, + "theoretical_loss": 3.5065212928899996, + "tokens_seen": 1547173888 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002683149448345035, + "loss": 1.333, + "theoretical_loss": 3.506508274993664, + "tokens_seen": 1547239424 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002683049147442327, + "loss": 1.4811, + "theoretical_loss": 3.5064952578030955, + "tokens_seen": 1547304960 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002682948846539619, + "loss": 1.3474, + "theoretical_loss": 3.5064822413182255, + "tokens_seen": 1547370496 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026828485456369106, + "loss": 1.4505, + "theoretical_loss": 3.5064692255389867, + "tokens_seen": 1547436032 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002682748244734203, + "loss": 1.3329, + "theoretical_loss": 3.50645621046531, + "tokens_seen": 1547501568 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002682647943831494, + "loss": 1.2783, + "theoretical_loss": 3.5064431960971287, + "tokens_seen": 1547567104 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026825476429287866, + "loss": 1.3519, + "theoretical_loss": 3.5064301824343733, + "tokens_seen": 1547632640 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002682447342026078, + "loss": 1.297, + "theoretical_loss": 3.5064171694769763, + "tokens_seen": 1547698176 + }, + { + "epoch": 0.47, + "learning_rate": 0.000268234704112337, + "loss": 1.2708, + "theoretical_loss": 3.5064041572248694, + "tokens_seen": 1547763712 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002682246740220662, + "loss": 1.3105, + "theoretical_loss": 3.506391145677985, + "tokens_seen": 1547829248 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002682146439317954, + "loss": 1.4298, + "theoretical_loss": 3.506378134836254, + "tokens_seen": 1547894784 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026820461384152456, + "loss": 1.416, + "theoretical_loss": 3.5063651246996095, + "tokens_seen": 1547960320 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002681945837512538, + "loss": 1.3903, + "theoretical_loss": 3.5063521152679833, + "tokens_seen": 1548025856 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002681845536609829, + "loss": 1.3792, + "theoretical_loss": 3.506339106541306, + "tokens_seen": 1548091392 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026817452357071216, + "loss": 1.3779, + "theoretical_loss": 3.5063260985195113, + "tokens_seen": 1548156928 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002681644934804413, + "loss": 1.2609, + "theoretical_loss": 3.5063130912025304, + "tokens_seen": 1548222464 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.0011279992759227753, + "objective/train/docs_used": 878530, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6500141620635986, + "objective/train/original_loss": 2.6500141620635986, + "objective/train/theoretical_loss": 3.506300084590295, + "objective/train/tokens_used": 1568748000, + "objective/train/value_avg": -0.005130767822265625, + "objective/train/value_loss": 7.494973397115245e-05, + "objective/train/value_max": -0.00033283233642578125, + "objective/train/value_min": -0.1513671875, + "objective/train/value_reward_corr": 0.4005278661800279, + "objective/train/value_std": 0.004566192626953125, + "objective/train/weight_avg": 0.9922328591346741, + "objective/train/weighted_lm_loss": 2.6355669498443604, + "objective/train/weights_max": 4.222811698913574, + "objective/train/weights_min": 0.20813994109630585, + "theoretical_loss": 3.506300084590295, + "tokens_seen": 1548288000 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002681544633901705, + "loss": 1.3744, + "theoretical_loss": 3.506300084590295, + "tokens_seen": 1548288000 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002681444332998997, + "loss": 1.4204, + "theoretical_loss": 3.5062870786827376, + "tokens_seen": 1548353536 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002681344032096289, + "loss": 1.3779, + "theoretical_loss": 3.5062740734797897, + "tokens_seen": 1548419072 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026812437311935807, + "loss": 1.3916, + "theoretical_loss": 3.506261068981384, + "tokens_seen": 1548484608 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026811434302908725, + "loss": 1.3819, + "theoretical_loss": 3.506248065187452, + "tokens_seen": 1548550144 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026810431293881643, + "loss": 1.4875, + "theoretical_loss": 3.5062350620979252, + "tokens_seen": 1548615680 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026809428284854566, + "loss": 1.409, + "theoretical_loss": 3.506222059712737, + "tokens_seen": 1548681216 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002680842527582748, + "loss": 1.3446, + "theoretical_loss": 3.5062090580318186, + "tokens_seen": 1548746752 + }, + { + "epoch": 0.47, + "learning_rate": 0.000268074222668004, + "loss": 1.3459, + "theoretical_loss": 3.506196057055102, + "tokens_seen": 1548812288 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026806419257773315, + "loss": 1.3343, + "theoretical_loss": 3.5061830567825196, + "tokens_seen": 1548877824 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002680541624874624, + "loss": 1.3277, + "theoretical_loss": 3.5061700572140033, + "tokens_seen": 1548943360 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026804413239719157, + "loss": 1.3984, + "theoretical_loss": 3.506157058349485, + "tokens_seen": 1549008896 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026803410230692075, + "loss": 1.365, + "theoretical_loss": 3.5061440601888973, + "tokens_seen": 1549074432 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026802407221664993, + "loss": 1.2393, + "theoretical_loss": 3.5061310627321713, + "tokens_seen": 1549139968 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026801404212637917, + "loss": 1.4437, + "theoretical_loss": 3.5061180659792406, + "tokens_seen": 1549205504 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026800401203610835, + "loss": 1.3672, + "theoretical_loss": 3.506105069930036, + "tokens_seen": 1549271040 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026799398194583753, + "loss": 1.267, + "theoretical_loss": 3.5060920745844903, + "tokens_seen": 1549336576 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002679839518555667, + "loss": 1.4032, + "theoretical_loss": 3.506079079942536, + "tokens_seen": 1549402112 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002679739217652959, + "loss": 1.4257, + "theoretical_loss": 3.506066086004104, + "tokens_seen": 1549467648 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002679638916750251, + "loss": 1.343, + "theoretical_loss": 3.506053092769127, + "tokens_seen": 1549533184 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026795386158475425, + "loss": 1.2836, + "theoretical_loss": 3.506040100237538, + "tokens_seen": 1549598720 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002679438314944835, + "loss": 1.3224, + "theoretical_loss": 3.5060271084092682, + "tokens_seen": 1549664256 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002679338014042126, + "loss": 1.3585, + "theoretical_loss": 3.5060141172842503, + "tokens_seen": 1549729792 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026792377131394185, + "loss": 1.3767, + "theoretical_loss": 3.506001126862416, + "tokens_seen": 1549795328 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026791374122367103, + "loss": 1.3673, + "theoretical_loss": 3.505988137143698, + "tokens_seen": 1549860864 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.00025606952840462327, + "objective/train/docs_used": 879841, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.396853446960449, + "objective/train/original_loss": 2.39685320854187, + "objective/train/theoretical_loss": 3.5059751481280284, + "objective/train/tokens_used": 1570386400, + "objective/train/value_avg": -0.005306243896484375, + "objective/train/value_loss": 5.1740829803748056e-05, + "objective/train/value_max": -0.0002868175506591797, + "objective/train/value_min": -0.0775146484375, + "objective/train/value_reward_corr": 0.474936692458366, + "objective/train/value_std": 0.004405975341796875, + "objective/train/weight_avg": 0.9998553395271301, + "objective/train/weighted_lm_loss": 2.407195568084717, + "objective/train/weights_max": 1.6923240423202515, + "objective/train/weights_min": 0.3834691643714905, + "theoretical_loss": 3.5059751481280284, + "tokens_seen": 1549926400 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002679037111334002, + "loss": 1.1856, + "theoretical_loss": 3.5059751481280284, + "tokens_seen": 1549926400 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002678936810431294, + "loss": 1.4134, + "theoretical_loss": 3.5059621598153394, + "tokens_seen": 1549991936 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026788365095285863, + "loss": 1.4133, + "theoretical_loss": 3.505949172205563, + "tokens_seen": 1550057472 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026787362086258776, + "loss": 1.3616, + "theoretical_loss": 3.5059361852986317, + "tokens_seen": 1550123008 + }, + { + "epoch": 0.47, + "learning_rate": 0.000267863590772317, + "loss": 1.3564, + "theoretical_loss": 3.505923199094478, + "tokens_seen": 1550188544 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002678535606820461, + "loss": 1.3331, + "theoretical_loss": 3.5059102135930336, + "tokens_seen": 1550254080 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026784353059177535, + "loss": 1.3057, + "theoretical_loss": 3.5058972287942307, + "tokens_seen": 1550319616 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026783350050150453, + "loss": 1.3345, + "theoretical_loss": 3.5058842446980023, + "tokens_seen": 1550385152 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002678234704112337, + "loss": 1.265, + "theoretical_loss": 3.5058712613042804, + "tokens_seen": 1550450688 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002678134403209629, + "loss": 1.3415, + "theoretical_loss": 3.5058582786129966, + "tokens_seen": 1550516224 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002678034102306921, + "loss": 1.2854, + "theoretical_loss": 3.5058452966240843, + "tokens_seen": 1550581760 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026779338014042126, + "loss": 1.1674, + "theoretical_loss": 3.505832315337475, + "tokens_seen": 1550647296 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002677833500501505, + "loss": 1.3489, + "theoretical_loss": 3.5058193347531015, + "tokens_seen": 1550712832 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002677733199598796, + "loss": 1.241, + "theoretical_loss": 3.505806354870896, + "tokens_seen": 1550778368 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026776328986960886, + "loss": 1.3384, + "theoretical_loss": 3.505793375690791, + "tokens_seen": 1550843904 + }, + { + "epoch": 0.47, + "learning_rate": 0.000267753259779338, + "loss": 1.3157, + "theoretical_loss": 3.5057803972127184, + "tokens_seen": 1550909440 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002677432296890672, + "loss": 1.2447, + "theoretical_loss": 3.505767419436611, + "tokens_seen": 1550974976 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002677331995987964, + "loss": 1.4003, + "theoretical_loss": 3.505754442362401, + "tokens_seen": 1551040512 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002677231695085256, + "loss": 1.2921, + "theoretical_loss": 3.5057414659900203, + "tokens_seen": 1551106048 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026771313941825476, + "loss": 1.2086, + "theoretical_loss": 3.5057284903194024, + "tokens_seen": 1551171584 + }, + { + "epoch": 0.47, + "learning_rate": 0.000267703109327984, + "loss": 1.3046, + "theoretical_loss": 3.505715515350479, + "tokens_seen": 1551237120 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002676930792377131, + "loss": 1.3851, + "theoretical_loss": 3.505702541083182, + "tokens_seen": 1551302656 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026768304914744236, + "loss": 1.2363, + "theoretical_loss": 3.5056895675174453, + "tokens_seen": 1551368192 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002676730190571715, + "loss": 1.4134, + "theoretical_loss": 3.5056765946532003, + "tokens_seen": 1551433728 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002676629889669007, + "loss": 1.2874, + "theoretical_loss": 3.5056636224903794, + "tokens_seen": 1551499264 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.0003485331544652581, + "objective/train/docs_used": 880523, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.048173189163208, + "objective/train/original_loss": 3.048173189163208, + "objective/train/theoretical_loss": 3.5056506510289154, + "objective/train/tokens_used": 1572024800, + "objective/train/value_avg": -0.005115509033203125, + "objective/train/value_loss": 6.762588600395247e-05, + "objective/train/value_max": -0.0002913475036621094, + "objective/train/value_min": -0.07708740234375, + "objective/train/value_reward_corr": 0.5705376141307931, + "objective/train/value_std": 0.005214691162109375, + "objective/train/weight_avg": 0.9996607899665833, + "objective/train/weighted_lm_loss": 3.051745891571045, + "objective/train/weights_max": 1.8038166761398315, + "objective/train/weights_min": 0.39224252104759216, + "theoretical_loss": 3.5056506510289154, + "tokens_seen": 1551564800 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002676529588766299, + "loss": 1.3005, + "theoretical_loss": 3.5056506510289154, + "tokens_seen": 1551564800 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002676429287863591, + "loss": 1.3836, + "theoretical_loss": 3.50563768026874, + "tokens_seen": 1551630336 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026763289869608827, + "loss": 1.397, + "theoretical_loss": 3.505624710209787, + "tokens_seen": 1551695872 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026762286860581745, + "loss": 1.3565, + "theoretical_loss": 3.505611740851988, + "tokens_seen": 1551761408 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026761283851554663, + "loss": 1.3672, + "theoretical_loss": 3.505598772195275, + "tokens_seen": 1551826944 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026760280842527586, + "loss": 1.3124, + "theoretical_loss": 3.5055858042395815, + "tokens_seen": 1551892480 + }, + { + "epoch": 0.47, + "learning_rate": 0.000267592778335005, + "loss": 1.3616, + "theoretical_loss": 3.50557283698484, + "tokens_seen": 1551958016 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002675827482447342, + "loss": 1.2551, + "theoretical_loss": 3.5055598704309823, + "tokens_seen": 1552023552 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026757271815446335, + "loss": 1.315, + "theoretical_loss": 3.5055469045779413, + "tokens_seen": 1552089088 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002675626880641926, + "loss": 1.4072, + "theoretical_loss": 3.50553393942565, + "tokens_seen": 1552154624 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026755265797392177, + "loss": 1.3346, + "theoretical_loss": 3.50552097497404, + "tokens_seen": 1552220160 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026754262788365095, + "loss": 1.2375, + "theoretical_loss": 3.505508011223044, + "tokens_seen": 1552285696 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026753259779338013, + "loss": 1.4018, + "theoretical_loss": 3.5054950481725955, + "tokens_seen": 1552351232 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026752256770310937, + "loss": 1.3146, + "theoretical_loss": 3.5054820858226257, + "tokens_seen": 1552416768 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002675125376128385, + "loss": 1.3547, + "theoretical_loss": 3.505469124173068, + "tokens_seen": 1552482304 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026750250752256773, + "loss": 1.2707, + "theoretical_loss": 3.5054561632238554, + "tokens_seen": 1552547840 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026749247743229686, + "loss": 1.2052, + "theoretical_loss": 3.50544320297492, + "tokens_seen": 1552613376 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002674824473420261, + "loss": 1.227, + "theoretical_loss": 3.5054302434261944, + "tokens_seen": 1552678912 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026747241725175527, + "loss": 1.3962, + "theoretical_loss": 3.505417284577611, + "tokens_seen": 1552744448 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026746238716148445, + "loss": 1.1861, + "theoretical_loss": 3.5054043264291024, + "tokens_seen": 1552809984 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026745235707121363, + "loss": 1.2849, + "theoretical_loss": 3.505391368980602, + "tokens_seen": 1552875520 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002674423269809428, + "loss": 1.376, + "theoretical_loss": 3.5053784122320417, + "tokens_seen": 1552941056 + }, + { + "epoch": 0.47, + "learning_rate": 0.000267432296890672, + "loss": 1.4263, + "theoretical_loss": 3.5053654561833545, + "tokens_seen": 1553006592 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026742226680040123, + "loss": 1.4446, + "theoretical_loss": 3.505352500834473, + "tokens_seen": 1553072128 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026741223671013036, + "loss": 1.4553, + "theoretical_loss": 3.505339546185329, + "tokens_seen": 1553137664 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.0004572200123220682, + "objective/train/docs_used": 881131, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5482099056243896, + "objective/train/original_loss": 2.5482099056243896, + "objective/train/theoretical_loss": 3.505326592235857, + "objective/train/tokens_used": 1573663200, + "objective/train/value_avg": -0.0043487548828125, + "objective/train/value_loss": 4.0298353269463405e-05, + "objective/train/value_max": -0.0003275871276855469, + "objective/train/value_min": -0.07696533203125, + "objective/train/value_reward_corr": 0.3081350382677088, + "objective/train/value_std": 0.0036182403564453125, + "objective/train/weight_avg": 0.997340202331543, + "objective/train/weighted_lm_loss": 2.547450542449951, + "objective/train/weights_max": 1.9959484338760376, + "objective/train/weights_min": 0.4421577453613281, + "theoretical_loss": 3.505326592235857, + "tokens_seen": 1553203200 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002674022066198596, + "loss": 1.2995, + "theoretical_loss": 3.505326592235857, + "tokens_seen": 1553203200 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002673921765295888, + "loss": 1.3071, + "theoretical_loss": 3.5053136389859882, + "tokens_seen": 1553268736 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026738214643931796, + "loss": 1.2664, + "theoretical_loss": 3.505300686435656, + "tokens_seen": 1553334272 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026737211634904714, + "loss": 1.3271, + "theoretical_loss": 3.505287734584793, + "tokens_seen": 1553399808 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002673620862587763, + "loss": 1.3172, + "theoretical_loss": 3.5052747834333315, + "tokens_seen": 1553465344 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002673520561685055, + "loss": 1.3086, + "theoretical_loss": 3.505261832981205, + "tokens_seen": 1553530880 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026734202607823474, + "loss": 1.342, + "theoretical_loss": 3.5052488832283455, + "tokens_seen": 1553596416 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026733199598796386, + "loss": 1.4541, + "theoretical_loss": 3.505235934174686, + "tokens_seen": 1553661952 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002673219658976931, + "loss": 1.304, + "theoretical_loss": 3.5052229858201596, + "tokens_seen": 1553727488 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002673119358074222, + "loss": 1.3729, + "theoretical_loss": 3.505210038164698, + "tokens_seen": 1553793024 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026730190571715146, + "loss": 1.3375, + "theoretical_loss": 3.5051970912082355, + "tokens_seen": 1553858560 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026729187562688064, + "loss": 1.307, + "theoretical_loss": 3.5051841449507037, + "tokens_seen": 1553924096 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002672818455366098, + "loss": 1.3154, + "theoretical_loss": 3.505171199392036, + "tokens_seen": 1553989632 + }, + { + "epoch": 0.47, + "learning_rate": 0.000267271815446339, + "loss": 1.3723, + "theoretical_loss": 3.505158254532165, + "tokens_seen": 1554055168 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002672617853560682, + "loss": 1.3586, + "theoretical_loss": 3.5051453103710237, + "tokens_seen": 1554120704 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002672517552657974, + "loss": 1.2616, + "theoretical_loss": 3.505132366908545, + "tokens_seen": 1554186240 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002672417251755266, + "loss": 1.3177, + "theoretical_loss": 3.505119424144661, + "tokens_seen": 1554251776 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002672316950852558, + "loss": 1.3974, + "theoretical_loss": 3.505106482079305, + "tokens_seen": 1554317312 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026722166499498496, + "loss": 1.3369, + "theoretical_loss": 3.50509354071241, + "tokens_seen": 1554382848 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002672116349047142, + "loss": 1.3145, + "theoretical_loss": 3.5050806000439088, + "tokens_seen": 1554448384 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002672016048144433, + "loss": 1.3781, + "theoretical_loss": 3.5050676600737343, + "tokens_seen": 1554513920 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026719157472417256, + "loss": 1.3274, + "theoretical_loss": 3.505054720801819, + "tokens_seen": 1554579456 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002671815446339017, + "loss": 1.3832, + "theoretical_loss": 3.505041782228096, + "tokens_seen": 1554644992 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002671715145436309, + "loss": 1.4067, + "theoretical_loss": 3.505028844352499, + "tokens_seen": 1554710528 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002671614844533601, + "loss": 1.2822, + "theoretical_loss": 3.5050159071749594, + "tokens_seen": 1554776064 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": 0.00010718072735471651, + "objective/train/docs_used": 882511, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7621402740478516, + "objective/train/original_loss": 2.7621395587921143, + "objective/train/theoretical_loss": 3.5050029706954113, + "objective/train/tokens_used": 1575301600, + "objective/train/value_avg": -0.00440216064453125, + "objective/train/value_loss": 4.685802196036093e-05, + "objective/train/value_max": -0.00029587745666503906, + "objective/train/value_min": -0.052703857421875, + "objective/train/value_reward_corr": 0.2602943835235384, + "objective/train/value_std": 0.00330352783203125, + "objective/train/weight_avg": 1.0032296180725098, + "objective/train/weighted_lm_loss": 2.7766411304473877, + "objective/train/weights_max": 1.4593350887298584, + "objective/train/weights_min": 0.3416737914085388, + "theoretical_loss": 3.5050029706954113, + "tokens_seen": 1554841600 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002671514543630893, + "loss": 1.3228, + "theoretical_loss": 3.5050029706954113, + "tokens_seen": 1554841600 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026714142427281847, + "loss": 1.363, + "theoretical_loss": 3.504990034913787, + "tokens_seen": 1554907136 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026713139418254765, + "loss": 1.286, + "theoretical_loss": 3.504977099830019, + "tokens_seen": 1554972672 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026712136409227683, + "loss": 1.4124, + "theoretical_loss": 3.504964165444042, + "tokens_seen": 1555038208 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026711133400200606, + "loss": 1.3908, + "theoretical_loss": 3.5049512317557876, + "tokens_seen": 1555103744 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002671013039117352, + "loss": 1.4182, + "theoretical_loss": 3.5049382987651887, + "tokens_seen": 1555169280 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002670912738214644, + "loss": 1.3271, + "theoretical_loss": 3.504925366472179, + "tokens_seen": 1555234816 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026708124373119355, + "loss": 1.2043, + "theoretical_loss": 3.5049124348766902, + "tokens_seen": 1555300352 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002670712136409228, + "loss": 1.335, + "theoretical_loss": 3.504899503978657, + "tokens_seen": 1555365888 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026706118355065197, + "loss": 1.2337, + "theoretical_loss": 3.5048865737780113, + "tokens_seen": 1555431424 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026705115346038115, + "loss": 1.4088, + "theoretical_loss": 3.504873644274686, + "tokens_seen": 1555496960 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026704112337011033, + "loss": 1.3812, + "theoretical_loss": 3.504860715468615, + "tokens_seen": 1555562496 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026703109327983957, + "loss": 1.3778, + "theoretical_loss": 3.50484778735973, + "tokens_seen": 1555628032 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002670210631895687, + "loss": 1.2734, + "theoretical_loss": 3.504834859947966, + "tokens_seen": 1555693568 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026701103309929793, + "loss": 1.235, + "theoretical_loss": 3.504821933233254, + "tokens_seen": 1555759104 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026700100300902706, + "loss": 1.3576, + "theoretical_loss": 3.504809007215528, + "tokens_seen": 1555824640 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002669909729187563, + "loss": 1.2908, + "theoretical_loss": 3.504796081894721, + "tokens_seen": 1555890176 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026698094282848547, + "loss": 1.4129, + "theoretical_loss": 3.5047831572707664, + "tokens_seen": 1555955712 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026697091273821465, + "loss": 1.3448, + "theoretical_loss": 3.5047702333435966, + "tokens_seen": 1556021248 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026696088264794383, + "loss": 1.2881, + "theoretical_loss": 3.504757310113145, + "tokens_seen": 1556086784 + }, + { + "epoch": 0.47, + "learning_rate": 0.000266950852557673, + "loss": 1.3237, + "theoretical_loss": 3.504744387579345, + "tokens_seen": 1556152320 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002669408224674022, + "loss": 1.3133, + "theoretical_loss": 3.5047314657421293, + "tokens_seen": 1556217856 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026693079237713143, + "loss": 1.3705, + "theoretical_loss": 3.504718544601431, + "tokens_seen": 1556283392 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026692076228686056, + "loss": 1.3201, + "theoretical_loss": 3.5047056241571837, + "tokens_seen": 1556348928 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002669107321965898, + "loss": 1.3186, + "theoretical_loss": 3.5046927044093197, + "tokens_seen": 1556414464 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.0007732663070783019, + "objective/train/docs_used": 882978, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9015157222747803, + "objective/train/original_loss": 2.9015157222747803, + "objective/train/theoretical_loss": 3.504679785357773, + "objective/train/tokens_used": 1576940000, + "objective/train/value_avg": -0.004184722900390625, + "objective/train/value_loss": 9.964116907212883e-05, + "objective/train/value_max": -0.00034880638122558594, + "objective/train/value_min": -0.04638671875, + "objective/train/value_reward_corr": 0.23102758763517198, + "objective/train/value_std": 0.0030956268310546875, + "objective/train/weight_avg": 0.9964028596878052, + "objective/train/weighted_lm_loss": 2.903454542160034, + "objective/train/weights_max": 1.3454169034957886, + "objective/train/weights_min": 0.19138504564762115, + "theoretical_loss": 3.504679785357773, + "tokens_seen": 1556480000 + }, + { + "epoch": 0.47, + "learning_rate": 0.000266900702106319, + "loss": 1.2389, + "theoretical_loss": 3.504679785357773, + "tokens_seen": 1556480000 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026689067201604816, + "loss": 1.2216, + "theoretical_loss": 3.5046668670024768, + "tokens_seen": 1556545536 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026688064192577734, + "loss": 1.3298, + "theoretical_loss": 3.5046539493433633, + "tokens_seen": 1556611072 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002668706118355065, + "loss": 1.421, + "theoretical_loss": 3.5046410323803663, + "tokens_seen": 1556676608 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002668605817452357, + "loss": 1.2631, + "theoretical_loss": 3.5046281161134196, + "tokens_seen": 1556742144 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026685055165496494, + "loss": 1.2848, + "theoretical_loss": 3.504615200542455, + "tokens_seen": 1556807680 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026684052156469406, + "loss": 1.3598, + "theoretical_loss": 3.504602285667407, + "tokens_seen": 1556873216 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002668304914744233, + "loss": 1.2701, + "theoretical_loss": 3.5045893714882075, + "tokens_seen": 1556938752 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002668204613841524, + "loss": 1.3634, + "theoretical_loss": 3.5045764580047916, + "tokens_seen": 1557004288 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026681043129388166, + "loss": 1.3069, + "theoretical_loss": 3.5045635452170902, + "tokens_seen": 1557069824 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026680040120361084, + "loss": 1.2753, + "theoretical_loss": 3.5045506331250382, + "tokens_seen": 1557135360 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026679037111334, + "loss": 1.3002, + "theoretical_loss": 3.5045377217285685, + "tokens_seen": 1557200896 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002667803410230692, + "loss": 1.36, + "theoretical_loss": 3.504524811027614, + "tokens_seen": 1557266432 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002667703109327984, + "loss": 1.3381, + "theoretical_loss": 3.5045119010221084, + "tokens_seen": 1557331968 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026676028084252757, + "loss": 1.3637, + "theoretical_loss": 3.504498991711985, + "tokens_seen": 1557397504 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002667502507522568, + "loss": 1.2363, + "theoretical_loss": 3.5044860830971762, + "tokens_seen": 1557463040 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026674022066198593, + "loss": 1.346, + "theoretical_loss": 3.504473175177617, + "tokens_seen": 1557528576 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026673019057171516, + "loss": 1.2961, + "theoretical_loss": 3.5044602679532386, + "tokens_seen": 1557594112 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026672016048144434, + "loss": 1.3288, + "theoretical_loss": 3.504447361423976, + "tokens_seen": 1557659648 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002667101303911735, + "loss": 1.3235, + "theoretical_loss": 3.5044344555897617, + "tokens_seen": 1557725184 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002667001003009027, + "loss": 1.3403, + "theoretical_loss": 3.5044215504505294, + "tokens_seen": 1557790720 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002666900702106319, + "loss": 1.3178, + "theoretical_loss": 3.504408646006212, + "tokens_seen": 1557856256 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026668004012036107, + "loss": 1.3504, + "theoretical_loss": 3.504395742256743, + "tokens_seen": 1557921792 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002666700100300903, + "loss": 1.4035, + "theoretical_loss": 3.504382839202056, + "tokens_seen": 1557987328 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026665997993981943, + "loss": 1.2932, + "theoretical_loss": 3.5043699368420844, + "tokens_seen": 1558052864 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.000883480766788125, + "objective/train/docs_used": 884109, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6678574085235596, + "objective/train/original_loss": 2.6678574085235596, + "objective/train/theoretical_loss": 3.5043570351767617, + "objective/train/tokens_used": 1578578400, + "objective/train/value_avg": -0.006221771240234375, + "objective/train/value_loss": 0.00010816296708071604, + "objective/train/value_max": -0.0002913475036621094, + "objective/train/value_min": -0.07452392578125, + "objective/train/value_reward_corr": 0.3646364052432643, + "objective/train/value_std": 0.004825592041015625, + "objective/train/weight_avg": 0.9959673285484314, + "objective/train/weighted_lm_loss": 2.649351119995117, + "objective/train/weights_max": 2.106945514678955, + "objective/train/weights_min": 0.08522149920463562, + "theoretical_loss": 3.5043570351767617, + "tokens_seen": 1558118400 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026664994984954867, + "loss": 1.3919, + "theoretical_loss": 3.5043570351767617, + "tokens_seen": 1558118400 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002666399197592778, + "loss": 1.2373, + "theoretical_loss": 3.50434413420602, + "tokens_seen": 1558183936 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026662988966900703, + "loss": 1.2665, + "theoretical_loss": 3.5043312339297947, + "tokens_seen": 1558249472 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002666198595787362, + "loss": 1.3484, + "theoretical_loss": 3.5043183343480178, + "tokens_seen": 1558315008 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002666098294884654, + "loss": 1.3145, + "theoretical_loss": 3.504305435460623, + "tokens_seen": 1558380544 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026659979939819457, + "loss": 1.3311, + "theoretical_loss": 3.504292537267544, + "tokens_seen": 1558446080 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026658976930792375, + "loss": 1.2955, + "theoretical_loss": 3.504279639768714, + "tokens_seen": 1558511616 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026657973921765293, + "loss": 1.3822, + "theoretical_loss": 3.5042667429640666, + "tokens_seen": 1558577152 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026656970912738217, + "loss": 1.3666, + "theoretical_loss": 3.504253846853535, + "tokens_seen": 1558642688 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002665596790371113, + "loss": 1.4455, + "theoretical_loss": 3.504240951437053, + "tokens_seen": 1558708224 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026654964894684053, + "loss": 1.341, + "theoretical_loss": 3.5042280567145543, + "tokens_seen": 1558773760 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002665396188565697, + "loss": 1.3046, + "theoretical_loss": 3.504215162685971, + "tokens_seen": 1558839296 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002665295887662989, + "loss": 1.3252, + "theoretical_loss": 3.5042022693512385, + "tokens_seen": 1558904832 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002665195586760281, + "loss": 1.3676, + "theoretical_loss": 3.5041893767102885, + "tokens_seen": 1558970368 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026650952858575726, + "loss": 1.3336, + "theoretical_loss": 3.5041764847630557, + "tokens_seen": 1559035904 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002664994984954865, + "loss": 1.2833, + "theoretical_loss": 3.5041635935094737, + "tokens_seen": 1559101440 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026648946840521567, + "loss": 1.3223, + "theoretical_loss": 3.5041507029494747, + "tokens_seen": 1559166976 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026647943831494485, + "loss": 1.2745, + "theoretical_loss": 3.504137813082994, + "tokens_seen": 1559232512 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026646940822467403, + "loss": 1.2974, + "theoretical_loss": 3.5041249239099637, + "tokens_seen": 1559298048 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002664593781344032, + "loss": 1.3251, + "theoretical_loss": 3.5041120354303184, + "tokens_seen": 1559363584 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002664493480441324, + "loss": 1.4036, + "theoretical_loss": 3.5040991476439904, + "tokens_seen": 1559429120 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026643931795386163, + "loss": 1.3404, + "theoretical_loss": 3.5040862605509147, + "tokens_seen": 1559494656 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026642928786359076, + "loss": 1.3737, + "theoretical_loss": 3.504073374151024, + "tokens_seen": 1559560192 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026641925777332, + "loss": 1.3034, + "theoretical_loss": 3.504060488444252, + "tokens_seen": 1559625728 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002664092276830492, + "loss": 1.2897, + "theoretical_loss": 3.5040476034305326, + "tokens_seen": 1559691264 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": 0.00016976271581370384, + "objective/train/docs_used": 884823, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.501232624053955, + "objective/train/original_loss": 2.501232385635376, + "objective/train/theoretical_loss": 3.504034719109799, + "objective/train/tokens_used": 1580216800, + "objective/train/value_avg": -0.005138397216796875, + "objective/train/value_loss": 4.999504017177969e-05, + "objective/train/value_max": -0.00040459632873535156, + "objective/train/value_min": -0.0633544921875, + "objective/train/value_reward_corr": 0.338152756544428, + "objective/train/value_std": 0.004241943359375, + "objective/train/weight_avg": 1.0040690898895264, + "objective/train/weighted_lm_loss": 2.520690679550171, + "objective/train/weights_max": 1.7324062585830688, + "objective/train/weights_min": 0.33607012033462524, + "theoretical_loss": 3.504034719109799, + "tokens_seen": 1559756800 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026639919759277836, + "loss": 1.2812, + "theoretical_loss": 3.504034719109799, + "tokens_seen": 1559756800 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026638916750250754, + "loss": 1.2315, + "theoretical_loss": 3.5040218354819856, + "tokens_seen": 1559822336 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002663791374122367, + "loss": 1.3485, + "theoretical_loss": 3.504008952547025, + "tokens_seen": 1559887872 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002663691073219659, + "loss": 1.314, + "theoretical_loss": 3.5039960703048507, + "tokens_seen": 1559953408 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026635907723169514, + "loss": 1.2894, + "theoretical_loss": 3.503983188755398, + "tokens_seen": 1560018944 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026634904714142426, + "loss": 1.3449, + "theoretical_loss": 3.503970307898599, + "tokens_seen": 1560084480 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002663390170511535, + "loss": 1.2501, + "theoretical_loss": 3.5039574277343877, + "tokens_seen": 1560150016 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002663289869608826, + "loss": 1.3384, + "theoretical_loss": 3.503944548262698, + "tokens_seen": 1560215552 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026631895687061186, + "loss": 1.3211, + "theoretical_loss": 3.5039316694834635, + "tokens_seen": 1560281088 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026630892678034104, + "loss": 1.3292, + "theoretical_loss": 3.503918791396618, + "tokens_seen": 1560346624 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002662988966900702, + "loss": 1.2915, + "theoretical_loss": 3.503905914002095, + "tokens_seen": 1560412160 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002662888665997994, + "loss": 1.4606, + "theoretical_loss": 3.5038930372998283, + "tokens_seen": 1560477696 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002662788365095286, + "loss": 1.2607, + "theoretical_loss": 3.503880161289751, + "tokens_seen": 1560543232 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026626880641925777, + "loss": 1.3971, + "theoretical_loss": 3.5038672859717983, + "tokens_seen": 1560608768 + }, + { + "epoch": 0.47, + "learning_rate": 0.000266258776328987, + "loss": 1.2872, + "theoretical_loss": 3.5038544113459027, + "tokens_seen": 1560674304 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026624874623871613, + "loss": 1.4136, + "theoretical_loss": 3.5038415374119984, + "tokens_seen": 1560739840 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026623871614844536, + "loss": 1.3518, + "theoretical_loss": 3.503828664170019, + "tokens_seen": 1560805376 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026622868605817454, + "loss": 1.4577, + "theoretical_loss": 3.503815791619898, + "tokens_seen": 1560870912 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002662186559679037, + "loss": 1.3777, + "theoretical_loss": 3.50380291976157, + "tokens_seen": 1560936448 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002662086258776329, + "loss": 1.4389, + "theoretical_loss": 3.5037900485949676, + "tokens_seen": 1561001984 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002661985957873621, + "loss": 1.3763, + "theoretical_loss": 3.5037771781200258, + "tokens_seen": 1561067520 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026618856569709127, + "loss": 1.3151, + "theoretical_loss": 3.5037643083366774, + "tokens_seen": 1561133056 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002661785356068205, + "loss": 1.2959, + "theoretical_loss": 3.503751439244857, + "tokens_seen": 1561198592 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026616850551654963, + "loss": 1.2685, + "theoretical_loss": 3.503738570844498, + "tokens_seen": 1561264128 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026615847542627887, + "loss": 1.3359, + "theoretical_loss": 3.5037257031355344, + "tokens_seen": 1561329664 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.0002753601293079555, + "objective/train/docs_used": 885915, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 1.9510542154312134, + "objective/train/original_loss": 1.951054334640503, + "objective/train/theoretical_loss": 3.5037128361178995, + "objective/train/tokens_used": 1581855200, + "objective/train/value_avg": -0.004558563232421875, + "objective/train/value_loss": 4.479961353354156e-05, + "objective/train/value_max": -0.00033020973205566406, + "objective/train/value_min": -0.0546875, + "objective/train/value_reward_corr": 0.40129303993481014, + "objective/train/value_std": 0.003509521484375, + "objective/train/weight_avg": 0.9993802905082703, + "objective/train/weighted_lm_loss": 1.9518260955810547, + "objective/train/weights_max": 1.40261709690094, + "objective/train/weights_min": 0.48247164487838745, + "theoretical_loss": 3.5037128361178995, + "tokens_seen": 1561395200 + }, + { + "epoch": 0.47, + "learning_rate": 0.000266148445336008, + "loss": 1.1652, + "theoretical_loss": 3.5037128361178995, + "tokens_seen": 1561395200 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026613841524573723, + "loss": 1.3358, + "theoretical_loss": 3.503699969791528, + "tokens_seen": 1561460736 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002661283851554664, + "loss": 1.3406, + "theoretical_loss": 3.503687104156353, + "tokens_seen": 1561526272 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002661183550651956, + "loss": 1.3576, + "theoretical_loss": 3.503674239212309, + "tokens_seen": 1561591808 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026610832497492477, + "loss": 1.3754, + "theoretical_loss": 3.5036613749593295, + "tokens_seen": 1561657344 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026609829488465395, + "loss": 1.3407, + "theoretical_loss": 3.5036485113973486, + "tokens_seen": 1561722880 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026608826479438313, + "loss": 1.3934, + "theoretical_loss": 3.5036356485262994, + "tokens_seen": 1561788416 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026607823470411237, + "loss": 1.3561, + "theoretical_loss": 3.503622786346117, + "tokens_seen": 1561853952 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002660682046138415, + "loss": 1.3262, + "theoretical_loss": 3.5036099248567343, + "tokens_seen": 1561919488 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026605817452357073, + "loss": 1.3823, + "theoretical_loss": 3.503597064058086, + "tokens_seen": 1561985024 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002660481444332999, + "loss": 1.2073, + "theoretical_loss": 3.5035842039501057, + "tokens_seen": 1562050560 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002660381143430291, + "loss": 1.4689, + "theoretical_loss": 3.503571344532727, + "tokens_seen": 1562116096 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002660280842527583, + "loss": 1.4171, + "theoretical_loss": 3.5035584858058844, + "tokens_seen": 1562181632 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026601805416248746, + "loss": 1.3474, + "theoretical_loss": 3.503545627769511, + "tokens_seen": 1562247168 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026600802407221664, + "loss": 1.3281, + "theoretical_loss": 3.5035327704235417, + "tokens_seen": 1562312704 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026599799398194587, + "loss": 1.3124, + "theoretical_loss": 3.5035199137679105, + "tokens_seen": 1562378240 + }, + { + "epoch": 0.47, + "learning_rate": 0.000265987963891675, + "loss": 1.3739, + "theoretical_loss": 3.5035070578025502, + "tokens_seen": 1562443776 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026597793380140423, + "loss": 1.3161, + "theoretical_loss": 3.5034942025273965, + "tokens_seen": 1562509312 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026596790371113336, + "loss": 1.3141, + "theoretical_loss": 3.5034813479423823, + "tokens_seen": 1562574848 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002659578736208626, + "loss": 1.3676, + "theoretical_loss": 3.5034684940474414, + "tokens_seen": 1562640384 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002659478435305918, + "loss": 1.35, + "theoretical_loss": 3.5034556408425086, + "tokens_seen": 1562705920 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026593781344032096, + "loss": 1.2823, + "theoretical_loss": 3.503442788327517, + "tokens_seen": 1562771456 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026592778335005014, + "loss": 1.3319, + "theoretical_loss": 3.5034299365024015, + "tokens_seen": 1562836992 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002659177532597794, + "loss": 1.3954, + "theoretical_loss": 3.503417085367096, + "tokens_seen": 1562902528 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002659077231695085, + "loss": 1.3457, + "theoretical_loss": 3.503404234921534, + "tokens_seen": 1562968064 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.001450386713258922, + "objective/train/docs_used": 886454, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.2146260738372803, + "objective/train/original_loss": 2.2146265506744385, + "objective/train/theoretical_loss": 3.50339138516565, + "objective/train/tokens_used": 1583493600, + "objective/train/value_avg": -0.005184173583984375, + "objective/train/value_loss": 0.00010702557483455166, + "objective/train/value_max": -0.0004029273986816406, + "objective/train/value_min": -0.053497314453125, + "objective/train/value_reward_corr": 0.3735362012499704, + "objective/train/value_std": 0.004047393798828125, + "objective/train/weight_avg": 0.9896339178085327, + "objective/train/weighted_lm_loss": 2.2006797790527344, + "objective/train/weights_max": 1.5756248235702515, + "objective/train/weights_min": 0.08974236249923706, + "theoretical_loss": 3.50339138516565, + "tokens_seen": 1563033600 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026589769307923774, + "loss": 1.3247, + "theoretical_loss": 3.50339138516565, + "tokens_seen": 1563033600 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026588766298896686, + "loss": 1.3951, + "theoretical_loss": 3.5033785360993783, + "tokens_seen": 1563099136 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002658776328986961, + "loss": 1.4492, + "theoretical_loss": 3.5033656877226527, + "tokens_seen": 1563164672 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002658676028084253, + "loss": 1.2855, + "theoretical_loss": 3.503352840035407, + "tokens_seen": 1563230208 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026585757271815446, + "loss": 1.2425, + "theoretical_loss": 3.5033399930375757, + "tokens_seen": 1563295744 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026584754262788364, + "loss": 1.4223, + "theoretical_loss": 3.5033271467290925, + "tokens_seen": 1563361280 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002658375125376128, + "loss": 1.2626, + "theoretical_loss": 3.503314301109892, + "tokens_seen": 1563426816 + }, + { + "epoch": 0.47, + "learning_rate": 0.000265827482447342, + "loss": 1.3159, + "theoretical_loss": 3.503301456179908, + "tokens_seen": 1563492352 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026581745235707124, + "loss": 1.406, + "theoretical_loss": 3.503288611939075, + "tokens_seen": 1563557888 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026580742226680037, + "loss": 1.2592, + "theoretical_loss": 3.5032757683873266, + "tokens_seen": 1563623424 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002657973921765296, + "loss": 1.2776, + "theoretical_loss": 3.5032629255245977, + "tokens_seen": 1563688960 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026578736208625873, + "loss": 1.3205, + "theoretical_loss": 3.5032500833508218, + "tokens_seen": 1563754496 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026577733199598797, + "loss": 1.3646, + "theoretical_loss": 3.503237241865933, + "tokens_seen": 1563820032 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026576730190571715, + "loss": 1.3221, + "theoretical_loss": 3.5032244010698657, + "tokens_seen": 1563885568 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026575727181544633, + "loss": 1.3932, + "theoretical_loss": 3.5032115609625545, + "tokens_seen": 1563951104 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026574724172517556, + "loss": 1.4876, + "theoretical_loss": 3.5031987215439333, + "tokens_seen": 1564016640 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026573721163490474, + "loss": 1.292, + "theoretical_loss": 3.503185882813936, + "tokens_seen": 1564082176 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002657271815446339, + "loss": 1.4091, + "theoretical_loss": 3.503173044772497, + "tokens_seen": 1564147712 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002657171514543631, + "loss": 1.3467, + "theoretical_loss": 3.5031602074195503, + "tokens_seen": 1564213248 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002657071213640923, + "loss": 1.2647, + "theoretical_loss": 3.503147370755031, + "tokens_seen": 1564278784 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026569709127382147, + "loss": 1.3938, + "theoretical_loss": 3.5031345347788725, + "tokens_seen": 1564344320 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002656870611835507, + "loss": 1.3689, + "theoretical_loss": 3.5031216994910093, + "tokens_seen": 1564409856 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026567703109327983, + "loss": 1.2156, + "theoretical_loss": 3.5031088648913755, + "tokens_seen": 1564475392 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026566700100300907, + "loss": 1.3623, + "theoretical_loss": 3.503096030979906, + "tokens_seen": 1564540928 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002656569709127382, + "loss": 1.2524, + "theoretical_loss": 3.503083197756534, + "tokens_seen": 1564606464 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": 0.00022397887369152158, + "objective/train/docs_used": 887471, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.440706968307495, + "objective/train/original_loss": 2.440706968307495, + "objective/train/theoretical_loss": 3.5030703652211947, + "objective/train/tokens_used": 1585132000, + "objective/train/value_avg": -0.005443572998046875, + "objective/train/value_loss": 4.806679498869926e-05, + "objective/train/value_max": -0.00028014183044433594, + "objective/train/value_min": -0.060211181640625, + "objective/train/value_reward_corr": 0.49141410286537957, + "objective/train/value_std": 0.004962921142578125, + "objective/train/weight_avg": 1.0044764280319214, + "objective/train/weighted_lm_loss": 2.4654688835144043, + "objective/train/weights_max": 1.5451133251190186, + "objective/train/weights_min": 0.3055603802204132, + "theoretical_loss": 3.5030703652211947, + "tokens_seen": 1564672000 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026564694082246743, + "loss": 1.3903, + "theoretical_loss": 3.5030703652211947, + "tokens_seen": 1564672000 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002656369107321966, + "loss": 1.3385, + "theoretical_loss": 3.5030575333738225, + "tokens_seen": 1564737536 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002656268806419258, + "loss": 1.4558, + "theoretical_loss": 3.5030447022143507, + "tokens_seen": 1564803072 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026561685055165497, + "loss": 1.3803, + "theoretical_loss": 3.5030318717427145, + "tokens_seen": 1564868608 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026560682046138415, + "loss": 1.4027, + "theoretical_loss": 3.503019041958848, + "tokens_seen": 1564934144 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026559679037111333, + "loss": 1.41, + "theoretical_loss": 3.5030062128626853, + "tokens_seen": 1564999680 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026558676028084257, + "loss": 1.3546, + "theoretical_loss": 3.502993384454161, + "tokens_seen": 1565065216 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002655767301905717, + "loss": 1.3538, + "theoretical_loss": 3.5029805567332097, + "tokens_seen": 1565130752 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026556670010030093, + "loss": 1.3803, + "theoretical_loss": 3.5029677296997646, + "tokens_seen": 1565196288 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002655566700100301, + "loss": 1.3936, + "theoretical_loss": 3.5029549033537615, + "tokens_seen": 1565261824 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002655466399197593, + "loss": 1.4229, + "theoretical_loss": 3.5029420776951343, + "tokens_seen": 1565327360 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002655366098294885, + "loss": 1.2663, + "theoretical_loss": 3.502929252723817, + "tokens_seen": 1565392896 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026552657973921766, + "loss": 1.4464, + "theoretical_loss": 3.502916428439744, + "tokens_seen": 1565458432 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026551654964894684, + "loss": 1.2956, + "theoretical_loss": 3.5029036048428503, + "tokens_seen": 1565523968 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026550651955867607, + "loss": 1.4448, + "theoretical_loss": 3.50289078193307, + "tokens_seen": 1565589504 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002654964894684052, + "loss": 1.4149, + "theoretical_loss": 3.5028779597103377, + "tokens_seen": 1565655040 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026548645937813443, + "loss": 1.2612, + "theoretical_loss": 3.5028651381745872, + "tokens_seen": 1565720576 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026547642928786356, + "loss": 1.3533, + "theoretical_loss": 3.502852317325753, + "tokens_seen": 1565786112 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002654663991975928, + "loss": 1.3838, + "theoretical_loss": 3.502839497163771, + "tokens_seen": 1565851648 + }, + { + "epoch": 0.47, + "learning_rate": 0.000265456369107322, + "loss": 1.2207, + "theoretical_loss": 3.502826677688574, + "tokens_seen": 1565917184 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026544633901705116, + "loss": 1.3462, + "theoretical_loss": 3.502813858900097, + "tokens_seen": 1565982720 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026543630892678034, + "loss": 1.3491, + "theoretical_loss": 3.502801040798275, + "tokens_seen": 1566048256 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002654262788365096, + "loss": 1.3713, + "theoretical_loss": 3.502788223383041, + "tokens_seen": 1566113792 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002654162487462387, + "loss": 1.3559, + "theoretical_loss": 3.5027754066543313, + "tokens_seen": 1566179328 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026540621865596794, + "loss": 1.352, + "theoretical_loss": 3.502762590612079, + "tokens_seen": 1566244864 + }, + { + "epoch": 0.47, + "objective/train/advantage_avg": -0.0007116802153177559, + "objective/train/docs_used": 888126, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.904843330383301, + "objective/train/original_loss": 2.9048428535461426, + "objective/train/theoretical_loss": 3.5027497752562198, + "objective/train/tokens_used": 1586770400, + "objective/train/value_avg": -0.0059051513671875, + "objective/train/value_loss": 9.157808381132782e-05, + "objective/train/value_max": -0.0003528594970703125, + "objective/train/value_min": -0.080078125, + "objective/train/value_reward_corr": 0.3667091171795949, + "objective/train/value_std": 0.0044403076171875, + "objective/train/weight_avg": 0.996778130531311, + "objective/train/weighted_lm_loss": 2.8926773071289062, + "objective/train/weights_max": 1.8987709283828735, + "objective/train/weights_min": 0.20084697008132935, + "theoretical_loss": 3.5027497752562198, + "tokens_seen": 1566310400 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026539618856569707, + "loss": 1.3626, + "theoretical_loss": 3.5027497752562198, + "tokens_seen": 1566310400 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002653861584754263, + "loss": 1.3244, + "theoretical_loss": 3.502736960586687, + "tokens_seen": 1566375936 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002653761283851555, + "loss": 1.3356, + "theoretical_loss": 3.5027241466034167, + "tokens_seen": 1566441472 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026536609829488466, + "loss": 1.4297, + "theoretical_loss": 3.502711333306342, + "tokens_seen": 1566507008 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026535606820461384, + "loss": 1.3445, + "theoretical_loss": 3.502698520695398, + "tokens_seen": 1566572544 + }, + { + "epoch": 0.47, + "learning_rate": 0.000265346038114343, + "loss": 1.3588, + "theoretical_loss": 3.502685708770519, + "tokens_seen": 1566638080 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002653360080240722, + "loss": 1.1612, + "theoretical_loss": 3.50267289753164, + "tokens_seen": 1566703616 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026532597793380144, + "loss": 1.1448, + "theoretical_loss": 3.5026600869786955, + "tokens_seen": 1566769152 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026531594784353057, + "loss": 1.2843, + "theoretical_loss": 3.50264727711162, + "tokens_seen": 1566834688 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002653059177532598, + "loss": 1.3391, + "theoretical_loss": 3.502634467930348, + "tokens_seen": 1566900224 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026529588766298893, + "loss": 1.357, + "theoretical_loss": 3.502621659434814, + "tokens_seen": 1566965760 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026528585757271817, + "loss": 1.3193, + "theoretical_loss": 3.5026088516249523, + "tokens_seen": 1567031296 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026527582748244735, + "loss": 1.2233, + "theoretical_loss": 3.5025960445006987, + "tokens_seen": 1567096832 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026526579739217653, + "loss": 1.3142, + "theoretical_loss": 3.5025832380619866, + "tokens_seen": 1567162368 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002652557673019057, + "loss": 1.2981, + "theoretical_loss": 3.502570432308751, + "tokens_seen": 1567227904 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026524573721163494, + "loss": 1.3642, + "theoretical_loss": 3.502557627240927, + "tokens_seen": 1567293440 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026523570712136407, + "loss": 1.3621, + "theoretical_loss": 3.5025448228584493, + "tokens_seen": 1567358976 + }, + { + "epoch": 0.47, + "learning_rate": 0.0002652256770310933, + "loss": 1.215, + "theoretical_loss": 3.5025320191612517, + "tokens_seen": 1567424512 + }, + { + "epoch": 0.47, + "learning_rate": 0.00026521564694082243, + "loss": 1.3545, + "theoretical_loss": 3.5025192161492695, + "tokens_seen": 1567490048 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026520561685055167, + "loss": 1.3379, + "theoretical_loss": 3.502506413822437, + "tokens_seen": 1567555584 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026519558676028085, + "loss": 1.385, + "theoretical_loss": 3.5024936121806896, + "tokens_seen": 1567621120 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026518555667001003, + "loss": 1.3226, + "theoretical_loss": 3.5024808112239616, + "tokens_seen": 1567686656 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002651755265797392, + "loss": 1.4284, + "theoretical_loss": 3.5024680109521875, + "tokens_seen": 1567752192 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002651654964894684, + "loss": 1.3614, + "theoretical_loss": 3.5024552113653025, + "tokens_seen": 1567817728 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002651554663991976, + "loss": 1.3653, + "theoretical_loss": 3.5024424124632403, + "tokens_seen": 1567883264 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": 3.934768756153062e-05, + "objective/train/docs_used": 889384, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0271852016448975, + "objective/train/original_loss": 3.0271856784820557, + "objective/train/theoretical_loss": 3.502429614245937, + "objective/train/tokens_used": 1588408800, + "objective/train/value_avg": -0.0052490234375, + "objective/train/value_loss": 6.379525439115241e-05, + "objective/train/value_max": -0.0004355907440185547, + "objective/train/value_min": -0.0482177734375, + "objective/train/value_reward_corr": 0.3092044175769241, + "objective/train/value_std": 0.0036334991455078125, + "objective/train/weight_avg": 1.0033270120620728, + "objective/train/weighted_lm_loss": 3.039557933807373, + "objective/train/weights_max": 1.6195976734161377, + "objective/train/weights_min": 0.3627690076828003, + "theoretical_loss": 3.502429614245937, + "tokens_seen": 1567948800 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002651454363089268, + "loss": 1.3548, + "theoretical_loss": 3.502429614245937, + "tokens_seen": 1567948800 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026513540621865594, + "loss": 1.4365, + "theoretical_loss": 3.5024168167133265, + "tokens_seen": 1568014336 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026512537612838517, + "loss": 1.3325, + "theoretical_loss": 3.5024040198653434, + "tokens_seen": 1568079872 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002651153460381143, + "loss": 1.3914, + "theoretical_loss": 3.502391223701923, + "tokens_seen": 1568145408 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026510531594784353, + "loss": 1.4251, + "theoretical_loss": 3.5023784282230004, + "tokens_seen": 1568210944 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002650952858575727, + "loss": 1.3719, + "theoretical_loss": 3.502365633428509, + "tokens_seen": 1568276480 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002650852557673019, + "loss": 1.2628, + "theoretical_loss": 3.5023528393183847, + "tokens_seen": 1568342016 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002650752256770311, + "loss": 1.3444, + "theoretical_loss": 3.502340045892563, + "tokens_seen": 1568407552 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002650651955867603, + "loss": 1.272, + "theoretical_loss": 3.502327253150977, + "tokens_seen": 1568473088 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026505516549648944, + "loss": 1.4288, + "theoretical_loss": 3.5023144610935617, + "tokens_seen": 1568538624 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002650451354062187, + "loss": 1.3775, + "theoretical_loss": 3.5023016697202536, + "tokens_seen": 1568604160 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002650351053159478, + "loss": 1.3336, + "theoretical_loss": 3.502288879030986, + "tokens_seen": 1568669696 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026502507522567704, + "loss": 1.2654, + "theoretical_loss": 3.502276089025694, + "tokens_seen": 1568735232 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002650150451354062, + "loss": 1.3305, + "theoretical_loss": 3.502263299704313, + "tokens_seen": 1568800768 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002650050150451354, + "loss": 1.3899, + "theoretical_loss": 3.502250511066777, + "tokens_seen": 1568866304 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026499498495486464, + "loss": 1.4403, + "theoretical_loss": 3.502237723113022, + "tokens_seen": 1568931840 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026498495486459376, + "loss": 1.3289, + "theoretical_loss": 3.502224935842982, + "tokens_seen": 1568997376 + }, + { + "epoch": 0.48, + "learning_rate": 0.000264974924774323, + "loss": 1.3707, + "theoretical_loss": 3.502212149256592, + "tokens_seen": 1569062912 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002649648946840522, + "loss": 1.3317, + "theoretical_loss": 3.502199363353787, + "tokens_seen": 1569128448 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026495486459378136, + "loss": 1.3282, + "theoretical_loss": 3.502186578134502, + "tokens_seen": 1569193984 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026494483450351054, + "loss": 1.3515, + "theoretical_loss": 3.502173793598672, + "tokens_seen": 1569259520 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002649348044132398, + "loss": 1.3836, + "theoretical_loss": 3.5021610097462315, + "tokens_seen": 1569325056 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002649247743229689, + "loss": 1.3274, + "theoretical_loss": 3.502148226577116, + "tokens_seen": 1569390592 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026491474423269814, + "loss": 1.3259, + "theoretical_loss": 3.5021354440912607, + "tokens_seen": 1569456128 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026490471414242727, + "loss": 1.3757, + "theoretical_loss": 3.502122662288599, + "tokens_seen": 1569521664 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.0020706483628600836, + "objective/train/docs_used": 890057, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.83276104927063, + "objective/train/original_loss": 2.832760810852051, + "objective/train/theoretical_loss": 3.5021098811690674, + "objective/train/tokens_used": 1590047200, + "objective/train/value_avg": -0.005374908447265625, + "objective/train/value_loss": 0.00010187853331444785, + "objective/train/value_max": -0.0004992485046386719, + "objective/train/value_min": -0.08587646484375, + "objective/train/value_reward_corr": 0.3243144926900965, + "objective/train/value_std": 0.004486083984375, + "objective/train/weight_avg": 0.9840491414070129, + "objective/train/weighted_lm_loss": 2.828653573989868, + "objective/train/weights_max": 1.6096062660217285, + "objective/train/weights_min": 0.5465269088745117, + "theoretical_loss": 3.5021098811690674, + "tokens_seen": 1569587200 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002648946840521565, + "loss": 1.2972, + "theoretical_loss": 3.5021098811690674, + "tokens_seen": 1569587200 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002648846539618857, + "loss": 1.3045, + "theoretical_loss": 3.5020971007326, + "tokens_seen": 1569652736 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026487462387161486, + "loss": 1.3109, + "theoretical_loss": 3.5020843209791326, + "tokens_seen": 1569718272 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026486459378134404, + "loss": 1.3645, + "theoretical_loss": 3.5020715419085997, + "tokens_seen": 1569783808 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002648545636910732, + "loss": 1.3468, + "theoretical_loss": 3.502058763520936, + "tokens_seen": 1569849344 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002648445336008024, + "loss": 1.2911, + "theoretical_loss": 3.5020459858160766, + "tokens_seen": 1569914880 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026483450351053164, + "loss": 1.4324, + "theoretical_loss": 3.502033208793957, + "tokens_seen": 1569980416 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026482447342026077, + "loss": 1.2728, + "theoretical_loss": 3.502020432454512, + "tokens_seen": 1570045952 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026481444332999, + "loss": 1.3595, + "theoretical_loss": 3.5020076567976766, + "tokens_seen": 1570111488 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026480441323971913, + "loss": 1.202, + "theoretical_loss": 3.5019948818233857, + "tokens_seen": 1570177024 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026479438314944837, + "loss": 1.3474, + "theoretical_loss": 3.5019821075315747, + "tokens_seen": 1570242560 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026478435305917755, + "loss": 1.3537, + "theoretical_loss": 3.5019693339221787, + "tokens_seen": 1570308096 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026477432296890673, + "loss": 1.3543, + "theoretical_loss": 3.501956560995132, + "tokens_seen": 1570373632 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002647642928786359, + "loss": 1.376, + "theoretical_loss": 3.5019437887503697, + "tokens_seen": 1570439168 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026475426278836514, + "loss": 1.3574, + "theoretical_loss": 3.501931017187828, + "tokens_seen": 1570504704 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026474423269809427, + "loss": 1.3996, + "theoretical_loss": 3.5019182463074414, + "tokens_seen": 1570570240 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002647342026078235, + "loss": 1.3654, + "theoretical_loss": 3.501905476109145, + "tokens_seen": 1570635776 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026472417251755263, + "loss": 1.2171, + "theoretical_loss": 3.5018927065928733, + "tokens_seen": 1570701312 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026471414242728187, + "loss": 1.3206, + "theoretical_loss": 3.501879937758562, + "tokens_seen": 1570766848 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026470411233701105, + "loss": 1.3441, + "theoretical_loss": 3.5018671696061467, + "tokens_seen": 1570832384 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026469408224674023, + "loss": 1.2668, + "theoretical_loss": 3.501854402135562, + "tokens_seen": 1570897920 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002646840521564694, + "loss": 1.4572, + "theoretical_loss": 3.501841635346743, + "tokens_seen": 1570963456 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002646740220661986, + "loss": 1.2746, + "theoretical_loss": 3.501828869239625, + "tokens_seen": 1571028992 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002646639919759278, + "loss": 1.384, + "theoretical_loss": 3.501816103814143, + "tokens_seen": 1571094528 + }, + { + "epoch": 0.48, + "learning_rate": 0.000264653961885657, + "loss": 1.3718, + "theoretical_loss": 3.501803339070232, + "tokens_seen": 1571160064 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.0003150742850266397, + "objective/train/docs_used": 891281, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.054901361465454, + "objective/train/original_loss": 3.054901361465454, + "objective/train/theoretical_loss": 3.5017905750078278, + "objective/train/tokens_used": 1591685600, + "objective/train/value_avg": -0.0050506591796875, + "objective/train/value_loss": 8.325291855726391e-05, + "objective/train/value_max": -0.0004601478576660156, + "objective/train/value_min": -0.0469970703125, + "objective/train/value_reward_corr": 0.28508977818832604, + "objective/train/value_std": 0.003406524658203125, + "objective/train/weight_avg": 1.0005263090133667, + "objective/train/weighted_lm_loss": 3.0481371879577637, + "objective/train/weights_max": 1.3878540992736816, + "objective/train/weights_min": 0.14313025772571564, + "theoretical_loss": 3.5017905750078278, + "tokens_seen": 1571225600 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026464393179538614, + "loss": 1.4608, + "theoretical_loss": 3.5017905750078278, + "tokens_seen": 1571225600 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026463390170511537, + "loss": 1.371, + "theoretical_loss": 3.501777811626865, + "tokens_seen": 1571291136 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002646238716148445, + "loss": 1.3543, + "theoretical_loss": 3.50176504892728, + "tokens_seen": 1571356672 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026461384152457373, + "loss": 1.355, + "theoretical_loss": 3.5017522869090056, + "tokens_seen": 1571422208 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002646038114343029, + "loss": 1.3497, + "theoretical_loss": 3.5017395255719794, + "tokens_seen": 1571487744 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002645937813440321, + "loss": 1.2604, + "theoretical_loss": 3.5017267649161354, + "tokens_seen": 1571553280 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002645837512537613, + "loss": 1.3621, + "theoretical_loss": 3.5017140049414097, + "tokens_seen": 1571618816 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002645737211634905, + "loss": 1.5525, + "theoretical_loss": 3.5017012456477365, + "tokens_seen": 1571684352 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026456369107321964, + "loss": 1.2741, + "theoretical_loss": 3.5016884870350515, + "tokens_seen": 1571749888 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002645536609829489, + "loss": 1.4596, + "theoretical_loss": 3.5016757291032903, + "tokens_seen": 1571815424 + }, + { + "epoch": 0.48, + "learning_rate": 0.000264543630892678, + "loss": 1.2072, + "theoretical_loss": 3.5016629718523875, + "tokens_seen": 1571880960 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026453360080240724, + "loss": 1.4041, + "theoretical_loss": 3.501650215282279, + "tokens_seen": 1571946496 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002645235707121364, + "loss": 1.4034, + "theoretical_loss": 3.5016374593928994, + "tokens_seen": 1572012032 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002645135406218656, + "loss": 1.3711, + "theoretical_loss": 3.5016247041841853, + "tokens_seen": 1572077568 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002645035105315948, + "loss": 1.2636, + "theoretical_loss": 3.5016119496560707, + "tokens_seen": 1572143104 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026449348044132396, + "loss": 1.3302, + "theoretical_loss": 3.5015991958084913, + "tokens_seen": 1572208640 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026448345035105314, + "loss": 1.2719, + "theoretical_loss": 3.5015864426413827, + "tokens_seen": 1572274176 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002644734202607824, + "loss": 1.4532, + "theoretical_loss": 3.5015736901546797, + "tokens_seen": 1572339712 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002644633901705115, + "loss": 1.4582, + "theoretical_loss": 3.501560938348318, + "tokens_seen": 1572405248 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026445336008024074, + "loss": 1.3575, + "theoretical_loss": 3.501548187222233, + "tokens_seen": 1572470784 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026444332998996987, + "loss": 1.3295, + "theoretical_loss": 3.50153543677636, + "tokens_seen": 1572536320 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002644332998996991, + "loss": 1.3709, + "theoretical_loss": 3.5015226870106346, + "tokens_seen": 1572601856 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002644232698094283, + "loss": 1.3453, + "theoretical_loss": 3.5015099379249914, + "tokens_seen": 1572667392 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026441323971915747, + "loss": 1.2457, + "theoretical_loss": 3.501497189519367, + "tokens_seen": 1572732928 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026440320962888665, + "loss": 1.3685, + "theoretical_loss": 3.5014844417936954, + "tokens_seen": 1572798464 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.0006728452863171697, + "objective/train/docs_used": 891900, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5131943225860596, + "objective/train/original_loss": 2.5131943225860596, + "objective/train/theoretical_loss": 3.501471694747913, + "objective/train/tokens_used": 1593324000, + "objective/train/value_avg": -0.005130767822265625, + "objective/train/value_loss": 5.063635398983024e-05, + "objective/train/value_max": -0.00033664703369140625, + "objective/train/value_min": -0.064208984375, + "objective/train/value_reward_corr": 0.3775266787730332, + "objective/train/value_std": 0.00421142578125, + "objective/train/weight_avg": 0.9956116080284119, + "objective/train/weighted_lm_loss": 2.5161185264587402, + "objective/train/weights_max": 1.8025093078613281, + "objective/train/weights_min": 0.20441025495529175, + "theoretical_loss": 3.501471694747913, + "tokens_seen": 1572864000 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002643931795386159, + "loss": 1.3392, + "theoretical_loss": 3.501471694747913, + "tokens_seen": 1572864000 + }, + { + "epoch": 0.48, + "learning_rate": 0.000264383149448345, + "loss": 1.3419, + "theoretical_loss": 3.5014589483819547, + "tokens_seen": 1572929536 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026437311935807424, + "loss": 1.3933, + "theoretical_loss": 3.5014462026957567, + "tokens_seen": 1572995072 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026436308926780337, + "loss": 1.3684, + "theoretical_loss": 3.501433457689253, + "tokens_seen": 1573060608 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002643530591775326, + "loss": 1.3899, + "theoretical_loss": 3.5014207133623803, + "tokens_seen": 1573126144 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002643430290872618, + "loss": 1.4021, + "theoretical_loss": 3.501407969715074, + "tokens_seen": 1573191680 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026433299899699097, + "loss": 1.4884, + "theoretical_loss": 3.501395226747269, + "tokens_seen": 1573257216 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026432296890672015, + "loss": 1.4393, + "theoretical_loss": 3.501382484458901, + "tokens_seen": 1573322752 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026431293881644933, + "loss": 1.4102, + "theoretical_loss": 3.501369742849905, + "tokens_seen": 1573388288 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002643029087261785, + "loss": 1.2635, + "theoretical_loss": 3.5013570019202174, + "tokens_seen": 1573453824 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026429287863590775, + "loss": 1.4508, + "theoretical_loss": 3.5013442616697734, + "tokens_seen": 1573519360 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002642828485456369, + "loss": 1.3878, + "theoretical_loss": 3.501331522098508, + "tokens_seen": 1573584896 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002642728184553661, + "loss": 1.3426, + "theoretical_loss": 3.501318783206357, + "tokens_seen": 1573650432 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026426278836509534, + "loss": 1.3757, + "theoretical_loss": 3.501306044993256, + "tokens_seen": 1573715968 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026425275827482447, + "loss": 1.4284, + "theoretical_loss": 3.5012933074591404, + "tokens_seen": 1573781504 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002642427281845537, + "loss": 1.3864, + "theoretical_loss": 3.501280570603946, + "tokens_seen": 1573847040 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026423269809428283, + "loss": 1.233, + "theoretical_loss": 3.5012678344276082, + "tokens_seen": 1573912576 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026422266800401207, + "loss": 1.3263, + "theoretical_loss": 3.501255098930062, + "tokens_seen": 1573978112 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026421263791374125, + "loss": 1.4472, + "theoretical_loss": 3.501242364111244, + "tokens_seen": 1574043648 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026420260782347043, + "loss": 1.3673, + "theoretical_loss": 3.501229629971089, + "tokens_seen": 1574109184 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002641925777331996, + "loss": 1.455, + "theoretical_loss": 3.5012168965095327, + "tokens_seen": 1574174720 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002641825476429288, + "loss": 1.3965, + "theoretical_loss": 3.501204163726511, + "tokens_seen": 1574240256 + }, + { + "epoch": 0.48, + "learning_rate": 0.000264172517552658, + "loss": 1.3159, + "theoretical_loss": 3.5011914316219594, + "tokens_seen": 1574305792 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002641624874623872, + "loss": 1.4454, + "theoretical_loss": 3.5011787001958137, + "tokens_seen": 1574371328 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026415245737211634, + "loss": 1.3698, + "theoretical_loss": 3.5011659694480084, + "tokens_seen": 1574436864 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.00010118311183759943, + "objective/train/docs_used": 893256, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8394954204559326, + "objective/train/original_loss": 2.8394954204559326, + "objective/train/theoretical_loss": 3.5011532393784806, + "objective/train/tokens_used": 1594962400, + "objective/train/value_avg": -0.0062255859375, + "objective/train/value_loss": 6.384016160154715e-05, + "objective/train/value_max": -0.0004322528839111328, + "objective/train/value_min": -0.1026611328125, + "objective/train/value_reward_corr": 0.2569340509643822, + "objective/train/value_std": 0.0048675537109375, + "objective/train/weight_avg": 1.002123475074768, + "objective/train/weighted_lm_loss": 2.8464136123657227, + "objective/train/weights_max": 2.7915899753570557, + "objective/train/weights_min": 0.4419669806957245, + "theoretical_loss": 3.5011532393784806, + "tokens_seen": 1574502400 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026414242728184557, + "loss": 1.3269, + "theoretical_loss": 3.5011532393784806, + "tokens_seen": 1574502400 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002641323971915747, + "loss": 1.432, + "theoretical_loss": 3.501140509987165, + "tokens_seen": 1574567936 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026412236710130393, + "loss": 1.4297, + "theoretical_loss": 3.501127781273998, + "tokens_seen": 1574633472 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002641123370110331, + "loss": 1.4173, + "theoretical_loss": 3.501115053238914, + "tokens_seen": 1574699008 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002641023069207623, + "loss": 1.3417, + "theoretical_loss": 3.50110232588185, + "tokens_seen": 1574764544 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002640922768304915, + "loss": 1.3375, + "theoretical_loss": 3.501089599202741, + "tokens_seen": 1574830080 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002640822467402207, + "loss": 1.4817, + "theoretical_loss": 3.5010768732015225, + "tokens_seen": 1574895616 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026407221664994984, + "loss": 1.3238, + "theoretical_loss": 3.5010641478781306, + "tokens_seen": 1574961152 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002640621865596791, + "loss": 1.2787, + "theoretical_loss": 3.501051423232501, + "tokens_seen": 1575026688 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002640521564694082, + "loss": 1.3359, + "theoretical_loss": 3.5010386992645692, + "tokens_seen": 1575092224 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026404212637913744, + "loss": 1.3603, + "theoretical_loss": 3.5010259759742715, + "tokens_seen": 1575157760 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002640320962888666, + "loss": 1.4036, + "theoretical_loss": 3.501013253361542, + "tokens_seen": 1575223296 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002640220661985958, + "loss": 1.4387, + "theoretical_loss": 3.5010005314263184, + "tokens_seen": 1575288832 + }, + { + "epoch": 0.48, + "learning_rate": 0.000264012036108325, + "loss": 1.3464, + "theoretical_loss": 3.500987810168535, + "tokens_seen": 1575354368 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026400200601805416, + "loss": 1.3877, + "theoretical_loss": 3.5009750895881284, + "tokens_seen": 1575419904 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026399197592778334, + "loss": 1.4467, + "theoretical_loss": 3.500962369685034, + "tokens_seen": 1575485440 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002639819458375126, + "loss": 1.3496, + "theoretical_loss": 3.5009496504591873, + "tokens_seen": 1575550976 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002639719157472417, + "loss": 1.3166, + "theoretical_loss": 3.500936931910525, + "tokens_seen": 1575616512 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026396188565697094, + "loss": 1.2948, + "theoretical_loss": 3.500924214038982, + "tokens_seen": 1575682048 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026395185556670007, + "loss": 1.4041, + "theoretical_loss": 3.5009114968444943, + "tokens_seen": 1575747584 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002639418254764293, + "loss": 1.247, + "theoretical_loss": 3.500898780326998, + "tokens_seen": 1575813120 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002639317953861585, + "loss": 1.3197, + "theoretical_loss": 3.5008860644864286, + "tokens_seen": 1575878656 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026392176529588767, + "loss": 1.478, + "theoretical_loss": 3.500873349322722, + "tokens_seen": 1575944192 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026391173520561685, + "loss": 1.3714, + "theoretical_loss": 3.5008606348358136, + "tokens_seen": 1576009728 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002639017051153461, + "loss": 1.452, + "theoretical_loss": 3.50084792102564, + "tokens_seen": 1576075264 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.0007392663974314928, + "objective/train/docs_used": 893728, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7269673347473145, + "objective/train/original_loss": 2.7269668579101562, + "objective/train/theoretical_loss": 3.5008352078921368, + "objective/train/tokens_used": 1596600800, + "objective/train/value_avg": -0.00580596923828125, + "objective/train/value_loss": 9.965552453650162e-05, + "objective/train/value_max": -0.0004897117614746094, + "objective/train/value_min": -0.08648681640625, + "objective/train/value_reward_corr": 0.5687078978669466, + "objective/train/value_std": 0.005802154541015625, + "objective/train/weight_avg": 0.9971452355384827, + "objective/train/weighted_lm_loss": 2.722919464111328, + "objective/train/weights_max": 2.0200555324554443, + "objective/train/weights_min": 0.31433284282684326, + "theoretical_loss": 3.5008352078921368, + "tokens_seen": 1576140800 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002638916750250752, + "loss": 1.4517, + "theoretical_loss": 3.5008352078921368, + "tokens_seen": 1576140800 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026388164493480444, + "loss": 1.4348, + "theoretical_loss": 3.50082249543524, + "tokens_seen": 1576206336 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026387161484453357, + "loss": 1.4083, + "theoretical_loss": 3.5008097836548844, + "tokens_seen": 1576271872 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002638615847542628, + "loss": 1.2952, + "theoretical_loss": 3.5007970725510074, + "tokens_seen": 1576337408 + }, + { + "epoch": 0.48, + "learning_rate": 0.000263851554663992, + "loss": 1.4174, + "theoretical_loss": 3.500784362123544, + "tokens_seen": 1576402944 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026384152457372117, + "loss": 1.4202, + "theoretical_loss": 3.50077165237243, + "tokens_seen": 1576468480 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026383149448345035, + "loss": 1.3715, + "theoretical_loss": 3.5007589432976016, + "tokens_seen": 1576534016 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026382146439317953, + "loss": 1.3104, + "theoretical_loss": 3.5007462348989944, + "tokens_seen": 1576599552 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002638114343029087, + "loss": 1.3808, + "theoretical_loss": 3.500733527176545, + "tokens_seen": 1576665088 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026380140421263795, + "loss": 1.3454, + "theoretical_loss": 3.500720820130189, + "tokens_seen": 1576730624 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002637913741223671, + "loss": 1.3448, + "theoretical_loss": 3.5007081137598615, + "tokens_seen": 1576796160 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002637813440320963, + "loss": 1.3104, + "theoretical_loss": 3.5006954080655, + "tokens_seen": 1576861696 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002637713139418255, + "loss": 1.3238, + "theoretical_loss": 3.5006827030470387, + "tokens_seen": 1576927232 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026376128385155467, + "loss": 1.1956, + "theoretical_loss": 3.5006699987044154, + "tokens_seen": 1576992768 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026375125376128385, + "loss": 1.2722, + "theoretical_loss": 3.5006572950375645, + "tokens_seen": 1577058304 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026374122367101303, + "loss": 1.4192, + "theoretical_loss": 3.500644592046423, + "tokens_seen": 1577123840 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002637311935807422, + "loss": 1.3332, + "theoretical_loss": 3.5006318897309256, + "tokens_seen": 1577189376 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026372116349047145, + "loss": 1.4668, + "theoretical_loss": 3.50061918809101, + "tokens_seen": 1577254912 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002637111334002006, + "loss": 1.3255, + "theoretical_loss": 3.5006064871266114, + "tokens_seen": 1577320448 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002637011033099298, + "loss": 1.4031, + "theoretical_loss": 3.5005937868376655, + "tokens_seen": 1577385984 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026369107321965894, + "loss": 1.3065, + "theoretical_loss": 3.5005810872241083, + "tokens_seen": 1577451520 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002636810431293882, + "loss": 1.311, + "theoretical_loss": 3.5005683882858767, + "tokens_seen": 1577517056 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026367101303911736, + "loss": 1.3516, + "theoretical_loss": 3.500555690022906, + "tokens_seen": 1577582592 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026366098294884654, + "loss": 1.3802, + "theoretical_loss": 3.5005429924351317, + "tokens_seen": 1577648128 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002636509528585757, + "loss": 1.4615, + "theoretical_loss": 3.5005302955224913, + "tokens_seen": 1577713664 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.0003648870042525232, + "objective/train/docs_used": 894313, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.180392026901245, + "objective/train/original_loss": 3.180392026901245, + "objective/train/theoretical_loss": 3.50051759928492, + "objective/train/tokens_used": 1598239200, + "objective/train/value_avg": -0.005222320556640625, + "objective/train/value_loss": 7.304869359359145e-05, + "objective/train/value_max": -0.000396728515625, + "objective/train/value_min": -0.10174560546875, + "objective/train/value_reward_corr": 0.47619207933379, + "objective/train/value_std": 0.005016326904296875, + "objective/train/weight_avg": 0.9995678067207336, + "objective/train/weighted_lm_loss": 3.182648181915283, + "objective/train/weights_max": 1.9339051246643066, + "objective/train/weights_min": 0.08368801325559616, + "theoretical_loss": 3.50051759928492, + "tokens_seen": 1577779200 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002636409227683049, + "loss": 1.3673, + "theoretical_loss": 3.50051759928492, + "tokens_seen": 1577779200 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002636308926780341, + "loss": 1.3035, + "theoretical_loss": 3.5005049037223537, + "tokens_seen": 1577844736 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002636208625877633, + "loss": 1.426, + "theoretical_loss": 3.500492208834729, + "tokens_seen": 1577910272 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026361083249749244, + "loss": 1.2842, + "theoretical_loss": 3.500479514621982, + "tokens_seen": 1577975808 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002636008024072217, + "loss": 1.3355, + "theoretical_loss": 3.500466821084048, + "tokens_seen": 1578041344 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026359077231695086, + "loss": 1.417, + "theoretical_loss": 3.5004541282208637, + "tokens_seen": 1578106880 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026358074222668004, + "loss": 1.2724, + "theoretical_loss": 3.5004414360323652, + "tokens_seen": 1578172416 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002635707121364092, + "loss": 1.4332, + "theoretical_loss": 3.500428744518489, + "tokens_seen": 1578237952 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002635606820461384, + "loss": 1.4561, + "theoretical_loss": 3.5004160536791704, + "tokens_seen": 1578303488 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002635506519558676, + "loss": 1.3226, + "theoretical_loss": 3.500403363514346, + "tokens_seen": 1578369024 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002635406218655968, + "loss": 1.4861, + "theoretical_loss": 3.500390674023952, + "tokens_seen": 1578434560 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026353059177532595, + "loss": 1.3359, + "theoretical_loss": 3.5003779852079244, + "tokens_seen": 1578500096 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002635205616850552, + "loss": 1.4265, + "theoretical_loss": 3.5003652970661996, + "tokens_seen": 1578565632 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026351053159478436, + "loss": 1.3266, + "theoretical_loss": 3.5003526095987136, + "tokens_seen": 1578631168 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026350050150451354, + "loss": 1.3186, + "theoretical_loss": 3.500339922805402, + "tokens_seen": 1578696704 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002634904714142428, + "loss": 1.3493, + "theoretical_loss": 3.5003272366862026, + "tokens_seen": 1578762240 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002634804413239719, + "loss": 1.4522, + "theoretical_loss": 3.5003145512410496, + "tokens_seen": 1578827776 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026347041123370114, + "loss": 1.3528, + "theoretical_loss": 3.5003018664698806, + "tokens_seen": 1578893312 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026346038114343027, + "loss": 1.3888, + "theoretical_loss": 3.500289182372631, + "tokens_seen": 1578958848 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002634503510531595, + "loss": 1.497, + "theoretical_loss": 3.5002764989492383, + "tokens_seen": 1579024384 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002634403209628887, + "loss": 1.4617, + "theoretical_loss": 3.500263816199637, + "tokens_seen": 1579089920 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026343029087261787, + "loss": 1.4062, + "theoretical_loss": 3.500251134123765, + "tokens_seen": 1579155456 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026342026078234705, + "loss": 1.3005, + "theoretical_loss": 3.500238452721557, + "tokens_seen": 1579220992 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002634102306920763, + "loss": 1.3387, + "theoretical_loss": 3.500225771992951, + "tokens_seen": 1579286528 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002634002006018054, + "loss": 1.4037, + "theoretical_loss": 3.500213091937881, + "tokens_seen": 1579352064 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": 0.00021756874048151076, + "objective/train/docs_used": 895615, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.923929214477539, + "objective/train/original_loss": 2.9239296913146973, + "objective/train/theoretical_loss": 3.5002004125562856, + "objective/train/tokens_used": 1599877600, + "objective/train/value_avg": -0.004596710205078125, + "objective/train/value_loss": 4.8557751142652705e-05, + "objective/train/value_max": -0.0003101825714111328, + "objective/train/value_min": -0.041839599609375, + "objective/train/value_reward_corr": 0.30846228786991975, + "objective/train/value_std": 0.003353118896484375, + "objective/train/weight_avg": 1.0044950246810913, + "objective/train/weighted_lm_loss": 2.943798065185547, + "objective/train/weights_max": 1.3465936183929443, + "objective/train/weights_min": 0.5595852732658386, + "theoretical_loss": 3.5002004125562856, + "tokens_seen": 1579417600 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026339017051153464, + "loss": 1.2969, + "theoretical_loss": 3.5002004125562856, + "tokens_seen": 1579417600 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026338014042126377, + "loss": 1.3767, + "theoretical_loss": 3.500187733848099, + "tokens_seen": 1579483136 + }, + { + "epoch": 0.48, + "learning_rate": 0.000263370110330993, + "loss": 1.3592, + "theoretical_loss": 3.5001750558132594, + "tokens_seen": 1579548672 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002633600802407222, + "loss": 1.3403, + "theoretical_loss": 3.5001623784517024, + "tokens_seen": 1579614208 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026335005015045137, + "loss": 1.3619, + "theoretical_loss": 3.5001497017633634, + "tokens_seen": 1579679744 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026334002006018055, + "loss": 1.3223, + "theoretical_loss": 3.50013702574818, + "tokens_seen": 1579745280 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026332998996990973, + "loss": 1.3748, + "theoretical_loss": 3.5001243504060877, + "tokens_seen": 1579810816 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002633199598796389, + "loss": 1.41, + "theoretical_loss": 3.5001116757370236, + "tokens_seen": 1579876352 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026330992978936815, + "loss": 1.3575, + "theoretical_loss": 3.5000990017409235, + "tokens_seen": 1579941888 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002632998996990973, + "loss": 1.2904, + "theoretical_loss": 3.5000863284177237, + "tokens_seen": 1580007424 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002632898696088265, + "loss": 1.4152, + "theoretical_loss": 3.500073655767361, + "tokens_seen": 1580072960 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002632798395185557, + "loss": 1.4022, + "theoretical_loss": 3.500060983789771, + "tokens_seen": 1580138496 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026326980942828487, + "loss": 1.3365, + "theoretical_loss": 3.500048312484891, + "tokens_seen": 1580204032 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026325977933801405, + "loss": 1.4232, + "theoretical_loss": 3.500035641852657, + "tokens_seen": 1580269568 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026324974924774323, + "loss": 1.3352, + "theoretical_loss": 3.5000229718930047, + "tokens_seen": 1580335104 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002632397191574724, + "loss": 1.3752, + "theoretical_loss": 3.5000103026058715, + "tokens_seen": 1580400640 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026322968906720165, + "loss": 1.3511, + "theoretical_loss": 3.4999976339911942, + "tokens_seen": 1580466176 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002632196589769308, + "loss": 1.3264, + "theoretical_loss": 3.499984966048908, + "tokens_seen": 1580531712 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026320962888666, + "loss": 1.4323, + "theoretical_loss": 3.4999722987789497, + "tokens_seen": 1580597248 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026319959879638914, + "loss": 1.2851, + "theoretical_loss": 3.4999596321812563, + "tokens_seen": 1580662784 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002631895687061184, + "loss": 1.3874, + "theoretical_loss": 3.4999469662557634, + "tokens_seen": 1580728320 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026317953861584756, + "loss": 1.2801, + "theoretical_loss": 3.4999343010024084, + "tokens_seen": 1580793856 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026316950852557674, + "loss": 1.2867, + "theoretical_loss": 3.499921636421127, + "tokens_seen": 1580859392 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002631594784353059, + "loss": 1.3904, + "theoretical_loss": 3.499908972511856, + "tokens_seen": 1580924928 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002631494483450351, + "loss": 1.3031, + "theoretical_loss": 3.499896309274532, + "tokens_seen": 1580990464 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -2.417980613245163e-05, + "objective/train/docs_used": 896135, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4363536834716797, + "objective/train/original_loss": 2.4363532066345215, + "objective/train/theoretical_loss": 3.499883646709091, + "objective/train/tokens_used": 1601516000, + "objective/train/value_avg": -0.005054473876953125, + "objective/train/value_loss": 5.092531137051992e-05, + "objective/train/value_max": -0.0002491474151611328, + "objective/train/value_min": -0.133544921875, + "objective/train/value_reward_corr": 0.2550498960264204, + "objective/train/value_std": 0.0037250518798828125, + "objective/train/weight_avg": 1.002130150794983, + "objective/train/weighted_lm_loss": 2.4383795261383057, + "objective/train/weights_max": 3.8017032146453857, + "objective/train/weights_min": 0.13762995600700378, + "theoretical_loss": 3.499883646709091, + "tokens_seen": 1581056000 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002631394182547643, + "loss": 1.2596, + "theoretical_loss": 3.499883646709091, + "tokens_seen": 1581056000 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002631293881644935, + "loss": 1.3154, + "theoretical_loss": 3.4998709848154697, + "tokens_seen": 1581121536 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026311935807422264, + "loss": 1.4116, + "theoretical_loss": 3.499858323593605, + "tokens_seen": 1581187072 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002631093279839519, + "loss": 1.3616, + "theoretical_loss": 3.4998456630434336, + "tokens_seen": 1581252608 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026309929789368106, + "loss": 1.3443, + "theoretical_loss": 3.499833003164891, + "tokens_seen": 1581318144 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026308926780341024, + "loss": 1.3016, + "theoretical_loss": 3.4998203439579147, + "tokens_seen": 1581383680 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002630792377131394, + "loss": 1.3068, + "theoretical_loss": 3.499807685422441, + "tokens_seen": 1581449216 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002630692076228686, + "loss": 1.3581, + "theoretical_loss": 3.499795027558406, + "tokens_seen": 1581514752 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002630591775325978, + "loss": 1.3238, + "theoretical_loss": 3.4997823703657467, + "tokens_seen": 1581580288 + }, + { + "epoch": 0.48, + "learning_rate": 0.000263049147442327, + "loss": 1.3699, + "theoretical_loss": 3.4997697138444, + "tokens_seen": 1581645824 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026303911735205615, + "loss": 1.3538, + "theoretical_loss": 3.4997570579943016, + "tokens_seen": 1581711360 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002630290872617854, + "loss": 1.4518, + "theoretical_loss": 3.499744402815389, + "tokens_seen": 1581776896 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002630190571715145, + "loss": 1.3661, + "theoretical_loss": 3.499731748307598, + "tokens_seen": 1581842432 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026300902708124374, + "loss": 1.4219, + "theoretical_loss": 3.4997190944708656, + "tokens_seen": 1581907968 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002629989969909729, + "loss": 1.4177, + "theoretical_loss": 3.499706441305128, + "tokens_seen": 1581973504 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002629889669007021, + "loss": 1.3282, + "theoretical_loss": 3.4996937888103226, + "tokens_seen": 1582039040 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002629789368104313, + "loss": 1.3991, + "theoretical_loss": 3.4996811369863856, + "tokens_seen": 1582104576 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026296890672016047, + "loss": 1.3605, + "theoretical_loss": 3.499668485833254, + "tokens_seen": 1582170112 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026295887662988965, + "loss": 1.3628, + "theoretical_loss": 3.4996558353508638, + "tokens_seen": 1582235648 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002629488465396189, + "loss": 1.3235, + "theoretical_loss": 3.499643185539152, + "tokens_seen": 1582301184 + }, + { + "epoch": 0.48, + "learning_rate": 0.000262938816449348, + "loss": 1.3203, + "theoretical_loss": 3.499630536398055, + "tokens_seen": 1582366720 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026292878635907725, + "loss": 1.2916, + "theoretical_loss": 3.49961788792751, + "tokens_seen": 1582432256 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026291875626880643, + "loss": 1.3077, + "theoretical_loss": 3.4996052401274538, + "tokens_seen": 1582497792 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002629087261785356, + "loss": 1.2186, + "theoretical_loss": 3.499592592997822, + "tokens_seen": 1582563328 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002628986960882648, + "loss": 1.2363, + "theoretical_loss": 3.499579946538552, + "tokens_seen": 1582628864 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.000574201054405421, + "objective/train/docs_used": 897089, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.759089231491089, + "objective/train/original_loss": 2.759089231491089, + "objective/train/theoretical_loss": 3.4995673007495816, + "objective/train/tokens_used": 1603154400, + "objective/train/value_avg": -0.00550079345703125, + "objective/train/value_loss": 9.511390817351639e-05, + "objective/train/value_max": -0.00035834312438964844, + "objective/train/value_min": -0.0640869140625, + "objective/train/value_reward_corr": 0.2931640688085513, + "objective/train/value_std": 0.004184722900390625, + "objective/train/weight_avg": 0.9983986616134644, + "objective/train/weighted_lm_loss": 2.7587666511535645, + "objective/train/weights_max": 1.773708462715149, + "objective/train/weights_min": 0.29530978202819824, + "theoretical_loss": 3.4995673007495816, + "tokens_seen": 1582694400 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026288866599799397, + "loss": 1.3642, + "theoretical_loss": 3.4995673007495816, + "tokens_seen": 1582694400 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026287863590772315, + "loss": 1.2704, + "theoretical_loss": 3.4995546556308454, + "tokens_seen": 1582759936 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002628686058174524, + "loss": 1.293, + "theoretical_loss": 3.4995420111822813, + "tokens_seen": 1582825472 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002628585757271815, + "loss": 1.4087, + "theoretical_loss": 3.499529367403826, + "tokens_seen": 1582891008 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026284854563691075, + "loss": 1.3953, + "theoretical_loss": 3.4995167242954164, + "tokens_seen": 1582956544 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002628385155466399, + "loss": 1.2787, + "theoretical_loss": 3.4995040818569887, + "tokens_seen": 1583022080 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002628284854563691, + "loss": 1.3879, + "theoretical_loss": 3.49949144008848, + "tokens_seen": 1583087616 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002628184553660983, + "loss": 1.4054, + "theoretical_loss": 3.499478798989827, + "tokens_seen": 1583153152 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002628084252758275, + "loss": 1.4775, + "theoretical_loss": 3.499466158560966, + "tokens_seen": 1583218688 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026279839518555666, + "loss": 1.4147, + "theoretical_loss": 3.499453518801835, + "tokens_seen": 1583284224 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002627883650952859, + "loss": 1.4349, + "theoretical_loss": 3.4994408797123704, + "tokens_seen": 1583349760 + }, + { + "epoch": 0.48, + "learning_rate": 0.000262778335005015, + "loss": 1.4094, + "theoretical_loss": 3.4994282412925077, + "tokens_seen": 1583415296 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026276830491474425, + "loss": 1.3027, + "theoretical_loss": 3.4994156035421855, + "tokens_seen": 1583480832 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026275827482447343, + "loss": 1.3899, + "theoretical_loss": 3.4994029664613393, + "tokens_seen": 1583546368 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002627482447342026, + "loss": 1.4201, + "theoretical_loss": 3.499390330049907, + "tokens_seen": 1583611904 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026273821464393185, + "loss": 1.3545, + "theoretical_loss": 3.4993776943078245, + "tokens_seen": 1583677440 + }, + { + "epoch": 0.48, + "learning_rate": 0.000262728184553661, + "loss": 1.4501, + "theoretical_loss": 3.4993650592350294, + "tokens_seen": 1583742976 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002627181544633902, + "loss": 1.2728, + "theoretical_loss": 3.499352424831458, + "tokens_seen": 1583808512 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026270812437311934, + "loss": 1.416, + "theoretical_loss": 3.4993397910970474, + "tokens_seen": 1583874048 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002626980942828486, + "loss": 1.3371, + "theoretical_loss": 3.4993271580317344, + "tokens_seen": 1583939584 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026268806419257776, + "loss": 1.4408, + "theoretical_loss": 3.499314525635456, + "tokens_seen": 1584005120 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026267803410230694, + "loss": 1.4426, + "theoretical_loss": 3.499301893908149, + "tokens_seen": 1584070656 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002626680040120361, + "loss": 1.3191, + "theoretical_loss": 3.49928926284975, + "tokens_seen": 1584136192 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002626579739217653, + "loss": 1.2779, + "theoretical_loss": 3.4992766324601967, + "tokens_seen": 1584201728 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002626479438314945, + "loss": 1.4513, + "theoretical_loss": 3.4992640027394253, + "tokens_seen": 1584267264 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": 0.0011657722061499953, + "objective/train/docs_used": 897735, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.649683952331543, + "objective/train/original_loss": 2.649683713912964, + "objective/train/theoretical_loss": 3.499251373687373, + "objective/train/tokens_used": 1604792800, + "objective/train/value_avg": -0.004207611083984375, + "objective/train/value_loss": 2.517068787710741e-05, + "objective/train/value_max": -0.0004305839538574219, + "objective/train/value_min": -0.043121337890625, + "objective/train/value_reward_corr": 0.3316241374687393, + "objective/train/value_std": 0.003116607666015625, + "objective/train/weight_avg": 1.012904405593872, + "objective/train/weighted_lm_loss": 2.682878017425537, + "objective/train/weights_max": 1.4137243032455444, + "objective/train/weights_min": 0.613807737827301, + "theoretical_loss": 3.499251373687373, + "tokens_seen": 1584332800 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002626379137412237, + "loss": 1.3657, + "theoretical_loss": 3.499251373687373, + "tokens_seen": 1584332800 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026262788365095284, + "loss": 1.3735, + "theoretical_loss": 3.499238745303977, + "tokens_seen": 1584398336 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002626178535606821, + "loss": 1.4399, + "theoretical_loss": 3.4992261175891732, + "tokens_seen": 1584463872 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026260782347041126, + "loss": 1.2846, + "theoretical_loss": 3.4992134905428998, + "tokens_seen": 1584529408 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026259779338014044, + "loss": 1.3405, + "theoretical_loss": 3.4992008641650933, + "tokens_seen": 1584594944 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002625877632898696, + "loss": 1.345, + "theoretical_loss": 3.49918823845569, + "tokens_seen": 1584660480 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002625777331995988, + "loss": 1.272, + "theoretical_loss": 3.4991756134146286, + "tokens_seen": 1584726016 + }, + { + "epoch": 0.48, + "learning_rate": 0.000262567703109328, + "loss": 1.2898, + "theoretical_loss": 3.499162989041844, + "tokens_seen": 1584791552 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002625576730190572, + "loss": 1.394, + "theoretical_loss": 3.4991503653372744, + "tokens_seen": 1584857088 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026254764292878635, + "loss": 1.3386, + "theoretical_loss": 3.4991377423008565, + "tokens_seen": 1584922624 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002625376128385156, + "loss": 1.4614, + "theoretical_loss": 3.4991251199325277, + "tokens_seen": 1584988160 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002625275827482447, + "loss": 1.3899, + "theoretical_loss": 3.4991124982322246, + "tokens_seen": 1585053696 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026251755265797394, + "loss": 1.3899, + "theoretical_loss": 3.4990998771998845, + "tokens_seen": 1585119232 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002625075225677031, + "loss": 1.2996, + "theoretical_loss": 3.499087256835444, + "tokens_seen": 1585184768 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002624974924774323, + "loss": 1.1831, + "theoretical_loss": 3.49907463713884, + "tokens_seen": 1585250304 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002624874623871615, + "loss": 1.3094, + "theoretical_loss": 3.4990620181100107, + "tokens_seen": 1585315840 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026247743229689067, + "loss": 1.2791, + "theoretical_loss": 3.499049399748892, + "tokens_seen": 1585381376 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026246740220661985, + "loss": 1.293, + "theoretical_loss": 3.4990367820554216, + "tokens_seen": 1585446912 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002624573721163491, + "loss": 1.3183, + "theoretical_loss": 3.499024165029536, + "tokens_seen": 1585512448 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002624473420260782, + "loss": 1.2956, + "theoretical_loss": 3.499011548671173, + "tokens_seen": 1585577984 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026243731193580745, + "loss": 1.4395, + "theoretical_loss": 3.4989989329802693, + "tokens_seen": 1585643520 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026242728184553663, + "loss": 1.3391, + "theoretical_loss": 3.498986317956762, + "tokens_seen": 1585709056 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002624172517552658, + "loss": 1.3162, + "theoretical_loss": 3.4989737036005883, + "tokens_seen": 1585774592 + }, + { + "epoch": 0.48, + "learning_rate": 0.000262407221664995, + "loss": 1.3529, + "theoretical_loss": 3.4989610899116856, + "tokens_seen": 1585840128 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026239719157472417, + "loss": 1.345, + "theoretical_loss": 3.49894847688999, + "tokens_seen": 1585905664 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.0012950043892487884, + "objective/train/docs_used": 899130, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5992236137390137, + "objective/train/original_loss": 2.5992238521575928, + "objective/train/theoretical_loss": 3.49893586453544, + "objective/train/tokens_used": 1606431200, + "objective/train/value_avg": -0.004878997802734375, + "objective/train/value_loss": 4.941031875205226e-05, + "objective/train/value_max": -0.00039196014404296875, + "objective/train/value_min": -0.046722412109375, + "objective/train/value_reward_corr": 0.32993940685228773, + "objective/train/value_std": 0.003528594970703125, + "objective/train/weight_avg": 0.9893919825553894, + "objective/train/weighted_lm_loss": 2.5798583030700684, + "objective/train/weights_max": 1.4462262392044067, + "objective/train/weights_min": 0.4071545898914337, + "theoretical_loss": 3.49893586453544, + "tokens_seen": 1585971200 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026238716148445335, + "loss": 1.3249, + "theoretical_loss": 3.49893586453544, + "tokens_seen": 1585971200 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002623771313941826, + "loss": 1.4491, + "theoretical_loss": 3.498923252847972, + "tokens_seen": 1586036736 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002623671013039117, + "loss": 1.429, + "theoretical_loss": 3.498910641827523, + "tokens_seen": 1586102272 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026235707121364095, + "loss": 1.3479, + "theoretical_loss": 3.4988980314740306, + "tokens_seen": 1586167808 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002623470411233701, + "loss": 1.411, + "theoretical_loss": 3.4988854217874317, + "tokens_seen": 1586233344 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002623370110330993, + "loss": 1.351, + "theoretical_loss": 3.4988728127676634, + "tokens_seen": 1586298880 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002623269809428285, + "loss": 1.3938, + "theoretical_loss": 3.4988602044146635, + "tokens_seen": 1586364416 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002623169508525577, + "loss": 1.4394, + "theoretical_loss": 3.4988475967283685, + "tokens_seen": 1586429952 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026230692076228686, + "loss": 1.298, + "theoretical_loss": 3.498834989708716, + "tokens_seen": 1586495488 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002622968906720161, + "loss": 1.3896, + "theoretical_loss": 3.498822383355643, + "tokens_seen": 1586561024 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002622868605817452, + "loss": 1.2789, + "theoretical_loss": 3.498809777669087, + "tokens_seen": 1586626560 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026227683049147445, + "loss": 1.3398, + "theoretical_loss": 3.4987971726489846, + "tokens_seen": 1586692096 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002622668004012036, + "loss": 1.4169, + "theoretical_loss": 3.4987845682952736, + "tokens_seen": 1586757632 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002622567703109328, + "loss": 1.2666, + "theoretical_loss": 3.4987719646078914, + "tokens_seen": 1586823168 + }, + { + "epoch": 0.48, + "learning_rate": 0.000262246740220662, + "loss": 1.363, + "theoretical_loss": 3.4987593615867745, + "tokens_seen": 1586888704 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002622367101303912, + "loss": 1.4233, + "theoretical_loss": 3.498746759231861, + "tokens_seen": 1586954240 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026222668004012036, + "loss": 1.457, + "theoretical_loss": 3.498734157543088, + "tokens_seen": 1587019776 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026221664994984954, + "loss": 1.307, + "theoretical_loss": 3.498721556520392, + "tokens_seen": 1587085312 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002622066198595787, + "loss": 1.3616, + "theoretical_loss": 3.4987089561637115, + "tokens_seen": 1587150848 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026219658976930796, + "loss": 1.1852, + "theoretical_loss": 3.4986963564729834, + "tokens_seen": 1587216384 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002621865596790371, + "loss": 1.3164, + "theoretical_loss": 3.4986837574481444, + "tokens_seen": 1587281920 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002621765295887663, + "loss": 1.3819, + "theoretical_loss": 3.498671159089132, + "tokens_seen": 1587347456 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026216649949849545, + "loss": 1.2966, + "theoretical_loss": 3.4986585613958843, + "tokens_seen": 1587412992 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002621564694082247, + "loss": 1.3243, + "theoretical_loss": 3.498645964368338, + "tokens_seen": 1587478528 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026214643931795386, + "loss": 1.3553, + "theoretical_loss": 3.4986333680064297, + "tokens_seen": 1587544064 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.0011085834121331573, + "objective/train/docs_used": 899656, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.797250747680664, + "objective/train/original_loss": 2.797250747680664, + "objective/train/theoretical_loss": 3.4986207723100984, + "objective/train/tokens_used": 1608069600, + "objective/train/value_avg": -0.004550933837890625, + "objective/train/value_loss": 4.703642844106071e-05, + "objective/train/value_max": -0.0002453327178955078, + "objective/train/value_min": -0.06707763671875, + "objective/train/value_reward_corr": 0.4004510433848919, + "objective/train/value_std": 0.0035400390625, + "objective/train/weight_avg": 0.9911633729934692, + "objective/train/weighted_lm_loss": 2.776946783065796, + "objective/train/weights_max": 1.6768721342086792, + "objective/train/weights_min": 0.6008287668228149, + "theoretical_loss": 3.4986207723100984, + "tokens_seen": 1587609600 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026213640922768304, + "loss": 1.4561, + "theoretical_loss": 3.4986207723100984, + "tokens_seen": 1587609600 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002621263791374122, + "loss": 1.3612, + "theoretical_loss": 3.49860817727928, + "tokens_seen": 1587675136 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026211634904714146, + "loss": 1.2446, + "theoretical_loss": 3.4985955829139135, + "tokens_seen": 1587740672 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002621063189568706, + "loss": 1.3342, + "theoretical_loss": 3.4985829892139346, + "tokens_seen": 1587806208 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002620962888665998, + "loss": 1.4197, + "theoretical_loss": 3.4985703961792813, + "tokens_seen": 1587871744 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026208625877632895, + "loss": 1.2724, + "theoretical_loss": 3.4985578038098915, + "tokens_seen": 1587937280 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002620762286860582, + "loss": 1.2248, + "theoretical_loss": 3.498545212105702, + "tokens_seen": 1588002816 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026206619859578737, + "loss": 1.3196, + "theoretical_loss": 3.4985326210666496, + "tokens_seen": 1588068352 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026205616850551655, + "loss": 1.3411, + "theoretical_loss": 3.4985200306926734, + "tokens_seen": 1588133888 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026204613841524573, + "loss": 1.4121, + "theoretical_loss": 3.4985074409837096, + "tokens_seen": 1588199424 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002620361083249749, + "loss": 1.2952, + "theoretical_loss": 3.4984948519396957, + "tokens_seen": 1588264960 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002620260782347041, + "loss": 1.4475, + "theoretical_loss": 3.49848226356057, + "tokens_seen": 1588330496 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002620160481444333, + "loss": 1.3618, + "theoretical_loss": 3.498469675846269, + "tokens_seen": 1588396032 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002620060180541625, + "loss": 1.2603, + "theoretical_loss": 3.49845708879673, + "tokens_seen": 1588461568 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002619959879638917, + "loss": 1.2553, + "theoretical_loss": 3.4984445024118918, + "tokens_seen": 1588527104 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026198595787362087, + "loss": 1.3647, + "theoretical_loss": 3.4984319166916906, + "tokens_seen": 1588592640 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026197592778335005, + "loss": 1.3037, + "theoretical_loss": 3.4984193316360646, + "tokens_seen": 1588658176 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002619658976930793, + "loss": 1.4217, + "theoretical_loss": 3.4984067472449514, + "tokens_seen": 1588723712 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002619558676028084, + "loss": 1.432, + "theoretical_loss": 3.4983941635182876, + "tokens_seen": 1588789248 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026194583751253765, + "loss": 1.3195, + "theoretical_loss": 3.498381580456011, + "tokens_seen": 1588854784 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026193580742226683, + "loss": 1.3281, + "theoretical_loss": 3.49836899805806, + "tokens_seen": 1588920320 + }, + { + "epoch": 0.48, + "learning_rate": 0.000261925777331996, + "loss": 1.3586, + "theoretical_loss": 3.4983564163243708, + "tokens_seen": 1588985856 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002619157472417252, + "loss": 1.3553, + "theoretical_loss": 3.4983438352548824, + "tokens_seen": 1589051392 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026190571715145437, + "loss": 1.2326, + "theoretical_loss": 3.498331254849531, + "tokens_seen": 1589116928 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026189568706118355, + "loss": 1.3404, + "theoretical_loss": 3.498318675108255, + "tokens_seen": 1589182464 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": 0.00016955874161794782, + "objective/train/docs_used": 900745, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0928406715393066, + "objective/train/original_loss": 3.0928406715393066, + "objective/train/theoretical_loss": 3.4983060960309915, + "objective/train/tokens_used": 1609708000, + "objective/train/value_avg": -0.005489349365234375, + "objective/train/value_loss": 4.521034861681983e-05, + "objective/train/value_max": -0.0003726482391357422, + "objective/train/value_min": -0.07708740234375, + "objective/train/value_reward_corr": 0.39470071675537705, + "objective/train/value_std": 0.0040435791015625, + "objective/train/weight_avg": 1.0038752555847168, + "objective/train/weighted_lm_loss": 3.1069087982177734, + "objective/train/weights_max": 1.7825350761413574, + "objective/train/weights_min": 0.42447343468666077, + "theoretical_loss": 3.4983060960309915, + "tokens_seen": 1589248000 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002618856569709128, + "loss": 1.3497, + "theoretical_loss": 3.4983060960309915, + "tokens_seen": 1589248000 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002618756268806419, + "loss": 1.3598, + "theoretical_loss": 3.4982935176176784, + "tokens_seen": 1589313536 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026186559679037115, + "loss": 1.3369, + "theoretical_loss": 3.4982809398682533, + "tokens_seen": 1589379072 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002618555667001003, + "loss": 1.4426, + "theoretical_loss": 3.498268362782653, + "tokens_seen": 1589444608 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002618455366098295, + "loss": 1.4277, + "theoretical_loss": 3.498255786360817, + "tokens_seen": 1589510144 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002618355065195587, + "loss": 1.3313, + "theoretical_loss": 3.4982432106026806, + "tokens_seen": 1589575680 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002618254764292879, + "loss": 1.3951, + "theoretical_loss": 3.4982306355081825, + "tokens_seen": 1589641216 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026181544633901706, + "loss": 1.2682, + "theoretical_loss": 3.49821806107726, + "tokens_seen": 1589706752 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002618054162487463, + "loss": 1.275, + "theoretical_loss": 3.4982054873098516, + "tokens_seen": 1589772288 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002617953861584754, + "loss": 1.3416, + "theoretical_loss": 3.4981929142058945, + "tokens_seen": 1589837824 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026178535606820465, + "loss": 1.3002, + "theoretical_loss": 3.498180341765326, + "tokens_seen": 1589903360 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002617753259779338, + "loss": 1.2825, + "theoretical_loss": 3.4981677699880835, + "tokens_seen": 1589968896 + }, + { + "epoch": 0.48, + "learning_rate": 0.000261765295887663, + "loss": 1.3148, + "theoretical_loss": 3.4981551988741053, + "tokens_seen": 1590034432 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002617552657973922, + "loss": 1.2718, + "theoretical_loss": 3.498142628423329, + "tokens_seen": 1590099968 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002617452357071214, + "loss": 1.3004, + "theoretical_loss": 3.4981300586356925, + "tokens_seen": 1590165504 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026173520561685056, + "loss": 1.42, + "theoretical_loss": 3.4981174895111327, + "tokens_seen": 1590231040 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026172517552657974, + "loss": 1.2109, + "theoretical_loss": 3.498104921049588, + "tokens_seen": 1590296576 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002617151454363089, + "loss": 1.3508, + "theoretical_loss": 3.4980923532509953, + "tokens_seen": 1590362112 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026170511534603816, + "loss": 1.3986, + "theoretical_loss": 3.4980797861152935, + "tokens_seen": 1590427648 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002616950852557673, + "loss": 1.3008, + "theoretical_loss": 3.4980672196424196, + "tokens_seen": 1590493184 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002616850551654965, + "loss": 1.3323, + "theoretical_loss": 3.4980546538323107, + "tokens_seen": 1590558720 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026167502507522565, + "loss": 1.2393, + "theoretical_loss": 3.498042088684906, + "tokens_seen": 1590624256 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002616649949849549, + "loss": 1.2443, + "theoretical_loss": 3.4980295242001422, + "tokens_seen": 1590689792 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026165496489468406, + "loss": 1.3191, + "theoretical_loss": 3.4980169603779574, + "tokens_seen": 1590755328 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026164493480441324, + "loss": 1.3507, + "theoretical_loss": 3.4980043972182897, + "tokens_seen": 1590820864 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -5.4582604207098484e-05, + "objective/train/docs_used": 902130, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6891260147094727, + "objective/train/original_loss": 2.6891262531280518, + "objective/train/theoretical_loss": 3.497991834721076, + "objective/train/tokens_used": 1611346400, + "objective/train/value_avg": -0.004825592041015625, + "objective/train/value_loss": 4.375218850327656e-05, + "objective/train/value_max": -0.00033020973205566406, + "objective/train/value_min": -0.0999755859375, + "objective/train/value_reward_corr": 0.4152321681456171, + "objective/train/value_std": 0.004177093505859375, + "objective/train/weight_avg": 1.0015270709991455, + "objective/train/weighted_lm_loss": 2.6888060569763184, + "objective/train/weights_max": 2.3174190521240234, + "objective/train/weights_min": 0.34202590584754944, + "theoretical_loss": 3.497991834721076, + "tokens_seen": 1590886400 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002616349047141424, + "loss": 1.4155, + "theoretical_loss": 3.497991834721076, + "tokens_seen": 1590886400 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026162487462387166, + "loss": 1.4221, + "theoretical_loss": 3.497979272886255, + "tokens_seen": 1590951936 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002616148445336008, + "loss": 1.312, + "theoretical_loss": 3.4979667117137634, + "tokens_seen": 1591017472 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026160481444333, + "loss": 1.3999, + "theoretical_loss": 3.4979541512035404, + "tokens_seen": 1591083008 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026159478435305915, + "loss": 1.3935, + "theoretical_loss": 3.4979415913555227, + "tokens_seen": 1591148544 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002615847542627884, + "loss": 1.1848, + "theoretical_loss": 3.4979290321696483, + "tokens_seen": 1591214080 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026157472417251757, + "loss": 1.3494, + "theoretical_loss": 3.4979164736458555, + "tokens_seen": 1591279616 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026156469408224675, + "loss": 1.2824, + "theoretical_loss": 3.497903915784082, + "tokens_seen": 1591345152 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026155466399197593, + "loss": 1.3909, + "theoretical_loss": 3.497891358584265, + "tokens_seen": 1591410688 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002615446339017051, + "loss": 1.2794, + "theoretical_loss": 3.497878802046343, + "tokens_seen": 1591476224 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002615346038114343, + "loss": 1.4098, + "theoretical_loss": 3.497866246170254, + "tokens_seen": 1591541760 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002615245737211635, + "loss": 1.4173, + "theoretical_loss": 3.4978536909559352, + "tokens_seen": 1591607296 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026151454363089265, + "loss": 1.3647, + "theoretical_loss": 3.497841136403325, + "tokens_seen": 1591672832 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002615045135406219, + "loss": 1.3716, + "theoretical_loss": 3.497828582512361, + "tokens_seen": 1591738368 + }, + { + "epoch": 0.48, + "learning_rate": 0.000261494483450351, + "loss": 1.3155, + "theoretical_loss": 3.4978160292829816, + "tokens_seen": 1591803904 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026148445336008025, + "loss": 1.337, + "theoretical_loss": 3.4978034767151236, + "tokens_seen": 1591869440 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026147442326980943, + "loss": 1.4484, + "theoretical_loss": 3.4977909248087258, + "tokens_seen": 1591934976 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002614643931795386, + "loss": 1.3167, + "theoretical_loss": 3.497778373563726, + "tokens_seen": 1592000512 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002614543630892678, + "loss": 1.378, + "theoretical_loss": 3.497765822980062, + "tokens_seen": 1592066048 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026144433299899703, + "loss": 1.3635, + "theoretical_loss": 3.497753273057672, + "tokens_seen": 1592131584 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026143430290872616, + "loss": 1.2781, + "theoretical_loss": 3.497740723796494, + "tokens_seen": 1592197120 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002614242728184554, + "loss": 1.3302, + "theoretical_loss": 3.4977281751964653, + "tokens_seen": 1592262656 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002614142427281845, + "loss": 1.4467, + "theoretical_loss": 3.4977156272575245, + "tokens_seen": 1592328192 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026140421263791375, + "loss": 1.3278, + "theoretical_loss": 3.4977030799796087, + "tokens_seen": 1592393728 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026139418254764293, + "loss": 1.414, + "theoretical_loss": 3.4976905333626567, + "tokens_seen": 1592459264 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": 0.0004648338072001934, + "objective/train/docs_used": 902721, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.746025562286377, + "objective/train/original_loss": 2.746025562286377, + "objective/train/theoretical_loss": 3.4976779874066066, + "objective/train/tokens_used": 1612984800, + "objective/train/value_avg": -0.004184722900390625, + "objective/train/value_loss": 5.635028355754912e-05, + "objective/train/value_max": -0.00014317035675048828, + "objective/train/value_min": -0.041778564453125, + "objective/train/value_reward_corr": 0.352015012416673, + "objective/train/value_std": 0.0031147003173828125, + "objective/train/weight_avg": 1.0072382688522339, + "objective/train/weighted_lm_loss": 2.7673799991607666, + "objective/train/weights_max": 1.4238914251327515, + "objective/train/weights_min": 0.3842305839061737, + "theoretical_loss": 3.4976779874066066, + "tokens_seen": 1592524800 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002613841524573721, + "loss": 1.3585, + "theoretical_loss": 3.4976779874066066, + "tokens_seen": 1592524800 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002613741223671013, + "loss": 1.412, + "theoretical_loss": 3.4976654421113955, + "tokens_seen": 1592590336 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002613640922768305, + "loss": 1.4374, + "theoretical_loss": 3.497652897476962, + "tokens_seen": 1592655872 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026135406218655966, + "loss": 1.2594, + "theoretical_loss": 3.497640353503244, + "tokens_seen": 1592721408 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002613440320962889, + "loss": 1.3185, + "theoretical_loss": 3.4976278101901803, + "tokens_seen": 1592786944 + }, + { + "epoch": 0.48, + "learning_rate": 0.000261334002006018, + "loss": 1.2899, + "theoretical_loss": 3.4976152675377077, + "tokens_seen": 1592852480 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026132397191574726, + "loss": 1.3491, + "theoretical_loss": 3.4976027255457645, + "tokens_seen": 1592918016 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002613139418254764, + "loss": 1.3276, + "theoretical_loss": 3.4975901842142894, + "tokens_seen": 1592983552 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002613039117352056, + "loss": 1.3476, + "theoretical_loss": 3.4975776435432198, + "tokens_seen": 1593049088 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002612938816449348, + "loss": 1.2799, + "theoretical_loss": 3.497565103532494, + "tokens_seen": 1593114624 + }, + { + "epoch": 0.48, + "learning_rate": 0.000261283851554664, + "loss": 1.3095, + "theoretical_loss": 3.49755256418205, + "tokens_seen": 1593180160 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026127382146439316, + "loss": 1.3332, + "theoretical_loss": 3.497540025491826, + "tokens_seen": 1593245696 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002612637913741224, + "loss": 1.4716, + "theoretical_loss": 3.4975274874617597, + "tokens_seen": 1593311232 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002612537612838516, + "loss": 1.2432, + "theoretical_loss": 3.49751495009179, + "tokens_seen": 1593376768 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026124373119358076, + "loss": 1.4011, + "theoretical_loss": 3.497502413381854, + "tokens_seen": 1593442304 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026123370110330994, + "loss": 1.448, + "theoretical_loss": 3.4974898773318905, + "tokens_seen": 1593507840 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002612236710130391, + "loss": 1.2925, + "theoretical_loss": 3.497477341941837, + "tokens_seen": 1593573376 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026121364092276836, + "loss": 1.347, + "theoretical_loss": 3.4974648072116326, + "tokens_seen": 1593638912 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002612036108324975, + "loss": 1.3312, + "theoretical_loss": 3.4974522731412145, + "tokens_seen": 1593704448 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002611935807422267, + "loss": 1.4101, + "theoretical_loss": 3.4974397397305217, + "tokens_seen": 1593769984 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026118355065195585, + "loss": 1.3655, + "theoretical_loss": 3.4974272069794914, + "tokens_seen": 1593835520 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002611735205616851, + "loss": 1.307, + "theoretical_loss": 3.4974146748880623, + "tokens_seen": 1593901056 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026116349047141426, + "loss": 1.4351, + "theoretical_loss": 3.4974021434561724, + "tokens_seen": 1593966592 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026115346038114344, + "loss": 1.3293, + "theoretical_loss": 3.4973896126837603, + "tokens_seen": 1594032128 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002611434302908726, + "loss": 1.3492, + "theoretical_loss": 3.4973770825707637, + "tokens_seen": 1594097664 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": 0.0002995907561853528, + "objective/train/docs_used": 903500, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.7508559226989746, + "objective/train/original_loss": 2.7508556842803955, + "objective/train/theoretical_loss": 3.4973645531171207, + "objective/train/tokens_used": 1614623200, + "objective/train/value_avg": -0.00677490234375, + "objective/train/value_loss": 5.777081605629064e-05, + "objective/train/value_max": -0.0005998611450195312, + "objective/train/value_min": -0.093994140625, + "objective/train/value_reward_corr": 0.5285699182609297, + "objective/train/value_std": 0.006496429443359375, + "objective/train/weight_avg": 1.0058090686798096, + "objective/train/weighted_lm_loss": 2.769676446914673, + "objective/train/weights_max": 2.1841161251068115, + "objective/train/weights_min": 0.46591442823410034, + "theoretical_loss": 3.4973645531171207, + "tokens_seen": 1594163200 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026113340020060186, + "loss": 1.429, + "theoretical_loss": 3.4973645531171207, + "tokens_seen": 1594163200 + }, + { + "epoch": 0.48, + "learning_rate": 0.000261123370110331, + "loss": 1.3402, + "theoretical_loss": 3.49735202432277, + "tokens_seen": 1594228736 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002611133400200602, + "loss": 1.3222, + "theoretical_loss": 3.4973394961876494, + "tokens_seen": 1594294272 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026110330992978935, + "loss": 1.3085, + "theoretical_loss": 3.4973269687116977, + "tokens_seen": 1594359808 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002610932798395186, + "loss": 1.3154, + "theoretical_loss": 3.497314441894852, + "tokens_seen": 1594425344 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026108324974924777, + "loss": 1.4094, + "theoretical_loss": 3.4973019157370517, + "tokens_seen": 1594490880 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026107321965897695, + "loss": 1.4117, + "theoretical_loss": 3.497289390238234, + "tokens_seen": 1594556416 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026106318956870613, + "loss": 1.3478, + "theoretical_loss": 3.497276865398338, + "tokens_seen": 1594621952 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002610531594784353, + "loss": 1.3502, + "theoretical_loss": 3.497264341217302, + "tokens_seen": 1594687488 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002610431293881645, + "loss": 1.4156, + "theoretical_loss": 3.497251817695064, + "tokens_seen": 1594753024 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002610330992978937, + "loss": 1.2273, + "theoretical_loss": 3.4972392948315614, + "tokens_seen": 1594818560 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026102306920762285, + "loss": 1.3025, + "theoretical_loss": 3.4972267726267336, + "tokens_seen": 1594884096 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002610130391173521, + "loss": 1.3373, + "theoretical_loss": 3.497214251080519, + "tokens_seen": 1594949632 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002610030090270812, + "loss": 1.4323, + "theoretical_loss": 3.497201730192855, + "tokens_seen": 1595015168 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026099297893681045, + "loss": 1.3807, + "theoretical_loss": 3.4971892099636808, + "tokens_seen": 1595080704 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026098294884653963, + "loss": 1.2619, + "theoretical_loss": 3.4971766903929344, + "tokens_seen": 1595146240 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002609729187562688, + "loss": 1.3458, + "theoretical_loss": 3.497164171480554, + "tokens_seen": 1595211776 + }, + { + "epoch": 0.48, + "learning_rate": 0.000260962888665998, + "loss": 1.4445, + "theoretical_loss": 3.4971516532264775, + "tokens_seen": 1595277312 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026095285857572723, + "loss": 1.4161, + "theoretical_loss": 3.4971391356306443, + "tokens_seen": 1595342848 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026094282848545636, + "loss": 1.3477, + "theoretical_loss": 3.4971266186929917, + "tokens_seen": 1595408384 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002609327983951856, + "loss": 1.3382, + "theoretical_loss": 3.497114102413458, + "tokens_seen": 1595473920 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002609227683049147, + "loss": 1.346, + "theoretical_loss": 3.497101586791983, + "tokens_seen": 1595539456 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026091273821464395, + "loss": 1.2392, + "theoretical_loss": 3.497089071828504, + "tokens_seen": 1595604992 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026090270812437313, + "loss": 1.3543, + "theoretical_loss": 3.497076557522959, + "tokens_seen": 1595670528 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002608926780341023, + "loss": 1.2011, + "theoretical_loss": 3.4970640438752874, + "tokens_seen": 1595736064 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.0011616046540439129, + "objective/train/docs_used": 904004, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 1.984488844871521, + "objective/train/original_loss": 1.9844887256622314, + "objective/train/theoretical_loss": 3.497051530885427, + "objective/train/tokens_used": 1616261600, + "objective/train/value_avg": -0.00446319580078125, + "objective/train/value_loss": 6.116964505054057e-05, + "objective/train/value_max": -0.0004425048828125, + "objective/train/value_min": -0.042236328125, + "objective/train/value_reward_corr": 0.33070815181466057, + "objective/train/value_std": 0.003147125244140625, + "objective/train/weight_avg": 0.9912167191505432, + "objective/train/weighted_lm_loss": 1.9665831327438354, + "objective/train/weights_max": 1.4131661653518677, + "objective/train/weights_min": 0.3401457667350769, + "theoretical_loss": 3.497051530885427, + "tokens_seen": 1595801600 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002608826479438315, + "loss": 1.2598, + "theoretical_loss": 3.497051530885427, + "tokens_seen": 1595801600 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002608726178535607, + "loss": 1.3032, + "theoretical_loss": 3.4970390185533167, + "tokens_seen": 1595867136 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026086258776328986, + "loss": 1.3407, + "theoretical_loss": 3.4970265068788944, + "tokens_seen": 1595932672 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002608525576730191, + "loss": 1.3771, + "theoretical_loss": 3.4970139958620985, + "tokens_seen": 1595998208 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002608425275827482, + "loss": 1.2713, + "theoretical_loss": 3.4970014855028673, + "tokens_seen": 1596063744 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026083249749247746, + "loss": 1.3739, + "theoretical_loss": 3.49698897580114, + "tokens_seen": 1596129280 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002608224674022066, + "loss": 1.3348, + "theoretical_loss": 3.4969764667568546, + "tokens_seen": 1596194816 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002608124373119358, + "loss": 1.4211, + "theoretical_loss": 3.496963958369949, + "tokens_seen": 1596260352 + }, + { + "epoch": 0.48, + "learning_rate": 0.000260802407221665, + "loss": 1.3776, + "theoretical_loss": 3.496951450640363, + "tokens_seen": 1596325888 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002607923771313942, + "loss": 1.4039, + "theoretical_loss": 3.4969389435680336, + "tokens_seen": 1596391424 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026078234704112336, + "loss": 1.3574, + "theoretical_loss": 3.4969264371529007, + "tokens_seen": 1596456960 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002607723169508526, + "loss": 1.2823, + "theoretical_loss": 3.496913931394902, + "tokens_seen": 1596522496 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002607622868605817, + "loss": 1.3076, + "theoretical_loss": 3.4969014262939755, + "tokens_seen": 1596588032 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026075225677031096, + "loss": 1.372, + "theoretical_loss": 3.4968889218500605, + "tokens_seen": 1596653568 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002607422266800401, + "loss": 1.317, + "theoretical_loss": 3.496876418063095, + "tokens_seen": 1596719104 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002607321965897693, + "loss": 1.4743, + "theoretical_loss": 3.4968639149330185, + "tokens_seen": 1596784640 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002607221664994985, + "loss": 1.3921, + "theoretical_loss": 3.496851412459768, + "tokens_seen": 1596850176 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002607121364092277, + "loss": 1.2794, + "theoretical_loss": 3.496838910643284, + "tokens_seen": 1596915712 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026070210631895687, + "loss": 1.3974, + "theoretical_loss": 3.4968264094835027, + "tokens_seen": 1596981248 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026069207622868605, + "loss": 1.4008, + "theoretical_loss": 3.4968139089803647, + "tokens_seen": 1597046784 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026068204613841523, + "loss": 1.2993, + "theoretical_loss": 3.4968014091338073, + "tokens_seen": 1597112320 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026067201604814446, + "loss": 1.3349, + "theoretical_loss": 3.49678890994377, + "tokens_seen": 1597177856 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002606619859578736, + "loss": 1.3868, + "theoretical_loss": 3.496776411410191, + "tokens_seen": 1597243392 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002606519558676028, + "loss": 1.1868, + "theoretical_loss": 3.496763913533008, + "tokens_seen": 1597308928 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026064192577733195, + "loss": 1.3978, + "theoretical_loss": 3.4967514163121614, + "tokens_seen": 1597374464 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.0011186620686203241, + "objective/train/docs_used": 904558, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.730773448944092, + "objective/train/original_loss": 2.730773448944092, + "objective/train/theoretical_loss": 3.496738919747588, + "objective/train/tokens_used": 1617900000, + "objective/train/value_avg": -0.005046844482421875, + "objective/train/value_loss": 5.423882248578593e-05, + "objective/train/value_max": -0.0002715587615966797, + "objective/train/value_min": -0.05572509765625, + "objective/train/value_reward_corr": 0.38893899964668677, + "objective/train/value_std": 0.0039520263671875, + "objective/train/weight_avg": 0.9913780093193054, + "objective/train/weighted_lm_loss": 2.711364269256592, + "objective/train/weights_max": 1.4910417795181274, + "objective/train/weights_min": 0.46138665080070496, + "theoretical_loss": 3.496738919747588, + "tokens_seen": 1597440000 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002606318956870612, + "loss": 1.4117, + "theoretical_loss": 3.496738919747588, + "tokens_seen": 1597440000 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026062186559679037, + "loss": 1.5149, + "theoretical_loss": 3.4967264238392275, + "tokens_seen": 1597505536 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026061183550651955, + "loss": 1.3633, + "theoretical_loss": 3.4967139285870186, + "tokens_seen": 1597571072 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026060180541624873, + "loss": 1.3788, + "theoretical_loss": 3.496701433990899, + "tokens_seen": 1597636608 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026059177532597797, + "loss": 1.4663, + "theoretical_loss": 3.496688940050808, + "tokens_seen": 1597702144 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002605817452357071, + "loss": 1.2406, + "theoretical_loss": 3.4966764467666844, + "tokens_seen": 1597767680 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026057171514543633, + "loss": 1.319, + "theoretical_loss": 3.4966639541384668, + "tokens_seen": 1597833216 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026056168505516546, + "loss": 1.3772, + "theoretical_loss": 3.4966514621660933, + "tokens_seen": 1597898752 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002605516549648947, + "loss": 1.3735, + "theoretical_loss": 3.496638970849503, + "tokens_seen": 1597964288 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026054162487462387, + "loss": 1.4407, + "theoretical_loss": 3.4966264801886346, + "tokens_seen": 1598029824 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026053159478435305, + "loss": 1.3315, + "theoretical_loss": 3.496613990183427, + "tokens_seen": 1598095360 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026052156469408223, + "loss": 1.4417, + "theoretical_loss": 3.4966015008338185, + "tokens_seen": 1598160896 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002605115346038114, + "loss": 1.291, + "theoretical_loss": 3.4965890121397476, + "tokens_seen": 1598226432 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026050150451354065, + "loss": 1.5315, + "theoretical_loss": 3.4965765241011537, + "tokens_seen": 1598291968 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026049147442326983, + "loss": 1.4772, + "theoretical_loss": 3.496564036717975, + "tokens_seen": 1598357504 + }, + { + "epoch": 0.48, + "learning_rate": 0.000260481444332999, + "loss": 1.4744, + "theoretical_loss": 3.4965515499901505, + "tokens_seen": 1598423040 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002604714142427282, + "loss": 1.3777, + "theoretical_loss": 3.4965390639176186, + "tokens_seen": 1598488576 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026046138415245743, + "loss": 1.4993, + "theoretical_loss": 3.4965265785003186, + "tokens_seen": 1598554112 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026045135406218656, + "loss": 1.4163, + "theoretical_loss": 3.4965140937381887, + "tokens_seen": 1598619648 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002604413239719158, + "loss": 1.3498, + "theoretical_loss": 3.496501609631168, + "tokens_seen": 1598685184 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002604312938816449, + "loss": 1.3432, + "theoretical_loss": 3.4964891261791955, + "tokens_seen": 1598750720 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026042126379137415, + "loss": 1.4115, + "theoretical_loss": 3.496476643382209, + "tokens_seen": 1598816256 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026041123370110333, + "loss": 1.3669, + "theoretical_loss": 3.4964641612401484, + "tokens_seen": 1598881792 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002604012036108325, + "loss": 1.2761, + "theoretical_loss": 3.4964516797529517, + "tokens_seen": 1598947328 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002603911735205617, + "loss": 1.3708, + "theoretical_loss": 3.496439198920558, + "tokens_seen": 1599012864 + }, + { + "epoch": 0.48, + "objective/train/advantage_avg": -0.0017554962541908026, + "objective/train/docs_used": 905731, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5930421352386475, + "objective/train/original_loss": 2.5930418968200684, + "objective/train/theoretical_loss": 3.4964267187429066, + "objective/train/tokens_used": 1619538400, + "objective/train/value_avg": -0.00533294677734375, + "objective/train/value_loss": 5.4587770137004554e-05, + "objective/train/value_max": -0.0003597736358642578, + "objective/train/value_min": -0.04107666015625, + "objective/train/value_reward_corr": 0.38561292906819156, + "objective/train/value_std": 0.003742218017578125, + "objective/train/weight_avg": 0.9850510954856873, + "objective/train/weighted_lm_loss": 2.564393997192383, + "objective/train/weights_max": 1.4465653896331787, + "objective/train/weights_min": 0.546975314617157, + "theoretical_loss": 3.4964267187429066, + "tokens_seen": 1599078400 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002603811434302909, + "loss": 1.3324, + "theoretical_loss": 3.4964267187429066, + "tokens_seen": 1599078400 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026037111334002006, + "loss": 1.3081, + "theoretical_loss": 3.4964142392199355, + "tokens_seen": 1599143936 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002603610832497493, + "loss": 1.293, + "theoretical_loss": 3.496401760351584, + "tokens_seen": 1599209472 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002603510531594784, + "loss": 1.3571, + "theoretical_loss": 3.496389282137791, + "tokens_seen": 1599275008 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026034102306920766, + "loss": 1.4234, + "theoretical_loss": 3.496376804578495, + "tokens_seen": 1599340544 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002603309929789368, + "loss": 1.3773, + "theoretical_loss": 3.4963643276736356, + "tokens_seen": 1599406080 + }, + { + "epoch": 0.48, + "learning_rate": 0.000260320962888666, + "loss": 1.4564, + "theoretical_loss": 3.4963518514231504, + "tokens_seen": 1599471616 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002603109327983952, + "loss": 1.4641, + "theoretical_loss": 3.496339375826979, + "tokens_seen": 1599537152 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002603009027081244, + "loss": 1.4024, + "theoretical_loss": 3.4963269008850606, + "tokens_seen": 1599602688 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026029087261785356, + "loss": 1.34, + "theoretical_loss": 3.4963144265973334, + "tokens_seen": 1599668224 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002602808425275828, + "loss": 1.3046, + "theoretical_loss": 3.4963019529637367, + "tokens_seen": 1599733760 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002602708124373119, + "loss": 1.3628, + "theoretical_loss": 3.4962894799842097, + "tokens_seen": 1599799296 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026026078234704116, + "loss": 1.2285, + "theoretical_loss": 3.4962770076586906, + "tokens_seen": 1599864832 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002602507522567703, + "loss": 1.4457, + "theoretical_loss": 3.4962645359871187, + "tokens_seen": 1599930368 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002602407221664995, + "loss": 1.4599, + "theoretical_loss": 3.4962520649694326, + "tokens_seen": 1599995904 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002602306920762287, + "loss": 1.3095, + "theoretical_loss": 3.496239594605572, + "tokens_seen": 1600061440 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002602206619859579, + "loss": 1.2683, + "theoretical_loss": 3.4962271248954755, + "tokens_seen": 1600126976 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026021063189568707, + "loss": 1.3596, + "theoretical_loss": 3.4962146558390814, + "tokens_seen": 1600192512 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026020060180541625, + "loss": 1.2823, + "theoretical_loss": 3.4962021874363294, + "tokens_seen": 1600258048 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026019057171514543, + "loss": 1.3107, + "theoretical_loss": 3.496189719687158, + "tokens_seen": 1600323584 + }, + { + "epoch": 0.48, + "learning_rate": 0.00026018054162487466, + "loss": 1.3609, + "theoretical_loss": 3.496177252591507, + "tokens_seen": 1600389120 + }, + { + "epoch": 0.48, + "learning_rate": 0.0002601705115346038, + "loss": 1.3635, + "theoretical_loss": 3.4961647861493144, + "tokens_seen": 1600454656 + }, + { + "epoch": 0.49, + "learning_rate": 0.000260160481444333, + "loss": 1.353, + "theoretical_loss": 3.49615232036052, + "tokens_seen": 1600520192 + }, + { + "epoch": 0.49, + "learning_rate": 0.00026015045135406215, + "loss": 1.3137, + "theoretical_loss": 3.4961398552250618, + "tokens_seen": 1600585728 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002601404212637914, + "loss": 1.4067, + "theoretical_loss": 3.49612739074288, + "tokens_seen": 1600651264 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0026205810718238354, + "objective/train/docs_used": 906398, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.3343887329101562, + "objective/train/original_loss": 2.3343887329101562, + "objective/train/theoretical_loss": 3.4961149269139127, + "objective/train/tokens_used": 1621176800, + "objective/train/value_avg": -0.0041961669921875, + "objective/train/value_loss": 0.00012985075591132045, + "objective/train/value_max": -0.00010889768600463867, + "objective/train/value_min": -0.03271484375, + "objective/train/value_reward_corr": 0.303538256968968, + "objective/train/value_std": 0.0029735565185546875, + "objective/train/weight_avg": 0.9795370697975159, + "objective/train/weighted_lm_loss": 2.323740005493164, + "objective/train/weights_max": 1.3574846982955933, + "objective/train/weights_min": 0.3028680682182312, + "theoretical_loss": 3.4961149269139127, + "tokens_seen": 1600716800 + }, + { + "epoch": 0.49, + "learning_rate": 0.00026013039117352057, + "loss": 1.2818, + "theoretical_loss": 3.4961149269139127, + "tokens_seen": 1600716800 + }, + { + "epoch": 0.49, + "learning_rate": 0.00026012036108324975, + "loss": 1.3619, + "theoretical_loss": 3.496102463738099, + "tokens_seen": 1600782336 + }, + { + "epoch": 0.49, + "learning_rate": 0.00026011033099297893, + "loss": 1.4289, + "theoretical_loss": 3.4960900012153786, + "tokens_seen": 1600847872 + }, + { + "epoch": 0.49, + "learning_rate": 0.00026010030090270817, + "loss": 1.338, + "theoretical_loss": 3.4960775393456904, + "tokens_seen": 1600913408 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002600902708124373, + "loss": 1.3846, + "theoretical_loss": 3.4960650781289724, + "tokens_seen": 1600978944 + }, + { + "epoch": 0.49, + "learning_rate": 0.00026008024072216653, + "loss": 1.3074, + "theoretical_loss": 3.496052617565165, + "tokens_seen": 1601044480 + }, + { + "epoch": 0.49, + "learning_rate": 0.00026007021063189566, + "loss": 1.3408, + "theoretical_loss": 3.496040157654207, + "tokens_seen": 1601110016 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002600601805416249, + "loss": 1.3137, + "theoretical_loss": 3.4960276983960368, + "tokens_seen": 1601175552 + }, + { + "epoch": 0.49, + "learning_rate": 0.00026005015045135407, + "loss": 1.4129, + "theoretical_loss": 3.4960152397905935, + "tokens_seen": 1601241088 + }, + { + "epoch": 0.49, + "learning_rate": 0.00026004012036108325, + "loss": 1.3872, + "theoretical_loss": 3.4960027818378174, + "tokens_seen": 1601306624 + }, + { + "epoch": 0.49, + "learning_rate": 0.00026003009027081243, + "loss": 1.3953, + "theoretical_loss": 3.495990324537646, + "tokens_seen": 1601372160 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002600200601805416, + "loss": 1.3467, + "theoretical_loss": 3.4959778678900193, + "tokens_seen": 1601437696 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002600100300902708, + "loss": 1.3078, + "theoretical_loss": 3.4959654118948764, + "tokens_seen": 1601503232 + }, + { + "epoch": 0.49, + "learning_rate": 0.00026000000000000003, + "loss": 1.3658, + "theoretical_loss": 3.4959529565521565, + "tokens_seen": 1601568768 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025998996990972916, + "loss": 1.4735, + "theoretical_loss": 3.4959405018617984, + "tokens_seen": 1601634304 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002599799398194584, + "loss": 1.3327, + "theoretical_loss": 3.4959280478237416, + "tokens_seen": 1601699840 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002599699097291876, + "loss": 1.3103, + "theoretical_loss": 3.4959155944379248, + "tokens_seen": 1601765376 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025995987963891676, + "loss": 1.3443, + "theoretical_loss": 3.4959031417042876, + "tokens_seen": 1601830912 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025994984954864594, + "loss": 1.3794, + "theoretical_loss": 3.495890689622769, + "tokens_seen": 1601896448 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002599398194583751, + "loss": 1.4021, + "theoretical_loss": 3.4958782381933076, + "tokens_seen": 1601961984 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002599297893681043, + "loss": 1.4302, + "theoretical_loss": 3.4958657874158434, + "tokens_seen": 1602027520 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025991975927783353, + "loss": 1.352, + "theoretical_loss": 3.495853337290315, + "tokens_seen": 1602093056 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025990972918756266, + "loss": 1.3666, + "theoretical_loss": 3.4958408878166622, + "tokens_seen": 1602158592 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002598996990972919, + "loss": 1.3602, + "theoretical_loss": 3.495828438994824, + "tokens_seen": 1602224128 + }, + { + "epoch": 0.49, + "learning_rate": 0.000259889669007021, + "loss": 1.3403, + "theoretical_loss": 3.49581599082474, + "tokens_seen": 1602289664 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0003024221514351666, + "objective/train/docs_used": 907690, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0079572200775146, + "objective/train/original_loss": 3.0079572200775146, + "objective/train/theoretical_loss": 3.495803543306348, + "objective/train/tokens_used": 1622815200, + "objective/train/value_avg": -0.005443572998046875, + "objective/train/value_loss": 6.334637146210298e-05, + "objective/train/value_max": -0.00016999244689941406, + "objective/train/value_min": -0.049957275390625, + "objective/train/value_reward_corr": 0.35191653114525473, + "objective/train/value_std": 0.00400543212890625, + "objective/train/weight_avg": 0.99992835521698, + "objective/train/weighted_lm_loss": 3.0110580921173096, + "objective/train/weights_max": 1.498305320739746, + "objective/train/weights_min": 0.3160642981529236, + "theoretical_loss": 3.495803543306348, + "tokens_seen": 1602355200 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025987963891675026, + "loss": 1.4473, + "theoretical_loss": 3.495803543306348, + "tokens_seen": 1602355200 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025986960882647944, + "loss": 1.4071, + "theoretical_loss": 3.4957910964395884, + "tokens_seen": 1602420736 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002598595787362086, + "loss": 1.3369, + "theoretical_loss": 3.4957786502244, + "tokens_seen": 1602486272 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002598495486459378, + "loss": 1.2671, + "theoretical_loss": 3.4957662046607227, + "tokens_seen": 1602551808 + }, + { + "epoch": 0.49, + "learning_rate": 0.000259839518555667, + "loss": 1.3144, + "theoretical_loss": 3.495753759748496, + "tokens_seen": 1602617344 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025982948846539616, + "loss": 1.4687, + "theoretical_loss": 3.495741315487657, + "tokens_seen": 1602682880 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002598194583751254, + "loss": 1.308, + "theoretical_loss": 3.495728871878147, + "tokens_seen": 1602748416 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002598094282848545, + "loss": 1.3795, + "theoretical_loss": 3.495716428919905, + "tokens_seen": 1602813952 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025979939819458376, + "loss": 1.3894, + "theoretical_loss": 3.4957039866128703, + "tokens_seen": 1602879488 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025978936810431294, + "loss": 1.3435, + "theoretical_loss": 3.4956915449569808, + "tokens_seen": 1602945024 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002597793380140421, + "loss": 1.392, + "theoretical_loss": 3.495679103952178, + "tokens_seen": 1603010560 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002597693079237713, + "loss": 1.2292, + "theoretical_loss": 3.4956666635983993, + "tokens_seen": 1603076096 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002597592778335005, + "loss": 1.43, + "theoretical_loss": 3.4956542238955857, + "tokens_seen": 1603141632 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002597492477432297, + "loss": 1.3794, + "theoretical_loss": 3.495641784843675, + "tokens_seen": 1603207168 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002597392176529589, + "loss": 1.3312, + "theoretical_loss": 3.495629346442607, + "tokens_seen": 1603272704 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002597291875626881, + "loss": 1.3252, + "theoretical_loss": 3.495616908692322, + "tokens_seen": 1603338240 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025971915747241727, + "loss": 1.3133, + "theoretical_loss": 3.495604471592758, + "tokens_seen": 1603403776 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025970912738214645, + "loss": 1.4942, + "theoretical_loss": 3.495592035143855, + "tokens_seen": 1603469312 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025969909729187563, + "loss": 1.3767, + "theoretical_loss": 3.495579599345553, + "tokens_seen": 1603534848 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025968906720160486, + "loss": 1.2586, + "theoretical_loss": 3.49556716419779, + "tokens_seen": 1603600384 + }, + { + "epoch": 0.49, + "learning_rate": 0.000259679037111334, + "loss": 1.3299, + "theoretical_loss": 3.495554729700506, + "tokens_seen": 1603665920 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002596690070210632, + "loss": 1.2752, + "theoretical_loss": 3.4955422958536406, + "tokens_seen": 1603731456 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025965897693079235, + "loss": 1.2543, + "theoretical_loss": 3.4955298626571336, + "tokens_seen": 1603796992 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002596489468405216, + "loss": 1.3857, + "theoretical_loss": 3.4955174301109233, + "tokens_seen": 1603862528 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025963891675025077, + "loss": 1.3988, + "theoretical_loss": 3.49550499821495, + "tokens_seen": 1603928064 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0012940892484039068, + "objective/train/docs_used": 908320, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6188666820526123, + "objective/train/original_loss": 2.6188666820526123, + "objective/train/theoretical_loss": 3.495492566969153, + "objective/train/tokens_used": 1624453600, + "objective/train/value_avg": -0.00598907470703125, + "objective/train/value_loss": 8.223343320423737e-05, + "objective/train/value_max": -0.00020992755889892578, + "objective/train/value_min": -0.06793212890625, + "objective/train/value_reward_corr": 0.37651396977778584, + "objective/train/value_std": 0.0048828125, + "objective/train/weight_avg": 0.9909438490867615, + "objective/train/weighted_lm_loss": 2.605348825454712, + "objective/train/weights_max": 1.9249639511108398, + "objective/train/weights_min": 0.3383985459804535, + "theoretical_loss": 3.495492566969153, + "tokens_seen": 1603993600 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025962888665997995, + "loss": 1.3536, + "theoretical_loss": 3.495492566969153, + "tokens_seen": 1603993600 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025961885656970913, + "loss": 1.4023, + "theoretical_loss": 3.4954801363734713, + "tokens_seen": 1604059136 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025960882647943837, + "loss": 1.2306, + "theoretical_loss": 3.4954677064278448, + "tokens_seen": 1604124672 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002595987963891675, + "loss": 1.2331, + "theoretical_loss": 3.495455277132213, + "tokens_seen": 1604190208 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025958876629889673, + "loss": 1.3902, + "theoretical_loss": 3.4954428484865145, + "tokens_seen": 1604255744 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025957873620862586, + "loss": 1.3643, + "theoretical_loss": 3.4954304204906896, + "tokens_seen": 1604321280 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002595687061183551, + "loss": 1.413, + "theoretical_loss": 3.4954179931446774, + "tokens_seen": 1604386816 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025955867602808427, + "loss": 1.3363, + "theoretical_loss": 3.495405566448418, + "tokens_seen": 1604452352 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025954864593781345, + "loss": 1.3577, + "theoretical_loss": 3.49539314040185, + "tokens_seen": 1604517888 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025953861584754263, + "loss": 1.2997, + "theoretical_loss": 3.495380715004914, + "tokens_seen": 1604583424 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002595285857572718, + "loss": 1.4141, + "theoretical_loss": 3.495368290257548, + "tokens_seen": 1604648960 + }, + { + "epoch": 0.49, + "learning_rate": 0.000259518555667001, + "loss": 1.2503, + "theoretical_loss": 3.4953558661596924, + "tokens_seen": 1604714496 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025950852557673023, + "loss": 1.3519, + "theoretical_loss": 3.4953434427112873, + "tokens_seen": 1604780032 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025949849548645936, + "loss": 1.4141, + "theoretical_loss": 3.495331019912271, + "tokens_seen": 1604845568 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002594884653961886, + "loss": 1.3269, + "theoretical_loss": 3.495318597762584, + "tokens_seen": 1604911104 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002594784353059178, + "loss": 1.3368, + "theoretical_loss": 3.4953061762621656, + "tokens_seen": 1604976640 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025946840521564696, + "loss": 1.3024, + "theoretical_loss": 3.495293755410955, + "tokens_seen": 1605042176 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025945837512537614, + "loss": 1.3491, + "theoretical_loss": 3.495281335208892, + "tokens_seen": 1605107712 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002594483450351053, + "loss": 1.4073, + "theoretical_loss": 3.4952689156559162, + "tokens_seen": 1605173248 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002594383149448345, + "loss": 1.3391, + "theoretical_loss": 3.495256496751967, + "tokens_seen": 1605238784 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025942828485456373, + "loss": 1.3012, + "theoretical_loss": 3.4952440784969845, + "tokens_seen": 1605304320 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025941825476429286, + "loss": 1.284, + "theoretical_loss": 3.4952316608909078, + "tokens_seen": 1605369856 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002594082246740221, + "loss": 1.3464, + "theoretical_loss": 3.4952192439336764, + "tokens_seen": 1605435392 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002593981945837512, + "loss": 1.438, + "theoretical_loss": 3.4952068276252306, + "tokens_seen": 1605500928 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025938816449348046, + "loss": 1.3754, + "theoretical_loss": 3.4951944119655094, + "tokens_seen": 1605566464 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0009922620374709368, + "objective/train/docs_used": 909776, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4513187408447266, + "objective/train/original_loss": 2.4513182640075684, + "objective/train/theoretical_loss": 3.495181996954453, + "objective/train/tokens_used": 1626092000, + "objective/train/value_avg": -0.005504608154296875, + "objective/train/value_loss": 8.938580867834389e-05, + "objective/train/value_max": -0.0004222393035888672, + "objective/train/value_min": -0.0634765625, + "objective/train/value_reward_corr": 0.34469621629866576, + "objective/train/value_std": 0.00409698486328125, + "objective/train/weight_avg": 0.9939494729042053, + "objective/train/weighted_lm_loss": 2.4361133575439453, + "objective/train/weights_max": 1.6895689964294434, + "objective/train/weights_min": 0.14466723799705505, + "theoretical_loss": 3.495181996954453, + "tokens_seen": 1605632000 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025937813440320964, + "loss": 1.3535, + "theoretical_loss": 3.495181996954453, + "tokens_seen": 1605632000 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002593681043129388, + "loss": 1.388, + "theoretical_loss": 3.495169582592, + "tokens_seen": 1605697536 + }, + { + "epoch": 0.49, + "learning_rate": 0.000259358074222668, + "loss": 1.3764, + "theoretical_loss": 3.495157168878091, + "tokens_seen": 1605763072 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002593480441323972, + "loss": 1.318, + "theoretical_loss": 3.4951447558126647, + "tokens_seen": 1605828608 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025933801404212636, + "loss": 1.3513, + "theoretical_loss": 3.495132343395662, + "tokens_seen": 1605894144 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002593279839518556, + "loss": 1.3099, + "theoretical_loss": 3.495119931627022, + "tokens_seen": 1605959680 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002593179538615847, + "loss": 1.2896, + "theoretical_loss": 3.4951075205066844, + "tokens_seen": 1606025216 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025930792377131396, + "loss": 1.3656, + "theoretical_loss": 3.4950951100345886, + "tokens_seen": 1606090752 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025929789368104314, + "loss": 1.3423, + "theoretical_loss": 3.4950827002106744, + "tokens_seen": 1606156288 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002592878635907723, + "loss": 1.2952, + "theoretical_loss": 3.495070291034882, + "tokens_seen": 1606221824 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002592778335005015, + "loss": 1.3748, + "theoretical_loss": 3.495057882507151, + "tokens_seen": 1606287360 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002592678034102307, + "loss": 1.4044, + "theoretical_loss": 3.49504547462742, + "tokens_seen": 1606352896 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025925777331995987, + "loss": 1.3558, + "theoretical_loss": 3.49503306739563, + "tokens_seen": 1606418432 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002592477432296891, + "loss": 1.3462, + "theoretical_loss": 3.4950206608117202, + "tokens_seen": 1606483968 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025923771313941823, + "loss": 1.436, + "theoretical_loss": 3.4950082548756307, + "tokens_seen": 1606549504 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025922768304914747, + "loss": 1.4045, + "theoretical_loss": 3.494995849587301, + "tokens_seen": 1606615040 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002592176529588766, + "loss": 1.3183, + "theoretical_loss": 3.4949834449466706, + "tokens_seen": 1606680576 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025920762286860583, + "loss": 1.2782, + "theoretical_loss": 3.4949710409536796, + "tokens_seen": 1606746112 + }, + { + "epoch": 0.49, + "learning_rate": 0.000259197592778335, + "loss": 1.3422, + "theoretical_loss": 3.4949586376082675, + "tokens_seen": 1606811648 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002591875626880642, + "loss": 1.2332, + "theoretical_loss": 3.4949462349103744, + "tokens_seen": 1606877184 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025917753259779337, + "loss": 1.3905, + "theoretical_loss": 3.49493383285994, + "tokens_seen": 1606942720 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025916750250752255, + "loss": 1.4484, + "theoretical_loss": 3.4949214314569046, + "tokens_seen": 1607008256 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025915747241725173, + "loss": 1.287, + "theoretical_loss": 3.4949090307012067, + "tokens_seen": 1607073792 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025914744232698097, + "loss": 1.2503, + "theoretical_loss": 3.494896630592787, + "tokens_seen": 1607139328 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002591374122367101, + "loss": 1.5059, + "theoretical_loss": 3.4948842311315853, + "tokens_seen": 1607204864 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0009816508973017335, + "objective/train/docs_used": 910493, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.604266881942749, + "objective/train/original_loss": 2.604266881942749, + "objective/train/theoretical_loss": 3.4948718323175414, + "objective/train/tokens_used": 1627730400, + "objective/train/value_avg": -0.005191802978515625, + "objective/train/value_loss": 6.113043491495773e-05, + "objective/train/value_max": -0.0004935264587402344, + "objective/train/value_min": -0.047607421875, + "objective/train/value_reward_corr": 0.31158484357761196, + "objective/train/value_std": 0.003490447998046875, + "objective/train/weight_avg": 0.9930211901664734, + "objective/train/weighted_lm_loss": 2.5852596759796143, + "objective/train/weights_max": 1.5685226917266846, + "objective/train/weights_min": 0.29981037974357605, + "theoretical_loss": 3.4948718323175414, + "tokens_seen": 1607270400 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025912738214643933, + "loss": 1.2595, + "theoretical_loss": 3.4948718323175414, + "tokens_seen": 1607270400 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002591173520561685, + "loss": 1.3052, + "theoretical_loss": 3.4948594341505954, + "tokens_seen": 1607335936 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002591073219658977, + "loss": 1.4177, + "theoretical_loss": 3.494847036630686, + "tokens_seen": 1607401472 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002590972918756269, + "loss": 1.3292, + "theoretical_loss": 3.4948346397577543, + "tokens_seen": 1607467008 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025908726178535606, + "loss": 1.3479, + "theoretical_loss": 3.4948222435317398, + "tokens_seen": 1607532544 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025907723169508524, + "loss": 1.3586, + "theoretical_loss": 3.4948098479525824, + "tokens_seen": 1607598080 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025906720160481447, + "loss": 1.3541, + "theoretical_loss": 3.4947974530202215, + "tokens_seen": 1607663616 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002590571715145436, + "loss": 1.3931, + "theoretical_loss": 3.494785058734598, + "tokens_seen": 1607729152 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025904714142427283, + "loss": 1.3498, + "theoretical_loss": 3.4947726650956508, + "tokens_seen": 1607794688 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025903711133400196, + "loss": 1.3279, + "theoretical_loss": 3.4947602721033197, + "tokens_seen": 1607860224 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002590270812437312, + "loss": 1.4499, + "theoretical_loss": 3.4947478797575453, + "tokens_seen": 1607925760 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002590170511534604, + "loss": 1.2512, + "theoretical_loss": 3.4947354880582675, + "tokens_seen": 1607991296 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025900702106318956, + "loss": 1.2713, + "theoretical_loss": 3.4947230970054264, + "tokens_seen": 1608056832 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002589969909729188, + "loss": 1.2562, + "theoretical_loss": 3.4947107065989607, + "tokens_seen": 1608122368 + }, + { + "epoch": 0.49, + "learning_rate": 0.000258986960882648, + "loss": 1.339, + "theoretical_loss": 3.4946983168388117, + "tokens_seen": 1608187904 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025897693079237716, + "loss": 1.2786, + "theoretical_loss": 3.4946859277249187, + "tokens_seen": 1608253440 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025896690070210634, + "loss": 1.2827, + "theoretical_loss": 3.4946735392572217, + "tokens_seen": 1608318976 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002589568706118355, + "loss": 1.3652, + "theoretical_loss": 3.494661151435661, + "tokens_seen": 1608384512 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002589468405215647, + "loss": 1.2756, + "theoretical_loss": 3.4946487642601762, + "tokens_seen": 1608450048 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025893681043129393, + "loss": 1.3339, + "theoretical_loss": 3.4946363777307075, + "tokens_seen": 1608515584 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025892678034102306, + "loss": 1.3776, + "theoretical_loss": 3.494623991847195, + "tokens_seen": 1608581120 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002589167502507523, + "loss": 1.3553, + "theoretical_loss": 3.494611606609578, + "tokens_seen": 1608646656 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002589067201604814, + "loss": 1.3037, + "theoretical_loss": 3.494599222017797, + "tokens_seen": 1608712192 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025889669007021066, + "loss": 1.3232, + "theoretical_loss": 3.4945868380717924, + "tokens_seen": 1608777728 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025888665997993984, + "loss": 1.2994, + "theoretical_loss": 3.494574454771504, + "tokens_seen": 1608843264 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0010219078976660967, + "objective/train/docs_used": 911872, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.462357521057129, + "objective/train/original_loss": 2.462357759475708, + "objective/train/theoretical_loss": 3.4945620721168713, + "objective/train/tokens_used": 1629368800, + "objective/train/value_avg": -0.0055389404296875, + "objective/train/value_loss": 9.510491508990526e-05, + "objective/train/value_max": -0.00046181678771972656, + "objective/train/value_min": -0.053192138671875, + "objective/train/value_reward_corr": 0.2243238067675628, + "objective/train/value_std": 0.00444793701171875, + "objective/train/weight_avg": 0.9938604235649109, + "objective/train/weighted_lm_loss": 2.4607341289520264, + "objective/train/weights_max": 1.7021996974945068, + "objective/train/weights_min": 0.13862977921962738, + "theoretical_loss": 3.4945620721168713, + "tokens_seen": 1608908800 + }, + { + "epoch": 0.49, + "learning_rate": 0.000258876629889669, + "loss": 1.3486, + "theoretical_loss": 3.4945620721168713, + "tokens_seen": 1608908800 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002588665997993982, + "loss": 1.3255, + "theoretical_loss": 3.494549690107835, + "tokens_seen": 1608974336 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002588565697091274, + "loss": 1.2666, + "theoretical_loss": 3.4945373087443348, + "tokens_seen": 1609039872 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025884653961885657, + "loss": 1.3175, + "theoretical_loss": 3.4945249280263107, + "tokens_seen": 1609105408 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002588365095285858, + "loss": 1.3954, + "theoretical_loss": 3.494512547953703, + "tokens_seen": 1609170944 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025882647943831493, + "loss": 1.2303, + "theoretical_loss": 3.4945001685264514, + "tokens_seen": 1609236480 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025881644934804416, + "loss": 1.3509, + "theoretical_loss": 3.494487789744497, + "tokens_seen": 1609302016 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025880641925777334, + "loss": 1.289, + "theoretical_loss": 3.494475411607778, + "tokens_seen": 1609367552 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002587963891675025, + "loss": 1.2968, + "theoretical_loss": 3.494463034116236, + "tokens_seen": 1609433088 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002587863590772317, + "loss": 1.3029, + "theoretical_loss": 3.494450657269811, + "tokens_seen": 1609498624 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002587763289869609, + "loss": 1.2208, + "theoretical_loss": 3.494438281068443, + "tokens_seen": 1609564160 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025876629889669007, + "loss": 1.3598, + "theoretical_loss": 3.494425905512071, + "tokens_seen": 1609629696 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002587562688064193, + "loss": 1.3245, + "theoretical_loss": 3.494413530600637, + "tokens_seen": 1609695232 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025874623871614843, + "loss": 1.3425, + "theoretical_loss": 3.4944011563340798, + "tokens_seen": 1609760768 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025873620862587767, + "loss": 1.2449, + "theoretical_loss": 3.49438878271234, + "tokens_seen": 1609826304 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002587261785356068, + "loss": 1.3549, + "theoretical_loss": 3.494376409735358, + "tokens_seen": 1609891840 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025871614844533603, + "loss": 1.3113, + "theoretical_loss": 3.4943640374030736, + "tokens_seen": 1609957376 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002587061183550652, + "loss": 1.3503, + "theoretical_loss": 3.4943516657154268, + "tokens_seen": 1610022912 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002586960882647944, + "loss": 1.3226, + "theoretical_loss": 3.4943392946723577, + "tokens_seen": 1610088448 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025868605817452357, + "loss": 1.2826, + "theoretical_loss": 3.4943269242738073, + "tokens_seen": 1610153984 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025867602808425275, + "loss": 1.3853, + "theoretical_loss": 3.494314554519715, + "tokens_seen": 1610219520 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025866599799398193, + "loss": 1.456, + "theoretical_loss": 3.494302185410022, + "tokens_seen": 1610285056 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025865596790371117, + "loss": 1.335, + "theoretical_loss": 3.494289816944667, + "tokens_seen": 1610350592 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002586459378134403, + "loss": 1.2958, + "theoretical_loss": 3.4942774491235906, + "tokens_seen": 1610416128 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025863590772316953, + "loss": 1.3681, + "theoretical_loss": 3.4942650819467342, + "tokens_seen": 1610481664 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": 0.0008412501774728298, + "objective/train/docs_used": 912654, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.511878252029419, + "objective/train/original_loss": 2.51187801361084, + "objective/train/theoretical_loss": 3.494252715414037, + "objective/train/tokens_used": 1631007200, + "objective/train/value_avg": -0.00469207763671875, + "objective/train/value_loss": 3.162517896271311e-05, + "objective/train/value_max": -0.0004239082336425781, + "objective/train/value_min": -0.0604248046875, + "objective/train/value_reward_corr": 0.396010730964455, + "objective/train/value_std": 0.00351715087890625, + "objective/train/weight_avg": 1.0099703073501587, + "objective/train/weighted_lm_loss": 2.5376367568969727, + "objective/train/weights_max": 1.7262287139892578, + "objective/train/weights_min": 0.7300071120262146, + "theoretical_loss": 3.494252715414037, + "tokens_seen": 1610547200 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002586258776328987, + "loss": 1.2804, + "theoretical_loss": 3.494252715414037, + "tokens_seen": 1610547200 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002586158475426279, + "loss": 1.3369, + "theoretical_loss": 3.4942403495254393, + "tokens_seen": 1610612736 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002586058174523571, + "loss": 1.3181, + "theoretical_loss": 3.4942279842808817, + "tokens_seen": 1610678272 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025859578736208626, + "loss": 1.2366, + "theoretical_loss": 3.494215619680304, + "tokens_seen": 1610743808 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025858575727181544, + "loss": 1.3013, + "theoretical_loss": 3.4942032557236473, + "tokens_seen": 1610809344 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025857572718154467, + "loss": 1.2529, + "theoretical_loss": 3.4941908924108507, + "tokens_seen": 1610874880 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002585656970912738, + "loss": 1.4834, + "theoretical_loss": 3.4941785297418555, + "tokens_seen": 1610940416 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025855566700100303, + "loss": 1.3581, + "theoretical_loss": 3.4941661677166014, + "tokens_seen": 1611005952 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025854563691073216, + "loss": 1.2648, + "theoretical_loss": 3.494153806335029, + "tokens_seen": 1611071488 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002585356068204614, + "loss": 1.4531, + "theoretical_loss": 3.4941414455970787, + "tokens_seen": 1611137024 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002585255767301906, + "loss": 1.2441, + "theoretical_loss": 3.4941290855026903, + "tokens_seen": 1611202560 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025851554663991976, + "loss": 1.3357, + "theoretical_loss": 3.4941167260518045, + "tokens_seen": 1611268096 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025850551654964894, + "loss": 1.3834, + "theoretical_loss": 3.4941043672443612, + "tokens_seen": 1611333632 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002584954864593782, + "loss": 1.4398, + "theoretical_loss": 3.494092009080301, + "tokens_seen": 1611399168 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002584854563691073, + "loss": 1.3824, + "theoretical_loss": 3.4940796515595647, + "tokens_seen": 1611464704 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025847542627883654, + "loss": 1.3455, + "theoretical_loss": 3.4940672946820923, + "tokens_seen": 1611530240 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025846539618856566, + "loss": 1.2361, + "theoretical_loss": 3.494054938447824, + "tokens_seen": 1611595776 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002584553660982949, + "loss": 1.2622, + "theoretical_loss": 3.4940425828567, + "tokens_seen": 1611661312 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002584453360080241, + "loss": 1.3349, + "theoretical_loss": 3.4940302279086612, + "tokens_seen": 1611726848 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025843530591775326, + "loss": 1.3269, + "theoretical_loss": 3.494017873603648, + "tokens_seen": 1611792384 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025842527582748244, + "loss": 1.3904, + "theoretical_loss": 3.4940055199416005, + "tokens_seen": 1611857920 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002584152457372116, + "loss": 1.2551, + "theoretical_loss": 3.4939931669224586, + "tokens_seen": 1611923456 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002584052156469408, + "loss": 1.2605, + "theoretical_loss": 3.4939808145461635, + "tokens_seen": 1611988992 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025839518555667004, + "loss": 1.325, + "theoretical_loss": 3.4939684628126555, + "tokens_seen": 1612054528 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025838515546639917, + "loss": 1.2591, + "theoretical_loss": 3.4939561117218743, + "tokens_seen": 1612120064 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -2.41225916397525e-05, + "objective/train/docs_used": 913987, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.880720853805542, + "objective/train/original_loss": 2.880721092224121, + "objective/train/theoretical_loss": 3.493943761273761, + "objective/train/tokens_used": 1632645600, + "objective/train/value_avg": -0.005519866943359375, + "objective/train/value_loss": 5.1238257583463565e-05, + "objective/train/value_max": -8.153915405273438e-05, + "objective/train/value_min": -0.09912109375, + "objective/train/value_reward_corr": 0.39048084575069003, + "objective/train/value_std": 0.0045928955078125, + "objective/train/weight_avg": 1.0022518634796143, + "objective/train/weighted_lm_loss": 2.8840925693511963, + "objective/train/weights_max": 2.5302493572235107, + "objective/train/weights_min": 0.5003027319908142, + "theoretical_loss": 3.493943761273761, + "tokens_seen": 1612185600 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002583751253761284, + "loss": 1.3247, + "theoretical_loss": 3.493943761273761, + "tokens_seen": 1612185600 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025836509528585753, + "loss": 1.3536, + "theoretical_loss": 3.4939314114682567, + "tokens_seen": 1612251136 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025835506519558677, + "loss": 1.3055, + "theoretical_loss": 3.4939190623053005, + "tokens_seen": 1612316672 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025834503510531595, + "loss": 1.4032, + "theoretical_loss": 3.4939067137848334, + "tokens_seen": 1612382208 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025833500501504513, + "loss": 1.3562, + "theoretical_loss": 3.493894365906796, + "tokens_seen": 1612447744 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002583249749247743, + "loss": 1.212, + "theoretical_loss": 3.493882018671129, + "tokens_seen": 1612513280 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025831494483450354, + "loss": 1.2693, + "theoretical_loss": 3.4938696720777718, + "tokens_seen": 1612578816 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025830491474423267, + "loss": 1.2156, + "theoretical_loss": 3.493857326126666, + "tokens_seen": 1612644352 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002582948846539619, + "loss": 1.2544, + "theoretical_loss": 3.4938449808177516, + "tokens_seen": 1612709888 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025828485456369103, + "loss": 1.4557, + "theoretical_loss": 3.493832636150969, + "tokens_seen": 1612775424 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025827482447342027, + "loss": 1.3572, + "theoretical_loss": 3.4938202921262596, + "tokens_seen": 1612840960 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025826479438314945, + "loss": 1.2245, + "theoretical_loss": 3.4938079487435623, + "tokens_seen": 1612906496 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025825476429287863, + "loss": 1.3359, + "theoretical_loss": 3.493795606002819, + "tokens_seen": 1612972032 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025824473420260787, + "loss": 1.2617, + "theoretical_loss": 3.4937832639039703, + "tokens_seen": 1613037568 + }, + { + "epoch": 0.49, + "learning_rate": 0.000258234704112337, + "loss": 1.2554, + "theoretical_loss": 3.4937709224469558, + "tokens_seen": 1613103104 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025822467402206623, + "loss": 1.306, + "theoretical_loss": 3.4937585816317167, + "tokens_seen": 1613168640 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002582146439317954, + "loss": 1.2465, + "theoretical_loss": 3.493746241458193, + "tokens_seen": 1613234176 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002582046138415246, + "loss": 1.3468, + "theoretical_loss": 3.4937339019263254, + "tokens_seen": 1613299712 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025819458375125377, + "loss": 1.3666, + "theoretical_loss": 3.493721563036055, + "tokens_seen": 1613365248 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025818455366098295, + "loss": 1.3525, + "theoretical_loss": 3.493709224787322, + "tokens_seen": 1613430784 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025817452357071213, + "loss": 1.2271, + "theoretical_loss": 3.493696887180067, + "tokens_seen": 1613496320 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025816449348044137, + "loss": 1.3264, + "theoretical_loss": 3.4936845502142306, + "tokens_seen": 1613561856 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002581544633901705, + "loss": 1.2851, + "theoretical_loss": 3.493672213889754, + "tokens_seen": 1613627392 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025814443329989973, + "loss": 1.4192, + "theoretical_loss": 3.493659878206577, + "tokens_seen": 1613692928 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002581344032096289, + "loss": 1.2739, + "theoretical_loss": 3.4936475431646397, + "tokens_seen": 1613758464 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0004220815608277917, + "objective/train/docs_used": 914763, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.681870698928833, + "objective/train/original_loss": 2.681870698928833, + "objective/train/theoretical_loss": 3.4936352087638842, + "objective/train/tokens_used": 1634284000, + "objective/train/value_avg": -0.00601959228515625, + "objective/train/value_loss": 5.048046659794636e-05, + "objective/train/value_max": -0.0003542900085449219, + "objective/train/value_min": -0.097412109375, + "objective/train/value_reward_corr": 0.3896949864847189, + "objective/train/value_std": 0.004619598388671875, + "objective/train/weight_avg": 0.998240053653717, + "objective/train/weighted_lm_loss": 2.6892919540405273, + "objective/train/weights_max": 2.6488380432128906, + "objective/train/weights_min": 0.5072400569915771, + "theoretical_loss": 3.4936352087638842, + "tokens_seen": 1613824000 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002581243731193581, + "loss": 1.2116, + "theoretical_loss": 3.4936352087638842, + "tokens_seen": 1613824000 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002581143430290873, + "loss": 1.3326, + "theoretical_loss": 3.49362287500425, + "tokens_seen": 1613889536 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025810431293881646, + "loss": 1.1645, + "theoretical_loss": 3.493610541885679, + "tokens_seen": 1613955072 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025809428284854564, + "loss": 1.467, + "theoretical_loss": 3.4935982094081104, + "tokens_seen": 1614020608 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025808425275827487, + "loss": 1.4027, + "theoretical_loss": 3.493585877571485, + "tokens_seen": 1614086144 + }, + { + "epoch": 0.49, + "learning_rate": 0.000258074222668004, + "loss": 1.4389, + "theoretical_loss": 3.493573546375745, + "tokens_seen": 1614151680 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025806419257773323, + "loss": 1.3452, + "theoretical_loss": 3.4935612158208293, + "tokens_seen": 1614217216 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025805416248746236, + "loss": 1.3626, + "theoretical_loss": 3.49354888590668, + "tokens_seen": 1614282752 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002580441323971916, + "loss": 1.4237, + "theoretical_loss": 3.493536556633236, + "tokens_seen": 1614348288 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002580341023069208, + "loss": 1.3533, + "theoretical_loss": 3.49352422800044, + "tokens_seen": 1614413824 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025802407221664996, + "loss": 1.2073, + "theoretical_loss": 3.4935119000082318, + "tokens_seen": 1614479360 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025801404212637914, + "loss": 1.2666, + "theoretical_loss": 3.493499572656552, + "tokens_seen": 1614544896 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002580040120361084, + "loss": 1.4153, + "theoretical_loss": 3.4934872459453414, + "tokens_seen": 1614610432 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002579939819458375, + "loss": 1.2817, + "theoretical_loss": 3.4934749198745405, + "tokens_seen": 1614675968 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025798395185556674, + "loss": 1.3077, + "theoretical_loss": 3.4934625944440905, + "tokens_seen": 1614741504 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025797392176529586, + "loss": 1.4189, + "theoretical_loss": 3.4934502696539322, + "tokens_seen": 1614807040 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002579638916750251, + "loss": 1.2859, + "theoretical_loss": 3.493437945504006, + "tokens_seen": 1614872576 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002579538615847543, + "loss": 1.329, + "theoretical_loss": 3.4934256219942523, + "tokens_seen": 1614938112 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025794383149448346, + "loss": 1.276, + "theoretical_loss": 3.4934132991246125, + "tokens_seen": 1615003648 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025793380140421264, + "loss": 1.3609, + "theoretical_loss": 3.4934009768950274, + "tokens_seen": 1615069184 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002579237713139418, + "loss": 1.3806, + "theoretical_loss": 3.493388655305438, + "tokens_seen": 1615134720 + }, + { + "epoch": 0.49, + "learning_rate": 0.000257913741223671, + "loss": 1.2597, + "theoretical_loss": 3.4933763343557844, + "tokens_seen": 1615200256 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025790371113340024, + "loss": 1.3253, + "theoretical_loss": 3.4933640140460076, + "tokens_seen": 1615265792 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025789368104312937, + "loss": 1.351, + "theoretical_loss": 3.4933516943760488, + "tokens_seen": 1615331328 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002578836509528586, + "loss": 1.2833, + "theoretical_loss": 3.493339375345848, + "tokens_seen": 1615396864 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -5.176855847821571e-05, + "objective/train/docs_used": 916021, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5687711238861084, + "objective/train/original_loss": 2.5687711238861084, + "objective/train/theoretical_loss": 3.493327056955347, + "objective/train/tokens_used": 1635922400, + "objective/train/value_avg": -0.005939483642578125, + "objective/train/value_loss": 7.818730955477804e-05, + "objective/train/value_max": -0.00037407875061035156, + "objective/train/value_min": -0.08197021484375, + "objective/train/value_reward_corr": 0.578365259203918, + "objective/train/value_std": 0.006011962890625, + "objective/train/weight_avg": 1.0031170845031738, + "objective/train/weighted_lm_loss": 2.5677154064178467, + "objective/train/weights_max": 1.9708478450775146, + "objective/train/weights_min": 0.32365334033966064, + "theoretical_loss": 3.493327056955347, + "tokens_seen": 1615462400 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025787362086258773, + "loss": 1.346, + "theoretical_loss": 3.493327056955347, + "tokens_seen": 1615462400 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025786359077231697, + "loss": 1.3442, + "theoretical_loss": 3.493314739204486, + "tokens_seen": 1615527936 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025785356068204615, + "loss": 1.3745, + "theoretical_loss": 3.4933024220932056, + "tokens_seen": 1615593472 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025784353059177533, + "loss": 1.2391, + "theoretical_loss": 3.4932901056214476, + "tokens_seen": 1615659008 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002578335005015045, + "loss": 1.3458, + "theoretical_loss": 3.4932777897891523, + "tokens_seen": 1615724544 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025782347041123374, + "loss": 1.4175, + "theoretical_loss": 3.49326547459626, + "tokens_seen": 1615790080 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025781344032096287, + "loss": 1.2605, + "theoretical_loss": 3.493253160042713, + "tokens_seen": 1615855616 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002578034102306921, + "loss": 1.32, + "theoretical_loss": 3.493240846128451, + "tokens_seen": 1615921152 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025779338014042123, + "loss": 1.3067, + "theoretical_loss": 3.493228532853415, + "tokens_seen": 1615986688 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025778335005015047, + "loss": 1.3034, + "theoretical_loss": 3.4932162202175467, + "tokens_seen": 1616052224 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025777331995987965, + "loss": 1.3949, + "theoretical_loss": 3.493203908220786, + "tokens_seen": 1616117760 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025776328986960883, + "loss": 1.3571, + "theoretical_loss": 3.493191596863074, + "tokens_seen": 1616183296 + }, + { + "epoch": 0.49, + "learning_rate": 0.000257753259779338, + "loss": 1.4111, + "theoretical_loss": 3.4931792861443522, + "tokens_seen": 1616248832 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002577432296890672, + "loss": 1.4138, + "theoretical_loss": 3.493166976064561, + "tokens_seen": 1616314368 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002577331995987964, + "loss": 1.3194, + "theoretical_loss": 3.493154666623642, + "tokens_seen": 1616379904 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002577231695085256, + "loss": 1.3279, + "theoretical_loss": 3.4931423578215357, + "tokens_seen": 1616445440 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025771313941825474, + "loss": 1.2711, + "theoretical_loss": 3.4931300496581823, + "tokens_seen": 1616510976 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025770310932798397, + "loss": 1.3311, + "theoretical_loss": 3.4931177421335238, + "tokens_seen": 1616576512 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002576930792377131, + "loss": 1.3255, + "theoretical_loss": 3.493105435247501, + "tokens_seen": 1616642048 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025768304914744233, + "loss": 1.3359, + "theoretical_loss": 3.4930931290000546, + "tokens_seen": 1616707584 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002576730190571715, + "loss": 1.2898, + "theoretical_loss": 3.4930808233911255, + "tokens_seen": 1616773120 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002576629889669007, + "loss": 1.2289, + "theoretical_loss": 3.493068518420655, + "tokens_seen": 1616838656 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002576529588766299, + "loss": 1.2896, + "theoretical_loss": 3.4930562140885844, + "tokens_seen": 1616904192 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002576429287863591, + "loss": 1.4371, + "theoretical_loss": 3.493043910394854, + "tokens_seen": 1616969728 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025763289869608824, + "loss": 1.3427, + "theoretical_loss": 3.4930316073394048, + "tokens_seen": 1617035264 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": 0.000306890084175393, + "objective/train/docs_used": 916782, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.30298113822937, + "objective/train/original_loss": 2.302981376647949, + "objective/train/theoretical_loss": 3.4930193049221785, + "objective/train/tokens_used": 1637560800, + "objective/train/value_avg": -0.004863739013671875, + "objective/train/value_loss": 4.355929195298813e-05, + "objective/train/value_max": -0.0003554821014404297, + "objective/train/value_min": -0.06304931640625, + "objective/train/value_reward_corr": 0.4930771557784846, + "objective/train/value_std": 0.004283905029296875, + "objective/train/weight_avg": 1.0050997734069824, + "objective/train/weighted_lm_loss": 2.3148162364959717, + "objective/train/weights_max": 1.5285316705703735, + "objective/train/weights_min": 0.4686948359012604, + "theoretical_loss": 3.4930193049221785, + "tokens_seen": 1617100800 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002576228686058175, + "loss": 1.2259, + "theoretical_loss": 3.4930193049221785, + "tokens_seen": 1617100800 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002576128385155466, + "loss": 1.3228, + "theoretical_loss": 3.4930070031431155, + "tokens_seen": 1617166336 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025760280842527584, + "loss": 1.3445, + "theoretical_loss": 3.4929947020021572, + "tokens_seen": 1617231872 + }, + { + "epoch": 0.49, + "learning_rate": 0.000257592778335005, + "loss": 1.2457, + "theoretical_loss": 3.4929824014992445, + "tokens_seen": 1617297408 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002575827482447342, + "loss": 1.3673, + "theoretical_loss": 3.4929701016343184, + "tokens_seen": 1617362944 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002575727181544634, + "loss": 1.3897, + "theoretical_loss": 3.4929578024073202, + "tokens_seen": 1617428480 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025756268806419256, + "loss": 1.3122, + "theoretical_loss": 3.492945503818191, + "tokens_seen": 1617494016 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025755265797392174, + "loss": 1.2762, + "theoretical_loss": 3.492933205866871, + "tokens_seen": 1617559552 + }, + { + "epoch": 0.49, + "learning_rate": 0.000257542627883651, + "loss": 1.3045, + "theoretical_loss": 3.4929209085533026, + "tokens_seen": 1617625088 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002575325977933801, + "loss": 1.3362, + "theoretical_loss": 3.492908611877426, + "tokens_seen": 1617690624 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025752256770310934, + "loss": 1.3755, + "theoretical_loss": 3.492896315839183, + "tokens_seen": 1617756160 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025751253761283847, + "loss": 1.3673, + "theoretical_loss": 3.492884020438514, + "tokens_seen": 1617821696 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002575025075225677, + "loss": 1.2442, + "theoretical_loss": 3.4928717256753603, + "tokens_seen": 1617887232 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025749247743229694, + "loss": 1.3047, + "theoretical_loss": 3.492859431549663, + "tokens_seen": 1617952768 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025748244734202606, + "loss": 1.4579, + "theoretical_loss": 3.4928471380613635, + "tokens_seen": 1618018304 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002574724172517553, + "loss": 1.2601, + "theoretical_loss": 3.4928348452104028, + "tokens_seen": 1618083840 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002574623871614845, + "loss": 1.342, + "theoretical_loss": 3.492822552996722, + "tokens_seen": 1618149376 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025745235707121366, + "loss": 1.3259, + "theoretical_loss": 3.4928102614202623, + "tokens_seen": 1618214912 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025744232698094284, + "loss": 1.3517, + "theoretical_loss": 3.492797970480965, + "tokens_seen": 1618280448 + }, + { + "epoch": 0.49, + "learning_rate": 0.000257432296890672, + "loss": 1.2616, + "theoretical_loss": 3.4927856801787707, + "tokens_seen": 1618345984 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002574222668004012, + "loss": 1.2911, + "theoretical_loss": 3.492773390513621, + "tokens_seen": 1618411520 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025741223671013044, + "loss": 1.2701, + "theoretical_loss": 3.492761101485457, + "tokens_seen": 1618477056 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025740220661985957, + "loss": 1.2769, + "theoretical_loss": 3.492748813094221, + "tokens_seen": 1618542592 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002573921765295888, + "loss": 1.2106, + "theoretical_loss": 3.4927365253398523, + "tokens_seen": 1618608128 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025738214643931793, + "loss": 1.2927, + "theoretical_loss": 3.492724238222293, + "tokens_seen": 1618673664 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0007907633553259075, + "objective/train/docs_used": 918149, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.0598795413970947, + "objective/train/original_loss": 3.0598795413970947, + "objective/train/theoretical_loss": 3.4927119517414846, + "objective/train/tokens_used": 1639199200, + "objective/train/value_avg": -0.00433349609375, + "objective/train/value_loss": 5.6662662245798856e-05, + "objective/train/value_max": -0.0002269744873046875, + "objective/train/value_min": -0.07171630859375, + "objective/train/value_reward_corr": 0.4256531206992316, + "objective/train/value_std": 0.003536224365234375, + "objective/train/weight_avg": 0.9947344064712524, + "objective/train/weighted_lm_loss": 3.0499672889709473, + "objective/train/weights_max": 1.9341961145401, + "objective/train/weights_min": 0.4001772403717041, + "theoretical_loss": 3.4927119517414846, + "tokens_seen": 1618739200 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025737211634904717, + "loss": 1.4046, + "theoretical_loss": 3.4927119517414846, + "tokens_seen": 1618739200 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025736208625877635, + "loss": 1.2766, + "theoretical_loss": 3.4926996658973675, + "tokens_seen": 1618804736 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025735205616850553, + "loss": 1.3142, + "theoretical_loss": 3.492687380689884, + "tokens_seen": 1618870272 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002573420260782347, + "loss": 1.375, + "theoretical_loss": 3.4926750961189743, + "tokens_seen": 1618935808 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025733199598796394, + "loss": 1.2603, + "theoretical_loss": 3.49266281218458, + "tokens_seen": 1619001344 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025732196589769307, + "loss": 1.2755, + "theoretical_loss": 3.492650528886643, + "tokens_seen": 1619066880 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002573119358074223, + "loss": 1.3207, + "theoretical_loss": 3.4926382462251047, + "tokens_seen": 1619132416 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025730190571715143, + "loss": 1.2673, + "theoretical_loss": 3.4926259641999047, + "tokens_seen": 1619197952 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025729187562688067, + "loss": 1.1416, + "theoretical_loss": 3.4926136828109855, + "tokens_seen": 1619263488 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025728184553660985, + "loss": 1.3326, + "theoretical_loss": 3.4926014020582885, + "tokens_seen": 1619329024 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025727181544633903, + "loss": 1.2198, + "theoretical_loss": 3.492589121941754, + "tokens_seen": 1619394560 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002572617853560682, + "loss": 1.3076, + "theoretical_loss": 3.492576842461325, + "tokens_seen": 1619460096 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002572517552657974, + "loss": 1.3483, + "theoretical_loss": 3.4925645636169413, + "tokens_seen": 1619525632 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002572417251755266, + "loss": 1.3016, + "theoretical_loss": 3.4925522854085447, + "tokens_seen": 1619591168 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002572316950852558, + "loss": 1.2781, + "theoretical_loss": 3.4925400078360767, + "tokens_seen": 1619656704 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025722166499498494, + "loss": 1.3299, + "theoretical_loss": 3.492527730899478, + "tokens_seen": 1619722240 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025721163490471417, + "loss": 1.4086, + "theoretical_loss": 3.492515454598691, + "tokens_seen": 1619787776 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002572016048144433, + "loss": 1.3381, + "theoretical_loss": 3.492503178933656, + "tokens_seen": 1619853312 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025719157472417253, + "loss": 1.3632, + "theoretical_loss": 3.492490903904315, + "tokens_seen": 1619918848 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002571815446339017, + "loss": 1.2933, + "theoretical_loss": 3.4924786295106096, + "tokens_seen": 1619984384 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002571715145436309, + "loss": 1.3653, + "theoretical_loss": 3.49246635575248, + "tokens_seen": 1620049920 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002571614844533601, + "loss": 1.3142, + "theoretical_loss": 3.492454082629869, + "tokens_seen": 1620115456 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002571514543630893, + "loss": 1.2273, + "theoretical_loss": 3.4924418101427173, + "tokens_seen": 1620180992 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025714142427281844, + "loss": 1.3727, + "theoretical_loss": 3.492429538290966, + "tokens_seen": 1620246528 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002571313941825477, + "loss": 1.301, + "theoretical_loss": 3.492417267074557, + "tokens_seen": 1620312064 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": 0.0006048058858141303, + "objective/train/docs_used": 918561, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.5957751274108887, + "objective/train/original_loss": 2.5957741737365723, + "objective/train/theoretical_loss": 3.4924049964934314, + "objective/train/tokens_used": 1640837600, + "objective/train/value_avg": -0.00527191162109375, + "objective/train/value_loss": 4.911828727927059e-05, + "objective/train/value_max": -0.00037550926208496094, + "objective/train/value_min": -0.044281005859375, + "objective/train/value_reward_corr": 0.27625053278390843, + "objective/train/value_std": 0.003803253173828125, + "objective/train/weight_avg": 1.0084115266799927, + "objective/train/weighted_lm_loss": 2.6077349185943604, + "objective/train/weights_max": 1.4453437328338623, + "objective/train/weights_min": 0.414291113615036, + "theoretical_loss": 3.4924049964934314, + "tokens_seen": 1620377600 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002571213640922768, + "loss": 1.2944, + "theoretical_loss": 3.4924049964934314, + "tokens_seen": 1620377600 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025711133400200604, + "loss": 1.2988, + "theoretical_loss": 3.492392726547531, + "tokens_seen": 1620443136 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002571013039117352, + "loss": 1.3961, + "theoretical_loss": 3.492380457236797, + "tokens_seen": 1620508672 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002570912738214644, + "loss": 1.2836, + "theoretical_loss": 3.4923681885611706, + "tokens_seen": 1620574208 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002570812437311936, + "loss": 1.4762, + "theoretical_loss": 3.492355920520594, + "tokens_seen": 1620639744 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025707121364092276, + "loss": 1.2618, + "theoretical_loss": 3.4923436531150074, + "tokens_seen": 1620705280 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025706118355065194, + "loss": 1.3039, + "theoretical_loss": 3.4923313863443535, + "tokens_seen": 1620770816 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002570511534603812, + "loss": 1.2379, + "theoretical_loss": 3.492319120208573, + "tokens_seen": 1620836352 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002570411233701103, + "loss": 1.4367, + "theoretical_loss": 3.4923068547076075, + "tokens_seen": 1620901888 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025703109327983954, + "loss": 1.2805, + "theoretical_loss": 3.492294589841399, + "tokens_seen": 1620967424 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025702106318956867, + "loss": 1.3917, + "theoretical_loss": 3.4922823256098883, + "tokens_seen": 1621032960 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002570110330992979, + "loss": 1.2991, + "theoretical_loss": 3.4922700620130174, + "tokens_seen": 1621098496 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002570010030090271, + "loss": 1.2614, + "theoretical_loss": 3.4922577990507273, + "tokens_seen": 1621164032 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025699097291875626, + "loss": 1.2454, + "theoretical_loss": 3.49224553672296, + "tokens_seen": 1621229568 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025698094282848545, + "loss": 1.3564, + "theoretical_loss": 3.4922332750296565, + "tokens_seen": 1621295104 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002569709127382147, + "loss": 1.2207, + "theoretical_loss": 3.492221013970759, + "tokens_seen": 1621360640 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002569608826479438, + "loss": 1.2896, + "theoretical_loss": 3.4922087535462083, + "tokens_seen": 1621426176 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025695085255767304, + "loss": 1.3958, + "theoretical_loss": 3.492196493755947, + "tokens_seen": 1621491712 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025694082246740217, + "loss": 1.19, + "theoretical_loss": 3.4921842345999154, + "tokens_seen": 1621557248 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002569307923771314, + "loss": 1.2775, + "theoretical_loss": 3.4921719760780556, + "tokens_seen": 1621622784 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002569207622868606, + "loss": 1.3067, + "theoretical_loss": 3.49215971819031, + "tokens_seen": 1621688320 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025691073219658977, + "loss": 1.353, + "theoretical_loss": 3.4921474609366188, + "tokens_seen": 1621753856 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025690070210631895, + "loss": 1.3438, + "theoretical_loss": 3.492135204316924, + "tokens_seen": 1621819392 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025689067201604813, + "loss": 1.1699, + "theoretical_loss": 3.4921229483311675, + "tokens_seen": 1621884928 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002568806419257773, + "loss": 1.2912, + "theoretical_loss": 3.492110692979291, + "tokens_seen": 1621950464 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.000715781410690397, + "objective/train/docs_used": 919775, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.866987943649292, + "objective/train/original_loss": 2.866987943649292, + "objective/train/theoretical_loss": 3.4920984382612357, + "objective/train/tokens_used": 1642476000, + "objective/train/value_avg": -0.005405426025390625, + "objective/train/value_loss": 5.3444782679434866e-05, + "objective/train/value_max": -0.0001233816146850586, + "objective/train/value_min": -0.054901123046875, + "objective/train/value_reward_corr": 0.44454285754277423, + "objective/train/value_std": 0.004730224609375, + "objective/train/weight_avg": 0.9954188466072083, + "objective/train/weighted_lm_loss": 2.8579812049865723, + "objective/train/weights_max": 1.5193647146224976, + "objective/train/weights_min": 0.5348273515701294, + "theoretical_loss": 3.4920984382612357, + "tokens_seen": 1622016000 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025687061183550655, + "loss": 1.414, + "theoretical_loss": 3.4920984382612357, + "tokens_seen": 1622016000 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002568605817452357, + "loss": 1.281, + "theoretical_loss": 3.492086184176943, + "tokens_seen": 1622081536 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002568505516549649, + "loss": 1.2806, + "theoretical_loss": 3.492073930726355, + "tokens_seen": 1622147072 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025684052156469404, + "loss": 1.3232, + "theoretical_loss": 3.4920616779094136, + "tokens_seen": 1622212608 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025683049147442327, + "loss": 1.3122, + "theoretical_loss": 3.4920494257260604, + "tokens_seen": 1622278144 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025682046138415245, + "loss": 1.389, + "theoretical_loss": 3.492037174176236, + "tokens_seen": 1622343680 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025681043129388163, + "loss": 1.3383, + "theoretical_loss": 3.4920249232598826, + "tokens_seen": 1622409216 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002568004012036108, + "loss": 1.3053, + "theoretical_loss": 3.4920126729769425, + "tokens_seen": 1622474752 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025679037111334005, + "loss": 1.4299, + "theoretical_loss": 3.492000423327357, + "tokens_seen": 1622540288 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002567803410230692, + "loss": 1.3947, + "theoretical_loss": 3.491988174311067, + "tokens_seen": 1622605824 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002567703109327984, + "loss": 1.2178, + "theoretical_loss": 3.491975925928015, + "tokens_seen": 1622671360 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025676028084252754, + "loss": 1.1517, + "theoretical_loss": 3.491963678178143, + "tokens_seen": 1622736896 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002567502507522568, + "loss": 1.3983, + "theoretical_loss": 3.491951431061392, + "tokens_seen": 1622802432 + }, + { + "epoch": 0.49, + "learning_rate": 0.000256740220661986, + "loss": 1.2723, + "theoretical_loss": 3.491939184577704, + "tokens_seen": 1622867968 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025673019057171514, + "loss": 1.5089, + "theoretical_loss": 3.491926938727021, + "tokens_seen": 1622933504 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025672016048144437, + "loss": 1.3477, + "theoretical_loss": 3.4919146935092833, + "tokens_seen": 1622999040 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002567101303911735, + "loss": 1.4092, + "theoretical_loss": 3.4919024489244346, + "tokens_seen": 1623064576 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025670010030090273, + "loss": 1.3827, + "theoretical_loss": 3.4918902049724156, + "tokens_seen": 1623130112 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002566900702106319, + "loss": 1.3125, + "theoretical_loss": 3.491877961653168, + "tokens_seen": 1623195648 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002566800401203611, + "loss": 1.3583, + "theoretical_loss": 3.4918657189666336, + "tokens_seen": 1623261184 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002566700100300903, + "loss": 1.3563, + "theoretical_loss": 3.4918534769127545, + "tokens_seen": 1623326720 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002566599799398195, + "loss": 1.3794, + "theoretical_loss": 3.491841235491472, + "tokens_seen": 1623392256 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025664994984954864, + "loss": 1.3843, + "theoretical_loss": 3.4918289947027286, + "tokens_seen": 1623457792 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002566399197592779, + "loss": 1.4064, + "theoretical_loss": 3.4918167545464653, + "tokens_seen": 1623523328 + }, + { + "epoch": 0.49, + "learning_rate": 0.000256629889669007, + "loss": 1.332, + "theoretical_loss": 3.4918045150226242, + "tokens_seen": 1623588864 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.008000208996236324, + "objective/train/docs_used": 920322, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4651973247528076, + "objective/train/original_loss": 2.4651975631713867, + "objective/train/theoretical_loss": 3.491792276131147, + "objective/train/tokens_used": 1644114400, + "objective/train/value_avg": -0.00433349609375, + "objective/train/value_loss": 0.001325575402006507, + "objective/train/value_max": -0.00021147727966308594, + "objective/train/value_min": -0.056854248046875, + "objective/train/value_reward_corr": -0.04714585026717078, + "objective/train/value_std": 0.00357818603515625, + "objective/train/weight_avg": 0.9599878191947937, + "objective/train/weighted_lm_loss": 2.431215524673462, + "objective/train/weights_max": 1.6567341089248657, + "objective/train/weights_min": 0.11419516801834106, + "theoretical_loss": 3.491792276131147, + "tokens_seen": 1623654400 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025661985957873624, + "loss": 1.2977, + "theoretical_loss": 3.491792276131147, + "tokens_seen": 1623654400 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002566098294884654, + "loss": 1.3451, + "theoretical_loss": 3.491780037871976, + "tokens_seen": 1623719936 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002565997993981946, + "loss": 1.35, + "theoretical_loss": 3.491767800245052, + "tokens_seen": 1623785472 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002565897693079238, + "loss": 1.2859, + "theoretical_loss": 3.491755563250318, + "tokens_seen": 1623851008 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025657973921765296, + "loss": 1.295, + "theoretical_loss": 3.491743326887715, + "tokens_seen": 1623916544 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025656970912738214, + "loss": 1.3649, + "theoretical_loss": 3.4917310911571855, + "tokens_seen": 1623982080 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002565596790371114, + "loss": 1.2674, + "theoretical_loss": 3.4917188560586707, + "tokens_seen": 1624047616 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002565496489468405, + "loss": 1.4296, + "theoretical_loss": 3.4917066215921126, + "tokens_seen": 1624113152 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025653961885656974, + "loss": 1.3517, + "theoretical_loss": 3.491694387757453, + "tokens_seen": 1624178688 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025652958876629887, + "loss": 1.3055, + "theoretical_loss": 3.4916821545546344, + "tokens_seen": 1624244224 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002565195586760281, + "loss": 1.2932, + "theoretical_loss": 3.491669921983598, + "tokens_seen": 1624309760 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002565095285857573, + "loss": 1.3081, + "theoretical_loss": 3.4916576900442857, + "tokens_seen": 1624375296 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025649949849548647, + "loss": 1.3463, + "theoretical_loss": 3.49164545873664, + "tokens_seen": 1624440832 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025648946840521565, + "loss": 1.2666, + "theoretical_loss": 3.4916332280606017, + "tokens_seen": 1624506368 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002564794383149449, + "loss": 1.4059, + "theoretical_loss": 3.4916209980161144, + "tokens_seen": 1624571904 + }, + { + "epoch": 0.49, + "learning_rate": 0.000256469408224674, + "loss": 1.4082, + "theoretical_loss": 3.4916087686031183, + "tokens_seen": 1624637440 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025645937813440324, + "loss": 1.3626, + "theoretical_loss": 3.491596539821556, + "tokens_seen": 1624702976 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025644934804413237, + "loss": 1.3833, + "theoretical_loss": 3.4915843116713696, + "tokens_seen": 1624768512 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002564393179538616, + "loss": 1.3431, + "theoretical_loss": 3.4915720841525006, + "tokens_seen": 1624834048 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002564292878635908, + "loss": 1.3236, + "theoretical_loss": 3.4915598572648916, + "tokens_seen": 1624899584 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025641925777331997, + "loss": 1.3932, + "theoretical_loss": 3.4915476310084843, + "tokens_seen": 1624965120 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025640922768304915, + "loss": 1.3409, + "theoretical_loss": 3.49153540538322, + "tokens_seen": 1625030656 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025639919759277833, + "loss": 1.2391, + "theoretical_loss": 3.4915231803890423, + "tokens_seen": 1625096192 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002563891675025075, + "loss": 1.4036, + "theoretical_loss": 3.4915109560258912, + "tokens_seen": 1625161728 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025637913741223675, + "loss": 1.3083, + "theoretical_loss": 3.4914987322937097, + "tokens_seen": 1625227264 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0012240931391716003, + "objective/train/docs_used": 921030, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.766627788543701, + "objective/train/original_loss": 2.7666280269622803, + "objective/train/theoretical_loss": 3.4914865091924394, + "objective/train/tokens_used": 1645752800, + "objective/train/value_avg": -0.005916595458984375, + "objective/train/value_loss": 9.978541493183002e-05, + "objective/train/value_max": -0.00030303001403808594, + "objective/train/value_min": -0.0849609375, + "objective/train/value_reward_corr": 0.3355302441354317, + "objective/train/value_std": 0.00440216064453125, + "objective/train/weight_avg": 0.9923166632652283, + "objective/train/weighted_lm_loss": 2.7645018100738525, + "objective/train/weights_max": 1.8021658658981323, + "objective/train/weights_min": 0.2339644879102707, + "theoretical_loss": 3.4914865091924394, + "tokens_seen": 1625292800 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002563691073219659, + "loss": 1.3718, + "theoretical_loss": 3.4914865091924394, + "tokens_seen": 1625292800 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002563590772316951, + "loss": 1.3116, + "theoretical_loss": 3.491474286722023, + "tokens_seen": 1625358336 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025634904714142424, + "loss": 1.3054, + "theoretical_loss": 3.491462064882402, + "tokens_seen": 1625423872 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025633901705115347, + "loss": 1.3184, + "theoretical_loss": 3.491449843673519, + "tokens_seen": 1625489408 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025632898696088265, + "loss": 1.3312, + "theoretical_loss": 3.4914376230953144, + "tokens_seen": 1625554944 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025631895687061183, + "loss": 1.3393, + "theoretical_loss": 3.4914254031477316, + "tokens_seen": 1625620480 + }, + { + "epoch": 0.49, + "learning_rate": 0.000256308926780341, + "loss": 1.3009, + "theoretical_loss": 3.491413183830713, + "tokens_seen": 1625686016 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025629889669007025, + "loss": 1.2665, + "theoretical_loss": 3.4914009651441997, + "tokens_seen": 1625751552 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002562888665997994, + "loss": 1.2918, + "theoretical_loss": 3.491388747088134, + "tokens_seen": 1625817088 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002562788365095286, + "loss": 1.3068, + "theoretical_loss": 3.491376529662458, + "tokens_seen": 1625882624 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025626880641925774, + "loss": 1.2841, + "theoretical_loss": 3.491364312867114, + "tokens_seen": 1625948160 + }, + { + "epoch": 0.49, + "learning_rate": 0.000256258776328987, + "loss": 1.3153, + "theoretical_loss": 3.4913520967020437, + "tokens_seen": 1626013696 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025624874623871616, + "loss": 1.296, + "theoretical_loss": 3.4913398811671894, + "tokens_seen": 1626079232 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025623871614844534, + "loss": 1.3348, + "theoretical_loss": 3.4913276662624932, + "tokens_seen": 1626144768 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002562286860581745, + "loss": 1.257, + "theoretical_loss": 3.491315451987897, + "tokens_seen": 1626210304 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002562186559679037, + "loss": 1.2686, + "theoretical_loss": 3.4913032383433427, + "tokens_seen": 1626275840 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002562086258776329, + "loss": 1.1543, + "theoretical_loss": 3.4912910253287732, + "tokens_seen": 1626341376 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002561985957873621, + "loss": 1.3881, + "theoretical_loss": 3.49127881294413, + "tokens_seen": 1626406912 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025618856569709124, + "loss": 1.3282, + "theoretical_loss": 3.4912666011893556, + "tokens_seen": 1626472448 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002561785356068205, + "loss": 1.424, + "theoretical_loss": 3.491254390064392, + "tokens_seen": 1626537984 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025616850551654966, + "loss": 1.4358, + "theoretical_loss": 3.4912421795691815, + "tokens_seen": 1626603520 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025615847542627884, + "loss": 1.3258, + "theoretical_loss": 3.4912299697036655, + "tokens_seen": 1626669056 + }, + { + "epoch": 0.49, + "learning_rate": 0.000256148445336008, + "loss": 1.3444, + "theoretical_loss": 3.4912177604677868, + "tokens_seen": 1626734592 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002561384152457372, + "loss": 1.3443, + "theoretical_loss": 3.491205551861488, + "tokens_seen": 1626800128 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002561283851554664, + "loss": 1.2729, + "theoretical_loss": 3.49119334388471, + "tokens_seen": 1626865664 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0002625184424687177, + "objective/train/docs_used": 922470, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.865013360977173, + "objective/train/original_loss": 2.865013599395752, + "objective/train/theoretical_loss": 3.4911811365373957, + "objective/train/tokens_used": 1647391200, + "objective/train/value_avg": -0.005481719970703125, + "objective/train/value_loss": 7.589413144160062e-05, + "objective/train/value_max": -0.0005154609680175781, + "objective/train/value_min": -0.053802490234375, + "objective/train/value_reward_corr": 0.4506628140154924, + "objective/train/value_std": 0.00469207763671875, + "objective/train/weight_avg": 1.0007046461105347, + "objective/train/weighted_lm_loss": 2.874375104904175, + "objective/train/weights_max": 1.4955990314483643, + "objective/train/weights_min": 0.08643288165330887, + "theoretical_loss": 3.4911811365373957, + "tokens_seen": 1626931200 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002561183550651956, + "loss": 1.3871, + "theoretical_loss": 3.4911811365373957, + "tokens_seen": 1626931200 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025610832497492475, + "loss": 1.3161, + "theoretical_loss": 3.491168929819488, + "tokens_seen": 1626996736 + }, + { + "epoch": 0.49, + "learning_rate": 0.000256098294884654, + "loss": 1.2873, + "theoretical_loss": 3.491156723730928, + "tokens_seen": 1627062272 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002560882647943831, + "loss": 1.2829, + "theoretical_loss": 3.4911445182716583, + "tokens_seen": 1627127808 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025607823470411234, + "loss": 1.3168, + "theoretical_loss": 3.4911323134416214, + "tokens_seen": 1627193344 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002560682046138415, + "loss": 1.3661, + "theoretical_loss": 3.491120109240759, + "tokens_seen": 1627258880 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002560581745235707, + "loss": 1.3405, + "theoretical_loss": 3.4911079056690135, + "tokens_seen": 1627324416 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002560481444332999, + "loss": 1.3257, + "theoretical_loss": 3.4910957027263274, + "tokens_seen": 1627389952 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025603811434302907, + "loss": 1.3139, + "theoretical_loss": 3.4910835004126426, + "tokens_seen": 1627455488 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025602808425275825, + "loss": 1.2582, + "theoretical_loss": 3.491071298727902, + "tokens_seen": 1627521024 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002560180541624875, + "loss": 1.2924, + "theoretical_loss": 3.4910590976720464, + "tokens_seen": 1627586560 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002560080240722166, + "loss": 1.2484, + "theoretical_loss": 3.4910468972450195, + "tokens_seen": 1627652096 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025599799398194585, + "loss": 1.3004, + "theoretical_loss": 3.4910346974467634, + "tokens_seen": 1627717632 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002559879638916751, + "loss": 1.3796, + "theoretical_loss": 3.4910224982772196, + "tokens_seen": 1627783168 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002559779338014042, + "loss": 1.2806, + "theoretical_loss": 3.491010299736331, + "tokens_seen": 1627848704 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025596790371113344, + "loss": 1.4435, + "theoretical_loss": 3.49099810182404, + "tokens_seen": 1627914240 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025595787362086257, + "loss": 1.2809, + "theoretical_loss": 3.4909859045402887, + "tokens_seen": 1627979776 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002559478435305918, + "loss": 1.2358, + "theoretical_loss": 3.4909737078850194, + "tokens_seen": 1628045312 + }, + { + "epoch": 0.49, + "learning_rate": 0.000255937813440321, + "loss": 1.3405, + "theoretical_loss": 3.490961511858174, + "tokens_seen": 1628110848 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025592778335005017, + "loss": 1.2902, + "theoretical_loss": 3.4909493164596954, + "tokens_seen": 1628176384 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025591775325977935, + "loss": 1.2538, + "theoretical_loss": 3.4909371216895257, + "tokens_seen": 1628241920 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025590772316950853, + "loss": 1.438, + "theoretical_loss": 3.4909249275476077, + "tokens_seen": 1628307456 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002558976930792377, + "loss": 1.4388, + "theoretical_loss": 3.4909127340338832, + "tokens_seen": 1628372992 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025588766298896695, + "loss": 1.4289, + "theoretical_loss": 3.490900541148295, + "tokens_seen": 1628438528 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002558776328986961, + "loss": 1.4452, + "theoretical_loss": 3.4908883488907847, + "tokens_seen": 1628504064 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0017117458628490567, + "objective/train/docs_used": 922615, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8974857330322266, + "objective/train/original_loss": 2.8974857330322266, + "objective/train/theoretical_loss": 3.4908761572612947, + "objective/train/tokens_used": 1649029600, + "objective/train/value_avg": -0.006504058837890625, + "objective/train/value_loss": 8.371443982468918e-05, + "objective/train/value_max": -0.00044417381286621094, + "objective/train/value_min": -0.0755615234375, + "objective/train/value_reward_corr": 0.4280419108486173, + "objective/train/value_std": 0.005069732666015625, + "objective/train/weight_avg": 0.9866976141929626, + "objective/train/weighted_lm_loss": 2.87284517288208, + "objective/train/weights_max": 1.7232190370559692, + "objective/train/weights_min": 0.30791226029396057, + "theoretical_loss": 3.4908761572612947, + "tokens_seen": 1628569600 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002558676028084253, + "loss": 1.3711, + "theoretical_loss": 3.4908761572612947, + "tokens_seen": 1628569600 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025585757271815444, + "loss": 1.2418, + "theoretical_loss": 3.4908639662597687, + "tokens_seen": 1628635136 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025584754262788367, + "loss": 1.3327, + "theoretical_loss": 3.490851775886148, + "tokens_seen": 1628700672 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025583751253761285, + "loss": 1.2722, + "theoretical_loss": 3.490839586140375, + "tokens_seen": 1628766208 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025582748244734203, + "loss": 1.3611, + "theoretical_loss": 3.4908273970223926, + "tokens_seen": 1628831744 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002558174523570712, + "loss": 1.321, + "theoretical_loss": 3.4908152085321427, + "tokens_seen": 1628897280 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025580742226680045, + "loss": 1.4418, + "theoretical_loss": 3.490803020669568, + "tokens_seen": 1628962816 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002557973921765296, + "loss": 1.3844, + "theoretical_loss": 3.4907908334346107, + "tokens_seen": 1629028352 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002557873620862588, + "loss": 1.4196, + "theoretical_loss": 3.4907786468272137, + "tokens_seen": 1629093888 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025577733199598794, + "loss": 1.2157, + "theoretical_loss": 3.4907664608473192, + "tokens_seen": 1629159424 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002557673019057172, + "loss": 1.4025, + "theoretical_loss": 3.4907542754948695, + "tokens_seen": 1629224960 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025575727181544636, + "loss": 1.2883, + "theoretical_loss": 3.490742090769807, + "tokens_seen": 1629290496 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025574724172517554, + "loss": 1.2802, + "theoretical_loss": 3.4907299066720743, + "tokens_seen": 1629356032 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002557372116349047, + "loss": 1.2004, + "theoretical_loss": 3.490717723201614, + "tokens_seen": 1629421568 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002557271815446339, + "loss": 1.2482, + "theoretical_loss": 3.490705540358369, + "tokens_seen": 1629487104 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002557171514543631, + "loss": 1.3728, + "theoretical_loss": 3.4906933581422805, + "tokens_seen": 1629552640 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002557071213640923, + "loss": 1.3153, + "theoretical_loss": 3.490681176553292, + "tokens_seen": 1629618176 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025569709127382144, + "loss": 1.5125, + "theoretical_loss": 3.4906689955913457, + "tokens_seen": 1629683712 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002556870611835507, + "loss": 1.2593, + "theoretical_loss": 3.490656815256384, + "tokens_seen": 1629749248 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025567703109327986, + "loss": 1.3216, + "theoretical_loss": 3.4906446355483496, + "tokens_seen": 1629814784 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025566700100300904, + "loss": 1.4633, + "theoretical_loss": 3.4906324564671856, + "tokens_seen": 1629880320 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002556569709127382, + "loss": 1.3742, + "theoretical_loss": 3.4906202780128335, + "tokens_seen": 1629945856 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002556469408224674, + "loss": 1.2358, + "theoretical_loss": 3.490608100185236, + "tokens_seen": 1630011392 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002556369107321966, + "loss": 1.3162, + "theoretical_loss": 3.4905959229843364, + "tokens_seen": 1630076928 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002556268806419258, + "loss": 1.3822, + "theoretical_loss": 3.490583746410077, + "tokens_seen": 1630142464 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0005584089667536318, + "objective/train/docs_used": 923942, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.725911855697632, + "objective/train/original_loss": 2.725911855697632, + "objective/train/theoretical_loss": 3.490571570462399, + "objective/train/tokens_used": 1650668000, + "objective/train/value_avg": -0.005092620849609375, + "objective/train/value_loss": 5.893180059501901e-05, + "objective/train/value_max": -0.0002846717834472656, + "objective/train/value_min": -0.07763671875, + "objective/train/value_reward_corr": 0.36256235038585855, + "objective/train/value_std": 0.004161834716796875, + "objective/train/weight_avg": 0.9971864223480225, + "objective/train/weighted_lm_loss": 2.7178750038146973, + "objective/train/weights_max": 1.9530950784683228, + "objective/train/weights_min": 0.3456130027770996, + "theoretical_loss": 3.490571570462399, + "tokens_seen": 1630208000 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025561685055165495, + "loss": 1.2686, + "theoretical_loss": 3.490571570462399, + "tokens_seen": 1630208000 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002556068204613842, + "loss": 1.3452, + "theoretical_loss": 3.490559395141247, + "tokens_seen": 1630273536 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002555967903711133, + "loss": 1.3, + "theoretical_loss": 3.4905472204465626, + "tokens_seen": 1630339072 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025558676028084254, + "loss": 1.3087, + "theoretical_loss": 3.4905350463782883, + "tokens_seen": 1630404608 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002555767301905717, + "loss": 1.3533, + "theoretical_loss": 3.490522872936367, + "tokens_seen": 1630470144 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002555667001003009, + "loss": 1.3412, + "theoretical_loss": 3.4905107001207414, + "tokens_seen": 1630535680 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002555566700100301, + "loss": 1.3184, + "theoretical_loss": 3.4904985279313534, + "tokens_seen": 1630601216 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025554663991975927, + "loss": 1.4069, + "theoretical_loss": 3.4904863563681463, + "tokens_seen": 1630666752 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025553660982948845, + "loss": 1.3668, + "theoretical_loss": 3.4904741854310624, + "tokens_seen": 1630732288 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002555265797392177, + "loss": 1.4327, + "theoretical_loss": 3.4904620151200447, + "tokens_seen": 1630797824 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002555165496489468, + "loss": 1.299, + "theoretical_loss": 3.490449845435035, + "tokens_seen": 1630863360 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025550651955867605, + "loss": 1.3461, + "theoretical_loss": 3.490437676375977, + "tokens_seen": 1630928896 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025549648946840523, + "loss": 1.4288, + "theoretical_loss": 3.490425507942813, + "tokens_seen": 1630994432 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002554864593781344, + "loss": 1.3958, + "theoretical_loss": 3.4904133401354853, + "tokens_seen": 1631059968 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002554764292878636, + "loss": 1.2685, + "theoretical_loss": 3.490401172953937, + "tokens_seen": 1631125504 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025546639919759277, + "loss": 1.2277, + "theoretical_loss": 3.49038900639811, + "tokens_seen": 1631191040 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025545636910732195, + "loss": 1.4664, + "theoretical_loss": 3.490376840467948, + "tokens_seen": 1631256576 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002554463390170512, + "loss": 1.3514, + "theoretical_loss": 3.490364675163393, + "tokens_seen": 1631322112 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002554363089267803, + "loss": 1.3157, + "theoretical_loss": 3.4903525104843887, + "tokens_seen": 1631387648 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025542627883650955, + "loss": 1.3181, + "theoretical_loss": 3.4903403464308767, + "tokens_seen": 1631453184 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002554162487462387, + "loss": 1.4527, + "theoretical_loss": 3.4903281830027995, + "tokens_seen": 1631518720 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002554062186559679, + "loss": 1.3836, + "theoretical_loss": 3.490316020200101, + "tokens_seen": 1631584256 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002553961885656971, + "loss": 1.3778, + "theoretical_loss": 3.490303858022723, + "tokens_seen": 1631649792 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002553861584754263, + "loss": 1.3571, + "theoretical_loss": 3.490291696470609, + "tokens_seen": 1631715328 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025537612838515546, + "loss": 1.2927, + "theoretical_loss": 3.490279535543701, + "tokens_seen": 1631780864 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": -0.0004263700102455914, + "objective/train/docs_used": 924504, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8494069576263428, + "objective/train/original_loss": 2.8494067192077637, + "objective/train/theoretical_loss": 3.4902673752419417, + "objective/train/tokens_used": 1652306400, + "objective/train/value_avg": -0.0048828125, + "objective/train/value_loss": 4.477901529753581e-05, + "objective/train/value_max": -0.0002913475036621094, + "objective/train/value_min": -0.042083740234375, + "objective/train/value_reward_corr": 0.41142024053931453, + "objective/train/value_std": 0.003704071044921875, + "objective/train/weight_avg": 0.9978681206703186, + "objective/train/weighted_lm_loss": 2.8577756881713867, + "objective/train/weights_max": 1.3211108446121216, + "objective/train/weights_min": 0.3738744556903839, + "theoretical_loss": 3.4902673752419417, + "tokens_seen": 1631846400 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025536609829488464, + "loss": 1.4576, + "theoretical_loss": 3.4902673752419417, + "tokens_seen": 1631846400 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002553560682046138, + "loss": 1.3273, + "theoretical_loss": 3.490255215565275, + "tokens_seen": 1631911936 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025534603811434305, + "loss": 1.2325, + "theoretical_loss": 3.490243056513642, + "tokens_seen": 1631977472 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002553360080240722, + "loss": 1.3685, + "theoretical_loss": 3.4902308980869874, + "tokens_seen": 1632043008 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002553259779338014, + "loss": 1.279, + "theoretical_loss": 3.4902187402852523, + "tokens_seen": 1632108544 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002553159478435306, + "loss": 1.2954, + "theoretical_loss": 3.49020658310838, + "tokens_seen": 1632174080 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002553059177532598, + "loss": 1.3206, + "theoretical_loss": 3.4901944265563136, + "tokens_seen": 1632239616 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025529588766298896, + "loss": 1.489, + "theoretical_loss": 3.4901822706289956, + "tokens_seen": 1632305152 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025528585757271814, + "loss": 1.3016, + "theoretical_loss": 3.490170115326369, + "tokens_seen": 1632370688 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002552758274824473, + "loss": 1.3665, + "theoretical_loss": 3.490157960648377, + "tokens_seen": 1632436224 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025526579739217656, + "loss": 1.3327, + "theoretical_loss": 3.4901458065949615, + "tokens_seen": 1632501760 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002552557673019057, + "loss": 1.3256, + "theoretical_loss": 3.490133653166066, + "tokens_seen": 1632567296 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002552457372116349, + "loss": 1.4036, + "theoretical_loss": 3.4901215003616333, + "tokens_seen": 1632632832 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002552357071213641, + "loss": 1.3709, + "theoretical_loss": 3.4901093481816057, + "tokens_seen": 1632698368 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002552256770310933, + "loss": 1.3028, + "theoretical_loss": 3.4900971966259267, + "tokens_seen": 1632763904 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002552156469408225, + "loss": 1.3879, + "theoretical_loss": 3.4900850456945385, + "tokens_seen": 1632829440 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025520561685055164, + "loss": 1.3757, + "theoretical_loss": 3.490072895387385, + "tokens_seen": 1632894976 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002551955867602809, + "loss": 1.378, + "theoretical_loss": 3.490060745704408, + "tokens_seen": 1632960512 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025518555667001006, + "loss": 1.3624, + "theoretical_loss": 3.4900485966455514, + "tokens_seen": 1633026048 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025517552657973924, + "loss": 1.3572, + "theoretical_loss": 3.4900364482107573, + "tokens_seen": 1633091584 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002551654964894684, + "loss": 1.4026, + "theoretical_loss": 3.490024300399969, + "tokens_seen": 1633157120 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002551554663991976, + "loss": 1.4188, + "theoretical_loss": 3.4900121532131294, + "tokens_seen": 1633222656 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002551454363089268, + "loss": 1.42, + "theoretical_loss": 3.490000006650181, + "tokens_seen": 1633288192 + }, + { + "epoch": 0.49, + "learning_rate": 0.000255135406218656, + "loss": 1.4015, + "theoretical_loss": 3.489987860711067, + "tokens_seen": 1633353728 + }, + { + "epoch": 0.49, + "learning_rate": 0.00025512537612838515, + "loss": 1.4637, + "theoretical_loss": 3.4899757153957305, + "tokens_seen": 1633419264 + }, + { + "epoch": 0.49, + "objective/train/advantage_avg": 0.0004369155503809452, + "objective/train/docs_used": 925849, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9567370414733887, + "objective/train/original_loss": 2.9567368030548096, + "objective/train/theoretical_loss": 3.489963570704114, + "objective/train/tokens_used": 1653944800, + "objective/train/value_avg": -0.005157470703125, + "objective/train/value_loss": 5.211452298681252e-05, + "objective/train/value_max": -0.0003859996795654297, + "objective/train/value_min": -0.0546875, + "objective/train/value_reward_corr": 0.32632839345453013, + "objective/train/value_std": 0.0037689208984375, + "objective/train/weight_avg": 1.0068604946136475, + "objective/train/weighted_lm_loss": 2.9956321716308594, + "objective/train/weights_max": 1.642747163772583, + "objective/train/weights_min": 0.3230612576007843, + "theoretical_loss": 3.489963570704114, + "tokens_seen": 1633484800 + }, + { + "epoch": 0.49, + "learning_rate": 0.0002551153460381144, + "loss": 1.3733, + "theoretical_loss": 3.489963570704114, + "tokens_seen": 1633484800 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002551053159478435, + "loss": 1.3757, + "theoretical_loss": 3.489951426636161, + "tokens_seen": 1633550336 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025509528585757274, + "loss": 1.4089, + "theoretical_loss": 3.489939283191814, + "tokens_seen": 1633615872 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002550852557673019, + "loss": 1.4773, + "theoretical_loss": 3.4899271403710164, + "tokens_seen": 1633681408 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002550752256770311, + "loss": 1.3486, + "theoretical_loss": 3.4899149981737105, + "tokens_seen": 1633746944 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002550651955867603, + "loss": 1.3096, + "theoretical_loss": 3.4899028565998407, + "tokens_seen": 1633812480 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025505516549648947, + "loss": 1.3212, + "theoretical_loss": 3.4898907156493477, + "tokens_seen": 1633878016 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025504513540621865, + "loss": 1.3891, + "theoretical_loss": 3.4898785753221766, + "tokens_seen": 1633943552 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002550351053159479, + "loss": 1.2913, + "theoretical_loss": 3.4898664356182696, + "tokens_seen": 1634009088 + }, + { + "epoch": 0.5, + "learning_rate": 0.000255025075225677, + "loss": 1.4583, + "theoretical_loss": 3.489854296537569, + "tokens_seen": 1634074624 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025501504513540625, + "loss": 1.4115, + "theoretical_loss": 3.4898421580800187, + "tokens_seen": 1634140160 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025500501504513543, + "loss": 1.3425, + "theoretical_loss": 3.489830020245562, + "tokens_seen": 1634205696 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002549949849548646, + "loss": 1.2575, + "theoretical_loss": 3.4898178830341413, + "tokens_seen": 1634271232 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002549849548645938, + "loss": 1.4094, + "theoretical_loss": 3.4898057464456995, + "tokens_seen": 1634336768 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025497492477432297, + "loss": 1.4911, + "theoretical_loss": 3.4897936104801803, + "tokens_seen": 1634402304 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025496489468405215, + "loss": 1.3436, + "theoretical_loss": 3.489781475137526, + "tokens_seen": 1634467840 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002549548645937814, + "loss": 1.2925, + "theoretical_loss": 3.4897693404176806, + "tokens_seen": 1634533376 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002549448345035105, + "loss": 1.2503, + "theoretical_loss": 3.489757206320586, + "tokens_seen": 1634598912 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025493480441323975, + "loss": 1.4263, + "theoretical_loss": 3.489745072846186, + "tokens_seen": 1634664448 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002549247743229689, + "loss": 1.2551, + "theoretical_loss": 3.4897329399944237, + "tokens_seen": 1634729984 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002549147442326981, + "loss": 1.3762, + "theoretical_loss": 3.489720807765242, + "tokens_seen": 1634795520 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002549047141424273, + "loss": 1.3888, + "theoretical_loss": 3.489708676158584, + "tokens_seen": 1634861056 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002548946840521565, + "loss": 1.2671, + "theoretical_loss": 3.489696545174393, + "tokens_seen": 1634926592 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025488465396188566, + "loss": 1.3627, + "theoretical_loss": 3.489684414812612, + "tokens_seen": 1634992128 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025487462387161484, + "loss": 1.2269, + "theoretical_loss": 3.489672285073184, + "tokens_seen": 1635057664 + }, + { + "epoch": 0.5, + "objective/train/advantage_avg": -0.0006586140370927751, + "objective/train/docs_used": 926453, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.4825971126556396, + "objective/train/original_loss": 2.4825968742370605, + "objective/train/theoretical_loss": 3.4896601559560523, + "objective/train/tokens_used": 1655583200, + "objective/train/value_avg": -0.00598907470703125, + "objective/train/value_loss": 6.0568137996597216e-05, + "objective/train/value_max": -0.00046563148498535156, + "objective/train/value_min": -0.076416015625, + "objective/train/value_reward_corr": 0.4742849385767004, + "objective/train/value_std": 0.00533294677734375, + "objective/train/weight_avg": 0.9962968230247498, + "objective/train/weighted_lm_loss": 2.4819841384887695, + "objective/train/weights_max": 1.7326931953430176, + "objective/train/weights_min": 0.3498043715953827, + "theoretical_loss": 3.4896601559560523, + "tokens_seen": 1635123200 + }, + { + "epoch": 0.5, + "learning_rate": 0.000254864593781344, + "loss": 1.3799, + "theoretical_loss": 3.4896601559560523, + "tokens_seen": 1635123200 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025485456369107325, + "loss": 1.3394, + "theoretical_loss": 3.4896480274611594, + "tokens_seen": 1635188736 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002548445336008024, + "loss": 1.3063, + "theoretical_loss": 3.4896358995884498, + "tokens_seen": 1635254272 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002548345035105316, + "loss": 1.3272, + "theoretical_loss": 3.489623772337865, + "tokens_seen": 1635319808 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002548244734202608, + "loss": 1.3424, + "theoretical_loss": 3.48961164570935, + "tokens_seen": 1635385344 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025481444332999, + "loss": 1.3417, + "theoretical_loss": 3.489599519702846, + "tokens_seen": 1635450880 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025480441323971916, + "loss": 1.4247, + "theoretical_loss": 3.4895873943182973, + "tokens_seen": 1635516416 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025479438314944834, + "loss": 1.2837, + "theoretical_loss": 3.4895752695556475, + "tokens_seen": 1635581952 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002547843530591775, + "loss": 1.4854, + "theoretical_loss": 3.489563145414839, + "tokens_seen": 1635647488 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025477432296890676, + "loss": 1.2519, + "theoretical_loss": 3.4895510218958146, + "tokens_seen": 1635713024 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002547642928786359, + "loss": 1.2663, + "theoretical_loss": 3.4895388989985188, + "tokens_seen": 1635778560 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002547542627883651, + "loss": 1.3509, + "theoretical_loss": 3.4895267767228937, + "tokens_seen": 1635844096 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025474423269809425, + "loss": 1.2785, + "theoretical_loss": 3.4895146550688834, + "tokens_seen": 1635909632 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002547342026078235, + "loss": 1.4455, + "theoretical_loss": 3.48950253403643, + "tokens_seen": 1635975168 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025472417251755266, + "loss": 1.3752, + "theoretical_loss": 3.489490413625478, + "tokens_seen": 1636040704 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025471414242728184, + "loss": 1.2468, + "theoretical_loss": 3.4894782938359694, + "tokens_seen": 1636106240 + }, + { + "epoch": 0.5, + "learning_rate": 0.000254704112337011, + "loss": 1.4528, + "theoretical_loss": 3.4894661746678484, + "tokens_seen": 1636171776 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025469408224674026, + "loss": 1.4474, + "theoretical_loss": 3.489454056121058, + "tokens_seen": 1636237312 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002546840521564694, + "loss": 1.2899, + "theoretical_loss": 3.489441938195541, + "tokens_seen": 1636302848 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002546740220661986, + "loss": 1.2638, + "theoretical_loss": 3.489429820891241, + "tokens_seen": 1636368384 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025466399197592775, + "loss": 1.3106, + "theoretical_loss": 3.4894177042081016, + "tokens_seen": 1636433920 + }, + { + "epoch": 0.5, + "learning_rate": 0.000254653961885657, + "loss": 1.2701, + "theoretical_loss": 3.4894055881460657, + "tokens_seen": 1636499456 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025464393179538616, + "loss": 1.2407, + "theoretical_loss": 3.4893934727050766, + "tokens_seen": 1636564992 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025463390170511535, + "loss": 1.3484, + "theoretical_loss": 3.4893813578850783, + "tokens_seen": 1636630528 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025462387161484453, + "loss": 1.3796, + "theoretical_loss": 3.4893692436860126, + "tokens_seen": 1636696064 + }, + { + "epoch": 0.5, + "objective/train/advantage_avg": -0.0010836475994437933, + "objective/train/docs_used": 927487, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.589848756790161, + "objective/train/original_loss": 2.5898489952087402, + "objective/train/theoretical_loss": 3.4893571301078246, + "objective/train/tokens_used": 1657221600, + "objective/train/value_avg": -0.005832672119140625, + "objective/train/value_loss": 8.034129132283852e-05, + "objective/train/value_max": -0.0003077983856201172, + "objective/train/value_min": -0.07763671875, + "objective/train/value_reward_corr": 0.4323710406318021, + "objective/train/value_std": 0.00499725341796875, + "objective/train/weight_avg": 0.992844820022583, + "objective/train/weighted_lm_loss": 2.578704833984375, + "objective/train/weights_max": 1.7521611452102661, + "objective/train/weights_min": 0.2663605809211731, + "theoretical_loss": 3.4893571301078246, + "tokens_seen": 1636761600 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002546138415245737, + "loss": 1.2438, + "theoretical_loss": 3.4893571301078246, + "tokens_seen": 1636761600 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002546038114343029, + "loss": 1.311, + "theoretical_loss": 3.4893450171504563, + "tokens_seen": 1636827136 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002545937813440321, + "loss": 1.3089, + "theoretical_loss": 3.4893329048138515, + "tokens_seen": 1636892672 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025458375125376125, + "loss": 1.4496, + "theoretical_loss": 3.4893207930979533, + "tokens_seen": 1636958208 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002545737211634905, + "loss": 1.3681, + "theoretical_loss": 3.4893086820027053, + "tokens_seen": 1637023744 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002545636910732196, + "loss": 1.3622, + "theoretical_loss": 3.489296571528051, + "tokens_seen": 1637089280 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025455366098294885, + "loss": 1.3529, + "theoretical_loss": 3.4892844616739334, + "tokens_seen": 1637154816 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025454363089267803, + "loss": 1.2575, + "theoretical_loss": 3.489272352440296, + "tokens_seen": 1637220352 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002545336008024072, + "loss": 1.3913, + "theoretical_loss": 3.4892602438270823, + "tokens_seen": 1637285888 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002545235707121364, + "loss": 1.3216, + "theoretical_loss": 3.4892481358342353, + "tokens_seen": 1637351424 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025451354062186563, + "loss": 1.4349, + "theoretical_loss": 3.489236028461699, + "tokens_seen": 1637416960 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002545035105315948, + "loss": 1.3461, + "theoretical_loss": 3.4892239217094163, + "tokens_seen": 1637482496 + }, + { + "epoch": 0.5, + "learning_rate": 0.000254493480441324, + "loss": 1.396, + "theoretical_loss": 3.4892118155773306, + "tokens_seen": 1637548032 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025448345035105317, + "loss": 1.3756, + "theoretical_loss": 3.4891997100653853, + "tokens_seen": 1637613568 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025447342026078235, + "loss": 1.2719, + "theoretical_loss": 3.4891876051735244, + "tokens_seen": 1637679104 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002544633901705116, + "loss": 1.3754, + "theoretical_loss": 3.489175500901691, + "tokens_seen": 1637744640 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002544533600802407, + "loss": 1.2873, + "theoretical_loss": 3.489163397249828, + "tokens_seen": 1637810176 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025444332998996995, + "loss": 1.2767, + "theoretical_loss": 3.4891512942178795, + "tokens_seen": 1637875712 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002544332998996991, + "loss": 1.3195, + "theoretical_loss": 3.489139191805789, + "tokens_seen": 1637941248 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002544232698094283, + "loss": 1.3176, + "theoretical_loss": 3.489127090013499, + "tokens_seen": 1638006784 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002544132397191575, + "loss": 1.2392, + "theoretical_loss": 3.489114988840954, + "tokens_seen": 1638072320 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002544032096288867, + "loss": 1.2485, + "theoretical_loss": 3.4891028882880972, + "tokens_seen": 1638137856 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025439317953861586, + "loss": 1.3466, + "theoretical_loss": 3.489090788354872, + "tokens_seen": 1638203392 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025438314944834504, + "loss": 1.2984, + "theoretical_loss": 3.4890786890412215, + "tokens_seen": 1638268928 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002543731193580742, + "loss": 1.2705, + "theoretical_loss": 3.48906659034709, + "tokens_seen": 1638334464 + }, + { + "debugging/Self-BLEU-5": 0.33530039370861947, + "debugging/distinct-1-grams": 0.7587313007572996, + "debugging/distinct-2-grams": 0.9641422307481502, + "debugging/entropy-1-grams": 5.04825258661308, + "debugging/entropy-2-grams": 5.737162228400578, + "debugging/length": 527.75, + "debugging/num_segments": 4, + "debugging/raw_token_scores_avg": 0.005318289622664452, + "debugging/raw_token_scores_std": 0.008705426938831806, + "debugging/score": 0.005611861062992998, + "debugging/score_std": 0.0047581399001464944, + "epoch": 0.5, + "objective/train/advantage_avg": -0.0003810616908594966, + "objective/train/docs_used": 928998, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.411527156829834, + "objective/train/original_loss": 3.4115264415740967, + "objective/train/theoretical_loss": 3.4890544922724205, + "objective/train/tokens_used": 1658860000, + "objective/train/value_avg": -0.004940032958984375, + "objective/train/value_loss": 6.436228431994095e-05, + "objective/train/value_max": -0.00038909912109375, + "objective/train/value_min": -0.053192138671875, + "objective/train/value_reward_corr": 0.39520284693401214, + "objective/train/value_std": 0.003925323486328125, + "objective/train/weight_avg": 0.9992018342018127, + "objective/train/weighted_lm_loss": 3.4019973278045654, + "objective/train/weights_max": 1.7021996974945068, + "objective/train/weights_min": 0.3963789641857147, + "theoretical_loss": 3.4890544922724205, + "tokens_seen": 1638400000 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025436308926780345, + "loss": 1.5542, + "theoretical_loss": 3.4890544922724205, + "tokens_seen": 1638400000 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002543530591775326, + "loss": 1.4243, + "theoretical_loss": 3.4890423948171563, + "tokens_seen": 1638465536 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002543430290872618, + "loss": 1.3004, + "theoretical_loss": 3.4890302979812415, + "tokens_seen": 1638531072 + }, + { + "epoch": 0.5, + "learning_rate": 0.000254332998996991, + "loss": 1.2949, + "theoretical_loss": 3.489018201764619, + "tokens_seen": 1638596608 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002543229689067202, + "loss": 1.2432, + "theoretical_loss": 3.489006106167233, + "tokens_seen": 1638662144 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025431293881644936, + "loss": 1.2886, + "theoretical_loss": 3.4889940111890265, + "tokens_seen": 1638727680 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025430290872617854, + "loss": 1.3699, + "theoretical_loss": 3.488981916829943, + "tokens_seen": 1638793216 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002542928786359077, + "loss": 1.3957, + "theoretical_loss": 3.4889698230899264, + "tokens_seen": 1638858752 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025428284854563696, + "loss": 1.4211, + "theoretical_loss": 3.48895772996892, + "tokens_seen": 1638924288 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002542728184553661, + "loss": 1.3301, + "theoretical_loss": 3.488945637466868, + "tokens_seen": 1638989824 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002542627883650953, + "loss": 1.3449, + "theoretical_loss": 3.4889335455837127, + "tokens_seen": 1639055360 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025425275827482445, + "loss": 1.2559, + "theoretical_loss": 3.488921454319399, + "tokens_seen": 1639120896 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002542427281845537, + "loss": 1.3505, + "theoretical_loss": 3.4889093636738697, + "tokens_seen": 1639186432 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025423269809428286, + "loss": 1.2965, + "theoretical_loss": 3.4888972736470687, + "tokens_seen": 1639251968 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025422266800401204, + "loss": 1.3744, + "theoretical_loss": 3.4888851842389395, + "tokens_seen": 1639317504 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002542126379137412, + "loss": 1.3653, + "theoretical_loss": 3.488873095449426, + "tokens_seen": 1639383040 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025420260782347046, + "loss": 1.3064, + "theoretical_loss": 3.488861007278471, + "tokens_seen": 1639448576 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002541925777331996, + "loss": 1.4756, + "theoretical_loss": 3.488848919726019, + "tokens_seen": 1639514112 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002541825476429288, + "loss": 1.4492, + "theoretical_loss": 3.488836832792013, + "tokens_seen": 1639579648 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025417251755265795, + "loss": 1.355, + "theoretical_loss": 3.4888247464763973, + "tokens_seen": 1639645184 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002541624874623872, + "loss": 1.3303, + "theoretical_loss": 3.4888126607791152, + "tokens_seen": 1639710720 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025415245737211637, + "loss": 1.4529, + "theoretical_loss": 3.48880057570011, + "tokens_seen": 1639776256 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025414242728184555, + "loss": 1.3245, + "theoretical_loss": 3.488788491239326, + "tokens_seen": 1639841792 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025413239719157473, + "loss": 1.3667, + "theoretical_loss": 3.4887764073967067, + "tokens_seen": 1639907328 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002541223671013039, + "loss": 1.3232, + "theoretical_loss": 3.4887643241721955, + "tokens_seen": 1639972864 + }, + { + "epoch": 0.5, + "objective/train/advantage_avg": -0.0011950418120250106, + "objective/train/docs_used": 929588, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.83590030670166, + "objective/train/original_loss": 2.8359005451202393, + "objective/train/theoretical_loss": 3.488752241565736, + "objective/train/tokens_used": 1660498400, + "objective/train/value_avg": -0.005611419677734375, + "objective/train/value_loss": 5.8494584664003924e-05, + "objective/train/value_max": -0.0004372596740722656, + "objective/train/value_min": -0.07989501953125, + "objective/train/value_reward_corr": 0.4361465130138702, + "objective/train/value_std": 0.00446319580078125, + "objective/train/weight_avg": 0.9908848404884338, + "objective/train/weighted_lm_loss": 2.8170762062072754, + "objective/train/weights_max": 2.223205804824829, + "objective/train/weights_min": 0.6623309850692749, + "theoretical_loss": 3.488752241565736, + "tokens_seen": 1640038400 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002541123370110331, + "loss": 1.2605, + "theoretical_loss": 3.488752241565736, + "tokens_seen": 1640038400 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002541023069207623, + "loss": 1.2806, + "theoretical_loss": 3.488740159577273, + "tokens_seen": 1640103936 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025409227683049145, + "loss": 1.3901, + "theoretical_loss": 3.4887280782067487, + "tokens_seen": 1640169472 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002540822467402207, + "loss": 1.4042, + "theoretical_loss": 3.488715997454107, + "tokens_seen": 1640235008 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002540722166499498, + "loss": 1.4037, + "theoretical_loss": 3.4887039173192926, + "tokens_seen": 1640300544 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025406218655967905, + "loss": 1.4023, + "theoretical_loss": 3.488691837802249, + "tokens_seen": 1640366080 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025405215646940823, + "loss": 1.2781, + "theoretical_loss": 3.488679758902919, + "tokens_seen": 1640431616 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002540421263791374, + "loss": 1.3806, + "theoretical_loss": 3.488667680621247, + "tokens_seen": 1640497152 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002540320962888666, + "loss": 1.3889, + "theoretical_loss": 3.488655602957177, + "tokens_seen": 1640562688 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025402206619859583, + "loss": 1.3258, + "theoretical_loss": 3.4886435259106525, + "tokens_seen": 1640628224 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025401203610832495, + "loss": 1.3211, + "theoretical_loss": 3.4886314494816166, + "tokens_seen": 1640693760 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002540020060180542, + "loss": 1.3709, + "theoretical_loss": 3.488619373670014, + "tokens_seen": 1640759296 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002539919759277833, + "loss": 1.371, + "theoretical_loss": 3.488607298475788, + "tokens_seen": 1640824832 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025398194583751255, + "loss": 1.3615, + "theoretical_loss": 3.4885952238988827, + "tokens_seen": 1640890368 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025397191574724173, + "loss": 1.3659, + "theoretical_loss": 3.4885831499392417, + "tokens_seen": 1640955904 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002539618856569709, + "loss": 1.2631, + "theoretical_loss": 3.4885710765968088, + "tokens_seen": 1641021440 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002539518555667001, + "loss": 1.3935, + "theoretical_loss": 3.4885590038715275, + "tokens_seen": 1641086976 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002539418254764293, + "loss": 1.382, + "theoretical_loss": 3.4885469317633424, + "tokens_seen": 1641152512 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025393179538615846, + "loss": 1.24, + "theoretical_loss": 3.488534860272196, + "tokens_seen": 1641218048 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002539217652958877, + "loss": 1.3411, + "theoretical_loss": 3.4885227893980337, + "tokens_seen": 1641283584 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002539117352056168, + "loss": 1.2942, + "theoretical_loss": 3.4885107191407982, + "tokens_seen": 1641349120 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025390170511534606, + "loss": 1.2974, + "theoretical_loss": 3.4884986495004338, + "tokens_seen": 1641414656 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002538916750250752, + "loss": 1.3051, + "theoretical_loss": 3.4884865804768843, + "tokens_seen": 1641480192 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002538816449348044, + "loss": 1.2313, + "theoretical_loss": 3.488474512070094, + "tokens_seen": 1641545728 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002538716148445336, + "loss": 1.4328, + "theoretical_loss": 3.488462444280006, + "tokens_seen": 1641611264 + }, + { + "epoch": 0.5, + "objective/train/advantage_avg": -0.0020256279967725277, + "objective/train/docs_used": 930445, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.9512789249420166, + "objective/train/original_loss": 2.9512789249420166, + "objective/train/theoretical_loss": 3.4884503771065636, + "objective/train/tokens_used": 1662136800, + "objective/train/value_avg": -0.006031036376953125, + "objective/train/value_loss": 8.177549170795828e-05, + "objective/train/value_max": -0.0005154609680175781, + "objective/train/value_min": -0.0703125, + "objective/train/value_reward_corr": 0.5125573932820948, + "objective/train/value_std": 0.006023406982421875, + "objective/train/weight_avg": 0.9835789203643799, + "objective/train/weighted_lm_loss": 2.9199023246765137, + "objective/train/weights_max": 1.594272255897522, + "objective/train/weights_min": 0.276386559009552, + "theoretical_loss": 3.4884503771065636, + "tokens_seen": 1641676800 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002538615847542628, + "loss": 1.3764, + "theoretical_loss": 3.4884503771065636, + "tokens_seen": 1641676800 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025385155466399196, + "loss": 1.3278, + "theoretical_loss": 3.488438310549712, + "tokens_seen": 1641742336 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002538415245737212, + "loss": 1.3202, + "theoretical_loss": 3.488426244609395, + "tokens_seen": 1641807872 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002538314944834503, + "loss": 1.338, + "theoretical_loss": 3.488414179285556, + "tokens_seen": 1641873408 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025382146439317956, + "loss": 1.3728, + "theoretical_loss": 3.4884021145781388, + "tokens_seen": 1641938944 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002538114343029087, + "loss": 1.4263, + "theoretical_loss": 3.4883900504870873, + "tokens_seen": 1642004480 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002538014042126379, + "loss": 1.2991, + "theoretical_loss": 3.4883779870123455, + "tokens_seen": 1642070016 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002537913741223671, + "loss": 1.2431, + "theoretical_loss": 3.4883659241538583, + "tokens_seen": 1642135552 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002537813440320963, + "loss": 1.4154, + "theoretical_loss": 3.488353861911568, + "tokens_seen": 1642201088 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025377131394182546, + "loss": 1.3182, + "theoretical_loss": 3.4883418002854194, + "tokens_seen": 1642266624 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025376128385155465, + "loss": 1.3648, + "theoretical_loss": 3.4883297392753563, + "tokens_seen": 1642332160 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002537512537612839, + "loss": 1.4011, + "theoretical_loss": 3.4883176788813226, + "tokens_seen": 1642397696 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025374122367101306, + "loss": 1.348, + "theoretical_loss": 3.4883056191032624, + "tokens_seen": 1642463232 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025373119358074224, + "loss": 1.2788, + "theoretical_loss": 3.48829355994112, + "tokens_seen": 1642528768 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002537211634904714, + "loss": 1.323, + "theoretical_loss": 3.4882815013948383, + "tokens_seen": 1642594304 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025371113340020066, + "loss": 1.269, + "theoretical_loss": 3.488269443464363, + "tokens_seen": 1642659840 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002537011033099298, + "loss": 1.3492, + "theoretical_loss": 3.4882573861496358, + "tokens_seen": 1642725376 + }, + { + "epoch": 0.5, + "learning_rate": 0.000253691073219659, + "loss": 1.3379, + "theoretical_loss": 3.488245329450603, + "tokens_seen": 1642790912 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025368104312938815, + "loss": 1.3429, + "theoretical_loss": 3.4882332733672063, + "tokens_seen": 1642856448 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002536710130391174, + "loss": 1.3582, + "theoretical_loss": 3.4882212178993917, + "tokens_seen": 1642921984 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025366098294884657, + "loss": 1.3222, + "theoretical_loss": 3.4882091630471024, + "tokens_seen": 1642987520 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025365095285857575, + "loss": 1.411, + "theoretical_loss": 3.488197108810282, + "tokens_seen": 1643053056 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025364092276830493, + "loss": 1.4604, + "theoretical_loss": 3.488185055188876, + "tokens_seen": 1643118592 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002536308926780341, + "loss": 1.3398, + "theoretical_loss": 3.4881730021828266, + "tokens_seen": 1643184128 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002536208625877633, + "loss": 1.3056, + "theoretical_loss": 3.488160949792079, + "tokens_seen": 1643249664 + }, + { + "epoch": 0.5, + "objective/train/advantage_avg": -3.73930306523107e-05, + "objective/train/docs_used": 931444, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.6944475173950195, + "objective/train/original_loss": 2.6944477558135986, + "objective/train/theoretical_loss": 3.4881488980165765, + "objective/train/tokens_used": 1663775200, + "objective/train/value_avg": -0.00598907470703125, + "objective/train/value_loss": 4.4672851799987257e-05, + "objective/train/value_max": -0.0004172325134277344, + "objective/train/value_min": -0.0816650390625, + "objective/train/value_reward_corr": 0.41627212624033494, + "objective/train/value_std": 0.0056304931640625, + "objective/train/weight_avg": 1.0018643140792847, + "objective/train/weighted_lm_loss": 2.7042105197906494, + "objective/train/weights_max": 1.655578374862671, + "objective/train/weights_min": 0.6074255108833313, + "theoretical_loss": 3.4881488980165765, + "tokens_seen": 1643315200 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002536108324974925, + "loss": 1.4256, + "theoretical_loss": 3.4881488980165765, + "tokens_seen": 1643315200 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025360080240722165, + "loss": 1.3237, + "theoretical_loss": 3.488136846856264, + "tokens_seen": 1643380736 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002535907723169509, + "loss": 1.4218, + "theoretical_loss": 3.488124796311085, + "tokens_seen": 1643446272 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025358074222668, + "loss": 1.2189, + "theoretical_loss": 3.488112746380984, + "tokens_seen": 1643511808 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025357071213640925, + "loss": 1.3128, + "theoretical_loss": 3.4881006970659048, + "tokens_seen": 1643577344 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025356068204613843, + "loss": 1.2388, + "theoretical_loss": 3.488088648365791, + "tokens_seen": 1643642880 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002535506519558676, + "loss": 1.1614, + "theoretical_loss": 3.4880766002805874, + "tokens_seen": 1643708416 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002535406218655968, + "loss": 1.2525, + "theoretical_loss": 3.488064552810238, + "tokens_seen": 1643773952 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025353059177532603, + "loss": 1.4002, + "theoretical_loss": 3.4880525059546867, + "tokens_seen": 1643839488 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025352056168505516, + "loss": 1.2986, + "theoretical_loss": 3.4880404597138783, + "tokens_seen": 1643905024 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002535105315947844, + "loss": 1.3342, + "theoretical_loss": 3.488028414087756, + "tokens_seen": 1643970560 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002535005015045135, + "loss": 1.3777, + "theoretical_loss": 3.488016369076264, + "tokens_seen": 1644036096 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025349047141424275, + "loss": 1.1906, + "theoretical_loss": 3.4880043246793466, + "tokens_seen": 1644101632 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025348044132397193, + "loss": 1.2916, + "theoretical_loss": 3.4879922808969486, + "tokens_seen": 1644167168 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002534704112337011, + "loss": 1.4604, + "theoretical_loss": 3.487980237729013, + "tokens_seen": 1644232704 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002534603811434303, + "loss": 1.2943, + "theoretical_loss": 3.487968195175485, + "tokens_seen": 1644298240 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002534503510531595, + "loss": 1.423, + "theoretical_loss": 3.4879561532363086, + "tokens_seen": 1644363776 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025344032096288866, + "loss": 1.218, + "theoretical_loss": 3.4879441119114274, + "tokens_seen": 1644429312 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002534302908726179, + "loss": 1.3968, + "theoretical_loss": 3.4879320712007855, + "tokens_seen": 1644494848 + }, + { + "epoch": 0.5, + "learning_rate": 0.000253420260782347, + "loss": 1.4507, + "theoretical_loss": 3.487920031104328, + "tokens_seen": 1644560384 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025341023069207626, + "loss": 1.3552, + "theoretical_loss": 3.4879079916219986, + "tokens_seen": 1644625920 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002534002006018054, + "loss": 1.3141, + "theoretical_loss": 3.4878959527537416, + "tokens_seen": 1644691456 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002533901705115346, + "loss": 1.3032, + "theoretical_loss": 3.487883914499501, + "tokens_seen": 1644756992 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002533801404212638, + "loss": 1.2924, + "theoretical_loss": 3.487871876859221, + "tokens_seen": 1644822528 + }, + { + "epoch": 0.5, + "learning_rate": 0.000253370110330993, + "loss": 1.3223, + "theoretical_loss": 3.487859839832846, + "tokens_seen": 1644888064 + }, + { + "epoch": 0.5, + "objective/train/advantage_avg": -0.00023067208530846983, + "objective/train/docs_used": 932028, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 3.083209276199341, + "objective/train/original_loss": 3.0832090377807617, + "objective/train/theoretical_loss": 3.48784780342032, + "objective/train/tokens_used": 1665413600, + "objective/train/value_avg": -0.00507354736328125, + "objective/train/value_loss": 5.4458840168081224e-05, + "objective/train/value_max": -0.0004582405090332031, + "objective/train/value_min": -0.058349609375, + "objective/train/value_reward_corr": 0.3638895520833974, + "objective/train/value_std": 0.0038738250732421875, + "objective/train/weight_avg": 1.0002583265304565, + "objective/train/weighted_lm_loss": 3.091245174407959, + "objective/train/weights_max": 1.6294751167297363, + "objective/train/weights_min": 0.4445752799510956, + "theoretical_loss": 3.48784780342032, + "tokens_seen": 1644953600 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025336008024072216, + "loss": 1.346, + "theoretical_loss": 3.48784780342032, + "tokens_seen": 1644953600 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002533500501504514, + "loss": 1.3514, + "theoretical_loss": 3.487835767621588, + "tokens_seen": 1645019136 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002533400200601805, + "loss": 1.3389, + "theoretical_loss": 3.4878237324365933, + "tokens_seen": 1645084672 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025332998996990976, + "loss": 1.3885, + "theoretical_loss": 3.48781169786528, + "tokens_seen": 1645150208 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002533199598796389, + "loss": 1.2918, + "theoretical_loss": 3.4877996639075937, + "tokens_seen": 1645215744 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002533099297893681, + "loss": 1.4078, + "theoretical_loss": 3.487787630563478, + "tokens_seen": 1645281280 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002532998996990973, + "loss": 1.394, + "theoretical_loss": 3.4877755978328766, + "tokens_seen": 1645346816 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002532898696088265, + "loss": 1.2875, + "theoretical_loss": 3.487763565715734, + "tokens_seen": 1645412352 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025327983951855566, + "loss": 1.3691, + "theoretical_loss": 3.4877515342119954, + "tokens_seen": 1645477888 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025326980942828485, + "loss": 1.2034, + "theoretical_loss": 3.487739503321604, + "tokens_seen": 1645543424 + }, + { + "epoch": 0.5, + "learning_rate": 0.000253259779338014, + "loss": 1.3642, + "theoretical_loss": 3.4877274730445045, + "tokens_seen": 1645608960 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025324974924774326, + "loss": 1.398, + "theoretical_loss": 3.487715443380641, + "tokens_seen": 1645674496 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002532397191574724, + "loss": 1.381, + "theoretical_loss": 3.4877034143299586, + "tokens_seen": 1645740032 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002532296890672016, + "loss": 1.276, + "theoretical_loss": 3.4876913858924015, + "tokens_seen": 1645805568 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025321965897693075, + "loss": 1.2947, + "theoretical_loss": 3.487679358067913, + "tokens_seen": 1645871104 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025320962888666, + "loss": 1.2866, + "theoretical_loss": 3.4876673308564383, + "tokens_seen": 1645936640 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025319959879638917, + "loss": 1.2368, + "theoretical_loss": 3.4876553042579213, + "tokens_seen": 1646002176 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025318956870611835, + "loss": 1.3019, + "theoretical_loss": 3.4876432782723064, + "tokens_seen": 1646067712 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025317953861584753, + "loss": 1.3162, + "theoretical_loss": 3.487631252899538, + "tokens_seen": 1646133248 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025316950852557677, + "loss": 1.3281, + "theoretical_loss": 3.4876192281395615, + "tokens_seen": 1646198784 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002531594784353059, + "loss": 1.3694, + "theoretical_loss": 3.4876072039923196, + "tokens_seen": 1646264320 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025314944834503513, + "loss": 1.3077, + "theoretical_loss": 3.487595180457758, + "tokens_seen": 1646329856 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025313941825476425, + "loss": 1.2891, + "theoretical_loss": 3.4875831575358207, + "tokens_seen": 1646395392 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002531293881644935, + "loss": 1.3305, + "theoretical_loss": 3.4875711352264513, + "tokens_seen": 1646460928 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025311935807422267, + "loss": 1.4117, + "theoretical_loss": 3.4875591135295956, + "tokens_seen": 1646526464 + }, + { + "epoch": 0.5, + "objective/train/advantage_avg": -0.0010230493498966098, + "objective/train/docs_used": 933210, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.338439702987671, + "objective/train/original_loss": 2.338439464569092, + "objective/train/theoretical_loss": 3.487547092445197, + "objective/train/tokens_used": 1667052000, + "objective/train/value_avg": -0.005176544189453125, + "objective/train/value_loss": 6.850550562376156e-05, + "objective/train/value_max": -0.00033926963806152344, + "objective/train/value_min": -0.0843505859375, + "objective/train/value_reward_corr": 0.3144845223100113, + "objective/train/value_std": 0.004032135009765625, + "objective/train/weight_avg": 0.9929271936416626, + "objective/train/weighted_lm_loss": 2.3128280639648438, + "objective/train/weights_max": 1.6874887943267822, + "objective/train/weights_min": 0.2610293924808502, + "theoretical_loss": 3.487547092445197, + "tokens_seen": 1646592000 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025310932798395185, + "loss": 1.2844, + "theoretical_loss": 3.487547092445197, + "tokens_seen": 1646592000 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025309929789368103, + "loss": 1.2527, + "theoretical_loss": 3.4875350719732, + "tokens_seen": 1646657536 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002530892678034102, + "loss": 1.3541, + "theoretical_loss": 3.48752305211355, + "tokens_seen": 1646723072 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002530792377131394, + "loss": 1.2827, + "theoretical_loss": 3.4875110328661902, + "tokens_seen": 1646788608 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025306920762286863, + "loss": 1.2101, + "theoretical_loss": 3.4874990142310653, + "tokens_seen": 1646854144 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025305917753259776, + "loss": 1.3549, + "theoretical_loss": 3.48748699620812, + "tokens_seen": 1646919680 + }, + { + "epoch": 0.5, + "learning_rate": 0.000253049147442327, + "loss": 1.3606, + "theoretical_loss": 3.4874749787972994, + "tokens_seen": 1646985216 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002530391173520562, + "loss": 1.3868, + "theoretical_loss": 3.4874629619985473, + "tokens_seen": 1647050752 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025302908726178536, + "loss": 1.3855, + "theoretical_loss": 3.4874509458118075, + "tokens_seen": 1647116288 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025301905717151454, + "loss": 1.3073, + "theoretical_loss": 3.4874389302370252, + "tokens_seen": 1647181824 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002530090270812437, + "loss": 1.3762, + "theoretical_loss": 3.4874269152741455, + "tokens_seen": 1647247360 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025299899699097295, + "loss": 1.3265, + "theoretical_loss": 3.4874149009231123, + "tokens_seen": 1647312896 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025298896690070213, + "loss": 1.4542, + "theoretical_loss": 3.4874028871838694, + "tokens_seen": 1647378432 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002529789368104313, + "loss": 1.3554, + "theoretical_loss": 3.4873908740563624, + "tokens_seen": 1647443968 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002529689067201605, + "loss": 1.3298, + "theoretical_loss": 3.4873788615405354, + "tokens_seen": 1647509504 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002529588766298897, + "loss": 1.2596, + "theoretical_loss": 3.4873668496363326, + "tokens_seen": 1647575040 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025294884653961886, + "loss": 1.1954, + "theoretical_loss": 3.487354838343699, + "tokens_seen": 1647640576 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002529388164493481, + "loss": 1.2745, + "theoretical_loss": 3.4873428276625793, + "tokens_seen": 1647706112 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002529287863590772, + "loss": 1.3638, + "theoretical_loss": 3.487330817592918, + "tokens_seen": 1647771648 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025291875626880646, + "loss": 1.3686, + "theoretical_loss": 3.4873188081346584, + "tokens_seen": 1647837184 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002529087261785356, + "loss": 1.3491, + "theoretical_loss": 3.4873067992877465, + "tokens_seen": 1647902720 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002528986960882648, + "loss": 1.3524, + "theoretical_loss": 3.4872947910521264, + "tokens_seen": 1647968256 + }, + { + "epoch": 0.5, + "learning_rate": 0.000252888665997994, + "loss": 1.3511, + "theoretical_loss": 3.487282783427743, + "tokens_seen": 1648033792 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002528786359077232, + "loss": 1.2824, + "theoretical_loss": 3.4872707764145403, + "tokens_seen": 1648099328 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025286860581745236, + "loss": 1.3863, + "theoretical_loss": 3.4872587700124633, + "tokens_seen": 1648164864 + }, + { + "epoch": 0.5, + "objective/train/advantage_avg": 0.0005404538824222982, + "objective/train/docs_used": 933788, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.8193140029907227, + "objective/train/original_loss": 2.8193137645721436, + "objective/train/theoretical_loss": 3.4872467642214566, + "objective/train/tokens_used": 1668690400, + "objective/train/value_avg": -0.005428314208984375, + "objective/train/value_loss": 6.116455188021064e-05, + "objective/train/value_max": -0.00038909912109375, + "objective/train/value_min": -0.060638427734375, + "objective/train/value_reward_corr": 0.3007180529004477, + "objective/train/value_std": 0.004283905029296875, + "objective/train/weight_avg": 1.0082814693450928, + "objective/train/weighted_lm_loss": 2.8462514877319336, + "objective/train/weights_max": 1.6578871011734009, + "objective/train/weights_min": 0.33160167932510376, + "theoretical_loss": 3.4872467642214566, + "tokens_seen": 1648230400 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002528585757271816, + "loss": 1.3494, + "theoretical_loss": 3.4872467642214566, + "tokens_seen": 1648230400 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002528485456369107, + "loss": 1.2729, + "theoretical_loss": 3.4872347590414643, + "tokens_seen": 1648295936 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025283851554663996, + "loss": 1.2426, + "theoretical_loss": 3.4872227544724312, + "tokens_seen": 1648361472 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002528284854563691, + "loss": 1.2405, + "theoretical_loss": 3.487210750514303, + "tokens_seen": 1648427008 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002528184553660983, + "loss": 1.4153, + "theoretical_loss": 3.4871987471670227, + "tokens_seen": 1648492544 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002528084252758275, + "loss": 1.2916, + "theoretical_loss": 3.487186744430536, + "tokens_seen": 1648558080 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002527983951855567, + "loss": 1.2066, + "theoretical_loss": 3.487174742304787, + "tokens_seen": 1648623616 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025278836509528586, + "loss": 1.4051, + "theoretical_loss": 3.4871627407897203, + "tokens_seen": 1648689152 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025277833500501505, + "loss": 1.2256, + "theoretical_loss": 3.487150739885281, + "tokens_seen": 1648754688 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002527683049147442, + "loss": 1.3241, + "theoretical_loss": 3.4871387395914137, + "tokens_seen": 1648820224 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025275827482447346, + "loss": 1.2517, + "theoretical_loss": 3.4871267399080628, + "tokens_seen": 1648885760 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002527482447342026, + "loss": 1.3231, + "theoretical_loss": 3.4871147408351733, + "tokens_seen": 1648951296 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002527382146439318, + "loss": 1.3101, + "theoretical_loss": 3.4871027423726897, + "tokens_seen": 1649016832 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025272818455366095, + "loss": 1.3584, + "theoretical_loss": 3.4870907445205566, + "tokens_seen": 1649082368 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002527181544633902, + "loss": 1.3996, + "theoretical_loss": 3.4870787472787184, + "tokens_seen": 1649147904 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025270812437311937, + "loss": 1.3526, + "theoretical_loss": 3.487066750647121, + "tokens_seen": 1649213440 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025269809428284855, + "loss": 1.2732, + "theoretical_loss": 3.4870547546257074, + "tokens_seen": 1649278976 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025268806419257773, + "loss": 1.4322, + "theoretical_loss": 3.4870427592144235, + "tokens_seen": 1649344512 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025267803410230697, + "loss": 1.3269, + "theoretical_loss": 3.487030764413214, + "tokens_seen": 1649410048 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002526680040120361, + "loss": 1.3547, + "theoretical_loss": 3.487018770222023, + "tokens_seen": 1649475584 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025265797392176533, + "loss": 1.2811, + "theoretical_loss": 3.487006776640796, + "tokens_seen": 1649541120 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025264794383149445, + "loss": 1.3263, + "theoretical_loss": 3.486994783669477, + "tokens_seen": 1649606656 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002526379137412237, + "loss": 1.3116, + "theoretical_loss": 3.4869827913080114, + "tokens_seen": 1649672192 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025262788365095287, + "loss": 1.2849, + "theoretical_loss": 3.486970799556344, + "tokens_seen": 1649737728 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025261785356068205, + "loss": 1.446, + "theoretical_loss": 3.4869588084144185, + "tokens_seen": 1649803264 + }, + { + "epoch": 0.5, + "objective/train/advantage_avg": -0.0003931217361241579, + "objective/train/docs_used": 934285, + "objective/train/instantaneous_batch_size": 16, + "objective/train/instantaneous_microbatch_size": 16384, + "objective/train/lm_loss": 2.728093147277832, + "objective/train/original_loss": 2.728093147277832, + "objective/train/theoretical_loss": 3.486946817882181, + "objective/train/tokens_used": 1670328800, + "objective/train/value_avg": -0.004894256591796875, + "objective/train/value_loss": 6.701856182189658e-05, + "objective/train/value_max": -0.00017130374908447266, + "objective/train/value_min": -0.062225341796875, + "objective/train/value_reward_corr": 0.29876349552986425, + "objective/train/value_std": 0.003841400146484375, + "objective/train/weight_avg": 0.9991167187690735, + "objective/train/weighted_lm_loss": 2.7412712574005127, + "objective/train/weights_max": 1.735910415649414, + "objective/train/weights_min": 0.3734860122203827, + "theoretical_loss": 3.486946817882181, + "tokens_seen": 1649868800 + }, + { + "epoch": 0.5, + "learning_rate": 0.00025260782347041123, + "loss": 1.2832, + "theoretical_loss": 3.486946817882181, + "tokens_seen": 1649868800 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002525977933801404, + "loss": 1.4721, + "theoretical_loss": 3.4869348279595758, + "tokens_seen": 1649934336 + }, + { + "epoch": 0.5, + "learning_rate": 0.0002525877632898696, + "loss": 1.3636, + "theoretical_loss": 3.486922838646547, + "tokens_seen": 1649999872 + } + ], + "max_steps": 50354, + "num_train_epochs": 9223372036854775807, + "total_flos": 8.420619378764022e+17, + "trial_name": null, + "trial_params": null +}